package client

import (
	"git.db.org.ai/dborg/internal/models"
	"encoding/json"
	"fmt"
	"net/url"
)

func (c *Client) ListAccounts() (*models.AdminResponse, error) {
	data, err := c.Get("/admin/accounts", nil)
	if err != nil {
		return nil, err
	}

	var response models.AdminResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse admin response: %w", err)
	}

	return &response, nil
}

func (c *Client) CreateAccount(req *models.AccountCreateRequest) (*models.AdminResponse, error) {
	data, err := c.Post("/admin/accounts", req)
	if err != nil {
		return nil, err
	}

	var account models.Account
	if err := json.Unmarshal(data, &account); err == nil && account.APIKey != "" {
		return &models.AdminResponse{
			Success: true,
			Account: &account,
		}, nil
	}

	var response models.AdminResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse admin response: %w", err)
	}

	return &response, nil
}

func (c *Client) DeleteAccount(apiKey string) (*models.AdminResponse, error) {
	path := fmt.Sprintf("/admin/accounts/%s", url.PathEscape(apiKey))
	data, err := c.Delete(path)
	if err != nil {
		return nil, err
	}

	var response models.AdminResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse admin response: %w", err)
	}

	return &response, nil
}

func (c *Client) UpdateCredits(apiKey string, credits int) (*models.AdminResponse, error) {
	path := fmt.Sprintf("/admin/accounts/%s/credits", url.PathEscape(apiKey))
	req := &models.AddCreditsRequest{
		Credits: credits,
	}

	data, err := c.Post(path, req)
	if err != nil {
		return nil, err
	}

	var response models.AdminResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse admin response: %w", err)
	}

	return &response, nil
}

func (c *Client) SetCredits(apiKey string, credits int) (*models.AdminResponse, error) {
	path := fmt.Sprintf("/admin/accounts/%s/credits", url.PathEscape(apiKey))
	req := &models.SetCreditsRequest{
		Credits: credits,
	}

	data, err := c.Put(path, req)
	if err != nil {
		return nil, err
	}

	var account models.Account
	if err := json.Unmarshal(data, &account); err == nil && account.APIKey != "" {
		return &models.AdminResponse{
			Success: true,
			Account: &account,
		}, nil
	}

	var response models.AdminResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse admin response: %w", err)
	}

	return &response, nil
}

func (c *Client) ToggleAccount(apiKey string, enable bool) (*models.AdminResponse, error) {
	path := fmt.Sprintf("/admin/accounts/%s/disable", url.PathEscape(apiKey))
	req := &models.DisableAccountRequest{
		Disabled: !enable,
	}

	data, err := c.Patch(path, req)
	if err != nil {
		return nil, err
	}

	var response models.AdminResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse admin response: %w", err)
	}

	return &response, nil
}