mirror of
https://github.com/OwO-Network/DeepLX.git
synced 2024-11-16 11:42:33 +08:00
fix: add pro support (may have no effect)
This commit is contained in:
Vincent Yang
parent
9edb997f06
commit
d98f71d1c5
114
main.go
114
main.go
|
|
@ -2,7 +2,7 @@
|
|||
* @Author: Vincent Yang
|
||||
* @Date: 2023-07-01 21:45:34
|
||||
* @LastEditors: Vincent Yang
|
||||
* @LastEditTime: 2024-11-01 00:42:58
|
||||
* @LastEditTime: 2024-11-01 13:04:50
|
||||
* @FilePath: /DeepLX/main.go
|
||||
* @Telegram: https://t.me/missuo
|
||||
* @GitHub: https://github.com/missuo
|
||||
|
|
@ -131,7 +131,7 @@ func main() {
|
|||
return
|
||||
}
|
||||
|
||||
result, err := translate.TranslateByDeepLX(sourceLang, targetLang, translateText, tagHandling, proxyURL)
|
||||
result, err := translate.TranslateByDeepLX(sourceLang, targetLang, translateText, tagHandling, proxyURL, "")
|
||||
if err != nil {
|
||||
log.Fatalf("Translation failed: %s", err)
|
||||
}
|
||||
|
|
@ -156,68 +156,68 @@ func main() {
|
|||
})
|
||||
|
||||
// Pro API endpoint, Pro Account required
|
||||
// r.POST("/v1/translate", authMiddleware(cfg), func(c *gin.Context) {
|
||||
// req := PayloadFree{}
|
||||
// c.BindJSON(&req)
|
||||
r.POST("/v1/translate", authMiddleware(cfg), func(c *gin.Context) {
|
||||
req := PayloadFree{}
|
||||
c.BindJSON(&req)
|
||||
|
||||
// sourceLang := req.SourceLang
|
||||
// targetLang := req.TargetLang
|
||||
// translateText := req.TransText
|
||||
// tagHandling := req.TagHandling
|
||||
// proxyURL := cfg.Proxy
|
||||
sourceLang := req.SourceLang
|
||||
targetLang := req.TargetLang
|
||||
translateText := req.TransText
|
||||
tagHandling := req.TagHandling
|
||||
proxyURL := cfg.Proxy
|
||||
|
||||
// dlSession := cfg.DlSession
|
||||
dlSession := cfg.DlSession
|
||||
|
||||
// if tagHandling != "" && tagHandling != "html" && tagHandling != "xml" {
|
||||
// c.JSON(http.StatusBadRequest, gin.H{
|
||||
// "code": http.StatusBadRequest,
|
||||
// "message": "Invalid tag_handling value. Allowed values are 'html' and 'xml'.",
|
||||
// })
|
||||
// return
|
||||
// }
|
||||
if tagHandling != "" && tagHandling != "html" && tagHandling != "xml" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{
|
||||
"code": http.StatusBadRequest,
|
||||
"message": "Invalid tag_handling value. Allowed values are 'html' and 'xml'.",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// cookie := c.GetHeader("Cookie")
|
||||
// if cookie != "" {
|
||||
// dlSession = strings.Replace(cookie, "dl_session=", "", -1)
|
||||
// }
|
||||
cookie := c.GetHeader("Cookie")
|
||||
if cookie != "" {
|
||||
dlSession = strings.Replace(cookie, "dl_session=", "", -1)
|
||||
}
|
||||
|
||||
// if dlSession == "" {
|
||||
// c.JSON(http.StatusUnauthorized, gin.H{
|
||||
// "code": http.StatusUnauthorized,
|
||||
// "message": "No dl_session Found",
|
||||
// })
|
||||
// return
|
||||
// } else if strings.Contains(dlSession, ".") {
|
||||
// c.JSON(http.StatusUnauthorized, gin.H{
|
||||
// "code": http.StatusUnauthorized,
|
||||
// "message": "Your account is not a Pro account. Please upgrade your account or switch to a different account.",
|
||||
// })
|
||||
// return
|
||||
// }
|
||||
if dlSession == "" {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{
|
||||
"code": http.StatusUnauthorized,
|
||||
"message": "No dl_session Found",
|
||||
})
|
||||
return
|
||||
} else if strings.Contains(dlSession, ".") {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{
|
||||
"code": http.StatusUnauthorized,
|
||||
"message": "Your account is not a Pro account. Please upgrade your account or switch to a different account.",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// result, err := translate.TranslateByDeepLXPro(sourceLang, targetLang, translateText, tagHandling, dlSession, proxyURL)
|
||||
// if err != nil {
|
||||
// log.Fatalf("Translation failed: %s", err)
|
||||
// }
|
||||
result, err := translate.TranslateByDeepLX(sourceLang, targetLang, translateText, tagHandling, proxyURL, dlSession)
|
||||
if err != nil {
|
||||
log.Fatalf("Translation failed: %s", err)
|
||||
}
|
||||
|
||||
// if result.Code == http.StatusOK {
|
||||
// c.JSON(http.StatusOK, gin.H{
|
||||
// "code": http.StatusOK,
|
||||
// "id": result.ID,
|
||||
// "data": result.Data,
|
||||
// "alternatives": result.Alternatives,
|
||||
// "source_lang": result.SourceLang,
|
||||
// "target_lang": result.TargetLang,
|
||||
// "method": result.Method,
|
||||
// })
|
||||
// } else {
|
||||
// c.JSON(result.Code, gin.H{
|
||||
// "code": result.Code,
|
||||
// "message": result.Message,
|
||||
// })
|
||||
if result.Code == http.StatusOK {
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"code": http.StatusOK,
|
||||
"id": result.ID,
|
||||
"data": result.Data,
|
||||
"alternatives": result.Alternatives,
|
||||
"source_lang": result.SourceLang,
|
||||
"target_lang": result.TargetLang,
|
||||
"method": result.Method,
|
||||
})
|
||||
} else {
|
||||
c.JSON(result.Code, gin.H{
|
||||
"code": result.Code,
|
||||
"message": result.Message,
|
||||
})
|
||||
|
||||
// }
|
||||
// })
|
||||
}
|
||||
})
|
||||
|
||||
// Free API endpoint, Consistent with the official API format
|
||||
r.POST("/v2/translate", authMiddleware(cfg), func(c *gin.Context) {
|
||||
|
|
@ -247,7 +247,7 @@ func main() {
|
|||
targetLang = jsonData.TargetLang
|
||||
}
|
||||
|
||||
result, err := translate.TranslateByDeepLX("", targetLang, translateText, "", proxyURL)
|
||||
result, err := translate.TranslateByDeepLX("", targetLang, translateText, "", proxyURL, "")
|
||||
if err != nil {
|
||||
log.Fatalf("Translation failed: %s", err)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -2,7 +2,7 @@
|
|||
* @Author: Vincent Young
|
||||
* @Date: 2024-09-16 11:59:24
|
||||
* @LastEditors: Vincent Yang
|
||||
* @LastEditTime: 2024-11-01 12:45:44
|
||||
* @LastEditTime: 2024-11-01 13:12:25
|
||||
* @FilePath: /DeepLX/translate/translate.go
|
||||
* @Telegram: https://t.me/missuo
|
||||
* @GitHub: https://github.com/missuo
|
||||
|
|
@ -27,7 +27,7 @@ import (
|
|||
const baseURL = "https://www2.deepl.com/jsonrpc"
|
||||
|
||||
// makeRequest makes an HTTP request to DeepL API
|
||||
func makeRequest(postData *PostData, urlMethod string, proxyURL string) (gjson.Result, error) {
|
||||
func makeRequest(postData *PostData, urlMethod string, proxyURL string, dlSession string) (gjson.Result, error) {
|
||||
urlFull := fmt.Sprintf("%s?client=chrome-extension,1.28.0&method=%s", baseURL, urlMethod)
|
||||
|
||||
postStr := formatPostString(postData)
|
||||
|
|
@ -36,7 +36,7 @@ func makeRequest(postData *PostData, urlMethod string, proxyURL string) (gjson.R
|
|||
return gjson.Result{}, err
|
||||
}
|
||||
|
||||
// Set headers
|
||||
if dlSession != "" {
|
||||
req.Header = http.Header{
|
||||
"Accept": []string{"*/*"},
|
||||
"Accept-Language": []string{"en-US,en;q=0.9,zh-CN;q=0.8,zh-TW;q=0.7,zh-HK;q=0.6,zh;q=0.5"},
|
||||
|
|
@ -53,6 +53,26 @@ func makeRequest(postData *PostData, urlMethod string, proxyURL string) (gjson.R
|
|||
"Sec-Fetch-Site": []string{"none"},
|
||||
"Sec-GPC": []string{"1"},
|
||||
"User-Agent": []string{"DeepLBrowserExtension/1.28.0 Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36"},
|
||||
"Cookie": []string{"dl_session=" + dlSession},
|
||||
}
|
||||
} else {
|
||||
req.Header = http.Header{
|
||||
"Accept": []string{"*/*"},
|
||||
"Accept-Language": []string{"en-US,en;q=0.9,zh-CN;q=0.8,zh-TW;q=0.7,zh-HK;q=0.6,zh;q=0.5"},
|
||||
"Authorization": []string{"None"},
|
||||
"Cache-Control": []string{"no-cache"},
|
||||
"Content-Type": []string{"application/json"},
|
||||
"DNT": []string{"1"},
|
||||
"Origin": []string{"chrome-extension://cofdbpoegempjloogbagkncekinflcnj"},
|
||||
"Pragma": []string{"no-cache"},
|
||||
"Priority": []string{"u=1, i"},
|
||||
"Referer": []string{"https://www.deepl.com/"},
|
||||
"Sec-Fetch-Dest": []string{"empty"},
|
||||
"Sec-Fetch-Mode": []string{"cors"},
|
||||
"Sec-Fetch-Site": []string{"none"},
|
||||
"Sec-GPC": []string{"1"},
|
||||
"User-Agent": []string{"DeepLBrowserExtension/1.28.0 Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36"},
|
||||
}
|
||||
}
|
||||
|
||||
// Setup client with proxy if provided
|
||||
|
|
@ -73,6 +93,20 @@ func makeRequest(postData *PostData, urlMethod string, proxyURL string) (gjson.R
|
|||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// Check status code before processing response
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
switch resp.StatusCode {
|
||||
case http.StatusTooManyRequests:
|
||||
return gjson.Result{}, fmt.Errorf("too many requests")
|
||||
case http.StatusUnauthorized:
|
||||
return gjson.Result{}, fmt.Errorf("unauthorized")
|
||||
case http.StatusForbidden:
|
||||
return gjson.Result{}, fmt.Errorf("forbidden")
|
||||
default:
|
||||
return gjson.Result{}, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
|
||||
}
|
||||
}
|
||||
|
||||
var bodyReader io.Reader
|
||||
if resp.Header.Get("Content-Encoding") == "br" {
|
||||
bodyReader = brotli.NewReader(resp.Body)
|
||||
|
|
@ -84,12 +118,11 @@ func makeRequest(postData *PostData, urlMethod string, proxyURL string) (gjson.R
|
|||
if err != nil {
|
||||
return gjson.Result{}, err
|
||||
}
|
||||
|
||||
return gjson.ParseBytes(body), nil
|
||||
}
|
||||
|
||||
// splitText splits the input text for translation
|
||||
func splitText(text string, tagHandling bool, proxyURL string) (gjson.Result, error) {
|
||||
func splitText(text string, tagHandling bool, proxyURL string, dlSession string) (gjson.Result, error) {
|
||||
postData := &PostData{
|
||||
Jsonrpc: "2.0",
|
||||
Method: "LMT_split_text",
|
||||
|
|
@ -106,11 +139,11 @@ func splitText(text string, tagHandling bool, proxyURL string) (gjson.Result, er
|
|||
},
|
||||
}
|
||||
|
||||
return makeRequest(postData, "LMT_split_text", proxyURL)
|
||||
return makeRequest(postData, "LMT_split_text", proxyURL, dlSession)
|
||||
}
|
||||
|
||||
// TranslateByDeepLX performs translation using DeepL API
|
||||
func TranslateByDeepLX(sourceLang, targetLang, text string, tagHandling string, proxyURL string) (DeepLXTranslationResult, error) {
|
||||
func TranslateByDeepLX(sourceLang, targetLang, text string, tagHandling string, proxyURL string, dlSession string) (DeepLXTranslationResult, error) {
|
||||
if text == "" {
|
||||
return DeepLXTranslationResult{
|
||||
Code: http.StatusNotFound,
|
||||
|
|
@ -119,7 +152,7 @@ func TranslateByDeepLX(sourceLang, targetLang, text string, tagHandling string,
|
|||
}
|
||||
|
||||
// Split text first
|
||||
splitResult, err := splitText(text, tagHandling == "html" || tagHandling == "xml", proxyURL)
|
||||
splitResult, err := splitText(text, tagHandling == "html" || tagHandling == "xml", proxyURL, dlSession)
|
||||
if err != nil {
|
||||
return DeepLXTranslationResult{
|
||||
Code: http.StatusServiceUnavailable,
|
||||
|
|
@ -188,10 +221,9 @@ func TranslateByDeepLX(sourceLang, targetLang, text string, tagHandling string,
|
|||
Timestamp: getTimeStamp(getICount(text)),
|
||||
},
|
||||
}
|
||||
fmt.Println(postData)
|
||||
|
||||
// Make translation request
|
||||
result, err := makeRequest(postData, "LMT_handle_jobs", proxyURL)
|
||||
result, err := makeRequest(postData, "LMT_handle_jobs", proxyURL, dlSession)
|
||||
if err != nil {
|
||||
return DeepLXTranslationResult{
|
||||
Code: http.StatusServiceUnavailable,
|
||||
|
|
@ -241,6 +273,6 @@ func TranslateByDeepLX(sourceLang, targetLang, text string, tagHandling string,
|
|||
Alternatives: alternatives,
|
||||
SourceLang: sourceLang,
|
||||
TargetLang: targetLang,
|
||||
Method: "Free",
|
||||
Method: map[bool]string{true: "Pro", false: "Free"}[dlSession != ""],
|
||||
}, nil
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user