eta_api/services/document_manage_service/document_manage_service.go

// @Author gmy 2024/9/19 16:45:00
package document_manage_service

import (
	"eta/eta_api/models"
	"eta/eta_api/models/document_manage_model"
	"eta/eta_api/utils"
	"fmt"
	"html"
	"strconv"

	"github.com/beego/beego/v2/core/logs"
	"github.com/google/uuid"
	"github.com/rdlucklib/rdluck_tools/paging"
)

func DocumentClassifyList() ([]models.ClassifyVO, error) {
	logs.Info("DocumentClassifyList")

	//获取所有已启用分类
	classifyList, err := models.GetClassifyListByKeyword(``, utils.IS_ENABLE)
	if err != nil {
		return nil, err
	}

	// 递归处理分类 拿到父级分类和对应的子级分类
	classifyVOList := make([]models.ClassifyVO, 0)
	for _, classify := range classifyList {
		if classify.ParentId == 0 {
			classifyVO := models.ClassifyVO{
				Id:           classify.Id,
				ClassifyName: classify.ClassifyName,
				ParentId:     classify.ParentId,
				Sort:         classify.Sort,
				Enabled:      classify.Enabled,
				Level:        classify.Level,
			}
			classifyVO.Children = getChildClassify(classifyList, classify.Id)
			classifyVOList = append(classifyVOList, classifyVO)
		}
	}

	return classifyVOList, nil
}

func getChildClassify(classifyList []*models.ClassifyList, classifyId int) *[]models.ClassifyVO {
	childList := make([]models.ClassifyVO, 0)
	for _, classify := range classifyList {
		if classify.ParentId == classifyId {
			classifyVO := models.ClassifyVO{
				Id:           classify.Id,
				ClassifyName: classify.ClassifyName,
				ParentId:     classify.ParentId,
				Sort:         classify.Sort,
				Enabled:      classify.Enabled,
			}
			classifyVO.Children = getChildClassify(classifyList, classify.Id)
			childList = append(childList, classifyVO)
		}
	}
	return &childList
}

func DocumentVarietyList() ([]models.ChartPermissionVO, error) {
	logs.Info("DocumentVarietyList")

	cp := new(models.ChartPermission)
	var condition string
	pars := make([]interface{}, 0)

	condition = ` and enabled=?`
	pars = append(pars, 1)
	condition += ` and product_id=1`

	chartPermissionList, err := cp.GetItemsByCondition(condition, pars)
	if err != nil {
		return nil, err
	}

	// 递归处理品种 拿到父级品种和对应的子级品种
	permissionVOList := make([]models.ChartPermissionVO, 0)
	for _, chartPermission := range chartPermissionList {
		if chartPermission.ParentId == 0 {
			permissionVO := models.ChartPermissionVO{
				ChartPermissionId:   chartPermission.ChartPermissionId,
				ChartPermissionName: chartPermission.ChartPermissionName,
				PermissionName:      chartPermission.PermissionName,
				ParentId:            chartPermission.ParentId,
				Enabled:             chartPermission.Enabled,
				Sort:                chartPermission.Sort,
				CreatedTime:         chartPermission.CreatedTime,
			}
			permissionVO.Children = getChildVariety(chartPermissionList, permissionVO.ChartPermissionId)
			permissionVOList = append(permissionVOList, permissionVO)
		}
	}

	return permissionVOList, nil
}

func getChildVariety(permissionList []*models.ChartPermission, permissionId int) []*models.ChartPermissionVO {
	childList := make([]*models.ChartPermissionVO, 0)
	for _, permission := range permissionList {
		if permission.ParentId == permissionId {
			permissionVO := models.ChartPermissionVO{
				ChartPermissionId:   permission.ChartPermissionId,
				ChartPermissionName: permission.ChartPermissionName,
				PermissionName:      permission.PermissionName,
				ParentId:            permission.ParentId,
				Enabled:             permission.Enabled,
				Sort:                permission.Sort,
				CreatedTime:         permission.CreatedTime,
			}
			permissionVO.Children = getChildVariety(permissionList, permissionVO.ChartPermissionId)
			childList = append(childList, &permissionVO)
		}
	}
	return childList

}

