// Package gen
// @Link  https://github.com/xiujiecn/xiu-admin
// @Copyright  Copyright (c) 2026 LiXiujie
// @Author  Lxj <xxx@example.com>
// @License  https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
// @AutoGenerate Version
package common

import (
	"context"
	"fmt"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/dao"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/contexts"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/exportcenter"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/websocket"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/xgorm/handler"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/xgorm/hook"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/entity"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/convert"
	"strings"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gtime"
	"github.com/gogf/gf/v2/util/gconv"
)

type sCommonTMessageLog struct{}

func NewCommonTMessageLog() *sCommonTMessageLog {
	return &sCommonTMessageLog{}
}

func init() {
	service.RegisterCommonTMessageLog(NewCommonTMessageLog())
}

// Model 消息日志ORM模型
func (s *sCommonTMessageLog) Model(ctx context.Context, option ...*handler.Option) *gdb.Model {
	if len(option) == 0 {
		option = append(option, &handler.Option{
			FilterTenant: true,
			FilterAuth:   true,
		})
	}
	return handler.Model(dao.TMessageLog.Ctx(ctx), option...)
}

// List 获取消息日志列表
func (s *sCommonTMessageLog) List(ctx context.Context, in *model.TMessageLogListParam) (list []*model.TMessageLogListModel, totalCount int, err error) {
	mod := dao.TMessageLog.Ctx(ctx)
	deptId := contexts.GetDeptId(ctx)

	// 字段过滤
	mod = mod.Fields(model.TMessageLogListModel{})
	g.Log().Info(ctx, "查询用户ID:", contexts.GetUserId(ctx))

	if in.MessageType != nil {
		mod = mod.Where(dao.TMessageLog.Columns().MessageType, in.MessageType)
	}

	if title := strings.TrimSpace(in.MessageTitle); title != "" {
		mod = mod.WhereLike(dao.TMessageLog.Columns().MessageTitle, "%"+title+"%")
	}

	// 查询发送时间范围
	if in.SendTimeStart != nil {
		mod = mod.WhereGTE(dao.TMessageLog.Columns().SendTime, in.SendTimeStart)
	}
	if in.SendTimeEnd != nil {
		mod = mod.WhereLTE(dao.TMessageLog.Columns().SendTime, in.SendTimeEnd.EndOfDay())
	}

	// 兼容旧的单日发送时间查询
	if in.SendTime != nil {
		mod = mod.Where(fmt.Sprintf("DATE(%s) = ?", dao.TMessageLog.Columns().SendTime), in.SendTime.Format("Y-m-d"))
	}

	// 查询用户ID（JSON字段，使用JSON_CONTAINS查询）
	userId := contexts.GetUserId(ctx)
	var sysUser *entity.SysUser
	dao.SysUser.Ctx(ctx).WherePri(userId).Scan(&sysUser)
	userType := sysUser.UserType

	// 1. 查找顶级部门（ParentId = 0）
	var topDept entity.SysDept
	err = dao.SysDept.Ctx(ctx).
		Where(dao.SysDept.Columns().ParentId, 0).
		Scan(&topDept)
	if err != nil {
		return
	}

	// 检查顶级部门是否存在
	if topDept.DeptId > 0 && deptId != topDept.DeptId {
		mod = mod.Where(dao.TMessageLog.Columns().IsRecall, 0)
		mod = mod.Where(fmt.Sprintf("JSON_CONTAINS(%s, CAST(? AS JSON))", dao.TMessageLog.Columns().UserIds), userId)
		if userType == "sys_user" {
			mod = mod.Where(dao.TMessageLog.Columns().SendUserType, 1)
		} else if userType == "app_user" {
			mod = mod.Where(dao.TMessageLog.Columns().SendUserType, 2)
		}
	}

	// 分页
	mod = mod.Page(in.Page, in.PageSize)

	// 排序
	mod = mod.OrderDesc(dao.TMessageLog.Columns().MessageLogId)

	// 操作人摘要信息
	mod = mod.Hook(hook.MemberSummary)

	// 查询数据
	if err = mod.ScanAndCount(&list, &totalCount, false); err != nil {
		err = gerror.Wrap(err, "获取消息日志列表失败，请稍后重试！")
		return
	}
	return
}

