106 lines
3.4 KiB
Go
106 lines
3.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/geminicli"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
|
|
"github.com/imroc/req/v3"
|
|
)
|
|
|
|
type geminiCliCodeAssistClient struct {
|
|
baseURL string
|
|
}
|
|
|
|
func NewGeminiCliCodeAssistClient() service.GeminiCliCodeAssistClient {
|
|
return &geminiCliCodeAssistClient{baseURL: geminicli.GeminiCliBaseURL}
|
|
}
|
|
|
|
func (c *geminiCliCodeAssistClient) LoadCodeAssist(ctx context.Context, accessToken, proxyURL string, reqBody *geminicli.LoadCodeAssistRequest) (*geminicli.LoadCodeAssistResponse, error) {
|
|
if reqBody == nil {
|
|
reqBody = defaultLoadCodeAssistRequest()
|
|
}
|
|
|
|
var out geminicli.LoadCodeAssistResponse
|
|
resp, err := createGeminiCliReqClient(proxyURL).R().
|
|
SetContext(ctx).
|
|
SetHeader("Authorization", "Bearer "+accessToken).
|
|
SetHeader("Content-Type", "application/json").
|
|
SetHeader("User-Agent", geminicli.GeminiCLIUserAgent).
|
|
SetBody(reqBody).
|
|
SetSuccessResult(&out).
|
|
Post(c.baseURL + "/v1internal:loadCodeAssist")
|
|
if err != nil {
|
|
fmt.Printf("[CodeAssist] LoadCodeAssist request error: %v\n", err)
|
|
return nil, fmt.Errorf("request failed: %w", err)
|
|
}
|
|
if !resp.IsSuccessState() {
|
|
body := geminicli.SanitizeBodyForLogs(resp.String())
|
|
fmt.Printf("[CodeAssist] LoadCodeAssist failed: status %d, body: %s\n", resp.StatusCode, body)
|
|
return nil, fmt.Errorf("loadCodeAssist failed: status %d, body: %s", resp.StatusCode, body)
|
|
}
|
|
fmt.Printf("[CodeAssist] LoadCodeAssist success: status %d, response: %+v\n", resp.StatusCode, out)
|
|
return &out, nil
|
|
}
|
|
|
|
func (c *geminiCliCodeAssistClient) OnboardUser(ctx context.Context, accessToken, proxyURL string, reqBody *geminicli.OnboardUserRequest) (*geminicli.OnboardUserResponse, error) {
|
|
if reqBody == nil {
|
|
reqBody = defaultOnboardUserRequest()
|
|
}
|
|
|
|
fmt.Printf("[CodeAssist] OnboardUser request body: %+v\n", reqBody)
|
|
|
|
var out geminicli.OnboardUserResponse
|
|
resp, err := createGeminiCliReqClient(proxyURL).R().
|
|
SetContext(ctx).
|
|
SetHeader("Authorization", "Bearer "+accessToken).
|
|
SetHeader("Content-Type", "application/json").
|
|
SetHeader("User-Agent", geminicli.GeminiCLIUserAgent).
|
|
SetBody(reqBody).
|
|
SetSuccessResult(&out).
|
|
Post(c.baseURL + "/v1internal:onboardUser")
|
|
if err != nil {
|
|
fmt.Printf("[CodeAssist] OnboardUser request error: %v\n", err)
|
|
return nil, fmt.Errorf("request failed: %w", err)
|
|
}
|
|
if !resp.IsSuccessState() {
|
|
body := geminicli.SanitizeBodyForLogs(resp.String())
|
|
fmt.Printf("[CodeAssist] OnboardUser failed: status %d, body: %s\n", resp.StatusCode, body)
|
|
return nil, fmt.Errorf("onboardUser failed: status %d, body: %s", resp.StatusCode, body)
|
|
}
|
|
fmt.Printf("[CodeAssist] OnboardUser success: status %d, response: %+v\n", resp.StatusCode, out)
|
|
return &out, nil
|
|
}
|
|
|
|
func createGeminiCliReqClient(proxyURL string) *req.Client {
|
|
client := req.C().SetTimeout(30 * time.Second)
|
|
if proxyURL != "" {
|
|
client.SetProxyURL(proxyURL)
|
|
}
|
|
return client
|
|
}
|
|
|
|
func defaultLoadCodeAssistRequest() *geminicli.LoadCodeAssistRequest {
|
|
return &geminicli.LoadCodeAssistRequest{
|
|
Metadata: geminicli.LoadCodeAssistMetadata{
|
|
IDEType: "ANTIGRAVITY",
|
|
Platform: "PLATFORM_UNSPECIFIED",
|
|
PluginType: "GEMINI",
|
|
},
|
|
}
|
|
}
|
|
|
|
func defaultOnboardUserRequest() *geminicli.OnboardUserRequest {
|
|
return &geminicli.OnboardUserRequest{
|
|
TierID: "LEGACY",
|
|
Metadata: geminicli.LoadCodeAssistMetadata{
|
|
IDEType: "ANTIGRAVITY",
|
|
Platform: "PLATFORM_UNSPECIFIED",
|
|
PluginType: "GEMINI",
|
|
},
|
|
}
|
|
}
|