ai-agent/internal/middleware/chat_middleware.go

package middleware

import (
	"cs-agent/internal/pkg/irisx"
	"cs-agent/internal/pkg/openidentity"
	"cs-agent/internal/services"

	"github.com/kataras/iris/v12"
	"github.com/mlogclub/simple/web"
)

func ExternalInfoMiddleware(ctx iris.Context) {
	channel := services.ChannelService.GetEnabledChannel(ctx)
	if channel == nil {
		ctx.StopExecution()
		_ = ctx.JSON(web.JsonErrorMsg("接入渠道异常"))
		return
	}
	secret := services.ChannelService.GetUserTokenSecret(channel)
	ext, err := openidentity.GetExternalInfo(ctx, secret)
	if err != nil {
		ctx.StopExecution()
		_ = ctx.JSON(web.JsonError(err))
		return
	}
	irisx.SetExternalInfo(ctx, ext)
	ctx.Next()
}