func DocumentReportList(documentType int, chartPermissionIdList []string, classifyIdList []string, keyword string, orderField, orderType string, startSize, pageSize int) (*document_manage_model.OutsideReportPage, error) {
	logs.Info("DocumentVarietyList")

	var condition string
	pars := make([]interface{}, 0)

	if documentType == 1 {
		condition = ` and t1.source!=3`
	}
	if len(classifyIdList) > 0 {
		// 递归查询子分类
		for _, classifyId := range classifyIdList {
			id, err := strconv.Atoi(classifyId)
			if err != nil {
				return nil, err
			}

			if id > 0 {
				childrenClassifyIdList, err := GetAllClassifyIdsByParentId(id)
				if err != nil {
					return nil, err
				}
				classifyIdList = append(classifyIdList, childrenClassifyIdList...)
			}
		}
		condition += ` and t1.classify_id in (` + utils.GetOrmInReplace(len(classifyIdList)) + `)`
		pars = append(pars, classifyIdList)
	}
	if len(chartPermissionIdList) > 0 {
		condition += ` and t2.chart_permission_id in (` + utils.GetOrmInReplace(len(chartPermissionIdList)) + `)`
		pars = append(pars, chartPermissionIdList)
	}
	if keyword != "" {
		condition += ` and (t1.title like ? or t1.sys_user_name like ?) `
		pars = append(pars, "%"+keyword+"%", "%"+keyword+"%")
	}
	if len(chartPermissionIdList) > 0 {
		condition += ` GROUP BY t1.outside_report_id HAVING COUNT(DISTINCT t2.chart_permission_id) = ` + strconv.Itoa(len(chartPermissionIdList))
	}

	count, err := document_manage_model.GetOutsideReportListByConditionCount(condition, pars, chartPermissionIdList)
	if err != nil {
		return nil, err
	}
	reportPage := document_manage_model.OutsideReportPage{}

	page := paging.GetPaging(startSize, pageSize, count)
	if count <= 0 {
		reportPage.Paging = page
		return &reportPage, nil
	}

	if orderField != "" && orderType != "" {
		condition += ` order by t1.` + orderField + ` ` + orderType
	} else {
		condition += ` order by t1.report_update_time desc`
	}

	condition += ` LIMIT ?,?`
	pars = append(pars, startSize, pageSize)

	outsideReportList, err := document_manage_model.GetOutsideReportListByCondition(condition, pars)
	if err != nil {
		return nil, err
	}
	reportPage.Paging = page
	reportPage.List = outsideReportList

	return &reportPage, nil
}

// GetAllClassifyIdsByParentId 递归查询子分类
func GetAllClassifyIdsByParentId(parentId int) ([]string, error) {
	var classifyIdList []string

	// 获取子分类
	classifyList, err := models.GetClassifyListByParentId(parentId)
	if err != nil {
		return nil, err
	}

	// 遍历子分类
	for _, classify := range classifyList {
		classifyIdList = append(classifyIdList, strconv.Itoa(classify.Id))
		// 递归调用
		subClassifyIds, err := GetAllClassifyIdsByParentId(classify.Id)
		if err != nil {
			return nil, err
		}
		classifyIdList = append(classifyIdList, subClassifyIds...)
	}

	return classifyIdList, nil
}

