OneAuth/libs/base/user_handler.go

package base

import (
	"github.com/veypi/OneAuth/cfg"
	"github.com/veypi/OneAuth/libs/oerr"
	"github.com/veypi/OneAuth/libs/token"
	"github.com/veypi/OneAuth/oalib"
	"github.com/veypi/OneBD"
	"github.com/veypi/OneBD/rfc"
)

type UserHandler struct {
	Payload      *oalib.PayLoad
	ignoreMethod map[rfc.Method]bool
}

func (a *UserHandler) Init(m OneBD.Meta) error {
	if a.ignoreMethod != nil && a.ignoreMethod[m.Method()] {
		return nil
	}
	return a.ParsePayload(m)
}

func (a *UserHandler) ParsePayload(m OneBD.Meta) error {
	a.Payload = new(oalib.PayLoad)
	tokenStr := m.GetHeader("auth_token")
	if tokenStr == "" {
		return oerr.NotLogin
	}
	ok, err := token.ParseToken(tokenStr, a.Payload, cfg.CFG.APPKey)
	if ok {
		return nil
	}
	return oerr.NotLogin.Attach(err)
}

func (a *UserHandler) 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 *UserHandler) GetAuth(ResourceID string, ResourceUUID ...string) oalib.AuthLevel {
	return a.Payload.GetAuth(ResourceID, ResourceUUID...)
}