package main

import (
   "github.com/goflyfox/gtoken/gtoken"
   "github.com/gogf/gf/frame/g"
   "github.com/gogf/gf/net/ghttp"
   "github.com/gogf/gf/os/glog"
)

var TestServerName string

//var TestServerName string = "gtoken"

func main() {
   glog.Info("########service start...")

   g.Cfg().SetPath("example/sample")
   s := g.Server(TestServerName)
   initRouter(s)

   glog.Info("########service finish.")
   s.Run()
}

var gfToken *gtoken.GfToken

/*
統(tǒng)一路由注冊(cè)
*/
func initRouter(s *ghttp.Server) {
   // 不認(rèn)證接口
   s.Group("/", func(group *ghttp.RouterGroup) {
      group.Middleware(CORS)

      // 調(diào)試路由
      group.ALL("/hello", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("hello"))
      })
   })

   // 認(rèn)證接口
   loginFunc := Login
   // 啟動(dòng)gtoken
   gfToken := &gtoken.GfToken{
      ServerName:       TestServerName,
      LoginPath:        "/login",
      LoginBeforeFunc:  loginFunc,
      LogoutPath:       "/user/logout",
      AuthExcludePaths: g.SliceStr{"/user/info", "/system/user/info"}, // 不攔截路徑 /user/info,/system/user/info,/system/user,
      MultiLogin:       g.Config().GetBool("gToken.MultiLogin"),
   }
   s.Group("/", func(group *ghttp.RouterGroup) {
      group.Middleware(CORS)
      gfToken.Middleware(group)

      group.ALL("/system/user", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("system user"))
      })
      group.ALL("/user/data", func(r *ghttp.Request) {
         r.Response.WriteJson(gfToken.GetTokenData(r))
      })
      group.ALL("/user/info", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("user info"))
      })
      group.ALL("/system/user/info", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("system user info"))
      })
   })

   // 啟動(dòng)gtoken
   gfAdminToken := &gtoken.GfToken{
      ServerName: TestServerName,
      //Timeout:         10 * 1000,
      LoginPath:        "/login",
      LoginBeforeFunc:  loginFunc,
      LogoutPath:       "/user/logout",
      AuthExcludePaths: g.SliceStr{"/admin/user/info", "/admin/system/user/info"}, // 不攔截路徑 /user/info,/system/user/info,/system/user,
      MultiLogin:       g.Config().GetBool("gToken.MultiLogin"),
   }
   s.Group("/admin", func(group *ghttp.RouterGroup) {
      group.Middleware(CORS)
      gfAdminToken.Middleware(group)

      group.ALL("/system/user", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("system user"))
      })
      group.ALL("/user/info", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("user info"))
      })
      group.ALL("/system/user/info", func(r *ghttp.Request) {
         r.Response.WriteJson(gtoken.Succ("system user info"))
      })
   })
}

func Login(r *ghttp.Request) (string, interface{}) {
   username := r.GetString("username")
   passwd := r.GetString("passwd")

   if username == "" || passwd == "" {
      r.Response.WriteJson(gtoken.Fail("賬號(hào)或密碼錯(cuò)誤."))
      r.ExitAll()
   }

   return username, "1"
   /**
   返回的第一個(gè)參數(shù)對(duì)應(yīng)：userKey
   返回的第二個(gè)參數(shù)對(duì)應(yīng)：data
   {
       "code": 0,
       "msg": "success",
       "data": {
           "createTime": 1652838582190,
           "data": "1",
           "refreshTime": 1653270582190,
           "userKey": "王中陽(yáng)",
           "uuid": "ac75676efeb906f9959cf35f779a1d38"
       }
   }
   */
}

// 跨域
func CORS(r *ghttp.Request) {
   r.Response.CORSDefault()
   r.Middleware.Next()
}

// GfToken gtoken結(jié)構(gòu)體
type GfToken struct {
   // GoFrame server name
   ServerName string
   // 緩存模式 1 gcache 2 gredis 默認(rèn)1
   CacheMode int8
   // 緩存key
   CacheKey string
   // 超時(shí)時(shí)間 默認(rèn)10天（毫秒）
   Timeout int
   // 緩存刷新時(shí)間 默認(rèn)為超時(shí)時(shí)間的一半（毫秒）
   MaxRefresh int
   // Token分隔符
   TokenDelimiter string
   // Token加密key
   EncryptKey []byte
   // 認(rèn)證失敗中文提示
   AuthFailMsg string
   // 是否支持多端登錄，默認(rèn)false
   MultiLogin bool
   // 是否是全局認(rèn)證，兼容歷史版本，已廢棄
   GlobalMiddleware bool
   // 中間件類型 1 GroupMiddleware 2 BindMiddleware  3 GlobalMiddleware
   MiddlewareType uint

   // 登錄路徑
   LoginPath string
   // 登錄驗(yàn)證方法 return userKey 用戶標(biāo)識(shí) 如果userKey為空，結(jié)束執(zhí)行
   LoginBeforeFunc func(r *ghttp.Request) (string, interface{})
   // 登錄返回方法
   LoginAfterFunc func(r *ghttp.Request, respData Resp)
   // 登出地址
   LogoutPath string
   // 登出驗(yàn)證方法 return true 繼續(xù)執(zhí)行，否則結(jié)束執(zhí)行
   LogoutBeforeFunc func(r *ghttp.Request) bool
   // 登出返回方法
   LogoutAfterFunc func(r *ghttp.Request, respData Resp)

   // 攔截地址
   AuthPaths g.SliceStr
   // 攔截排除地址
   AuthExcludePaths g.SliceStr
   // 認(rèn)證驗(yàn)證方法 return true 繼續(xù)執(zhí)行，否則結(jié)束執(zhí)行
   AuthBeforeFunc func(r *ghttp.Request) bool
   // 認(rèn)證返回方法
   AuthAfterFunc func(r *ghttp.Request, respData Resp)
}