// ==========================================================================
// GFast自动生成logic操作代码。
// 生成日期：2023-10-07 16:08:10
// 生成路径: internal/app/system/logic/xiangmuhuafenbiao.go
// 生成人：gfast
// desc:项目划分
// company:云南奇讯科技有限公司
// ==========================================================================

package logic

import (
	"context"
	"errors"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
	"github.com/tiger1103/gfast/v3/api/v1/common/coryCommon"
	"github.com/tiger1103/gfast/v3/api/v1/common/globe"
	"github.com/tiger1103/gfast/v3/api/v1/common/tool/excel"
	"github.com/tiger1103/gfast/v3/api/v1/system"
	"github.com/tiger1103/gfast/v3/internal/app/system/consts"
	"github.com/tiger1103/gfast/v3/internal/app/system/dao"
	"github.com/tiger1103/gfast/v3/internal/app/system/model"
	"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
	"github.com/tiger1103/gfast/v3/internal/app/system/service"
	"github.com/tiger1103/gfast/v3/library/liberr"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"
)

func init() {
	service.RegisterXiangmuhuafenbiao(New())
}

func New() *sXiangmuhuafenbiao {
	return &sXiangmuhuafenbiao{}
}

type sXiangmuhuafenbiao struct{}

func (s *sXiangmuhuafenbiao) EditExcelFunc(ctx context.Context, req *system.EditExcelFuncReq) (err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		//1、获取到document表对应的文件，然后返回这一行的数据
		var dt *model.DocumentInfoRes
		dao.Document.Ctx(ctx).Where("name", req.FileName).Limit(1).Scan(&dt)
		//2、根据
		var ey *model.XiangmuhuafenbiaoInfoRes
		dao.Xiangmuhuafenbiao.Ctx(ctx).WherePri(req.Id).Scan(&ey)
		var pathName = ""
		if ey != nil {
			dir := filepath.Dir(dt.FilenPath)
			dir = strings.ReplaceAll(dir, "/file", coryCommon.GetCWD()+"/resource/public")
			timestamp := time.Now().UnixNano() / int64(time.Millisecond)
			sjc := strconv.FormatInt(timestamp, 10)
			pathName = dir + "/" + strconv.FormatInt(ey.ProjectId, 10) + strconv.FormatInt(ey.Id, 10) + sjc + ".txt"
			flag, err := coryCommon.OutJson([]byte(req.Json), pathName)
			if err != nil || flag == false {
				err = errors.New("修改失败")
				return
			}
			_, err = dao.Xiangmuhuafenbiao.Ctx(ctx).WherePri(req.Id).Update(do.DocumentProject{
				Json: strings.ReplaceAll(pathName, "\\", "/"),
			})
			//把原本的json删除
			if err == nil && ey.Json != "" {
				dirTwo := strings.ReplaceAll(ey.Json, "/file", coryCommon.GetCWD()+"/resource/public")
				os.Remove(dirTwo)
			}
			liberr.ErrIsNil(ctx, err, "修改失败")
		}
	})
	return
}

