feat: AIのリソースを切り替えられるように変更

＊今はgRPCのサーバのみ実装
cocoide · mochi-yu · Sep 10, 2023 · Sep 9, 2023 · Sep 9, 2023 · Sep 9, 2023
commit d6ef59d26e9091c98e2628019f19c572415457dd
diff --git a/cmd/suggest.go b/cmd/suggest.go
@@ -7,6 +7,7 @@ import (
 	"strings"
 
 	tea "github.com/charmbracelet/bubbletea"
+	"github.com/cocoide/commitify/internal/entity"
 	"github.com/cocoide/commitify/internal/gateway"
 	"github.com/cocoide/commitify/util"
 	"github.com/fatih/color"
@@ -21,17 +22,20 @@ type model struct {
 }
 
 func (m *model) Init() tea.Cmd {
-	// 本当はこう書きたい。今は一旦全てgrpcサーバに呼び出し
-	// switch conf.endpoint {
-	// case openai:
-	// 	~~~
-	// case grpc_serve:
-	// 	~~~
-	// }
-	// var gateway gatewayInterface
+	conf, err := entity.ReadConfig()
+	if err != nil {
+		log.Fatal("設定情報の取得に失敗: ", err)
+	}
+
+	var gi gateway.GatewayInterface
+	switch conf.AISource {
+	case int(entity.WrapServer):
+		gi = gateway.NewGrpcServeGateway()
+	default:
+		gi = gateway.NewGrpcServeGateway()
+	}
 
-	gsg := gateway.NewGrpcServeGateway()
-	messages, err := gsg.FetchCommitMessages()
+	messages, err := gi.FetchCommitMessages()
 	if err != nil {
 		log.Fatal("コミットメッセージの生成に失敗: ", err)
 		os.Exit(-1)

diff --git a/internal/gateway/gateway_interface.go b/internal/gateway/gateway_interface.go
@@ -0,0 +1,5 @@
+package gateway
+
+type GatewayInterface interface {
+	FetchCommitMessages() ([]string, error)
+}