From d597d62c1c6ed2a0858dbb608f5c59f59b3cc266 Mon Sep 17 00:00:00 2001
From: nyne <me@nyne.dev>
Date: Sat, 31 May 2025 12:43:08 +0800
Subject: [PATCH] Add Quick Add Tag functionality and improve tag management in
 Edit and Publish pages

---
 frontend/src/components/tag_input.tsx     | 60 +++++++++++++++++++++++
 frontend/src/i18n.ts                      | 15 ++++++
 frontend/src/network/network.ts           | 16 ++++++
 frontend/src/pages/edit_resource_page.tsx | 29 +++++++++--
 frontend/src/pages/publish_page.tsx       | 29 +++++++++--
 server/api/tag.go                         | 44 +++++++++++++++++
 server/dao/tag.go                         |  9 ++++
 server/service/tag.go                     | 27 ++++++++++
 8 files changed, 221 insertions(+), 8 deletions(-)

diff --git a/frontend/src/components/tag_input.tsx b/frontend/src/components/tag_input.tsx
index babd78b..3ffc953 100644
--- a/frontend/src/components/tag_input.tsx
+++ b/frontend/src/components/tag_input.tsx
@@ -4,6 +4,9 @@ import {useTranslation} from "react-i18next";
 import {network} from "../network/network.ts";
 import {LuInfo} from "react-icons/lu";
 import {MdSearch} from "react-icons/md";
+import Button from "./button.tsx";
+import Input, {TextArea} from "./input.tsx";
+import {ErrorAlert} from "./alert.tsx";
 
 export default function TagInput({ onAdd, mainTag }: { onAdd: (tag: Tag) => void, mainTag?: boolean }) {
   const [keyword, setKeyword] = useState<string>("")
@@ -142,4 +145,61 @@ class Debounce {
       this.timer = null
     }
   }
+}
+
+export function QuickAddTagDialog({ onAdded }: { onAdded: (tags: Tag[]) => void }) {
+  const {t} = useTranslation();
+
+  const [text, setText] = useState<string>("")
+
+  const [type, setType] = useState<string>("")
+
+  const [error, setError] = useState<string | null>(null)
+
+  const handleSubmit = async () => {
+    if (text.trim().length === 0) {
+      return
+    }
+    setError(null)
+    const names = text.split(",").filter((n) => n.length > 0)
+    const res = await network.getOrCreateTags(names, type)
+    if (!res.success) {
+      setError(res.message)
+      return
+    }
+    const tags = res.data!
+    onAdded(tags)
+    setText("")
+    setType("")
+    const dialog = document.getElementById("quick_add_tag_dialog") as HTMLDialogElement
+    dialog.close()
+  }
+
+  return <>
+    <Button className={"btn-soft btn-primary"} onClick={() => {
+      const dialog = document.getElementById("quick_add_tag_dialog") as HTMLDialogElement
+      dialog.showModal()
+    }}>{t("Quick Add")}</Button>
+    <dialog id="quick_add_tag_dialog" className="modal">
+      <div className="modal-box">
+        <h3 className="font-bold text-lg">{t('Add Tags')}</h3>
+        <p className="py-2 text-sm">
+          {t("Input tags separated by commas.")}
+          <br/>
+          {t("If the tag does not exist, it will be created automatically.")}
+          <br/>
+          {t("Optionally, you can specify a type for the new tags.")}
+        </p>
+        <TextArea value={text} onChange={(e) => setText(e.target.value)} label={"Tags"}/>
+        <Input value={type} onChange={(e) => setType(e.target.value)} label={"Type"}/>
+        {error && <ErrorAlert className={"mt-2"} message={error}/>}
+        <div className="modal-action">
+          <form method="dialog">
+            <Button className="btn">{t("Cancel")}</Button>
+          </form>
+          <Button className={"btn-primary"} disabled={text === ""} onClick={handleSubmit}>{t("Submit")}</Button>
+        </div>
+      </div>
+    </dialog>
+  </>
 }
