package business

import (
	"../repository"
	"context"
	"encoding/json"
	"fmt"
	"gitee.com/cuihovah/hzmgo"
	"gitee.com/cuihovah/kratos"
	"gitee.com/cuihovah/kratos/examples/simple_server/schema"
	"github.com/Unknwon/goconfig"
	"github.com/julienschmidt/httprouter"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"log"
	"net/http"
)

type Serv struct {
	Db   *mongo.Database
	Cfg  *goconfig.ConfigFile
	repo *repository.Repository
}

func (s *Serv) MongoClient() *mongo.Client {
	return s.Db.Client()
}

func (s *Serv) SessionName() string {
	return "sdp_user_token"
}

func (s *Serv) GetApiName() string {
	return "/spdb_view/v1"
}

func (s *Serv) GetToken(ctx context.Context, jwtStr string) (kratos.Token, error) {
	claims, err := ParseJWT(jwtStr, "your_jwt_secret")
	if err != nil {
		return nil, err
	}
	user := schema.User{}
	// 这里根据你的 schema.User 定义，把 claims 转成对应类型
	// 例如：
	buffer, err := json.Marshal(claims)
	if err != nil {
		return nil, err
	}
	if err := json.Unmarshal(buffer, &user); err != nil {
		return nil, err
	}
	return user, nil
}

func New(url, dbname string) (*Serv, error) {
	var result Serv
	clientOptions := options.Client().ApplyURI(url)
	client, err := mongo.Connect(context.TODO(), clientOptions)

	if err != nil {
		log.Fatal(err)
	}

	err = client.Ping(context.TODO(), nil)

	if err != nil {
		return nil, err
	}
	result.Db = client.Database(dbname)
	mgo := hzmgo.NewMgo(result.Db)
	if err != nil {
		fmt.Println("必须设置目标文件")
		return nil, err
	}
	result.repo = repository.GetRepository(mgo)
	if err != nil {
		return &result, err
	}
	return &result, err
}

func (s *Serv) Start(addr string, router *httprouter.Router) error {
	return http.ListenAndServe(addr, router)
}