package logic

import (
	"context"
	"errors"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
	"github.com/tiger1103/gfast/v3/internal/app/businesses/dao"
	"github.com/tiger1103/gfast/v3/internal/app/businesses/model"
	"github.com/tiger1103/gfast/v3/internal/app/businesses/model/do"
	"github.com/tiger1103/gfast/v3/internal/app/businesses/service"
	"github.com/tiger1103/gfast/v3/internal/app/system/consts"
	"github.com/tiger1103/gfast/v3/library/liberr"
)

func init() {
	service.RegisterGeography(NewIGeography())
}

func NewIGeography() service.IGeography {
	return &sGeography{}
}

type sGeography struct{}

func (s *sGeography) List(ctx context.Context, req *model.GeographySearchReq) (listRes *model.GeographySearchRes, err error) {
	listRes = new(model.GeographySearchRes)
	err = g.Try(ctx, func(ctx context.Context) {
		m := dao.Geography.Ctx(ctx).WithAll()
		if req.Id != "" {
			m = m.Where(dao.Geography.Columns().Id+" = ?", req.Id)
		}
		if req.SpeciesCode != "" {
			m = m.Where(dao.Geography.Columns().SpeciesCode+" like ?", "%"+req.SpeciesCode+"%")
		}
		if req.CreateUser != "" {
			m = m.Where(dao.Geography.Columns().CreateUser+" = ?", gconv.Int(req.CreateUser))
		}
		if req.CreateDate != nil {
			start := req.CreateDate[0]
			end := req.CreateDate[1]
			if start != nil && end != nil {
				m = m.Where(
					dao.Geography.Columns().CreateDate+" >= ? AND "+
						dao.Geography.Columns().CreateDate+" <= ?",
					start,
					end,
				)
			}
		}
		if req.AuditUser != "" {
			m = m.Where(dao.Geography.Columns().AuditUser+" = ?", gconv.Int(req.AuditUser))
		}
		if req.AuditDate != nil {
			start := req.AuditDate[0]
			end := req.AuditDate[1]
			if start != nil && end != nil {
				m = m.Where(
					dao.Geography.Columns().AuditDate+" >= ? AND "+
						dao.Geography.Columns().AuditDate+" <= ?",
					start,
					end,
				)
			}
		}
		if req.AuditStatus != "" {
			m = m.Where(dao.Geography.Columns().AuditStatus+" = ?", gconv.Int(req.AuditStatus))
		}
		listRes.Total, err = m.Count()
		liberr.ErrIsNil(ctx, err, "获取总行数失败")
		if req.PageNum == 0 {
			req.PageNum = 1
		}
		listRes.CurrentPage = req.PageNum
		if req.PageSize == 0 {
			req.PageSize = consts.PageSize
		}
		order := "id asc"
		if req.OrderBy != "" {
			order = req.OrderBy
		}
		var res []*model.GeographyListRes
		err = m.Page(req.PageNum, req.PageSize).Order(order).Scan(&res)
		liberr.ErrIsNil(ctx, err, "获取数据失败")
		listRes.List = make([]*model.GeographyListRes, len(res))
		for k, v := range res {
			listRes.List[k] = &model.GeographyListRes{
				Id:               v.Id,
				SpeciesCode:      v.SpeciesCode,
				DistributionInfo: v.DistributionInfo,
				OriginInfo:       v.OriginInfo,
				AbroadInfo:       v.AbroadInfo,
				DomesticInfo:     v.DomesticInfo,
				CreateUser:       v.CreateUser,
				CreateDate:       v.CreateDate,
				AuditUser:        v.AuditUser,
				AuditDate:        v.AuditDate,
				AuditStatus:      v.AuditStatus,
				AuditView:        v.AuditView,
				Remark:           v.Remark,
				Version:          v.Version,
				CreatedAt:        v.CreatedAt,
			}
		}
	})
	return
}

func (s *sGeography) GetById(ctx context.Context, id int) (res *model.GeographyInfoRes, err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		err = dao.Geography.Ctx(ctx).WithAll().Where(dao.Geography.Columns().Id, id).Scan(&res)
		liberr.ErrIsNil(ctx, err, "获取信息失败")
	})
	return
}

