package pages

import (
	"code.gitea.io/sdk/gitea"
	"go.uber.org/zap"
	"strconv"
	"strings"
)

type AutoRedirect struct {
	Enabled bool
	Scheme  string
	Code    int
}

type PageClient struct {
	BaseDomain   string
	GiteaConfig  *GiteaConfig
	DomainAlias  *CustomDomains
	ErrorPages   *ErrorPages
	AutoRedirect *AutoRedirect
	OwnerCache   *OwnerCache
	DomainCache  *DomainCache
	logger       *zap.Logger
}

func (p *PageClient) Close() error {
	if p.OwnerCache != nil {
		p.OwnerCache.Cache.Flush()
	}
	if p.DomainCache != nil {
		_ = p.DomainCache.Close()
	}
	return nil
}

func NewPageClient(
	config *MiddlewareConfig,
	logger *zap.Logger,
) (*PageClient, error) {
	options := make([]gitea.ClientOption, 0)
	if config.Token != "" {
		options = append(options, gitea.SetToken(config.Token))
	}
	options = append(options, gitea.SetGiteaVersion(""))
	client, err := gitea.NewClient(config.Server, options...)
	if err != nil {
		return nil, err
	}
	alias, err := NewCustomDomains(config.Alias, config.SharedAlias)
	if err != nil {
		return nil, err
	}
	pages, err := NewErrorPages(config.ErrorPages)
	if err != nil {
		return nil, err
	}
	ownerCache := NewOwnerCache(config.CacheRefresh, config.CacheTimeout)
	giteaConfig := &GiteaConfig{
		Server:        config.Server,
		Token:         config.Token,
		Client:        client,
		Logger:        logger,
		CacheMaxSize:  config.CacheMaxSize,
		CustomHeaders: config.CustomHeaders,
	}
	domainCache := NewDomainCache(config.CacheRefresh, config.CacheTimeout)
	logger.Info("gitea cache ttl " + strconv.FormatInt(config.CacheTimeout.Milliseconds(), 10) + " ms .")
	return &PageClient{
		GiteaConfig:  giteaConfig,
		BaseDomain:   "." + strings.Trim(config.Domain, "."),
		DomainAlias:  alias,
		ErrorPages:   pages,
		logger:       logger,
		AutoRedirect: config.AutoRedirect,
		DomainCache:  &domainCache,
		OwnerCache:   &ownerCache,
	}, nil
}

func (p *PageClient) Validate() error {
	ver, _, err := p.GiteaConfig.Client.ServerVersion()
	p.logger.Info("Gitea Version ", zap.String("version", ver))
	if err != nil {
		p.logger.Warn("Failed to get Gitea version", zap.Error(err))
	}
	return nil
}