package middleware

import (
	"service/app/mix/service"
	"service/app/mix/service/logic"
	"service/dbstruct"
	"service/library/logger"

	"service/api/consts"
	notificationproto "service/api/proto/notification/proto"

	goproto "google.golang.org/protobuf/proto"

	"github.com/gin-gonic/gin"
)

var (
	DefaultNotifSender gin.HandlerFunc
)

func InitNotifSender(_DefaultNotification *logic.Notification, _DefaultNotifBcstCenter service.NotifBcstCenter) {
	DefaultNotifSender = func(ctx *gin.Context) {
		notifTempId := ctx.MustGet("notif_template_id").(int64)
		notifTempParams := ctx.MustGet("notif_template_params").([]string)

		notification := &dbstruct.Notification{}
		// 通知接收人mids不为空则默认认为是自定义发送
		if len(notif.ObjMids) > 0 {
			notification.ObjType = goproto.Int64(consts.Notification_ObjType_Customized)
		}

		notification.SubMid = goproto.Int64(0)
		notification.PushTime = goproto.Int64(0)

		err := _DefaultNotification.OpCreate(ctx, &notificationproto.OpCreateReq{
			Notification: notification,
		})
		if err != nil {
			logger.Error("通知创建失败:%v", err)
			ctx.Next()
		}

		nids := make([]int64, 0)
		nids = append(nids, notification.GetId())

		err = _DefaultNotifBcstCenter.BcstNotifs(ctx, nids, notification.GetObjType(), notification.ObjMids)
		if err != nil {
			logger.Error("通知广播失败:%v", err)
			ctx.Next()
		}
		ctx.Next()
	}
}

func NotifSender() gin.HandlerFunc {
	return DefaultNotifSender
}