package weixin

import (
	"gitee.com/h79/gothird/result"
	"gitee.com/h79/gothird/token"
	"gitee.com/h79/gothird/weixin/errors"
	"gitee.com/h79/gothird/weixin/oauth2"
	"gitee.com/h79/gothird/weixin/offia/ticket"
	"gitee.com/h79/gothird/weixin/session"
	"gitee.com/h79/goutils/common/data"
	"mime/multipart"
	http2 "net/http"
)

var firstService *acsReq
var webService *acsReq

func init() {
	firstService = &acsReq{
		get:    GetAccessToken,
		ref:    token.GetEmpty,
		auth:   oauth2.GetAccessToken,
		create: token.CreateChild,
	}

	webService = &acsReq{
		get:    token.GetEmpty,
		ref:    oauth2.RefreshToken,
		auth:   token.GetEmpty,
		create: token.CreateChildEmpty,
	}
}

func NewToken(app token.App) token.Token {
	return token.New(&app, firstService)
}

type acsReq struct {
	get    token.GetFunc
	ref    token.GetFunc
	auth   token.GetFunc
	create token.CreateChildFunc
}

// Execute token.Service interface
func (req *acsReq) Execute(tk token.Token, cmd string, d data.D) (interface{}, error) {

	if cmd == token.NTicket {
		return ticket.GetTicket(tk, d)
	}
	if cmd == token.NAccessToken {
		return req.get(tk, d)
	}
	if cmd == token.NRefreshToken {
		return req.ref(tk, d)
	}
	if cmd == token.NOauth2 { // 网页，个人授权，返回一个 openid
		acs, err := req.auth(tk, d)
		if err != nil {
			return nil, err
		}
		if se, ok := acs.(*session.Session); ok {
			ch := &token.Child{
				AliasId: se.Openid,
				Name:    "web",
				VMap:    token.VMap{},
			}
			token.SyncCreateChild(tk, webService, ch, acs) // 需要把个人openid 与 appid 对应
			token.SyncAddAlias(tk, []string{se.Openid}, token.AliasId(se.Openid))
		}
		return acs, err
	}
	if cmd == token.NCreateChild {
		if ch, ok := d.Value("child").(*token.Child); ok {
			return req.create(tk, webService, ch, nil)
		}
	}
	return nil, result.ErrNotSupported
}

func (req *acsReq) GetUrl(tk token.Token, acsKey string, uri string) string {
	return token.DefaultUrl(acsKey, uri)
}

func (req *acsReq) SetHead(tk token.Token, acsKey string, h *http2.Header) {
}

func (req *acsReq) CreateForm(w *multipart.Writer, form interface{}) error {
	return token.DefaultForm(w, "media", form.(string))
}

func (req *acsReq) CheckError(err error) error {
	if res, ok := err.(result.Result); ok {
		if errors.IsTokenExpired(res.Code) {
			return result.ErrExpired
		}
	}
	return err
}