\ No newline at end of file
diff --git a/frontend/src/i18n.ts b/frontend/src/i18n.ts
index ef8f2aa..5fdecf3 100644
--- a/frontend/src/i18n.ts
+++ b/frontend/src/i18n.ts
@@ -172,6 +172,11 @@ export const i18nData = {
       "About": "About",
       "Home": "Home",
       "Other": "Other",
+      "Quick Add": "Quick Add",
+      "Add Tags": "Add Tags",
+      "Input tags separated by commas.": "Input tags separated by commas.",
+      "If the tag does not exist, it will be created automatically.": "If the tag does not exist, it will be created automatically.",
+      "Optionally, you can specify a type for the new tags.": "Optionally, you can specify a type for the new tags.",
     }
   },
   "zh-CN": {
@@ -347,6 +352,11 @@ export const i18nData = {
       "About": "关于",
       "Home": "首页",
       "Other": "其他",
+      "Quick Add": "快速添加",
+      "Add Tags": "添加标签",
+      "Input tags separated by commas.": "输入标签, 用逗号分隔。",
+      "If the tag does not exist, it will be created automatically.": "如果标签不存在, 将自动创建。",
+      "Optionally, you can specify a type for the new tags.": "您可以选择为新标签指定一个类型。",
     }
   },
   "zh-TW": {
@@ -522,6 +532,11 @@ export const i18nData = {
       "About": "關於",
       "Home": "首頁",
       "Other": "其他",
+      "Quick Add": "快速添加",
+      "Add Tags": "添加標籤",
+      "Input tags separated by commas.": "輸入標籤, 用逗號分隔。",
+      "If the tag does not exist, it will be created automatically.": "如果標籤不存在, 將自動創建。",
+      "Optionally, you can specify a type for the new tags.": "您可以選擇為新標籤指定一個類型。",
     }
   }
 }
diff --git a/frontend/src/network/network.ts b/frontend/src/network/network.ts
index dadc3e3..1caf993 100644
--- a/frontend/src/network/network.ts
+++ b/frontend/src/network/network.ts
@@ -330,6 +330,22 @@ class Network {
     }
   }
 
+  async getOrCreateTags(names: string[], tagType: string): Promise<Response<Tag[]>> {
+    try {
+      const response = await axios.post(`${this.apiBaseUrl}/tag/batch`, {
+        names,
+        type: tagType
+      })
+      return response.data
+    } catch (e: any) {
+      console.error(e)
+      return {
+        success: false,
+        message: e.toString(),
+      }
+    }
+  }
+
   async getTagByName(name: string): Promise<Response<Tag>> {
     try {
       const response = await axios.get(`${this.apiBaseUrl}/tag/${name}`)
diff --git a/frontend/src/pages/edit_resource_page.tsx b/frontend/src/pages/edit_resource_page.tsx
index 7f703bd..50345e2 100644
--- a/frontend/src/pages/edit_resource_page.tsx
+++ b/frontend/src/pages/edit_resource_page.tsx
@@ -8,7 +8,7 @@ import { useTranslation } from "react-i18next";
 import { app } from "../app.ts";
 import { ErrorAlert } from "../components/alert.tsx";
 import Loading from "../components/loading.tsx";
-import TagInput from "../components/tag_input.tsx";
+import TagInput, {QuickAddTagDialog} from "../components/tag_input.tsx";
 
 export default function EditResourcePage() {
   const [title, setTitle] = useState<string>("")
@@ -178,9 +178,30 @@ export default function EditResourcePage() {
         })
       }
     </p>
-    <TagInput onAdd={(tag) => {
-      setTags([...tags, tag])
-    }} />
+    <div className={"flex items-center"}>
+      <TagInput onAdd={(tag) => {
+        setTags((prev) => {
+          const existingTag = prev.find(t => t.id === tag.id);
+          if (existingTag) {
+            return prev; // If the tag already exists, do not add it again
+          }
+          return [...prev, tag];
+        })
+      }} />
+      <span className={"w-4"}/>
+      <QuickAddTagDialog onAdded={(tags) => {
+        setTags((prev) => {
+          const newTags = [...prev];
+          for (const tag of tags) {
+            const existingTag = newTags.find(t => t.id === tag.id);
+            if (!existingTag) {
+              newTags.push(tag);
+            }
+          }
+          return newTags;
+        })
+      }}/>
+    </div>
     <div className={"h-4"}></div>
     <p className={"my-1"}>{t("Description")}</p>
     <textarea className="textarea w-full min-h-80 p-4" value={article} onChange={(e) => setArticle(e.target.value)} />
diff --git a/frontend/src/pages/publish_page.tsx b/frontend/src/pages/publish_page.tsx
index fad0336..e0b2b78 100644
--- a/frontend/src/pages/publish_page.tsx
+++ b/frontend/src/pages/publish_page.tsx
@@ -8,7 +8,7 @@ import { useTranslation } from "react-i18next";
 import { app } from "../app.ts";
 import { ErrorAlert } from "../components/alert.tsx";
 import {useAppContext} from "../components/AppContext.tsx";
-import TagInput from "../components/tag_input.tsx";
+import TagInput, {QuickAddTagDialog} from "../components/tag_input.tsx";
 
 export default function PublishPage() {
   const [title, setTitle] = useState<string>("")
@@ -154,9 +154,30 @@ export default function PublishPage() {
         })
       }
     </p>