func RuiSiReportList(classifyIdFirst, classifyIdSecond, classifyIdThird int, chartPermissionIdList []string, keyword string, orderField, orderType string, startSize, pageSize int) (*models.ReportListResp, error) {
	logs.Info("RuiSiReportList")

	var condition string
	pars := make([]interface{}, 0)

	if classifyIdFirst > 0 {
		condition += ` AND a.classify_id_first = ? `
		pars = append(pars, classifyIdFirst)
	}
	if classifyIdSecond > 0 {
		condition += ` AND a.classify_id_second = ? `
		pars = append(pars, classifyIdSecond)
	}
	if classifyIdThird > 0 {
		condition += ` AND a.classify_id_third = ? `
		pars = append(pars, classifyIdThird)
	}
	if keyword != "" {
		condition += ` and ( a.title like ? or a.admin_real_name like ? ) `
		pars = append(pars, "%"+keyword+"%", "%"+keyword+"%")
	}

	// 作者为 全球市场战略研究中心 PCI Research
	//condition += ` and a.author = '战研中心 PCIR'`
	// 已发布的报告
	condition += ` AND a.state IN (2,6) AND a.is_public_publish = 1 `

	if len(chartPermissionIdList) > 0 {
		classifyIds, err := models.GetClassifyIdsByPermissionId(chartPermissionIdList)
		if err != nil {
			return nil, err
		}

		if len(classifyIds) > 0 {
			condition += ` AND ( (a.classify_id_first IN (` + utils.GetOrmInReplace(len(classifyIds)) + `) AND a.classify_id_second = 0) 
			OR (a.classify_id_second IN (` + utils.GetOrmInReplace(len(classifyIds)) + `) AND a.classify_id_third = 0) 
			OR a.classify_id_third IN (` + utils.GetOrmInReplace(len(classifyIds)) + `) )`
			for _, classifyId := range classifyIds {
				pars = append(pars, classifyId)
			}
			for _, classifyId := range classifyIds {
				pars = append(pars, classifyId)
			}
			for _, classifyId := range classifyIds {
				pars = append(pars, classifyId)
			}
		}
	}

	count, err := models.GetReportListCountV1(condition, pars)
	if err != nil {
		return nil, err
	}
	reportPage := models.ReportListResp{}

	page := paging.GetPaging(startSize, pageSize, count)
	if count <= 0 {
		reportPage.Paging = page
		return &reportPage, nil
	}

	if orderField != "" && orderType != "" {
		condition += ` order by a.` + orderField + ` ` + orderType
	} else {
		condition += ` order by a.publish_time desc`
	}

	reportList, err := models.GetReportListByCondition(condition, pars, startSize, pageSize)
	if err != nil {
		return nil, err
	}

	listLen := len(reportList)
	if listLen > 0 {
		classifyIdList := make([]int, 0)              // 关联的指标分类id
		classifyIdMap := make(map[int]bool)           // 分类id map
		classifyMap := make(map[int]*models.Classify) // 分类map

		for i := 0; i < listLen; i++ {
			if reportList[i].ClassifyIdFirst > 0 {
				if _, ok := classifyIdMap[reportList[i].ClassifyIdFirst]; !ok {
					classifyIdList = append(classifyIdList, reportList[i].ClassifyIdFirst)
				}
			}
			if reportList[i].ClassifyIdSecond > 0 {
				if _, ok := classifyIdMap[reportList[i].ClassifyIdSecond]; !ok {
					classifyIdList = append(classifyIdList, reportList[i].ClassifyIdSecond)
				}
			}
			if reportList[i].ClassifyIdThird > 0 {
				if _, ok := classifyIdMap[reportList[i].ClassifyIdThird]; !ok {
					classifyIdList = append(classifyIdList, reportList[i].ClassifyIdThird)
				}
			}
		}

		if len(classifyIdList) > 0 {
			classifyList, e := models.GetClassifyListByIdList(classifyIdList)
			if e != nil {
				return &reportPage, e
			}
			for _, v := range classifyList {
				classifyMap[v.Id] = v
			}
		}

		for i := 0; i < listLen; i++ {
			if reportList[i].ClassifyIdThird > 0 {
				if classify, ok := classifyMap[reportList[i].ClassifyIdThird]; ok {
					if classify.Enabled == 1 {
						reportList[i].ClassifyEnabled = true
					}
				}
			} else if reportList[i].ClassifyIdSecond > 0 {
				if classify, ok := classifyMap[reportList[i].ClassifyIdSecond]; ok {
					if classify.Enabled == 1 {
						reportList[i].ClassifyEnabled = true
					}
				}
			} else if reportList[i].ClassifyIdFirst > 0 {
				if classify, ok := classifyMap[reportList[i].ClassifyIdFirst]; ok {
					if classify.Enabled == 1 {
						reportList[i].ClassifyEnabled = true
					}
				}
			}
		}
	}

	reportPage.Paging = page
	reportPage.List = reportList

	return &reportPage, nil
}

func DocumentRuiSiDetail(reportId int) (*models.ReportDetail, error) {
	logs.Info("DocumentRuiSiDetail")

	reportDetail, err := models.GetReportById(reportId)
	if err != nil {
		if utils.IsErrNoRow(err) {
			return nil, fmt.Errorf("报告已被删除")
		}
		return nil, err
	}
	reportDetail.Content = html.UnescapeString(reportDetail.Content)

	return reportDetail, nil
}

