package ccloudv2

import (
	"context"
	"net/http"

	aiv1 "github.com/confluentinc/ccloud-sdk-go-v2/ai/v1"

	"github.com/confluentinc/cli/v4/pkg/errors"
)

func newAiClient(httpClient *http.Client, url, userAgent string, unsafeTrace bool) *aiv1.APIClient {
	cfg := aiv1.NewConfiguration()
	cfg.Debug = unsafeTrace
	cfg.HTTPClient = httpClient
	cfg.Servers = aiv1.ServerConfigurations{{URL: url}}
	cfg.UserAgent = userAgent

	return aiv1.NewAPIClient(cfg)
}

func (c *Client) aiApiContext() context.Context {
	return context.WithValue(context.Background(), aiv1.ContextAccessToken, c.cfg.Context().GetAuthToken())
}

func (c *Client) GetAvailability() (aiv1.AiV1Availability, error) {
	res, httpResp, err := c.AiClient.AvailabilitiesAiV1Api.GetAiV1Availability(c.aiApiContext()).Execute()
	return res, errors.CatchCCloudV2Error(err, httpResp)
}

func (c *Client) QueryChatCompletion(req aiv1.AiV1ChatCompletionsRequest) (aiv1.AiV1ChatCompletionsReply, error) {
	res, httpResp, err := c.AiClient.ChatCompletionsAiV1Api.QueryAiV1ChatCompletion(c.aiApiContext()).AiV1ChatCompletionsRequest(req).Execute()
	return res, errors.CatchCCloudV2Error(err, httpResp)
}

func (c *Client) CreateChatCompletionFeedback(chatCompletionId string, feedback aiv1.AiV1Feedback) error {
	httpResp, err := c.AiClient.FeedbacksAiV1Api.CreateAiV1ChatCompletionFeedback(c.aiApiContext(), chatCompletionId).AiV1Feedback(feedback).Execute()
	return errors.CatchCCloudV2Error(err, httpResp)
}
