package impl

import (
	"context"
	"fmt"
	"gitee.com/MinKo-U/keyauth/apps/token"
	"github.com/infraboard/mcube/exception"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
)

func (s *service) save(ctx context.Context, ins *token.Token) error {

	if _, err := s.col.InsertOne(ctx, ins); err != nil {
		return exception.NewInternalServerError("inserted token(%s) document error, %s",
			ins.AccessToken, err)
	}
	return nil
}
func (s *service) get(ctx context.Context, req *token.ValidateTokenRequest) (*token.Token, error) {
	filter := bson.M{"_id": req.AccessToken, "domain": req.Domain}
	ins := token.NewToken()
	if err := s.col.FindOne(ctx, filter).Decode(ins); err != nil {
		if err == mongo.ErrNoDocuments {
			return nil, exception.NewNotFound("token id %s not found", req.AccessToken)
		}
		return nil, exception.NewInternalServerError("find token id  %s error, %s", req.AccessToken, err)
	}

	return ins, nil
}
func (s *service) delete(ctx context.Context, ins *token.Token) error {
	if ins == nil || ins.AccessToken == "" {
		return fmt.Errorf("token is nil")
	}

	result, err := s.col.DeleteOne(ctx, bson.M{"_id": ins.AccessToken, "domain": ins.Domain})
	if err != nil {
		return exception.NewInternalServerError("delete book(%s) error, %s", ins.AccessToken, err)
	}

	if result.DeletedCount == 0 {
		return exception.NewNotFound("book %s not found", ins.AccessToken)
	}

	return nil
}
func (s *service) update(ctx context.Context, ins *token.Token) error {
	s.log.Debugf("ins req:%s", ins)
	_, err := s.col.UpdateOne(ctx, bson.M{"_id": ins.AccessToken}, bson.M{"$set": ins})
	if err != nil {
		return exception.NewInternalServerError("inserted token(%s) document error, %s",
			ins.AccessToken, err)
	}

	return nil
}