package impl

import (
	"context"
	"fmt"

	"gitee.com/talk-about-articles/keyauth-tz/apps/user"

	"github.com/infraboard/mcube/exception"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

func (s *service) save(ctx context.Context, ins *user.User) error {
	// insertOne 和 insertMany 会根据定义模型的bson标签完成对应的文档类型入库，主要是bson标签
	if _, err := s.col.InsertOne(ctx, ins); err != nil {
		return exception.NewInternalServerError("inserted user(%s) document error, %s",
			ins.Data.Name, err)
	}
	return nil
}

// GET,Describe
// filter 过滤器（Collection）,类似于mysql的where条件
// 过滤器内容就是按照文档中的任意一个键值对作为过滤条件 完成过滤查询
// 调用Decode方法来进行反序列化 bytes --> Object   (查询出来数据的反序列化)
func (s *service) get(ctx context.Context, req *user.DescribeUserRequest) (*user.User, error) {
	filter := bson.M{}
	s.log.Debugf("req %s", req)
	switch req.DescribeBy {
	case user.DescribeBy_USER_ID:
		// 以主键的键值对为条件过滤   就是类似于在json串中匹配  "_id: 1" 的json串
		// 这里就等于是赋值声明，等同于 filter := bson.M{"_id",req.Id}
		filter["_id"] = req.Id
	case user.DescribeBy_USER_NAME:
		filter["data.name"] = req.UserName
		filter["data.domain"] = req.Domain
	default:
		return nil, fmt.Errorf("unknown describe_by %s", req.DescribeBy)
	}

	ins := user.NewDefaultUser()
	// 通过Decode反序列化为Object对象
	if err := s.col.FindOne(ctx, filter).Decode(ins); err != nil {
		if err == mongo.ErrNoDocuments {
			return nil, exception.NewNotFound("user %s not found", req.Id)
		}

		return nil, exception.NewInternalServerError("find user %s error, %s", req.Id, err)
	}

	return ins, nil
}

func newQueryUserRequest(r *user.QueryUserRequest) *queryUserRequest {
	return &queryUserRequest{
		r,
	}
}

type queryUserRequest struct {
	*user.QueryUserRequest
}

func (r *queryUserRequest) FindOptions() *options.FindOptions {
	pageSize := int64(r.Page.PageSize)
	skip := int64(r.Page.PageSize) * int64(r.Page.PageNumber-1)

	opt := &options.FindOptions{
		// 排序： Order By create_at Desc
		Sort: bson.D{
			{Key: "create_at", Value: -1},
		},
		Limit: &pageSize,
		Skip:  &skip,
	}

	return opt
}

// 过滤条件
// 由于Mongodb支持嵌套，JSON,如何过滤嵌套里面的条件，使用.访问嵌套对象属性
func (r *queryUserRequest) FindFilter() bson.M {
	filter := bson.M{}

	return filter
}

func (s *service) query(ctx context.Context, req *queryUserRequest) (*user.UserSet, error) {
	resp, err := s.col.Find(ctx, req.FindFilter(), req.FindOptions())

	if err != nil {
		return nil, exception.NewInternalServerError("find user error, error is %s", err)
	}

	set := user.NewUserSet()
	// 循环
	for resp.Next(ctx) {
		ins := user.NewDefaultUser()
		if err := resp.Decode(ins); err != nil {
			return nil, exception.NewInternalServerError("decode user error, error is %s", err)
		}

		set.Add(ins)
	}

	// count
	count, err := s.col.CountDocuments(ctx, req.FindFilter())
	if err != nil {
		return nil, exception.NewInternalServerError("get user count error, error is %s", err)
	}
	set.Total = count

	return set, nil
}

func (s *service) update(ctx context.Context, ins *user.User) error {
	if _, err := s.col.UpdateByID(ctx, ins.Id, ins); err != nil {
		return exception.NewInternalServerError("inserted user(%s) document error, %s",
			ins.Data.Name, err)
	}

	return nil
}

func (s *service) deleteUser(ctx context.Context, ins *user.User) error {
	if ins == nil || ins.Id == "" {
		return fmt.Errorf("user is nil")
	}

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

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

	return nil
}