267 lines
7.8 KiB
Go
267 lines
7.8 KiB
Go
package utils
|
||
|
||
import (
|
||
"encoding/binary"
|
||
"log"
|
||
"os"
|
||
"testing"
|
||
)
|
||
|
||
type (
|
||
// BaiDuError 错误提示
|
||
BaiDuError struct {
|
||
Error string `json:"error"`
|
||
ErrorDescription string `json:"error_description"`
|
||
}
|
||
// BaiDuAccessToken AccessToken信息
|
||
BaiDuAccessToken struct {
|
||
RefreshToken string `json:"refresh_token"`
|
||
ExpiresIn int `json:"expires_in"`
|
||
Scope string `json:"scope"`
|
||
SessionKey string `json:"session_key"`
|
||
AccessToken string `json:"access_token"`
|
||
SessionSecret string `json:"session_secret"`
|
||
BaiDuError
|
||
}
|
||
// BaiDuSpeechQuick SpeechQuick信息
|
||
BaiDuSpeechQuick struct {
|
||
ErrNo int `json:"err_no"`
|
||
ErrMsg string `json:"err_msg"`
|
||
SN string `json:"sn"`
|
||
Result []string `json:"result"`
|
||
}
|
||
// BaiDuRobotDialogue 机器人对话
|
||
BaiDuRobotDialogue struct {
|
||
ErrorCode int `json:"error_code"`
|
||
ErrorMsg string `json:"error_msg"`
|
||
Result struct {
|
||
Version string `json:"version"`
|
||
ServiceID string `json:"service_id"`
|
||
LogID string `json:"log_id"`
|
||
InteractionID string `json:"interaction_id"`
|
||
Response struct {
|
||
Status int `json:"status"`
|
||
Msg string `json:"msg"`
|
||
ActionList []struct {
|
||
Confidence float64 `json:"confidence"`
|
||
ActionID string `json:"action_id"`
|
||
Say string `json:"say"`
|
||
CustomReply string `json:"custom_reply"`
|
||
Type string `json:"type"`
|
||
} `json:"action_list"`
|
||
Schema struct {
|
||
Confidence float64 `json:"confidence"`
|
||
Intent string `json:"intent"`
|
||
IntentConfidence float64 `json:"intent_confidence"`
|
||
Slots []struct {
|
||
Confidence float64 `json:"confidence"`
|
||
Begin int `json:"begin"`
|
||
Length int `json:"length"`
|
||
OriginalWord string `json:"original_word"`
|
||
NormalizedWord string `json:"normalized_word"`
|
||
WordType string `json:"word_type"`
|
||
Name string `json:"name"`
|
||
SessionOffset string `json:"session_offset"`
|
||
MergeMethod string `json:"merge_method"`
|
||
} `json:"slots"`
|
||
} `json:"schema"`
|
||
} `json:"response"`
|
||
} `json:"result"`
|
||
}
|
||
)
|
||
|
||
const (
|
||
baiDuClientID string = "Sy0tLT7bHWE2RhollVOqelHq"
|
||
baiDuClientSecret string = "jSr0a2Isaivi1yvgk2TXlB7tqg21Gf1m"
|
||
//baiDuClientID string = "MDNsII2jkUtbF729GQOZt7FS"
|
||
//baiDuClientSecret string = "0vWCVCLsbWHMSH1wjvxaDq4VmvCZM2O9"
|
||
|
||
// baiDuRequestURLForAccessToken 获取token地址
|
||
baiDuRequestURLForAccessToken string = "https://aip.baidubce.com/oauth/2.0/token"
|
||
// baiDuRequestURLForVoiceQuick 语音识别极速版
|
||
baiDuRequestURLForVoiceQuick string = "https://vop.baidu.com/pro_api"
|
||
// baiDuRequestURLForRobotDialogue 语音机器人对话
|
||
baiDuRequestURLForRobotDialogue string = "https://aip.baidubce.com/rpc/2.0/unit/bot/chat"
|
||
//baiDuRequestURLForRobotDialogue string = "https://aip.baidubce.com/rpc/2.0/unit/service/chat"
|
||
)
|
||
|
||
func TestNewClient(t *testing.T) {
|
||
//file := "../upload/20210624/16k1.pcm"
|
||
//
|
||
client2 := NewClient(baiDuRequestURLForAccessToken, MethodForPost, map[string]interface{}{
|
||
"grant_type": "client_credentials", "client_id": baiDuClientID, "client_secret": baiDuClientSecret,
|
||
})
|
||
resp2, err := client2.Request(RequestBodyFormatForFormData)
|
||
|
||
if err != nil {
|
||
t.Log(err)
|
||
return
|
||
}
|
||
response := new(BaiDuAccessToken)
|
||
|
||
_ = FromJSONBytes(resp2, response)
|
||
|
||
if response.Error != "" {
|
||
t.Logf("获取百度AccessToken错误：%v - %v", response.Error, response.ErrorDescription)
|
||
return
|
||
}
|
||
//t.Log(response.AccessToken)
|
||
//
|
||
//reader, err := os.OpenFile(file, os.O_RDONLY, 0666)
|
||
//
|
||
//if err != nil {
|
||
// t.Log(err)
|
||
// return
|
||
//}
|
||
//defer reader.Close()
|
||
//
|
||
//content, _ := ioutil.ReadAll(reader)
|
||
//
|
||
//cuid := ""
|
||
//
|
||
//netitfs, err := net.Interfaces()
|
||
//
|
||
//if err != nil {
|
||
// cuid = "anonymous_sqzn"
|
||
//} else {
|
||
// for _, itf := range netitfs {
|
||
// if cuid = itf.HardwareAddr.String(); len(cuid) > 0 {
|
||
// break
|
||
// }
|
||
// }
|
||
//}
|
||
//t.Log(cuid)
|
||
//t.Log(base64.StdEncoding.EncodeToString(content))
|
||
//t.Log(fmt.Sprintf("%d", len(content)))
|
||
//
|
||
//_params := map[string]interface{}{
|
||
// "format": file[len(file)-3:],
|
||
// "rate": 16000,
|
||
// "dev_pid": 1537,
|
||
// "channel": 1,
|
||
// "token": "24.1f876b06d070d7403c90832dddb813cb.2592000.1627110943.282335-24431674",
|
||
// "cuid": cuid,
|
||
// "speech": base64.StdEncoding.EncodeToString(content),
|
||
// "len": len(content),
|
||
//}
|
||
//_json, _ := json.Marshal(_params)
|
||
//
|
||
//req, err := http.NewRequest("GET", "http://vop.baidu.com/server_api", bytes.NewBuffer(_json))
|
||
//
|
||
//if err != nil {
|
||
// t.Log(err)
|
||
// return
|
||
//}
|
||
//resp := new(http.Response)
|
||
//
|
||
//client := new(http.Client)
|
||
//
|
||
//if resp, err = client.Do(req); err != nil {
|
||
// t.Log(err)
|
||
// return
|
||
//}
|
||
//bytes, err := ioutil.ReadAll(resp.Body)
|
||
//defer resp.Body.Close()
|
||
//
|
||
//response1 := new(BaiDuSpeechQuick)
|
||
//
|
||
//_ = FromJSONBytes(bytes, response1)
|
||
//
|
||
//t.Logf("resp：%v\n", AnyToJSON(response1))
|
||
|
||
serviceID := "1101579"
|
||
|
||
params2 := map[string]interface{}{
|
||
"version": "2.0",
|
||
//"service_id": serviceID,
|
||
"bot_id": serviceID,
|
||
"log_id": Md5String(AnyToJSON(2040256374931197952), serviceID),
|
||
"session_id": Sha256String(AnyToJSON(2040256374931197952) + serviceID),
|
||
"request": map[string]interface{}{
|
||
"query": "公告",
|
||
"user_id": AnyToJSON(2040256374931197952),
|
||
"query_info": map[string]interface{}{
|
||
"asr_candidates": []string{},
|
||
"source": "KEYBOARD",
|
||
"type": "TEXT",
|
||
},
|
||
},
|
||
"bernard_level": 1,
|
||
}
|
||
t.Log(params2)
|
||
client3 := NewClient(baiDuRequestURLForRobotDialogue+"?access_token="+response.AccessToken,
|
||
MethodForPost, params2)
|
||
|
||
resp3, err := client3.Request(RequestBodyFormatForRaw, Headers{ContentType: "application/json; charset=UTF-8"})
|
||
|
||
if err != nil {
|
||
t.Log(err)
|
||
return
|
||
}
|
||
response3 := new(BaiDuRobotDialogue)
|
||
|
||
_ = FromJSONBytes(resp3, response3)
|
||
|
||
t.Log(AnyToJSON(response3))
|
||
|
||
return
|
||
//
|
||
//client1 := NewClient("http://vop.baidu.com/server_api", MethodForPost, params)
|
||
//
|
||
//resp1 := make([]byte, 0)
|
||
//
|
||
//if resp1, err = client1.Request(RequestBodyFormatForRaw, Headers{
|
||
// ContentType: "application/json",
|
||
//}); err != nil {
|
||
// t.Log(err)
|
||
// return
|
||
//}
|
||
//response1 := new(BaiDuSpeechQuick)
|
||
//
|
||
//_ = FromJSONBytes(resp1, response1)
|
||
//
|
||
//t.Logf("resp：%v\n", AnyToJSON(response1))
|
||
}
|
||
|
||
func TestClient_Request(t *testing.T) {
|
||
request := NewClient("https://image1.ljcdn.com/hdic-resblock/4494aa6e-4165-4f4a-b7ba-4ab095dd1ffa.JPG.710x400.jpg", "GET", nil)
|
||
|
||
resp, err := request.Request(RequestBodyFormatForFormData, Headers{
|
||
Others: map[string]string{
|
||
"Referer": "http://drc.hefei.gov.cn/group4/M00/07/4D/wKgEIWEM9X-AXLhsAAONk965l5o088.png",
|
||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.72 Safari/537.36",
|
||
"Cookie": "__jsluid_h=9fcece02433a024c638dd5e8d4cf6f92; __jsl_clearance=1628842172.968|0|8rBRZzH5SoW3MMG1%2FWkYpLUeRXA%3D",
|
||
},
|
||
})
|
||
f, err := os.Create("test.jpg")
|
||
if err != nil {
|
||
log.Fatal("Couldn't open file")
|
||
}
|
||
defer f.Close()
|
||
|
||
err = binary.Write(f, binary.LittleEndian, resp)
|
||
|
||
if err != nil {
|
||
log.Fatal("Write failed")
|
||
}
|
||
|
||
//resp, err := request.Request(RequestBodyFormatForFormData, Headers{
|
||
// //Others: map[string]string{
|
||
// // "Referer": "http://drc.hefei.gov.cn/group4/M00/07/4D/wKgEIWEM9X-AXLhsAAONk965l5o088.png",
|
||
// // "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.72 Safari/537.36",
|
||
// // "Cookie": "__jsluid_h=2844bb494bad8b1cd372e28c65844abd; __jsl_clearance=1628840623.068|0|vNUfDD1V4muQrHrWy%2BmhoGbOFr0%3D",
|
||
// //},
|
||
//})
|
||
//f, err := os.Create("test.jpg")
|
||
//if err != nil {
|
||
// log.Fatal("Couldn't open file")
|
||
//}
|
||
//defer f.Close()
|
||
//
|
||
//err = binary.Write(f, binary.LittleEndian, resp)
|
||
//
|
||
//if err != nil {
|
||
// log.Fatal("Write failed")
|
||
//}
|
||
}
|