package indexer

import (
	"fmt"

	"github.com/rs/zerolog/log"

	"github.com/autobrr/autobrr/internal/domain"
	"github.com/autobrr/autobrr/pkg/btn"
	"github.com/autobrr/autobrr/pkg/ggn"
	"github.com/autobrr/autobrr/pkg/ptp"
)

type APIService interface {
	TestConnection(indexer string) (bool, error)
	GetTorrentByID(indexer string, torrentID string) (*domain.TorrentBasic, error)
	AddClient(indexer string, settings map[string]string) error
	RemoveClient(indexer string) error
}

type apiClient interface {
	GetTorrentByID(torrentID string) (*domain.TorrentBasic, error)
	TestAPI() (bool, error)
}

type apiService struct {
	apiClients map[string]apiClient
}

func NewAPIService() APIService {
	return &apiService{
		apiClients: make(map[string]apiClient),
	}
}

func (s *apiService) GetTorrentByID(indexer string, torrentID string) (*domain.TorrentBasic, error) {
	v, ok := s.apiClients[indexer]
	if !ok {
		return nil, nil
	}

	log.Trace().Str("service", "api").Str("method", "GetTorrentByID").Msgf("'%v' trying to fetch torrent from api", indexer)

	t, err := v.GetTorrentByID(torrentID)
	if err != nil {
		log.Error().Stack().Err(err).Msgf("could not get torrent: '%v' from: %v", torrentID, indexer)
		return nil, err
	}

	log.Trace().Str("service", "api").Str("method", "GetTorrentByID").Msgf("'%v' successfully fetched torrent from api: %+v", indexer, t)

	return t, nil
}

func (s *apiService) TestConnection(indexer string) (bool, error) {
	v, ok := s.apiClients[indexer]
	if !ok {
		return false, nil
	}

	t, err := v.TestAPI()
	if err != nil {
		return false, err
	}

	return t, nil
}

func (s *apiService) AddClient(indexer string, settings map[string]string) error {
	// basic validation
	if indexer == "" {
		return fmt.Errorf("api_service.add_client: validation falied: indexer can't be empty")
	} else if len(settings) == 0 {
		return fmt.Errorf("api_service.add_client: validation falied: settings can't be empty")
	}

	log.Trace().Msgf("api-service.add_client: init api client for '%v'", indexer)

	// init client
	switch indexer {
	case "btn":
		key, ok := settings["api_key"]
		if !ok || key == "" {
			return fmt.Errorf("api_service: could not initialize btn client: missing var 'api_key'")
		}
		s.apiClients[indexer] = btn.NewClient("", key)

	case "ptp":
		user, ok := settings["api_user"]
		if !ok || user == "" {
			return fmt.Errorf("api_service: could not initialize ptp client: missing var 'api_user'")
		}

		key, ok := settings["api_key"]
		if !ok || key == "" {
			return fmt.Errorf("api_service: could not initialize ptp client: missing var 'api_key'")
		}
		s.apiClients[indexer] = ptp.NewClient("", user, key)

	case "ggn":
		key, ok := settings["api_key"]
		if !ok || key == "" {
			return fmt.Errorf("api_service: could not initialize ggn client: missing var 'api_key'")
		}
		s.apiClients[indexer] = ggn.NewClient("", key)

	default:
		return fmt.Errorf("api_service: could not initialize client: unsupported indexer '%v'", indexer)

	}
	return nil
}

func (s *apiService) RemoveClient(indexer string) error {
	_, ok := s.apiClients[indexer]
	if ok {
		delete(s.apiClients, indexer)
	}

	return nil
}