func (s *sXiangmuhuafenbiao) List(ctx context.Context, req *system.XiangmuhuafenbiaoSearchReq) (listRes *system.XiangmuhuafenbiaoSearchRes, err error) {
	listRes = new(system.XiangmuhuafenbiaoSearchRes)
	err = g.Try(ctx, func(ctx context.Context) {
		m := dao.Xiangmuhuafenbiao.Ctx(ctx).WithAll()
		if req.Id != "" {
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Id+" = ?", req.Id)
		}
		if req.Danweigongcheng != "" {
			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Danweigongcheng+" = ?", req.Danweigongcheng)
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Danweigongcheng+" like ?", "%"+req.Danweigongcheng+"%")

		}
		if req.Zidanweigongcheng != "" {
			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Zidanweigongcheng+" = ?", req.Zidanweigongcheng)
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Zidanweigongcheng+" like ?", "%"+req.Zidanweigongcheng+"%")

		}
		if req.Fenbugongcheng != "" {
			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Fenbugongcheng+" = ?", req.Fenbugongcheng)
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Fenbugongcheng+" like ?", "%"+req.Fenbugongcheng+"%")

		}
		if req.Zifenbugongcheng != "" {
			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Zifenbugongcheng+" = ?", req.Zifenbugongcheng)
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Zifenbugongcheng+" like ?", "%"+req.Zifenbugongcheng+"%")

		}
		if req.Fenxianggongcheng != "" {
			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Fenxianggongcheng+" = ?", req.Fenxianggongcheng)
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Fenxianggongcheng+" like ?", "%"+req.Fenxianggongcheng+"%")

		}
		if req.Jianyanpi != "" {
			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Jianyanpi+" = ?", req.Jianyanpi)
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Jianyanpi+" like ?", "%"+req.Jianyanpi+"%")

		}
		if req.Name != "" {
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Name+" like ?", "%"+req.Name+"%")
		}
		if req.Zhixingbiaohao != "" {
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().Zhixingbiaohao+" like ?", "%"+req.Zhixingbiaohao+"%")

			//m = m.Where(dao.Xiangmuhuafenbiao.Columns().Zhixingbiaohao+" = ?", req.Zhixingbiaohao)
		}
		//if req.Json != "" {
		//	m = m.Where(dao.Xiangmuhuafenbiao.Columns().Json+" = ?", req.Json)
		//}
		if req.PrintStatus != "" {
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().PrintStatus+" = ?", gconv.Int(req.PrintStatus))
		}
		if req.ProjectId != "" {
			m = m.Where(dao.Xiangmuhuafenbiao.Columns().ProjectId+" = ?", gconv.Int(req.ProjectId))
		}
		//if req.CreateBy != "" {
		//	m = m.Where(dao.Xiangmuhuafenbiao.Columns().CreateBy+" = ?", req.CreateBy)
		//}
		//if req.UpdateBy != "" {
		//	m = m.Where(dao.Xiangmuhuafenbiao.Columns().UpdateBy+" = ?", req.UpdateBy)
		//}
		//if req.CreateddAt != "" {
		//	m = m.Where(dao.Xiangmuhuafenbiao.Columns().CreateddAt+" = ?", gconv.Time(req.CreateddAt))
		//}
		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.XiangmuhuafenbiaoInfoRes
		err = m.Fields(system.XiangmuhuafenbiaoSearchRes{}).Page(req.PageNum, req.PageSize).Order(order).Scan(&res)
		liberr.ErrIsNil(ctx, err, "获取数据失败")
		listRes.List = make([]*model.XiangmuhuafenbiaoListRes, len(res))
		for k, v := range res {
			listRes.List[k] = &model.XiangmuhuafenbiaoListRes{
				Id:                v.Id,
				Danweigongcheng:   v.Danweigongcheng,
				Zidanweigongcheng: v.Zidanweigongcheng,
				Fenbugongcheng:    v.Fenbugongcheng,
				Zifenbugongcheng:  v.Zifenbugongcheng,
				Fenxianggongcheng: v.Fenxianggongcheng,
				Jianyanpi:         v.Jianyanpi,
				Name:              v.Name,
				Zhixingbiaohao:    v.Zhixingbiaohao,
				Json:              v.Json,
				PrintStatus:       v.PrintStatus,
				ProjectId:         v.ProjectId,
				CreateBy:          v.CreateBy,
				UpdateBy:          v.UpdateBy,
				CreateddAt:        v.CreateddAt,
			}
		}
	})
	return
}

func (s *sXiangmuhuafenbiao) GetById(ctx context.Context, id int) (res *model.XiangmuhuafenbiaoInfoRes, err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		err = dao.Xiangmuhuafenbiao.Ctx(ctx).WithAll().Where(dao.Xiangmuhuafenbiao.Columns().Id, id).Scan(&res)
		json := res.Json
		if json != "" {
			data, err2 := coryCommon.PutJson(json)
			if err2 != nil {
				err = err2
			}
			res.Json = data
		}
		liberr.ErrIsNil(ctx, err, "获取信息失败")
	})
	return
}

