From aa627fb61d1ff7f80ddd5788f19f2a532bd05f31 Mon Sep 17 00:00:00 2001
From: Zhang Minghan <zmh13054618081@dingtalk.com>
Date: Sat, 23 Dec 2023 20:14:14 +0800
Subject: [PATCH] feat: update deeptrain mode

---
 app/src-tauri/tauri.conf.json   |  2 +-
 app/src/conf.ts                 |  2 +-
 app/src/resources/i18n/cn.json  |  1 +
 app/src/resources/i18n/en.json  |  3 ++-
 app/src/resources/i18n/ja.json  |  3 ++-
 app/src/resources/i18n/ru.json  |  3 ++-
 app/src/routes/admin/System.tsx | 13 +++++++++++++
 auth/auth.go                    |  6 +++++-
 auth/call.go                    |  4 ++++
 auth/controller.go              |  6 +++---
 auth/package.go                 |  4 ++++
 auth/payment.go                 |  8 ++++++++
 middleware/throttle.go          |  3 +++
 13 files changed, 49 insertions(+), 9 deletions(-)

diff --git a/app/src-tauri/tauri.conf.json b/app/src-tauri/tauri.conf.json
index 5a3e931..7ff2ea1 100644
--- a/app/src-tauri/tauri.conf.json
+++ b/app/src-tauri/tauri.conf.json
@@ -8,7 +8,7 @@
   },
   "package": {
     "productName": "chatnio",
-    "version": "3.7.8"
+    "version": "3.8.0"
   },
   "tauri": {
     "allowlist": {
diff --git a/app/src/conf.ts b/app/src/conf.ts
index ae39b13..996af21 100644
--- a/app/src/conf.ts
+++ b/app/src/conf.ts
@@ -12,7 +12,7 @@ import { getMemory } from "@/utils/memory.ts";
 import { Compass, Image, Newspaper } from "lucide-react";
 import React from "react";
 
-export const version = "3.7.8";
+export const version = "3.8.0";
 export const dev: boolean = getDev();
 export const deploy: boolean = true;
 export let rest_api: string = getRestApi(deploy);
diff --git a/app/src/resources/i18n/cn.json b/app/src/resources/i18n/cn.json
index 76e667f..01d5b53 100644
--- a/app/src/resources/i18n/cn.json
+++ b/app/src/resources/i18n/cn.json
@@ -453,6 +453,7 @@
       "mailPort": "SMTP 端口",
       "mailUser": "用户名",
       "mailPass": "密码",
+      "mailFrom": "发件人",
       "searchEndpoint": "搜索接入点",
       "searchQuery": "最大搜索结果数",
       "searchTip": "DuckDuckGo 搜索接入点，如不填写自动使用 WebPilot 和 New Bing 逆向进行搜索功能。\\nDuckDuckGo API 项目搭建：[duckduckgo-api](https://github.com/binjie09/duckduckgo-api)。"
diff --git a/app/src/resources/i18n/en.json b/app/src/resources/i18n/en.json
index 8aba0b5..b7153ed 100644
--- a/app/src/resources/i18n/en.json
+++ b/app/src/resources/i18n/en.json
@@ -416,7 +416,8 @@
       "mailPass": "Password",
       "searchEndpoint": "Search Endpoint",
       "searchQuery": "Max Search Results",
-      "searchTip": "DuckDuckGo search endpoint, if not filled in, use WebPilot and New Bing reverse search function by default.\nDuckDuckGo API project build: [duckduckgo-api](https://github.com/binjie09/duckduckgo-api)."
+      "searchTip": "DuckDuckGo search endpoint, if not filled in, use WebPilot and New Bing reverse search function by default.\nDuckDuckGo API project build: [duckduckgo-api](https://github.com/binjie09/duckduckgo-api).",
+      "mailFrom": "Sender"
     }
   },
   "mask": {
diff --git a/app/src/resources/i18n/ja.json b/app/src/resources/i18n/ja.json
index 019451a..7152a0c 100644
--- a/app/src/resources/i18n/ja.json
+++ b/app/src/resources/i18n/ja.json
@@ -416,7 +416,8 @@
       "mailPass": "パスワード",
       "searchEndpoint": "アクセスポイントを検索",
       "searchQuery": "検索結果の最大数",
-      "searchTip": "DuckDuckGoは、入力せずにWebPilotやNew Bing Reverse Searchなどのアクセスポイントを自動的に検索します。\\ nDuckDuckGo APIプロジェクトビルド：[ duckduckgo - api ]（ https://github.com/binjie09/duckduckgo-api ）。"
+      "searchTip": "DuckDuckGoは、入力せずにWebPilotやNew Bing Reverse Searchなどのアクセスポイントを自動的に検索します。\\ nDuckDuckGo APIプロジェクトビルド：[ duckduckgo - api ]（ https://github.com/binjie09/duckduckgo-api ）。",
+      "mailFrom": "発信元"
     }
   },
   "mask": {
diff --git a/app/src/resources/i18n/ru.json b/app/src/resources/i18n/ru.json
index a3d1cd2..2866afc 100644
--- a/app/src/resources/i18n/ru.json
+++ b/app/src/resources/i18n/ru.json
@@ -416,7 +416,8 @@
       "mailPass": "Пароль",
       "searchEndpoint": "Конечная точка поиска",
       "searchQuery": "Максимальное количество результатов поиска",
-      "searchTip": "Конечная точка поиска DuckDuckGo, если она не заполнена, по умолчанию используется функция обратного поиска WebPilot и New Bing.\nСборка проекта DuckDuckGo API: [duckduckgo-api](https://github.com/binjie09/duckduckgo-api)."
+      "searchTip": "Конечная точка поиска DuckDuckGo, если она не заполнена, по умолчанию используется функция обратного поиска WebPilot и New Bing.\nСборка проекта DuckDuckGo API: [duckduckgo-api](https://github.com/binjie09/duckduckgo-api).",
+      "mailFrom": "От"
     }
   },
   "mask": {
diff --git a/app/src/routes/admin/System.tsx b/app/src/routes/admin/System.tsx
index c17d77e..698119c 100644
--- a/app/src/routes/admin/System.tsx
+++ b/app/src/routes/admin/System.tsx
@@ -130,6 +130,19 @@ function Mail({ data, dispatch, onChange }: CompProps<MailState>) {
           placeholder={t("admin.system.mailPass")}
         />
       </ParagraphItem>
+      <ParagraphItem>
+        <Label>{t("admin.system.mailFrom")}</Label>
+        <Input
+          value={data.from}
+          onChange={(e) =>
+            dispatch({
+              type: "update:mail.from",
+              value: e.target.value,
+            })
+          }
+          placeholder={`bot@${location.host}`}
+        />
+      </ParagraphItem>
       <ParagraphFooter>
         <div className={`grow`} />
         <Button size={`sm`} loading={true} onClick={onChange}>
diff --git a/auth/auth.go b/auth/auth.go
index e044e21..c8784bb 100644
--- a/auth/auth.go
+++ b/auth/auth.go
@@ -54,7 +54,11 @@ func ParseApiKey(c *gin.Context, key string) *User {
 	return &user
 }
 
-func Login(c *gin.Context, token string) (string, error) {
+func DeepLogin(c *gin.Context, token string) (string, error) {
+	if !useDeeptrain() {
+		return "", errors.New("deeptrain feature is disabled")
+	}
+
 	user := Validate(token)
 	if user == nil {
 		return "", errors.New("cannot validate access token")
diff --git a/auth/call.go b/auth/call.go
index 9c2c86d..74fc6f5 100644
--- a/auth/call.go
+++ b/auth/call.go
@@ -16,6 +16,10 @@ func getDeeptrainApi(path string) string {
 	return viper.GetString("auth.endpoint") + path
 }
 
+func useDeeptrain() bool {
+	return viper.GetBool("auth.use_deeptrain")
+}
+
 func Validate(token string) *ValidateUserResponse {
 	res, err := utils.Post(getDeeptrainApi("/app/validate"), map[string]string{
 		"Content-Type": "application/json",
diff --git a/auth/controller.go b/auth/controller.go
index 80f54c3..8bca9c6 100644
--- a/auth/controller.go
+++ b/auth/controller.go
@@ -7,7 +7,7 @@ import (
 	"strings"
 )
 
-type LoginForm struct {
+type DeepLoginForm struct {
 	Token string `form:"token" binding:"required"`
 }
 
@@ -112,7 +112,7 @@ func RequireEnterprise(c *gin.Context) *User {
 }
 
 func LoginAPI(c *gin.Context) {
-	var form LoginForm
+	var form DeepLoginForm
 	if err := c.ShouldBind(&form); err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"status": false,
@@ -121,7 +121,7 @@ func LoginAPI(c *gin.Context) {
 		return
 	}
 
-	token, err := Login(c, form.Token)
+	token, err := DeepLogin(c, form.Token)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"status": false,
diff --git a/auth/package.go b/auth/package.go
index 8e21ccb..5fdcc93 100644
--- a/auth/package.go
+++ b/auth/package.go
@@ -59,6 +59,10 @@ func NewTeenagerPackage(db *sql.DB, user *User) bool {
 }
 
 func RefreshPackage(db *sql.DB, user *User) *GiftResponse {
+	if !useDeeptrain() {
+		return nil
+	}
+
 	resp := Cert(user.Username)
 	if resp == nil || resp.Status == false {
 		return nil
diff --git a/auth/payment.go b/auth/payment.go
index 1522300..ad4bece 100644
--- a/auth/payment.go
+++ b/auth/payment.go
@@ -24,6 +24,10 @@ func GenerateOrder() string {
 }
 
 func GetBalance(username string) float32 {
+	if !useDeeptrain() {
+		return 0.
+	}
+
 	order := GenerateOrder()
 	res, err := utils.Post(getDeeptrainApi("/app/balance"), map[string]string{
 		"Content-Type": "application/json",
@@ -45,6 +49,10 @@ func GetBalance(username string) float32 {
 }
 
 func Pay(username string, amount float32) bool {
+	if !useDeeptrain() {
+		return false
+	}
+
 	order := GenerateOrder()
 	res, err := utils.Post(getDeeptrainApi("/app/payment"), map[string]string{
 		"Content-Type": "application/json",
diff --git a/middleware/throttle.go b/middleware/throttle.go
index 1d5123f..0235f8b 100644
--- a/middleware/throttle.go
+++ b/middleware/throttle.go
@@ -22,6 +22,9 @@ func (l *Limiter) RateLimit(client *redis.Client, ip string, path string) bool {
 
 var limits = map[string]Limiter{
 	"/login":        {Duration: 10, Count: 20},
+	"/register":     {Duration: 120, Count: 10},
+	"/verify":       {Duration: 120, Count: 5},
+	"/reset":        {Duration: 120, Count: 10},
 	"/apikey":       {Duration: 1, Count: 2},
 	"/package":      {Duration: 1, Count: 2},
 	"/quota":        {Duration: 1, Count: 2},