Refactored the code

2025-12-18 20:54:18 +01:00 · 2025-06-20 23:16:32 +02:00
parent 7ea5713bf5
commit be38f711b7
4 changed files with 437 additions and 466 deletions
--- a/immich-souvenirs/internals/immich/immich.go
+++ b/immich-souvenirs/internals/immich/immich.go
@@ -0,0 +1,169 @@
+package immich
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"time"
+)
+
+type Album struct {
+        ID                      string          `json:"id"`
+        Name                    string          `json:"albumName"`
+        Description             string          `json:"description"`
+        Shared                  bool            `json:"shared"`
+        HasSharedLink           bool            `json:"hasSharedLink"`
+        StartDate               time.Time       `json:"startDate"`
+        CreatedAt               time.Time       `json:"createdAt"`
+        AlbumThumbnailAssetId   string  `json:"albumThumbnailAssetId"`
+}
+
+type Key struct {
+        ID              string          `json:"id"`
+        Key             string          `json:"key"`
+        Album           *Album          `json:"album"`
+}
+
+type ImmichClient struct {
+	BaseURL string
+	APIKey  string
+}
+
+// Fonction pour créer une instance d'ImmichClient
+func New(baseURL string, apiKey string) *ImmichClient {
+	return &ImmichClient{
+		BaseURL: baseURL,
+		APIKey:  apiKey,
+	}
+}
+
+// Récupérer la liste des albums
+func (ic *ImmichClient) FetchAlbums() ([]Album, error) {
+	client := &http.Client{Timeout: 10 * time.Second}
+	req, err := http.NewRequest(http.MethodGet, ic.BaseURL+"/api/albums", nil)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("x-api-key", ic.APIKey)
+	req.Header.Set("Accept", "application/json")
+
+	res, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode != 200 {
+		return nil, fmt.Errorf("Status code %d", res.StatusCode)
+	}
+
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	var albums []Album
+	if err := json.Unmarshal(body, &albums); err != nil {
+		return nil, err
+	}
+	return albums, nil
+}
+
+// Obtenir la clé de partage pour un album
+func (ic *ImmichClient) GetSharingKey(album Album) (string, error) {
+	client := &http.Client{Timeout: 10 * time.Second}
+	if album.HasSharedLink {
+		// Récupérer la clé existante
+		req, err := http.NewRequest(http.MethodGet, ic.BaseURL+"/api/shared-links", nil)
+		if err != nil {
+			return "", err
+		}
+		req.Header.Set("x-api-key", ic.APIKey)
+		req.Header.Set("Accept", "application/json")
+		res, err := client.Do(req)
+		if err != nil {
+			return "", err
+		}
+		defer res.Body.Close()
+
+		if res.StatusCode != 200 {
+			return "", fmt.Errorf("Status code %d", res.StatusCode)
+		}
+
+		body, err := ioutil.ReadAll(res.Body)
+		if err != nil {
+			return "", err
+		}
+
+		var keys []Key
+		if err := json.Unmarshal(body, &keys); err != nil {
+			return "", err
+		}
+		for _, key := range keys {
+			if key.Album != nil && key.Album.ID == album.ID {
+				return key.Key, nil
+			}
+		}
+		return "", fmt.Errorf("No sharing key found for album '%s'", album.Name)
+	} else {
+		// Créer un nouveau lien partagé
+		jsonData := []byte(`{
+			"albumId": "` + album.ID + `",
+			"allowDownload": true,
+			"allowUpload": false,
+			"showMetadata": true,
+			"type": "ALBUM"
+		}`)
+		req, err := http.NewRequest(http.MethodPost, ic.BaseURL+"/api/shared-links", bytes.NewBuffer(jsonData))
+		if err != nil {
+			return "", err
+		}
+		req.Header.Set("x-api-key", ic.APIKey)
+		req.Header.Set("Content-Type", "application/json")
+		req.Header.Set("Accept", "application/octet-stream")
+		res, err := client.Do(req)
+		if err != nil {
+			return "", err
+		}
+		defer res.Body.Close()
+
+		if res.StatusCode != 201 {
+			return "", fmt.Errorf("Status code %d", res.StatusCode)
+		}
+
+		body, err := ioutil.ReadAll(res.Body)
+		if err != nil {
+			return "", err
+		}
+
+		var key Key
+		if err := json.Unmarshal(body, &key); err != nil {
+			return "", err
+		}
+		return key.Key, nil
+	}
+}
+
+// Récupérer la miniature d'un album
+func (ic *ImmichClient) GetThumbnail(assetID string) ([]byte, error) {
+	client := &http.Client{Timeout: 10 * time.Second}
+	req, err := http.NewRequest(http.MethodGet, ic.BaseURL+"/api/assets/"+assetID+"/thumbnail?size=preview", nil)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("x-api-key", ic.APIKey)
+	req.Header.Set("Accept", "application/octet-stream")
+	res, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode != 200 {
+		return nil, fmt.Errorf("Status code %d", res.StatusCode)
+	}
+
+	return ioutil.ReadAll(res.Body)
+}
--- a/immich-souvenirs/internals/whatsapp/whatsapp.go
+++ b/immich-souvenirs/internals/whatsapp/whatsapp.go
@@ -0,0 +1,84 @@
+package whatsapp
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+        _ "github.com/mattn/go-sqlite3"
+	"github.com/mdp/qrterminal/v3"
+	"go.mau.fi/whatsmeow"
+	"go.mau.fi/whatsmeow/store/sqlstore"
+	waLog "go.mau.fi/whatsmeow/util/log"
+	"go.mau.fi/whatsmeow/types"
+	waProto "go.mau.fi/whatsmeow/binary/proto"
+	"google.golang.org/protobuf/proto"
+)
+
+type WhatsAppClient struct {
+	Client *whatsmeow.Client
+}
+
+// Fonction pour initialiser la connexion et retourner une instance de WhatsAppClient
+func New(sessionFile string) (*WhatsAppClient, error) {
+	dbLog := waLog.Stdout("Database", "ERROR", true)
+	container, err := sqlstore.New(context.Background(), "sqlite3", "file:"+sessionFile+"?_foreign_keys=on", dbLog)
+	if err != nil {
+		return nil, err
+	}
+	deviceStore, err := container.GetFirstDevice(context.Background())
+	if err != nil {
+		return nil, err
+	}
+	clientLog := waLog.Stdout("Client", "ERROR", true)
+	client := whatsmeow.NewClient(deviceStore, clientLog)
+
+	if client.Store.ID == nil {
+		// No ID stored, nouvelle connexion
+		qrChan, _ := client.GetQRChannel(context.Background())
+		err = client.Connect()
+		if err != nil {
+			return nil, err
+		}
+		for evt := range qrChan {
+			if evt.Event == "code" {
+				qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout)
+				fmt.Println("QR code:", evt.Code)
+			} else {
+				fmt.Println("Event:", evt.Event)
+			}
+		}
+	} else {
+		// Connexion existante
+		err = client.Connect()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return &WhatsAppClient{Client: client}, nil
+}
+
+// Méthode pour envoyer un message
+func (wac *WhatsAppClient) SendMessage(group string, title string, description string, message string, url string, thumbnail []byte) error {
+	jid, err := types.ParseJID(group)
+	if err != nil {
+		return fmt.Errorf("Incorrect group identifier '%s': %v", group, err)
+	}
+
+	msg := &waProto.Message{
+		ExtendedTextMessage: &waProto.ExtendedTextMessage{
+			Title:         proto.String(title),
+			Description:   proto.String(description),
+			Text:          proto.String(message),
+			MatchedText:   proto.String(url),
+			JPEGThumbnail: thumbnail,
+		},
+	}
+	ts, err := wac.Client.SendMessage(context.Background(), jid, msg)
+	if err != nil {
+		return fmt.Errorf("Error sending message with title '%s': %v", title, err)
+	}
+	fmt.Printf("Message with title '%s' sent (timestamp: %s)\n", title, ts)
+	return nil
+}