feat: コミットメッセージの生成元を追加

mochi-yu · mochi-yu · commit b9f4dbce0d3a · 2023-12-16T08:32:21.000+09:00
diff --git a/cmd/suggest.go b/cmd/suggest.go
@@ -2,14 +2,15 @@ package cmd
 
 import (
 	"fmt"
+	"log"
+	"os"
+	"strings"
+
 	"github.com/cocoide/commitify/internal/entity"
 	"github.com/cocoide/commitify/internal/gateway"
 	"github.com/cocoide/commitify/internal/service"
 	"github.com/cocoide/commitify/internal/usecase"
 	"golang.org/x/net/context"
-	"log"
-	"os"
-	"strings"
 
 	"github.com/charmbracelet/bubbles/spinner"
 	"github.com/charmbracelet/bubbles/textinput"
@@ -135,6 +136,10 @@ func NewSuggestModel() *suggestModel {
 		commitMessageService = gateway.NewClientCommitMessageGateway(nlp)
 	case entity.Server:
 		commitMessageService = gateway.NewGrpcServerGateway()
+	case entity.Qdrant:
+		commitMessageService = gateway.NewQdrantServerGateway()
+	case entity.Gemini:
+		commitMessageService = gateway.NewGeminiServerGateway()
 	}
 	suggestCmdUsecase := usecase.NewSuggestCmdUsecase(commitMessageService, github)
 
diff --git a/internal/gateway/gemini.go b/internal/gateway/gemini.go
@@ -0,0 +1,53 @@
+package gateway
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/cocoide/commitify/internal/entity"
+	"github.com/cocoide/commitify/internal/service"
+)
+
+type geminiServerGateway struct {
+	client *HttpClient
+}
+
+func NewGeminiServerGateway() service.CommitMessageService {
+	c := NewHttpClient().
+		WithBaseURL("http://suwageeks.org:5215").
+		WithPath("/gemini").
+		WithHeader("Content-Type", "application/json")
+
+	return &geminiServerGateway{client: c}
+}
+
+func (qs *geminiServerGateway) GenerateCommitMessageList(diff string, conf entity.Config) ([]string, error) {
+	type geminiBody struct {
+		Diff string `json:"diff"`
+	}
+
+	body := &geminiBody{
+		Diff: diff,
+	}
+	b, err := json.Marshal(body)
+	if err != nil {
+		return nil, err
+	}
+
+	res, err := qs.client.WithBody(b).Execute(POST)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Print(string(res))
+
+	type geminiResponse struct {
+		Messages []string `json:"messages"`
+	}
+
+	values := new(geminiResponse)
+	if err = json.Unmarshal(res, values); err != nil {
+		return nil, err
+	}
+
+	return values.Messages, nil
+}
diff --git a/internal/gateway/qdrant.go b/internal/gateway/qdrant.go
@@ -0,0 +1,54 @@
+package gateway
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/cocoide/commitify/internal/entity"
+	"github.com/cocoide/commitify/internal/service"
+)
+
+type qdrantServerGateway struct {
+	client *HttpClient
+}
+
+func NewQdrantServerGateway() service.CommitMessageService {
+	c := NewHttpClient()
+
+	return &qdrantServerGateway{client: c}
+}
+
+func (qs *qdrantServerGateway) GenerateCommitMessageList(diff string, conf entity.Config) ([]string, error) {
+	type qdrantBody struct {
+		Diff string `json:"diff"`
+	}
+
+	body := &qdrantBody{
+		Diff: diff,
+	}
+	b, err := json.Marshal(body)
+	if err != nil {
+		return nil, err
+	}
+
+	res, err := qs.client.
+		WithBaseURL("http://suwageeks.org:5215").
+		WithPath("/search").
+		WithHeader("Content-Type", "application/json").
+		WithBody(b).Execute(POST)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Print(string(res))
+
+	type qdrantResponse struct {
+		Messages []string `json:"messages"`
+	}
+
+	values := new(qdrantResponse)
+	if err = json.Unmarshal(res, values); err != nil {
+		return nil, err
+	}
+
+	return values.Messages, nil
+}
