77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package global
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"errors"
|
|
|
|
"git.u8t.cn/open/goutil"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
var (
|
|
corpId = 1000
|
|
)
|
|
|
|
type VerifyCode struct {
|
|
Username string `json:"username"`
|
|
Timestamp string `json:"timestamp"`
|
|
}
|
|
type StaffInfo struct {
|
|
Username string `json:"username"`
|
|
Realname string `json:"realname"`
|
|
Phone string `json:"phone"`
|
|
}
|
|
type StaffUser struct {
|
|
Enterprise
|
|
}
|
|
|
|
func NewStaffUser(baseUrl, token string) *StaffUser {
|
|
return &StaffUser{
|
|
Enterprise: Enterprise{
|
|
baseUrl: baseUrl,
|
|
token: token,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (e *StaffUser) SendVerifyCode(username, scene string) (*VerifyCode, error) {
|
|
var reqBody string
|
|
reqBody = fmt.Sprintf(`{"username":"%s","scene":"%s"}`, username, scene)
|
|
reqUrl := e.GetBaseUrl() + "/api/staff/verify/code"
|
|
|
|
body, err := goutil.HttpPost(reqUrl, e.GetHeader(), []byte(reqBody))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
g := gjson.ParseBytes(body)
|
|
if g.Get("code").Int() != 0 {
|
|
return nil, errors.New(string(body))
|
|
}
|
|
r := new(VerifyCode)
|
|
r.Timestamp = g.Get("data.timestamp").String()
|
|
r.Username = g.Get("data.username").String()
|
|
|
|
return r, nil
|
|
}
|
|
|
|
func (e *StaffUser) Login(username, timestamp, code string) (*StaffInfo, error) {
|
|
reqBody := fmt.Sprintf(`{"username":"%s","timestamp":"%s","code":"%s"}`, username, timestamp, code)
|
|
reqUrl := e.GetBaseUrl() + "/api/staff/login"
|
|
body, err := goutil.HttpPost(reqUrl, e.GetHeader(), []byte(reqBody))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
g := gjson.ParseBytes(body)
|
|
if g.Get("code").Int() != 0 {
|
|
return nil, errors.New(string(body))
|
|
}
|
|
|
|
r := new(StaffInfo)
|
|
r.Realname = g.Get("data.realname").String()
|
|
r.Username = g.Get("data.username").String()
|
|
r.Phone = g.Get("data.phone").String()
|
|
|
|
return r, nil
|
|
}
|