package services

import (
	"context"
	"errors"
	"strings"
	"sync"

	"code.tczkiot.com/wlw/ai-agent/contract"
)

type platformAIService struct {
	mu       sync.RWMutex
	provider contract.PlatformAIProvider
}

var PlatformAIService = &platformAIService{}

func SetPlatformAIProvider(provider contract.PlatformAIProvider) {
	PlatformAIService.mu.Lock()
	defer PlatformAIService.mu.Unlock()
	PlatformAIService.provider = provider
}

func (s *platformAIService) ModelSource(ctx context.Context) (string, error) {
	provider := s.current()
	if provider == nil {
		return contract.ModelSourceCustom, nil
	}
	source, err := provider.ModelSource(ctx)
	if err != nil {
		return "", err
	}
	if strings.EqualFold(strings.TrimSpace(source), contract.ModelSourcePlatform) {
		return contract.ModelSourcePlatform, nil
	}
	return contract.ModelSourceCustom, nil
}

func (s *platformAIService) IsPlatform(ctx context.Context) (bool, error) {
	source, err := s.ModelSource(ctx)
	return source == contract.ModelSourcePlatform, err
}

func (s *platformAIService) Config(ctx context.Context) (*contract.PlatformAIConfig, error) {
	provider := s.current()
	if provider == nil {
		return nil, errors.New("platform AI provider is not initialized")
	}
	return provider.Config(ctx)
}

func (s *platformAIService) Status(ctx context.Context) (*contract.PlatformAIStatus, error) {
	provider := s.current()
	if provider == nil {
		return nil, errors.New("platform AI provider is not initialized")
	}
	return provider.Status(ctx)
}

func (s *platformAIService) current() contract.PlatformAIProvider {
	s.mu.RLock()
	defer s.mu.RUnlock()
	return s.provider
}