hk4e/dispatch/controller/controller.go

package controller

import (
	"encoding/base64"
	"encoding/binary"
	"github.com/nats-io/nats.go"
	"net/http"
	"strconv"

	"hk4e/common/config"
	"hk4e/common/region"
	"hk4e/dispatch/dao"
	"hk4e/pkg/logger"

	"github.com/gin-gonic/gin"
	pb "google.golang.org/protobuf/proto"
)

type Controller struct {
	dao              *dao.Dao
	regionListBase64 string
	regionCurrBase64 string
	signRsaKey       []byte
	encRsaKeyMap     map[string][]byte
	pwdRsaKey        []byte
}

func NewController(dao *dao.Dao) (r *Controller) {
	r = new(Controller)
	r.dao = dao
	r.regionListBase64 = ""
	r.regionCurrBase64 = ""
	regionCurr, regionList, dispatchEc2b := region.InitRegion(config.CONF.Hk4e.KcpAddr, config.CONF.Hk4e.KcpPort)

	// TODO 临时写一下用来传递新的密钥后面改RPC
	conn, err := nats.Connect(config.CONF.MQ.NatsUrl)
	if err != nil {
		logger.LOG.Error("connect nats error: %v", err)
		return nil
	}
	natsMsg := nats.NewMsg("GATE_KEY_HK4E")
	natsMsg.Data = make([]byte, 8)
	dispatchEc2bSeed := dispatchEc2b.Seed()
	binary.BigEndian.PutUint64(natsMsg.Data, dispatchEc2bSeed)
	err = conn.PublishMsg(natsMsg)
	if err != nil {
		logger.LOG.Error("nats publish msg error: %v", err)
		return nil
	}
	conn.Close()
	logger.LOG.Debug("send new dispatch ec2b seed: %v", dispatchEc2bSeed)

	r.signRsaKey, r.encRsaKeyMap, r.pwdRsaKey = region.LoadRsaKey()
	regionCurrModify, err := pb.Marshal(regionCurr)
	if err != nil {
		logger.LOG.Error("Marshal QueryCurrRegionHttpRsp error")
		return nil
	}
	r.regionCurrBase64 = base64.StdEncoding.EncodeToString(regionCurrModify)
	regionListModify, err := pb.Marshal(regionList)
	if err != nil {
		logger.LOG.Error("Marshal QueryRegionListHttpRsp error")
		return nil
	}
	r.regionListBase64 = base64.StdEncoding.EncodeToString(regionListModify)
	go r.registerRouter()
	return r
}

func (c *Controller) authorize() gin.HandlerFunc {
	return func(context *gin.Context) {
		// TODO auth token或其他验证方式
		ok := true
		if ok {
			context.Next()
			return
		}
		context.Abort()
		context.JSON(http.StatusOK, gin.H{
			"code": "10001",
			"msg":  "没有访问权限",
		})
	}
}

func (c *Controller) registerRouter() {
	if config.CONF.Logger.Level == "DEBUG" {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}
	engine := gin.Default()
	{
		// 404
		engine.NoRoute(func(context *gin.Context) {
			logger.LOG.Info("no route find, fallback to fuck mhy, url: %v", context.Request.RequestURI)
			context.Header("Content-type", "text/html; charset=UTF-8")
			context.Status(http.StatusNotFound)
			_, _ = context.Writer.WriteString("FUCK MHY")
		})
	}
	{
		// 调度
		// dispatchosglobal.yuanshen.com
		engine.GET("/query_security_file", c.query_security_file)
		engine.GET("/query_region_list", c.query_region_list)
		// osusadispatch.yuanshen.com
		engine.GET("/query_cur_region", c.query_cur_region)
	}
	{
		// 登录
		// hk4e-sdk-os.hoyoverse.com
		// 账号登录
		engine.POST("/hk4e_global/mdk/shield/api/login", c.apiLogin)
		// token登录
		engine.POST("/hk4e_global/mdk/shield/api/verify", c.apiVerify)
		// 获取combo token
		engine.POST("/hk4e_global/combo/granter/login/v2/login", c.v2Login)
	}
	{
		// 日志
		engine.POST("/sdk/dataUpload", c.sdkDataUpload)
		engine.GET("/perf/config/verify", c.perfConfigVerify)
		engine.POST("/perf/dataUpload", c.perfDataUpload)
		engine.POST("/log", c.log8888)
		engine.POST("/crash/dataUpload", c.crashDataUpload)
	}
	{
		// 返回固定数据
		// Windows
		engine.GET("/hk4e_global/mdk/agreement/api/getAgreementInfos", c.getAgreementInfos)
		engine.POST("/hk4e_global/combo/granter/api/compareProtocolVersion", c.postCompareProtocolVersion)
		engine.POST("/account/risky/api/check", c.check)
		engine.GET("/combo/box/api/config/sdk/combo", c.combo)
		engine.GET("/hk4e_global/combo/granter/api/getConfig", c.getConfig)
		engine.GET("/hk4e_global/mdk/shield/api/loadConfig", c.loadConfig)
		engine.POST("/data_abtest_api/config/experiment/list", c.list)
		engine.GET("/admin/mi18n/plat_oversea/m2020030410/m2020030410-version.json", c.version10Json)
		engine.GET("/admin/mi18n/plat_oversea/m2020030410/m2020030410-zh-cn.json", c.zhCN10Json)
		engine.GET("/geetestV2.html", c.geetestV2)
		// Android
		engine.POST("/common/h5log/log/batch", c.batch)
		engine.GET("/hk4e_global/combo/granter/api/getFont", c.getFont)
		engine.GET("/admin/mi18n/plat_oversea/m202003049/m202003049-version.json", c.version9Json)
		engine.GET("/admin/mi18n/plat_oversea/m202003049/m202003049-zh-cn.json", c.zhCN9Json)
		engine.GET("/hk4e_global/combo/granter/api/compareProtocolVersion", c.getCompareProtocolVersion)
		// Android geetest
		engine.GET("/gettype.php", c.gettype)
		engine.GET("/get.php", c.get)
		engine.POST("/ajax.php", c.ajax)
		engine.GET("/ajax.php", c.ajax)
		engine.GET("/static/appweb/app3-index.html", c.app3Index)
		engine.GET("/static/js/slide.7.8.6.js", c.slideJs)
		engine.GET("/favicon.ico", c.faviconIco)
		engine.GET("/static/js/gct.e7810b5b525994e2fb1f89135f8df14a.js", c.js)
		engine.GET("/static/ant/style_https.1.2.6.css", c.css)
		engine.GET("/pictures/gt/a330cf996/a330cf996.webp", c.webp)
		engine.GET("/pictures/gt/a330cf996/bg/86f9db021.webp", c.bgWebp)
		engine.GET("/pictures/gt/a330cf996/slice/86f9db021.png", c.slicePng)
		engine.GET("/static/ant/sprite2x.1.2.6.png", c.sprite2xPng)
	}
	engine.Use(c.authorize())
	engine.POST("/gate/token/verify", c.gateTokenVerify)
	port := config.CONF.HttpPort
	addr := ":" + strconv.Itoa(port)
	err := engine.Run(addr)
	if err != nil {
		logger.LOG.Error("gin run error: %v", err)
	}
}