115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package laosiji
|
|
|
|
import (
|
|
"91porn-server/common/log"
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
func getComicsSearchUrl() string {
|
|
return fmt.Sprintf("%s/lsjapi/comics/search", APIUrl)
|
|
}
|
|
|
|
func getComicsDetailUrl() string {
|
|
return fmt.Sprintf("%s/lsjapi/comics/detail", APIUrl)
|
|
}
|
|
|
|
func ComicsSearch(ctx context.Context, req ComicsSearchListReq) (resp ComicsSearchListResp, err error) {
|
|
endpoint := getComicsSearchUrl()
|
|
req.Need_total_info = "y"
|
|
var data map[string]interface{}
|
|
data, err = StructToMapViaJSON(req)
|
|
if err != nil {
|
|
log.Error("GenerateUndress post fail", log.E(err))
|
|
return resp, err
|
|
}
|
|
|
|
err = comicsPost(ctx, endpoint, data, &resp)
|
|
if err != nil {
|
|
log.Error("GenerateUndress post fail", log.Any("data", data), log.E(err))
|
|
}
|
|
return
|
|
}
|
|
|
|
func ComicsDetail(ctx context.Context, req ComicsDetailReq) (resp ComicsDetailResp, err error) {
|
|
endpoint := getComicsDetailUrl()
|
|
|
|
var data map[string]interface{}
|
|
data, err = StructToMapViaJSON(req)
|
|
if err != nil {
|
|
log.Error("GenerateUndress post fail", log.E(err))
|
|
return resp, err
|
|
}
|
|
|
|
err = comicsPost(ctx, endpoint, data, &resp)
|
|
if err != nil {
|
|
log.Error("GenerateUndress post fail", log.Any("data", data), log.E(err))
|
|
}
|
|
return
|
|
}
|
|
|
|
func comicsPost(ctx context.Context, endpoint string, req map[string]interface{}, response interface{}) (err error) {
|
|
jsonData, err := json.Marshal(req)
|
|
if err != nil {
|
|
log.Warn("GenerateAiUndress", log.E(err))
|
|
return
|
|
}
|
|
encryptedData, err := encryptBase64(string(jsonData), APIKey)
|
|
if err != nil {
|
|
log.Warn("GenerateAiUndress encryptBase64", log.E(err))
|
|
return
|
|
}
|
|
// 创建 POST 请求
|
|
httpReq, err := http.NewRequest(http.MethodPost, endpoint, bytes.NewBuffer([]byte(encryptedData)))
|
|
if err != nil {
|
|
return
|
|
}
|
|
httpReq.Header.Set("Content-Type", "application/json")
|
|
httpReq.Header.Set("Content-Length", fmt.Sprintf("%d", len(encryptedData)))
|
|
httpReq.Header.Set("appid", Appid)
|
|
|
|
// 发送请求
|
|
client := &http.Client{Timeout: 10 * time.Second}
|
|
resp, err := client.Do(httpReq)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// 读取响应体
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// 检查 HTTP 状态码
|
|
if resp.StatusCode != http.StatusOK {
|
|
return
|
|
}
|
|
// 解析 JSON
|
|
var res Response
|
|
if err = json.Unmarshal(body, &res); err != nil {
|
|
return
|
|
}
|
|
if res.Status != "y" {
|
|
err = fmt.Errorf("errorCode:%v error:%v", res.ErrorCode, res.Error)
|
|
return
|
|
}
|
|
|
|
// 进行解密
|
|
dataStr, err := decryptBase64(res.Data, APIKey)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = json.Unmarshal([]byte(dataStr), response)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|