From ec85ee3e82a97e26f080cf710793099876139e56 Mon Sep 17 00:00:00 2001 From: nyne Date: Sat, 15 Nov 2025 19:57:47 +0800 Subject: [PATCH] Get characters from vndb --- server/api/resource.go | 17 ++++ server/service/resource.go | 163 +++++++++++++++++++++++++++++++++++++ 2 files changed, 180 insertions(+) diff --git a/server/api/resource.go b/server/api/resource.go index f89a2ad..d2e5973 100644 --- a/server/api/resource.go +++ b/server/api/resource.go @@ -282,6 +282,22 @@ func handleGetPinnedResources(c fiber.Ctx) error { }) } +func handleGetCharactorsFromVndb(c fiber.Ctx) error { + vnID := c.Query("vnid") + if vnID == "" { + return model.NewRequestError("VNDB ID is required") + } + characters, err := service.GetCharactorsFromVndb(vnID) + if err != nil { + return err + } + return c.Status(fiber.StatusOK).JSON(model.Response[[]service.CharactorParams]{ + Success: true, + Data: characters, + Message: "Characters retrieved successfully", + }) +} + func AddResourceRoutes(api fiber.Router) { resource := api.Group("/resource") { @@ -290,6 +306,7 @@ func AddResourceRoutes(api fiber.Router) { resource.Get("/", handleListResources) resource.Get("/random", handleGetRandomResource) resource.Get("/pinned", handleGetPinnedResources) + resource.Get("/vndb/characters", handleGetCharactorsFromVndb) resource.Get("/:id", handleGetResource) resource.Delete("/:id", handleDeleteResource) resource.Get("/tag/:tag", handleListResourcesWithTag) diff --git a/server/service/resource.go b/server/service/resource.go index ca8600a..5ff9ad3 100644 --- a/server/service/resource.go +++ b/server/service/resource.go @@ -1,6 +1,10 @@ package service import ( + "encoding/json" + "fmt" + "io" + "net/http" "net/url" "nysoure/server/config" "nysoure/server/dao" @@ -10,6 +14,7 @@ import ( "slices" "strconv" "strings" + "time" "github.com/gofiber/fiber/v3" "github.com/gofiber/fiber/v3/log" @@ -590,3 +595,161 @@ func GetPinnedResources() ([]model.ResourceView, error) { } return views, nil } + +func GetCharactorsFromVndb(vnID string) ([]CharactorParams, error) { + client := http.Client{} + jsonStr := fmt.Sprintf(` + { + "filters": ["id", "=", "%s"], + "fields": "va.character.name, va.staff.name, va.staff.original, va.character.original, va.character.image.url, va.character.vns.role" + } + `, vnID) + jsonStr = strings.TrimSpace(jsonStr) + reader := strings.NewReader(jsonStr) + resp, err := client.Post("https://api.vndb.org/kana/vn", "application/json", reader) + if err != nil { + return nil, model.NewInternalServerError("Failed to fetch data from VNDB") + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + return nil, model.NewInternalServerError("Failed to fetch data from VNDB") + } + // 定义 VNDB API 响应结构 + type VndbResponse struct { + Results []struct { + ID string `json:"id"` + VA []struct { + Character struct { + ID string `json:"id"` + Name string `json:"name"` + Original string `json:"original"` + Image struct { + URL string `json:"url"` + } `json:"image"` + VNS []struct { + ID string `json:"id"` + Role string `json:"role"` + } `json:"vns"` + } `json:"character"` + Staff struct { + ID string `json:"id"` + Name string `json:"name"` + Original string `json:"original"` + } `json:"staff"` + } `json:"va"` + } `json:"results"` + } + + // 解析响应 + var vndbResp VndbResponse + if err := json.NewDecoder(resp.Body).Decode(&vndbResp); err != nil { + return nil, model.NewInternalServerError("Failed to parse VNDB response") + } + + if len(vndbResp.Results) == 0 { + return []CharactorParams{}, nil + } + + result := vndbResp.Results[0] + var charactors []CharactorParams + processedCharacters := make(map[string]bool) // 避免重复角色 + + // 遍历声优信息 + for _, va := range result.VA { + // 检查角色是否为主要角色 + isPrimary := false + for _, vn := range va.Character.VNS { + if vn.Role == "primary" { + isPrimary = true + break + } + } + + // 只处理主要角色 + if !isPrimary { + continue + } + + // 避免重复角色 + if processedCharacters[va.Character.ID] { + continue + } + processedCharacters[va.Character.ID] = true + + // 优先使用 original 字段作为角色名，如果没有则使用 name + characterName := strings.ReplaceAll(va.Character.Original, " ", "") + if characterName == "" { + characterName = va.Character.Name + } + if characterName == "" { + continue // 跳过没有名字的角色 + } + + // 使用 original 字段作为声优名，如果没有则使用 name + cvName := strings.ReplaceAll(va.Staff.Original, " ", "") + if cvName == "" { + cvName = va.Staff.Name + } + + charactor := CharactorParams{ + Name: characterName, + Alias: []string{}, // 按要求不添加别名 + CV: cvName, + Image: 0, // 默认值，下面会下载图片 + } + + // 下载并保存角色图片 + if va.Character.Image.URL != "" { + imageID, err := downloadAndCreateImage(va.Character.Image.URL) + if err != nil { + log.Error("Failed to download character image:", err) + // 继续处理，即使图片下载失败 + } else { + charactor.Image = imageID + } + } + + charactors = append(charactors, charactor) + } + + return charactors, nil +} + +// downloadAndCreateImage 下载图片并使用 CreateImage 保存 +func downloadAndCreateImage(imageURL string) (uint, error) { + // 创建 HTTP 客户端 + client := &http.Client{ + Timeout: 30 * time.Second, + } + + // 下载图片 + resp, err := client.Get(imageURL) + if err != nil { + return 0, fmt.Errorf("failed to download image: %w", err) + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + return 0, fmt.Errorf("failed to download image: HTTP %d", resp.StatusCode) + } + + // 读取图片数据 + imageData, err := io.ReadAll(resp.Body) + if err != nil { + return 0, fmt.Errorf("failed to read image data: %w", err) + } + + // 限制图片大小，防止内存溢出 + if len(imageData) > 8*1024*1024 { // 8MB 限制 + return 0, fmt.Errorf("image too large") + } + + // 使用系统用户ID (假设为1) 创建图片 + // 注意：这里使用系统账户，实际使用时可能需要调整 + imageID, err := CreateImage(1, "127.0.0.1", imageData) + if err != nil { + return 0, fmt.Errorf("failed to create image: %w", err) + } + + return imageID, nil +}