update

libs/auth/auth.go

@@ -0,0 +1,43 @@
+package auth
+
+import (
+	"OneAuth/cfg"
+	"OneAuth/libs/oerr"
+	"OneAuth/models"
+	"github.com/veypi/OneBD"
+	"github.com/veypi/OneBD/rfc"
+)
+
+type Auth struct {
+	Payload      *models.PayLoad
+	ignoreMethod map[rfc.Method]bool
+}
+
+func (a *Auth) Init(m OneBD.Meta) error {
+	if a.ignoreMethod != nil && a.ignoreMethod[m.Method()] {
+		return nil
+	}
+	a.Payload = new(models.PayLoad)
+	token := m.GetHeader("auth_token")
+	if token == "" {
+		return oerr.NotLogin
+	}
+	ok, err := models.ParseToken(token, cfg.CFG.Key, a.Payload)
+	if ok {
+		return nil
+	}
+	return oerr.NotLogin.Attach(err)
+}
+
+func (a *Auth) Ignore(methods ...rfc.Method) {
+	if a.ignoreMethod == nil {
+		a.ignoreMethod = make(map[rfc.Method]bool)
+	}
+	for _, m := range methods {
+		a.ignoreMethod[m] = true
+	}
+}
+
+func (a *Auth) CheckAuth(name string, tags ...string) models.AuthLevel {
+	return a.Payload.CheckAuth(name, tags...)
+}