func (s *sXiangmuhuafenbiao) Add(ctx context.Context, req *system.XiangmuhuafenbiaoAddReq) (err error) {
	err = g.Try(ctx, func(ctx context.Context) {

		res, e := service.SysProject().GetByProjectId(ctx, req.ProjectId)
		if e != nil {
			liberr.ErrIsNil(ctx, e)
			return
		}
		if res == nil {
			liberr.ErrIsNil(ctx, errors.New("项目不存在"))
			return
		}
		str, err1 := coryCommon.UploadFile(ctx, req.File, coryCommon.ProjectSplitTable)
		if err1 != nil {
			liberr.ErrIsNil(ctx, err1, "上传失败！")
		}
		if !strings.HasSuffix(str, ".xls") {
			liberr.ErrIsNil(ctx, err, "仅支持xls格式数据")
		}
		err2, sheets := excel.ReadXlsx(str)
		if err2 != nil {
			liberr.ErrIsNil(ctx, err2, "表格读取失败")
		}

		dao.Xiangmuhuafenbiao.Ctx(ctx).Where(dao.Xiangmuhuafenbiao.Columns().ProjectId, req.ProjectId).Delete()
		var rows []do.Xiangmuhuafenbiao
		for _, sheet := range sheets {
			for _, row := range sheet.Rows {
				r := do.Xiangmuhuafenbiao{
					ProjectId:   req.ProjectId,
					PrintStatus: globe.DISABLE,
				}
				l := len(row)
				if l > 0 {
					r.Danweigongcheng = row[0]
				}
				if l > 1 {
					r.Zidanweigongcheng = row[1]
				}
				if l > 2 {
					r.Fenbugongcheng = row[2]
				}
				if l > 3 {
					r.Zifenbugongcheng = row[3]
				}
				if l > 4 {
					r.Fenxianggongcheng = row[4]
				}
				if l > 5 {
					r.Jianyanpi = row[5]
				}
				if l > 6 {
					r.Name = row[6]
				}
				if l > 7 {
					r.Zhixingbiaohao = row[7]
				}
				rows = append(rows, r)
				if len(rows) == 1000 {
					dao.Xiangmuhuafenbiao.Ctx(ctx).Insert(&rows)
					rows = []do.Xiangmuhuafenbiao{}
				}
			}
		}
		if len(rows) > 0 {
			dao.Xiangmuhuafenbiao.Ctx(ctx).Insert(&rows)
		}
		liberr.ErrIsNil(ctx, err, "添加失败")
	})
	return
}

func (s *sXiangmuhuafenbiao) Edit(ctx context.Context, req *system.XiangmuhuafenbiaoEditReq) (err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		xiangmuhuafenbiao := do.Xiangmuhuafenbiao{
			Danweigongcheng:   req.Danweigongcheng,
			Zidanweigongcheng: req.Zidanweigongcheng,
			Fenbugongcheng:    req.Fenbugongcheng,
			Zifenbugongcheng:  req.Zifenbugongcheng,
			Fenxianggongcheng: req.Fenxianggongcheng,
			Jianyanpi:         req.Jianyanpi,
			Name:              req.Name,
			Zhixingbiaohao:    req.Zhixingbiaohao,
			PrintStatus:       req.PrintStatus,
			//ProjectId:         req.ProjectId,
			//CreateBy:          req.CreateBy,
			//UpdateBy:          req.UpdateBy,
			//CreateddAt:        req.CreateddAt,
		}
		//json不为空表示更改了文件，把原本的文件删除了
		if req.Json != "" {
			value, _ := dao.Xiangmuhuafenbiao.Ctx(ctx).WherePri(req.Id).Fields("json").Value()
			dir, _ := os.Getwd()
			newPath := strings.ReplaceAll(value.String(), "/file", dir+"/resource/public")
			os.Remove(newPath)
		}
		_, err = dao.Xiangmuhuafenbiao.Ctx(ctx).OmitEmpty().WherePri(req.Id).Update(xiangmuhuafenbiao)
		_, err = dao.Xiangmuhuafenbiao.Ctx(ctx).WherePri(req.Id).Update(g.Map{"json": nil})
		liberr.ErrIsNil(ctx, err, "修改失败")
	})
	return
}

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