70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package erp
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.u8t.cn/open/goutil"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
type StaffCode struct {
|
|
Timestamp string `json:"timestamp"`
|
|
}
|
|
type StaffInfo struct {
|
|
Username string `json:"username"`
|
|
Realname string `json:"realname"`
|
|
Phone string `json:"phone"`
|
|
}
|
|
type StaffUser struct {
|
|
Erp
|
|
}
|
|
|
|
func NewStaffUser(baseUrl, token string) *StaffUser {
|
|
return &StaffUser{
|
|
Erp: Erp{
|
|
baseUrl: baseUrl,
|
|
token: token,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (e *StaffUser) SendCode(username, scene string) (*StaffCode, error) {
|
|
var reqBody string
|
|
reqBody = fmt.Sprintf(`{"username":"%s","scene":"%s"}`, username, scene)
|
|
reqUrl := e.GetBaseUrl() + "/ext/staff/user/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(StaffCode)
|
|
r.Timestamp = g.Get("data.timestamp").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() + "/ext/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
|
|
}
|