85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package video_hat
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/tiger1103/gfast/v3/api/constant"
|
|
)
|
|
|
|
type RefreshTokenReq struct {
|
|
g.Meta `path:"/video/refreshToken" method:"post" tags:"视频安全帽相关" summary:"刷新Token(不需要前端调用)"`
|
|
}
|
|
|
|
type RefreshTokenRes struct{}
|
|
|
|
func (v VideoHat) RefreshToken(ctx context.Context, req *RefreshTokenReq) (res *RefreshTokenRes, err error) {
|
|
res = new(RefreshTokenRes)
|
|
InitToken()
|
|
return res, nil
|
|
}
|
|
|
|
// 用于接收获取 Token 的响应
|
|
type TokenRes struct {
|
|
Status bool `json:"status"`
|
|
Token string `json:"token"`
|
|
SessionId string `json:"session_id"`
|
|
MsgCode string `json:"msg_code"`
|
|
}
|
|
|
|
type PKeyResponse struct {
|
|
Status bool `json:"status"`
|
|
Data string `json:"data"`
|
|
Msg string `json:"msg"`
|
|
MsgCode string `json:"msg_code"`
|
|
}
|
|
|
|
type TokenResponse struct {
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
// 发送 POST 请求
|
|
func fetchPostForm(url string, data url.Values) ([]byte, error) {
|
|
resp, err := http.PostForm(url, data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
return io.ReadAll(resp.Body)
|
|
}
|
|
|
|
// InitToken 项目初始化启动获取 Token【以及刷新 Token】
|
|
func InitToken() error {
|
|
data := url.Values{}
|
|
data.Set("user_name", constant.Username)
|
|
data.Set("pwd", constant.Password)
|
|
|
|
respBody, err := fetchPostForm("https://caps.runde.pro/api/index.php?ctl=tool&act=get_pkey", data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var pkeyRes PKeyResponse
|
|
if err := json.Unmarshal(respBody, &pkeyRes); err != nil {
|
|
return err
|
|
}
|
|
|
|
data.Set("pkey", pkeyRes.Data)
|
|
respBody, err = fetchPostForm("https://caps.runde.pro/api/index.php?ctl=tool&act=get_token", data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var tokenRes TokenResponse
|
|
if err := json.Unmarshal(respBody, &tokenRes); err != nil {
|
|
return err
|
|
}
|
|
|
|
constant.Token = tokenRes.Token
|
|
return nil
|
|
}
|