func (s *sGeography) Add(ctx context.Context, req *model.GeographyAddReq) (err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		//查询是否存在
		speciesNameCount, err := dao.SpeciesName.Ctx(ctx).Where(dao.SpeciesName.Columns().SpeciesCode+" = ?", req.SpeciesCode).Count()
		if err != nil {
			liberr.ErrIsNil(ctx, err, "获取物种编码失败")
			return
		}
		if speciesNameCount == 0 {
			liberr.ErrIsNil(ctx, errors.New("物种编码不存在"), "物种编码不存在")
			return
		}
		//查重
		count, err := dao.Geography.Ctx(ctx).Where(dao.Geography.Columns().SpeciesCode+" = ?", req.SpeciesCode).Count()
		if err != nil {
			liberr.ErrIsNil(ctx, err, "获取物种编码失败")
			return

		}
		if count > 0 {
			liberr.ErrIsNil(ctx, errors.New("请勿重复添加"), "请勿重复添加")
			return
		}
		_, err = dao.Geography.Ctx(ctx).Insert(do.Geography{
			Id:               req.Id,
			SpeciesCode:      req.SpeciesCode,
			DistributionInfo: req.DistributionInfo,
			OriginInfo:       req.OriginInfo,
			AbroadInfo:       req.AbroadInfo,
			DomesticInfo:     req.DomesticInfo,
			CreateUser:       req.CreateUser,
			CreateDate:       req.CreateDate,
			//AuditUser:        req.AuditUser,
			//AuditDate:        req.AuditDate,
			AuditStatus: req.AuditStatus,
			//AuditView:        req.AuditView,
			Remark: req.Remark,
		})
		liberr.ErrIsNil(ctx, err, "添加失败")
	})
	return
}

func (s *sGeography) Edit(ctx context.Context, req *model.GeographyEditReq) (err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		// 查询物种编码是否存在
		speciesNameCount, err := dao.SpeciesName.Ctx(ctx).Where(dao.SpeciesName.Columns().SpeciesCode+" = ?", req.SpeciesCode).Count()
		if err != nil {
			liberr.ErrIsNil(ctx, err, "物种编码不存在")
			return
		}
		if speciesNameCount == 0 {
			liberr.ErrIsNil(ctx, errors.New("物种编码不存在"), "物种编码不存在")
			return
		}
		count, err := dao.Geography.Ctx(ctx).
			Where(dao.Geography.Columns().SpeciesCode+" = ?", req.SpeciesCode).
			WhereNot(dao.Geography.Columns().Id, req.Id). // 排除当前记录
			Count()
		if err != nil {
			liberr.ErrIsNil(ctx, err, "获取物种编码失败")
			return
		}
		if count > 0 {
			// 使用非nil错误触发错误处理
			liberr.ErrIsNil(ctx, errors.New("物种编码已存在"), "请勿重复添加")
			return
		}
		_, err = dao.Geography.Ctx(ctx).Where(dao.Geography.Columns().Id+" = ?", req.Id).Update(do.Geography{
			SpeciesCode:      req.SpeciesCode,
			DistributionInfo: req.DistributionInfo,
			OriginInfo:       req.OriginInfo,
			AbroadInfo:       req.AbroadInfo,
			DomesticInfo:     req.DomesticInfo,
			CreateUser:       req.CreateUser,
			CreateDate:       req.CreateDate,
			//AuditUser:        req.AuditUser,
			//AuditDate:        req.AuditDate,
			AuditStatus: req.AuditStatus,
			//AuditView:        req.AuditView,
			Remark:  req.Remark,
			Version: req.Version,
		})
		liberr.ErrIsNil(ctx, err, "修改失败")
	})
	return
}

func (s *sGeography) Delete(ctx context.Context, ids []int) (err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		_, err = dao.Geography.Ctx(ctx).Delete(dao.Geography.Columns().Id+" in (?)", ids)
		liberr.ErrIsNil(ctx, err, "删除失败")
	})
	return
}