-    <TagInput onAdd={(tag) => {
-      setTags([...tags, tag])
-    }} />
+    <div className={"flex items-center"}>
+      <TagInput onAdd={(tag) => {
+        setTags((prev) => {
+          const existingTag = prev.find(t => t.id === tag.id);
+          if (existingTag) {
+            return prev; // If the tag already exists, do not add it again
+          }
+          return [...prev, tag];
+        })
+      }} />
+      <span className={"w-4"}/>
+      <QuickAddTagDialog onAdded={(tags) => {
+        setTags((prev) => {
+          const newTags = [...prev];
+          for (const tag of tags) {
+            const existingTag = newTags.find(t => t.id === tag.id);
+            if (!existingTag) {
+              newTags.push(tag);
+            }
+          }
+          return newTags;
+        })
+      }}/>
+    </div>
     <div className={"h-4"}></div>
     <p className={"my-1"}>{t("Description")}</p>
     <textarea className="textarea w-full min-h-80 p-4" value={article} onChange={(e) => setArticle(e.target.value)} />
diff --git a/server/api/tag.go b/server/api/tag.go
index db0907e..550be42 100644
--- a/server/api/tag.go
+++ b/server/api/tag.go
@@ -137,6 +137,49 @@ func getAllTags(c fiber.Ctx) error {
 	})
 }
 
+func getOrCreateTags(c fiber.Ctx) error {
+	uid, ok := c.Locals("uid").(uint)
+	if !ok {
+		return model.NewUnAuthorizedError("You must be logged in to get or create tags")
+	}
+
+	type GetOrCreateTagsRequest struct {
+		Names   []string `json:"names"`
+		TagType string   `json:"type"`
+	}
+
+	var req GetOrCreateTagsRequest
+	if err := c.Bind().JSON(&req); err != nil {
+		return model.NewRequestError("Invalid request format")
+	}
+
+	names := make([]string, 0, len(req.Names))
+	for _, name := range req.Names {
+		name = strings.TrimSpace(name)
+		if name == "" {
+			continue
+		}
+		names = append(names, name)
+	}
+
+	if len(names) == 0 {
+		return model.NewRequestError("At least one tag name is required")
+	}
+
+	tagType := strings.TrimSpace(req.TagType)
+
+	tags, err := service.GetOrCreateTags(uid, names, tagType)
+	if err != nil {
+		return err
+	}
+
+	return c.Status(fiber.StatusOK).JSON(model.Response[*[]model.TagView]{
+		Success: true,
+		Data:    &tags,
+		Message: "Tags retrieved or created successfully",
+	})
+}
+
 func AddTagRoutes(api fiber.Router) {
 	tag := api.Group("/tag")
 	{
@@ -146,5 +189,6 @@ func AddTagRoutes(api fiber.Router) {
 		tag.Put("/:id/info", handleSetTagInfo)
 		tag.Get("/:name", handleGetTagByName)
 		tag.Get("/", getAllTags)
+		tag.Post("/batch", getOrCreateTags)
 	}
 }
diff --git a/server/dao/tag.go b/server/dao/tag.go
index 6b96559..05af143 100644
--- a/server/dao/tag.go
+++ b/server/dao/tag.go
@@ -16,6 +16,15 @@ func CreateTag(tag string) (model.Tag, error) {
 	return t, nil
 }
 
+func CreateTagWithType(tag string, tagType string) (model.Tag, error) {
+	// Create a new tag with a specific type in the database
+	t := model.Tag{Name: tag, Type: tagType}
+	if err := db.Create(&t).Error; err != nil {
+		return model.Tag{}, err
+	}
+	return t, nil
+}
+
 func SearchTag(keyword string, mainTag bool) ([]model.Tag, error) {
 	// Search for a tag by its name in the database
 	var t []model.Tag
diff --git a/server/service/tag.go b/server/service/tag.go
index 5c8ec2f..00db2a9 100644
--- a/server/service/tag.go
+++ b/server/service/tag.go
@@ -134,3 +134,30 @@ func GetTagList() ([]model.TagViewWithCount, error) {
 	}
 	return cachedTagList, nil
 }
+
+func GetOrCreateTags(uid uint, names []string, tagType string) ([]model.TagView, error) {
+	canUpload, err := checkUserCanUpload(uid)
+	if err != nil {
+		log.Error("Error checking user permissions:", err)
+		return nil, model.NewInternalServerError("Error checking user permissions")
+	}
+	if !canUpload {
+		return nil, model.NewUnAuthorizedError("User cannot create tags")
+	}
+	tags := make([]model.TagView, 0, len(names))
+	for _, name := range names {
+		t, err := dao.GetTagByName(name)
+		if err != nil {
+			if model.IsNotFoundError(err) {
+				t, err = dao.CreateTagWithType(name, tagType)
+				if err != nil {
+					return nil, err
+				}
+			} else {
+				return nil, err
+			}
+		}
+		tags = append(tags, *t.ToView())
+	}
+	return tags, updateCachedTagList()
+}