// Export 导出消息日志（异步：提交任务后到下载中心取文件）
func (s *sCommonTMessageLog) Export(ctx context.Context, in *model.TMessageLogListParam) (err error) {
	_, err = exportcenter.Submit(ctx, &exportcenter.Request{
		BizKey:     "tMessageLog",
		BizName:    "消息日志",
		PermCode:   "cpm:common:tMessageLog:export",
		QueryParam: in,
		Tags: func() ([]string, error) {
			return convert.GetEntityDescTags(model.TMessageLogListModel{})
		},
		Fetch: func(ctx context.Context, page, pageSize int) (rows []interface{}, total int, err error) {
			query := *in
			query.Page = page
			query.PageSize = pageSize
			list, count, err := s.List(ctx, &query)
			if err != nil {
				return nil, 0, err
			}
			rows = make([]interface{}, 0, len(list))
			for _, item := range list {
				rows = append(rows, item)
			}
			return rows, count, nil
		},
	})
	return err
}

// Edit 修改/新增消息日志
func (s *sCommonTMessageLog) Edit(ctx context.Context, in *model.TMessageLogEditParam) (err error) {
	var (
		pushUserIds []uint64
		pushPayload g.Map
	)

	err = g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) (err error) {

		// 修改
		if in.MessageLogId > 0 {
			tenantId := contexts.GetTenantId(ctx)
			in.TenantId = &tenantId
			updatedBy := contexts.GetUserId(ctx)
			in.UpdatedBy = &updatedBy
			in.UpdatedAt = gtime.Now()
			if _, err = s.Model(ctx).
				Fields(model.TMessageLogUpdateFields{}).
				WherePri(in.MessageLogId).Data(in).Update(); err != nil {
				err = gerror.Wrap(err, "修改消息日志失败，请稍后重试！")
			}
			return
		}

		// 新增
		tenantId := contexts.GetTenantId(ctx)
		in.TenantId = &tenantId
		createdDept := contexts.GetDeptId(ctx)
		in.CreatedDept = &createdDept
		createdBy := contexts.GetUserId(ctx)
		in.CreatedBy = &createdBy
		in.CreatedAt = gtime.Now()
		in.SendTime = gtime.Now()

		result, err := tx.Ctx(ctx).Model(dao.TMessageLog.Table()).
			Fields(model.TMessageLogInsertFields{}).
			Data(in).Insert()
		if err != nil {
			err = gerror.Wrap(err, "新增消息日志失败，请稍后重试！")
			return
		}

		// 获取插入的消息日志ID
		messageLogId, err := result.LastInsertId()
		if err != nil {
			err = gerror.Wrap(err, "获取消息日志ID失败")
			return
		}
		pushPayload = g.Map{
			"messageLogId":   messageLogId,
			"messageType":    gconv.Uint(in.MessageType),
			"sendUserType":   gconv.Uint(in.SendUserType),
			"messageTitle":   gconv.String(in.MessageTitle),
			"messageContent": gconv.String(in.MessageContent),
			"messageLink":    gconv.String(in.MessageLink),
			"sendTime":       in.SendTime,
			"isRecall":       gconv.Uint(in.IsRecall),
			"isRead":         0,
		}

		optionalUserMessageColumns := s.userMessageOptionalColumns(ctx)
		// 根据 user_ids 往用户消息表插入记录
		if in.UserIds != nil && len(in.UserIds.Interfaces()) > 0 {
			userIds := in.UserIds.Interfaces()
			for _, userId := range userIds {
				userIdInt := gconv.Int64(userId)
				if userIdInt <= 0 {
					continue
				}
				userMessageData := g.Map{
					dao.TUserMessage.Columns().MessageLogId: messageLogId,
					dao.TUserMessage.Columns().UserId:       userIdInt,
					dao.TUserMessage.Columns().TenantId:     tenantId,
					dao.TUserMessage.Columns().CreatedDept:  createdDept,
					dao.TUserMessage.Columns().CreatedBy:    createdBy,
					dao.TUserMessage.Columns().CreatedAt:    in.CreatedAt,
				}
				if _, ok := optionalUserMessageColumns["user_rank"]; ok {
					userMessageData["user_rank"] = 0
				}
				if _, ok := optionalUserMessageColumns["title"]; ok {
					userMessageData["title"] = messageLogStringValue(in.MessageTitle)
				}
				if _, ok := optionalUserMessageColumns["content"]; ok {
					userMessageData["content"] = messageLogStringValue(in.MessageContent)
				}
				if _, ok := optionalUserMessageColumns["link"]; ok {
					userMessageData["link"] = messageLogStringValue(in.MessageLink)
				}
				_, err = tx.Ctx(ctx).Model(dao.TUserMessage.Table()).Data(userMessageData).Insert()
				if err != nil {
					err = gerror.Wrap(err, "新增用户消息失败，请稍后重试！")
					return
				}
				pushUserIds = append(pushUserIds, uint64(userIdInt))
			}
		}
		return
	})
	if err != nil {
		return
	}
	for _, userId := range pushUserIds {
		g.Log().Info(ctx, "推送消息到用户 %v", userId)
		websocket.SendToUser(userId, &websocket.WResponse{
			Event: "message:new",
			Data:  pushPayload,
		})
	}
	return
}

