main.go

package main

import (
	"github.com/beego/beego/v2/server/web/context"

	_ "g8ink/models"
	_ "g8ink/routers"
	"g8ink/tools"

	beego "github.com/beego/beego/v2/server/web"
)

var ADMIN_LOGIN_PASS, _ = beego.AppConfig.String("ADMIN_LOGIN_PASS")

func main() {

	//过滤未登录的
	var FilterUser = func(ctx *context.Context) {
		pass, ok := ctx.GetSecureCookie(tools.GetCookiePass(), "Password")
		if !ok || pass != ADMIN_LOGIN_PASS {
			ctx.Abort(404, "404")
		}
	}

	//过滤刷api的
	var FilterTimes = func(ctx *context.Context) {
		if ctx.Input.Method() == "POST" && tools.LimitAccess(ctx.Input.IP()) {
			re := make(map[string]interface{})
			re["Code"] = -2
			re["Message"] = "太快啦~~要被玩坏啦~＞︿＜"
			ctx.Output.JSON(&re, true, false)
		}
	}

	beego.InsertFilter("/admin/"+tools.GetAdminUrl()+"/api/*", beego.BeforeRouter, FilterUser)
	beego.InsertFilter("/admin/"+tools.GetAdminUrl()+"/home", beego.BeforeRouter, FilterUser)
	beego.InsertFilter("/admin/"+tools.GetAdminUrl()+"/links", beego.BeforeRouter, FilterUser)
	beego.InsertFilter("/admin/"+tools.GetAdminUrl()+"/ban", beego.BeforeRouter, FilterUser)
	beego.InsertFilter("/admin/"+tools.GetAdminUrl()+"/limitips", beego.BeforeRouter, FilterUser)
	beego.InsertFilter("/admin/"+tools.GetAdminUrl()+"/whitelist", beego.BeforeRouter, FilterUser)

	beego.InsertFilter("/", beego.BeforeRouter, FilterTimes)

	// orm.Debug = true

	beego.Run()
}