package svc

import (
	"github.com/Kagami/go-face"
	"github.com/minio/minio-go/v7"
	"github.com/redis/go-redis/v9"
	"gocv.io/x/gocv"
	"schisandra-album-cloud-microservices/app/aisvc/model/mysql"
	"schisandra-album-cloud-microservices/app/aisvc/model/mysql/query"
	"schisandra-album-cloud-microservices/app/aisvc/rpc/internal/config"
	"schisandra-album-cloud-microservices/common/caffe_classifier"
	"schisandra-album-cloud-microservices/common/clarity"
	"schisandra-album-cloud-microservices/common/face_recognizer"
	"schisandra-album-cloud-microservices/common/miniox"
	"schisandra-album-cloud-microservices/common/redisx"
	"schisandra-album-cloud-microservices/common/tf_classifier"
)

type ServiceContext struct {
	Config         config.Config
	FaceRecognizer *face.Recognizer
	DB             *query.Query
	RedisClient    *redis.Client
	TfNet          *gocv.Net
	TfDesc         []string
	CaffeNet       *gocv.Net
	CaffeDesc      []string
	MinioClient    *minio.Client
	Clarity        *clarity.Detector
}

func NewServiceContext(c config.Config) *ServiceContext {
	redisClient := redisx.NewRedis(c.RedisConf.Host, c.RedisConf.Pass, c.RedisConf.DB)
	_, queryDB := mysql.NewMySQL(c.Mysql.DataSource, c.Mysql.MaxOpenConn, c.Mysql.MaxIdleConn, redisClient)
	tfClassifier, tfDesc := tf_classifier.NewTFClassifier()
	caffeClassifier, caffeDesc := caffe_classifier.NewCaffeClassifier()
	return &ServiceContext{
		Config:         c,
		FaceRecognizer: face_recognizer.NewFaceRecognition(),
		DB:             queryDB,
		RedisClient:    redisClient,
		TfNet:          tfClassifier,
		TfDesc:         tfDesc,
		CaffeNet:       caffeClassifier,
		CaffeDesc:      caffeDesc,
		MinioClient:    miniox.NewMinio(c.Minio.Endpoint, c.Minio.AccessKeyID, c.Minio.SecretAccessKey, c.Minio.UseSSL),
		Clarity:        clarity.NewDetector(clarity.WithConcurrency(8), clarity.WithBaseThreshold(90), clarity.WithEdgeBoost(1.2), clarity.WithSampleScale(1)),
	}
}