func (s *sCommonTMessageLog) userMessageOptionalColumns(ctx context.Context) map[string]struct{} {
	result, err := g.DB().GetAll(ctx, "SHOW COLUMNS FROM "+dao.TUserMessage.Table())
	if err != nil {
		return map[string]struct{}{}
	}
	columns := make(map[string]struct{}, len(result))
	for _, row := range result {
		field := row["Field"].String()
		if field != "" {
			columns[field] = struct{}{}
		}
	}
	return columns
}

func messageLogStringValue(value *string) string {
	if value == nil {
		return ""
	}
	return *value
}

// Delete 删除消息日志
func (s *sCommonTMessageLog) Delete(ctx context.Context, in *model.TMessageLogDeleteParam) (err error) {

	if _, err = s.Model(ctx).WherePri(in.MessageLogId).Unscoped().Delete(); err != nil {
		err = gerror.Wrap(err, "删除消息日志失败，请稍后重试！")
		return
	}
	return
}

// View 获取消息日志指定信息，并标记已读
func (s *sCommonTMessageLog) View(ctx context.Context, in *model.TMessageLogViewParam) (res *model.TMessageLogViewModel, err error) {
	if err = dao.TMessageLog.Ctx(ctx).WherePri(in.MessageLogId).Hook(hook.MemberSummary).Scan(&res); err != nil {
		err = gerror.Wrap(err, "获取消息日志信息，请稍后重试！")
		return
	}

	// 标记用户消息为已读
	userId := contexts.GetUserId(ctx)
	now := gtime.Now()
	_, err = dao.TUserMessage.Ctx(ctx).Where(g.Map{
		dao.TUserMessage.Columns().MessageLogId: in.MessageLogId,
		dao.TUserMessage.Columns().UserId:       userId,
		dao.TUserMessage.Columns().IsRead:       0,
	}).Data(g.Map{
		dao.TUserMessage.Columns().IsRead:    1,
		dao.TUserMessage.Columns().UpdatedBy: userId,
		dao.TUserMessage.Columns().UpdatedAt: now,
	}).Update()
	if err != nil {
		// 已读标记失败不影响主业务，仅记录日志
		g.Log().Warning(ctx, "标记消息已读失败: %v", err)
		err = nil
	}

	return
}

// 未读通知的默认与最大返回条数，避免一次拉取过多消息。
const (
	messageUnreadDefaultLimit = 50
	messageUnreadMaxLimit     = 200
)

// unreadQuery 构造当前用户未读消息查询。
// 以用户消息表为准：只查本人的未读记录，关联消息日志排除已撤回消息，
// 消息日志已删除（关联为空）时也不提醒。
// 这里不按 send_user_type 过滤：收件人 user_id 全部来自 sys_user 单表，
// 该字段只表示消息面向哪个控制台，按其过滤会漏掉真正发给本人的消息。
func (s *sCommonTMessageLog) unreadQuery(ctx context.Context, userId int64) *gdb.Model {
	var (
		userTable   = dao.TUserMessage.Table()
		logTable    = dao.TMessageLog.Table()
		userColumns = dao.TUserMessage.Columns()
		logColumns  = dao.TMessageLog.Columns()
	)
	return dao.TUserMessage.Ctx(ctx).
		LeftJoin(logTable+" ON "+logTable+"."+logColumns.MessageLogId+" = "+userTable+"."+userColumns.MessageLogId).
		Where(userTable+"."+userColumns.UserId, userId).
		Where(userTable+"."+userColumns.IsRead, 0).
		Where(logTable+"."+logColumns.IsRecall, 0)
}

