feat: you can summary screen shot directly

2026-07-07 00:59:26 +08:00
parent 39f0aaae02
commit 712a2cbb6a
17 changed files with 1432 additions and 51 deletions
@@ -0,0 +1,86 @@
+package application
+
+import (
+	"context"
+	"testing"
+)
+
+type fakeOSSProvider struct {
+	key         string
+	data        []byte
+	contentType string
+	url         string
+}
+
+func (f *fakeOSSProvider) Upload(_ context.Context, key string, data []byte, contentType string) (string, error) {
+	f.key = key
+	f.data = append([]byte(nil), data...)
+	f.contentType = contentType
+	if f.url == "" {
+		return "https://cdn.example.com/" + key, nil
+	}
+	return f.url, nil
+}
+
+func (f *fakeOSSProvider) Name() string { return "fake" }
+
+type fakeSummarizer struct {
+	prompt string
+	url    string
+	text   string
+}
+
+func (f *fakeSummarizer) SummarizeImage(_ context.Context, prompt string, imageURL string) (string, error) {
+	f.prompt = prompt
+	f.url = imageURL
+	if f.text == "" {
+		return "summary text", nil
+	}
+	return f.text, nil
+}
+
+func TestCaptureSummaryServiceUploadsSummarizesAndCopies(t *testing.T) {
+	ctx := context.Background()
+	provider := &fakeOSSProvider{}
+	summarizer := &fakeSummarizer{}
+	clip := &fakeClipboard{}
+	svc := &CaptureSummaryService{
+		Provider:   provider,
+		Summarizer: summarizer,
+		Clipboard:  clip,
+		Prompt:     "describe screenshot",
+		PathPrefix: "snapgo/",
+	}
+
+	uploaded, err := svc.UploadImage(ctx, []byte("png"))
+	if err != nil {
+		t.Fatalf("upload image: %v", err)
+	}
+	if provider.contentType != "image/png" {
+		t.Fatalf("expected image/png content type, got %q", provider.contentType)
+	}
+	if string(provider.data) != "png" {
+		t.Fatalf("expected uploaded png bytes, got %q", string(provider.data))
+	}
+
+	summary, err := svc.Summarize(ctx, uploaded.URL)
+	if err != nil {
+		t.Fatalf("summarize: %v", err)
+	}
+	if summarizer.prompt != "describe screenshot" {
+		t.Fatalf("expected prompt forwarded, got %q", summarizer.prompt)
+	}
+	if summarizer.url != uploaded.URL {
+		t.Fatalf("expected uploaded url forwarded, got %q", summarizer.url)
+	}
+	if summary != "summary text" {
+		t.Fatalf("expected summary text, got %q", summary)
+	}
+
+	if err := svc.CopySummary(ctx, summary); err != nil {
+		t.Fatalf("copy summary: %v", err)
+	}
+	if clip.text != "summary text" {
+		t.Fatalf("expected summary copied, got %q", clip.text)
+	}
+}