func DocumentSave(outsideReport *document_manage_model.OutsideReportBO) error {
	logs.Info("DocumentSave")

	// 保存报告
	report := document_manage_model.OutsideReport{
		Source:           outsideReport.Source,
		Title:            outsideReport.Title,
		Abstract:         outsideReport.Abstract,
		ClassifyId:       outsideReport.ClassifyId,
		ClassifyName:     outsideReport.ClassifyName,
		Content:          outsideReport.Content,
		SysUserId:        outsideReport.SysUserId,
		SysUserName:      outsideReport.SysUserName,
		ReportUpdateTime: utils.GetCurrentTime(),
		ModifyTime:       utils.GetCurrentTime(),
		CreateTime:       utils.GetCurrentTime(),
		ReportCode:       uuid.New().String(),
	}
	id, err := document_manage_model.SaveOutsideReport(report)
	if err != nil {
		return err
	}

	// 保存附件
	attachmentList := outsideReport.AttachmentList
	if len(attachmentList) > 0 {
		for _, attachment := range attachmentList {
			if attachment.Title == "" || attachment.Url == "" {
				continue
			}
			attachment.OutsideReportId = int(id)
			attachment.CreateTime = utils.GetCurrentTime()
			_, err := document_manage_model.SaveOutsideReportAttachment(attachment)
			if err != nil {
				return err
			}
		}
	}
	return err
}

func DocumentReportDetail(outsideReportId int) (*document_manage_model.OutsideReportBO, error) {
	logs.Info("DocumentReportDetail")

	outsideReport, err := document_manage_model.GetOutsideReportById(outsideReportId)
	if err != nil {
		return nil, err
	}

	attachmentList, err := document_manage_model.GetOutsideReportAttachmentListByReportId(outsideReportId)
	if err != nil {
		return nil, err
	}

	outsideReportBO := document_manage_model.OutsideReportBO{
		OutsideReportId: outsideReportId,
		Source:          outsideReport.Source,
		Title:           outsideReport.Title,
		Abstract:        outsideReport.Abstract,
		ClassifyId:      outsideReport.ClassifyId,
		ClassifyName:    outsideReport.ClassifyName,
		Content:         html.UnescapeString(outsideReport.Content),
		SysUserId:       outsideReport.SysUserId,
		SysUserName:     outsideReport.SysUserName,
		CreateTime:      outsideReport.CreateTime,
		ModifyTime:      outsideReport.ModifyTime,
		ReportCode:      outsideReport.ReportCode,
		AttachmentList:  attachmentList,
	}
	return &outsideReportBO, nil
}

func DocumentUpdate(outsideReport *document_manage_model.OutsideReportBO) error {
	logs.Info("DocumentUpdate")
	report, err := document_manage_model.GetOutsideReportById(outsideReport.OutsideReportId)
	if err != nil {
		return err
	}

	if report == nil {
		return fmt.Errorf("报告不存在")
	}
	// 更新报告
	if outsideReport.Title != "" {
		report.Title = outsideReport.Title
	}

	report.Abstract = outsideReport.Abstract

	if outsideReport.ClassifyId > 0 {
		report.ClassifyId = outsideReport.ClassifyId
	}
	if outsideReport.ClassifyName != "" {
		report.ClassifyName = outsideReport.ClassifyName
	}
	if outsideReport.Content != "" {
		report.Content = outsideReport.Content
	}
	report.ModifyTime = utils.GetCurrentTime()
	//report.ReportUpdateTime = utils.GetCurrentTime()
	err = document_manage_model.UpdateOutsideReport(report, []string{"Title", "Abstract", "ClassifyId", "ClassifyName", "Content", "ModifyTime"})
	if err != nil {
		return fmt.Errorf("更新报告失败, Err: %s", err.Error())
	}

	// 更新报告附件
	attachmentList := outsideReport.AttachmentList
	if len(attachmentList) > 0 {
		err = document_manage_model.DeleteReportAttachmentByReportId(outsideReport.OutsideReportId)
		if err != nil {
			return fmt.Errorf("删除报告附件失败, Err: %s", err.Error())
		}

		for _, attachment := range attachmentList {
			if attachment.Title == "" || attachment.Url == "" {
				continue
			}
			attachment.OutsideReportId = outsideReport.OutsideReportId
			attachment.CreateTime = utils.GetCurrentTime()
			_, err := document_manage_model.SaveOutsideReportAttachment(attachment)
			if err != nil {
				return err
			}
		}
	}
	return nil
}

func DocumentDelete(outsideReportId int) error {
	logs.Info("DocumentDelete")

	report, err := document_manage_model.GetOutsideReportById(outsideReportId)
	if err != nil {
		return err
	}
	if report == nil {
		return fmt.Errorf("报告不存在")
	}

	err = document_manage_model.DeleteOutsideReport(outsideReportId)
	if err != nil {
		return err
	}
	err = document_manage_model.DeleteReportAttachmentByReportId(outsideReportId)
	if err != nil {
		return err
	}
	return nil
}