// Package stashbox provides a client interface to a stash-box server instance.
package stashbox

import (
	"context"
	"net/http"
	"regexp"

	"github.com/Yamashou/gqlgenc/clientv2"
	"github.com/stashapp/stash/pkg/match"
	"github.com/stashapp/stash/pkg/models"
	"github.com/stashapp/stash/pkg/scraper"
	"github.com/stashapp/stash/pkg/scraper/stashbox/graphql"
	"github.com/stashapp/stash/pkg/txn"
)

type SceneReader interface {
	models.SceneGetter
	models.StashIDLoader
	models.VideoFileLoader
}

type PerformerReader interface {
	models.PerformerGetter
	match.PerformerFinder
	models.AliasLoader
	models.StashIDLoader
	models.URLLoader
	FindBySceneID(ctx context.Context, sceneID int) ([]*models.Performer, error)
	GetImage(ctx context.Context, performerID int) ([]byte, error)
}

type StudioReader interface {
	models.StudioGetter
	match.StudioFinder
	models.StashIDLoader
}

type TagFinder interface {
	models.TagQueryer
	FindBySceneID(ctx context.Context, sceneID int) ([]*models.Tag, error)
}

type Repository struct {
	TxnManager models.TxnManager

	Scene     SceneReader
	Performer PerformerReader
	Tag       TagFinder
	Studio    StudioReader
}

func NewRepository(repo models.Repository) Repository {
	return Repository{
		TxnManager: repo.TxnManager,
		Scene:      repo.Scene,
		Performer:  repo.Performer,
		Tag:        repo.Tag,
		Studio:     repo.Studio,
	}
}

func (r *Repository) WithReadTxn(ctx context.Context, fn txn.TxnFunc) error {
	return txn.WithReadTxn(ctx, r.TxnManager, fn)
}

// Client represents the client interface to a stash-box server instance.
type Client struct {
	client     *graphql.Client
	repository Repository
	box        models.StashBox

	// tag patterns to be excluded
	excludeTagRE []*regexp.Regexp
}

// NewClient returns a new instance of a stash-box client.
func NewClient(box models.StashBox, repo Repository, excludeTagPatterns []string) *Client {
	authHeader := func(ctx context.Context, req *http.Request, gqlInfo *clientv2.GQLRequestInfo, res interface{}, next clientv2.RequestInterceptorFunc) error {
		req.Header.Set("ApiKey", box.APIKey)
		return next(ctx, req, gqlInfo, res)
	}

	client := &graphql.Client{
		Client: clientv2.NewClient(http.DefaultClient, box.Endpoint, nil, authHeader),
	}

	return &Client{
		client:       client,
		repository:   repo,
		box:          box,
		excludeTagRE: scraper.CompileExclusionRegexps(excludeTagPatterns),
	}
}

func (c Client) getHTTPClient() *http.Client {
	return c.client.Client.Client
}

func (c Client) GetUser(ctx context.Context) (*graphql.Me, error) {
	return c.client.Me(ctx)
}