package openai

import (
	"time"

	"github.com/go-errors/errors"
	"github.com/go-resty/resty/v2"
)

type Client struct {
	rest *resty.Client
}

func NewClient(apiKey string) *Client {
	cli := resty.New().
		SetBaseURL("https://api.openai.com").
		SetHeader("Authorization", "Bearer "+apiKey).
		SetTimeout(30 * time.Second)

	return &Client{rest: cli}
}

func (c *Client) ChatCompletion(request ChatRequest) (*ChatResponse, error) {
	resp, err := c.rest.R().
		SetBody(request).
		SetHeader("Content-Type", "application/json").
		SetResult(&ChatResponse{}).
		Post(ChatAPIPath)

	if err != nil {
		return nil, errors.Wrap(err, 0)
	}

	if resp.StatusCode() != 200 {
		return nil, errors.Errorf("unexpected status code: %d", resp.StatusCode())
	}

	return resp.Result().(*ChatResponse), nil
}