package auth

import (
	"gitee.com/h79/goutils/auth/token"
	"gitee.com/h79/goutils/common/result"
)

type authMap map[string]token.Authenticate

var (
	_ token.Authenticate = (*route)(nil)
	_ token.Route        = (*route)(nil)
)

type route struct {
	auths authMap
}

func (auth *route) HasAuth(method string) bool {
	_, ok := auth.auths[method]
	return ok
}

func (auth *route) Use(a token.Authenticate) token.Route {
	auth.auths[a.Type()] = a
	return auth
}

func (auth *route) Auth() token.Authenticate {
	return auth
}

func (auth *route) Type() string {
	return "engine"
}

func (auth *route) Create(source token.Source, expireSeconds int64, opts ...token.Option) (token.Token, error) {
	item, err := auth.useAuth(source)
	if err != nil {
		return nil, err
	}
	return item.Create(source, expireSeconds, opts...)
}

func (auth *route) Decode(source token.Source, tok string, opts ...token.Option) (token.Token, error) {
	item, err := auth.useAuth(source)
	if err != nil {
		return nil, err
	}
	return item.Decode(source, tok, opts...)
}

func (auth *route) Check(source token.Source, tok string, opts ...token.Option) (token.Token, error) {
	item, err := auth.useAuth(source)
	if err != nil {
		return nil, err
	}
	return item.Check(source, tok, opts...)
}

func (auth *route) EnableFlag(source token.Source) int {
	item, err := auth.useAuth(source)
	if err != nil {
		return 2
	}
	return item.EnableFlag(source)
}

func (auth *route) SetEnabled(source token.Source, enable bool) {
	item, err := auth.useAuth(source)
	if err != nil {
		return
	}
	item.SetEnabled(source, enable)
}

func (auth *route) useAuth(source token.Source) (token.Authenticate, error) {
	if len(auth.auths) <= 0 {
		return nil, result.Error(result.ErrNil, "Not exist auth")
	}
	item, ok := auth.auths[source.Method]
	if !ok {
		return nil, result.Errorf(result.ErrNotFound, "NOT exist auth for source= '%+v'", source)
	}
	return item, nil
}