From 9d9a2545f930f4cbfb63d63e99cb1d06e7581265 Mon Sep 17 00:00:00 2001
From: nyne <me@nyne.dev>
Date: Sun, 16 Nov 2025 18:37:33 +0800
Subject: [PATCH] Role for character

---
 ...{charactor_edit.tsx => character_edit.tsx} | 13 ++++++++--
 frontend/src/network/models.ts                |  3 +++
 frontend/src/pages/edit_resource_page.tsx     |  4 ++--
 frontend/src/pages/publish_page.tsx           |  4 ++--
 server/model/charactor.go                     |  5 +++-
 server/service/resource.go                    | 24 +++++++++++++------
 6 files changed, 39 insertions(+), 14 deletions(-)
 rename frontend/src/components/{charactor_edit.tsx => character_edit.tsx} (89%)

diff --git a/frontend/src/components/charactor_edit.tsx b/frontend/src/components/character_edit.tsx
similarity index 89%
rename from frontend/src/components/charactor_edit.tsx
rename to frontend/src/components/character_edit.tsx
index cff457c..55e0b90 100644
--- a/frontend/src/components/charactor_edit.tsx
+++ b/frontend/src/components/character_edit.tsx
@@ -1,11 +1,11 @@
 import { useState } from "react";
-import { CharacterParams } from "../network/models";
+import { CharacterParams, CharacterRole } from "../network/models";
 import { network } from "../network/network";
 import showToast from "./toast";
 import { useTranslation } from "../utils/i18n";
 import Button from "./button";
 