// UnreadList 获取当前登录用户的未读消息通知。
// 这是右上角消息提醒的数据来源：登录、刷新、断线重连后都必须重新拉取，
// 否则离线期间的消息会因错过 websocket 推送而不再出现提醒。
func (s *sCommonTMessageLog) UnreadList(ctx context.Context, in *model.TMessageLogUnreadListParam) (list []*model.TMessageLogUnreadListModel, totalCount int, err error) {
	userId := contexts.GetUserId(ctx)
	if userId <= 0 {
		err = gerror.New("未获取到当前登录用户信息")
		return
	}

	limit := messageUnreadDefaultLimit
	if in != nil && in.Limit > 0 {
		limit = in.Limit
	}
	if limit > messageUnreadMaxLimit {
		limit = messageUnreadMaxLimit
	}

	var (
		userTable  = dao.TUserMessage.Table()
		logTable   = dao.TMessageLog.Table()
		userColumn = dao.TUserMessage.Columns()
		logColumn  = dao.TMessageLog.Columns()
	)
	if err = s.unreadQuery(ctx, userId).
		Fields(userTable + "." + userColumn.MessageLogId + "," +
			logTable + "." + logColumn.MessageType + "," +
			logTable + "." + logColumn.SendUserType + "," +
			logTable + "." + logColumn.MessageTitle + "," +
			logTable + "." + logColumn.MessageContent + "," +
			logTable + "." + logColumn.MessageLink + "," +
			logTable + "." + logColumn.SendTime + "," +
			userTable + "." + userColumn.IsRead).
		OrderDesc(userTable + "." + userColumn.MessageId).
		Limit(limit).
		Scan(&list); err != nil {
		err = gerror.Wrap(err, "获取未读消息失败，请稍后重试！")
		return
	}
	if list == nil {
		list = []*model.TMessageLogUnreadListModel{}
	}
	// 总数不受列表条数上限影响，用于判断是否还有更多未读。
	if totalCount, err = s.unreadQuery(ctx, userId).Count(); err != nil {
		err = gerror.Wrap(err, "获取未读消息数量失败，请稍后重试！")
	}
	return
}

// Read 标记当前登录用户的指定消息为已读。
// 只允许操作本人的用户消息记录，避免越权修改他人已读状态。
func (s *sCommonTMessageLog) Read(ctx context.Context, in *model.TMessageLogReadParam) (err error) {
	userId := contexts.GetUserId(ctx)
	if userId <= 0 {
		return gerror.New("未获取到当前登录用户信息")
	}
	if in == nil || in.MessageLogId <= 0 {
		return gerror.New("消息ID不能为空")
	}
	if _, err = dao.TUserMessage.Ctx(ctx).Where(g.Map{
		dao.TUserMessage.Columns().MessageLogId: in.MessageLogId,
		dao.TUserMessage.Columns().UserId:       userId,
	}).Data(g.Map{
		dao.TUserMessage.Columns().IsRead:    1,
		dao.TUserMessage.Columns().UpdatedBy: userId,
		dao.TUserMessage.Columns().UpdatedAt: gtime.Now(),
	}).Update(); err != nil {
		err = gerror.Wrap(err, "标记消息已读失败，请稍后重试！")
	}
	return
}

// ReadAll 标记当前登录用户的全部未读消息为已读。
func (s *sCommonTMessageLog) ReadAll(ctx context.Context, in *model.TMessageLogReadAllParam) (err error) {
	userId := contexts.GetUserId(ctx)
	if userId <= 0 {
		return gerror.New("未获取到当前登录用户信息")
	}
	if _, err = dao.TUserMessage.Ctx(ctx).
		Where(dao.TUserMessage.Columns().UserId, userId).
		Where(dao.TUserMessage.Columns().IsRead, 0).
		Data(g.Map{
			dao.TUserMessage.Columns().IsRead:    1,
			dao.TUserMessage.Columns().UpdatedBy: userId,
			dao.TUserMessage.Columns().UpdatedAt: gtime.Now(),
		}).Update(); err != nil {
		err = gerror.Wrap(err, "标记全部消息已读失败，请稍后重试！")
	}
	return
}