-export default function CharactorEditor({charactor, setCharactor, onDelete}: {
+export default function CharacterEditer({charactor, setCharactor, onDelete}: {
     charactor: CharacterParams;
     setCharactor: (charactor: CharacterParams) => void;
     onDelete: () => void;
@@ -82,6 +82,15 @@ export default function CharactorEditor({charactor, setCharactor, onDelete}: {
           onChange={(e) => setCharactor({ ...charactor, cv: e.target.value })}
         />
         
+        <select
+          className="select select-sm select-bordered"
+          value={charactor.role}
+          onChange={(e) => setCharactor({ ...charactor, role: e.target.value as CharacterRole })}
+        >
+          <option value="primary">{t("Primary Role")}</option>
+          <option value="side">{t("Side Role")}</option>
+        </select>
+        
         <div className="flex-1">
           <textarea
             className="textarea textarea-bordered w-full h-full resize-none text-xs"
diff --git a/frontend/src/network/models.ts b/frontend/src/network/models.ts
index 8ea005f..84e944b 100644
--- a/frontend/src/network/models.ts
+++ b/frontend/src/network/models.ts
@@ -52,11 +52,14 @@ export interface CreateResourceParams {
   characters: CharacterParams[];
 }
 
+export type CharacterRole = 'primary' | 'side';
+
 export interface CharacterParams {
   name: string;
   alias: string[];
   cv: string;
   image: number;
+  role: CharacterRole;
 }
 
 export interface Image {
diff --git a/frontend/src/pages/edit_resource_page.tsx b/frontend/src/pages/edit_resource_page.tsx
index 05bcba0..6b27d8f 100644
--- a/frontend/src/pages/edit_resource_page.tsx
+++ b/frontend/src/pages/edit_resource_page.tsx
@@ -20,7 +20,7 @@ import {
   SelectAndUploadImageButton,
   UploadClipboardImageButton,
 } from "../components/image_selector.tsx";
-import CharactorEditor, { FetchVndbCharactersButton } from "../components/charactor_edit.tsx";
+import CharacterEditer, { FetchVndbCharactersButton } from "../components/character_edit.tsx";
 
 export default function EditResourcePage() {
   const [title, setTitle] = useState<string>("");
@@ -429,7 +429,7 @@ export default function EditResourcePage() {
           <div className="grid grid-cols-1 md:grid-cols-2 my-2 gap-4">
             {
               charactors.map((charactor, index) => {
-                return <CharactorEditor 
+                return <CharacterEditer 
                   charactor={charactor} 
                   setCharactor={(newCharactor) => {
                     const newCharactors = [...charactors];
diff --git a/frontend/src/pages/publish_page.tsx b/frontend/src/pages/publish_page.tsx
index 12186f3..708060d 100644
--- a/frontend/src/pages/publish_page.tsx
+++ b/frontend/src/pages/publish_page.tsx
@@ -19,7 +19,7 @@ import {
   SelectAndUploadImageButton,
   UploadClipboardImageButton,
 } from "../components/image_selector.tsx";
-import CharactorEditor from "../components/charactor_edit.tsx";
+import CharacterEditer from "../components/character_edit.tsx";
 
 export default function PublishPage() {
   const [title, setTitle] = useState<string>("");
@@ -436,7 +436,7 @@ export default function PublishPage() {
           <div className="grid grid-cols-1 md:grid-cols-2 my-2 gap-4">
             {
               charactors.map((charactor, index) => {
-                return <CharactorEditor 
+                return <CharacterEditer 
                   charactor={charactor} 
                   setCharactor={(newCharactor) => {
                     const newCharactors = [...charactors];
diff --git a/server/model/charactor.go b/server/model/charactor.go
index 84db719..0a078e8 100644
--- a/server/model/charactor.go
+++ b/server/model/charactor.go
@@ -5,6 +5,7 @@ type Charactor struct {
 	Name       string   `gorm:"type:varchar(100);not null"`
 	Alias      []string `gorm:"serializer:json"`
 	CV         string   `gorm:"type:varchar(100)"`
+	Role       string   `gorm:"type:varchar(20);default:primary"`
 	ImageID    uint
 	ResourceID uint
 	Image      *Image `gorm:"foreignKey:ImageID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;"`
@@ -15,6 +16,7 @@ type CharactorView struct {
 	Name  string   `json:"name"`
 	Alias []string `json:"alias"`
 	CV    string   `json:"cv"`
+	Role  string   `json:"role"`
 	Image uint     `json:"image"`
 }
 
@@ -24,12 +26,13 @@ func (c *Charactor) ToView() *CharactorView {
 		Name:  c.Name,
 		Alias: c.Alias,
 		CV:    c.CV,
+		Role:  c.Role,
 		Image: c.ImageID,
 	}
 }
 
 func (c *Charactor) Equal(other *Charactor) bool {
-	if c.Name != other.Name || c.CV != other.CV || c.ImageID != other.ImageID {
+	if c.Name != other.Name || c.CV != other.CV || c.Role != other.Role || c.ImageID != other.ImageID {
 		return false
 	}
 	if len(c.Alias) != len(other.Alias) {
diff --git a/server/service/resource.go b/server/service/resource.go
index cb83739..9703c34 100644
--- a/server/service/resource.go
+++ b/server/service/resource.go
@@ -42,6 +42,7 @@ type CharactorParams struct {
 	Name  string   `json:"name" binding:"required"`
 	Alias []string `json:"alias"`
 	CV    string   `json:"cv"`
+	Role  string   `json:"role"`
 	Image uint     `json:"image"`
 }
 
@@ -84,10 +85,15 @@ func CreateResource(uid uint, params *ResourceParams) (uint, error) {
 	}
 	charactors := make([]model.Charactor, len(params.Charactors))
 	for i, c := range params.Charactors {
+		role := c.Role
+		if role == "" {
+			role = "primary"
+		}
 		charactors[i] = model.Charactor{
 			Name:    c.Name,
 			Alias:   c.Alias,
 			CV:      c.CV,
+			Role:    role,
 			ImageID: c.Image,
 		}
 	}
@@ -502,10 +508,15 @@ func UpdateResource(uid, rid uint, params *ResourceParams) error {
 	}
 	charactors := make([]model.Charactor, len(params.Charactors))
 	for i, c := range params.Charactors {
+		role := c.Role
+		if role == "" {
+			role = "primary"
+		}
 		charactors[i] = model.Charactor{
 			Name:    c.Name,
 			Alias:   c.Alias,
 			CV:      c.CV,
+			Role:    role,
 			ImageID: c.Image,
 		}
 	}
@@ -656,17 +667,15 @@ func GetCharactorsFromVndb(vnID string) ([]CharactorParams, error) {
 
 	// 遍历声优信息
 	for _, va := range result.VA {
-		// 检查角色是否为主要角色
-		isPrimary := false
+		role := "Unknown"
 		for _, vn := range va.Character.VNS {
-			if vn.Role == "primary" {
-				isPrimary = true
+			if vn.ID == vnID {
+				role = vn.Role
 				break
 			}
 		}
 
-		// 只处理主要角色
-		if !isPrimary {
+		if role != "primary" && role != "side" {
 			continue
 		}
 
@@ -693,8 +702,9 @@ func GetCharactorsFromVndb(vnID string) ([]CharactorParams, error) {
 
 		charactor := CharactorParams{
 			Name:  characterName,
-			Alias: []string{}, // 按要求不添加别名
+			Alias: []string{},
 			CV:    cvName,
+			Role:  role,
 			Image: 0, // 默认值，下面会下载图片
 		}