初始化
This commit is contained in:
Ran
+21
@@ -0,0 +1,21 @@
|
|||||||
|
# 环境变量和本地敏感配置
|
||||||
|
.env
|
||||||
|
.env.*
|
||||||
|
!.env.example
|
||||||
|
# 前端依赖
|
||||||
|
node_modules/
|
||||||
|
# 前端构建产物
|
||||||
|
ADMIN/dist/
|
||||||
|
WEB/dist/
|
||||||
|
# 后端编译产物和运行时模型
|
||||||
|
/API/bin/
|
||||||
|
/API/models/
|
||||||
|
# 本地工具二进制
|
||||||
|
/API/tools/ffmpeg/bin/
|
||||||
|
*.exe
|
||||||
|
# 测试、覆盖率和日志产物
|
||||||
|
coverage/
|
||||||
|
coverage.out
|
||||||
|
*.log
|
||||||
|
# 不应上传的文本配置、说明和本地文档
|
||||||
|
*.txt
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
{
|
||||||
|
"$schema": "https://json.schemastore.org/prettierrc",
|
||||||
|
"printWidth": 120,
|
||||||
|
"tabWidth": 2,
|
||||||
|
"useTabs": false,
|
||||||
|
"semi": true,
|
||||||
|
"singleQuote": false,
|
||||||
|
"trailingComma": "all",
|
||||||
|
"bracketSameLine": false,
|
||||||
|
"singleAttributePerLine": true
|
||||||
|
}
|
||||||
Vendored
+1
@@ -0,0 +1 @@
|
|||||||
|
/// <reference types="vite/client" />
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8" />
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||||
|
<link rel="icon" type="image/png" href="/assets/logo.png?v=1" />
|
||||||
|
<title>剧游AI-后台管理</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div id="app"></div>
|
||||||
|
<script type="module" src="/src/main.ts"></script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Generated
+2912
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,32 @@
|
|||||||
|
{
|
||||||
|
"name": "juhe-factory-admin",
|
||||||
|
"version": "0.0.1",
|
||||||
|
"private": true,
|
||||||
|
"type": "module",
|
||||||
|
"scripts": {
|
||||||
|
"dev": "vite",
|
||||||
|
"build": "vue-tsc --noEmit && vite build",
|
||||||
|
"format": "prettier --write src",
|
||||||
|
"format:check": "prettier --check src",
|
||||||
|
"preview": "vite preview",
|
||||||
|
"lint": "eslint . --ext .vue,.ts,.tsx --fix"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@element-plus/icons-vue": "^2.3.1",
|
||||||
|
"axios": "^1.18.1",
|
||||||
|
"element-plus": "^2.5.6",
|
||||||
|
"pinia": "^2.1.7",
|
||||||
|
"vue": "^3.4.21",
|
||||||
|
"vue-router": "^4.3.0"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@vitejs/plugin-vue": "^5.0.4",
|
||||||
|
"@vue/tsconfig": "^0.5.1",
|
||||||
|
"prettier": "3.9.6",
|
||||||
|
"sass": "^1.71.1",
|
||||||
|
"sass-embedded": "^1.100.0",
|
||||||
|
"typescript": "^5.9.3",
|
||||||
|
"vite": "^5.1.5",
|
||||||
|
"vue-tsc": "^2.2.12"
|
||||||
|
}
|
||||||
|
}
|
||||||
Binary file not shown.
|
After Width: | Height: | Size: 97 KiB |
@@ -0,0 +1,9 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { RouterView } from "vue-router";
|
||||||
|
import ToastContainer from "@/components/ToastContainer.vue";
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<RouterView />
|
||||||
|
<ToastContainer />
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,67 @@
|
|||||||
|
import axios, { AxiosError, type InternalAxiosRequestConfig } from "axios";
|
||||||
|
|
||||||
|
const http = axios.create({ baseURL: "/api/admin", timeout: 20000 });
|
||||||
|
let refreshPromise: Promise<string> | null = null;
|
||||||
|
|
||||||
|
http.interceptors.request.use((config) => {
|
||||||
|
const token = localStorage.getItem("admin_access_token");
|
||||||
|
if (token) config.headers.Authorization = `Bearer ${token}`;
|
||||||
|
return config;
|
||||||
|
});
|
||||||
|
|
||||||
|
http.interceptors.response.use(
|
||||||
|
(response) => response,
|
||||||
|
async (error: AxiosError) => {
|
||||||
|
const request = error.config as (InternalAxiosRequestConfig & { _retried?: boolean }) | undefined;
|
||||||
|
const canRefresh = !request?.url?.includes("/auth/");
|
||||||
|
if (error.response?.status === 401 && request && !request._retried && canRefresh) {
|
||||||
|
request._retried = true;
|
||||||
|
try {
|
||||||
|
const token = await refreshAccessToken();
|
||||||
|
request.headers.Authorization = `Bearer ${token}`;
|
||||||
|
return http(request);
|
||||||
|
} catch {
|
||||||
|
clearAuth();
|
||||||
|
if (!location.pathname.endsWith("/login")) location.href = "/admin/login";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Promise.reject(error);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
async function refreshAccessToken() {
|
||||||
|
if (!refreshPromise) {
|
||||||
|
refreshPromise = axios
|
||||||
|
.post("/api/admin/auth/refresh", { refresh_token: localStorage.getItem("admin_refresh_token") })
|
||||||
|
.then(({ data }) => {
|
||||||
|
saveTokens(data.data);
|
||||||
|
return data.data.access_token as string;
|
||||||
|
})
|
||||||
|
.finally(() => {
|
||||||
|
refreshPromise = null;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return refreshPromise;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function saveTokens(data: { access_token: string; refresh_token: string; admin?: unknown }) {
|
||||||
|
localStorage.setItem("admin_access_token", data.access_token);
|
||||||
|
localStorage.setItem("admin_refresh_token", data.refresh_token);
|
||||||
|
if (data.admin) localStorage.setItem("admin_profile", JSON.stringify(data.admin));
|
||||||
|
}
|
||||||
|
|
||||||
|
export function clearAuth() {
|
||||||
|
localStorage.removeItem("admin_access_token");
|
||||||
|
localStorage.removeItem("admin_refresh_token");
|
||||||
|
localStorage.removeItem("admin_profile");
|
||||||
|
}
|
||||||
|
|
||||||
|
export function apiError(error: unknown) {
|
||||||
|
const axiosError = error as AxiosError<{ code?: string; message?: string; trace_id?: string }>;
|
||||||
|
const body = axiosError.response?.data;
|
||||||
|
const message = body?.message || axiosError.message || "请求失败";
|
||||||
|
const copyText = `错误码：${body?.code || "request_failed"}\n错误描述：${message}\n请求追踪 ID：${body?.trace_id || axiosError.response?.headers?.["x-request-id"] || "无"}`;
|
||||||
|
return { message, copyText };
|
||||||
|
}
|
||||||
|
|
||||||
|
export default http;
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
import logoUrl from "../../../WEB/src/assets/logo.png";
|
||||||
|
|
||||||
|
export default logoUrl;
|
||||||
@@ -0,0 +1,837 @@
|
|||||||
|
@use "./variables.scss";
|
||||||
|
|
||||||
|
*,
|
||||||
|
*::before,
|
||||||
|
*::after {
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
html {
|
||||||
|
font-size: 16px;
|
||||||
|
-webkit-font-smoothing: antialiased;
|
||||||
|
-moz-osx-font-smoothing: grayscale;
|
||||||
|
}
|
||||||
|
body {
|
||||||
|
margin: 0;
|
||||||
|
min-width: 320px;
|
||||||
|
font-family: Inter, "PingFang SC", "Microsoft YaHei", Arial, sans-serif;
|
||||||
|
color: var(--color-text);
|
||||||
|
background: var(--color-bg);
|
||||||
|
line-height: 1.5;
|
||||||
|
}
|
||||||
|
button,
|
||||||
|
input,
|
||||||
|
textarea,
|
||||||
|
select {
|
||||||
|
font: inherit;
|
||||||
|
}
|
||||||
|
button {
|
||||||
|
touch-action: manipulation;
|
||||||
|
}
|
||||||
|
#app {
|
||||||
|
width: 100%;
|
||||||
|
min-height: 100dvh;
|
||||||
|
}
|
||||||
|
a {
|
||||||
|
color: var(--color-primary);
|
||||||
|
text-decoration: none;
|
||||||
|
}
|
||||||
|
h1,
|
||||||
|
h2,
|
||||||
|
h3,
|
||||||
|
h4,
|
||||||
|
p {
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
img {
|
||||||
|
max-width: 100%;
|
||||||
|
height: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
.admin-shell {
|
||||||
|
min-height: 100dvh;
|
||||||
|
padding-left: var(--sidebar-width);
|
||||||
|
}
|
||||||
|
.sidebar {
|
||||||
|
position: fixed;
|
||||||
|
inset: 0 auto 0 0;
|
||||||
|
z-index: 40;
|
||||||
|
width: var(--sidebar-width);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
padding: 12px 10px;
|
||||||
|
background: #111827;
|
||||||
|
color: var(--color-sidebar-text);
|
||||||
|
border-right: 1px solid #202c3e;
|
||||||
|
box-shadow: 4px 0 16px rgba(15, 23, 42, 0.08);
|
||||||
|
transition: width 0.2s ease;
|
||||||
|
}
|
||||||
|
.brand {
|
||||||
|
height: 54px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 0 8px 10px;
|
||||||
|
color: #f8fafc;
|
||||||
|
font-size: 17px;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
.brand strong {
|
||||||
|
font-weight: 700;
|
||||||
|
letter-spacing: 0.02em;
|
||||||
|
}
|
||||||
|
.brand-mark {
|
||||||
|
width: 34px;
|
||||||
|
height: 34px;
|
||||||
|
flex: 0 0 34px;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
overflow: hidden;
|
||||||
|
border: 1px solid #314058;
|
||||||
|
border-radius: 9px;
|
||||||
|
background: #182235;
|
||||||
|
}
|
||||||
|
.brand-mark img {
|
||||||
|
width: 28px;
|
||||||
|
height: 28px;
|
||||||
|
object-fit: contain;
|
||||||
|
}
|
||||||
|
.menu {
|
||||||
|
display: flex;
|
||||||
|
flex: 1;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 3px;
|
||||||
|
min-height: 0;
|
||||||
|
margin-top: 8px;
|
||||||
|
overflow-y: auto;
|
||||||
|
scrollbar-width: thin;
|
||||||
|
scrollbar-color: #344258 transparent;
|
||||||
|
}
|
||||||
|
.menu-item {
|
||||||
|
width: 100%;
|
||||||
|
min-height: 38px;
|
||||||
|
border: 0;
|
||||||
|
border-radius: 7px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 0 10px;
|
||||||
|
background: transparent;
|
||||||
|
color: #9cabc1;
|
||||||
|
cursor: pointer;
|
||||||
|
text-align: left;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
transition:
|
||||||
|
background-color 0.15s ease,
|
||||||
|
color 0.15s ease,
|
||||||
|
box-shadow 0.15s ease;
|
||||||
|
}
|
||||||
|
.menu-item .el-icon {
|
||||||
|
color: #8495ad;
|
||||||
|
transition: color 0.15s ease;
|
||||||
|
}
|
||||||
|
.menu-item:hover {
|
||||||
|
background: #182335;
|
||||||
|
color: #dce5f2;
|
||||||
|
}
|
||||||
|
.menu-item:hover .el-icon {
|
||||||
|
color: #b7c6da;
|
||||||
|
}
|
||||||
|
.menu-item.active {
|
||||||
|
background: #24344d;
|
||||||
|
color: #fff;
|
||||||
|
box-shadow: inset 2px 0 #60a5fa;
|
||||||
|
}
|
||||||
|
.menu-item.active .el-icon {
|
||||||
|
color: #e5edf8;
|
||||||
|
}
|
||||||
|
.menu-item.logout {
|
||||||
|
flex: 0 0 auto;
|
||||||
|
margin-top: 10px;
|
||||||
|
background: transparent;
|
||||||
|
color: #8493aa;
|
||||||
|
}
|
||||||
|
.menu-item.logout:hover {
|
||||||
|
background: #182335;
|
||||||
|
color: #dce5f2;
|
||||||
|
}
|
||||||
|
.is-collapsed {
|
||||||
|
padding-left: var(--sidebar-collapsed-width);
|
||||||
|
}
|
||||||
|
.is-collapsed .sidebar {
|
||||||
|
width: var(--sidebar-collapsed-width);
|
||||||
|
padding-inline: 12px;
|
||||||
|
}
|
||||||
|
.is-collapsed .brand {
|
||||||
|
justify-content: center;
|
||||||
|
padding-inline: 0;
|
||||||
|
}
|
||||||
|
.is-collapsed .menu-item {
|
||||||
|
justify-content: center;
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
|
.shell-main {
|
||||||
|
min-width: 0;
|
||||||
|
}
|
||||||
|
.topbar {
|
||||||
|
position: sticky;
|
||||||
|
top: 0;
|
||||||
|
z-index: 30;
|
||||||
|
height: var(--topbar-height);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 16px;
|
||||||
|
padding: 0 28px;
|
||||||
|
border-bottom: 1px solid var(--color-border-light);
|
||||||
|
background: rgba(255, 255, 255, 0.96);
|
||||||
|
}
|
||||||
|
.topbar h1 {
|
||||||
|
font-size: 18px;
|
||||||
|
line-height: 1.3;
|
||||||
|
}
|
||||||
|
.topbar p {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
}
|
||||||
|
.topbar-menu {
|
||||||
|
flex: 0 0 44px;
|
||||||
|
}
|
||||||
|
.admin-profile {
|
||||||
|
margin-left: auto;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
.page-content {
|
||||||
|
width: 100%;
|
||||||
|
max-width: 1600px;
|
||||||
|
margin: 0 auto;
|
||||||
|
padding: 24px 28px 40px;
|
||||||
|
}
|
||||||
|
.page-stack {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 20px;
|
||||||
|
}
|
||||||
|
.management-page {
|
||||||
|
height: max(814px, calc(100dvh - 64px));
|
||||||
|
min-height: 814px;
|
||||||
|
}
|
||||||
|
.management-page > .table-panel {
|
||||||
|
display: flex;
|
||||||
|
flex: 1;
|
||||||
|
min-height: 0;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
.management-page > .table-panel > .el-pagination {
|
||||||
|
margin-top: auto;
|
||||||
|
padding-top: 18px;
|
||||||
|
}
|
||||||
|
.page-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 20px;
|
||||||
|
}
|
||||||
|
.page-header h2 {
|
||||||
|
font-size: var(--font-size-xl);
|
||||||
|
line-height: 1.3;
|
||||||
|
}
|
||||||
|
.page-header p {
|
||||||
|
margin-top: 5px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
.page-actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
gap: 10px;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
}
|
||||||
|
.panel {
|
||||||
|
min-width: 0;
|
||||||
|
padding: 20px;
|
||||||
|
border: 1px solid var(--color-border-light);
|
||||||
|
border-radius: var(--radius-lg);
|
||||||
|
background: var(--color-surface);
|
||||||
|
box-shadow: var(--shadow-sm);
|
||||||
|
}
|
||||||
|
.table-panel {
|
||||||
|
overflow: auto;
|
||||||
|
}
|
||||||
|
.filter-bar {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
}
|
||||||
|
.filter-bar > .el-input {
|
||||||
|
width: min(280px, 100%);
|
||||||
|
}
|
||||||
|
.filter-bar > .el-select {
|
||||||
|
width: 180px;
|
||||||
|
}
|
||||||
|
.filter-bar > .el-date-editor {
|
||||||
|
max-width: 420px;
|
||||||
|
}
|
||||||
|
.batch-toolbar {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
padding: 12px 16px;
|
||||||
|
border: 1px solid #bfdbfe;
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
background: var(--color-primary-soft);
|
||||||
|
}
|
||||||
|
.batch-toolbar strong {
|
||||||
|
margin-right: auto;
|
||||||
|
}
|
||||||
|
.section-heading {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 12px;
|
||||||
|
margin-bottom: 14px;
|
||||||
|
}
|
||||||
|
.section-heading h3 {
|
||||||
|
font-size: 17px;
|
||||||
|
}
|
||||||
|
.section-heading span {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
.channel-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(240px, 1fr));
|
||||||
|
gap: 16px;
|
||||||
|
}
|
||||||
|
.balance-card {
|
||||||
|
padding: 18px;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-lg);
|
||||||
|
background: #fff;
|
||||||
|
}
|
||||||
|
.balance-card.warning {
|
||||||
|
border-color: #f59e0b;
|
||||||
|
background: #fffbeb;
|
||||||
|
}
|
||||||
|
.balance-card__header {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
.balance-card__header strong,
|
||||||
|
.balance-card__header span {
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
.balance-card__header span,
|
||||||
|
.balance-card p {
|
||||||
|
margin-top: 3px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
.balance-value {
|
||||||
|
margin: 22px 0 14px;
|
||||||
|
font-size: 28px;
|
||||||
|
font-weight: 650;
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
.balance-value small {
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
}
|
||||||
|
.line-chart {
|
||||||
|
width: 100%;
|
||||||
|
height: 360px;
|
||||||
|
}
|
||||||
|
.icon-button {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
padding: 0;
|
||||||
|
border: 0;
|
||||||
|
border-radius: 6px;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
background: transparent;
|
||||||
|
color: inherit;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
.icon-button:hover {
|
||||||
|
background: rgba(15, 23, 42, 0.08);
|
||||||
|
}
|
||||||
|
.el-button .el-icon {
|
||||||
|
margin-right: 6px;
|
||||||
|
}
|
||||||
|
.field-full {
|
||||||
|
width: 100% !important;
|
||||||
|
}
|
||||||
|
.field-helper {
|
||||||
|
width: 100%;
|
||||||
|
margin-top: 6px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
line-height: 1.55;
|
||||||
|
}
|
||||||
|
.form-grid,
|
||||||
|
.dialog-form-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(2, minmax(0, 1fr));
|
||||||
|
gap: 0 18px;
|
||||||
|
}
|
||||||
|
.dialog-field--full {
|
||||||
|
grid-column: 1/-1;
|
||||||
|
}
|
||||||
|
.style-thumb {
|
||||||
|
width: 64px;
|
||||||
|
height: 48px;
|
||||||
|
border-radius: 6px;
|
||||||
|
object-fit: cover;
|
||||||
|
background: var(--color-bg);
|
||||||
|
}
|
||||||
|
.upload-preview {
|
||||||
|
display: block;
|
||||||
|
width: 120px;
|
||||||
|
height: 84px;
|
||||||
|
margin-top: 10px;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: 8px;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
.code-output {
|
||||||
|
margin-top: 16px;
|
||||||
|
}
|
||||||
|
.el-pagination {
|
||||||
|
justify-content: flex-end;
|
||||||
|
margin-top: 18px;
|
||||||
|
}
|
||||||
|
.el-table {
|
||||||
|
--el-table-header-bg-color: #f8fafc;
|
||||||
|
--el-table-row-hover-bg-color: #f8fafc;
|
||||||
|
}
|
||||||
|
.el-table th.el-table__cell {
|
||||||
|
color: #39465a;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
.mobile-card-list {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
.data-card {
|
||||||
|
padding: 16px;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
}
|
||||||
|
.data-card dl {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: 110px minmax(0, 1fr);
|
||||||
|
gap: 8px;
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
.data-card dt {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
}
|
||||||
|
.data-card dd {
|
||||||
|
margin: 0;
|
||||||
|
overflow-wrap: anywhere;
|
||||||
|
}
|
||||||
|
.card-actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
gap: 8px;
|
||||||
|
margin-top: 14px;
|
||||||
|
}
|
||||||
|
.el-overlay {
|
||||||
|
background: rgba(15, 23, 42, 0.58);
|
||||||
|
}
|
||||||
|
.el-overlay-dialog {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
.el-dialog {
|
||||||
|
margin: 0;
|
||||||
|
border: 1px solid #d3dbe7;
|
||||||
|
border-radius: 14px;
|
||||||
|
background: #fff;
|
||||||
|
box-shadow: 0 24px 70px rgba(15, 23, 42, 0.24);
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
.el-dialog__header {
|
||||||
|
margin: 0;
|
||||||
|
padding: 19px 24px 16px;
|
||||||
|
border-bottom: 1px solid var(--color-border-light);
|
||||||
|
background: #f8fafc;
|
||||||
|
}
|
||||||
|
.el-dialog__title {
|
||||||
|
color: var(--color-text);
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 650;
|
||||||
|
letter-spacing: 0.01em;
|
||||||
|
}
|
||||||
|
.el-dialog__headerbtn {
|
||||||
|
top: 12px;
|
||||||
|
right: 14px;
|
||||||
|
width: 36px;
|
||||||
|
height: 36px;
|
||||||
|
border-radius: 8px;
|
||||||
|
}
|
||||||
|
.el-dialog__headerbtn:hover {
|
||||||
|
background: #e9eef5;
|
||||||
|
}
|
||||||
|
.el-dialog__headerbtn .el-dialog__close {
|
||||||
|
color: #718096;
|
||||||
|
font-size: 18px;
|
||||||
|
}
|
||||||
|
.el-dialog__body {
|
||||||
|
max-height: calc(100dvh - 180px);
|
||||||
|
padding: 22px 24px 6px;
|
||||||
|
overflow-y: auto;
|
||||||
|
}
|
||||||
|
.el-dialog.managed-scroll-dialog {
|
||||||
|
margin: 0;
|
||||||
|
height: min(760px, calc(100dvh - 40px));
|
||||||
|
max-height: calc(100dvh - 40px);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
.el-dialog.managed-scroll-dialog .el-dialog__header,
|
||||||
|
.el-dialog.managed-scroll-dialog .el-dialog__footer {
|
||||||
|
flex: 0 0 auto;
|
||||||
|
}
|
||||||
|
.el-dialog.managed-scroll-dialog .el-dialog__body {
|
||||||
|
min-height: 0;
|
||||||
|
max-height: none;
|
||||||
|
flex: 1 1 auto;
|
||||||
|
overflow-y: auto;
|
||||||
|
overscroll-behavior: contain;
|
||||||
|
scrollbar-gutter: stable;
|
||||||
|
}
|
||||||
|
.managed-scroll-overlay .el-overlay-dialog {
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
overflow: hidden;
|
||||||
|
overscroll-behavior: none;
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__footer {
|
||||||
|
padding: 14px 24px 16px;
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
gap: 10px;
|
||||||
|
border-top: 0;
|
||||||
|
background: transparent;
|
||||||
|
}
|
||||||
|
.el-dialog__footer .el-button {
|
||||||
|
min-width: 82px;
|
||||||
|
height: 38px;
|
||||||
|
margin-left: 0;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
.el-dialog .el-form-item {
|
||||||
|
margin-bottom: 19px;
|
||||||
|
}
|
||||||
|
.el-dialog .el-form-item__label {
|
||||||
|
padding-bottom: 7px;
|
||||||
|
color: #39465a;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 600;
|
||||||
|
line-height: 1.4;
|
||||||
|
}
|
||||||
|
.el-dialog .el-input__wrapper,
|
||||||
|
.el-dialog .el-select__wrapper {
|
||||||
|
min-height: 40px;
|
||||||
|
border-radius: 8px;
|
||||||
|
box-shadow: 0 0 0 1px #d5dce6 inset;
|
||||||
|
}
|
||||||
|
.el-dialog .el-input__wrapper:hover,
|
||||||
|
.el-dialog .el-select__wrapper:hover {
|
||||||
|
box-shadow: 0 0 0 1px #9bb8e7 inset;
|
||||||
|
}
|
||||||
|
.el-dialog .el-input__wrapper.is-focus,
|
||||||
|
.el-dialog .el-select__wrapper.is-focused {
|
||||||
|
box-shadow: 0 0 0 1px var(--color-primary) inset;
|
||||||
|
}
|
||||||
|
.el-dialog .el-textarea__inner {
|
||||||
|
border-radius: 8px;
|
||||||
|
box-shadow: 0 0 0 1px #d5dce6 inset;
|
||||||
|
line-height: 1.65;
|
||||||
|
}
|
||||||
|
.el-dialog .el-input-number {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
.el-dialog .el-upload .el-button {
|
||||||
|
height: 38px;
|
||||||
|
border-radius: 8px;
|
||||||
|
}
|
||||||
|
.el-dialog .el-switch {
|
||||||
|
--el-switch-on-color: var(--color-primary);
|
||||||
|
}
|
||||||
|
.toast-region {
|
||||||
|
position: fixed;
|
||||||
|
z-index: 3000;
|
||||||
|
top: 18px;
|
||||||
|
left: 50%;
|
||||||
|
width: min(440px, calc(100vw - 32px));
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 10px;
|
||||||
|
transform: translateX(-50%);
|
||||||
|
pointer-events: none;
|
||||||
|
}
|
||||||
|
.app-toast {
|
||||||
|
position: relative;
|
||||||
|
min-height: 68px;
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 14px 12px 16px;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-left: 4px solid var(--color-info);
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
background: #fff;
|
||||||
|
box-shadow: var(--shadow-md);
|
||||||
|
overflow: hidden;
|
||||||
|
pointer-events: auto;
|
||||||
|
}
|
||||||
|
.app-toast--success {
|
||||||
|
border-left-color: var(--color-success);
|
||||||
|
}
|
||||||
|
.app-toast--warning {
|
||||||
|
border-left-color: var(--color-warning);
|
||||||
|
}
|
||||||
|
.app-toast--error {
|
||||||
|
border-left-color: var(--color-danger);
|
||||||
|
}
|
||||||
|
.app-toast__icon {
|
||||||
|
margin-top: 3px;
|
||||||
|
}
|
||||||
|
.app-toast__content {
|
||||||
|
min-width: 0;
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
.app-toast__content span {
|
||||||
|
margin-top: 2px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 13px;
|
||||||
|
overflow-wrap: anywhere;
|
||||||
|
}
|
||||||
|
.app-toast .icon-button {
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
flex: 0 0 32px;
|
||||||
|
}
|
||||||
|
.app-toast__progress {
|
||||||
|
position: absolute;
|
||||||
|
left: 0;
|
||||||
|
bottom: 0;
|
||||||
|
height: 3px;
|
||||||
|
background: var(--color-primary);
|
||||||
|
transition: width 0.05s linear;
|
||||||
|
}
|
||||||
|
.toast-fade-enter-active,
|
||||||
|
.toast-fade-leave-active {
|
||||||
|
transition: opacity 0.2s ease;
|
||||||
|
}
|
||||||
|
.toast-fade-enter-from,
|
||||||
|
.toast-fade-leave-to {
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
.login-page {
|
||||||
|
min-height: 100dvh;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
padding: 24px;
|
||||||
|
background: #eef2f7;
|
||||||
|
}
|
||||||
|
.login-panel {
|
||||||
|
width: min(440px, 100%);
|
||||||
|
padding: 34px;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-lg);
|
||||||
|
background: #fff;
|
||||||
|
box-shadow: var(--shadow-md);
|
||||||
|
}
|
||||||
|
.login-brand {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
font-weight: 650;
|
||||||
|
}
|
||||||
|
.login-heading {
|
||||||
|
margin: 32px 0 24px;
|
||||||
|
}
|
||||||
|
.login-heading h1 {
|
||||||
|
font-size: 26px;
|
||||||
|
}
|
||||||
|
.login-heading p {
|
||||||
|
margin-top: 6px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
}
|
||||||
|
.login-submit {
|
||||||
|
width: 100%;
|
||||||
|
margin-top: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.admin-shell {
|
||||||
|
padding-left: 0;
|
||||||
|
}
|
||||||
|
.page-content {
|
||||||
|
padding: 18px 14px 32px;
|
||||||
|
}
|
||||||
|
.topbar {
|
||||||
|
height: 64px;
|
||||||
|
padding: 0 14px;
|
||||||
|
}
|
||||||
|
.admin-profile span {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
.management-page {
|
||||||
|
height: auto;
|
||||||
|
min-height: 0;
|
||||||
|
}
|
||||||
|
.page-header {
|
||||||
|
align-items: stretch;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
.page-actions {
|
||||||
|
justify-content: flex-start;
|
||||||
|
}
|
||||||
|
.page-actions .el-button {
|
||||||
|
margin-left: 0;
|
||||||
|
}
|
||||||
|
.panel {
|
||||||
|
padding: 14px;
|
||||||
|
}
|
||||||
|
.filter-bar {
|
||||||
|
align-items: stretch;
|
||||||
|
}
|
||||||
|
.filter-bar > * {
|
||||||
|
width: 100% !important;
|
||||||
|
max-width: none !important;
|
||||||
|
}
|
||||||
|
.form-grid,
|
||||||
|
.dialog-form-grid {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
.dialog-field--full {
|
||||||
|
grid-column: auto;
|
||||||
|
}
|
||||||
|
.desktop-table {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
.mobile-card-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
.line-chart {
|
||||||
|
height: 300px;
|
||||||
|
}
|
||||||
|
.channel-grid {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
.login-panel {
|
||||||
|
padding: 26px 20px;
|
||||||
|
}
|
||||||
|
.el-overlay-dialog {
|
||||||
|
padding: 12px;
|
||||||
|
}
|
||||||
|
.el-dialog {
|
||||||
|
margin: 0 !important;
|
||||||
|
}
|
||||||
|
.el-dialog__header {
|
||||||
|
padding: 17px 18px 14px;
|
||||||
|
}
|
||||||
|
.el-dialog__body {
|
||||||
|
max-height: calc(100dvh - 150px);
|
||||||
|
padding: 18px 18px 4px;
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__footer {
|
||||||
|
padding: 18px 18px 12px;
|
||||||
|
}
|
||||||
|
.el-pagination {
|
||||||
|
justify-content: center;
|
||||||
|
}
|
||||||
|
.menu {
|
||||||
|
margin-bottom: 12px;
|
||||||
|
}
|
||||||
|
.mobile-drawer .el-drawer__body {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
padding: 16px 12px;
|
||||||
|
background: var(--color-sidebar);
|
||||||
|
color: var(--color-sidebar-text);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.desktop-user-table {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__header {
|
||||||
|
padding: 8px 24px 2px;
|
||||||
|
border-bottom: 0;
|
||||||
|
background: transparent;
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__headerbtn {
|
||||||
|
top: 7px;
|
||||||
|
right: 11px;
|
||||||
|
}
|
||||||
|
.el-dialog__body {
|
||||||
|
padding-top: 16px;
|
||||||
|
}
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.el-dialog .el-dialog__header {
|
||||||
|
padding: 6px 18px 2px;
|
||||||
|
}
|
||||||
|
.el-dialog__body {
|
||||||
|
padding-top: 14px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.el-dialog[style*="720px"] .el-form {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(2, minmax(0, 1fr));
|
||||||
|
gap: 0 18px;
|
||||||
|
}
|
||||||
|
.el-dialog[style*="720px"] .el-form > .form-grid {
|
||||||
|
grid-column: 1/-1;
|
||||||
|
}
|
||||||
|
.el-dialog[style*="720px"] .el-form > .el-form-item:has(.el-textarea__inner),
|
||||||
|
.el-dialog[style*="720px"] .el-form > .el-form-item:has(.el-upload),
|
||||||
|
.el-dialog[style*="720px"] .el-form > .el-form-item:has(.el-input__password) {
|
||||||
|
grid-column: 1/-1;
|
||||||
|
}
|
||||||
|
.el-dialog[style*="720px"] .el-form > .el-form-item {
|
||||||
|
margin-bottom: 16px;
|
||||||
|
}
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.el-dialog[style*="720px"] .el-form {
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__headerbtn {
|
||||||
|
top: 4px;
|
||||||
|
right: 10px;
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
border-radius: 0;
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__headerbtn .el-dialog__close {
|
||||||
|
color: #718096;
|
||||||
|
font-size: 15px;
|
||||||
|
transition: color 0.16s ease;
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__headerbtn:hover {
|
||||||
|
background: transparent;
|
||||||
|
}
|
||||||
|
.el-dialog .el-dialog__headerbtn:hover .el-dialog__close {
|
||||||
|
color: var(--color-primary);
|
||||||
|
}
|
||||||
@@ -0,0 +1,36 @@
|
|||||||
|
:root {
|
||||||
|
--color-primary: #4f46e5;
|
||||||
|
--color-primary-hover: #4338ca;
|
||||||
|
--color-primary-soft: #eef2ff;
|
||||||
|
--color-success: #15803d;
|
||||||
|
--color-warning: #b45309;
|
||||||
|
--color-danger: #b91c1c;
|
||||||
|
--color-info: #64748b;
|
||||||
|
--color-text: #172033;
|
||||||
|
--color-text-secondary: #64748b;
|
||||||
|
--color-text-placeholder: #94a3b8;
|
||||||
|
--color-border: #e2e8f0;
|
||||||
|
--color-border-light: #edf1f6;
|
||||||
|
--color-bg: #f4f6fb;
|
||||||
|
--color-surface: #ffffff;
|
||||||
|
--color-sidebar: #111827;
|
||||||
|
--color-sidebar-text: #cbd5e1;
|
||||||
|
--space-xs: 4px;
|
||||||
|
--space-sm: 8px;
|
||||||
|
--space-md: 16px;
|
||||||
|
--space-lg: 24px;
|
||||||
|
--space-xl: 32px;
|
||||||
|
--font-size-xs: 12px;
|
||||||
|
--font-size-sm: 14px;
|
||||||
|
--font-size-md: 16px;
|
||||||
|
--font-size-lg: 20px;
|
||||||
|
--font-size-xl: 26px;
|
||||||
|
--radius-sm: 5px;
|
||||||
|
--radius-md: 8px;
|
||||||
|
--radius-lg: 12px;
|
||||||
|
--shadow-sm: 0 1px 3px rgba(15, 23, 42, 0.08);
|
||||||
|
--shadow-md: 0 10px 30px rgba(15, 23, 42, 0.12);
|
||||||
|
--sidebar-width: 224px;
|
||||||
|
--sidebar-collapsed-width: 72px;
|
||||||
|
--topbar-height: 72px;
|
||||||
|
}
|
||||||
@@ -0,0 +1,41 @@
|
|||||||
|
<!-- 管理后台左侧菜单组件，统一维护菜单入口、图标和展示顺序。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import { adminMenuItems } from "@/router/adminNavigation";
|
||||||
|
|
||||||
|
defineProps<{ activePath: string; collapsed?: boolean }>();
|
||||||
|
const emit = defineEmits<{ navigate: [path: string] }>();
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<nav
|
||||||
|
class="menu admin-menu"
|
||||||
|
:class="{ 'admin-menu--collapsed': collapsed }"
|
||||||
|
>
|
||||||
|
<button
|
||||||
|
v-for="item in adminMenuItems"
|
||||||
|
:key="item.path"
|
||||||
|
class="menu-item"
|
||||||
|
:class="{ active: activePath === item.path }"
|
||||||
|
type="button"
|
||||||
|
:title="collapsed ? item.label : undefined"
|
||||||
|
@click="emit('navigate', item.path)"
|
||||||
|
>
|
||||||
|
<AppIcon
|
||||||
|
:name="item.icon"
|
||||||
|
:size="20"
|
||||||
|
/>
|
||||||
|
<span v-if="!collapsed">{{ item.label }}</span>
|
||||||
|
</button>
|
||||||
|
</nav>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.admin-menu {
|
||||||
|
gap: 5px;
|
||||||
|
padding: 12px 1px 8px;
|
||||||
|
}
|
||||||
|
.admin-menu--collapsed {
|
||||||
|
padding-inline: 0;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,123 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { ref, watch } from "vue";
|
||||||
|
|
||||||
|
interface Props {
|
||||||
|
modelValue?: number | string | null;
|
||||||
|
min?: number;
|
||||||
|
max?: number;
|
||||||
|
precision?: 0 | 2;
|
||||||
|
disabled?: boolean;
|
||||||
|
readonly?: boolean;
|
||||||
|
placeholder?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
const props = withDefaults(defineProps<Props>(), {
|
||||||
|
modelValue: null,
|
||||||
|
precision: 2,
|
||||||
|
disabled: false,
|
||||||
|
readonly: false,
|
||||||
|
placeholder: "",
|
||||||
|
});
|
||||||
|
const emit = defineEmits<{ "update:modelValue": [value: number | null] }>();
|
||||||
|
const text = ref(formatValue(props.modelValue));
|
||||||
|
const focused = ref(false);
|
||||||
|
|
||||||
|
watch(
|
||||||
|
() => props.modelValue,
|
||||||
|
(value) => {
|
||||||
|
if (!focused.value) text.value = formatValue(value);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
function formatValue(value: number | string | null | undefined) {
|
||||||
|
if (value === null || value === undefined || value === "") return "";
|
||||||
|
const numeric = Number(value);
|
||||||
|
if (!Number.isFinite(numeric)) return "";
|
||||||
|
const factor = 10 ** props.precision;
|
||||||
|
return String(Math.round(numeric * factor) / factor);
|
||||||
|
}
|
||||||
|
|
||||||
|
function sanitize(value: string) {
|
||||||
|
const normalized = value.replace(/,/g, ".").replace(/[^\d.-]/g, "");
|
||||||
|
const negative = normalized.startsWith("-");
|
||||||
|
const unsigned = normalized.replace(/-/g, "");
|
||||||
|
const [integer = "", ...decimalParts] = unsigned.split(".");
|
||||||
|
const sign = negative ? "-" : "";
|
||||||
|
if (props.precision === 0) return sign + integer;
|
||||||
|
const hasDecimalPoint = unsigned.includes(".");
|
||||||
|
const decimal = decimalParts.join("").slice(0, props.precision);
|
||||||
|
return sign + integer + (hasDecimalPoint ? `.${decimal}` : "");
|
||||||
|
}
|
||||||
|
|
||||||
|
function update(value: string) {
|
||||||
|
text.value = sanitize(value);
|
||||||
|
if (text.value === "" || text.value === "-" || text.value === "." || text.value === "-.") {
|
||||||
|
emit("update:modelValue", null);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const numeric = Number(text.value);
|
||||||
|
if (Number.isFinite(numeric)) emit("update:modelValue", numeric);
|
||||||
|
}
|
||||||
|
|
||||||
|
function commit() {
|
||||||
|
focused.value = false;
|
||||||
|
if (text.value === "" || text.value === "-" || text.value === "." || text.value === "-.") {
|
||||||
|
text.value = "";
|
||||||
|
emit("update:modelValue", null);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let value = Number(text.value);
|
||||||
|
if (!Number.isFinite(value)) {
|
||||||
|
text.value = formatValue(props.modelValue);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (props.min !== undefined) value = Math.max(props.min, value);
|
||||||
|
if (props.max !== undefined) value = Math.min(props.max, value);
|
||||||
|
const factor = 10 ** props.precision;
|
||||||
|
value = Math.round(value * factor) / factor;
|
||||||
|
text.value = String(value);
|
||||||
|
emit("update:modelValue", value);
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<el-input
|
||||||
|
class="admin-number-input"
|
||||||
|
:class="{ 'is-readonly': readonly }"
|
||||||
|
:model-value="text"
|
||||||
|
type="text"
|
||||||
|
:inputmode="precision === 0 ? 'numeric' : 'decimal'"
|
||||||
|
:disabled="disabled"
|
||||||
|
:readonly="readonly"
|
||||||
|
:placeholder="placeholder"
|
||||||
|
@focus="focused = true"
|
||||||
|
@input="update"
|
||||||
|
@blur="commit"
|
||||||
|
/>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.admin-number-input {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.admin-number-input :deep(.el-input__wrapper) {
|
||||||
|
min-height: 40px;
|
||||||
|
padding: 0 12px;
|
||||||
|
border-radius: var(--radius-sm);
|
||||||
|
box-shadow: 0 0 0 1px var(--color-border) inset;
|
||||||
|
}
|
||||||
|
|
||||||
|
.admin-number-input :deep(.el-input__wrapper.is-focus) {
|
||||||
|
box-shadow: 0 0 0 1px var(--color-primary) inset;
|
||||||
|
}
|
||||||
|
|
||||||
|
.admin-number-input :deep(.el-input__inner) {
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
text-align: left;
|
||||||
|
}
|
||||||
|
|
||||||
|
.admin-number-input.is-readonly :deep(.el-input__wrapper) {
|
||||||
|
background: var(--color-bg);
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
<!-- 剧游AI管理后台图标组件，统一封装第三方图标库的使用入口 -->
|
||||||
|
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { computed } from "vue";
|
||||||
|
import * as Icons from "@element-plus/icons-vue";
|
||||||
|
|
||||||
|
type IconName = keyof typeof Icons;
|
||||||
|
|
||||||
|
interface Props {
|
||||||
|
name: IconName;
|
||||||
|
size?: number | string;
|
||||||
|
color?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
const props = withDefaults(defineProps<Props>(), {
|
||||||
|
size: 16,
|
||||||
|
});
|
||||||
|
|
||||||
|
const iconComponent = computed(() => Icons[props.name]);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<el-icon
|
||||||
|
:size="size"
|
||||||
|
:color="color"
|
||||||
|
>
|
||||||
|
<component :is="iconComponent" />
|
||||||
|
</el-icon>
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,326 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, ref } from "vue";
|
||||||
|
import AdminNumberInput from "@/components/AdminNumberInput.vue";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
|
||||||
|
const props = defineProps<{
|
||||||
|
form: Record<string, any>;
|
||||||
|
editing: boolean;
|
||||||
|
}>();
|
||||||
|
const replacingAPIKey = ref(false);
|
||||||
|
|
||||||
|
const normalizedRate = computed(() => {
|
||||||
|
const cny = Number(props.form.exchange_cny_amount);
|
||||||
|
const points = Number(props.form.exchange_point_amount);
|
||||||
|
if (!Number.isFinite(cny) || cny <= 0 || !Number.isFinite(points) || points <= 0) return "--";
|
||||||
|
return (points / cny).toFixed(2);
|
||||||
|
});
|
||||||
|
|
||||||
|
function cancelAPIKeyReplacement() {
|
||||||
|
props.form.api_key = "";
|
||||||
|
replacingAPIKey.value = false;
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="channel-editor">
|
||||||
|
<section class="channel-section">
|
||||||
|
<header class="channel-section__header">
|
||||||
|
<span class="channel-section__icon"><AppIcon name="Connection" /></span>
|
||||||
|
<div>
|
||||||
|
<h3>基础连接</h3>
|
||||||
|
<p>配置渠道标识、接口地址和访问凭证。</p>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
<div class="channel-grid">
|
||||||
|
<el-form-item
|
||||||
|
label="渠道名"
|
||||||
|
prop="name"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model.trim="form.name"
|
||||||
|
placeholder="输入渠道名称"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="渠道 URL"
|
||||||
|
prop="base_url"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model.trim="form.base_url"
|
||||||
|
placeholder="https://api.example.com"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
class="channel-field--full"
|
||||||
|
label="API Key"
|
||||||
|
prop="api_key"
|
||||||
|
:required="!editing"
|
||||||
|
>
|
||||||
|
<template v-if="editing && !replacingAPIKey">
|
||||||
|
<div class="masked-key-row">
|
||||||
|
<el-input
|
||||||
|
:model-value="form.api_key_masked || '未配置'"
|
||||||
|
readonly
|
||||||
|
/>
|
||||||
|
<el-button @click="replacingAPIKey = true">更换 API Key</el-button>
|
||||||
|
</div>
|
||||||
|
<div class="field-helper">仅显示前后各 5 位，完整密钥不会返回管理端</div>
|
||||||
|
</template>
|
||||||
|
<template v-else>
|
||||||
|
<el-input
|
||||||
|
v-model="form.api_key"
|
||||||
|
type="password"
|
||||||
|
show-password
|
||||||
|
placeholder="输入新 API Key"
|
||||||
|
/>
|
||||||
|
<div class="api-key-helper-row">
|
||||||
|
<div class="field-helper">凭证将加密保存</div>
|
||||||
|
<el-button
|
||||||
|
v-if="editing"
|
||||||
|
link
|
||||||
|
@click="cancelAPIKeyReplacement"
|
||||||
|
>取消更换</el-button
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section class="channel-section">
|
||||||
|
<header class="channel-section__header">
|
||||||
|
<span class="channel-section__icon"><AppIcon name="Coin" /></span>
|
||||||
|
<div>
|
||||||
|
<h3>渠道汇率</h3>
|
||||||
|
<p>设置人民币与渠道积分的兑换关系，用于模型价格换算。</p>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
<div class="exchange-rate-row">
|
||||||
|
<el-form-item
|
||||||
|
label="人民币金额"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="form.exchange_cny_amount"
|
||||||
|
:min="0.01"
|
||||||
|
:precision="2"
|
||||||
|
placeholder="1.00"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<span class="exchange-rate-row__equals">=</span>
|
||||||
|
<el-form-item
|
||||||
|
label="渠道积分"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="form.exchange_point_amount"
|
||||||
|
:min="0.01"
|
||||||
|
:precision="2"
|
||||||
|
placeholder="1.00"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
<div class="exchange-rate-summary">
|
||||||
|
<span>归一化汇率</span>
|
||||||
|
<strong>1.00 元 = {{ normalizedRate }} 渠道积分</strong>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section class="channel-section">
|
||||||
|
<header class="channel-section__header">
|
||||||
|
<span class="channel-section__icon"><AppIcon name="Setting" /></span>
|
||||||
|
<div>
|
||||||
|
<h3>运行限制</h3>
|
||||||
|
<p>配置余额预警和任务并发上限。</p>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
<div class="channel-grid channel-grid--three">
|
||||||
|
<el-form-item
|
||||||
|
label="余额预警阈值"
|
||||||
|
prop="warning_threshold"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="form.warning_threshold"
|
||||||
|
:min="0"
|
||||||
|
:precision="2"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="最大并发数"
|
||||||
|
prop="max_concurrency"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="form.max_concurrency"
|
||||||
|
:min="1"
|
||||||
|
:max="5000"
|
||||||
|
:precision="0"
|
||||||
|
/>
|
||||||
|
<div class="field-helper">已提交且未进入终态的总任务数</div>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="单用户最高并发数"
|
||||||
|
prop="max_user_concurrency"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="form.max_user_concurrency"
|
||||||
|
:min="1"
|
||||||
|
:max="500"
|
||||||
|
:precision="0"
|
||||||
|
/>
|
||||||
|
<div class="field-helper">超出后继续排队，不计入轮询请求</div>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.channel-editor {
|
||||||
|
display: grid;
|
||||||
|
gap: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-section {
|
||||||
|
padding: 18px;
|
||||||
|
border: 1px solid var(--color-border-light);
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
background: var(--color-surface);
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-section__header {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
gap: 12px;
|
||||||
|
margin-bottom: 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-section__icon {
|
||||||
|
width: 34px;
|
||||||
|
height: 34px;
|
||||||
|
flex: 0 0 34px;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-sm);
|
||||||
|
background: var(--color-primary-soft);
|
||||||
|
color: var(--color-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-section__header h3 {
|
||||||
|
color: var(--color-text);
|
||||||
|
font-size: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-section__header p {
|
||||||
|
margin-top: 3px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
line-height: 1.5;
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(2, minmax(0, 1fr));
|
||||||
|
gap: 0 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-grid--three {
|
||||||
|
grid-template-columns: repeat(3, minmax(0, 1fr));
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-field--full {
|
||||||
|
grid-column: 1 / -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.masked-key-row {
|
||||||
|
width: 100%;
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: minmax(0, 1fr) auto;
|
||||||
|
gap: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.masked-key-row :deep(.el-input__inner) {
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
.masked-key-row > .el-button {
|
||||||
|
min-height: 40px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.api-key-helper-row {
|
||||||
|
width: 100%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-row {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: minmax(0, 1fr) 32px minmax(0, 1fr);
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-row__equals {
|
||||||
|
margin-top: 2px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 18px;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-summary {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 16px;
|
||||||
|
padding: 11px 13px;
|
||||||
|
border: 1px solid var(--color-border-light);
|
||||||
|
border-radius: var(--radius-sm);
|
||||||
|
background: var(--color-bg);
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-summary strong {
|
||||||
|
color: var(--color-text);
|
||||||
|
font-size: 13px;
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.channel-section {
|
||||||
|
padding: 14px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.channel-grid,
|
||||||
|
.channel-grid--three {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-row {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
gap: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-row__equals {
|
||||||
|
margin: -4px 0 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exchange-rate-summary {
|
||||||
|
align-items: flex-start;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 3px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.masked-key-row {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,100 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
defineProps<{ title: string; description: string; emptyText: string; hasSelection: boolean }>();
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<header class="page-header master-detail-heading">
|
||||||
|
<div>
|
||||||
|
<h2>{{ title }}</h2>
|
||||||
|
</div>
|
||||||
|
<div class="page-actions"><slot name="header-actions" /></div>
|
||||||
|
</header>
|
||||||
|
<section class="master-detail-panel">
|
||||||
|
<aside class="master-list">
|
||||||
|
<div class="master-list-toolbar"><slot name="list-toolbar" /></div>
|
||||||
|
<div class="master-list-items"><slot name="list" /></div>
|
||||||
|
</aside>
|
||||||
|
<section class="master-editor">
|
||||||
|
<slot
|
||||||
|
v-if="hasSelection"
|
||||||
|
name="detail"
|
||||||
|
/>
|
||||||
|
<div
|
||||||
|
v-else
|
||||||
|
class="master-empty"
|
||||||
|
>
|
||||||
|
{{ emptyText }}
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
</section>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.master-detail-heading {
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
.master-detail-panel {
|
||||||
|
display: grid;
|
||||||
|
flex: none;
|
||||||
|
height: max(760px, calc(100dvh - 118px));
|
||||||
|
min-height: 760px;
|
||||||
|
grid-template-columns: 300px minmax(0, 1fr);
|
||||||
|
overflow: hidden;
|
||||||
|
border: 1px solid var(--color-border-light);
|
||||||
|
border-radius: var(--radius-lg);
|
||||||
|
background: #fff;
|
||||||
|
box-shadow: var(--shadow-sm);
|
||||||
|
}
|
||||||
|
.master-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
min-width: 0;
|
||||||
|
min-height: 0;
|
||||||
|
border-right: 1px solid var(--color-border-light);
|
||||||
|
background: #f8fafc;
|
||||||
|
}
|
||||||
|
.master-list-toolbar {
|
||||||
|
padding: 16px;
|
||||||
|
border-bottom: 1px solid var(--color-border-light);
|
||||||
|
}
|
||||||
|
.master-list-items {
|
||||||
|
flex: 1;
|
||||||
|
min-height: 0;
|
||||||
|
overflow: auto;
|
||||||
|
padding: 8px;
|
||||||
|
}
|
||||||
|
.master-editor {
|
||||||
|
min-width: 0;
|
||||||
|
padding: 26px 30px;
|
||||||
|
background: #fff;
|
||||||
|
}
|
||||||
|
.master-empty {
|
||||||
|
height: 100%;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
.master-detail-heading :deep(.page-actions) {
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
@media (max-width: 800px) {
|
||||||
|
.master-detail-panel {
|
||||||
|
display: grid;
|
||||||
|
flex: none;
|
||||||
|
height: auto;
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
min-height: 0;
|
||||||
|
}
|
||||||
|
.master-list {
|
||||||
|
max-height: 280px;
|
||||||
|
border-right: 0;
|
||||||
|
border-bottom: 1px solid var(--color-border-light);
|
||||||
|
}
|
||||||
|
.master-editor {
|
||||||
|
padding: 20px 16px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
defineProps<{ title: string; description?: string }>();
|
||||||
|
</script>
|
||||||
|
<template>
|
||||||
|
<header class="page-header">
|
||||||
|
<div>
|
||||||
|
<h2>{{ title }}</h2>
|
||||||
|
</div>
|
||||||
|
<div class="page-actions"><slot /></div>
|
||||||
|
</header>
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,78 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, onBeforeUnmount, reactive, watch } from "vue";
|
||||||
|
import { Check, Close, CopyDocument, InfoFilled, WarningFilled } from "@element-plus/icons-vue";
|
||||||
|
import { useToastStore, type ToastItem } from "@/stores/toast";
|
||||||
|
|
||||||
|
const store = useToastStore();
|
||||||
|
const elapsed = reactive<Record<number, number>>({});
|
||||||
|
const paused = reactive<Record<number, boolean>>({});
|
||||||
|
let lastTick = performance.now();
|
||||||
|
const timer = window.setInterval(() => {
|
||||||
|
const now = performance.now();
|
||||||
|
const delta = now - lastTick;
|
||||||
|
lastTick = now;
|
||||||
|
for (const item of store.visible) {
|
||||||
|
if (!paused[item.id]) elapsed[item.id] = (elapsed[item.id] || 0) + delta;
|
||||||
|
if (elapsed[item.id] >= item.duration) store.remove(item.id);
|
||||||
|
}
|
||||||
|
}, 50);
|
||||||
|
|
||||||
|
watch(
|
||||||
|
() => store.visible.map((item) => item.id),
|
||||||
|
(ids) => {
|
||||||
|
ids.forEach((id) => {
|
||||||
|
if (elapsed[id] === undefined) elapsed[id] = 0;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
);
|
||||||
|
onBeforeUnmount(() => window.clearInterval(timer));
|
||||||
|
|
||||||
|
const icons = { success: Check, warning: WarningFilled, error: WarningFilled, info: InfoFilled };
|
||||||
|
const progress = (item: ToastItem) =>
|
||||||
|
computed(() => Math.max(0, 100 - ((elapsed[item.id] || 0) / item.duration) * 100)).value;
|
||||||
|
async function copy(item: ToastItem) {
|
||||||
|
await navigator.clipboard.writeText(item.copyText || item.message);
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="toast-region">
|
||||||
|
<TransitionGroup name="toast-fade">
|
||||||
|
<section
|
||||||
|
v-for="item in store.visible"
|
||||||
|
:key="item.id"
|
||||||
|
class="app-toast"
|
||||||
|
:class="`app-toast--${item.type}`"
|
||||||
|
@mouseenter="paused[item.id] = true"
|
||||||
|
@mouseleave="paused[item.id] = false"
|
||||||
|
>
|
||||||
|
<el-icon class="app-toast__icon"><component :is="icons[item.type]" /></el-icon>
|
||||||
|
<div class="app-toast__content">
|
||||||
|
<strong>{{ item.title }}</strong>
|
||||||
|
<span>{{ item.message }}</span>
|
||||||
|
</div>
|
||||||
|
<button
|
||||||
|
v-if="item.type === 'error'"
|
||||||
|
class="icon-button"
|
||||||
|
type="button"
|
||||||
|
title="复制完整错误"
|
||||||
|
@click="copy(item)"
|
||||||
|
>
|
||||||
|
<el-icon><CopyDocument /></el-icon>
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
class="icon-button"
|
||||||
|
type="button"
|
||||||
|
title="关闭提示"
|
||||||
|
@click="store.remove(item.id)"
|
||||||
|
>
|
||||||
|
<el-icon><Close /></el-icon>
|
||||||
|
</button>
|
||||||
|
<div
|
||||||
|
class="app-toast__progress"
|
||||||
|
:style="{ width: `${progress(item)}%` }"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
</TransitionGroup>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,148 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, onBeforeUnmount, onMounted, ref } from "vue";
|
||||||
|
import { ElMessageBox } from "element-plus";
|
||||||
|
import { RouterView, useRoute, useRouter } from "vue-router";
|
||||||
|
import AdminMenu from "@/components/AdminMenu.vue";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import logoUrl from "@/assets/logo";
|
||||||
|
import { useAuthStore } from "@/stores/auth";
|
||||||
|
|
||||||
|
const route = useRoute();
|
||||||
|
const router = useRouter();
|
||||||
|
const auth = useAuthStore();
|
||||||
|
const width = ref(window.innerWidth);
|
||||||
|
const drawerOpen = ref(false);
|
||||||
|
const loggingOut = ref(false);
|
||||||
|
const collapsed = computed(() => width.value < 1200);
|
||||||
|
const mobile = computed(() => width.value < 768);
|
||||||
|
const updateWidth = () => {
|
||||||
|
width.value = window.innerWidth;
|
||||||
|
if (!mobile.value) drawerOpen.value = false;
|
||||||
|
};
|
||||||
|
onMounted(() => window.addEventListener("resize", updateWidth, { passive: true }));
|
||||||
|
onBeforeUnmount(() => window.removeEventListener("resize", updateWidth));
|
||||||
|
|
||||||
|
/** 确认管理员主动退出后注销当前会话，桌面侧栏与移动抽屉共用该流程。 */
|
||||||
|
async function logout() {
|
||||||
|
if (loggingOut.value) return;
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm("退出后需要重新登录，确定退出管理后台吗？", "退出登录", {
|
||||||
|
type: "warning",
|
||||||
|
confirmButtonText: "退出登录",
|
||||||
|
cancelButtonText: "取消",
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
if (error === "cancel" || error === "close") return;
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
loggingOut.value = true;
|
||||||
|
try {
|
||||||
|
await auth.logout();
|
||||||
|
await router.replace("/login");
|
||||||
|
} finally {
|
||||||
|
loggingOut.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function navigate(path: string) {
|
||||||
|
router.push(path);
|
||||||
|
drawerOpen.value = false;
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div
|
||||||
|
class="admin-shell"
|
||||||
|
:class="{ 'is-collapsed': collapsed && !mobile }"
|
||||||
|
>
|
||||||
|
<aside
|
||||||
|
v-if="!mobile"
|
||||||
|
class="sidebar"
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
class="brand"
|
||||||
|
:title="collapsed ? '剧游AI' : undefined"
|
||||||
|
>
|
||||||
|
<span class="brand-mark"
|
||||||
|
><img
|
||||||
|
:src="logoUrl"
|
||||||
|
alt="" /></span
|
||||||
|
><strong v-if="!collapsed">剧游AI</strong>
|
||||||
|
</div>
|
||||||
|
<AdminMenu
|
||||||
|
:active-path="route.path"
|
||||||
|
:collapsed="collapsed"
|
||||||
|
@navigate="navigate"
|
||||||
|
/>
|
||||||
|
<button
|
||||||
|
class="menu-item logout"
|
||||||
|
type="button"
|
||||||
|
:title="collapsed ? '退出登录' : undefined"
|
||||||
|
:disabled="loggingOut"
|
||||||
|
@click="logout"
|
||||||
|
>
|
||||||
|
<AppIcon
|
||||||
|
name="SwitchButton"
|
||||||
|
:size="20"
|
||||||
|
/><span v-if="!collapsed">退出登录</span>
|
||||||
|
</button>
|
||||||
|
</aside>
|
||||||
|
|
||||||
|
<el-drawer
|
||||||
|
v-model="drawerOpen"
|
||||||
|
direction="ltr"
|
||||||
|
size="240px"
|
||||||
|
:with-header="false"
|
||||||
|
class="mobile-drawer"
|
||||||
|
>
|
||||||
|
<div class="brand">
|
||||||
|
<span class="brand-mark"
|
||||||
|
><img
|
||||||
|
:src="logoUrl"
|
||||||
|
alt="" /></span
|
||||||
|
><strong>剧游AI</strong>
|
||||||
|
</div>
|
||||||
|
<AdminMenu
|
||||||
|
:active-path="route.path"
|
||||||
|
@navigate="navigate"
|
||||||
|
/>
|
||||||
|
<button
|
||||||
|
class="menu-item logout"
|
||||||
|
type="button"
|
||||||
|
:disabled="loggingOut"
|
||||||
|
@click="logout"
|
||||||
|
>
|
||||||
|
<AppIcon
|
||||||
|
name="SwitchButton"
|
||||||
|
:size="20"
|
||||||
|
/><span>退出登录</span>
|
||||||
|
</button>
|
||||||
|
</el-drawer>
|
||||||
|
|
||||||
|
<div class="shell-main">
|
||||||
|
<main class="page-content">
|
||||||
|
<button
|
||||||
|
v-if="mobile"
|
||||||
|
class="icon-button mobile-menu-trigger"
|
||||||
|
type="button"
|
||||||
|
title="打开菜单"
|
||||||
|
@click="drawerOpen = true"
|
||||||
|
>
|
||||||
|
<AppIcon
|
||||||
|
name="Menu"
|
||||||
|
:size="22"
|
||||||
|
/>
|
||||||
|
</button>
|
||||||
|
<RouterView />
|
||||||
|
</main>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.mobile-menu-trigger {
|
||||||
|
margin-bottom: 16px;
|
||||||
|
background: #fff;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
box-shadow: var(--shadow-sm);
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
// 剧游AI管理后台入口，负责挂载 Vue 应用、注册路由与状态管理
|
||||||
|
|
||||||
|
import { createApp } from "vue";
|
||||||
|
import { createPinia } from "pinia";
|
||||||
|
import ElementPlus from "element-plus";
|
||||||
|
import zhCn from "element-plus/es/locale/lang/zh-cn";
|
||||||
|
|
||||||
|
import App from "./App.vue";
|
||||||
|
import router from "./router";
|
||||||
|
|
||||||
|
import "./assets/styles/main.scss";
|
||||||
|
import "element-plus/dist/index.css";
|
||||||
|
|
||||||
|
const app = createApp(App);
|
||||||
|
|
||||||
|
app.use(createPinia());
|
||||||
|
app.use(router);
|
||||||
|
app.use(ElementPlus, { locale: zhCn });
|
||||||
|
|
||||||
|
app.mount("#app");
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
// 管理后台主导航配置，统一维护菜单顺序和默认入口。
|
||||||
|
|
||||||
|
/** 管理后台左侧菜单项，顺序同时决定默认打开的页面。 */
|
||||||
|
export const adminMenuItems = [
|
||||||
|
{ path: "/users", label: "用户管理", icon: "User" },
|
||||||
|
{ path: "/redemption-codes", label: "兑换码管理", icon: "Key" },
|
||||||
|
{ path: "/styles", label: "风格管理", icon: "Picture" },
|
||||||
|
{ path: "/prompts", label: "提示词管理", icon: "Document" },
|
||||||
|
{ path: "/channels", label: "渠道管理", icon: "Connection" },
|
||||||
|
{ path: "/models", label: "模型管理", icon: "Cpu" },
|
||||||
|
{ path: "/audit-logs", label: "审计日志", icon: "Tickets" },
|
||||||
|
{ path: "/security", label: "账号安全管理", icon: "Lock" },
|
||||||
|
] as const;
|
||||||
|
|
||||||
|
/** 管理后台未指定目标页面时打开的第一个菜单路径。 */
|
||||||
|
export const DEFAULT_ADMIN_PATH = adminMenuItems[0].path;
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
// 管理后台路由配置，负责页面注册、默认入口和登录访问控制。
|
||||||
|
import { createRouter, createWebHistory } from "vue-router";
|
||||||
|
import { DEFAULT_ADMIN_PATH } from "@/router/adminNavigation";
|
||||||
|
|
||||||
|
const router = createRouter({
|
||||||
|
history: createWebHistory(import.meta.env.BASE_URL),
|
||||||
|
routes: [
|
||||||
|
{ path: "/login", name: "login", component: () => import("@/views/LoginView.vue"), meta: { public: true } },
|
||||||
|
{
|
||||||
|
path: "/",
|
||||||
|
component: () => import("@/layouts/AdminLayout.vue"),
|
||||||
|
children: [
|
||||||
|
{ path: "", redirect: DEFAULT_ADMIN_PATH },
|
||||||
|
{
|
||||||
|
path: "redemption-codes",
|
||||||
|
name: "redemption-codes",
|
||||||
|
component: () => import("@/views/RedemptionsView.vue"),
|
||||||
|
meta: { title: "兑换码管理" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "styles",
|
||||||
|
name: "styles",
|
||||||
|
component: () => import("@/views/StylesView.vue"),
|
||||||
|
meta: { title: "风格管理" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "prompts",
|
||||||
|
name: "prompts",
|
||||||
|
component: () => import("@/views/PromptsView.vue"),
|
||||||
|
meta: { title: "提示词管理" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "channels",
|
||||||
|
name: "channels",
|
||||||
|
component: () => import("@/views/ChannelsView.vue"),
|
||||||
|
meta: { title: "渠道管理" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "models",
|
||||||
|
name: "models",
|
||||||
|
component: () => import("@/views/ModelsView.vue"),
|
||||||
|
meta: { title: "模型管理" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "users",
|
||||||
|
name: "users",
|
||||||
|
component: () => import("@/views/UsersView.vue"),
|
||||||
|
meta: { title: "用户管理" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "audit-logs",
|
||||||
|
name: "audit-logs",
|
||||||
|
component: () => import("@/views/AuditLogsView.vue"),
|
||||||
|
meta: { title: "审计日志" },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: "security",
|
||||||
|
name: "security",
|
||||||
|
component: () => import("@/views/SecurityView.vue"),
|
||||||
|
meta: { title: "账号安全" },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
{ path: "/:pathMatch(.*)*", redirect: DEFAULT_ADMIN_PATH },
|
||||||
|
],
|
||||||
|
});
|
||||||
|
|
||||||
|
router.beforeEach((to) => {
|
||||||
|
const authenticated = Boolean(localStorage.getItem("admin_access_token"));
|
||||||
|
if (!to.meta.public && !authenticated) return { name: "login", query: { redirect: to.fullPath } };
|
||||||
|
if (to.name === "login" && authenticated) return { path: DEFAULT_ADMIN_PATH };
|
||||||
|
document.title = "剧游AI-后台管理";
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
|
||||||
|
export default router;
|
||||||
@@ -0,0 +1,31 @@
|
|||||||
|
import { defineStore } from "pinia";
|
||||||
|
import http, { clearAuth, saveTokens } from "@/api/http";
|
||||||
|
|
||||||
|
export const useAuthStore = defineStore("auth", {
|
||||||
|
state: () => ({
|
||||||
|
profile: JSON.parse(localStorage.getItem("admin_profile") || "null") as { id: string; username: string } | null,
|
||||||
|
loading: false,
|
||||||
|
}),
|
||||||
|
getters: { authenticated: () => Boolean(localStorage.getItem("admin_access_token")) },
|
||||||
|
actions: {
|
||||||
|
async login(username: string, password: string) {
|
||||||
|
this.loading = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.post("/auth/login", { username, password });
|
||||||
|
saveTokens(data.data);
|
||||||
|
this.profile = data.data.admin;
|
||||||
|
} finally {
|
||||||
|
this.loading = false;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
async logout() {
|
||||||
|
const refreshToken = localStorage.getItem("admin_refresh_token");
|
||||||
|
try {
|
||||||
|
await http.post("/auth/logout", { refresh_token: refreshToken });
|
||||||
|
} finally {
|
||||||
|
clearAuth();
|
||||||
|
this.profile = null;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
// 剧游AI管理后台计数器示例 Store，演示 Pinia 状态管理用法
|
||||||
|
|
||||||
|
import { ref, computed } from "vue";
|
||||||
|
import { defineStore } from "pinia";
|
||||||
|
|
||||||
|
export const useCounterStore = defineStore("counter", () => {
|
||||||
|
const count = ref(0);
|
||||||
|
const doubleCount = computed(() => count.value * 2);
|
||||||
|
|
||||||
|
function increment() {
|
||||||
|
count.value++;
|
||||||
|
}
|
||||||
|
|
||||||
|
return { count, doubleCount, increment };
|
||||||
|
});
|
||||||
@@ -0,0 +1,50 @@
|
|||||||
|
import { defineStore } from "pinia";
|
||||||
|
|
||||||
|
export type ToastType = "success" | "warning" | "error" | "info";
|
||||||
|
export interface ToastItem {
|
||||||
|
id: number;
|
||||||
|
type: ToastType;
|
||||||
|
title: string;
|
||||||
|
message: string;
|
||||||
|
copyText?: string;
|
||||||
|
duration: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
let nextId = 1;
|
||||||
|
|
||||||
|
export const useToastStore = defineStore("toast", {
|
||||||
|
state: () => ({ visible: [] as ToastItem[], queued: [] as ToastItem[] }),
|
||||||
|
actions: {
|
||||||
|
show(type: ToastType, message: string, options?: { title?: string; copyText?: string }) {
|
||||||
|
const item: ToastItem = {
|
||||||
|
id: nextId++,
|
||||||
|
type,
|
||||||
|
title: options?.title || typeTitle(type),
|
||||||
|
message,
|
||||||
|
copyText: options?.copyText,
|
||||||
|
duration: 5000,
|
||||||
|
};
|
||||||
|
if (this.visible.length < 3) this.visible.push(item);
|
||||||
|
else this.queued.push(item);
|
||||||
|
return item.id;
|
||||||
|
},
|
||||||
|
success(message: string) {
|
||||||
|
return this.show("success", message);
|
||||||
|
},
|
||||||
|
warning(message: string) {
|
||||||
|
return this.show("warning", message);
|
||||||
|
},
|
||||||
|
error(message: string, copyText?: string) {
|
||||||
|
return this.show("error", message, { copyText });
|
||||||
|
},
|
||||||
|
remove(id: number) {
|
||||||
|
this.visible = this.visible.filter((item) => item.id !== id);
|
||||||
|
const next = this.queued.shift();
|
||||||
|
if (next) this.visible.push(next);
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
function typeTitle(type: ToastType) {
|
||||||
|
return { success: "操作成功", warning: "请注意", error: "操作失败", info: "提示" }[type];
|
||||||
|
}
|
||||||
@@ -0,0 +1,243 @@
|
|||||||
|
<!-- 管理后台审计日志页，负责只读查询和展示关键管理操作。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { onMounted, reactive, ref } from "vue";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
const toast = useToastStore();
|
||||||
|
const loading = ref(false);
|
||||||
|
const items = ref<Record<string, any>[]>([]);
|
||||||
|
const total = ref(0);
|
||||||
|
const query = reactive({ keyword: "", page: 1, page_size: 15 });
|
||||||
|
|
||||||
|
const actionLabels: Record<string, string> = {
|
||||||
|
create: "新建",
|
||||||
|
batch_create: "批量新建",
|
||||||
|
update: "修改",
|
||||||
|
batch_update: "批量修改",
|
||||||
|
toggle: "状态变更",
|
||||||
|
delete: "删除",
|
||||||
|
batch_delete: "批量删除",
|
||||||
|
reorder: "调整顺序",
|
||||||
|
generate: "生成",
|
||||||
|
save: "保存",
|
||||||
|
save_draft: "保存草稿",
|
||||||
|
grant_points: "增加积分",
|
||||||
|
password_changed: "修改密码",
|
||||||
|
password_change_failed: "修改密码失败",
|
||||||
|
};
|
||||||
|
const resourceLabels: Record<string, string> = {
|
||||||
|
users: "用户",
|
||||||
|
"redemption-codes": "兑换码",
|
||||||
|
styles: "风格",
|
||||||
|
prompts: "提示词",
|
||||||
|
channels: "渠道",
|
||||||
|
models: "模型",
|
||||||
|
"admin-auth": "管理员账号",
|
||||||
|
};
|
||||||
|
|
||||||
|
/** 按当前筛选条件加载审计日志。 */
|
||||||
|
async function load() {
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.get("/audit-logs", { params: query });
|
||||||
|
items.value = data.data.items;
|
||||||
|
total.value = data.data.total;
|
||||||
|
} catch (error) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 清空关键字并返回第一页。 */
|
||||||
|
function resetQuery() {
|
||||||
|
query.keyword = "";
|
||||||
|
query.page = 1;
|
||||||
|
void load();
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 将审计动作代码转换为中文业务名称。 */
|
||||||
|
function actionText(value: string) {
|
||||||
|
return actionLabels[value] || value || "未知操作";
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 将审计资源代码转换为中文业务名称。 */
|
||||||
|
function resourceText(value: string) {
|
||||||
|
return resourceLabels[value] || value || "未知资源";
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 使用中国时区格式化审计发生时间。 */
|
||||||
|
function formatTime(value: unknown) {
|
||||||
|
if (!value) return "--";
|
||||||
|
return new Date(String(value)).toLocaleString("zh-CN", { timeZone: "Asia/Shanghai", hour12: false });
|
||||||
|
}
|
||||||
|
|
||||||
|
onMounted(load);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<PageHeader title="审计日志" />
|
||||||
|
<section class="panel filter-bar">
|
||||||
|
<el-input
|
||||||
|
v-model.trim="query.keyword"
|
||||||
|
clearable
|
||||||
|
placeholder="管理员、操作、资源、原因、对象 ID 或追踪 ID"
|
||||||
|
@keyup.enter="
|
||||||
|
query.page = 1;
|
||||||
|
load();
|
||||||
|
"
|
||||||
|
/>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
@click="
|
||||||
|
query.page = 1;
|
||||||
|
load();
|
||||||
|
"
|
||||||
|
>查询</el-button
|
||||||
|
>
|
||||||
|
<el-button @click="resetQuery">重置</el-button>
|
||||||
|
</section>
|
||||||
|
<section
|
||||||
|
class="panel table-panel"
|
||||||
|
v-loading="loading"
|
||||||
|
>
|
||||||
|
<el-table
|
||||||
|
class="desktop-table"
|
||||||
|
:data="items"
|
||||||
|
>
|
||||||
|
<el-table-column
|
||||||
|
prop="admin_username"
|
||||||
|
label="管理员"
|
||||||
|
min-width="120"
|
||||||
|
show-overflow-tooltip
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
label="操作"
|
||||||
|
min-width="120"
|
||||||
|
>
|
||||||
|
<template #default="{ row }">
|
||||||
|
<el-tag effect="plain">{{ actionText(row.action) }}</el-tag>
|
||||||
|
</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
label="资源"
|
||||||
|
min-width="120"
|
||||||
|
>
|
||||||
|
<template #default="{ row }">{{ resourceText(row.resource_type) }}</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
prop="resource_id"
|
||||||
|
min-width="180"
|
||||||
|
show-overflow-tooltip
|
||||||
|
>
|
||||||
|
<template #header>
|
||||||
|
<span class="column-label">
|
||||||
|
对象 ID
|
||||||
|
<el-tooltip
|
||||||
|
content="被操作数据的唯一标识；即使对象改名或删除，仍可用它准确定位记录。"
|
||||||
|
placement="top"
|
||||||
|
>
|
||||||
|
<span class="column-help"><AppIcon name="QuestionFilled" /></span>
|
||||||
|
</el-tooltip>
|
||||||
|
</span>
|
||||||
|
</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
prop="reason"
|
||||||
|
label="操作原因"
|
||||||
|
min-width="180"
|
||||||
|
show-overflow-tooltip
|
||||||
|
>
|
||||||
|
<template #default="{ row }">{{ row.reason || "--" }}</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
prop="trace_id"
|
||||||
|
min-width="180"
|
||||||
|
show-overflow-tooltip
|
||||||
|
>
|
||||||
|
<template #header>
|
||||||
|
<span class="column-label">
|
||||||
|
请求追踪 ID
|
||||||
|
<el-tooltip
|
||||||
|
content="本次请求的唯一追踪编号，用于在服务器日志中查找并关联同一次请求。"
|
||||||
|
placement="top"
|
||||||
|
>
|
||||||
|
<span class="column-help"><AppIcon name="QuestionFilled" /></span>
|
||||||
|
</el-tooltip>
|
||||||
|
</span>
|
||||||
|
</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
label="操作时间"
|
||||||
|
min-width="180"
|
||||||
|
>
|
||||||
|
<template #default="{ row }">{{ formatTime(row.created_at) }}</template>
|
||||||
|
</el-table-column>
|
||||||
|
</el-table>
|
||||||
|
|
||||||
|
<div
|
||||||
|
v-if="items.length"
|
||||||
|
class="mobile-card-list"
|
||||||
|
>
|
||||||
|
<article
|
||||||
|
v-for="row in items"
|
||||||
|
:key="row.id"
|
||||||
|
class="data-card"
|
||||||
|
>
|
||||||
|
<dl>
|
||||||
|
<dt>管理员</dt>
|
||||||
|
<dd>{{ row.admin_username }}</dd>
|
||||||
|
<dt>操作</dt>
|
||||||
|
<dd>{{ actionText(row.action) }}</dd>
|
||||||
|
<dt>资源</dt>
|
||||||
|
<dd>{{ resourceText(row.resource_type) }}</dd>
|
||||||
|
<dt>对象 ID</dt>
|
||||||
|
<dd>{{ row.resource_id || "--" }}</dd>
|
||||||
|
<dt>操作原因</dt>
|
||||||
|
<dd>{{ row.reason || "--" }}</dd>
|
||||||
|
<dt>请求追踪 ID</dt>
|
||||||
|
<dd>{{ row.trace_id || "--" }}</dd>
|
||||||
|
<dt>操作时间</dt>
|
||||||
|
<dd>{{ formatTime(row.created_at) }}</dd>
|
||||||
|
</dl>
|
||||||
|
</article>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<el-empty
|
||||||
|
v-if="!loading && !items.length"
|
||||||
|
description="暂无审计日志"
|
||||||
|
/>
|
||||||
|
<el-pagination
|
||||||
|
v-if="total > query.page_size"
|
||||||
|
v-model:current-page="query.page"
|
||||||
|
background
|
||||||
|
layout="total,prev,pager,next"
|
||||||
|
:total="total"
|
||||||
|
:page-size="query.page_size"
|
||||||
|
@current-change="load"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.column-label,
|
||||||
|
.column-help {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.column-label {
|
||||||
|
gap: var(--space-xs);
|
||||||
|
}
|
||||||
|
|
||||||
|
.column-help {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
cursor: help;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,345 @@
|
|||||||
|
<!-- 渠道管理页，负责渠道配置、余额状态和启停操作。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { onMounted, reactive, ref } from "vue";
|
||||||
|
import { ElMessageBox, type FormInstance } from "element-plus";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import ChannelFormFields from "@/components/ChannelFormFields.vue";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
const toast = useToastStore();
|
||||||
|
const loading = ref(false);
|
||||||
|
const saving = ref(false);
|
||||||
|
const dialogOpen = ref(false);
|
||||||
|
const editingId = ref("");
|
||||||
|
const formRef = ref<FormInstance>();
|
||||||
|
const items = ref<Record<string, any>[]>([]);
|
||||||
|
const total = ref(0);
|
||||||
|
const query = reactive({ keyword: "", enabled: "", page: 1, page_size: 15 });
|
||||||
|
const form = reactive<Record<string, any>>({});
|
||||||
|
const rules = {
|
||||||
|
name: [{ required: true, message: "请填写渠道名", trigger: "blur" }],
|
||||||
|
base_url: [{ required: true, message: "请填写渠道 URL", trigger: "blur" }],
|
||||||
|
api_key: [
|
||||||
|
{
|
||||||
|
validator: (_rule: unknown, value: string, callback: (error?: Error) => void) => {
|
||||||
|
if (!editingId.value && !String(value || "").trim()) callback(new Error("请填写 API Key"));
|
||||||
|
else callback();
|
||||||
|
},
|
||||||
|
trigger: "blur",
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
|
||||||
|
async function load() {
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.get("/resources/channels", { params: query });
|
||||||
|
items.value = data.data.items || [];
|
||||||
|
total.value = Number(data.data.total || 0);
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function reset(row?: Record<string, any>) {
|
||||||
|
Object.keys(form).forEach((key) => delete form[key]);
|
||||||
|
Object.assign(form, {
|
||||||
|
name: row?.name || "",
|
||||||
|
channel_type: row?.channel_type || "relay",
|
||||||
|
base_url: row?.base_url || "",
|
||||||
|
api_key: "",
|
||||||
|
api_key_masked: row?.api_key_masked || "",
|
||||||
|
warning_threshold: Number(row?.warning_threshold ?? 500),
|
||||||
|
exchange_cny_amount: 1,
|
||||||
|
exchange_point_amount: Number(row?.channel_points_per_cny ?? 1),
|
||||||
|
max_concurrency: Number(row?.max_concurrency ?? 500),
|
||||||
|
max_user_concurrency: Number(row?.max_user_concurrency ?? 10),
|
||||||
|
enabled: row?.enabled ?? true,
|
||||||
|
});
|
||||||
|
editingId.value = row?.id || "";
|
||||||
|
}
|
||||||
|
function openCreate() {
|
||||||
|
reset();
|
||||||
|
dialogOpen.value = true;
|
||||||
|
}
|
||||||
|
function openEdit(row: Record<string, any>) {
|
||||||
|
reset(row);
|
||||||
|
dialogOpen.value = true;
|
||||||
|
}
|
||||||
|
async function save() {
|
||||||
|
if (!(await formRef.value?.validate().catch(() => false))) return;
|
||||||
|
const cnyAmount = Number(form.exchange_cny_amount);
|
||||||
|
const pointAmount = Number(form.exchange_point_amount);
|
||||||
|
if (!Number.isFinite(cnyAmount) || cnyAmount <= 0 || !Number.isFinite(pointAmount) || pointAmount <= 0) {
|
||||||
|
toast.warning("请填写大于 0 的人民币金额和渠道积分");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (editingId.value && String(form.api_key || "").trim()) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm("此次修改将覆盖已加密保存的 API Key，是否继续？", "确认覆盖敏感配置", {
|
||||||
|
type: "warning",
|
||||||
|
confirmButtonText: "继续覆盖",
|
||||||
|
cancelButtonText: "取消",
|
||||||
|
});
|
||||||
|
} catch {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
saving.value = true;
|
||||||
|
try {
|
||||||
|
const payload = {
|
||||||
|
...form,
|
||||||
|
channel_points_per_cny: Number((pointAmount / cnyAmount).toFixed(6)),
|
||||||
|
};
|
||||||
|
delete payload.exchange_cny_amount;
|
||||||
|
delete payload.exchange_point_amount;
|
||||||
|
delete payload.api_key_masked;
|
||||||
|
if (!payload.api_key) delete payload.api_key;
|
||||||
|
if (editingId.value) await http.put(`/resources/channels/${editingId.value}`, payload);
|
||||||
|
else await http.post("/resources/channels", payload);
|
||||||
|
dialogOpen.value = false;
|
||||||
|
toast.success(editingId.value ? "渠道已更新" : "渠道已创建");
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
saving.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function toggle(row: Record<string, any>, enabled: boolean) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm(`${enabled ? "启用" : "停用"}后将立即生效，是否继续？`, "确认状态变更", {
|
||||||
|
type: "warning",
|
||||||
|
confirmButtonText: "确认",
|
||||||
|
cancelButtonText: "取消",
|
||||||
|
});
|
||||||
|
await http.patch(`/resources/channels/${row.id}/enabled`, { enabled });
|
||||||
|
row.enabled = enabled;
|
||||||
|
toast.success("渠道状态已更新");
|
||||||
|
} catch (error) {
|
||||||
|
row.enabled = !enabled;
|
||||||
|
if (error !== "cancel" && error !== "close") showError(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function remove(row: Record<string, any>) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm(`确定删除渠道“${row.name}”吗？`, "删除渠道", {
|
||||||
|
type: "warning",
|
||||||
|
confirmButtonText: "删除",
|
||||||
|
cancelButtonText: "取消",
|
||||||
|
});
|
||||||
|
await http.delete(`/resources/channels/${row.id}`);
|
||||||
|
toast.success("渠道已删除");
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
if (error !== "cancel" && error !== "close") showError(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function search() {
|
||||||
|
query.page = 1;
|
||||||
|
void load();
|
||||||
|
}
|
||||||
|
function resetQuery() {
|
||||||
|
query.keyword = "";
|
||||||
|
query.enabled = "";
|
||||||
|
search();
|
||||||
|
}
|
||||||
|
function balance(row: Record<string, any>) {
|
||||||
|
return Number(row.balance) < 0 ? "无限" : `${Number(row.balance || 0).toFixed(2)} ${row.currency || ""}`;
|
||||||
|
}
|
||||||
|
function showError(error: unknown) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
}
|
||||||
|
|
||||||
|
onMounted(load);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<PageHeader
|
||||||
|
title="渠道管理"
|
||||||
|
description="维护 API 地址、密钥、汇率、余额和并发配置"
|
||||||
|
>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
@click="openCreate"
|
||||||
|
><AppIcon name="Plus" />新建渠道</el-button
|
||||||
|
>
|
||||||
|
</PageHeader>
|
||||||
|
<section class="panel filter-bar">
|
||||||
|
<el-input
|
||||||
|
v-model="query.keyword"
|
||||||
|
clearable
|
||||||
|
placeholder="输入渠道名或地址"
|
||||||
|
><template #prefix><AppIcon name="Search" /></template
|
||||||
|
></el-input>
|
||||||
|
<el-select
|
||||||
|
v-model="query.enabled"
|
||||||
|
clearable
|
||||||
|
placeholder="启用状态"
|
||||||
|
><el-option
|
||||||
|
label="已启用"
|
||||||
|
value="true" /><el-option
|
||||||
|
label="已停用"
|
||||||
|
value="false"
|
||||||
|
/></el-select>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
@click="search"
|
||||||
|
>查询</el-button
|
||||||
|
>
|
||||||
|
<el-button @click="resetQuery">重置</el-button>
|
||||||
|
</section>
|
||||||
|
<section
|
||||||
|
v-loading="loading"
|
||||||
|
class="panel table-panel"
|
||||||
|
>
|
||||||
|
<el-table
|
||||||
|
v-if="items.length"
|
||||||
|
:data="items"
|
||||||
|
class="desktop-table"
|
||||||
|
>
|
||||||
|
<el-table-column
|
||||||
|
prop="name"
|
||||||
|
label="渠道名"
|
||||||
|
min-width="130"
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
prop="base_url"
|
||||||
|
label="渠道 URL"
|
||||||
|
min-width="240"
|
||||||
|
show-overflow-tooltip
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
prop="api_key_masked"
|
||||||
|
label="API Key"
|
||||||
|
min-width="190"
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
label="渠道汇率"
|
||||||
|
min-width="180"
|
||||||
|
><template #default="{ row }"
|
||||||
|
>1.00 元 = {{ Number(row.channel_points_per_cny || 0).toFixed(2) }} 积分</template
|
||||||
|
></el-table-column
|
||||||
|
>
|
||||||
|
<el-table-column
|
||||||
|
prop="max_concurrency"
|
||||||
|
label="最大并发"
|
||||||
|
min-width="100"
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
prop="max_user_concurrency"
|
||||||
|
label="单用户并发"
|
||||||
|
min-width="120"
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
label="当前余额"
|
||||||
|
min-width="130"
|
||||||
|
><template #default="{ row }">{{ balance(row) }}</template></el-table-column
|
||||||
|
>
|
||||||
|
<el-table-column
|
||||||
|
label="启用"
|
||||||
|
min-width="90"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-switch
|
||||||
|
:model-value="row.enabled"
|
||||||
|
@change="toggle(row, Boolean($event))" /></template
|
||||||
|
></el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
label="操作"
|
||||||
|
width="140"
|
||||||
|
fixed="right"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-button
|
||||||
|
link
|
||||||
|
type="primary"
|
||||||
|
@click="openEdit(row)"
|
||||||
|
>编辑</el-button
|
||||||
|
><el-button
|
||||||
|
link
|
||||||
|
type="danger"
|
||||||
|
@click="remove(row)"
|
||||||
|
>删除</el-button
|
||||||
|
></template
|
||||||
|
></el-table-column
|
||||||
|
>
|
||||||
|
</el-table>
|
||||||
|
<div
|
||||||
|
v-if="items.length"
|
||||||
|
class="mobile-card-list"
|
||||||
|
>
|
||||||
|
<article
|
||||||
|
v-for="row in items"
|
||||||
|
:key="row.id"
|
||||||
|
class="data-card"
|
||||||
|
>
|
||||||
|
<dl>
|
||||||
|
<dt>渠道名</dt>
|
||||||
|
<dd>{{ row.name }}</dd>
|
||||||
|
<dt>地址</dt>
|
||||||
|
<dd>{{ row.base_url }}</dd>
|
||||||
|
<dt>渠道汇率</dt>
|
||||||
|
<dd>{{ Number(row.channel_points_per_cny || 0).toFixed(2) }}</dd>
|
||||||
|
<dt>当前余额</dt>
|
||||||
|
<dd>{{ balance(row) }}</dd>
|
||||||
|
<dt>最大并发</dt>
|
||||||
|
<dd>{{ row.max_concurrency }} / 用户 {{ row.max_user_concurrency }}</dd>
|
||||||
|
</dl>
|
||||||
|
<div class="card-actions">
|
||||||
|
<el-button @click="openEdit(row)">编辑</el-button
|
||||||
|
><el-button
|
||||||
|
type="danger"
|
||||||
|
plain
|
||||||
|
@click="remove(row)"
|
||||||
|
>删除</el-button
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</article>
|
||||||
|
</div>
|
||||||
|
<el-empty
|
||||||
|
v-if="!items.length"
|
||||||
|
description="暂无渠道"
|
||||||
|
/>
|
||||||
|
<el-pagination
|
||||||
|
v-if="total > query.page_size"
|
||||||
|
v-model:current-page="query.page"
|
||||||
|
:page-size="query.page_size"
|
||||||
|
:total="total"
|
||||||
|
layout="total, prev, pager, next"
|
||||||
|
@current-change="load"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<el-dialog
|
||||||
|
v-model="dialogOpen"
|
||||||
|
:title="editingId ? '编辑渠道' : '新建渠道'"
|
||||||
|
width="min(820px, calc(100vw - 32px))"
|
||||||
|
class="managed-scroll-dialog"
|
||||||
|
align-center
|
||||||
|
destroy-on-close
|
||||||
|
>
|
||||||
|
<el-form
|
||||||
|
ref="formRef"
|
||||||
|
:model="form"
|
||||||
|
:rules="rules"
|
||||||
|
label-position="top"
|
||||||
|
><ChannelFormFields
|
||||||
|
:form="form"
|
||||||
|
:editing="Boolean(editingId)"
|
||||||
|
/></el-form>
|
||||||
|
<template #footer
|
||||||
|
><el-button @click="dialogOpen = false">取消</el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="saving"
|
||||||
|
@click="save"
|
||||||
|
>保存</el-button
|
||||||
|
></template
|
||||||
|
>
|
||||||
|
</el-dialog>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,85 @@
|
|||||||
|
<!-- 管理后台登录页，负责管理员凭据校验和登录后页面跳转。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { reactive, ref } from "vue";
|
||||||
|
import { useRoute, useRouter } from "vue-router";
|
||||||
|
import type { FormInstance, FormRules } from "element-plus";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import logoUrl from "@/assets/logo";
|
||||||
|
import { apiError } from "@/api/http";
|
||||||
|
import { useAuthStore } from "@/stores/auth";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
import { DEFAULT_ADMIN_PATH } from "@/router/adminNavigation";
|
||||||
|
|
||||||
|
const auth = useAuthStore();
|
||||||
|
const toast = useToastStore();
|
||||||
|
const router = useRouter();
|
||||||
|
const route = useRoute();
|
||||||
|
const form = reactive({ username: "", password: "" });
|
||||||
|
const formRef = ref<FormInstance>();
|
||||||
|
const rules: FormRules = {
|
||||||
|
username: [{ required: true, message: "请输入管理员账号", trigger: "blur" }],
|
||||||
|
password: [{ required: true, message: "请输入密码", trigger: "blur" }],
|
||||||
|
};
|
||||||
|
async function submit() {
|
||||||
|
if (auth.loading) return;
|
||||||
|
if (!(await formRef.value?.validate().catch(() => false))) return;
|
||||||
|
try {
|
||||||
|
await auth.login(form.username, form.password);
|
||||||
|
router.replace(String(route.query.redirect || DEFAULT_ADMIN_PATH));
|
||||||
|
} catch (error) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<main class="login-page">
|
||||||
|
<section class="login-panel">
|
||||||
|
<div class="login-brand">
|
||||||
|
<span class="brand-mark"
|
||||||
|
><img
|
||||||
|
:src="logoUrl"
|
||||||
|
alt="" /></span
|
||||||
|
><span>剧游AI</span>
|
||||||
|
</div>
|
||||||
|
<div class="login-heading"><h1 id="login-title">管理后台登录</h1></div>
|
||||||
|
<el-form
|
||||||
|
ref="formRef"
|
||||||
|
:model="form"
|
||||||
|
:rules="rules"
|
||||||
|
label-position="top"
|
||||||
|
size="large"
|
||||||
|
@submit.prevent="submit"
|
||||||
|
>
|
||||||
|
<el-form-item
|
||||||
|
label="管理员账号"
|
||||||
|
prop="username"
|
||||||
|
><el-input
|
||||||
|
v-model.trim="form.username"
|
||||||
|
autocomplete="username"
|
||||||
|
placeholder="请输入管理员账号"
|
||||||
|
><template #prefix><AppIcon name="User" /></template></el-input
|
||||||
|
></el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="密码"
|
||||||
|
prop="password"
|
||||||
|
><el-input
|
||||||
|
v-model="form.password"
|
||||||
|
type="password"
|
||||||
|
autocomplete="current-password"
|
||||||
|
show-password
|
||||||
|
placeholder="请输入密码"
|
||||||
|
><template #prefix><AppIcon name="Lock" /></template></el-input
|
||||||
|
></el-form-item>
|
||||||
|
<el-button
|
||||||
|
class="login-submit"
|
||||||
|
type="primary"
|
||||||
|
native-type="submit"
|
||||||
|
:loading="auth.loading"
|
||||||
|
>登录</el-button
|
||||||
|
>
|
||||||
|
</el-form>
|
||||||
|
</section>
|
||||||
|
</main>
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,548 @@
|
|||||||
|
<!-- 模型管理页，负责模型能力、渠道归属和计费配置。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, onMounted, reactive, ref } from "vue";
|
||||||
|
import { ElMessageBox } from "element-plus";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import AdminNumberInput from "@/components/AdminNumberInput.vue";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
type PriceField = [key: string, unit: string, label: string];
|
||||||
|
|
||||||
|
const toast = useToastStore();
|
||||||
|
const loading = ref(false);
|
||||||
|
const dialog = ref(false);
|
||||||
|
const items = ref<Record<string, any>[]>([]);
|
||||||
|
const channels = ref<Record<string, any>[]>([]);
|
||||||
|
const editing = ref("");
|
||||||
|
const query = reactive({ keyword: "", model_type: "", channel_id: "" });
|
||||||
|
const form = reactive<Record<string, any>>({});
|
||||||
|
|
||||||
|
const selectedChannel = computed(() => channels.value.find((channel) => channel.id === form.channel_id));
|
||||||
|
|
||||||
|
const priceFields = computed<PriceField[]>(() => {
|
||||||
|
if (form.model_type === "text") {
|
||||||
|
if (form.text_billing_mode === "per_token") {
|
||||||
|
return [
|
||||||
|
["input", "M Token", "输入 Token"],
|
||||||
|
["output", "M Token", "输出 Token"],
|
||||||
|
];
|
||||||
|
}
|
||||||
|
return [["default", "次", "每次价格"]];
|
||||||
|
}
|
||||||
|
if (form.model_type === "image")
|
||||||
|
return [
|
||||||
|
["1K", "次", "1K"],
|
||||||
|
["2K", "次", "2K"],
|
||||||
|
["4K", "次", "4K"],
|
||||||
|
];
|
||||||
|
return [
|
||||||
|
["480p", "秒", "480p"],
|
||||||
|
["720p", "秒", "720p"],
|
||||||
|
["1080p", "秒", "1080p"],
|
||||||
|
];
|
||||||
|
});
|
||||||
|
|
||||||
|
function parsePrices(value: unknown): Record<string, any>[] {
|
||||||
|
if (Array.isArray(value)) return value;
|
||||||
|
if (typeof value !== "string") return [];
|
||||||
|
try {
|
||||||
|
const parsed = JSON.parse(value);
|
||||||
|
return Array.isArray(parsed) ? parsed : [];
|
||||||
|
} catch {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function hasTokenPrices(value: unknown) {
|
||||||
|
return parsePrices(value).some((price) => ["input", "output"].includes(String(price.price_key).toLowerCase()));
|
||||||
|
}
|
||||||
|
|
||||||
|
function isTokenBillingRow(row: Record<string, any>) {
|
||||||
|
return row.model_type === "text" && (row.text_billing_mode === "per_token" || hasTokenPrices(row.prices));
|
||||||
|
}
|
||||||
|
|
||||||
|
function displayPrices(row: Record<string, any>) {
|
||||||
|
const prices = parsePrices(row.prices);
|
||||||
|
if (row.model_type !== "text") return prices;
|
||||||
|
const keys = isTokenBillingRow(row) ? ["input", "output"] : ["default"];
|
||||||
|
return prices.filter((price) => keys.includes(String(price.price_key).toLowerCase()));
|
||||||
|
}
|
||||||
|
|
||||||
|
function reset(row?: Record<string, any>) {
|
||||||
|
const rowPrices = parsePrices(row?.prices);
|
||||||
|
const prices: Record<string, number> = {};
|
||||||
|
for (const price of rowPrices) prices[price.price_key] = Number(price.price);
|
||||||
|
const billingMode = hasTokenPrices(rowPrices) ? "per_token" : row?.text_billing_mode || "per_request";
|
||||||
|
Object.assign(form, {
|
||||||
|
channel_id: row?.channel_id || "",
|
||||||
|
name: row?.name || "",
|
||||||
|
model_type: row?.model_type || "text",
|
||||||
|
multimodal: row?.multimodal || false,
|
||||||
|
text_billing_mode: billingMode,
|
||||||
|
enabled: row?.enabled ?? true,
|
||||||
|
price_values: prices,
|
||||||
|
});
|
||||||
|
editing.value = row?.id || "";
|
||||||
|
dialog.value = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function fetchAllModels() {
|
||||||
|
const result: Record<string, any>[] = [];
|
||||||
|
let page = 1;
|
||||||
|
while (true) {
|
||||||
|
const { data } = await http.get("/models", { params: { ...query, page, page_size: 100 } });
|
||||||
|
const current = data.data.items || [];
|
||||||
|
result.push(...current);
|
||||||
|
if (!current.length || result.length >= Number(data.data.total || 0)) return result;
|
||||||
|
page += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function load() {
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const [models, { data: channelData }] = await Promise.all([
|
||||||
|
fetchAllModels(),
|
||||||
|
http.get("/resources/channels", { params: { page_size: 100 } }),
|
||||||
|
]);
|
||||||
|
items.value = models.map((row: Record<string, any>) => ({ ...row, prices: parsePrices(row.prices) }));
|
||||||
|
channels.value = channelData.data.items;
|
||||||
|
} catch (error) {
|
||||||
|
show(error);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function save() {
|
||||||
|
if (!form.channel_id || !form.name) {
|
||||||
|
toast.warning("请填写模型名并选择渠道");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const prices = priceFields.value.map(([priceKey, unit]) => ({
|
||||||
|
price_key: priceKey,
|
||||||
|
unit,
|
||||||
|
price: Number(Number(form.price_values[priceKey] || 0).toFixed(2)),
|
||||||
|
}));
|
||||||
|
try {
|
||||||
|
const payload = { ...form, prices };
|
||||||
|
delete payload.price_values;
|
||||||
|
if (editing.value) await http.put(`/models/${editing.value}`, payload);
|
||||||
|
else await http.post("/models", payload);
|
||||||
|
toast.success(editing.value ? "模型已更新" : "模型已创建");
|
||||||
|
dialog.value = false;
|
||||||
|
load();
|
||||||
|
} catch (error) {
|
||||||
|
show(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function toggle(row: Record<string, any>, enabled: boolean) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm(`确认${enabled ? "启用" : "停用"}模型 ${row.name}？`, "状态变更", { type: "warning" });
|
||||||
|
await http.patch(`/resources/models/${row.id}/enabled`, { enabled });
|
||||||
|
row.enabled = enabled;
|
||||||
|
toast.success("模型状态已更新");
|
||||||
|
} catch (error) {
|
||||||
|
row.enabled = !enabled;
|
||||||
|
if (error !== "cancel" && error !== "close") show(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function remove(row: Record<string, any>) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm("模型将被软删除；已有使用记录不会删除。", "确认删除", { type: "warning" });
|
||||||
|
await http.delete(`/resources/models/${row.id}`);
|
||||||
|
toast.success("模型已删除");
|
||||||
|
load();
|
||||||
|
} catch (error) {
|
||||||
|
if (error !== "cancel" && error !== "close") show(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function show(error: unknown) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
}
|
||||||
|
|
||||||
|
const typeText = (value: string) => ({ text: "文本", image: "图片", video: "视频" })[value] || value;
|
||||||
|
const priceLabel = (value: string) => ({ default: "基础", input: "输入", output: "输出" })[value] || value;
|
||||||
|
onMounted(load);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<PageHeader
|
||||||
|
title="模型管理"
|
||||||
|
description="模型实际可用状态由模型和所属渠道的启用状态共同决定"
|
||||||
|
>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
@click="reset()"
|
||||||
|
><AppIcon name="Plus" />新建模型</el-button
|
||||||
|
>
|
||||||
|
</PageHeader>
|
||||||
|
<section class="panel filter-bar">
|
||||||
|
<el-input
|
||||||
|
v-model="query.keyword"
|
||||||
|
clearable
|
||||||
|
placeholder="模型名"
|
||||||
|
/>
|
||||||
|
<el-select
|
||||||
|
v-model="query.model_type"
|
||||||
|
clearable
|
||||||
|
placeholder="模型类型"
|
||||||
|
>
|
||||||
|
<el-option
|
||||||
|
label="文本"
|
||||||
|
value="text"
|
||||||
|
/><el-option
|
||||||
|
label="图片"
|
||||||
|
value="image"
|
||||||
|
/><el-option
|
||||||
|
label="视频"
|
||||||
|
value="video"
|
||||||
|
/>
|
||||||
|
</el-select>
|
||||||
|
<el-select
|
||||||
|
v-model="query.channel_id"
|
||||||
|
clearable
|
||||||
|
placeholder="所属渠道"
|
||||||
|
>
|
||||||
|
<el-option
|
||||||
|
v-for="channel in channels"
|
||||||
|
:key="channel.id"
|
||||||
|
:label="channel.name"
|
||||||
|
:value="channel.id"
|
||||||
|
/>
|
||||||
|
</el-select>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
@click="load"
|
||||||
|
>查询</el-button
|
||||||
|
>
|
||||||
|
</section>
|
||||||
|
<section
|
||||||
|
class="panel table-panel"
|
||||||
|
v-loading="loading"
|
||||||
|
>
|
||||||
|
<el-table :data="items">
|
||||||
|
<el-table-column
|
||||||
|
prop="name"
|
||||||
|
label="模型名"
|
||||||
|
min-width="150"
|
||||||
|
/>
|
||||||
|
<el-table-column
|
||||||
|
label="类型"
|
||||||
|
width="90"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-tag
|
||||||
|
size="small"
|
||||||
|
effect="plain"
|
||||||
|
>{{ typeText(row.model_type) }}</el-tag
|
||||||
|
></template
|
||||||
|
></el-table-column
|
||||||
|
>
|
||||||
|
<el-table-column
|
||||||
|
label="所属渠道"
|
||||||
|
min-width="130"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-tag
|
||||||
|
size="small"
|
||||||
|
type="info"
|
||||||
|
effect="plain"
|
||||||
|
>{{ row.channel_name }}</el-tag
|
||||||
|
></template
|
||||||
|
></el-table-column
|
||||||
|
>
|
||||||
|
<el-table-column
|
||||||
|
label="能力"
|
||||||
|
min-width="180"
|
||||||
|
>
|
||||||
|
<template #default="{ row }">
|
||||||
|
<span v-if="row.model_type === 'text'">{{ row.multimodal ? "支持多模态" : "仅文本" }}</span>
|
||||||
|
<span v-else-if="row.model_type === 'video'">视频生成</span>
|
||||||
|
<span v-else>图片生成</span>
|
||||||
|
</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
label="模型价格"
|
||||||
|
min-width="240"
|
||||||
|
>
|
||||||
|
<template #default="{ row }">
|
||||||
|
<div
|
||||||
|
v-if="displayPrices(row).length"
|
||||||
|
class="price-list"
|
||||||
|
>
|
||||||
|
<span
|
||||||
|
v-for="price in displayPrices(row)"
|
||||||
|
:key="price.price_key"
|
||||||
|
>{{ priceLabel(price.price_key) }}：{{ Number(price.price).toFixed(2) }} 积分/{{ price.unit }}</span
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
<span v-else>--</span>
|
||||||
|
</template>
|
||||||
|
</el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
label="启用"
|
||||||
|
width="130"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-tag
|
||||||
|
v-if="!row.channel_enabled"
|
||||||
|
size="small"
|
||||||
|
type="warning"
|
||||||
|
effect="plain"
|
||||||
|
>渠道已禁用</el-tag
|
||||||
|
><el-switch
|
||||||
|
v-else
|
||||||
|
:model-value="row.enabled"
|
||||||
|
@change="toggle(row, Boolean($event))" /></template
|
||||||
|
></el-table-column>
|
||||||
|
<el-table-column
|
||||||
|
label="操作"
|
||||||
|
width="140"
|
||||||
|
fixed="right"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-button
|
||||||
|
link
|
||||||
|
type="primary"
|
||||||
|
@click="reset(row)"
|
||||||
|
>编辑</el-button
|
||||||
|
><el-button
|
||||||
|
link
|
||||||
|
type="danger"
|
||||||
|
@click="remove(row)"
|
||||||
|
>删除</el-button
|
||||||
|
></template
|
||||||
|
></el-table-column
|
||||||
|
>
|
||||||
|
</el-table>
|
||||||
|
<el-empty
|
||||||
|
v-if="!loading && !items.length"
|
||||||
|
description="暂无模型"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<el-dialog
|
||||||
|
v-model="dialog"
|
||||||
|
:title="editing ? '编辑模型' : '新建模型'"
|
||||||
|
width="min(820px, calc(100vw - 32px))"
|
||||||
|
class="managed-scroll-dialog model-dialog"
|
||||||
|
modal-class="managed-scroll-overlay"
|
||||||
|
:lock-scroll="true"
|
||||||
|
>
|
||||||
|
<el-form
|
||||||
|
:model="form"
|
||||||
|
label-position="top"
|
||||||
|
>
|
||||||
|
<div class="form-grid model-identity-grid">
|
||||||
|
<el-form-item
|
||||||
|
label="模型名"
|
||||||
|
required
|
||||||
|
><el-input v-model.trim="form.name"
|
||||||
|
/></el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="所属渠道"
|
||||||
|
required
|
||||||
|
><el-select
|
||||||
|
v-model="form.channel_id"
|
||||||
|
class="field-full"
|
||||||
|
><el-option
|
||||||
|
v-for="channel in channels"
|
||||||
|
:key="channel.id"
|
||||||
|
:label="channel.name"
|
||||||
|
:value="channel.id" /></el-select
|
||||||
|
></el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="模型类型"
|
||||||
|
required
|
||||||
|
><el-select
|
||||||
|
v-model="form.model_type"
|
||||||
|
class="field-full"
|
||||||
|
><el-option
|
||||||
|
label="文本"
|
||||||
|
value="text" /><el-option
|
||||||
|
label="图片"
|
||||||
|
value="image" /><el-option
|
||||||
|
label="视频"
|
||||||
|
value="video" /></el-select
|
||||||
|
></el-form-item>
|
||||||
|
</div>
|
||||||
|
<el-form-item
|
||||||
|
v-if="form.model_type === 'text'"
|
||||||
|
label="支持多模态"
|
||||||
|
><el-switch v-model="form.multimodal"
|
||||||
|
/></el-form-item>
|
||||||
|
<div
|
||||||
|
v-if="form.model_type === 'text'"
|
||||||
|
class="text-billing-settings"
|
||||||
|
>
|
||||||
|
<el-form-item
|
||||||
|
label="计费方式"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<el-radio-group v-model="form.text_billing_mode">
|
||||||
|
<el-radio-button label="per_request">按次计费</el-radio-button>
|
||||||
|
<el-radio-button label="per_token">按 Token 计费</el-radio-button>
|
||||||
|
</el-radio-group>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
<section class="model-price-section">
|
||||||
|
<header class="model-price-section__header">
|
||||||
|
<div>
|
||||||
|
<h3>模型价格</h3>
|
||||||
|
<p>输入用户执行对应任务时实际扣除的积分。</p>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
<div class="model-price-grid">
|
||||||
|
<div
|
||||||
|
v-for="[key, unit, label] in priceFields"
|
||||||
|
:key="key"
|
||||||
|
class="model-price-card"
|
||||||
|
>
|
||||||
|
<el-form-item :label="`${label}计费（积分/${unit}）`">
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="form.price_values[key]"
|
||||||
|
class="price-input"
|
||||||
|
:min="0"
|
||||||
|
:precision="2"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
</el-form>
|
||||||
|
<template #footer
|
||||||
|
><el-button @click="dialog = false">取消</el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
@click="save"
|
||||||
|
>保存</el-button
|
||||||
|
></template
|
||||||
|
>
|
||||||
|
</el-dialog>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
:global(.el-dialog.managed-scroll-dialog.model-dialog) {
|
||||||
|
height: auto;
|
||||||
|
}
|
||||||
|
:global(.el-dialog.managed-scroll-dialog.model-dialog .el-dialog__body) {
|
||||||
|
max-height: calc(100dvh - 180px);
|
||||||
|
flex: 0 1 auto;
|
||||||
|
}
|
||||||
|
.model-identity-grid {
|
||||||
|
grid-template-columns: repeat(3, minmax(0, 1fr));
|
||||||
|
}
|
||||||
|
.price-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 4px;
|
||||||
|
line-height: 1.45;
|
||||||
|
}
|
||||||
|
.model-price-section {
|
||||||
|
margin-top: 2px;
|
||||||
|
padding: 18px;
|
||||||
|
border: 1px solid var(--color-border-light);
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
background: var(--color-surface);
|
||||||
|
}
|
||||||
|
.model-price-section__header {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 20px;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
}
|
||||||
|
.model-price-section h3 {
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
.model-price-section__header p {
|
||||||
|
margin-top: 4px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
line-height: 1.5;
|
||||||
|
}
|
||||||
|
.model-price-rates {
|
||||||
|
flex: 0 0 auto;
|
||||||
|
display: grid;
|
||||||
|
gap: 4px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
text-align: right;
|
||||||
|
}
|
||||||
|
.model-price-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(220px, 1fr));
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
.model-price-card {
|
||||||
|
min-width: 0;
|
||||||
|
padding: 14px;
|
||||||
|
border: 1px solid var(--color-border-light);
|
||||||
|
border-radius: var(--radius-sm);
|
||||||
|
background: var(--color-bg);
|
||||||
|
}
|
||||||
|
.model-price-card :deep(.el-form-item) {
|
||||||
|
margin-bottom: 12px;
|
||||||
|
}
|
||||||
|
.price-conversion {
|
||||||
|
display: grid;
|
||||||
|
gap: 8px;
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
.price-conversion > div {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 10px;
|
||||||
|
}
|
||||||
|
.price-conversion dt {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
.price-conversion dd {
|
||||||
|
margin: 0;
|
||||||
|
color: var(--color-text);
|
||||||
|
font-size: 12px;
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
text-align: right;
|
||||||
|
}
|
||||||
|
.text-billing-settings {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: minmax(0, 1fr) minmax(180px, 0.7fr);
|
||||||
|
gap: 16px;
|
||||||
|
align-items: start;
|
||||||
|
}
|
||||||
|
.field-hint {
|
||||||
|
margin-top: 6px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
line-height: 1.5;
|
||||||
|
}
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.model-identity-grid {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
.model-price-section {
|
||||||
|
padding: 14px;
|
||||||
|
}
|
||||||
|
.model-price-section__header {
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 10px;
|
||||||
|
}
|
||||||
|
.model-price-rates {
|
||||||
|
text-align: left;
|
||||||
|
}
|
||||||
|
.model-price-grid {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
.text-billing-settings {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,475 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, onMounted, reactive, ref } from "vue";
|
||||||
|
import { ElMessageBox } from "element-plus";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import MasterDetailLayout from "@/components/MasterDetailLayout.vue";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
type Prompt = Record<string, any>;
|
||||||
|
interface PromptBackupItem {
|
||||||
|
title: string;
|
||||||
|
type: string;
|
||||||
|
content: string;
|
||||||
|
}
|
||||||
|
const promptTypes = [
|
||||||
|
"剧本反推",
|
||||||
|
"剧本分析",
|
||||||
|
"角色生成",
|
||||||
|
"场景生成",
|
||||||
|
"道具生成",
|
||||||
|
"分镜图生成",
|
||||||
|
"首尾帧生成",
|
||||||
|
"视频生成",
|
||||||
|
];
|
||||||
|
const toast = useToastStore();
|
||||||
|
const loading = ref(false);
|
||||||
|
const saving = ref(false);
|
||||||
|
const exporting = ref(false);
|
||||||
|
const restoring = ref(false);
|
||||||
|
const restoreInput = ref<HTMLInputElement>();
|
||||||
|
const items = ref<Prompt[]>([]);
|
||||||
|
const selectedId = ref("");
|
||||||
|
const creating = ref(false);
|
||||||
|
const query = reactive({ keyword: "", page: 1, page_size: 100 });
|
||||||
|
const form = reactive({ name: "", type: "", content: "" });
|
||||||
|
const selected = computed(() => items.value.find((item) => item.id === selectedId.value));
|
||||||
|
|
||||||
|
function reset(row?: Prompt) {
|
||||||
|
Object.assign(form, {
|
||||||
|
name: row?.name || "",
|
||||||
|
type: row?.type || row?.category || "",
|
||||||
|
content: row?.content || row?.version_content || "",
|
||||||
|
});
|
||||||
|
selectedId.value = row?.id || "";
|
||||||
|
creating.value = !row;
|
||||||
|
}
|
||||||
|
async function load() {
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.get("/prompts", { params: query });
|
||||||
|
items.value = data.data?.items || [];
|
||||||
|
if (selectedId.value) {
|
||||||
|
const current = items.value.find((item) => item.id === selectedId.value);
|
||||||
|
if (current) reset(current);
|
||||||
|
else reset();
|
||||||
|
} else if (items.value.length) reset(items.value[0]);
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function create() {
|
||||||
|
reset();
|
||||||
|
}
|
||||||
|
function validate() {
|
||||||
|
if (!form.name.trim()) return "请输入提示词名称";
|
||||||
|
if (!form.type) return "请选择适用类型";
|
||||||
|
if (!form.content.trim()) return "请输入提示词内容";
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
async function save() {
|
||||||
|
const message = validate();
|
||||||
|
if (message) {
|
||||||
|
toast.warning(message);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
saving.value = true;
|
||||||
|
try {
|
||||||
|
const payload = { name: form.name.trim(), type: form.type, content: form.content };
|
||||||
|
if (selectedId.value) await http.put(`/prompts/${selectedId.value}`, payload);
|
||||||
|
else {
|
||||||
|
const { data } = await http.post("/prompts", payload);
|
||||||
|
selectedId.value = data.data.id;
|
||||||
|
}
|
||||||
|
creating.value = false;
|
||||||
|
toast.success("提示词已保存");
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
saving.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function remove() {
|
||||||
|
if (!selected.value) return;
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm(
|
||||||
|
`确定要删除提示词“${selected.value.name || "未命名提示词"}”吗？此操作不可恢复。`,
|
||||||
|
"确认删除",
|
||||||
|
{ type: "warning" },
|
||||||
|
);
|
||||||
|
await http.delete(`/prompts/${selectedId.value}`);
|
||||||
|
toast.success("提示词已删除");
|
||||||
|
selectedId.value = "";
|
||||||
|
reset();
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
if (error !== "cancel" && error !== "close") showError(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function fetchAllPrompts() {
|
||||||
|
const result: Prompt[] = [];
|
||||||
|
let page = 1;
|
||||||
|
while (true) {
|
||||||
|
const { data } = await http.get("/prompts", { params: { page, page_size: 100 } });
|
||||||
|
const current = data.data?.items || [];
|
||||||
|
result.push(...current);
|
||||||
|
if (result.length >= Number(data.data?.total || 0) || !current.length) return result;
|
||||||
|
page += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function backupFilename() {
|
||||||
|
const value = new Date()
|
||||||
|
.toLocaleString("sv-SE", { timeZone: "Asia/Shanghai", hour12: false })
|
||||||
|
.replace(/[-:]/g, "")
|
||||||
|
.replace(" ", "-");
|
||||||
|
return `prompts-backup-${value}.json`;
|
||||||
|
}
|
||||||
|
async function exportPrompts() {
|
||||||
|
exporting.value = true;
|
||||||
|
try {
|
||||||
|
const prompts = await fetchAllPrompts();
|
||||||
|
const backup = {
|
||||||
|
format: "juhe-factory-prompts",
|
||||||
|
version: 1,
|
||||||
|
exported_at: new Date().toISOString(),
|
||||||
|
prompts: prompts.map((item) => ({
|
||||||
|
title: String(item.name || ""),
|
||||||
|
type: String(item.type || ""),
|
||||||
|
content: String(item.content || ""),
|
||||||
|
})),
|
||||||
|
};
|
||||||
|
const url = URL.createObjectURL(
|
||||||
|
new Blob([JSON.stringify(backup, null, 2)], { type: "application/json;charset=utf-8" }),
|
||||||
|
);
|
||||||
|
const link = document.createElement("a");
|
||||||
|
link.href = url;
|
||||||
|
link.download = backupFilename();
|
||||||
|
link.click();
|
||||||
|
URL.revokeObjectURL(url);
|
||||||
|
toast.success(`已导出 ${prompts.length} 条提示词`);
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
exporting.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function parseBackup(value: string): PromptBackupItem[] {
|
||||||
|
const parsed = JSON.parse(value) as { prompts?: unknown } | unknown[];
|
||||||
|
const source = Array.isArray(parsed) ? parsed : parsed?.prompts;
|
||||||
|
if (!Array.isArray(source) || !source.length) throw new Error("JSON 中没有可恢复的提示词");
|
||||||
|
const titles = new Set<string>();
|
||||||
|
return source.map((raw, index) => {
|
||||||
|
if (!raw || typeof raw !== "object") throw new Error(`第 ${index + 1} 条提示词格式无效`);
|
||||||
|
const item = raw as Record<string, unknown>;
|
||||||
|
const title = String(item.title || item.name || "").trim();
|
||||||
|
const type = String(item.type || "").trim();
|
||||||
|
const content = String(item.content || "").trim();
|
||||||
|
if (!title || !type || !content) throw new Error(`第 ${index + 1} 条提示词缺少标题、类型或内容`);
|
||||||
|
if (titles.has(title)) throw new Error(`JSON 中存在重复标题：${title}`);
|
||||||
|
titles.add(title);
|
||||||
|
return { title, type, content };
|
||||||
|
});
|
||||||
|
}
|
||||||
|
function chooseRestore() {
|
||||||
|
restoreInput.value?.click();
|
||||||
|
}
|
||||||
|
async function restorePrompts(event: Event) {
|
||||||
|
const input = event.target as HTMLInputElement;
|
||||||
|
const file = input.files?.[0];
|
||||||
|
if (!file) return;
|
||||||
|
restoring.value = true;
|
||||||
|
try {
|
||||||
|
const backup = parseBackup(await file.text());
|
||||||
|
const existing = await fetchAllPrompts();
|
||||||
|
const byTitle = new Map<string, Prompt>();
|
||||||
|
for (const item of existing)
|
||||||
|
if (!byTitle.has(String(item.name || "").trim())) byTitle.set(String(item.name || "").trim(), item);
|
||||||
|
const overwriteCount = backup.filter((item) => byTitle.has(item.title)).length;
|
||||||
|
await ElMessageBox.confirm(
|
||||||
|
`文件包含 ${backup.length} 条提示词，将覆盖 ${overwriteCount} 条、新增 ${backup.length - overwriteCount} 条。是否继续恢复？`,
|
||||||
|
"确认恢复提示词",
|
||||||
|
{ type: "warning", confirmButtonText: "确认恢复", cancelButtonText: "取消" },
|
||||||
|
);
|
||||||
|
for (const item of backup) {
|
||||||
|
const payload = { name: item.title, type: item.type, content: item.content };
|
||||||
|
const current = byTitle.get(item.title);
|
||||||
|
if (current) await http.put(`/prompts/${current.id}`, payload);
|
||||||
|
else await http.post("/prompts", payload);
|
||||||
|
}
|
||||||
|
toast.success(`恢复完成：覆盖 ${overwriteCount} 条，新增 ${backup.length - overwriteCount} 条`);
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
if (error !== "cancel" && error !== "close") {
|
||||||
|
if (error instanceof SyntaxError) toast.error("JSON 文件格式错误");
|
||||||
|
else if (error instanceof Error && !("response" in error)) toast.error(error.message);
|
||||||
|
else showError(error);
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
restoring.value = false;
|
||||||
|
input.value = "";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function showError(error: unknown) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
}
|
||||||
|
onMounted(load);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<MasterDetailLayout
|
||||||
|
title="提示词管理"
|
||||||
|
description="管理系统提示词，保存后立即生效"
|
||||||
|
empty-text="请从左侧选择提示词，或点击新增提示词"
|
||||||
|
:has-selection="Boolean(selectedId) || creating"
|
||||||
|
>
|
||||||
|
<template #header-actions>
|
||||||
|
<div class="header-actions">
|
||||||
|
<input
|
||||||
|
ref="restoreInput"
|
||||||
|
class="restore-input"
|
||||||
|
type="file"
|
||||||
|
accept="application/json,.json"
|
||||||
|
@change="restorePrompts"
|
||||||
|
/>
|
||||||
|
<el-button
|
||||||
|
:loading="exporting"
|
||||||
|
:disabled="restoring"
|
||||||
|
@click="exportPrompts"
|
||||||
|
><AppIcon name="Upload" />导出</el-button
|
||||||
|
>
|
||||||
|
<el-button
|
||||||
|
:loading="restoring"
|
||||||
|
:disabled="exporting"
|
||||||
|
@click="chooseRestore"
|
||||||
|
><AppIcon name="Download" />恢复</el-button
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
<template #list-toolbar
|
||||||
|
><el-input
|
||||||
|
v-model="query.keyword"
|
||||||
|
clearable
|
||||||
|
placeholder="搜索提示词名称"
|
||||||
|
><template #prefix><AppIcon name="Search" /></template></el-input
|
||||||
|
></template>
|
||||||
|
<template #list>
|
||||||
|
<el-button
|
||||||
|
class="create-prompt-button"
|
||||||
|
type="primary"
|
||||||
|
@click="create"
|
||||||
|
><AppIcon name="Plus" />新建提示词</el-button
|
||||||
|
>
|
||||||
|
<button
|
||||||
|
v-for="item in items"
|
||||||
|
:key="item.id"
|
||||||
|
class="master-item"
|
||||||
|
:class="{ active: item.id === selectedId }"
|
||||||
|
type="button"
|
||||||
|
@click="reset(item)"
|
||||||
|
>
|
||||||
|
<strong>{{ item.name || "未命名提示词" }}</strong
|
||||||
|
><span>{{ item.type || item.category || "未分类" }}</span>
|
||||||
|
</button>
|
||||||
|
<el-empty
|
||||||
|
v-if="!loading && !items.length"
|
||||||
|
description="暂无提示词"
|
||||||
|
:image-size="70"
|
||||||
|
/>
|
||||||
|
</template>
|
||||||
|
<template #detail>
|
||||||
|
<header class="editor-heading">
|
||||||
|
<div>
|
||||||
|
<span>{{ selectedId ? "编辑提示词" : "新增提示词" }}</span>
|
||||||
|
<h3>{{ form.name || "未命名提示词" }}</h3>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
<el-form
|
||||||
|
class="editor-form"
|
||||||
|
:model="form"
|
||||||
|
label-position="top"
|
||||||
|
>
|
||||||
|
<div class="form-grid">
|
||||||
|
<el-form-item
|
||||||
|
label="提示词名称"
|
||||||
|
required
|
||||||
|
><el-input
|
||||||
|
v-model.trim="form.name"
|
||||||
|
placeholder="请输入提示词名称"
|
||||||
|
/></el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="适用类型"
|
||||||
|
required
|
||||||
|
><el-select
|
||||||
|
v-model="form.type"
|
||||||
|
placeholder="请选择适用类型"
|
||||||
|
class="type-select"
|
||||||
|
filterable
|
||||||
|
><el-option
|
||||||
|
v-for="type in promptTypes"
|
||||||
|
:key="type"
|
||||||
|
:label="type"
|
||||||
|
:value="type" /></el-select
|
||||||
|
></el-form-item>
|
||||||
|
</div>
|
||||||
|
<el-form-item
|
||||||
|
class="prompt-content-field"
|
||||||
|
label="提示词内容"
|
||||||
|
required
|
||||||
|
><el-input
|
||||||
|
v-model="form.content"
|
||||||
|
type="textarea"
|
||||||
|
:rows="16"
|
||||||
|
resize="none"
|
||||||
|
placeholder="请输入提示词内容"
|
||||||
|
/></el-form-item>
|
||||||
|
</el-form>
|
||||||
|
<footer class="editor-actions">
|
||||||
|
<el-button
|
||||||
|
v-if="selectedId"
|
||||||
|
type="danger"
|
||||||
|
plain
|
||||||
|
@click="remove"
|
||||||
|
><AppIcon name="Delete" />删除</el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="saving"
|
||||||
|
@click="save"
|
||||||
|
>保存</el-button
|
||||||
|
>
|
||||||
|
</footer>
|
||||||
|
</template>
|
||||||
|
</MasterDetailLayout>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.header-actions {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: flex-end;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
.header-actions :deep(.el-button) {
|
||||||
|
margin-left: 0;
|
||||||
|
}
|
||||||
|
.restore-input {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
.master-item {
|
||||||
|
width: 100%;
|
||||||
|
padding: 12px 13px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 5px;
|
||||||
|
border: 1px solid transparent;
|
||||||
|
border-radius: 9px;
|
||||||
|
background: transparent;
|
||||||
|
color: var(--color-text);
|
||||||
|
text-align: left;
|
||||||
|
cursor: pointer;
|
||||||
|
transition:
|
||||||
|
background 0.16s,
|
||||||
|
border-color 0.16s;
|
||||||
|
}
|
||||||
|
.master-item:hover {
|
||||||
|
background: #eef4ff;
|
||||||
|
}
|
||||||
|
.master-item.active {
|
||||||
|
border-color: #b8cff5;
|
||||||
|
background: #eaf2ff;
|
||||||
|
}
|
||||||
|
.master-item strong {
|
||||||
|
font-size: 14px;
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
.master-item span {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
.create-prompt-button {
|
||||||
|
width: 100%;
|
||||||
|
margin: 0 0 8px;
|
||||||
|
justify-content: center;
|
||||||
|
}
|
||||||
|
.editor-heading {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 16px;
|
||||||
|
padding-bottom: 18px;
|
||||||
|
border-bottom: 1px solid var(--color-border-light);
|
||||||
|
}
|
||||||
|
.editor-heading span {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
.editor-heading h3 {
|
||||||
|
margin-top: 4px;
|
||||||
|
font-size: 20px;
|
||||||
|
}
|
||||||
|
.editor-form {
|
||||||
|
display: flex;
|
||||||
|
flex: 1;
|
||||||
|
min-height: 0;
|
||||||
|
flex-direction: column;
|
||||||
|
padding-top: 22px;
|
||||||
|
}
|
||||||
|
.form-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: minmax(0, 1fr) minmax(0, 1fr);
|
||||||
|
gap: 16px;
|
||||||
|
}
|
||||||
|
.type-select {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
.prompt-content-field {
|
||||||
|
display: flex;
|
||||||
|
flex: 1;
|
||||||
|
min-height: 0;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
.prompt-content-field :deep(.el-form-item__content) {
|
||||||
|
flex: 1;
|
||||||
|
min-height: 0;
|
||||||
|
}
|
||||||
|
.prompt-content-field :deep(.el-textarea) {
|
||||||
|
height: 100%;
|
||||||
|
}
|
||||||
|
.prompt-content-field :deep(.el-textarea__inner) {
|
||||||
|
height: 100%;
|
||||||
|
min-height: 420px !important;
|
||||||
|
}
|
||||||
|
.editor-actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
gap: 8px;
|
||||||
|
padding-top: 18px;
|
||||||
|
border-top: 1px solid var(--color-border-light);
|
||||||
|
}
|
||||||
|
:deep(.master-editor) {
|
||||||
|
display: flex;
|
||||||
|
min-height: 0;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
@media (max-width: 640px) {
|
||||||
|
.header-actions {
|
||||||
|
width: 100%;
|
||||||
|
justify-content: flex-start;
|
||||||
|
}
|
||||||
|
.form-grid {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
.prompt-content-field :deep(.el-textarea__inner) {
|
||||||
|
min-height: 360px !important;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,271 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { onMounted, reactive, ref } from "vue";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import AdminNumberInput from "@/components/AdminNumberInput.vue";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
const toast = useToastStore(),
|
||||||
|
loading = ref(false),
|
||||||
|
generating = ref(false),
|
||||||
|
dialog = ref(false),
|
||||||
|
resultDialog = ref(false),
|
||||||
|
items = ref<Record<string, any>[]>([]),
|
||||||
|
codes = ref<string[]>([]),
|
||||||
|
total = ref(0);
|
||||||
|
async function copyCode(code: string) {
|
||||||
|
await navigator.clipboard.writeText(code);
|
||||||
|
toast.success("兑换码已复制");
|
||||||
|
}
|
||||||
|
const query = reactive({ keyword: "", status: "", page: 1, page_size: 15 });
|
||||||
|
const form = reactive({ name: "", points: 10, quantity: 1, expires_at: "" });
|
||||||
|
async function load() {
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.get("/redemption-codes", { params: query });
|
||||||
|
items.value = data.data.items;
|
||||||
|
total.value = data.data.total;
|
||||||
|
} catch (e) {
|
||||||
|
show(e);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function open() {
|
||||||
|
Object.assign(form, { name: "", points: 10, quantity: 1, expires_at: "" });
|
||||||
|
dialog.value = true;
|
||||||
|
}
|
||||||
|
async function generate() {
|
||||||
|
const points = Number(form.points);
|
||||||
|
const quantity = Number(form.quantity);
|
||||||
|
if (!Number.isFinite(points) || points < 1 || points > 100) {
|
||||||
|
toast.warning("单码积分数量必须为 1 至 100");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!Number.isInteger(quantity) || quantity < 1 || quantity > 10) {
|
||||||
|
toast.warning("生成数量必须为 1 至 10");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
generating.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.post("/redemption-batches", form);
|
||||||
|
codes.value = data.data.codes;
|
||||||
|
dialog.value = false;
|
||||||
|
resultDialog.value = true;
|
||||||
|
toast.success("兑换码批次生成成功");
|
||||||
|
load();
|
||||||
|
} catch (e) {
|
||||||
|
show(e);
|
||||||
|
} finally {
|
||||||
|
generating.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function copyAll() {
|
||||||
|
await navigator.clipboard.writeText(codes.value.join("\n"));
|
||||||
|
toast.success("完整兑换码已复制");
|
||||||
|
}
|
||||||
|
function exportCodes() {
|
||||||
|
const blob = new Blob([codes.value.join("\r\n")], { type: "text/plain;charset=utf-8" });
|
||||||
|
const url = URL.createObjectURL(blob);
|
||||||
|
const a = document.createElement("a");
|
||||||
|
a.href = url;
|
||||||
|
a.download = `redemption-codes-${Date.now()}.txt`;
|
||||||
|
a.click();
|
||||||
|
URL.revokeObjectURL(url);
|
||||||
|
}
|
||||||
|
function closeResult() {
|
||||||
|
codes.value = [];
|
||||||
|
resultDialog.value = false;
|
||||||
|
}
|
||||||
|
function show(e: unknown) {
|
||||||
|
const d = apiError(e);
|
||||||
|
toast.error(d.message, d.copyText);
|
||||||
|
}
|
||||||
|
const statusText = (v: string) => ({ unused: "未使用", used: "已使用", expired: "已过期" })[v] || v;
|
||||||
|
const fmt = (v: any) =>
|
||||||
|
v ? new Date(v).toLocaleString("zh-CN", { timeZone: "Asia/Shanghai", hour12: false }) : "长期有效";
|
||||||
|
onMounted(load);
|
||||||
|
</script>
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<PageHeader
|
||||||
|
title="兑换码管理"
|
||||||
|
description="完整兑换码可在列表中直接查看并复制"
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
@click="open"
|
||||||
|
>
|
||||||
|
<AppIcon name="Plus" />生成兑换码
|
||||||
|
</el-button></PageHeader
|
||||||
|
>
|
||||||
|
<section class="panel filter-bar">
|
||||||
|
<el-input
|
||||||
|
v-model="query.keyword"
|
||||||
|
clearable
|
||||||
|
placeholder="批次、兑换码或用户 UID"
|
||||||
|
/><el-select
|
||||||
|
v-model="query.status"
|
||||||
|
clearable
|
||||||
|
placeholder="状态"
|
||||||
|
><el-option
|
||||||
|
label="未使用"
|
||||||
|
value="unused" /><el-option
|
||||||
|
label="已使用"
|
||||||
|
value="used" /><el-option
|
||||||
|
label="已过期"
|
||||||
|
value="expired" /></el-select
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
@click="
|
||||||
|
query.page = 1;
|
||||||
|
load();
|
||||||
|
"
|
||||||
|
>查询</el-button
|
||||||
|
>
|
||||||
|
</section>
|
||||||
|
<section
|
||||||
|
class="panel table-panel"
|
||||||
|
v-loading="loading"
|
||||||
|
>
|
||||||
|
<el-table :data="items"
|
||||||
|
><el-table-column
|
||||||
|
prop="code"
|
||||||
|
label="兑换码"
|
||||||
|
min-width="230"
|
||||||
|
show-overflow-tooltip
|
||||||
|
><template #default="{ row }"
|
||||||
|
><span class="code-cell">{{ row.code || "--" }}</span></template
|
||||||
|
></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="操作"
|
||||||
|
width="90"
|
||||||
|
fixed="right"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-button
|
||||||
|
v-if="row.code"
|
||||||
|
link
|
||||||
|
type="primary"
|
||||||
|
@click="copyCode(row.code)"
|
||||||
|
><AppIcon name="CopyDocument" />复制</el-button
|
||||||
|
></template
|
||||||
|
></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
prop="batch_name"
|
||||||
|
label="所属批次"
|
||||||
|
min-width="140" /><el-table-column
|
||||||
|
label="积分数量"
|
||||||
|
width="110"
|
||||||
|
><template #default="{ row }">{{ Number(row.points).toFixed(2) }}</template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="状态"
|
||||||
|
width="100"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-tag effect="plain">{{ statusText(row.status) }}</el-tag></template
|
||||||
|
></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
prop="redeemed_uid"
|
||||||
|
label="兑换用户 UID"
|
||||||
|
width="130" /><el-table-column
|
||||||
|
label="兑换时间"
|
||||||
|
min-width="180"
|
||||||
|
><template #default="{ row }">{{ row.redeemed_at ? fmt(row.redeemed_at) : "--" }}</template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="失效时间"
|
||||||
|
min-width="180"
|
||||||
|
><template #default="{ row }">{{
|
||||||
|
row.status === "used" ? "--" : fmt(row.expires_at)
|
||||||
|
}}</template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
prop="created_by"
|
||||||
|
label="创建管理员"
|
||||||
|
width="130" /></el-table
|
||||||
|
><el-empty
|
||||||
|
v-if="!loading && !items.length"
|
||||||
|
description="暂无兑换码"
|
||||||
|
/><el-pagination
|
||||||
|
v-if="total > query.page_size"
|
||||||
|
v-model:current-page="query.page"
|
||||||
|
background
|
||||||
|
layout="total,prev,pager,next"
|
||||||
|
:total="total"
|
||||||
|
@current-change="load"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
<el-dialog
|
||||||
|
v-model="dialog"
|
||||||
|
title="按批次生成兑换码"
|
||||||
|
width="min(560px, calc(100vw - 32px))"
|
||||||
|
><el-form
|
||||||
|
:model="form"
|
||||||
|
label-position="top"
|
||||||
|
><el-form-item
|
||||||
|
label="批次名称"
|
||||||
|
required
|
||||||
|
><el-input v-model.trim="form.name"
|
||||||
|
/></el-form-item>
|
||||||
|
<div class="form-grid">
|
||||||
|
<el-form-item
|
||||||
|
label="单码积分数量"
|
||||||
|
required
|
||||||
|
><AdminNumberInput
|
||||||
|
v-model="form.points"
|
||||||
|
:min="1"
|
||||||
|
:max="100"
|
||||||
|
:precision="2" /></el-form-item
|
||||||
|
><el-form-item
|
||||||
|
label="生成数量"
|
||||||
|
required
|
||||||
|
><AdminNumberInput
|
||||||
|
v-model="form.quantity"
|
||||||
|
:min="1"
|
||||||
|
:max="10"
|
||||||
|
:precision="0"
|
||||||
|
/></el-form-item>
|
||||||
|
</div>
|
||||||
|
<el-form-item label="有效期"
|
||||||
|
><el-date-picker
|
||||||
|
v-model="form.expires_at"
|
||||||
|
type="datetime"
|
||||||
|
value-format="YYYY-MM-DDTHH:mm:ssZ"
|
||||||
|
class="field-full"
|
||||||
|
/></el-form-item> </el-form
|
||||||
|
><template #footer
|
||||||
|
><el-button @click="dialog = false">取消</el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="generating"
|
||||||
|
:disabled="generating"
|
||||||
|
@click="generate"
|
||||||
|
>生成</el-button
|
||||||
|
></template
|
||||||
|
></el-dialog
|
||||||
|
>
|
||||||
|
<el-dialog
|
||||||
|
:model-value="resultDialog"
|
||||||
|
title="完整兑换码（仅展示一次）"
|
||||||
|
width="min(720px, calc(100vw - 32px))"
|
||||||
|
:close-on-click-modal="false"
|
||||||
|
:close-on-press-escape="false"
|
||||||
|
:show-close="false"
|
||||||
|
><el-alert
|
||||||
|
type="warning"
|
||||||
|
:closable="false"
|
||||||
|
title="关闭后系统无法再次查看完整兑换码，请立即复制或导出。"
|
||||||
|
/><el-input
|
||||||
|
class="code-output"
|
||||||
|
:model-value="codes.join('\n')"
|
||||||
|
type="textarea"
|
||||||
|
:rows="12"
|
||||||
|
readonly
|
||||||
|
/><template #footer
|
||||||
|
><el-button @click="copyAll"> <AppIcon name="CopyDocument" />复制全部 </el-button
|
||||||
|
><el-button @click="exportCodes"> <AppIcon name="Download" />导出文件 </el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
@click="closeResult"
|
||||||
|
>我已保存，关闭</el-button
|
||||||
|
></template
|
||||||
|
></el-dialog
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
@@ -0,0 +1,181 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { reactive, ref } from "vue";
|
||||||
|
import { useRouter } from "vue-router";
|
||||||
|
import type { FormInstance, FormRules } from "element-plus";
|
||||||
|
import http, { apiError, clearAuth } from "@/api/http";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import { useAuthStore } from "@/stores/auth";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
const router = useRouter();
|
||||||
|
const auth = useAuthStore();
|
||||||
|
const toast = useToastStore();
|
||||||
|
const formRef = ref<FormInstance>();
|
||||||
|
const saving = ref(false);
|
||||||
|
const form = reactive({ currentPassword: "", newPassword: "", confirmPassword: "" });
|
||||||
|
|
||||||
|
const rules: FormRules = {
|
||||||
|
currentPassword: [{ required: true, message: "请输入当前密码", trigger: "blur" }],
|
||||||
|
newPassword: [
|
||||||
|
{ required: true, message: "请输入新密码", trigger: "blur" },
|
||||||
|
{ min: 8, max: 20, message: "密码长度必须为 8～20 位", trigger: "blur" },
|
||||||
|
],
|
||||||
|
confirmPassword: [
|
||||||
|
{ required: true, message: "请再次输入新密码", trigger: "blur" },
|
||||||
|
{
|
||||||
|
validator: (_rule, value, callback) =>
|
||||||
|
value === form.newPassword ? callback() : callback(new Error("两次输入的新密码不一致")),
|
||||||
|
trigger: "blur",
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
|
||||||
|
async function submit() {
|
||||||
|
if (!(await formRef.value?.validate().catch(() => false))) return;
|
||||||
|
saving.value = true;
|
||||||
|
try {
|
||||||
|
await http.put("/auth/password", {
|
||||||
|
current_password: form.currentPassword,
|
||||||
|
new_password: form.newPassword,
|
||||||
|
confirm_password: form.confirmPassword,
|
||||||
|
});
|
||||||
|
clearAuth();
|
||||||
|
auth.profile = null;
|
||||||
|
toast.success("密码已修改，请使用新密码重新登录");
|
||||||
|
await router.replace("/login");
|
||||||
|
} catch (error) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
} finally {
|
||||||
|
saving.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="page-stack security-page">
|
||||||
|
<PageHeader title="账号安全" />
|
||||||
|
|
||||||
|
<section class="panel security-panel">
|
||||||
|
<div class="security-intro">
|
||||||
|
<span class="security-icon"
|
||||||
|
><AppIcon
|
||||||
|
name="Lock"
|
||||||
|
:size="22"
|
||||||
|
/></span>
|
||||||
|
<div>
|
||||||
|
<h3>修改管理员密码</h3>
|
||||||
|
<p>需验证当前密码。修改成功后将退出当前账号，请使用新密码重新登录。</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<el-form
|
||||||
|
ref="formRef"
|
||||||
|
:model="form"
|
||||||
|
:rules="rules"
|
||||||
|
label-position="top"
|
||||||
|
class="password-form"
|
||||||
|
>
|
||||||
|
<el-form-item
|
||||||
|
label="当前密码"
|
||||||
|
prop="currentPassword"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model="form.currentPassword"
|
||||||
|
type="password"
|
||||||
|
autocomplete="current-password"
|
||||||
|
show-password
|
||||||
|
placeholder="请输入当前密码"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="新密码"
|
||||||
|
prop="newPassword"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model="form.newPassword"
|
||||||
|
type="password"
|
||||||
|
autocomplete="new-password"
|
||||||
|
show-password
|
||||||
|
placeholder="请输入 8～20 位新密码"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="确认新密码"
|
||||||
|
prop="confirmPassword"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model="form.confirmPassword"
|
||||||
|
type="password"
|
||||||
|
autocomplete="new-password"
|
||||||
|
show-password
|
||||||
|
placeholder="请再次输入新密码"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<div class="form-actions">
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="saving"
|
||||||
|
@click="submit"
|
||||||
|
>确认修改</el-button
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</el-form>
|
||||||
|
</section>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.security-page {
|
||||||
|
max-width: 760px;
|
||||||
|
}
|
||||||
|
.security-panel {
|
||||||
|
padding: 24px;
|
||||||
|
}
|
||||||
|
.security-intro {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
gap: 14px;
|
||||||
|
padding-bottom: 20px;
|
||||||
|
border-bottom: 1px solid var(--color-border-light);
|
||||||
|
}
|
||||||
|
.security-icon {
|
||||||
|
width: 42px;
|
||||||
|
height: 42px;
|
||||||
|
flex: 0 0 42px;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
color: var(--color-primary);
|
||||||
|
background: var(--color-primary-soft);
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
}
|
||||||
|
.security-intro h3 {
|
||||||
|
font-size: 16px;
|
||||||
|
}
|
||||||
|
.security-intro p {
|
||||||
|
margin-top: 5px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
.password-form {
|
||||||
|
max-width: 480px;
|
||||||
|
margin-top: 22px;
|
||||||
|
}
|
||||||
|
.form-actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
padding-top: 4px;
|
||||||
|
}
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.security-panel {
|
||||||
|
padding: 18px;
|
||||||
|
}
|
||||||
|
.password-form {
|
||||||
|
max-width: none;
|
||||||
|
}
|
||||||
|
.form-actions .el-button {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,437 @@
|
|||||||
|
<!-- 视觉风格管理页，负责风格图片、名称和排序维护。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, onMounted, reactive, ref } from "vue";
|
||||||
|
import { ElMessageBox, type FormInstance, type UploadRequestOptions } from "element-plus";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
type UploadError = Error & { status: number; method: string; url: string };
|
||||||
|
const toast = useToastStore();
|
||||||
|
const loading = ref(false);
|
||||||
|
const saving = ref(false);
|
||||||
|
const dialogOpen = ref(false);
|
||||||
|
const editingId = ref("");
|
||||||
|
const items = ref<Record<string, any>[]>([]);
|
||||||
|
const formRef = ref<FormInstance>();
|
||||||
|
const form = reactive<Record<string, any>>({ name: "", image: "", image_key: "" });
|
||||||
|
const draggedId = ref("");
|
||||||
|
const originalOrder = ref<string[]>([]);
|
||||||
|
const reordering = ref(false);
|
||||||
|
const canReorder = computed(() => items.value.length > 1 && !reordering.value);
|
||||||
|
const rules = { name: [{ required: true, message: "请填写风格名", trigger: "blur" }] };
|
||||||
|
|
||||||
|
// 加载风格列表，并保留接口返回的排序结果。
|
||||||
|
async function load() {
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.get("/resources/styles", { params: { page: 1, page_size: 100 } });
|
||||||
|
items.value = data.data.items || [];
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 重置弹窗表单；编辑时回填当前风格及其图片信息。
|
||||||
|
function reset(row?: Record<string, any>) {
|
||||||
|
editingId.value = row?.id || "";
|
||||||
|
Object.assign(form, {
|
||||||
|
name: row?.name || "",
|
||||||
|
image: row?.image_url || "",
|
||||||
|
image_key: row?.image_key || "",
|
||||||
|
image_url: row?.image_url || "",
|
||||||
|
image_mime: row?.image_mime || "",
|
||||||
|
image_size: row?.image_size || "",
|
||||||
|
image_width: row?.image_width || "",
|
||||||
|
image_height: row?.image_height || "",
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// 打开新建风格弹窗。
|
||||||
|
function openCreate() {
|
||||||
|
reset();
|
||||||
|
dialogOpen.value = true;
|
||||||
|
}
|
||||||
|
// 打开编辑风格弹窗并回填目标风格。
|
||||||
|
function openEdit(row: Record<string, any>) {
|
||||||
|
reset(row);
|
||||||
|
dialogOpen.value = true;
|
||||||
|
}
|
||||||
|
// 上传风格图片，并将服务端返回的图片元数据写入表单。
|
||||||
|
async function upload(options: UploadRequestOptions) {
|
||||||
|
const data = new FormData();
|
||||||
|
data.append("file", options.file);
|
||||||
|
try {
|
||||||
|
const response = await http.post("/uploads/style-images", data, {
|
||||||
|
headers: { "Content-Type": "multipart/form-data" },
|
||||||
|
});
|
||||||
|
Object.assign(form, response.data.data);
|
||||||
|
form.image = response.data.data.image_url || response.data.data.image_key;
|
||||||
|
options.onSuccess(response.data);
|
||||||
|
} catch (error) {
|
||||||
|
const uploadError = Object.assign(error instanceof Error ? error : new Error(String(error)), {
|
||||||
|
status: 0,
|
||||||
|
method: "POST",
|
||||||
|
url: "/uploads/style-images",
|
||||||
|
}) as UploadError;
|
||||||
|
options.onError(uploadError);
|
||||||
|
showError(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 校验并保存风格；图片上传成功后才允许提交。
|
||||||
|
async function save() {
|
||||||
|
if (!(await formRef.value?.validate().catch(() => false))) return;
|
||||||
|
if (!form.image_key) {
|
||||||
|
toast.warning("请先上传风格图片");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
saving.value = true;
|
||||||
|
try {
|
||||||
|
const payload = { ...form };
|
||||||
|
delete payload.image;
|
||||||
|
if (editingId.value) await http.put(`/resources/styles/${editingId.value}`, payload);
|
||||||
|
else await http.post("/resources/styles", payload);
|
||||||
|
dialogOpen.value = false;
|
||||||
|
toast.success(editingId.value ? "风格已更新" : "风格已创建");
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
saving.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 删除指定风格，并在成功后刷新列表。
|
||||||
|
async function remove(row: Record<string, any>) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm(`确定删除风格“${row.name}”吗？`, "删除风格", {
|
||||||
|
type: "warning",
|
||||||
|
confirmButtonText: "删除",
|
||||||
|
cancelButtonText: "取消",
|
||||||
|
});
|
||||||
|
await http.delete(`/resources/styles/${row.id}`);
|
||||||
|
toast.success("风格已删除");
|
||||||
|
await load();
|
||||||
|
} catch (error) {
|
||||||
|
if (error !== "cancel" && error !== "close") showError(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 记录拖拽起点和原始顺序，用于提交排序或失败回滚。
|
||||||
|
function startDrag(row: Record<string, any>, event: DragEvent) {
|
||||||
|
if (!canReorder.value) return;
|
||||||
|
draggedId.value = row.id;
|
||||||
|
originalOrder.value = items.value.map((item) => item.id);
|
||||||
|
event.dataTransfer?.setData("text/plain", row.id);
|
||||||
|
}
|
||||||
|
// 根据当前悬停位置实时调整风格卡片顺序。
|
||||||
|
function moveOver(targetId: string) {
|
||||||
|
const sourceIndex = items.value.findIndex((item) => item.id === draggedId.value);
|
||||||
|
const targetIndex = items.value.findIndex((item) => item.id === targetId);
|
||||||
|
if (sourceIndex < 0 || targetIndex < 0 || sourceIndex === targetIndex) return;
|
||||||
|
const [moved] = items.value.splice(sourceIndex, 1);
|
||||||
|
items.value.splice(targetIndex, 0, moved);
|
||||||
|
}
|
||||||
|
// 提交拖拽后的顺序；失败时恢复原始顺序。
|
||||||
|
async function finishDrag() {
|
||||||
|
if (!draggedId.value) return;
|
||||||
|
const nextOrder = items.value.map((item) => item.id);
|
||||||
|
draggedId.value = "";
|
||||||
|
if (nextOrder.every((id, index) => id === originalOrder.value[index])) return;
|
||||||
|
reordering.value = true;
|
||||||
|
try {
|
||||||
|
await http.put("/resources/styles/reorder", { ids: nextOrder });
|
||||||
|
toast.success("风格排序已保存");
|
||||||
|
} catch (error) {
|
||||||
|
const positions = new Map(originalOrder.value.map((id, index) => [id, index]));
|
||||||
|
items.value.sort((a, b) => (positions.get(a.id) ?? 0) - (positions.get(b.id) ?? 0));
|
||||||
|
showError(error);
|
||||||
|
} finally {
|
||||||
|
reordering.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 将接口异常转换为管理端统一提示。
|
||||||
|
function showError(error: unknown) {
|
||||||
|
const detail = apiError(error);
|
||||||
|
toast.error(detail.message, detail.copyText);
|
||||||
|
}
|
||||||
|
|
||||||
|
onMounted(load);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<PageHeader
|
||||||
|
title="风格管理"
|
||||||
|
description="管理项目风格、展示图片与排序"
|
||||||
|
>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
@click="openCreate"
|
||||||
|
><AppIcon name="Plus" />新建风格</el-button
|
||||||
|
>
|
||||||
|
</PageHeader>
|
||||||
|
<section
|
||||||
|
v-loading="loading"
|
||||||
|
class="panel style-grid-panel"
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
v-if="items.length"
|
||||||
|
class="style-card-grid"
|
||||||
|
>
|
||||||
|
<article
|
||||||
|
v-for="row in items"
|
||||||
|
:key="row.id"
|
||||||
|
class="style-card"
|
||||||
|
:class="{ 'is-dragging': draggedId === row.id }"
|
||||||
|
@dragover.prevent="moveOver(row.id)"
|
||||||
|
@drop.prevent="finishDrag"
|
||||||
|
>
|
||||||
|
<div class="style-card__visual">
|
||||||
|
<img
|
||||||
|
:src="row.image_url"
|
||||||
|
alt=""
|
||||||
|
draggable="false"
|
||||||
|
/>
|
||||||
|
<span
|
||||||
|
class="style-card__drag"
|
||||||
|
:draggable="canReorder"
|
||||||
|
title="拖动排序"
|
||||||
|
@dragstart="startDrag(row, $event)"
|
||||||
|
@dragend="finishDrag"
|
||||||
|
>
|
||||||
|
<AppIcon name="Rank" />
|
||||||
|
</span>
|
||||||
|
<div class="style-card__actions">
|
||||||
|
<el-button
|
||||||
|
circle
|
||||||
|
title="编辑风格"
|
||||||
|
@click="openEdit(row)"
|
||||||
|
><AppIcon name="EditPen"
|
||||||
|
/></el-button>
|
||||||
|
<el-button
|
||||||
|
circle
|
||||||
|
type="danger"
|
||||||
|
plain
|
||||||
|
title="删除风格"
|
||||||
|
@click="remove(row)"
|
||||||
|
><AppIcon name="Delete"
|
||||||
|
/></el-button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<strong>{{ row.name }}</strong>
|
||||||
|
</article>
|
||||||
|
</div>
|
||||||
|
<el-empty
|
||||||
|
v-else
|
||||||
|
description="暂无风格"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<el-dialog
|
||||||
|
v-model="dialogOpen"
|
||||||
|
:title="editingId ? '编辑风格' : '新建风格'"
|
||||||
|
class="style-dialog"
|
||||||
|
width="min(720px, calc(100vw - 32px))"
|
||||||
|
align-center
|
||||||
|
>
|
||||||
|
<el-form
|
||||||
|
ref="formRef"
|
||||||
|
:model="form"
|
||||||
|
:rules="rules"
|
||||||
|
label-position="top"
|
||||||
|
class="style-form"
|
||||||
|
>
|
||||||
|
<div class="style-form__preview-column">
|
||||||
|
<span class="style-form__label">图片预览</span>
|
||||||
|
<div class="upload-preview-frame">
|
||||||
|
<img
|
||||||
|
v-if="form.image"
|
||||||
|
class="upload-preview"
|
||||||
|
:src="form.image"
|
||||||
|
alt=""
|
||||||
|
/>
|
||||||
|
<div
|
||||||
|
v-else
|
||||||
|
class="upload-preview-empty"
|
||||||
|
>
|
||||||
|
<AppIcon name="Picture" />
|
||||||
|
<span>9:16 图片预览</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="style-form__fields">
|
||||||
|
<el-form-item
|
||||||
|
label="风格名"
|
||||||
|
prop="name"
|
||||||
|
><el-input v-model.trim="form.name"
|
||||||
|
/></el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="风格图片"
|
||||||
|
required
|
||||||
|
>
|
||||||
|
<el-upload
|
||||||
|
:show-file-list="false"
|
||||||
|
:http-request="upload"
|
||||||
|
accept="image/jpeg,image/png,image/gif"
|
||||||
|
>
|
||||||
|
<el-button><AppIcon name="Upload" />{{ form.image ? "重新上传" : "上传图片" }}</el-button>
|
||||||
|
</el-upload>
|
||||||
|
<p class="field-helper">推荐 9:16，支持 JPEG、PNG、GIF，最大 10MB</p>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
</el-form>
|
||||||
|
<template #footer
|
||||||
|
><el-button @click="dialogOpen = false">取消</el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="saving"
|
||||||
|
@click="save"
|
||||||
|
>保存</el-button
|
||||||
|
></template
|
||||||
|
>
|
||||||
|
</el-dialog>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.style-grid-panel {
|
||||||
|
flex: 1;
|
||||||
|
min-height: 0;
|
||||||
|
overflow: auto;
|
||||||
|
}
|
||||||
|
.style-card-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fill, minmax(170px, 1fr));
|
||||||
|
gap: 16px;
|
||||||
|
}
|
||||||
|
.style-card {
|
||||||
|
min-width: 0;
|
||||||
|
overflow: hidden;
|
||||||
|
background: #fff;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-lg);
|
||||||
|
transition:
|
||||||
|
border-color 0.18s,
|
||||||
|
transform 0.18s;
|
||||||
|
}
|
||||||
|
.style-card:hover {
|
||||||
|
border-color: var(--color-primary);
|
||||||
|
transform: translateY(-2px);
|
||||||
|
}
|
||||||
|
.style-card.is-dragging {
|
||||||
|
opacity: 0.55;
|
||||||
|
}
|
||||||
|
.style-card__visual {
|
||||||
|
height: 230px;
|
||||||
|
position: relative;
|
||||||
|
overflow: hidden;
|
||||||
|
background: var(--color-bg);
|
||||||
|
}
|
||||||
|
.style-card__visual img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
.style-card__drag {
|
||||||
|
position: absolute;
|
||||||
|
top: 10px;
|
||||||
|
left: 10px;
|
||||||
|
width: 34px;
|
||||||
|
height: 34px;
|
||||||
|
display: grid;
|
||||||
|
place-items: center;
|
||||||
|
color: #fff;
|
||||||
|
background: #111827;
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: grab;
|
||||||
|
}
|
||||||
|
.style-card__actions {
|
||||||
|
position: absolute;
|
||||||
|
top: 10px;
|
||||||
|
right: 10px;
|
||||||
|
display: flex;
|
||||||
|
gap: 6px;
|
||||||
|
}
|
||||||
|
.style-card > strong {
|
||||||
|
padding: 13px 14px;
|
||||||
|
display: block;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
.style-form {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: 180px minmax(0, 1fr);
|
||||||
|
gap: var(--space-lg);
|
||||||
|
align-items: start;
|
||||||
|
}
|
||||||
|
.style-form__preview-column,
|
||||||
|
.style-form__fields {
|
||||||
|
min-width: 0;
|
||||||
|
}
|
||||||
|
.style-form__label {
|
||||||
|
display: block;
|
||||||
|
margin-bottom: var(--space-sm);
|
||||||
|
color: var(--color-text);
|
||||||
|
font-size: var(--font-size-sm);
|
||||||
|
line-height: 22px;
|
||||||
|
}
|
||||||
|
.style-form__label::before {
|
||||||
|
margin-right: var(--space-xs);
|
||||||
|
color: var(--color-danger);
|
||||||
|
content: "*";
|
||||||
|
}
|
||||||
|
.upload-preview-frame {
|
||||||
|
width: 100%;
|
||||||
|
aspect-ratio: 9 / 16;
|
||||||
|
overflow: hidden;
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-md);
|
||||||
|
background: var(--color-bg);
|
||||||
|
}
|
||||||
|
.upload-preview {
|
||||||
|
display: block;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: contain;
|
||||||
|
}
|
||||||
|
.upload-preview-empty {
|
||||||
|
display: flex;
|
||||||
|
height: 100%;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: var(--space-sm);
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: var(--font-size-sm);
|
||||||
|
}
|
||||||
|
.upload-preview-empty :deep(.el-icon) {
|
||||||
|
font-size: 28px;
|
||||||
|
}
|
||||||
|
.style-form__fields :deep(.el-form-item:last-child) {
|
||||||
|
margin-bottom: 0;
|
||||||
|
}
|
||||||
|
.field-helper {
|
||||||
|
width: 100%;
|
||||||
|
margin: var(--space-sm) 0 0;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: var(--font-size-sm);
|
||||||
|
line-height: 1.6;
|
||||||
|
}
|
||||||
|
@media (max-width: 600px) {
|
||||||
|
.style-card-grid {
|
||||||
|
grid-template-columns: repeat(2, minmax(0, 1fr));
|
||||||
|
}
|
||||||
|
.style-card__visual {
|
||||||
|
height: 190px;
|
||||||
|
}
|
||||||
|
.style-form {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
gap: var(--space-md);
|
||||||
|
}
|
||||||
|
.upload-preview-frame {
|
||||||
|
width: min(180px, 100%);
|
||||||
|
margin: 0 auto;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,730 @@
|
|||||||
|
<!-- 管理后台用户管理页，负责用户创建、状态维护、每日限额和积分发放。 -->
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { computed, onBeforeUnmount, onMounted, reactive, ref } from "vue";
|
||||||
|
import { ElMessageBox, type FormInstance } from "element-plus";
|
||||||
|
import http, { apiError } from "@/api/http";
|
||||||
|
import AppIcon from "@/components/AppIcon.vue";
|
||||||
|
import AdminNumberInput from "@/components/AdminNumberInput.vue";
|
||||||
|
import PageHeader from "@/components/PageHeader.vue";
|
||||||
|
import { useToastStore } from "@/stores/toast";
|
||||||
|
|
||||||
|
const toast = useToastStore(),
|
||||||
|
loading = ref(false),
|
||||||
|
items = ref<Record<string, any>[]>([]),
|
||||||
|
total = ref(0),
|
||||||
|
selected = ref<Record<string, any>[]>([]),
|
||||||
|
dialog = ref<"single" | "batch" | "">(""),
|
||||||
|
pointGrantOpen = ref(false),
|
||||||
|
pointGrantTarget = ref<Record<string, any>>(),
|
||||||
|
grantingPoints = ref(false),
|
||||||
|
editingLimitId = ref(""),
|
||||||
|
editingLimitValue = ref<number | null>(null),
|
||||||
|
savingLimitId = ref("");
|
||||||
|
const query = reactive({ keyword: "", enabled: "", page: 1, page_size: 15 });
|
||||||
|
const form = reactive({
|
||||||
|
account: "",
|
||||||
|
prefix: "",
|
||||||
|
start_sequence: "001",
|
||||||
|
end_sequence: "001",
|
||||||
|
password: "",
|
||||||
|
daily_limit: 9999,
|
||||||
|
});
|
||||||
|
const pointGrantForm = reactive({ points: null as number | null, reason: "", request_id: "" });
|
||||||
|
const formRef = ref<FormInstance>();
|
||||||
|
const dialogOpen = computed({
|
||||||
|
get: () => dialog.value !== "",
|
||||||
|
set: (value: boolean) => {
|
||||||
|
if (!value) dialog.value = "";
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const canGrantPoints = computed(
|
||||||
|
() =>
|
||||||
|
Number.isFinite(Number(pointGrantForm.points)) &&
|
||||||
|
Number(pointGrantForm.points) > 0 &&
|
||||||
|
Boolean(pointGrantForm.reason.trim()),
|
||||||
|
);
|
||||||
|
const batchPreview = computed(() => {
|
||||||
|
const prefix = form.prefix.trim();
|
||||||
|
const startRaw = form.start_sequence.trim();
|
||||||
|
const endRaw = form.end_sequence.trim();
|
||||||
|
if (!prefix) return { error: "请输入账号前缀", count: 0, first: "", last: "" };
|
||||||
|
if (!/^\d+$/.test(startRaw) || !/^\d+$/.test(endRaw)) {
|
||||||
|
return { error: "起始序号和终止序号必须为数字", count: 0, first: "", last: "" };
|
||||||
|
}
|
||||||
|
const start = Number(startRaw);
|
||||||
|
const end = Number(endRaw);
|
||||||
|
if (!Number.isSafeInteger(start) || !Number.isSafeInteger(end)) {
|
||||||
|
return { error: "序号超出有效范围", count: 0, first: "", last: "" };
|
||||||
|
}
|
||||||
|
if (start > end) return { error: "起始序号不能大于终止序号", count: 0, first: "", last: "" };
|
||||||
|
const count = end - start + 1;
|
||||||
|
if (count > 50) return { error: "每次最多批量创建 50 个用户", count, first: "", last: "" };
|
||||||
|
const width = Math.max(startRaw.length, endRaw.length);
|
||||||
|
return {
|
||||||
|
error: "",
|
||||||
|
count,
|
||||||
|
first: `${prefix}-${String(start).padStart(width, "0")}`,
|
||||||
|
last: `${prefix}-${String(end).padStart(width, "0")}`,
|
||||||
|
};
|
||||||
|
});
|
||||||
|
async function load(silent = false) {
|
||||||
|
if (silent && editingLimitId.value) return;
|
||||||
|
if (!silent) loading.value = true;
|
||||||
|
try {
|
||||||
|
const { data } = await http.get("/users", { params: query });
|
||||||
|
items.value = data.data.items;
|
||||||
|
total.value = data.data.total;
|
||||||
|
} catch (e) {
|
||||||
|
showError(e);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function editDailyLimit(row: Record<string, any>) {
|
||||||
|
editingLimitId.value = row.id;
|
||||||
|
editingLimitValue.value = Number(row.daily_limit);
|
||||||
|
}
|
||||||
|
function cancelDailyLimitEdit() {
|
||||||
|
editingLimitId.value = "";
|
||||||
|
editingLimitValue.value = null;
|
||||||
|
}
|
||||||
|
function blurLimitInput(event: KeyboardEvent) {
|
||||||
|
(event.target as HTMLInputElement).blur();
|
||||||
|
}
|
||||||
|
async function saveDailyLimit(row: Record<string, any>) {
|
||||||
|
if (editingLimitId.value !== row.id) return;
|
||||||
|
const value = Number(editingLimitValue.value);
|
||||||
|
if (!Number.isFinite(value) || value < 0) {
|
||||||
|
toast.warning("每日限额不能小于 0");
|
||||||
|
cancelDailyLimitEdit();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (value === Number(row.daily_limit)) {
|
||||||
|
cancelDailyLimitEdit();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
savingLimitId.value = row.id;
|
||||||
|
try {
|
||||||
|
await http.patch("/users/batch", {
|
||||||
|
ids: [row.id],
|
||||||
|
daily_limit: value.toFixed(2),
|
||||||
|
reason: "管理员原地修改每日限额",
|
||||||
|
});
|
||||||
|
row.daily_limit = value;
|
||||||
|
toast.success("每日限额已保存");
|
||||||
|
} catch (e) {
|
||||||
|
showError(e);
|
||||||
|
} finally {
|
||||||
|
savingLimitId.value = "";
|
||||||
|
cancelDailyLimitEdit();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function open(type: "single" | "batch") {
|
||||||
|
Object.assign(form, {
|
||||||
|
account: "",
|
||||||
|
prefix: "",
|
||||||
|
start_sequence: "001",
|
||||||
|
end_sequence: "001",
|
||||||
|
password: "",
|
||||||
|
daily_limit: 9999,
|
||||||
|
});
|
||||||
|
dialog.value = type;
|
||||||
|
}
|
||||||
|
async function create() {
|
||||||
|
if (!(await formRef.value?.validate().catch(() => false))) return;
|
||||||
|
if (dialog.value === "batch" && batchPreview.value.error) {
|
||||||
|
toast.warning(batchPreview.value.error);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
if (dialog.value === "single")
|
||||||
|
await http.post("/users", {
|
||||||
|
account: form.account,
|
||||||
|
password: form.password,
|
||||||
|
daily_limit: form.daily_limit,
|
||||||
|
});
|
||||||
|
else
|
||||||
|
await http.post("/users/batch", {
|
||||||
|
prefix: form.prefix,
|
||||||
|
start_sequence: form.start_sequence,
|
||||||
|
end_sequence: form.end_sequence,
|
||||||
|
password: form.password,
|
||||||
|
daily_limit: form.daily_limit,
|
||||||
|
});
|
||||||
|
toast.success("用户创建成功");
|
||||||
|
dialog.value = "";
|
||||||
|
load();
|
||||||
|
} catch (e) {
|
||||||
|
showError(e);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function toggle(row: Record<string, any>, enabled: boolean) {
|
||||||
|
try {
|
||||||
|
await ElMessageBox.confirm(
|
||||||
|
`${enabled ? "启用" : "停用"}用户 ${row.account}（${row.username}）？停用后会撤销其有效会话。`,
|
||||||
|
"确认状态变更",
|
||||||
|
{ type: "warning" },
|
||||||
|
);
|
||||||
|
await http.patch("/users/batch", {
|
||||||
|
ids: [row.id],
|
||||||
|
enabled,
|
||||||
|
reason: `管理员${enabled ? "启用" : "停用"}用户`,
|
||||||
|
});
|
||||||
|
row.enabled = enabled;
|
||||||
|
toast.success("用户状态已更新");
|
||||||
|
} catch (e) {
|
||||||
|
row.enabled = !enabled;
|
||||||
|
if (e !== "cancel" && e !== "close") showError(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function adjustLimit() {
|
||||||
|
try {
|
||||||
|
const { value } = await ElMessageBox.prompt(
|
||||||
|
"请输入新的每日限额（0 表示禁止消耗，9999 表示不限制）",
|
||||||
|
"批量调整每日限额",
|
||||||
|
{
|
||||||
|
inputType: "text",
|
||||||
|
inputValue: "9999",
|
||||||
|
inputValidator: (v) => Number(v) >= 0 || "限额不能小于 0",
|
||||||
|
},
|
||||||
|
);
|
||||||
|
const reason = await ElMessageBox.prompt("请填写调整原因", "填写审计原因", {
|
||||||
|
inputValidator: (v) => Boolean(v.trim()) || "调整原因不能为空",
|
||||||
|
});
|
||||||
|
await http.patch("/users/batch", {
|
||||||
|
ids: selected.value.map((i) => i.id),
|
||||||
|
daily_limit: Number(value).toFixed(2),
|
||||||
|
reason: reason.value,
|
||||||
|
});
|
||||||
|
toast.success("每日限额已调整");
|
||||||
|
selected.value = [];
|
||||||
|
load();
|
||||||
|
} catch (e) {
|
||||||
|
if (e !== "cancel" && e !== "close") showError(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/** 打开单用户积分发放弹窗，并为本次操作生成幂等请求号。 */
|
||||||
|
function openPointGrant(row: Record<string, any>) {
|
||||||
|
pointGrantTarget.value = row;
|
||||||
|
Object.assign(pointGrantForm, { points: null, reason: "", request_id: crypto.randomUUID() });
|
||||||
|
pointGrantOpen.value = true;
|
||||||
|
}
|
||||||
|
/** 校验正数积分并通过账务接口完成入账，成功后同步列表余额。 */
|
||||||
|
async function grantPoints() {
|
||||||
|
if (!pointGrantTarget.value || !canGrantPoints.value) return;
|
||||||
|
grantingPoints.value = true;
|
||||||
|
try {
|
||||||
|
const points = Number(pointGrantForm.points).toFixed(2);
|
||||||
|
const { data } = await http.post(`/users/${pointGrantTarget.value.id}/points`, {
|
||||||
|
request_id: pointGrantForm.request_id,
|
||||||
|
points,
|
||||||
|
reason: pointGrantForm.reason.trim(),
|
||||||
|
});
|
||||||
|
pointGrantTarget.value.point_balance = data.data.balance;
|
||||||
|
toast.success(`已增加 ${points} 积分`);
|
||||||
|
pointGrantOpen.value = false;
|
||||||
|
} catch (e) {
|
||||||
|
showError(e);
|
||||||
|
} finally {
|
||||||
|
grantingPoints.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function remove(ids: string[]) {
|
||||||
|
try {
|
||||||
|
const reason = await ElMessageBox.prompt(
|
||||||
|
`确认软删除 ${ids.length} 个用户？其订单、流水和使用记录将保留。`,
|
||||||
|
"确认删除",
|
||||||
|
{
|
||||||
|
type: "warning",
|
||||||
|
inputPlaceholder: "请输入删除原因",
|
||||||
|
inputValidator: (v) => Boolean(v.trim()) || "删除原因不能为空",
|
||||||
|
confirmButtonText: "确认删除",
|
||||||
|
},
|
||||||
|
);
|
||||||
|
await http.delete("/users/batch", { data: { ids, reason: reason.value } });
|
||||||
|
toast.success("用户已删除");
|
||||||
|
selected.value = [];
|
||||||
|
load();
|
||||||
|
} catch (e) {
|
||||||
|
if (e !== "cancel" && e !== "close") showError(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const fmt = (v: any) =>
|
||||||
|
v
|
||||||
|
? new Date(v).toLocaleString("zh-CN", {
|
||||||
|
timeZone: "Asia/Shanghai",
|
||||||
|
hour12: false,
|
||||||
|
})
|
||||||
|
: "从未在线";
|
||||||
|
function showError(e: unknown) {
|
||||||
|
const d = apiError(e);
|
||||||
|
toast.error(d.message, d.copyText);
|
||||||
|
}
|
||||||
|
const timer = window.setInterval(() => load(true), 300000);
|
||||||
|
onMounted(load);
|
||||||
|
onBeforeUnmount(() => clearInterval(timer));
|
||||||
|
</script>
|
||||||
|
<template>
|
||||||
|
<div class="page-stack management-page">
|
||||||
|
<PageHeader
|
||||||
|
title="用户管理"
|
||||||
|
description="管理 WEB 用户账号、每日限额和启用状态"
|
||||||
|
><el-button @click="open('batch')"> <AppIcon name="UserFilled" />批量新建 </el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
@click="open('single')"
|
||||||
|
>
|
||||||
|
<AppIcon name="Plus" />新建用户
|
||||||
|
</el-button></PageHeader
|
||||||
|
>
|
||||||
|
<section class="panel filter-bar">
|
||||||
|
<el-input
|
||||||
|
v-model="query.keyword"
|
||||||
|
clearable
|
||||||
|
placeholder="账号、用户名或 UID"
|
||||||
|
/><el-select
|
||||||
|
v-model="query.enabled"
|
||||||
|
clearable
|
||||||
|
placeholder="启用状态"
|
||||||
|
><el-option
|
||||||
|
label="已启用"
|
||||||
|
value="true" /><el-option
|
||||||
|
label="已停用"
|
||||||
|
value="false" /></el-select
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
@click="
|
||||||
|
query.page = 1;
|
||||||
|
load();
|
||||||
|
"
|
||||||
|
>查询</el-button
|
||||||
|
><el-button
|
||||||
|
@click="
|
||||||
|
query.keyword = '';
|
||||||
|
query.enabled = '';
|
||||||
|
query.page = 1;
|
||||||
|
load();
|
||||||
|
"
|
||||||
|
>重置</el-button
|
||||||
|
>
|
||||||
|
</section>
|
||||||
|
<section
|
||||||
|
v-if="selected.length"
|
||||||
|
class="batch-toolbar"
|
||||||
|
>
|
||||||
|
<strong>已选择 {{ selected.length }} 项</strong><el-button @click="selected = []">取消选择</el-button
|
||||||
|
><el-button @click="adjustLimit">调整每日限额</el-button
|
||||||
|
><el-button
|
||||||
|
type="danger"
|
||||||
|
@click="remove(selected.map((i) => i.id))"
|
||||||
|
>批量删除</el-button
|
||||||
|
>
|
||||||
|
</section>
|
||||||
|
<section
|
||||||
|
class="panel table-panel"
|
||||||
|
v-loading="loading"
|
||||||
|
>
|
||||||
|
<el-table
|
||||||
|
class="desktop-user-table"
|
||||||
|
:data="items"
|
||||||
|
@selection-change="selected = $event"
|
||||||
|
><el-table-column
|
||||||
|
type="selection"
|
||||||
|
width="46"
|
||||||
|
/><el-table-column
|
||||||
|
prop="account"
|
||||||
|
label="账号"
|
||||||
|
min-width="150"
|
||||||
|
show-overflow-tooltip
|
||||||
|
/><el-table-column
|
||||||
|
prop="username"
|
||||||
|
label="用户名"
|
||||||
|
min-width="170"
|
||||||
|
/><el-table-column
|
||||||
|
prop="uid"
|
||||||
|
label="UID"
|
||||||
|
min-width="125"
|
||||||
|
/><el-table-column
|
||||||
|
label="积分余额"
|
||||||
|
min-width="120"
|
||||||
|
><template #default="{ row }">{{ Number(row.point_balance).toFixed(2) }}</template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="今日消耗"
|
||||||
|
min-width="120"
|
||||||
|
><template #default="{ row }">{{ Number(row.today_consumption).toFixed(2) }}</template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="每日限额"
|
||||||
|
min-width="170"
|
||||||
|
><template #default="{ row }">
|
||||||
|
<div class="limit-cell">
|
||||||
|
<AdminNumberInput
|
||||||
|
v-if="editingLimitId === row.id"
|
||||||
|
v-model="editingLimitValue"
|
||||||
|
class="limit-input"
|
||||||
|
:min="0"
|
||||||
|
:precision="2"
|
||||||
|
:disabled="savingLimitId === row.id"
|
||||||
|
autofocus
|
||||||
|
@blur="saveDailyLimit(row)"
|
||||||
|
/><template v-else
|
||||||
|
><span>{{ Number(row.daily_limit) === 9999 ? "不限制" : Number(row.daily_limit).toFixed(2) }}</span
|
||||||
|
><el-button
|
||||||
|
class="limit-edit-button"
|
||||||
|
text
|
||||||
|
circle
|
||||||
|
type="primary"
|
||||||
|
title="编辑每日限额"
|
||||||
|
@click="editDailyLimit(row)"
|
||||||
|
>
|
||||||
|
<AppIcon
|
||||||
|
name="Edit"
|
||||||
|
:size="15"
|
||||||
|
/> </el-button
|
||||||
|
></template>
|
||||||
|
</div> </template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="启用状态"
|
||||||
|
width="96"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-switch
|
||||||
|
:model-value="row.enabled"
|
||||||
|
@change="toggle(row, Boolean($event))" /></template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="最后在线时间"
|
||||||
|
min-width="190"
|
||||||
|
><template #default="{ row }">{{ fmt(row.last_online_at) }}</template></el-table-column
|
||||||
|
><el-table-column
|
||||||
|
label="操作"
|
||||||
|
width="150"
|
||||||
|
><template #default="{ row }"
|
||||||
|
><el-button
|
||||||
|
link
|
||||||
|
type="primary"
|
||||||
|
:disabled="!row.enabled"
|
||||||
|
:title="row.enabled ? '增加积分' : '停用用户不可增加积分'"
|
||||||
|
@click="openPointGrant(row)"
|
||||||
|
>增加积分</el-button
|
||||||
|
><el-button
|
||||||
|
link
|
||||||
|
type="danger"
|
||||||
|
@click="remove([row.id])"
|
||||||
|
>删除</el-button
|
||||||
|
></template
|
||||||
|
></el-table-column
|
||||||
|
></el-table
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
v-if="items.length"
|
||||||
|
class="mobile-card-list"
|
||||||
|
>
|
||||||
|
<article
|
||||||
|
v-for="row in items"
|
||||||
|
:key="row.id"
|
||||||
|
class="data-card"
|
||||||
|
>
|
||||||
|
<dl>
|
||||||
|
<dt>账号</dt>
|
||||||
|
<dd>{{ row.account }}</dd>
|
||||||
|
<dt>用户名</dt>
|
||||||
|
<dd>{{ row.username }}</dd>
|
||||||
|
<dt>UID</dt>
|
||||||
|
<dd>{{ row.uid }}</dd>
|
||||||
|
<dt>积分余额</dt>
|
||||||
|
<dd>{{ Number(row.point_balance).toFixed(2) }}</dd>
|
||||||
|
<dt>今日消耗</dt>
|
||||||
|
<dd>{{ Number(row.today_consumption).toFixed(2) }}</dd>
|
||||||
|
<dt>每日限额</dt>
|
||||||
|
<dd>
|
||||||
|
<div class="limit-cell">
|
||||||
|
<AdminNumberInput
|
||||||
|
v-if="editingLimitId === row.id"
|
||||||
|
v-model="editingLimitValue"
|
||||||
|
class="limit-input"
|
||||||
|
:min="0"
|
||||||
|
:precision="2"
|
||||||
|
:disabled="savingLimitId === row.id"
|
||||||
|
autofocus
|
||||||
|
@blur="saveDailyLimit(row)"
|
||||||
|
/><template v-else
|
||||||
|
><span>{{ Number(row.daily_limit) === 9999 ? "不限制" : Number(row.daily_limit).toFixed(2) }}</span
|
||||||
|
><el-button
|
||||||
|
class="limit-edit-button"
|
||||||
|
text
|
||||||
|
circle
|
||||||
|
type="primary"
|
||||||
|
title="编辑每日限额"
|
||||||
|
@click="editDailyLimit(row)"
|
||||||
|
>
|
||||||
|
<AppIcon
|
||||||
|
name="Edit"
|
||||||
|
:size="15"
|
||||||
|
/> </el-button
|
||||||
|
></template>
|
||||||
|
</div>
|
||||||
|
</dd>
|
||||||
|
<dt>最后在线</dt>
|
||||||
|
<dd>{{ fmt(row.last_online_at) }}</dd>
|
||||||
|
</dl>
|
||||||
|
<div class="card-actions">
|
||||||
|
<el-switch
|
||||||
|
:model-value="row.enabled"
|
||||||
|
@change="toggle(row, Boolean($event))"
|
||||||
|
/>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
plain
|
||||||
|
:disabled="!row.enabled"
|
||||||
|
@click="openPointGrant(row)"
|
||||||
|
>增加积分</el-button
|
||||||
|
>
|
||||||
|
<el-button
|
||||||
|
type="danger"
|
||||||
|
plain
|
||||||
|
@click="remove([row.id])"
|
||||||
|
>删除</el-button
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</article>
|
||||||
|
</div>
|
||||||
|
<el-empty
|
||||||
|
v-if="!loading && !items.length"
|
||||||
|
description="暂无用户"
|
||||||
|
/><el-pagination
|
||||||
|
v-if="total > query.page_size"
|
||||||
|
v-model:current-page="query.page"
|
||||||
|
background
|
||||||
|
layout="total,prev,pager,next"
|
||||||
|
:total="total"
|
||||||
|
@current-change="load"
|
||||||
|
/>
|
||||||
|
</section>
|
||||||
|
<el-dialog
|
||||||
|
v-model="pointGrantOpen"
|
||||||
|
title="增加用户积分"
|
||||||
|
width="min(480px, calc(100vw - 32px))"
|
||||||
|
:close-on-click-modal="!grantingPoints"
|
||||||
|
:show-close="!grantingPoints"
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
v-if="pointGrantTarget"
|
||||||
|
class="point-grant-summary"
|
||||||
|
>
|
||||||
|
<strong>{{ pointGrantTarget.account }}（{{ pointGrantTarget.username }}）</strong>
|
||||||
|
<span>当前余额 {{ Number(pointGrantTarget.point_balance).toFixed(2) }} 积分</span>
|
||||||
|
</div>
|
||||||
|
<el-form label-position="top">
|
||||||
|
<el-form-item label="增加积分">
|
||||||
|
<AdminNumberInput
|
||||||
|
v-model="pointGrantForm.points"
|
||||||
|
:min="0.01"
|
||||||
|
:precision="2"
|
||||||
|
:disabled="grantingPoints"
|
||||||
|
placeholder="请输入大于 0 的积分数量"
|
||||||
|
/>
|
||||||
|
<div class="field-helper">仅支持增加积分，到账后会写入用户积分变动记录</div>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item label="操作原因">
|
||||||
|
<el-input
|
||||||
|
v-model.trim="pointGrantForm.reason"
|
||||||
|
maxlength="200"
|
||||||
|
show-word-limit
|
||||||
|
:disabled="grantingPoints"
|
||||||
|
placeholder="用于管理后台审计，不会向用户展示"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
</el-form>
|
||||||
|
<template #footer>
|
||||||
|
<el-button
|
||||||
|
:disabled="grantingPoints"
|
||||||
|
@click="pointGrantOpen = false"
|
||||||
|
>取消</el-button
|
||||||
|
>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="grantingPoints"
|
||||||
|
:disabled="!canGrantPoints"
|
||||||
|
@click="grantPoints"
|
||||||
|
>确认增加</el-button
|
||||||
|
>
|
||||||
|
</template>
|
||||||
|
</el-dialog>
|
||||||
|
<el-dialog
|
||||||
|
v-model="dialogOpen"
|
||||||
|
:title="dialog === 'single' ? '新建用户' : '批量新建用户'"
|
||||||
|
width="min(560px, calc(100vw - 32px))"
|
||||||
|
><el-form
|
||||||
|
ref="formRef"
|
||||||
|
:model="form"
|
||||||
|
label-position="top"
|
||||||
|
><el-form-item
|
||||||
|
v-if="dialog === 'single'"
|
||||||
|
label="账号"
|
||||||
|
prop="account"
|
||||||
|
:rules="[{ required: true, message: '请输入账号' }]"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model.trim="form.account"
|
||||||
|
placeholder="用于用户登录，创建后不可修改"
|
||||||
|
/>
|
||||||
|
<div class="field-helper">系统将自动生成随机用户名，账号创建后不可修改</div>
|
||||||
|
</el-form-item>
|
||||||
|
<template v-else>
|
||||||
|
<div class="dialog-form-grid batch-account-grid">
|
||||||
|
<el-form-item
|
||||||
|
class="dialog-field--full"
|
||||||
|
label="账号前缀"
|
||||||
|
prop="prefix"
|
||||||
|
:rules="[{ required: true, message: '请输入账号前缀' }]"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model.trim="form.prefix"
|
||||||
|
placeholder="例如 RR"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="起始序号"
|
||||||
|
prop="start_sequence"
|
||||||
|
:rules="[
|
||||||
|
{ required: true, message: '请输入起始序号' },
|
||||||
|
{ pattern: /^\d+$/, message: '起始序号必须为数字' },
|
||||||
|
]"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model.trim="form.start_sequence"
|
||||||
|
placeholder="例如 001"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item
|
||||||
|
label="终止序号"
|
||||||
|
prop="end_sequence"
|
||||||
|
:rules="[
|
||||||
|
{ required: true, message: '请输入终止序号' },
|
||||||
|
{ pattern: /^\d+$/, message: '终止序号必须为数字' },
|
||||||
|
]"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
v-model.trim="form.end_sequence"
|
||||||
|
placeholder="例如 050"
|
||||||
|
/>
|
||||||
|
</el-form-item>
|
||||||
|
</div>
|
||||||
|
<el-alert
|
||||||
|
class="batch-preview"
|
||||||
|
:closable="false"
|
||||||
|
:type="batchPreview.error ? 'error' : 'info'"
|
||||||
|
:title="
|
||||||
|
batchPreview.error
|
||||||
|
? batchPreview.error
|
||||||
|
: `将创建 ${batchPreview.count} 个账号：${batchPreview.first}${batchPreview.count > 1 ? ` 至 ${batchPreview.last}` : ''}`
|
||||||
|
"
|
||||||
|
/>
|
||||||
|
<div class="field-helper batch-helper">
|
||||||
|
账号之间使用短横线分隔；序号会保留输入位数，每次最多创建 50 个用户
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
<el-form-item
|
||||||
|
label="密码"
|
||||||
|
prop="password"
|
||||||
|
:rules="[{ required: true, min: 8, message: '密码至少 8 个字符' }]"
|
||||||
|
><el-input
|
||||||
|
v-model="form.password"
|
||||||
|
type="password"
|
||||||
|
show-password
|
||||||
|
autocomplete="new-password" /></el-form-item
|
||||||
|
><el-form-item label="每日限额"
|
||||||
|
><AdminNumberInput
|
||||||
|
v-model="form.daily_limit"
|
||||||
|
:min="0"
|
||||||
|
:precision="2"
|
||||||
|
/>
|
||||||
|
<div class="field-helper">0 表示当日禁止消耗，9999 表示不限制</div>
|
||||||
|
</el-form-item></el-form
|
||||||
|
><template #footer
|
||||||
|
><el-button @click="dialog = ''">取消</el-button
|
||||||
|
><el-button
|
||||||
|
type="primary"
|
||||||
|
:loading="loading"
|
||||||
|
:disabled="dialog === 'batch' && Boolean(batchPreview.error)"
|
||||||
|
@click="create"
|
||||||
|
>创建</el-button
|
||||||
|
></template
|
||||||
|
></el-dialog
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.desktop-user-table {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview {
|
||||||
|
margin-bottom: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-helper {
|
||||||
|
margin-bottom: 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-cell {
|
||||||
|
min-width: 0;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-cell > span {
|
||||||
|
min-width: 58px;
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-cell .el-button {
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-edit-button {
|
||||||
|
width: 28px;
|
||||||
|
height: 28px;
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-input {
|
||||||
|
width: 118px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-input :deep(input) {
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
.point-grant-summary {
|
||||||
|
margin-bottom: var(--space-md);
|
||||||
|
padding: var(--space-sm) var(--space-md);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: var(--space-xs);
|
||||||
|
color: var(--color-text);
|
||||||
|
background: var(--color-bg);
|
||||||
|
border: 1px solid var(--color-border);
|
||||||
|
border-radius: var(--radius-sm);
|
||||||
|
}
|
||||||
|
|
||||||
|
.point-grant-summary span {
|
||||||
|
color: var(--color-text-secondary);
|
||||||
|
font-size: var(--font-size-sm);
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (max-width: 767px) {
|
||||||
|
.limit-cell {
|
||||||
|
justify-content: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.limit-input {
|
||||||
|
width: min(128px, 100%);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
{
|
||||||
|
"extends": "@vue/tsconfig/tsconfig.dom.json",
|
||||||
|
"include": ["env.d.ts", "src/**/*", "src/**/*.vue"],
|
||||||
|
"exclude": ["src/**/__tests__/*"],
|
||||||
|
"compilerOptions": {
|
||||||
|
"composite": true,
|
||||||
|
"tsBuildInfoFile": "./node_modules/.tmp/tsconfig.app.tsbuildinfo",
|
||||||
|
"baseUrl": ".",
|
||||||
|
"paths": {
|
||||||
|
"@/*": ["./src/*"]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
{
|
||||||
|
"files": [],
|
||||||
|
"references": [
|
||||||
|
{
|
||||||
|
"path": "./tsconfig.node.json"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"path": "./tsconfig.app.json"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
{
|
||||||
|
"include": ["vite.config.ts"],
|
||||||
|
"compilerOptions": {
|
||||||
|
"composite": true,
|
||||||
|
"noEmit": true,
|
||||||
|
"module": "ESNext",
|
||||||
|
"moduleResolution": "bundler",
|
||||||
|
"target": "ESNext",
|
||||||
|
"strict": true,
|
||||||
|
"esModuleInterop": true,
|
||||||
|
"skipLibCheck": true,
|
||||||
|
"types": ["node"]
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
// 剧游AI管理后台 Vite 构建配置，配置 Vue 插件、路径别名、代理与输出目录
|
||||||
|
|
||||||
|
import { fileURLToPath, URL } from 'node:url'
|
||||||
|
|
||||||
|
import { defineConfig } from 'vite'
|
||||||
|
import vue from '@vitejs/plugin-vue'
|
||||||
|
import { findAvailablePort } from '../vite-port'
|
||||||
|
|
||||||
|
const devHost = '127.0.0.1'
|
||||||
|
const devPort = await findAvailablePort(devHost, 5500)
|
||||||
|
|
||||||
|
export default defineConfig({
|
||||||
|
base: '/admin/',
|
||||||
|
plugins: [vue()],
|
||||||
|
resolve: {
|
||||||
|
alias: {
|
||||||
|
'@': fileURLToPath(new URL('./src', import.meta.url)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
css: {
|
||||||
|
preprocessorOptions: {
|
||||||
|
scss: {
|
||||||
|
api: 'modern-compiler',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
build: {
|
||||||
|
outDir: '../API/static/admin',
|
||||||
|
emptyOutDir: true,
|
||||||
|
},
|
||||||
|
server: {
|
||||||
|
host: devHost,
|
||||||
|
port: devPort,
|
||||||
|
open: false,
|
||||||
|
proxy: {
|
||||||
|
'/api': {
|
||||||
|
target: 'http://localhost:8123',
|
||||||
|
changeOrigin: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
@@ -0,0 +1,226 @@
|
|||||||
|
# AGENTS.md
|
||||||
|
|
||||||
|
<INSTRUCTIONS>
|
||||||
|
|
||||||
|
## 一、基本原则
|
||||||
|
|
||||||
|
1. 回答问题时避免过分夸赞。用户和 AI 的判断都可能不准确，应结合代码、日志、测试结果和现有实现反复核对，优先保证准确性。
|
||||||
|
|
||||||
|
2. 回答应保持结构化、条理清晰，明确区分：
|
||||||
|
- 已确认的事实
|
||||||
|
- 基于现有信息的推断
|
||||||
|
- 尚未确认的问题
|
||||||
|
- 实际修改和验证结果
|
||||||
|
|
||||||
|
3. 如果缺少的信息会显著影响实现方向、安全性或兼容性，应主动向用户索要补充信息或证据。对于不影响整体方向的小问题，可以结合项目现状作出合理假设，但必须说明假设内容。
|
||||||
|
|
||||||
|
4. 开始修改前，必须先阅读相关代码、项目结构、配置和调用链，确认问题根因及现有实现。禁止在未了解上下文的情况下直接创建一套平行实现。
|
||||||
|
|
||||||
|
5. 遵循最小改动原则：
|
||||||
|
- 只修改完成当前需求或解决根因所必需的代码。
|
||||||
|
- 禁止顺带重构无关代码。
|
||||||
|
- 禁止擅自升级依赖。
|
||||||
|
- 禁止调整无关代码格式。
|
||||||
|
- 禁止改变无关功能和既有行为。
|
||||||
|
- 禁止扩大用户未授权的修改范围。
|
||||||
|
|
||||||
|
6. 优先复用项目已有代码、组件、工具函数、服务和基础设施。相同逻辑、算法、功能、页面或样式应抽取为公共实现，禁止在多个位置重复实现，除非存在明确的独立实现理由。
|
||||||
|
|
||||||
|
7. 修改代码时，应保持现有接口、参数、返回结构、错误语义和业务行为兼容。确实需要破坏兼容性时，必须提前说明原因和影响范围，用户确认后才能实施。
|
||||||
|
|
||||||
|
8. 默认仅提供分析、解释、诊断、方案或修改建议。除非用户明确要求实施、修改、修复、创建、删除或重构，否则禁止主动更改任何代码、配置、数据库、依赖、文件或外部系统状态。
|
||||||
|
- “分析一下”“检查一下”“看看问题”“给出方案”“怎么实现”等表述，仅授权只读检查和提供建议，不代表授权实施修改。
|
||||||
|
- “帮我修改”“直接修复”“实现这个功能”“按方案执行”“开干”等明确表述，才视为授权实施。
|
||||||
|
- 获得实施授权后，只能修改当前需求明确涉及的范围，并遵循最小改动原则。
|
||||||
|
- 如果用户的表述无法判断是否授权修改，应先询问用户，不得自行实施。
|
||||||
|
- 读取文件、搜索代码、查看日志、运行不改变数据和外部状态的诊断命令，不视为代码修改。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 二、文件与代码组织
|
||||||
|
|
||||||
|
1. 每个文件应具有明确职责，避免将路由、参数处理、业务逻辑、数据库操作和第三方服务调用全部堆放在同一个文件中。
|
||||||
|
|
||||||
|
2. 代码文件接近或超过 800 行时，应检查是否需要按照功能、业务领域、页面或组件职责进行拆分。
|
||||||
|
|
||||||
|
3. 800 行不是强制凑齐的目标：
|
||||||
|
- 800 行是一个参考值，根据项目需求和代码复杂度，可以适当调整。
|
||||||
|
- 超出 800 行的文件，应检查是否需要按照功能、业务领域、页面或组件职责进行拆分，不得盲目拆分。
|
||||||
|
- 禁止为了减少行数而将多行代码压缩成一行。
|
||||||
|
- 禁止为了形式上的拆分创建大量缺乏独立职责的小文件。
|
||||||
|
- 文件是否拆分应以职责边界、可维护性和复用价值为依据。
|
||||||
|
|
||||||
|
4. 新增公共能力前，必须先搜索项目中是否已经存在相同或相近实现。
|
||||||
|
|
||||||
|
5. 禁止使用 PowerShell、Python、重定向符或其他临时脚本拼接、生成或覆盖源代码文件。修改已有文件时，在原文件编码受支持且修改范围适合的情况下，应优先使用精确补丁，避免为少量改动重新写入整个文件；如果补丁方式不适用，应采用能够保持原文件编码、BOM 状态和换行格式的安全编辑方式。
|
||||||
|
|
||||||
|
6. 文件、模块、函数及数据库定义必须使用中文注释说明其用途：
|
||||||
|
- 每个支持注释的代码文件顶部，必须使用中文注释简要说明该文件的作用和主要职责。
|
||||||
|
- 每个模块、类、接口、组件和函数顶部，必须使用中文注释简要说明其用途；存在重要参数、返回值、副作用、异常或使用限制时，应一并说明。
|
||||||
|
- 数据库定义文件中，每个模型、表和字段都必须使用中文注释说明其业务含义；对于枚举值、关联关系、默认值、索引和约束，也应说明其用途。
|
||||||
|
- 修改代码功能、参数或字段含义时，必须同步更新对应注释，禁止保留与实际实现不一致的注释。
|
||||||
|
- 注释应说明职责、业务含义或设计原因，禁止仅将代码名称直译成没有实际信息的注释。
|
||||||
|
- JSON 等语法本身不支持注释的文件，不得为了添加注释而使用非标准语法或破坏文件有效性。
|
||||||
|
- 第三方依赖、自动生成文件、构建产物及无法安全添加注释的文件不受此规则约束，除非用户明确要求修改。
|
||||||
|
|
||||||
|
7. 必须保护文件编码和换行格式，禁止因读写方式不一致造成中文乱码：
|
||||||
|
- 修改文件前，应识别并保持原文件的字符编码、BOM 状态和换行格式。
|
||||||
|
- 修改已有文件时，禁止擅自将 UTF-8、UTF-8 with BOM、GBK、UTF-16 等编码相互转换。
|
||||||
|
- 新建文本文件默认使用 UTF-8 编码；如果项目已有明确编码规范，则遵循项目现有规范。
|
||||||
|
- 禁止通过读取全文后重新写入的方式完成少量修改。
|
||||||
|
- 修改后必须检查本次变更中的中文内容是否正常，重点检查是否出现 `�`、异常问号或类似“涓枃”的乱码。
|
||||||
|
- 如果无法确定文件原始编码，应停止写入并先说明情况，不得猜测编码后直接覆盖。
|
||||||
|
- 禁止为了统一编码而批量转换无关文件；确需转换时，必须获得用户明确授权。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 三、后端模块化与解耦
|
||||||
|
|
||||||
|
1. 后端应按照业务领域和职责进行模块化设计，实现关注点分离和低耦合。
|
||||||
|
|
||||||
|
2. 上传、删除、查询、转换、通知等不同业务能力，应根据实际职责划分为独立模块。禁止仅通过复制文件或移动代码实现表面上的模块化。
|
||||||
|
|
||||||
|
3. 每个后端业务模块必须使用独立目录组织。即使该业务模块当前只有一个功能或一个实现文件，也必须为其创建单独目录，禁止将不同业务模块的实现文件直接平铺在公共目录中。
|
||||||
|
|
||||||
|
- 本规则所称“业务模块”，是指具有独立业务职责、领域边界或外部服务集成边界的功能，例如用户、订单、支付、文件管理、AI 中转站和对象存储。
|
||||||
|
- 普通工具函数、常量、类型定义、内部辅助类和仅服务于某个模块的实现，不视为独立业务模块，不要求分别创建目录。
|
||||||
|
- 模块相关的业务逻辑、数据访问、类型、配置和外部服务适配代码，应集中放置在该模块目录中。
|
||||||
|
- 模块只有一个简单功能时，可以只包含一个实现文件；不得为了填充目录而创建没有实际职责的文件。
|
||||||
|
- 模块包含多个职责明确的子功能时，应按照子功能分别创建文件；子功能不需要建立子目录。
|
||||||
|
- 基础 CRUD 操作通常是同一业务实体的数据访问方法，不得仅按照新增、查询、修改和删除机械拆分为四个模块。
|
||||||
|
- 不强制为每个模块创建 `index`、`types`、`service` 等固定文件，只有存在实际职责时才创建。
|
||||||
|
- 新增模块以及本次需求实际修改的模块必须遵守本规则；禁止为了统一目录形式而擅自重构与当前需求无关的既有模块。
|
||||||
|
- 目录和文件命名应遵循项目现有命名规范。
|
||||||
|
|
||||||
|
4. 模块之间应通过明确的公开接口进行调用，包括：
|
||||||
|
- 函数参数
|
||||||
|
- 返回值
|
||||||
|
- 类型或接口
|
||||||
|
- 服务抽象
|
||||||
|
- 依赖注入
|
||||||
|
- 事件机制（仅在确有异步解耦需求时使用）
|
||||||
|
|
||||||
|
5. 禁止一个模块直接访问另一个模块的内部变量、私有实现、内部数据库细节或非公开文件。
|
||||||
|
|
||||||
|
6. 模块应遵循单一职责原则。根据项目技术栈和现有结构，可以按以下职责组织：
|
||||||
|
- Router：定义路由和中间件
|
||||||
|
- Controller：接收请求、校验输入、调用服务、组织响应
|
||||||
|
- Service：实现业务逻辑
|
||||||
|
- Repository：封装数据库或持久化操作
|
||||||
|
- Adapter：封装外部存储、第三方 API 等基础设施
|
||||||
|
- Types/Interfaces：声明模块对外契约
|
||||||
|
|
||||||
|
7. 上述分层是职责参考，不是强制要求。对于简单功能，可以合并没有独立价值的层级，绝对禁止为了套用架构而过度设计。
|
||||||
|
|
||||||
|
8. 其他代码调用某个模块时，应只使用该模块明确对外提供的函数、参数和返回值，禁止依赖该模块的内部变量、内部文件或具体实现方式。模块内部实现被替换或扩展时，应尽量避免要求其他无关代码同步修改。
|
||||||
|
|
||||||
|
9. 模块内部实现可以变化，但对外接口应尽量保持稳定。
|
||||||
|
|
||||||
|
10. 禁止循环依赖。发现循环依赖时，应优先检查：
|
||||||
|
- 职责是否划分错误
|
||||||
|
- 是否存在应抽取的公共能力
|
||||||
|
- 是否需要使用抽象接口
|
||||||
|
- 是否错误地让业务模块相互了解内部实现
|
||||||
|
|
||||||
|
11. 公共逻辑只能保留一套权威实现。禁止在上传、删除或其他模块中分别复制相同的：
|
||||||
|
- 权限校验
|
||||||
|
- 路径处理
|
||||||
|
- 文件校验
|
||||||
|
- 错误转换
|
||||||
|
- 数据查询
|
||||||
|
- 状态判断
|
||||||
|
|
||||||
|
12. 模块化改造应保持现有接口路径、请求参数、响应结构和错误行为兼容，除非用户明确要求修改。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 四、前端开发
|
||||||
|
|
||||||
|
1. 前端禁止使用多渐变色，仅能使用同色系渐变，且不能太艳丽。
|
||||||
|
|
||||||
|
2. 前端禁止使用 Emoji 作为图标。图标必须统一使用第三方图标库；如果项目已经选定图标库，禁止再引入另一套图标库。
|
||||||
|
|
||||||
|
3. 前端禁止主动增加无障碍设置，包括但不限于额外的 ARIA 属性、无障碍模式和无障碍专用交互。
|
||||||
|
|
||||||
|
4. 前端禁止增加键盘快捷键、按键监听或依赖键盘完成的交互，除非用户明确要求。
|
||||||
|
|
||||||
|
5. 前端必须采用响应式布局，应适配项目现有支持的桌面端、平板端和移动端尺寸。
|
||||||
|
|
||||||
|
6. 前端必须使用统一的样式系统：
|
||||||
|
- 使用统一的颜色、间距、字号、圆角、阴影和层级变量。
|
||||||
|
- 使用语义化变量名，禁止在业务组件中散落大量无语义的硬编码样式值。
|
||||||
|
- 样式应放入项目现有的统一样式文件、主题文件或设计令牌系统。
|
||||||
|
- 优先复用现有组件和样式类。
|
||||||
|
- 禁止同一种视觉效果在多个页面分别实现。
|
||||||
|
|
||||||
|
7. 除非用户明确要求，否则前端禁止使用英文副标题。界面文案应与项目现有语言和表达方式保持一致。
|
||||||
|
|
||||||
|
8. 页面和组件应根据职责合理拆分。公共交互、公共布局和公共业务展示应抽取为公共组件，禁止在多个页面复制相同实现。
|
||||||
|
|
||||||
|
9. 修复前端问题时，仅修改解决根因所需的组件、样式和逻辑，禁止顺带重做页面设计。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 五、数据库修改
|
||||||
|
|
||||||
|
1. 涉及数据库结构修改时，必须同时完成：
|
||||||
|
- 更新数据库定义文件或 Schema。
|
||||||
|
- 创建对应的数据库迁移文件。
|
||||||
|
- 实际执行数据库迁移。
|
||||||
|
- 检查迁移执行结果。
|
||||||
|
- 验证应用代码与新结构兼容。
|
||||||
|
|
||||||
|
2. 禁止只修改或新增迁移文件而不更新数据库定义文件。
|
||||||
|
|
||||||
|
3. 禁止只更新数据库定义文件而不创建和执行迁移。
|
||||||
|
|
||||||
|
4. 修改字段、索引、约束、默认值或数据类型前，应检查现有数据兼容性和迁移风险。
|
||||||
|
|
||||||
|
5. 涉及删除字段、修改字段类型、增加非空约束等可能造成数据丢失或迁移失败的操作时，必须先说明风险，并制定数据迁移或兼容方案。
|
||||||
|
|
||||||
|
6. 数据库迁移完成后，应验证：
|
||||||
|
- 当前数据库版本
|
||||||
|
- 新旧数据兼容性
|
||||||
|
- 相关查询和写入逻辑
|
||||||
|
- 回滚或恢复风险
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 六、验证要求
|
||||||
|
|
||||||
|
1. 修改完成后，应优先运行与本次修改直接相关的：
|
||||||
|
- 单元测试
|
||||||
|
- 集成测试
|
||||||
|
- 类型检查
|
||||||
|
- 编译或构建检查
|
||||||
|
- Lint 检查
|
||||||
|
- 数据库迁移检查
|
||||||
|
|
||||||
|
2. 除非用户明确要求，否则必须禁止主动使用浏览器进行验证，也禁止主动调用浏览器自动化工具。
|
||||||
|
|
||||||
|
3. 如果前端修改无法通过现有自动化测试充分验证，应说明未验证的具体部分，不得将未验证内容描述为已经确认正常。
|
||||||
|
|
||||||
|
4. 测试失败时，应区分：
|
||||||
|
- 本次修改导致的失败
|
||||||
|
- 项目原本存在的失败
|
||||||
|
- 环境或依赖导致的失败
|
||||||
|
|
||||||
|
5. 禁止为了让测试通过而删除测试、降低断言强度、屏蔽错误或跳过必要检查。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 七、任务完成后的输出
|
||||||
|
|
||||||
|
完成任务后，应简要说明：
|
||||||
|
|
||||||
|
1. 问题根因或实现依据。
|
||||||
|
2. 实际修改的文件和主要内容。
|
||||||
|
3. 模块之间的调用关系。
|
||||||
|
4. 是否涉及接口或兼容性变化。
|
||||||
|
5. 是否涉及数据库迁移及执行结果。
|
||||||
|
6. 已运行的验证命令及结果。
|
||||||
|
7. 尚未验证的内容、已知限制和潜在风险。
|
||||||
|
|
||||||
|
禁止声称未实际执行的测试、迁移或验证已经通过。
|
||||||
|
|
||||||
|
</INSTRUCTIONS>
|
||||||
@@ -0,0 +1,87 @@
|
|||||||
|
# 剧游 AI 后端环境变量示例。
|
||||||
|
# 使用方式：复制为 API/.env，再填写实际配置；API/.env 不应提交到 Git。
|
||||||
|
# 标记为“必填”的变量不能继续使用下方占位符，否则服务可能无法正常启动。
|
||||||
|
|
||||||
|
# 应用基础配置
|
||||||
|
APP_NAME=剧游AI API
|
||||||
|
APP_ENV=development
|
||||||
|
DEBUG=true
|
||||||
|
SERVER_ADDRESS=:8900
|
||||||
|
|
||||||
|
# PostgreSQL
|
||||||
|
# 若使用根目录 docker-compose.yml，请将这三项填写为 postgres 服务的实际配置。
|
||||||
|
DATABASE_HOST=localhost
|
||||||
|
DATABASE_PORT=25432
|
||||||
|
# 本地开发示例值，不要填写生产数据库凭据。
|
||||||
|
DATABASE_NAME=juyou_ai_prod
|
||||||
|
DATABASE_USER=juyou_ai_app
|
||||||
|
DATABASE_PASSWORD=replace-with-local-postgres-password
|
||||||
|
# 填写后将优先使用该连接串；留空时由上面的 PostgreSQL 参数自动拼接。
|
||||||
|
DATABASE_URL=
|
||||||
|
|
||||||
|
# Redis
|
||||||
|
REDIS_HOST=localhost
|
||||||
|
REDIS_PORT=26379
|
||||||
|
REDIS_PASSWORD=
|
||||||
|
REDIS_DB=0
|
||||||
|
|
||||||
|
# Asynq 使用独立 Redis DB
|
||||||
|
ASYNQ_REDIS_DB=1
|
||||||
|
|
||||||
|
# 前端构建产物
|
||||||
|
STATIC_DIRECTORY=static
|
||||||
|
|
||||||
|
# 可选：外部模型服务需要代理时配置
|
||||||
|
HTTP_PROXY=
|
||||||
|
HTTPS_PROXY=
|
||||||
|
NO_PROXY=localhost,127.0.0.1,::1,.myqcloud.com,cos.youjuhui.xyz
|
||||||
|
|
||||||
|
# JWT（必填：请替换为随机长字符串）
|
||||||
|
JWT_SECRET_KEY=replace-with-a-random-secret-at-least-32-characters
|
||||||
|
JWT_ALGORITHM=HS256
|
||||||
|
JWT_ACCESS_TOKEN_EXPIRE_MINUTES=60
|
||||||
|
JWT_REFRESH_TOKEN_EXPIRE_HOURS=168
|
||||||
|
|
||||||
|
# 首次启动时仅在管理员表为空的情况下创建初始管理员；密码只用于初始化。
|
||||||
|
ADMIN_BOOTSTRAP_USERNAME=admin
|
||||||
|
ADMIN_BOOTSTRAP_PASSWORD=replace-with-a-strong-initial-password
|
||||||
|
|
||||||
|
# Argon2 密码哈希
|
||||||
|
ARGON2_TIME_COST=2
|
||||||
|
ARGON2_MEMORY_COST=19456
|
||||||
|
ARGON2_PARALLELISM=1
|
||||||
|
ARGON2_HASH_LENGTH=32
|
||||||
|
ARGON2_SALT_LENGTH=16
|
||||||
|
|
||||||
|
# 敏感配置加密（必填）。CONFIG_ENCRYPTION_KEY 必须是 32 字节随机值的标准 Base64 编码。
|
||||||
|
CONFIG_ENCRYPTION_KEY_VERSION=v1
|
||||||
|
CONFIG_ENCRYPTION_KEY=replace-with-base64-encoded-32-byte-key
|
||||||
|
|
||||||
|
# 腾讯云 COS（使用对象存储时必填）。密钥只配置在 API 服务端，不得暴露给前端。
|
||||||
|
COS_SECRET_ID=replace-with-tencent-secret-id
|
||||||
|
COS_SECRET_KEY=replace-with-tencent-secret-key
|
||||||
|
COS_BUCKET=replace-with-cos-bucket-name
|
||||||
|
COS_REGION=ap-chengdu
|
||||||
|
COS_ENDPOINT=https://cos.ap-chengdu.myqcloud.com
|
||||||
|
# 必填。视频反推与生成模型需要通过该域名读取参考图和抽帧。
|
||||||
|
COS_PUBLIC_BASE_URL=https://media.example.com
|
||||||
|
COS_MAX_IMAGE_SIZE_MB=20
|
||||||
|
COS_MAX_VIDEO_SIZE_MB=100
|
||||||
|
COS_MAX_AUDIO_SIZE_MB=30
|
||||||
|
|
||||||
|
# 持久任务与本地媒体处理。
|
||||||
|
AI_WORKER_CONCURRENCY=100
|
||||||
|
AI_POLL_INTERVAL_SECONDS=10
|
||||||
|
AI_HTTP_MAX_CONNECTIONS=200
|
||||||
|
FFMPEG_PATH=ffmpeg
|
||||||
|
FFPROBE_PATH=ffprobe
|
||||||
|
PYTHON_PATH=python3
|
||||||
|
ASR_SCRIPT_PATH=workers/transcribe.py
|
||||||
|
# 可填写模型目录；留空时由转写脚本使用 FASTER_WHISPER_MODEL_SIZE。
|
||||||
|
FASTER_WHISPER_MODEL_PATH=
|
||||||
|
FASTER_WHISPER_MODEL_SIZE=small
|
||||||
|
FASTER_WHISPER_DEVICE=cpu
|
||||||
|
FASTER_WHISPER_COMPUTE_TYPE=int8
|
||||||
|
|
||||||
|
# CORS（多个域名用逗号分隔）
|
||||||
|
CORS_ORIGINS=http://localhost:5500,http://localhost:5501
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
# acme.sh 本地目录
|
||||||
|
|
||||||
|
这里保存部署使用的 acme.sh 单文件版本，不从 GitHub clone。
|
||||||
|
|
||||||
|
更新方式：在 Linux 服务器或 WSL 中执行 `bash ../deploy/update_acme.sh`。
|
||||||
+9226
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,138 @@
|
|||||||
|
// API 进程入口，负责装配依赖、启动 HTTP 服务和异步任务处理器并执行平滑关闭。
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"log/slog"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/cache"
|
||||||
|
"juhe-factory/api/internal/config"
|
||||||
|
"juhe-factory/api/internal/database"
|
||||||
|
"juhe-factory/api/internal/datetime"
|
||||||
|
"juhe-factory/api/internal/handler"
|
||||||
|
adminmodule "juhe-factory/api/internal/modules/admin"
|
||||||
|
"juhe-factory/api/internal/modules/productimage"
|
||||||
|
"juhe-factory/api/internal/modules/prompt"
|
||||||
|
"juhe-factory/api/internal/queue"
|
||||||
|
"juhe-factory/api/internal/security"
|
||||||
|
"juhe-factory/api/internal/server"
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
"juhe-factory/api/internal/storage"
|
||||||
|
"juhe-factory/api/internal/worker"
|
||||||
|
|
||||||
|
"github.com/hibiken/asynq"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
time.Local = datetime.Location
|
||||||
|
cfg, err := config.Load()
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("加载配置失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
gormDB, sqlDB, err := database.Open(cfg)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("连接 PostgreSQL 失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
defer sqlDB.Close()
|
||||||
|
|
||||||
|
redisClient, err := cache.Open(cfg)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("连接 Redis 失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
defer redisClient.Close()
|
||||||
|
|
||||||
|
queueClient := queue.NewClient(cfg)
|
||||||
|
defer queueClient.Close()
|
||||||
|
|
||||||
|
healthHandler := handler.NewHealth(sqlDB, redisClient)
|
||||||
|
passwords := security.PasswordHasher{Time: cfg.Argon2Time, Memory: cfg.Argon2Memory, Parallelism: cfg.Argon2Parallelism, HashLength: cfg.Argon2HashLength, SaltLength: cfg.Argon2SaltLength}
|
||||||
|
tokens := security.TokenService{Secret: []byte(cfg.JWTSecretKey), AccessTTL: time.Duration(cfg.JWTAccessMinutes) * time.Minute, RefreshTTL: time.Duration(cfg.JWTRefreshHours) * time.Hour}
|
||||||
|
encryptor, err := security.NewEncryptor(cfg.EncryptionKey)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("加载敏感配置加密密钥失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
authService := service.NewAuth(gormDB, passwords, tokens)
|
||||||
|
if err := authService.Bootstrap(cfg.BootstrapAdminUsername, cfg.BootstrapAdminPassword); err != nil {
|
||||||
|
slog.Error("初始化管理员失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
cos, err := storage.NewCOS(context.Background(), cfg)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("初始化 COS 失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
adminDataService := &service.AdminData{DB: gormDB, Passwords: passwords, Encryptor: encryptor, HTTPClient: http.DefaultClient, EncryptionKeyVersion: cfg.EncryptionKeyVersion}
|
||||||
|
webService := &service.Web{DB: gormDB, Passwords: passwords, Tokens: security.WebTokenService{Secret: []byte(cfg.JWTSecretKey), AccessTTL: time.Duration(cfg.JWTAccessMinutes) * time.Minute}, RefreshTTL: time.Duration(cfg.JWTRefreshHours) * time.Hour}
|
||||||
|
creativeService := &service.Creative{DB: gormDB, Queue: queueClient, Encryptor: encryptor, HTTPClient: http.DefaultClient}
|
||||||
|
adminService := adminmodule.NewService(gormDB)
|
||||||
|
promptService := prompt.NewService(gormDB)
|
||||||
|
productImageService := productimage.NewService(gormDB, queueClient)
|
||||||
|
creativeHandler := handler.NewCreative(creativeService, promptService, cos)
|
||||||
|
httpServer := server.New(cfg, healthHandler, handler.NewAdminAuth(authService), handler.NewAdminData(adminDataService, adminService, promptService, cos), handler.NewWeb(webService, cos), creativeHandler, handler.NewProductImage(productImageService, cos))
|
||||||
|
|
||||||
|
aiHTTP := worker.NewHTTPClient(cfg.AIHTTPMaxConnections)
|
||||||
|
generationWorker := worker.NewGeneration(gormDB, queueClient, cos, encryptor, aiHTTP, time.Duration(cfg.AIPollIntervalSeconds)*time.Second)
|
||||||
|
mediaWorker := worker.NewMedia(gormDB, cos, encryptor, aiHTTP, cfg.FFmpegPath, cfg.FFprobePath, cfg.PythonPath, cfg.ASRScriptPath)
|
||||||
|
dramaParseWorker := worker.NewDramaParse(gormDB, queueClient, encryptor, aiHTTP)
|
||||||
|
scriptAnalysisWorker := worker.NewScriptAnalysis(gormDB, queueClient, encryptor, aiHTTP)
|
||||||
|
workerMux := asynq.NewServeMux()
|
||||||
|
generationWorker.Register(workerMux)
|
||||||
|
mediaWorker.Register(workerMux)
|
||||||
|
dramaParseWorker.Register(workerMux)
|
||||||
|
scriptAnalysisWorker.Register(workerMux)
|
||||||
|
queueServer := queue.NewServer(cfg)
|
||||||
|
workerErr := make(chan error, 1)
|
||||||
|
go func() {
|
||||||
|
slog.Info("AI 任务 Worker 已启动", "concurrency", cfg.AIWorkerConcurrency)
|
||||||
|
workerErr <- queueServer.Run(workerMux)
|
||||||
|
}()
|
||||||
|
if err := generationWorker.Recover(context.Background()); err != nil {
|
||||||
|
slog.Error("恢复 AI 任务失败", "error", err)
|
||||||
|
}
|
||||||
|
if err := dramaParseWorker.Recover(context.Background()); err != nil {
|
||||||
|
slog.Error("恢复短剧创作解析任务失败", "error", err)
|
||||||
|
}
|
||||||
|
if err := scriptAnalysisWorker.Recover(context.Background()); err != nil {
|
||||||
|
slog.Error("恢复剧本分析任务失败", "error", err)
|
||||||
|
}
|
||||||
|
serverErr := make(chan error, 1)
|
||||||
|
go func() {
|
||||||
|
slog.Info("API 服务已启动", "address", cfg.ServerAddress)
|
||||||
|
serverErr <- httpServer.ListenAndServe()
|
||||||
|
}()
|
||||||
|
|
||||||
|
stop := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case sig := <-stop:
|
||||||
|
slog.Info("收到停止信号", "signal", sig.String())
|
||||||
|
case err = <-serverErr:
|
||||||
|
if !errors.Is(err, http.ErrServerClosed) {
|
||||||
|
slog.Error("API 服务异常退出", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
case err = <-workerErr:
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("AI 任务 Worker 异常退出", "error", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
queueServer.Shutdown()
|
||||||
|
if err := httpServer.Shutdown(ctx); err != nil {
|
||||||
|
slog.Error("API 服务关闭失败", "error", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,53 @@
|
|||||||
|
# 剧核工厂部署脚本
|
||||||
|
|
||||||
|
## 一键推送
|
||||||
|
|
||||||
|
在 Windows 项目根目录执行：
|
||||||
|
|
||||||
|
```bat
|
||||||
|
API\deploy\push.bat root@服务器IP
|
||||||
|
```
|
||||||
|
|
||||||
|
脚本会构建 WEB、ADMIN，交叉编译 Linux amd64 API，并上传二进制、静态文件、迁移、部署脚本以及 `workers`。默认增量推送不会上传本地 `API/.env`，服务器继续使用 `/etc/juyou_ai/juyou_ai.env`，避免 Windows 开发环境配置覆盖 Linux 生产配置。
|
||||||
|
|
||||||
|
仅上传、不执行服务器部署时也可以显式使用：
|
||||||
|
|
||||||
|
```bat
|
||||||
|
API\deploy\push.bat root@服务器IP stage-only
|
||||||
|
```
|
||||||
|
|
||||||
|
只有首次部署或明确需要整体替换服务器环境配置时，才使用：
|
||||||
|
|
||||||
|
```bat
|
||||||
|
API\deploy\push.bat root@服务器IP with-env
|
||||||
|
```
|
||||||
|
|
||||||
|
`with-env` 会把本地 `API/.env` 暂存为服务器 `/tmp/juyou_ai.env`。使用前必须确认其中的 FFmpeg、Python 和脚本路径适用于 Linux；普通增量部署不要使用该选项。
|
||||||
|
|
||||||
|
## 一键部署
|
||||||
|
|
||||||
|
上传完成后，在服务器执行：
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo bash /opt/juyou_ai/deploy/one_click_deployment.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
普通增量部署会继续使用 `/etc/juyou_ai/juyou_ai.env`。使用 `with-env` 推送时，部署脚本才会用 `/tmp/juyou_ai.env` 更新生产环境配置。
|
||||||
|
|
||||||
|
部署脚本会安装 `python3-pip`，并在缺失时安装 `faster-whisper 1.2.1`；同时会将 FFmpeg、Python 和音频识别脚本路径规范为 Linux 运行路径。`push.bat` 会在本机下载默认 `small` 模型，并在服务器缺少模型时上传到 `/opt/juyou_ai/models/faster-whisper-small`。服务器部署和运行阶段不会访问 Hugging Face；后续增量部署会保留并复用已经上传的模型。
|
||||||
|
|
||||||
|
项目目录固定为 `/opt/juyou_ai`，systemd 服务名固定为 `juyou_ai`，API 内部端口固定为 `8123`。脚本仅保存域名和证书邮箱；再次执行会自动复用配置。多个域名会全部作为 SAN 追加到同一张证书。脚本只复用输入域名对应的现有证书，不会扫描其他项目的证书；没有匹配证书但填写邮箱时使用本地 `API/acme/acme.sh` 申请 HTTPS，域名填 `_` 或未填写邮箱时使用 HTTP。systemd 从 `/etc/juyou_ai/juyou_ai.env` 加载环境变量，项目目录不会保留 `.env`。
|
||||||
|
|
||||||
|
API、AI Worker 和 HTTP 请求日志统一写入 `/var/log/juyou_ai/juyou_ai.log`。日志每天轮转，最多保留 14 份，单个文件超过 50MB 时会提前轮转；可使用 `sudo tail -f /var/log/juyou_ai/juyou_ai.log` 实时查看。
|
||||||
|
|
||||||
|
完整操作说明参见 [部署文档.md](部署文档.md)。
|
||||||
|
|
||||||
|
第三方接口、网址和密钥字段说明参见 [第三方说明及密钥.md](第三方说明及密钥.md)。
|
||||||
|
|
||||||
|
## 更新 acme.sh
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo bash /opt/juyou_ai/deploy/update_acme.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
更新脚本通过 jsDelivr CDN 下载 acmesh-official/acme.sh 的最新 master 单文件版本，不执行 GitHub clone。
|
||||||
@@ -0,0 +1,212 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# 剧核工厂 Linux 一键部署脚本。
|
||||||
|
# 负责安装 Nginx/PostgreSQL/Redis/FFmpeg、执行迁移、创建 systemd 服务，
|
||||||
|
# 并按需使用本地 API/acme/acme.sh 申请或安装 HTTPS 证书。
|
||||||
|
|
||||||
|
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; CYAN='\033[0;36m'; RESET='\033[0m'
|
||||||
|
info(){ echo -e "${CYAN} -> $*${RESET}"; }
|
||||||
|
ok(){ echo -e "${GREEN} OK $*${RESET}"; }
|
||||||
|
warn(){ echo -e "${YELLOW} !! $*${RESET}"; }
|
||||||
|
fail(){ echo -e "${RED} ERR $*${RESET}" >&2; exit 1; }
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
API_ROOT="$(realpath "${1:-$SCRIPT_DIR/..}")"
|
||||||
|
PROJECT_ROOT="$(realpath "$API_ROOT/..")"
|
||||||
|
CONFIG_FILE="$SCRIPT_DIR/.deploy.conf"
|
||||||
|
SYSTEM_ENV_FILE="/etc/juyou_ai/juyou_ai.env"
|
||||||
|
[ -f "$SCRIPT_DIR/prepare_server.sh" ] || fail "缺少服务器初始化脚本。"
|
||||||
|
chmod +x "$API_ROOT/bin/jcf-api" 2>/dev/null || true
|
||||||
|
[ -x "$API_ROOT/bin/jcf-api" ] || fail "缺少 Linux API 二进制 $API_ROOT/bin/jcf-api，请先执行 push.bat。"
|
||||||
|
[ -f "$API_ROOT/static/web/index.html" ] || fail "缺少 WEB 构建产物。"
|
||||||
|
[ -f "$API_ROOT/static/admin/index.html" ] || fail "缺少 ADMIN 构建产物。"
|
||||||
|
|
||||||
|
if [ -s "$CONFIG_FILE" ]; then
|
||||||
|
# shellcheck disable=SC1090
|
||||||
|
source "$CONFIG_FILE"
|
||||||
|
fi
|
||||||
|
SERVICE_NAME="juyou_ai"
|
||||||
|
APP_PORT="8123"
|
||||||
|
DOMAIN_INPUT="${SAVED_DOMAIN_INPUT:-_}"
|
||||||
|
USER_EMAIL="${SAVED_USER_EMAIL:-}"
|
||||||
|
DOMAINS=()
|
||||||
|
if [ "$DOMAIN_INPUT" != "_" ]; then
|
||||||
|
for domain in $DOMAIN_INPUT; do
|
||||||
|
[[ "$domain" =~ ^([A-Za-z0-9]([A-Za-z0-9-]{0,61}[A-Za-z0-9])?\.)+[A-Za-z]{2,63}$ ]] || fail "域名格式无效：$domain"
|
||||||
|
DOMAINS+=("$domain")
|
||||||
|
done
|
||||||
|
[ "${#DOMAINS[@]}" -gt 0 ] || fail "至少需要一个有效域名。"
|
||||||
|
DOMAIN_INPUT="${DOMAINS[*]}"
|
||||||
|
FIRST_DOMAIN="${DOMAINS[0]}"
|
||||||
|
fi
|
||||||
|
FIRST_DOMAIN="${FIRST_DOMAIN:-_}"
|
||||||
|
command -v sudo >/dev/null 2>&1 || fail "系统缺少 sudo。"
|
||||||
|
sudo env SAVED_DOMAIN_INPUT="$DOMAIN_INPUT" JCF_ENV_SOURCE="${JCF_ENV_SOURCE:-}" bash "$SCRIPT_DIR/prepare_server.sh" "$API_ROOT"
|
||||||
|
[ -f "$SYSTEM_ENV_FILE" ] || fail "服务器初始化后仍缺少 $SYSTEM_ENV_FILE。"
|
||||||
|
SOCKET_PATH="/run/$SERVICE_NAME/app.sock"
|
||||||
|
CERT_DIR="/etc/nginx/ssl"
|
||||||
|
NGINX_FILE="/etc/nginx/conf.d/$SERVICE_NAME.conf"
|
||||||
|
SERVICE_FILE="/etc/systemd/system/$SERVICE_NAME.service"
|
||||||
|
LOG_DIR="/var/log/$SERVICE_NAME"
|
||||||
|
LOG_FILE="$LOG_DIR/$SERVICE_NAME.log"
|
||||||
|
LOGROTATE_FILE="/etc/logrotate.d/$SERVICE_NAME"
|
||||||
|
|
||||||
|
# 只读取迁移所需的数据库变量，不直接 source .env。
|
||||||
|
# APP_NAME 等值可能包含空格，直接 source 会被 Bash 误解析。
|
||||||
|
env_value() {
|
||||||
|
local key="$1" value
|
||||||
|
value="$(sudo sed -n -E "s/^${key}=//p" "$SYSTEM_ENV_FILE" | head -n 1)"
|
||||||
|
value="${value#\"}"; value="${value%\"}"
|
||||||
|
printf '%s' "$value"
|
||||||
|
}
|
||||||
|
DATABASE_URL="$(env_value DATABASE_URL)"
|
||||||
|
DATABASE_NAME="$(env_value DATABASE_NAME)"
|
||||||
|
DATABASE_HOST="$(env_value DATABASE_HOST)"
|
||||||
|
DATABASE_PORT="$(env_value DATABASE_PORT)"
|
||||||
|
DATABASE_USER="$(env_value DATABASE_USER)"
|
||||||
|
DATABASE_PASSWORD="$(env_value DATABASE_PASSWORD)"
|
||||||
|
PSQL_ARGS=()
|
||||||
|
if [ -n "${DATABASE_URL:-}" ]; then
|
||||||
|
PSQL_TARGET="$DATABASE_URL"
|
||||||
|
else
|
||||||
|
PSQL_TARGET="dbname=${DATABASE_NAME:-juchuang_factory} host=${DATABASE_HOST:-127.0.0.1} port=${DATABASE_PORT:-5432} user=${DATABASE_USER:-postgres}"
|
||||||
|
export PGPASSWORD="${DATABASE_PASSWORD:-}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 使用独立记录表避免重复执行同一个 SQL 迁移。
|
||||||
|
psql "$PSQL_TARGET" -v ON_ERROR_STOP=1 -c 'CREATE TABLE IF NOT EXISTS jcf_schema_migrations (version varchar(128) PRIMARY KEY, applied_at timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP);' >/dev/null
|
||||||
|
for migration in "$API_ROOT"/migrations/*.up.sql; do
|
||||||
|
[ -f "$migration" ] || continue
|
||||||
|
version="$(basename "$migration")"
|
||||||
|
already="$(psql "$PSQL_TARGET" -At -v ON_ERROR_STOP=1 -c "SELECT 1 FROM jcf_schema_migrations WHERE version='${version//\'/\'\'}' LIMIT 1")"
|
||||||
|
if [ "$already" = "1" ]; then
|
||||||
|
info "跳过已执行迁移：$version"
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
info "执行数据库迁移：$version"
|
||||||
|
psql "$PSQL_TARGET" -v ON_ERROR_STOP=1 -f "$migration" >/dev/null
|
||||||
|
psql "$PSQL_TARGET" -v ON_ERROR_STOP=1 -c "INSERT INTO jcf_schema_migrations(version) VALUES ('${version//\'/\'\'}')" >/dev/null
|
||||||
|
done
|
||||||
|
unset PGPASSWORD
|
||||||
|
|
||||||
|
sudo chown -R www-data:www-data "$API_ROOT"
|
||||||
|
sudo rm -f "$API_ROOT/.env"
|
||||||
|
sudo mkdir -p "$(dirname "$SOCKET_PATH")" "$CERT_DIR" /var/www/jcf-acme
|
||||||
|
sudo install -d -m 0750 -o www-data -g www-data "$LOG_DIR"
|
||||||
|
sudo touch "$LOG_FILE"
|
||||||
|
sudo chown www-data:www-data "$LOG_FILE"
|
||||||
|
sudo chmod 0640 "$LOG_FILE"
|
||||||
|
|
||||||
|
# 每天轮转 API 与 Worker 的统一日志，并在单个文件过大时提前轮转。
|
||||||
|
sudo tee "$LOGROTATE_FILE" >/dev/null <<EOF
|
||||||
|
$LOG_FILE {
|
||||||
|
daily
|
||||||
|
rotate 14
|
||||||
|
maxsize 50M
|
||||||
|
missingok
|
||||||
|
notifempty
|
||||||
|
compress
|
||||||
|
delaycompress
|
||||||
|
copytruncate
|
||||||
|
}
|
||||||
|
EOF
|
||||||
|
|
||||||
|
# 创建 API 与队列 Worker 同进程 systemd 服务。
|
||||||
|
sudo tee "$SERVICE_FILE" >/dev/null <<EOF
|
||||||
|
[Unit]
|
||||||
|
Description=JuheFactory Go API and AI Worker
|
||||||
|
After=network.target postgresql.service redis-server.service
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
User=www-data
|
||||||
|
Group=www-data
|
||||||
|
WorkingDirectory=$API_ROOT
|
||||||
|
EnvironmentFile=$SYSTEM_ENV_FILE
|
||||||
|
Environment=HOME=$API_ROOT
|
||||||
|
CacheDirectory=juyou_ai
|
||||||
|
ExecStart=$API_ROOT/bin/jcf-api
|
||||||
|
StandardOutput=append:$LOG_FILE
|
||||||
|
StandardError=append:$LOG_FILE
|
||||||
|
Restart=always
|
||||||
|
RestartSec=3
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
EOF
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl enable "$SERVICE_NAME" --now
|
||||||
|
|
||||||
|
mkdir -p "$API_ROOT/static/web" "$API_ROOT/static/admin"
|
||||||
|
EXISTING_CERT_BUNDLE="$CERT_DIR/${FIRST_DOMAIN}_bundle.crt"
|
||||||
|
EXISTING_CERT_KEY="$CERT_DIR/${FIRST_DOMAIN}.key"
|
||||||
|
if [ "$FIRST_DOMAIN" != "_" ] && [ -f "$EXISTING_CERT_BUNDLE" ] && [ -f "$EXISTING_CERT_KEY" ]; then
|
||||||
|
info "检测到现有 SSL 证书，跳过 ACME 申请并复用证书"
|
||||||
|
USE_SSL=true
|
||||||
|
elif [ "$FIRST_DOMAIN" != "_" ] && [ -n "$USER_EMAIL" ]; then
|
||||||
|
# 先用 HTTP 临时站点提供 ACME HTTP-01 验证目录。
|
||||||
|
sudo tee "$NGINX_FILE" >/dev/null <<EOF
|
||||||
|
server {
|
||||||
|
listen 80;
|
||||||
|
server_name $DOMAIN_INPUT;
|
||||||
|
location ^~ /.well-known/acme-challenge/ { root /var/www/jcf-acme; }
|
||||||
|
location / { root $API_ROOT/static/web; try_files \$uri \$uri/ /index.html; }
|
||||||
|
}
|
||||||
|
EOF
|
||||||
|
sudo nginx -t && sudo systemctl reload nginx
|
||||||
|
ACME_BIN="$API_ROOT/acme/acme.sh"
|
||||||
|
[ -f "$ACME_BIN" ] || fail "缺少本地 ACME 脚本，请先执行 API/deploy/update_acme.sh。"
|
||||||
|
chmod +x "$ACME_BIN"
|
||||||
|
export LE_CONFIG_HOME="/root/.acme.sh"
|
||||||
|
bash "$ACME_BIN" --set-default-ca --server letsencrypt
|
||||||
|
ACME_DOMAIN_ARGS=()
|
||||||
|
for domain in "${DOMAINS[@]}"; do ACME_DOMAIN_ARGS+=( -d "$domain" ); done
|
||||||
|
bash "$ACME_BIN" --issue --webroot /var/www/jcf-acme "${ACME_DOMAIN_ARGS[@]}" --accountemail "$USER_EMAIL"
|
||||||
|
bash "$ACME_BIN" --install-cert -d "$FIRST_DOMAIN" \
|
||||||
|
--key-file "$CERT_DIR/${FIRST_DOMAIN}.key" \
|
||||||
|
--fullchain-file "$CERT_DIR/${FIRST_DOMAIN}_bundle.crt" \
|
||||||
|
--reloadcmd "systemctl reload nginx"
|
||||||
|
USE_SSL=true
|
||||||
|
else
|
||||||
|
USE_SSL=false
|
||||||
|
[ "$FIRST_DOMAIN" = "_" ] || warn "未填写证书邮箱，部署为 HTTP。"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "$USE_SSL" = true ]; then
|
||||||
|
PROTOCOL=https
|
||||||
|
sudo tee "$NGINX_FILE" >/dev/null <<EOF
|
||||||
|
server { listen 80; server_name $DOMAIN_INPUT; return 301 https://\$host\$request_uri; }
|
||||||
|
server {
|
||||||
|
listen 443 ssl http2;
|
||||||
|
server_name $DOMAIN_INPUT;
|
||||||
|
ssl_certificate $CERT_DIR/${FIRST_DOMAIN}_bundle.crt;
|
||||||
|
ssl_certificate_key $CERT_DIR/${FIRST_DOMAIN}.key;
|
||||||
|
ssl_protocols TLSv1.2 TLSv1.3;
|
||||||
|
client_max_body_size 2g;
|
||||||
|
location /api/ { proxy_pass http://127.0.0.1:$APP_PORT; proxy_set_header Host \$host; proxy_set_header X-Real-IP \$remote_addr; proxy_set_header X-Forwarded-For \$proxy_add_x_forwarded_for; proxy_read_timeout 300; proxy_buffering off; }
|
||||||
|
location ~ ^/(health|docs|redoc|openapi.json)(/.*)?\$ { proxy_pass http://127.0.0.1:$APP_PORT; proxy_set_header Host \$host; }
|
||||||
|
location /admin/ { alias $API_ROOT/static/admin/; try_files \$uri \$uri/ /admin/index.html; }
|
||||||
|
location / { root $API_ROOT/static/web; try_files \$uri \$uri/ /index.html; }
|
||||||
|
}
|
||||||
|
EOF
|
||||||
|
else
|
||||||
|
PROTOCOL=http
|
||||||
|
sudo tee "$NGINX_FILE" >/dev/null <<EOF
|
||||||
|
server {
|
||||||
|
listen 80;
|
||||||
|
server_name $DOMAIN_INPUT;
|
||||||
|
client_max_body_size 2g;
|
||||||
|
location /api/ { proxy_pass http://127.0.0.1:$APP_PORT; proxy_set_header Host \$host; proxy_set_header X-Real-IP \$remote_addr; proxy_set_header X-Forwarded-For \$proxy_add_x_forwarded_for; proxy_read_timeout 300; proxy_buffering off; }
|
||||||
|
location ~ ^/(health|docs|redoc|openapi.json)(/.*)?\$ { proxy_pass http://127.0.0.1:$APP_PORT; proxy_set_header Host \$host; }
|
||||||
|
location /admin/ { alias $API_ROOT/static/admin/; try_files \$uri \$uri/ /admin/index.html; }
|
||||||
|
location / { root $API_ROOT/static/web; try_files \$uri \$uri/ /index.html; }
|
||||||
|
}
|
||||||
|
EOF
|
||||||
|
fi
|
||||||
|
sudo nginx -t
|
||||||
|
sudo systemctl reload nginx || sudo systemctl restart nginx
|
||||||
|
sudo systemctl restart "$SERVICE_NAME"
|
||||||
|
if [ -n "${JCF_ENV_SOURCE:-}" ]; then
|
||||||
|
rm -f -- "$JCF_ENV_SOURCE"
|
||||||
|
fi
|
||||||
|
ok "部署完成：${PROTOCOL}://${FIRST_DOMAIN}，服务名：$SERVICE_NAME"
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from faster_whisper.utils import download_model
|
||||||
|
|
||||||
|
|
||||||
|
MODEL_SIZE = "small"
|
||||||
|
MODEL_DIR = Path(__file__).resolve().parent.parent / "models" / f"faster-whisper-{MODEL_SIZE}"
|
||||||
|
REQUIRED_FILES = ("config.json", "model.bin", "tokenizer.json", "vocabulary.txt")
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> int:
|
||||||
|
missing = [name for name in REQUIRED_FILES if not (MODEL_DIR / name).is_file()]
|
||||||
|
if not missing:
|
||||||
|
print(f"Whisper model already exists: {MODEL_DIR}")
|
||||||
|
return 0
|
||||||
|
|
||||||
|
MODEL_DIR.mkdir(parents=True, exist_ok=True)
|
||||||
|
print(f"Downloading faster-whisper-{MODEL_SIZE} to {MODEL_DIR} ...")
|
||||||
|
download_model(MODEL_SIZE, output_dir=str(MODEL_DIR))
|
||||||
|
missing = [name for name in REQUIRED_FILES if not (MODEL_DIR / name).is_file()]
|
||||||
|
if missing:
|
||||||
|
raise RuntimeError(f"Whisper model download is incomplete; missing: {', '.join(missing)}")
|
||||||
|
print(f"Whisper model is ready: {MODEL_DIR}")
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
raise SystemExit(main())
|
||||||
@@ -0,0 +1,52 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# 剧核工厂一键部署入口。
|
||||||
|
# 第一次执行会询问部署参数并保存到 API/deploy/.deploy.conf；以后执行会自动复用。
|
||||||
|
# 完整部署流程在同目录的 deploy_with_ssl.sh 中，便于后续单独重复执行。
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
API_ROOT="$(realpath "$SCRIPT_DIR/..")"
|
||||||
|
PROJECT_ROOT="$(realpath "$API_ROOT/..")"
|
||||||
|
CONFIG_FILE="$SCRIPT_DIR/.deploy.conf"
|
||||||
|
|
||||||
|
if [ -s "$CONFIG_FILE" ]; then
|
||||||
|
# shellcheck disable=SC1090
|
||||||
|
source "$CONFIG_FILE"
|
||||||
|
else
|
||||||
|
SAVED_DOMAIN_INPUT=""
|
||||||
|
SAVED_USER_EMAIL=""
|
||||||
|
EXISTING_NGINX_CONFIG="/etc/nginx/conf.d/juyou_ai.conf"
|
||||||
|
if [ -f "$EXISTING_NGINX_CONFIG" ]; then
|
||||||
|
SAVED_DOMAIN_INPUT="$(sed -n -E 's/^[[:space:]]*server_name[[:space:]]+([^;]+);.*$/\1/p' "$EXISTING_NGINX_CONFIG" | head -n 1)"
|
||||||
|
SAVED_DOMAIN_INPUT="${SAVED_DOMAIN_INPUT//_/}"
|
||||||
|
SAVED_DOMAIN_INPUT="$(printf '%s' "$SAVED_DOMAIN_INPUT" | xargs)"
|
||||||
|
fi
|
||||||
|
if [ -z "$SAVED_DOMAIN_INPUT" ]; then
|
||||||
|
read -r -p "域名（多个域名空格分隔，将全部追加到证书；不申请 SSL 填 _）[_]: " SAVED_DOMAIN_INPUT
|
||||||
|
SAVED_DOMAIN_INPUT="${SAVED_DOMAIN_INPUT:-_}"
|
||||||
|
fi
|
||||||
|
FIRST_SAVED_DOMAIN="${SAVED_DOMAIN_INPUT%% *}"
|
||||||
|
EXISTING_CERT_READY=false
|
||||||
|
if [ -n "$FIRST_SAVED_DOMAIN" ] && [ -f "/etc/nginx/ssl/${FIRST_SAVED_DOMAIN}_bundle.crt" ] && [ -f "/etc/nginx/ssl/${FIRST_SAVED_DOMAIN}.key" ]; then
|
||||||
|
EXISTING_CERT_READY=true
|
||||||
|
fi
|
||||||
|
if [ "$SAVED_DOMAIN_INPUT" != "_" ] && [ "$EXISTING_CERT_READY" = true ]; then
|
||||||
|
echo "检测到当前域名的现有证书，复用域名和证书：$SAVED_DOMAIN_INPUT"
|
||||||
|
elif [ "$SAVED_DOMAIN_INPUT" != "_" ]; then
|
||||||
|
read -r -p "SSL 证书邮箱（申请新证书；不申请 SSL 直接回车）: " SAVED_USER_EMAIL
|
||||||
|
fi
|
||||||
|
cat >"$CONFIG_FILE" <<EOF
|
||||||
|
# 部署脚本自动保存的配置，可直接编辑后重新运行。
|
||||||
|
SAVED_DOMAIN_INPUT=$(printf '%q' "$SAVED_DOMAIN_INPUT")
|
||||||
|
SAVED_USER_EMAIL=$(printf '%q' "$SAVED_USER_EMAIL")
|
||||||
|
EOF
|
||||||
|
fi
|
||||||
|
|
||||||
|
SAVED_DOMAIN_INPUT="${SAVED_DOMAIN_INPUT//$'\r'/}"
|
||||||
|
SAVED_USER_EMAIL="${SAVED_USER_EMAIL//$'\r'/}"
|
||||||
|
export SAVED_DOMAIN_INPUT SAVED_USER_EMAIL
|
||||||
|
if [ -z "${JCF_ENV_SOURCE:-}" ] && [ -f /tmp/juyou_ai.env ]; then
|
||||||
|
export JCF_ENV_SOURCE=/tmp/juyou_ai.env
|
||||||
|
fi
|
||||||
|
exec bash "$SCRIPT_DIR/deploy_with_ssl.sh" "$API_ROOT"
|
||||||
@@ -0,0 +1,166 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# Prepare the Linux host and install the uploaded environment file outside the app directory.
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
API_ROOT="$(realpath "${1:-$SCRIPT_DIR/..}")"
|
||||||
|
SOURCE_ENV="${JCF_ENV_SOURCE:-}"
|
||||||
|
SYSTEM_ENV_DIR="/etc/juyou_ai"
|
||||||
|
SYSTEM_ENV_FILE="$SYSTEM_ENV_DIR/juyou_ai.env"
|
||||||
|
ASR_CACHE_ROOT="/var/cache/juyou_ai"
|
||||||
|
ASR_HF_HOME="$ASR_CACHE_ROOT/huggingface"
|
||||||
|
ASR_MODEL_PATH="$API_ROOT/models/faster-whisper-small"
|
||||||
|
|
||||||
|
info(){ printf ' -> %s\n' "$*"; }
|
||||||
|
fail(){ printf ' ERR %s\n' "$*" >&2; exit 1; }
|
||||||
|
|
||||||
|
[ "$(id -u)" -eq 0 ] || fail "请使用 sudo 或 root 执行。"
|
||||||
|
|
||||||
|
info "安装 PostgreSQL、Redis、Nginx、FFmpeg 与基础工具..."
|
||||||
|
if command -v apt-get >/dev/null 2>&1; then
|
||||||
|
apt-get update -qq
|
||||||
|
DEBIAN_FRONTEND=noninteractive apt-get install -y -qq sudo nginx postgresql postgresql-contrib redis-server ffmpeg curl socat ca-certificates openssl logrotate python3 python3-pip
|
||||||
|
elif command -v dnf >/dev/null 2>&1; then
|
||||||
|
dnf install -y sudo nginx postgresql-server postgresql-contrib redis ffmpeg curl socat ca-certificates openssl logrotate python3 python3-pip
|
||||||
|
elif command -v yum >/dev/null 2>&1; then
|
||||||
|
yum install -y sudo nginx postgresql-server postgresql-contrib redis ffmpeg curl socat ca-certificates openssl logrotate python3 python3-pip
|
||||||
|
else
|
||||||
|
fail "不支持当前 Linux 发行版：未找到 apt-get、dnf 或 yum。"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! python3 -c 'import faster_whisper' >/dev/null 2>&1; then
|
||||||
|
PIP_ARGS=(install --disable-pip-version-check)
|
||||||
|
if python3 -m pip install --help 2>/dev/null | grep -q -- '--break-system-packages'; then
|
||||||
|
PIP_ARGS+=(--break-system-packages)
|
||||||
|
fi
|
||||||
|
python3 -m pip "${PIP_ARGS[@]}" 'faster-whisper==1.2.1'
|
||||||
|
fi
|
||||||
|
python3 -c 'import faster_whisper' >/dev/null 2>&1 || fail "faster-whisper installation failed."
|
||||||
|
|
||||||
|
if command -v postgresql-setup >/dev/null 2>&1 && [ ! -f /var/lib/pgsql/data/PG_VERSION ]; then
|
||||||
|
info "初始化 PostgreSQL 数据目录..."
|
||||||
|
postgresql-setup --initdb
|
||||||
|
fi
|
||||||
|
systemctl enable --now postgresql >/dev/null 2>&1 || fail "PostgreSQL 启动失败。"
|
||||||
|
if systemctl list-unit-files redis-server.service >/dev/null 2>&1; then
|
||||||
|
systemctl enable --now redis-server >/dev/null
|
||||||
|
else
|
||||||
|
systemctl enable --now redis >/dev/null || fail "Redis 启动失败。"
|
||||||
|
fi
|
||||||
|
systemctl enable --now nginx >/dev/null
|
||||||
|
sudo -u postgres pg_isready >/dev/null || fail "PostgreSQL 未就绪。"
|
||||||
|
redis-cli ping | grep -q '^PONG$' || fail "Redis 未就绪。"
|
||||||
|
|
||||||
|
install -d -m 700 -o root -g root "$SYSTEM_ENV_DIR"
|
||||||
|
if [ -n "$SOURCE_ENV" ] && [ -f "$SOURCE_ENV" ]; then
|
||||||
|
install -m 600 -o root -g root "$SOURCE_ENV" "$SYSTEM_ENV_FILE"
|
||||||
|
info "已将上传的环境配置安装到 $SYSTEM_ENV_FILE。"
|
||||||
|
elif [ -f "$SYSTEM_ENV_FILE" ]; then
|
||||||
|
info "未上传新配置，继续使用 $SYSTEM_ENV_FILE。"
|
||||||
|
else
|
||||||
|
fail "缺少上传的 API/.env，且服务器不存在 $SYSTEM_ENV_FILE。"
|
||||||
|
fi
|
||||||
|
# The application directory must not retain deployment secrets.
|
||||||
|
rm -f "$API_ROOT/.env"
|
||||||
|
|
||||||
|
env_value() {
|
||||||
|
local key="$1"
|
||||||
|
sed -n -E "s/^${key}=//p" "$SYSTEM_ENV_FILE" | head -n 1 | tr -d '\r' | sed -E 's/^"//;s/"$//'
|
||||||
|
}
|
||||||
|
set_env() {
|
||||||
|
local key="$1" value="$2" temporary
|
||||||
|
temporary="$(mktemp "$SYSTEM_ENV_DIR/env.XXXXXX")"
|
||||||
|
awk -v key="$key" -v value="$value" '
|
||||||
|
BEGIN { found=0 }
|
||||||
|
index($0, key "=")==1 { print key "=" value; found=1; next }
|
||||||
|
{ print }
|
||||||
|
END { if (!found) print key "=" value }
|
||||||
|
' "$SYSTEM_ENV_FILE" >"$temporary"
|
||||||
|
install -m 600 -o root -g root "$temporary" "$SYSTEM_ENV_FILE"
|
||||||
|
rm -f "$temporary"
|
||||||
|
}
|
||||||
|
require_env() {
|
||||||
|
local key="$1" value
|
||||||
|
value="$(env_value "$key")"
|
||||||
|
[ -n "$value" ] || fail "$key 未配置。"
|
||||||
|
[[ "$value" != *replace-with* && "$value" != *example.com* && "$value" != *change-me* && "$value" != "juchuang_dev" ]] || fail "$key 仍是示例或开发环境值。"
|
||||||
|
}
|
||||||
|
|
||||||
|
APP_PORT="8123"
|
||||||
|
DOMAIN_INPUT="${SAVED_DOMAIN_INPUT:-_}"
|
||||||
|
if ! [[ "$APP_PORT" =~ ^[0-9]+$ ]] || [ "$APP_PORT" -lt 1 ] || [ "$APP_PORT" -gt 65535 ]; then
|
||||||
|
fail "API 端口无效：$APP_PORT"
|
||||||
|
fi
|
||||||
|
CORS_ORIGINS=""
|
||||||
|
if [ "$DOMAIN_INPUT" != "_" ]; then
|
||||||
|
for domain in $DOMAIN_INPUT; do
|
||||||
|
[[ "$domain" =~ ^([A-Za-z0-9]([A-Za-z0-9-]{0,61}[A-Za-z0-9])?\.)+[A-Za-z]{2,63}$ ]] || fail "域名格式无效：$domain"
|
||||||
|
[ -n "$CORS_ORIGINS" ] && CORS_ORIGINS+=","
|
||||||
|
CORS_ORIGINS+="https://$domain,http://$domain"
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Infrastructure values are normalized for native Linux services; application secrets stay unchanged.
|
||||||
|
set_env APP_ENV production
|
||||||
|
set_env DEBUG false
|
||||||
|
set_env SERVER_ADDRESS "127.0.0.1:$APP_PORT"
|
||||||
|
set_env DATABASE_HOST 127.0.0.1
|
||||||
|
set_env DATABASE_PORT 5432
|
||||||
|
set_env DATABASE_URL ""
|
||||||
|
set_env REDIS_HOST 127.0.0.1
|
||||||
|
set_env REDIS_PORT 6379
|
||||||
|
set_env HTTP_PROXY ""
|
||||||
|
set_env HTTPS_PROXY ""
|
||||||
|
set_env ALL_PROXY ""
|
||||||
|
set_env http_proxy ""
|
||||||
|
set_env https_proxy ""
|
||||||
|
set_env all_proxy ""
|
||||||
|
set_env CORS_ORIGINS "$CORS_ORIGINS"
|
||||||
|
set_env FFMPEG_PATH ffmpeg
|
||||||
|
set_env FFPROBE_PATH ffprobe
|
||||||
|
set_env PYTHON_PATH python3
|
||||||
|
set_env ASR_SCRIPT_PATH "$API_ROOT/workers/transcribe.py"
|
||||||
|
set_env JCF_ASR_CACHE_DIR "$ASR_CACHE_ROOT"
|
||||||
|
set_env HF_HOME "$ASR_HF_HOME"
|
||||||
|
set_env FASTER_WHISPER_MODEL_PATH "$ASR_MODEL_PATH"
|
||||||
|
[ -f "$API_ROOT/workers/transcribe.py" ] || fail "Missing ASR script: $API_ROOT/workers/transcribe.py"
|
||||||
|
for model_file in config.json model.bin tokenizer.json vocabulary.txt; do
|
||||||
|
[ -f "$ASR_MODEL_PATH/$model_file" ] || fail "离线 Whisper 模型不完整，缺少：$ASR_MODEL_PATH/$model_file。请重新执行 push.bat 上传模型。"
|
||||||
|
done
|
||||||
|
|
||||||
|
install -d -m 750 -o www-data -g www-data "$ASR_CACHE_ROOT" "$ASR_HF_HOME"
|
||||||
|
chown -R www-data:www-data "$ASR_MODEL_PATH"
|
||||||
|
info "使用离线 faster-whisper 模型：$ASR_MODEL_PATH。"
|
||||||
|
|
||||||
|
for key in DATABASE_NAME DATABASE_USER DATABASE_PASSWORD JWT_SECRET_KEY ADMIN_BOOTSTRAP_PASSWORD CONFIG_ENCRYPTION_KEY COS_SECRET_ID COS_SECRET_KEY COS_BUCKET COS_REGION COS_ENDPOINT COS_PUBLIC_BASE_URL; do
|
||||||
|
require_env "$key"
|
||||||
|
done
|
||||||
|
|
||||||
|
DB_NAME="$(env_value DATABASE_NAME)"
|
||||||
|
DB_USER="$(env_value DATABASE_USER)"
|
||||||
|
DB_PASSWORD="$(env_value DATABASE_PASSWORD)"
|
||||||
|
[[ "$DB_NAME" =~ ^[A-Za-z_][A-Za-z0-9_]*$ ]] || fail "DATABASE_NAME 只允许字母、数字和下划线。"
|
||||||
|
[[ "$DB_USER" =~ ^[A-Za-z_][A-Za-z0-9_]*$ ]] || fail "DATABASE_USER 只允许字母、数字和下划线。"
|
||||||
|
|
||||||
|
if ! sudo -u postgres psql -tAc "SELECT 1 FROM pg_roles WHERE rolname='$DB_USER'" | grep -q 1; then
|
||||||
|
sudo -u postgres psql -v ON_ERROR_STOP=1 --set=password="$DB_PASSWORD" >/dev/null <<SQL
|
||||||
|
SET password_encryption = 'scram-sha-256';
|
||||||
|
CREATE ROLE "$DB_USER" LOGIN PASSWORD :'password';
|
||||||
|
SQL
|
||||||
|
else
|
||||||
|
sudo -u postgres psql -v ON_ERROR_STOP=1 --set=password="$DB_PASSWORD" >/dev/null <<SQL
|
||||||
|
SET password_encryption = 'scram-sha-256';
|
||||||
|
ALTER ROLE "$DB_USER" WITH LOGIN PASSWORD :'password';
|
||||||
|
SQL
|
||||||
|
fi
|
||||||
|
if ! sudo -u postgres psql -tAc "SELECT 1 FROM pg_database WHERE datname='$DB_NAME'" | grep -q 1; then
|
||||||
|
sudo -u postgres createdb -O "$DB_USER" "$DB_NAME"
|
||||||
|
fi
|
||||||
|
sudo -u postgres psql -v ON_ERROR_STOP=1 -d "$DB_NAME" -c "ALTER DATABASE \"$DB_NAME\" SET timezone TO 'Asia/Shanghai';" >/dev/null
|
||||||
|
HBA_FILE="$(sudo -u postgres psql -Atc 'SHOW hba_file')"
|
||||||
|
if ! grep -q '^# JCF local application access$' "$HBA_FILE"; then
|
||||||
|
sed -i "1i# JCF local application access\nhost all all 127.0.0.1/32 scram-sha-256\nhost all all ::1/128 scram-sha-256" "$HBA_FILE"
|
||||||
|
systemctl reload postgresql
|
||||||
|
fi
|
||||||
|
info "PostgreSQL 数据库与用户已初始化，Redis 已启动。"
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
@echo off
|
||||||
|
REM JCF deployment: build locally, upload over SSH, and deploy on Linux.
|
||||||
|
REM Usage: push.bat user@host [stage-only|with-env]
|
||||||
|
|
||||||
|
setlocal DisableDelayedExpansion
|
||||||
|
cd /d "%~dp0..\.."
|
||||||
|
if "%~1"=="" (
|
||||||
|
echo Usage: push.bat user@host [stage-only^|with-env]
|
||||||
|
echo Example: push.bat root@1.2.3.4
|
||||||
|
exit /b 1
|
||||||
|
)
|
||||||
|
set "REMOTE=%~1"
|
||||||
|
set "REMOTE_ROOT=/opt/juyou_ai"
|
||||||
|
set "PACKAGE_ENV="
|
||||||
|
if "%~2"=="with-env" set "PACKAGE_ENV=.env"
|
||||||
|
if not "%~2"=="" if not "%~2"=="stage-only" if not "%~2"=="with-env" (
|
||||||
|
echo The second argument can only be stage-only or with-env.
|
||||||
|
exit /b 1
|
||||||
|
)
|
||||||
|
if "%~2"=="with-env" if not exist "API\.env" (
|
||||||
|
echo Missing API\.env. Deployment environment cannot be uploaded.
|
||||||
|
exit /b 1
|
||||||
|
)
|
||||||
|
if not "%~3"=="" (
|
||||||
|
echo Too many arguments.
|
||||||
|
exit /b 1
|
||||||
|
)
|
||||||
|
set "ARCHIVE=%TEMP%\juyou-deploy-%RANDOM%-%RANDOM%.tar.gz"
|
||||||
|
set "REMOTE_ARCHIVE=/tmp/juyou-deploy.tar.gz"
|
||||||
|
set "REMOTE_ENV=/tmp/juyou_ai.env"
|
||||||
|
set "PACKAGE_MODEL="
|
||||||
|
|
||||||
|
echo [0/5] Checking offline Whisper model...
|
||||||
|
ssh "%REMOTE%" "test -f %REMOTE_ROOT%/models/faster-whisper-small/config.json && test -f %REMOTE_ROOT%/models/faster-whisper-small/model.bin && test -f %REMOTE_ROOT%/models/faster-whisper-small/tokenizer.json && test -f %REMOTE_ROOT%/models/faster-whisper-small/vocabulary.txt" >nul 2>nul
|
||||||
|
if errorlevel 1 (
|
||||||
|
python "API\deploy\download_whisper_model.py"
|
||||||
|
if errorlevel 1 exit /b 1
|
||||||
|
set "PACKAGE_MODEL=models"
|
||||||
|
) else (
|
||||||
|
echo Offline Whisper model already exists on the server.
|
||||||
|
)
|
||||||
|
|
||||||
|
echo [1/5] Building WEB and ADMIN...
|
||||||
|
call npm --prefix WEB run build
|
||||||
|
if errorlevel 1 exit /b 1
|
||||||
|
call npm --prefix ADMIN run build
|
||||||
|
if errorlevel 1 exit /b 1
|
||||||
|
|
||||||
|
echo [2/5] Building Linux amd64 API...
|
||||||
|
pushd API
|
||||||
|
set "GOOS=linux"
|
||||||
|
set "GOARCH=amd64"
|
||||||
|
go build -trimpath -ldflags "-s -w" -o bin/jcf-api ./cmd/api
|
||||||
|
if errorlevel 1 (popd & exit /b 1)
|
||||||
|
popd
|
||||||
|
|
||||||
|
echo [3/5] Packaging deployment files...
|
||||||
|
tar --exclude=.git --exclude=tools/ffmpeg --exclude=deploy/.deploy.conf -C API -czf "%ARCHIVE%" bin static deploy migrations acme workers %PACKAGE_MODEL% %PACKAGE_ENV%
|
||||||
|
if errorlevel 1 exit /b 1
|
||||||
|
|
||||||
|
echo [4/5] Uploading and staging deployment files...
|
||||||
|
scp "%ARCHIVE%" "%REMOTE%:%REMOTE_ARCHIVE%"
|
||||||
|
if errorlevel 1 exit /b 1
|
||||||
|
ssh "%REMOTE%" "bash -lc 'set -e; WORK=/tmp/juyou-deploy-work; ENV_UPLOAD=%REMOTE_ENV%; cleanup(){ rm -rf $WORK %REMOTE_ARCHIVE%; }; trap cleanup EXIT; rm -rf $WORK; mkdir -p $WORK; tar -xzf %REMOTE_ARCHIVE% -C $WORK; if [ -f $WORK/.env ]; then install -m 600 $WORK/.env $ENV_UPLOAD; else rm -f $ENV_UPLOAD; fi; rm -f $WORK/.env; mkdir -p %REMOTE_ROOT%; if command -v rsync >/dev/null 2>&1; then rsync -a --delete --exclude=/.env --exclude=/media/ --exclude=/backups/ --exclude=/models/ $WORK/ %REMOTE_ROOT%/; if [ -d $WORK/models ]; then mkdir -p %REMOTE_ROOT%/models; rsync -a $WORK/models/ %REMOTE_ROOT%/models/; fi; else cp -a $WORK/. %REMOTE_ROOT%/; fi; echo upload complete; echo run manually: bash %REMOTE_ROOT%/deploy/one_click_deployment.sh'"
|
||||||
|
if errorlevel 1 exit /b 1
|
||||||
|
|
||||||
|
echo [5/5] Cleaning local temporary archive...
|
||||||
|
del "%ARCHIVE%" >nul 2>nul
|
||||||
|
echo Deployment push complete.
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# 一键更新本地 acme.sh。
|
||||||
|
# 使用 jsDelivr CDN 获取 acmesh-official/acme.sh 的 master 最新版本，
|
||||||
|
# 不执行 GitHub clone，也不依赖 GitHub 下载地址；脚本保存到 API/acme/。
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
API_ROOT="$(realpath "$SCRIPT_DIR/..")"
|
||||||
|
ACME_DIR="$API_ROOT/acme"
|
||||||
|
ACME_FILE="$ACME_DIR/acme.sh"
|
||||||
|
SOURCE_URL="https://cdn.jsdelivr.net/gh/acmesh-official/acme.sh@master/acme.sh"
|
||||||
|
mkdir -p "$ACME_DIR"
|
||||||
|
TEMP_FILE="$(mktemp "$ACME_DIR/acme.sh.XXXXXX")"
|
||||||
|
trap 'rm -f "$TEMP_FILE"' EXIT
|
||||||
|
|
||||||
|
command -v curl >/dev/null 2>&1 || { echo "缺少 curl，请先安装 curl。" >&2; exit 1; }
|
||||||
|
echo "正在从非 GitHub CDN 下载 acme.sh 最新版本..."
|
||||||
|
curl -fsSL --retry 3 --connect-timeout 15 "$SOURCE_URL" -o "$TEMP_FILE"
|
||||||
|
grep -q '^VER=' "$TEMP_FILE" || { echo "下载内容不是有效的 acme.sh 文件。" >&2; exit 1; }
|
||||||
|
chmod 755 "$TEMP_FILE"
|
||||||
|
mv -f "$TEMP_FILE" "$ACME_FILE"
|
||||||
|
echo "acme.sh 已更新到：$ACME_FILE"
|
||||||
|
grep '^VER=' "$ACME_FILE" | head -n 1
|
||||||
+79
@@ -0,0 +1,79 @@
|
|||||||
|
module juhe-factory/api
|
||||||
|
|
||||||
|
go 1.26.0
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/aws/aws-sdk-go-v2 v1.41.1
|
||||||
|
github.com/aws/aws-sdk-go-v2/config v1.32.7
|
||||||
|
github.com/aws/aws-sdk-go-v2/credentials v1.19.7
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/s3 v1.95.1
|
||||||
|
github.com/gin-gonic/gin v1.12.0
|
||||||
|
github.com/golang-jwt/jwt/v5 v5.3.0
|
||||||
|
github.com/google/uuid v1.6.0
|
||||||
|
github.com/hibiken/asynq v0.26.0
|
||||||
|
github.com/joho/godotenv v1.5.1
|
||||||
|
github.com/redis/go-redis/v9 v9.21.0
|
||||||
|
github.com/richardlehane/mscfb v1.0.7
|
||||||
|
golang.org/x/crypto v0.48.0
|
||||||
|
golang.org/x/text v0.34.0
|
||||||
|
gorm.io/driver/postgres v1.6.0
|
||||||
|
gorm.io/gorm v1.31.2
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.7.4 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.18.17 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/configsources v1.4.17 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.7.17 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/ini v1.8.4 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/v4a v1.4.17 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.13.4 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/checksum v1.9.8 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.13.17 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/s3shared v1.19.17 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/signin v1.0.5 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/sso v1.30.9 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/ssooidc v1.35.13 // indirect
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/sts v1.41.6 // indirect
|
||||||
|
github.com/aws/smithy-go v1.24.0 // indirect
|
||||||
|
github.com/bytedance/gopkg v0.1.3 // indirect
|
||||||
|
github.com/bytedance/sonic v1.15.0 // indirect
|
||||||
|
github.com/bytedance/sonic/loader v0.5.0 // indirect
|
||||||
|
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||||
|
github.com/cloudwego/base64x v0.1.6 // indirect
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.12 // indirect
|
||||||
|
github.com/gin-contrib/sse v1.1.0 // indirect
|
||||||
|
github.com/go-playground/locales v0.14.1 // indirect
|
||||||
|
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||||
|
github.com/go-playground/validator/v10 v10.30.1 // indirect
|
||||||
|
github.com/goccy/go-json v0.10.5 // indirect
|
||||||
|
github.com/goccy/go-yaml v1.19.2 // indirect
|
||||||
|
github.com/jackc/pgpassfile v1.0.0 // indirect
|
||||||
|
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
|
||||||
|
github.com/jackc/pgx/v5 v5.6.0 // indirect
|
||||||
|
github.com/jackc/puddle/v2 v2.2.2 // indirect
|
||||||
|
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||||
|
github.com/jinzhu/now v1.1.5 // indirect
|
||||||
|
github.com/json-iterator/go v1.1.12 // indirect
|
||||||
|
github.com/klauspost/cpuid/v2 v2.3.0 // indirect
|
||||||
|
github.com/leodido/go-urn v1.4.0 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||||
|
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.4 // indirect
|
||||||
|
github.com/quic-go/qpack v0.6.0 // indirect
|
||||||
|
github.com/quic-go/quic-go v0.59.0 // indirect
|
||||||
|
github.com/richardlehane/msoleps v1.0.3 // indirect
|
||||||
|
github.com/robfig/cron/v3 v3.0.1 // indirect
|
||||||
|
github.com/spf13/cast v1.10.0 // indirect
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
||||||
|
github.com/ugorji/go/codec v1.3.1 // indirect
|
||||||
|
go.mongodb.org/mongo-driver/v2 v2.5.0 // indirect
|
||||||
|
go.uber.org/atomic v1.11.0 // indirect
|
||||||
|
golang.org/x/arch v0.22.0 // indirect
|
||||||
|
golang.org/x/net v0.51.0 // indirect
|
||||||
|
golang.org/x/sync v0.19.0 // indirect
|
||||||
|
golang.org/x/sys v0.41.0 // indirect
|
||||||
|
golang.org/x/time v0.14.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.36.10 // indirect
|
||||||
|
)
|
||||||
+190
@@ -0,0 +1,190 @@
|
|||||||
|
github.com/aws/aws-sdk-go-v2 v1.41.1 h1:ABlyEARCDLN034NhxlRUSZr4l71mh+T5KAeGh6cerhU=
|
||||||
|
github.com/aws/aws-sdk-go-v2 v1.41.1/go.mod h1:MayyLB8y+buD9hZqkCW3kX1AKq07Y5pXxtgB+rRFhz0=
|
||||||
|
github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.7.4 h1:489krEF9xIGkOaaX3CE/Be2uWjiXrkCH6gUX+bZA/BU=
|
||||||
|
github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.7.4/go.mod h1:IOAPF6oT9KCsceNTvvYMNHy0+kMF8akOjeDvPENWxp4=
|
||||||
|
github.com/aws/aws-sdk-go-v2/config v1.32.7 h1:vxUyWGUwmkQ2g19n7JY/9YL8MfAIl7bTesIUykECXmY=
|
||||||
|
github.com/aws/aws-sdk-go-v2/config v1.32.7/go.mod h1:2/Qm5vKUU/r7Y+zUk/Ptt2MDAEKAfUtKc1+3U1Mo3oY=
|
||||||
|
github.com/aws/aws-sdk-go-v2/credentials v1.19.7 h1:tHK47VqqtJxOymRrNtUXN5SP/zUTvZKeLx4tH6PGQc8=
|
||||||
|
github.com/aws/aws-sdk-go-v2/credentials v1.19.7/go.mod h1:qOZk8sPDrxhf+4Wf4oT2urYJrYt3RejHSzgAquYeppw=
|
||||||
|
github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.18.17 h1:I0GyV8wiYrP8XpA70g1HBcQO1JlQxCMTW9npl5UbDHY=
|
||||||
|
github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.18.17/go.mod h1:tyw7BOl5bBe/oqvoIeECFJjMdzXoa/dfVz3QQ5lgHGA=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/configsources v1.4.17 h1:xOLELNKGp2vsiteLsvLPwxC+mYmO6OZ8PYgiuPJzF8U=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/configsources v1.4.17/go.mod h1:5M5CI3D12dNOtH3/mk6minaRwI2/37ifCURZISxA/IQ=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.7.17 h1:WWLqlh79iO48yLkj1v3ISRNiv+3KdQoZ6JWyfcsyQik=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.7.17/go.mod h1:EhG22vHRrvF8oXSTYStZhJc1aUgKtnJe+aOiFEV90cM=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/ini v1.8.4 h1:WKuaxf++XKWlHWu9ECbMlha8WOEGm0OUEZqm4K/Gcfk=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/ini v1.8.4/go.mod h1:ZWy7j6v1vWGmPReu0iSGvRiise4YI5SkR3OHKTZ6Wuc=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/v4a v1.4.17 h1:JqcdRG//czea7Ppjb+g/n4o8i/R50aTBHkA7vu0lK+k=
|
||||||
|
github.com/aws/aws-sdk-go-v2/internal/v4a v1.4.17/go.mod h1:CO+WeGmIdj/MlPel2KwID9Gt7CNq4M65HUfBW97liM0=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.13.4 h1:0ryTNEdJbzUCEWkVXEXoqlXV72J5keC1GvILMOuD00E=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.13.4/go.mod h1:HQ4qwNZh32C3CBeO6iJLQlgtMzqeG17ziAA/3KDJFow=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/checksum v1.9.8 h1:Z5EiPIzXKewUQK0QTMkutjiaPVeVYXX7KIqhXu/0fXs=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/checksum v1.9.8/go.mod h1:FsTpJtvC4U1fyDXk7c71XoDv3HlRm8V3NiYLeYLh5YE=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.13.17 h1:RuNSMoozM8oXlgLG/n6WLaFGoea7/CddrCfIiSA+xdY=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.13.17/go.mod h1:F2xxQ9TZz5gDWsclCtPQscGpP0VUOc8RqgFM3vDENmU=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/s3shared v1.19.17 h1:bGeHBsGZx0Dvu/eJC0Lh9adJa3M1xREcndxLNZlve2U=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/internal/s3shared v1.19.17/go.mod h1:dcW24lbU0CzHusTE8LLHhRLI42ejmINN8Lcr22bwh/g=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/s3 v1.95.1 h1:C2dUPSnEpy4voWFIq3JNd8gN0Y5vYGDo44eUE58a/p8=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/s3 v1.95.1/go.mod h1:5jggDlZ2CLQhwJBiZJb4vfk4f0GxWdEDruWKEJ1xOdo=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/signin v1.0.5 h1:VrhDvQib/i0lxvr3zqlUwLwJP4fpmpyD9wYG1vfSu+Y=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/signin v1.0.5/go.mod h1:k029+U8SY30/3/ras4G/Fnv/b88N4mAfliNn08Dem4M=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/sso v1.30.9 h1:v6EiMvhEYBoHABfbGB4alOYmCIrcgyPPiBE1wZAEbqk=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/sso v1.30.9/go.mod h1:yifAsgBxgJWn3ggx70A3urX2AN49Y5sJTD1UQFlfqBw=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/ssooidc v1.35.13 h1:gd84Omyu9JLriJVCbGApcLzVR3XtmC4ZDPcAI6Ftvds=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/ssooidc v1.35.13/go.mod h1:sTGThjphYE4Ohw8vJiRStAcu3rbjtXRsdNB0TvZ5wwo=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/sts v1.41.6 h1:5fFjR/ToSOzB2OQ/XqWpZBmNvmP/pJ1jOWYlFDJTjRQ=
|
||||||
|
github.com/aws/aws-sdk-go-v2/service/sts v1.41.6/go.mod h1:qgFDZQSD/Kys7nJnVqYlWKnh0SSdMjAi0uSwON4wgYQ=
|
||||||
|
github.com/aws/smithy-go v1.24.0 h1:LpilSUItNPFr1eY85RYgTIg5eIEPtvFbskaFcmmIUnk=
|
||||||
|
github.com/aws/smithy-go v1.24.0/go.mod h1:LEj2LM3rBRQJxPZTB4KuzZkaZYnZPnvgIhb4pu07mx0=
|
||||||
|
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
|
||||||
|
github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c=
|
||||||
|
github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
|
||||||
|
github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0=
|
||||||
|
github.com/bytedance/gopkg v0.1.3 h1:TPBSwH8RsouGCBcMBktLt1AymVo2TVsBVCY4b6TnZ/M=
|
||||||
|
github.com/bytedance/gopkg v0.1.3/go.mod h1:576VvJ+eJgyCzdjS+c4+77QF3p7ubbtiKARP3TxducM=
|
||||||
|
github.com/bytedance/sonic v1.15.0 h1:/PXeWFaR5ElNcVE84U0dOHjiMHQOwNIx3K4ymzh/uSE=
|
||||||
|
github.com/bytedance/sonic v1.15.0/go.mod h1:tFkWrPz0/CUCLEF4ri4UkHekCIcdnkqXw9VduqpJh0k=
|
||||||
|
github.com/bytedance/sonic/loader v0.5.0 h1:gXH3KVnatgY7loH5/TkeVyXPfESoqSBSBEiDd5VjlgE=
|
||||||
|
github.com/bytedance/sonic/loader v0.5.0/go.mod h1:AR4NYCk5DdzZizZ5djGqQ92eEhCCcdf5x77udYiSJRo=
|
||||||
|
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||||
|
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
|
github.com/cloudwego/base64x v0.1.6 h1:t11wG9AECkCDk5fMSoxmufanudBtJ+/HemLstXDLI2M=
|
||||||
|
github.com/cloudwego/base64x v0.1.6/go.mod h1:OFcloc187FXDaYHvrNIjxSe8ncn0OOM8gEHfghB2IPU=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
|
||||||
|
github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.12 h1:e9hWvmLYvtp846tLHam2o++qitpguFiYCKbn0w9jyqw=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.12/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s=
|
||||||
|
github.com/gin-contrib/sse v1.1.0 h1:n0w2GMuUpWDVp7qSpvze6fAu9iRxJY4Hmj6AmBOU05w=
|
||||||
|
github.com/gin-contrib/sse v1.1.0/go.mod h1:hxRZ5gVpWMT7Z0B0gSNYqqsSCNIJMjzvm6fqCz9vjwM=
|
||||||
|
github.com/gin-gonic/gin v1.12.0 h1:b3YAbrZtnf8N//yjKeU2+MQsh2mY5htkZidOM7O0wG8=
|
||||||
|
github.com/gin-gonic/gin v1.12.0/go.mod h1:VxccKfsSllpKshkBWgVgRniFFAzFb9csfngsqANjnLc=
|
||||||
|
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
||||||
|
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
|
||||||
|
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
|
||||||
|
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
||||||
|
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
||||||
|
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||||
|
github.com/go-playground/validator/v10 v10.30.1 h1:f3zDSN/zOma+w6+1Wswgd9fLkdwy06ntQJp0BBvFG0w=
|
||||||
|
github.com/go-playground/validator/v10 v10.30.1/go.mod h1:oSuBIQzuJxL//3MelwSLD5hc2Tu889bF0Idm9Dg26cM=
|
||||||
|
github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4=
|
||||||
|
github.com/goccy/go-json v0.10.5/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
|
||||||
|
github.com/goccy/go-yaml v1.19.2 h1:PmFC1S6h8ljIz6gMRBopkjP1TVT7xuwrButHID66PoM=
|
||||||
|
github.com/goccy/go-yaml v1.19.2/go.mod h1:XBurs7gK8ATbW4ZPGKgcbrY1Br56PdM69F7LkFRi1kA=
|
||||||
|
github.com/golang-jwt/jwt/v5 v5.3.0 h1:pv4AsKCKKZuqlgs5sUmn4x8UlGa0kEVt/puTpKx9vvo=
|
||||||
|
github.com/golang-jwt/jwt/v5 v5.3.0/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE=
|
||||||
|
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||||
|
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
||||||
|
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
|
github.com/hibiken/asynq v0.26.0 h1:1Zxr92MlDnb1Zt/QR5g2vSCqUS03i95lUfqx5X7/wrw=
|
||||||
|
github.com/hibiken/asynq v0.26.0/go.mod h1:Qk4e57bTnWDoyJ67VkchuV6VzSM9IQW2nPvAGuDyw58=
|
||||||
|
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
|
||||||
|
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
|
||||||
|
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
|
||||||
|
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=
|
||||||
|
github.com/jackc/pgx/v5 v5.6.0 h1:SWJzexBzPL5jb0GEsrPMLIsi/3jOo7RHlzTjcAeDrPY=
|
||||||
|
github.com/jackc/pgx/v5 v5.6.0/go.mod h1:DNZ/vlrUnhWCoFGxHAG8U2ljioxukquj7utPDgtQdTw=
|
||||||
|
github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo=
|
||||||
|
github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
|
||||||
|
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
|
||||||
|
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
|
||||||
|
github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
|
||||||
|
github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
|
||||||
|
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
||||||
|
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
|
||||||
|
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||||
|
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.3.0 h1:S4CRMLnYUhGeDFDqkGriYKdfoFlDnMtqTiI/sFzhA9Y=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.3.0/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0=
|
||||||
|
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
|
||||||
|
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
|
||||||
|
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||||
|
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||||
|
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
|
||||||
|
github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
|
||||||
|
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||||
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
|
github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
|
||||||
|
github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
||||||
|
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8=
|
||||||
|
github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII=
|
||||||
|
github.com/quic-go/quic-go v0.59.0 h1:OLJkp1Mlm/aS7dpKgTc6cnpynnD2Xg7C1pwL6vy/SAw=
|
||||||
|
github.com/quic-go/quic-go v0.59.0/go.mod h1:upnsH4Ju1YkqpLXC305eW3yDZ4NfnNbmQRCMWS58IKU=
|
||||||
|
github.com/redis/go-redis/v9 v9.21.0 h1:FPBE4hhbAke+TLmcY3WkpbDffJEomdqPn3HYiqAtL9E=
|
||||||
|
github.com/redis/go-redis/v9 v9.21.0/go.mod h1:v/M13XI1PVCDcm01VtPFOADfZtHf8YW3baQf57KlIkA=
|
||||||
|
github.com/richardlehane/mscfb v1.0.7 h1:oeoiM0WE79vHwE8RpIYYvIAc8ajTH2mb6UZm55/+EB0=
|
||||||
|
github.com/richardlehane/mscfb v1.0.7/go.mod h1:pe0+IUIc0AHh0+teNzBlJCtSyZdFOGgV4ZK9bsoV+Jo=
|
||||||
|
github.com/richardlehane/msoleps v1.0.3 h1:aznSZzrwYRl3rLKRT3gUk9am7T/mLNSnJINvN0AQoVM=
|
||||||
|
github.com/richardlehane/msoleps v1.0.3/go.mod h1:BWev5JBpU9Ko2WAgmZEuiz4/u3ZYTKbjLycmwiWUfWg=
|
||||||
|
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
||||||
|
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
||||||
|
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||||
|
github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=
|
||||||
|
github.com/spf13/cast v1.10.0 h1:h2x0u2shc1QuLHfxi+cTJvs30+ZAHOGRic8uyGTDWxY=
|
||||||
|
github.com/spf13/cast v1.10.0/go.mod h1:jNfB8QC9IA6ZuY2ZjDp0KtFO2LZZlg4S/7bzP6qqeHo=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||||
|
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||||
|
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||||
|
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
|
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
|
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
|
||||||
|
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||||
|
github.com/ugorji/go/codec v1.3.1 h1:waO7eEiFDwidsBN6agj1vJQ4AG7lh2yqXyOXqhgQuyY=
|
||||||
|
github.com/ugorji/go/codec v1.3.1/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4=
|
||||||
|
github.com/zeebo/xxh3 v1.1.0 h1:s7DLGDK45Dyfg7++yxI0khrfwq9661w9EN78eP/UZVs=
|
||||||
|
github.com/zeebo/xxh3 v1.1.0/go.mod h1:IisAie1LELR4xhVinxWS5+zf1lA4p0MW4T+w+W07F5s=
|
||||||
|
go.mongodb.org/mongo-driver/v2 v2.5.0 h1:yXUhImUjjAInNcpTcAlPHiT7bIXhshCTL3jVBkF3xaE=
|
||||||
|
go.mongodb.org/mongo-driver/v2 v2.5.0/go.mod h1:yOI9kBsufol30iFsl1slpdq1I0eHPzybRWdyYUs8K/0=
|
||||||
|
go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE=
|
||||||
|
go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=
|
||||||
|
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
|
||||||
|
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
|
||||||
|
go.uber.org/mock v0.6.0 h1:hyF9dfmbgIX5EfOdasqLsWD6xqpNZlXblLB/Dbnwv3Y=
|
||||||
|
go.uber.org/mock v0.6.0/go.mod h1:KiVJ4BqZJaMj4svdfmHM0AUx4NJYO8ZNpPnZn1Z+BBU=
|
||||||
|
golang.org/x/arch v0.22.0 h1:c/Zle32i5ttqRXjdLyyHZESLD/bB90DCU1g9l/0YBDI=
|
||||||
|
golang.org/x/arch v0.22.0/go.mod h1:dNHoOeKiyja7GTvF9NJS1l3Z2yntpQNzgrjh1cU103A=
|
||||||
|
golang.org/x/crypto v0.48.0 h1:/VRzVqiRSggnhY7gNRxPauEQ5Drw9haKdM0jqfcCFts=
|
||||||
|
golang.org/x/crypto v0.48.0/go.mod h1:r0kV5h3qnFPlQnBSrULhlsRfryS2pmewsg+XfMgkVos=
|
||||||
|
golang.org/x/net v0.51.0 h1:94R/GTO7mt3/4wIKpcR5gkGmRLOuE/2hNGeWq/GBIFo=
|
||||||
|
golang.org/x/net v0.51.0/go.mod h1:aamm+2QF5ogm02fjy5Bb7CQ0WMt1/WVM7FtyaTLlA9Y=
|
||||||
|
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||||
|
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||||
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
|
||||||
|
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
|
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
|
||||||
|
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
|
||||||
|
golang.org/x/time v0.14.0 h1:MRx4UaLrDotUKUdCIqzPC48t1Y9hANFKIRpNx+Te8PI=
|
||||||
|
golang.org/x/time v0.14.0/go.mod h1:eL/Oa2bBBK0TkX57Fyni+NgnyQQN4LitPmob2Hjnqw4=
|
||||||
|
google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE=
|
||||||
|
google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gorm.io/driver/postgres v1.6.0 h1:2dxzU8xJ+ivvqTRph34QX+WrRaJlmfyPqXmoGVjMBa4=
|
||||||
|
gorm.io/driver/postgres v1.6.0/go.mod h1:vUw0mrGgrTK+uPHEhAdV4sfFELrByKVGnaVRkXDhtWo=
|
||||||
|
gorm.io/driver/sqlite v1.6.0 h1:WHRRrIiulaPiPFmDcod6prc4l2VGVWHz80KspNsxSfQ=
|
||||||
|
gorm.io/driver/sqlite v1.6.0/go.mod h1:AO9V1qIQddBESngQUKWL9yoH93HIeA1X6V633rBwyT8=
|
||||||
|
gorm.io/gorm v1.31.2 h1:3o8FXNo9v9S858gil+3LlZA1LkCOzgb4g5BL64FgaCo=
|
||||||
|
gorm.io/gorm v1.31.2/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs=
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
package billing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
// PrechargeGenerationTask creates a task and reserves its full quoted cost in
|
||||||
|
// the same transaction. A zero price is allowed only when the model has an
|
||||||
|
// explicit zero-price record; callers must reject a missing price record.
|
||||||
|
func PrechargeGenerationTask(tx *gorm.DB, task *model.GenerationTask, unitPrice string, quantity int, remark string) error {
|
||||||
|
price, err := strconv.ParseFloat(unitPrice, 64)
|
||||||
|
if err != nil || price < 0 || quantity <= 0 {
|
||||||
|
return errors.New("生成任务价格无效")
|
||||||
|
}
|
||||||
|
var amount string
|
||||||
|
if err := tx.Raw("SELECT round(?::numeric * ?::numeric,2)::text", unitPrice, quantity).Scan(&amount).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if amount == "" {
|
||||||
|
return errors.New("生成任务价格无效")
|
||||||
|
}
|
||||||
|
task.EstimatedPoints = amount
|
||||||
|
task.PrepaidPoints = amount
|
||||||
|
if task.ID == uuid.Nil {
|
||||||
|
task.ID = uuid.New()
|
||||||
|
}
|
||||||
|
|
||||||
|
if amountValue, _ := strconv.ParseFloat(amount, 64); amountValue > 0 {
|
||||||
|
if _, err := DebitPoints(tx, task.UserID, amount, "generation_hold", task.ID.String(), remark, "generation:hold:"+task.ID.String()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := tx.Create(task).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RefundGenerationTask refunds a prepaid task exactly once. The caller must
|
||||||
|
// hold a row lock on task inside the same transaction.
|
||||||
|
func RefundGenerationTask(tx *gorm.DB, task *model.GenerationTask, remark string) (bool, error) {
|
||||||
|
amount, _ := strconv.ParseFloat(task.PrepaidPoints, 64)
|
||||||
|
if task.CostRefunded || amount <= 0 {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
refunded, err := RefundDebit(tx, task.UserID, "generation:hold:"+task.ID.String(), "generation:refund:"+task.ID.String(),
|
||||||
|
"generation_refund", task.ID.String(), remark)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
task.CostRefunded = true
|
||||||
|
return refunded, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,250 @@
|
|||||||
|
// 积分账务模块，负责积分入账、扣减、退款及批次分配记录维护。
|
||||||
|
package billing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ErrInsufficientPoints = errors.New("积分余额不足，请前往充值")
|
||||||
|
|
||||||
|
// pointGrant 表示参与扣减的可用积分批次。
|
||||||
|
type pointGrant struct {
|
||||||
|
ID uuid.UUID
|
||||||
|
AvailableAmount string
|
||||||
|
}
|
||||||
|
|
||||||
|
// pointAllocation 表示原扣减流水关联的积分批次和分配金额。
|
||||||
|
type pointAllocation struct {
|
||||||
|
GrantID uuid.UUID
|
||||||
|
Amount string
|
||||||
|
}
|
||||||
|
|
||||||
|
// parsePointCents 将最多两位小数的积分字符串转换为整数分值。
|
||||||
|
func parsePointCents(value string) (int64, error) {
|
||||||
|
value = strings.TrimSpace(value)
|
||||||
|
if value == "" {
|
||||||
|
return 0, errors.New("积分数量为空")
|
||||||
|
}
|
||||||
|
parts := strings.Split(value, ".")
|
||||||
|
if len(parts) > 2 || strings.HasPrefix(parts[0], "-") {
|
||||||
|
return 0, fmt.Errorf("无效积分数量 %q", value)
|
||||||
|
}
|
||||||
|
whole, err := strconv.ParseInt(parts[0], 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("无效积分数量 %q", value)
|
||||||
|
}
|
||||||
|
fraction := ""
|
||||||
|
if len(parts) == 2 {
|
||||||
|
fraction = parts[1]
|
||||||
|
}
|
||||||
|
if len(fraction) > 2 {
|
||||||
|
if strings.Trim(fraction[2:], "0") != "" {
|
||||||
|
return 0, fmt.Errorf("积分数量最多保留两位小数")
|
||||||
|
}
|
||||||
|
fraction = fraction[:2]
|
||||||
|
}
|
||||||
|
fraction += strings.Repeat("0", 2-len(fraction))
|
||||||
|
fractionValue, err := strconv.ParseInt(fraction, 10, 64)
|
||||||
|
if err != nil || whole > (int64(^uint64(0)>>1)-fractionValue)/100 {
|
||||||
|
return 0, fmt.Errorf("无效积分数量 %q", value)
|
||||||
|
}
|
||||||
|
return whole*100 + fractionValue, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// formatPointCents 将整数分值格式化为两位小数的积分字符串。
|
||||||
|
func formatPointCents(value int64) string {
|
||||||
|
return fmt.Sprintf("%d.%02d", value/100, value%100)
|
||||||
|
}
|
||||||
|
|
||||||
|
// lockUser 锁定用户积分余额，避免并发账务操作导致余额和批次不一致。
|
||||||
|
func lockUser(tx *gorm.DB, userID uuid.UUID, requireActive bool) (string, error) {
|
||||||
|
var balance string
|
||||||
|
userQuery := "SELECT point_balance::text FROM web_users WHERE id=? FOR UPDATE"
|
||||||
|
if requireActive {
|
||||||
|
userQuery = "SELECT point_balance::text FROM web_users WHERE id=? AND enabled=true AND deleted_at IS NULL FOR UPDATE"
|
||||||
|
}
|
||||||
|
if err := tx.Raw(userQuery, userID).Scan(&balance).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if balance == "" {
|
||||||
|
return "", gorm.ErrRecordNotFound
|
||||||
|
}
|
||||||
|
return balance, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreditPoints 创建永久有效的积分批次，并原子写入用户余额和积分流水。
|
||||||
|
func CreditPoints(tx *gorm.DB, userID uuid.UUID, sourceType, sourceID, points, remark string) (string, bool, error) {
|
||||||
|
amount, err := parsePointCents(points)
|
||||||
|
if err != nil || amount <= 0 {
|
||||||
|
return "", false, errors.New("入账积分必须大于零")
|
||||||
|
}
|
||||||
|
balance, err := lockUser(tx, userID, true)
|
||||||
|
if err != nil {
|
||||||
|
return "", false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
grantID := uuid.New()
|
||||||
|
result := tx.Exec(`INSERT INTO point_grants(id,user_id,source_type,source_id,granted_amount,available_amount)
|
||||||
|
VALUES(?,?,?,?,?::numeric,?::numeric) ON CONFLICT(source_type,source_id) DO NOTHING`,
|
||||||
|
grantID, userID, sourceType, sourceID, points, points)
|
||||||
|
if result.Error != nil {
|
||||||
|
return "", false, result.Error
|
||||||
|
}
|
||||||
|
if result.RowsAffected == 0 {
|
||||||
|
return balance, false, nil
|
||||||
|
}
|
||||||
|
if err := tx.Raw(`UPDATE web_users SET point_balance=point_balance+?::numeric
|
||||||
|
WHERE id=? RETURNING point_balance::text`, points, userID).Scan(&balance).Error; err != nil {
|
||||||
|
return "", false, err
|
||||||
|
}
|
||||||
|
if err := tx.Exec(`INSERT INTO point_ledger(user_id,change_amount,balance_after,business_type,business_id,remark,idempotency_key)
|
||||||
|
VALUES(?,?::numeric,?::numeric,?,?,?,?)`, userID, points, balance, sourceType, sourceID, remark, sourceType+":"+sourceID).Error; err != nil {
|
||||||
|
return "", false, err
|
||||||
|
}
|
||||||
|
return balance, true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreditRedemptionPoints 将兑换码对应积分以永久有效批次计入用户账户。
|
||||||
|
func CreditRedemptionPoints(tx *gorm.DB, userID, codeID uuid.UUID, points string) (string, bool, error) {
|
||||||
|
return CreditPoints(tx, userID, "redemption", codeID.String(), points, "兑换码兑换")
|
||||||
|
}
|
||||||
|
|
||||||
|
// DebitPoints 按批次创建顺序扣减积分，并记录分配明细供后续退款恢复。
|
||||||
|
func DebitPoints(tx *gorm.DB, userID uuid.UUID, points, businessType, businessID, remark, idempotencyKey string) (string, error) {
|
||||||
|
requested, err := parsePointCents(points)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
balance, err := lockUser(tx, userID, true)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if requested == 0 {
|
||||||
|
return balance, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var grants []pointGrant
|
||||||
|
if err := tx.Raw(`SELECT id,available_amount::text AS available_amount
|
||||||
|
FROM point_grants WHERE user_id=? AND available_amount>0
|
||||||
|
ORDER BY created_at,id FOR UPDATE`, userID).Scan(&grants).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
remaining := requested
|
||||||
|
type usedGrant struct {
|
||||||
|
id uuid.UUID
|
||||||
|
amount int64
|
||||||
|
}
|
||||||
|
used := make([]usedGrant, 0, len(grants))
|
||||||
|
for _, grant := range grants {
|
||||||
|
available, parseErr := parsePointCents(grant.AvailableAmount)
|
||||||
|
if parseErr != nil {
|
||||||
|
return "", parseErr
|
||||||
|
}
|
||||||
|
use := available
|
||||||
|
if use > remaining {
|
||||||
|
use = remaining
|
||||||
|
}
|
||||||
|
if use > 0 {
|
||||||
|
used = append(used, usedGrant{id: grant.ID, amount: use})
|
||||||
|
remaining -= use
|
||||||
|
}
|
||||||
|
if remaining == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if remaining > 0 {
|
||||||
|
return "", ErrInsufficientPoints
|
||||||
|
}
|
||||||
|
for _, allocation := range used {
|
||||||
|
if err := tx.Exec(`UPDATE point_grants SET available_amount=available_amount-?::numeric WHERE id=?`,
|
||||||
|
formatPointCents(allocation.amount), allocation.id).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := tx.Raw(`UPDATE web_users SET point_balance=point_balance-?::numeric
|
||||||
|
WHERE id=? RETURNING point_balance::text`, points, userID).Scan(&balance).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
var ledgerID int64
|
||||||
|
if err := tx.Raw(`INSERT INTO point_ledger(user_id,change_amount,balance_after,business_type,business_id,remark,idempotency_key)
|
||||||
|
VALUES(?,(-?::numeric),?::numeric,?,?,?,?) RETURNING id`,
|
||||||
|
userID, points, balance, businessType, businessID, remark, idempotencyKey).Scan(&ledgerID).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
for _, allocation := range used {
|
||||||
|
if err := tx.Exec(`INSERT INTO point_ledger_allocations(ledger_id,grant_id,allocation_type,amount)
|
||||||
|
VALUES(?,?,'consume',?::numeric)`, ledgerID, allocation.id, formatPointCents(allocation.amount)).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return balance, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RefundDebit 将原扣减记录恢复到对应的永久积分批次，并写入退款流水。
|
||||||
|
func RefundDebit(tx *gorm.DB, userID uuid.UUID, originalKey, refundKey, businessType, businessID, remark string) (bool, error) {
|
||||||
|
balance, err := lockUser(tx, userID, false)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
var existing int64
|
||||||
|
if err := tx.Raw("SELECT count(*) FROM point_ledger WHERE idempotency_key=?", refundKey).Scan(&existing).Error; err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
if existing > 0 {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var allocations []pointAllocation
|
||||||
|
if err := tx.Raw(`SELECT a.grant_id,a.amount::text AS amount
|
||||||
|
FROM point_ledger original
|
||||||
|
JOIN point_ledger_allocations a ON a.ledger_id=original.id AND a.allocation_type='consume'
|
||||||
|
JOIN point_grants g ON g.id=a.grant_id
|
||||||
|
WHERE original.idempotency_key=? ORDER BY g.created_at,g.id FOR UPDATE OF g`, originalKey).Scan(&allocations).Error; err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
if len(allocations) == 0 {
|
||||||
|
return false, errors.New("未找到原积分扣减分配记录")
|
||||||
|
}
|
||||||
|
restored := int64(0)
|
||||||
|
restoredAllocations := make([]pointAllocation, 0, len(allocations))
|
||||||
|
for _, allocation := range allocations {
|
||||||
|
amount, parseErr := parsePointCents(allocation.Amount)
|
||||||
|
if parseErr != nil {
|
||||||
|
return false, parseErr
|
||||||
|
}
|
||||||
|
result := tx.Exec(`UPDATE point_grants SET available_amount=available_amount+?::numeric
|
||||||
|
WHERE id=?`, allocation.Amount, allocation.GrantID)
|
||||||
|
if result.Error != nil {
|
||||||
|
return false, result.Error
|
||||||
|
}
|
||||||
|
if result.RowsAffected > 0 {
|
||||||
|
restored += amount
|
||||||
|
restoredAllocations = append(restoredAllocations, allocation)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if restored > 0 {
|
||||||
|
if err := tx.Raw(`UPDATE web_users SET point_balance=point_balance+?::numeric
|
||||||
|
WHERE id=? RETURNING point_balance::text`, formatPointCents(restored), userID).Scan(&balance).Error; err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var refundLedgerID int64
|
||||||
|
if err := tx.Raw(`INSERT INTO point_ledger(user_id,change_amount,balance_after,business_type,business_id,remark,idempotency_key)
|
||||||
|
VALUES(?,?::numeric,?::numeric,?,?,?,?) RETURNING id`, userID, formatPointCents(restored), balance,
|
||||||
|
businessType, businessID, remark, refundKey).Scan(&refundLedgerID).Error; err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
for _, allocation := range restoredAllocations {
|
||||||
|
if err := tx.Exec(`INSERT INTO point_ledger_allocations(ledger_id,grant_id,allocation_type,amount)
|
||||||
|
VALUES(?,?,'refund',?::numeric)`, refundLedgerID, allocation.GrantID, allocation.Amount).Error; err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
// 积分账务单元测试，验证积分精度解析和非法输入处理。
|
||||||
|
package billing
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
// TestParsePointCents 验证积分字符串能够稳定转换为整数分值。
|
||||||
|
func TestParsePointCents(t *testing.T) {
|
||||||
|
tests := map[string]int64{
|
||||||
|
"0": 0,
|
||||||
|
"1": 100,
|
||||||
|
"1.2": 120,
|
||||||
|
"1.23": 123,
|
||||||
|
"1.2300": 123,
|
||||||
|
"999.99": 99999,
|
||||||
|
}
|
||||||
|
for input, expected := range tests {
|
||||||
|
actual, err := parsePointCents(input)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("parse %q: %v", input, err)
|
||||||
|
}
|
||||||
|
if actual != expected {
|
||||||
|
t.Fatalf("parse %q: expected %d, got %d", input, expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, input := range []string{"", "-1", "1.234", "abc"} {
|
||||||
|
if _, err := parsePointCents(input); err == nil {
|
||||||
|
t.Fatalf("expected %q to be rejected", input)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,321 @@
|
|||||||
|
package billing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
|
"math/big"
|
||||||
|
"strings"
|
||||||
|
"unicode"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
TextBillingPerRequest = "per_request"
|
||||||
|
TextBillingPerToken = "per_token"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TextPricing struct {
|
||||||
|
Mode string `json:"mode"`
|
||||||
|
PerRequest string `json:"per_request,omitempty"`
|
||||||
|
InputPerM string `json:"input_per_m,omitempty"`
|
||||||
|
OutputPerM string `json:"output_per_m,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type TextUsage struct {
|
||||||
|
Input int64
|
||||||
|
Output int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateTextGenerationTask(tx *gorm.DB, task *model.GenerationTask, pricing TextPricing, remark string) error {
|
||||||
|
if err := pricing.Validate(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
task.BillingSnapshot = pricing.MarshalSnapshot()
|
||||||
|
if pricing.Mode == TextBillingPerRequest {
|
||||||
|
return PrechargeGenerationTask(tx, task, pricing.PerRequest, 1, remark)
|
||||||
|
}
|
||||||
|
task.EstimatedPoints = "0.00"
|
||||||
|
task.PrepaidPoints = "0.00"
|
||||||
|
if task.ID == uuid.Nil {
|
||||||
|
task.ID = uuid.New()
|
||||||
|
}
|
||||||
|
return tx.Create(task).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
func ChargeTextRequest(tx *gorm.DB, userID uuid.UUID, referenceID string, pricing TextPricing, remark string) (string, error) {
|
||||||
|
if pricing.Mode != TextBillingPerRequest {
|
||||||
|
return "", errors.New("文本模型不是按次计费")
|
||||||
|
}
|
||||||
|
businessID := textCallBusinessID(referenceID, "request")
|
||||||
|
idempotencyKey := "generation:hold:" + businessID
|
||||||
|
if existing, found, err := debitAmount(tx, idempotencyKey); err != nil {
|
||||||
|
return "", err
|
||||||
|
} else if found {
|
||||||
|
return existing, nil
|
||||||
|
}
|
||||||
|
if pricing.PerRequest == "0" || pricing.PerRequest == "0.00" || pricing.PerRequest == "0.0000" {
|
||||||
|
return "0.00", nil
|
||||||
|
}
|
||||||
|
_, err := DebitPoints(tx, userID, pricing.PerRequest, "generation_hold", businessID, remark, idempotencyKey)
|
||||||
|
return pricing.PerRequest, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func RefundTextGenerationTask(tx *gorm.DB, task *model.GenerationTask, remark string) (bool, error) {
|
||||||
|
pricing, err := ParseTextPricingSnapshot(task.BillingSnapshot)
|
||||||
|
if err != nil || pricing.Mode == TextBillingPerRequest {
|
||||||
|
return RefundGenerationTask(tx, task, remark)
|
||||||
|
}
|
||||||
|
if task.CostRefunded {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
refunded, err := RefundTextCalls(tx, task.UserID, task.ID.String(), remark)
|
||||||
|
if err == nil {
|
||||||
|
task.CostRefunded = true
|
||||||
|
}
|
||||||
|
return refunded, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadTextPricing(db *gorm.DB, modelID uuid.UUID) (TextPricing, error) {
|
||||||
|
var pricing TextPricing
|
||||||
|
err := db.Table("models m").
|
||||||
|
Select(`coalesce(m.text_billing_mode,'') AS mode,
|
||||||
|
coalesce(max(CASE WHEN lower(p.price_key)='default' THEN p.price::text END),'') AS per_request,
|
||||||
|
coalesce(max(CASE WHEN lower(p.price_key)='input' THEN p.price::text END),'') AS input_per_m,
|
||||||
|
coalesce(max(CASE WHEN lower(p.price_key)='output' THEN p.price::text END),'') AS output_per_m`).
|
||||||
|
Joins("LEFT JOIN model_prices p ON p.model_id=m.id").
|
||||||
|
Where("m.id=? AND m.model_type='text' AND m.enabled=true AND m.deleted_at IS NULL", modelID).
|
||||||
|
Group("m.id,m.text_billing_mode").
|
||||||
|
Take(&pricing).Error
|
||||||
|
if err != nil {
|
||||||
|
return pricing, err
|
||||||
|
}
|
||||||
|
if err := pricing.Validate(); err != nil {
|
||||||
|
return pricing, err
|
||||||
|
}
|
||||||
|
return pricing, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pricing TextPricing) Validate() error {
|
||||||
|
switch pricing.Mode {
|
||||||
|
case TextBillingPerRequest:
|
||||||
|
if !validNonNegativeDecimal(pricing.PerRequest) {
|
||||||
|
return errors.New("文本模型未配置按次价格")
|
||||||
|
}
|
||||||
|
case TextBillingPerToken:
|
||||||
|
if !validNonNegativeDecimal(pricing.InputPerM) || !validNonNegativeDecimal(pricing.OutputPerM) {
|
||||||
|
return errors.New("文本模型未完整配置输入、输出 Token 价格")
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return errors.New("文本模型计费模式无效")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pricing TextPricing) MarshalSnapshot() json.RawMessage {
|
||||||
|
data, _ := json.Marshal(pricing)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseTextPricingSnapshot(value json.RawMessage) (TextPricing, error) {
|
||||||
|
var pricing TextPricing
|
||||||
|
if len(value) == 0 || string(value) == "{}" {
|
||||||
|
return pricing, errors.New("文本模型计费快照缺失")
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(value, &pricing); err != nil {
|
||||||
|
return pricing, errors.New("文本模型计费快照无效")
|
||||||
|
}
|
||||||
|
return pricing, pricing.Validate()
|
||||||
|
}
|
||||||
|
|
||||||
|
func CalculateTextPoints(pricing TextPricing, usage TextUsage) (string, error) {
|
||||||
|
if pricing.Mode != TextBillingPerToken || usage.Input < 0 || usage.Output < 0 {
|
||||||
|
return "", errors.New("Token 计费参数无效")
|
||||||
|
}
|
||||||
|
inputRate, ok := new(big.Rat).SetString(pricing.InputPerM)
|
||||||
|
if !ok || inputRate.Sign() < 0 {
|
||||||
|
return "", errors.New("输入 Token 价格无效")
|
||||||
|
}
|
||||||
|
outputRate, ok := new(big.Rat).SetString(pricing.OutputPerM)
|
||||||
|
if !ok || outputRate.Sign() < 0 {
|
||||||
|
return "", errors.New("输出 Token 价格无效")
|
||||||
|
}
|
||||||
|
amount := new(big.Rat).Mul(inputRate, new(big.Rat).SetInt64(usage.Input))
|
||||||
|
amount.Add(amount, new(big.Rat).Mul(outputRate, new(big.Rat).SetInt64(usage.Output)))
|
||||||
|
amount.Quo(amount, big.NewRat(1000000, 1))
|
||||||
|
return ceilPointCents(amount), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func EstimateTextTokens(value string) int64 {
|
||||||
|
var tokens int64
|
||||||
|
ascii := 0
|
||||||
|
flush := func() {
|
||||||
|
if ascii > 0 {
|
||||||
|
tokens += int64((ascii + 3) / 4)
|
||||||
|
ascii = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, r := range value {
|
||||||
|
if r <= unicode.MaxASCII {
|
||||||
|
ascii++
|
||||||
|
} else {
|
||||||
|
flush()
|
||||||
|
tokens++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
flush()
|
||||||
|
if tokens == 0 && value != "" {
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return tokens
|
||||||
|
}
|
||||||
|
|
||||||
|
func ReserveTextCall(tx *gorm.DB, userID uuid.UUID, referenceID, callKey string, pricing TextPricing, estimatedInputTokens int64, remark string) (string, error) {
|
||||||
|
if pricing.Mode != TextBillingPerToken || estimatedInputTokens < 0 {
|
||||||
|
return "", errors.New("Token 预授权参数无效")
|
||||||
|
}
|
||||||
|
amount, err := CalculateTextReservePoints(pricing, estimatedInputTokens)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
businessID := textCallBusinessID(referenceID, callKey) + ":reserve"
|
||||||
|
idempotencyKey := "text:reserve:" + businessID
|
||||||
|
if existing, found, err := debitAmount(tx, idempotencyKey); err != nil {
|
||||||
|
return "", err
|
||||||
|
} else if found {
|
||||||
|
return existing, nil
|
||||||
|
}
|
||||||
|
if amount == "0.00" {
|
||||||
|
return amount, nil
|
||||||
|
}
|
||||||
|
_, err = DebitPoints(tx, userID, amount, "text_token_reserve", businessID, remark+"预授权", idempotencyKey)
|
||||||
|
return amount, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func CalculateTextReservePoints(pricing TextPricing, estimatedInputTokens int64) (string, error) {
|
||||||
|
if estimatedInputTokens < 0 || estimatedInputTokens > math.MaxInt64/2 {
|
||||||
|
return "", errors.New("Token 预授权参数无效")
|
||||||
|
}
|
||||||
|
return CalculateTextPoints(pricing, TextUsage{Input: estimatedInputTokens, Output: estimatedInputTokens * 2})
|
||||||
|
}
|
||||||
|
|
||||||
|
func SettleTextCall(tx *gorm.DB, userID uuid.UUID, referenceID, callKey string, pricing TextPricing, usage TextUsage, remark string) (string, error) {
|
||||||
|
if usage.Input <= 0 && usage.Output <= 0 {
|
||||||
|
return "", errors.New("上游未返回 Token 用量，无法完成计费")
|
||||||
|
}
|
||||||
|
actual, err := CalculateTextPoints(pricing, usage)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
baseID := textCallBusinessID(referenceID, callKey)
|
||||||
|
reserveID := baseID + ":reserve"
|
||||||
|
if _, found, err := debitAmount(tx, "text:reserve:"+reserveID); err != nil {
|
||||||
|
return "", err
|
||||||
|
} else if found {
|
||||||
|
if _, err := RefundDebit(tx, userID, "text:reserve:"+reserveID, "text:reserve-refund:"+reserveID,
|
||||||
|
"text_token_reserve_refund", reserveID, remark+"预授权释放"); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
chargeKey := "generation:hold:" + baseID
|
||||||
|
if existing, found, err := debitAmount(tx, chargeKey); err != nil {
|
||||||
|
return "", err
|
||||||
|
} else if found {
|
||||||
|
return existing, nil
|
||||||
|
}
|
||||||
|
if actual == "0.00" {
|
||||||
|
return actual, nil
|
||||||
|
}
|
||||||
|
_, err = DebitPoints(tx, userID, actual, "generation_hold", baseID, remark, chargeKey)
|
||||||
|
return actual, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func RefundTextCalls(tx *gorm.DB, userID uuid.UUID, referenceID, remark string) (bool, error) {
|
||||||
|
type debit struct {
|
||||||
|
BusinessType string
|
||||||
|
BusinessID string
|
||||||
|
IdempotencyKey string
|
||||||
|
}
|
||||||
|
debits := make([]debit, 0)
|
||||||
|
if err := tx.Table("point_ledger").
|
||||||
|
Select("business_type,business_id,idempotency_key").
|
||||||
|
Where("user_id=? AND change_amount<0 AND business_id LIKE ? AND business_type IN ?", userID, referenceID+":text:%", []string{"generation_hold", "text_token_reserve"}).
|
||||||
|
Order("id").Find(&debits).Error; err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
refunded := false
|
||||||
|
for _, debit := range debits {
|
||||||
|
refundKey := "generation:refund:" + debit.BusinessID
|
||||||
|
refundType := "generation_refund"
|
||||||
|
if debit.BusinessType == "text_token_reserve" {
|
||||||
|
refundKey = "text:reserve-refund:" + debit.BusinessID
|
||||||
|
refundType = "text_token_reserve_refund"
|
||||||
|
}
|
||||||
|
created, err := RefundDebit(tx, userID, debit.IdempotencyKey, refundKey, refundType, debit.BusinessID, remark)
|
||||||
|
if err != nil {
|
||||||
|
return refunded, err
|
||||||
|
}
|
||||||
|
refunded = refunded || created
|
||||||
|
}
|
||||||
|
return refunded, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func CommitTextReserves(tx *gorm.DB, userID uuid.UUID, referenceID, remark string) (string, error) {
|
||||||
|
type reserve struct {
|
||||||
|
BusinessID string
|
||||||
|
IdempotencyKey string
|
||||||
|
}
|
||||||
|
items := make([]reserve, 0)
|
||||||
|
if err := tx.Table("point_ledger").Select("business_id,idempotency_key").
|
||||||
|
Where("user_id=? AND business_type='text_token_reserve' AND business_id LIKE ? AND NOT EXISTS (SELECT 1 FROM point_ledger refund WHERE refund.idempotency_key='text:reserve-refund:'||point_ledger.business_id)", userID, referenceID+":text:%").
|
||||||
|
Find(&items).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
for _, item := range items {
|
||||||
|
baseID := strings.TrimSuffix(item.BusinessID, ":reserve")
|
||||||
|
if err := tx.Exec("UPDATE point_ledger SET business_type='generation_hold',business_id=?,idempotency_key=?,remark=? WHERE idempotency_key=?", baseID, "generation:hold:"+baseID, remark, item.IdempotencyKey).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var amount string
|
||||||
|
if err := tx.Table("point_ledger").Select("coalesce(sum(-change_amount),0)::text").
|
||||||
|
Where("user_id=? AND business_type='generation_hold' AND business_id LIKE ?", userID, referenceID+":text:%").Scan(&amount).Error; err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return amount, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func validNonNegativeDecimal(value string) bool {
|
||||||
|
number, ok := new(big.Rat).SetString(strings.TrimSpace(value))
|
||||||
|
return ok && number.Sign() >= 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func ceilPointCents(value *big.Rat) string {
|
||||||
|
if value == nil || value.Sign() <= 0 {
|
||||||
|
return "0.00"
|
||||||
|
}
|
||||||
|
cents := new(big.Rat).Mul(value, big.NewRat(100, 1))
|
||||||
|
quotient := new(big.Int).Quo(cents.Num(), cents.Denom())
|
||||||
|
if new(big.Int).Mod(cents.Num(), cents.Denom()).Sign() > 0 {
|
||||||
|
quotient.Add(quotient, big.NewInt(1))
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d.%02d", new(big.Int).Quo(quotient, big.NewInt(100)), new(big.Int).Mod(quotient, big.NewInt(100)))
|
||||||
|
}
|
||||||
|
|
||||||
|
func textCallBusinessID(referenceID, callKey string) string {
|
||||||
|
return referenceID + ":text:" + strings.TrimSpace(callKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
func debitAmount(tx *gorm.DB, idempotencyKey string) (string, bool, error) {
|
||||||
|
var amount string
|
||||||
|
result := tx.Table("point_ledger").Select("(-change_amount)::text").Where("idempotency_key=? AND change_amount<0", idempotencyKey).Limit(1).Scan(&amount)
|
||||||
|
if result.Error != nil {
|
||||||
|
return "", false, result.Error
|
||||||
|
}
|
||||||
|
return amount, strings.TrimSpace(amount) != "", nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,70 @@
|
|||||||
|
package billing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCalculateTextPointsRoundsUpToPointCent(t *testing.T) {
|
||||||
|
pricing := TextPricing{Mode: TextBillingPerToken, InputPerM: "100.0000", OutputPerM: "300.0000"}
|
||||||
|
amount, err := CalculateTextPoints(pricing, TextUsage{Input: 120, Output: 30})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if amount != "0.03" {
|
||||||
|
t.Fatalf("expected 0.03 points, got %s", amount)
|
||||||
|
}
|
||||||
|
amount, err = CalculateTextPoints(pricing, TextUsage{Input: 1})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if amount != "0.01" {
|
||||||
|
t.Fatalf("expected minimum positive charge 0.01, got %s", amount)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTextPricingSnapshot(t *testing.T) {
|
||||||
|
pricing := TextPricing{Mode: TextBillingPerToken, InputPerM: "1250.0000", OutputPerM: "2500.0000"}
|
||||||
|
parsed, err := ParseTextPricingSnapshot(pricing.MarshalSnapshot())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if parsed != pricing {
|
||||||
|
t.Fatalf("unexpected snapshot: %s", json.RawMessage(pricing.MarshalSnapshot()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTextPricingSnapshotRejectsLegacyThousandTokenRates(t *testing.T) {
|
||||||
|
_, err := ParseTextPricingSnapshot(json.RawMessage(`{"mode":"per_token","input_per_k":"0.1000","output_per_k":"0.3000","max_output_tokens":1000}`))
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected legacy thousand-token snapshot to be rejected")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCalculateTextPointsUsesMillionTokens(t *testing.T) {
|
||||||
|
pricing := TextPricing{Mode: TextBillingPerToken, InputPerM: "10", OutputPerM: "20"}
|
||||||
|
amount, err := CalculateTextPoints(pricing, TextUsage{Input: 1_000_000})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if amount != "10.00" {
|
||||||
|
t.Fatalf("expected 10.00 points for one million input tokens, got %s", amount)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCalculateTextReservePointsUsesDoubleInputAsEstimatedOutput(t *testing.T) {
|
||||||
|
pricing := TextPricing{Mode: TextBillingPerToken, InputPerM: "10", OutputPerM: "20"}
|
||||||
|
amount, err := CalculateTextReservePoints(pricing, 1_000_000)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if amount != "50.00" {
|
||||||
|
t.Fatalf("expected 50.00 points for input plus double-input output estimate, got %s", amount)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEstimateTextTokens(t *testing.T) {
|
||||||
|
if got := EstimateTextTokens("中文ABCD"); got != 3 {
|
||||||
|
t.Fatalf("expected 3 estimated tokens, got %d", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
Vendored
+25
@@ -0,0 +1,25 @@
|
|||||||
|
package cache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/config"
|
||||||
|
|
||||||
|
"github.com/redis/go-redis/v9"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Open(cfg config.Config) (*redis.Client, error) {
|
||||||
|
client := redis.NewClient(&redis.Options{
|
||||||
|
Addr: cfg.RedisAddress,
|
||||||
|
Password: cfg.RedisPassword,
|
||||||
|
DB: cfg.RedisDB,
|
||||||
|
})
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
if err := client.Ping(ctx).Err(); err != nil {
|
||||||
|
client.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return client, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,241 @@
|
|||||||
|
// Package config 读取并校验 API 服务运行所需的环境变量。
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/joho/godotenv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Config 汇总数据库、缓存、认证、腾讯云 COS 和本地媒体处理配置。
|
||||||
|
type Config struct {
|
||||||
|
AppEnv string
|
||||||
|
Debug bool
|
||||||
|
ServerAddress string
|
||||||
|
DatabaseURL string
|
||||||
|
RedisAddress string
|
||||||
|
RedisPassword string
|
||||||
|
RedisDB int
|
||||||
|
AsynqRedisDB int
|
||||||
|
CORSOrigins []string
|
||||||
|
StaticDirectory string
|
||||||
|
JWTSecretKey string
|
||||||
|
JWTAccessMinutes int
|
||||||
|
JWTRefreshHours int
|
||||||
|
BootstrapAdminUsername string
|
||||||
|
BootstrapAdminPassword string
|
||||||
|
Argon2Time uint32
|
||||||
|
Argon2Memory uint32
|
||||||
|
Argon2Parallelism uint8
|
||||||
|
Argon2HashLength uint32
|
||||||
|
Argon2SaltLength uint32
|
||||||
|
EncryptionKeyVersion string
|
||||||
|
EncryptionKey string
|
||||||
|
COSSecretID string
|
||||||
|
COSSecretKey string
|
||||||
|
COSBucket string
|
||||||
|
COSRegion string
|
||||||
|
COSEndpoint string
|
||||||
|
COSPublicBaseURL string
|
||||||
|
COSMaxImageSizeMB int
|
||||||
|
COSMaxVideoSizeMB int
|
||||||
|
COSMaxAudioSizeMB int
|
||||||
|
AIWorkerConcurrency int
|
||||||
|
AIPollIntervalSeconds int
|
||||||
|
AIHTTPMaxConnections int
|
||||||
|
FFmpegPath string
|
||||||
|
FFprobePath string
|
||||||
|
PythonPath string
|
||||||
|
ASRScriptPath string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load 从本地环境文件和进程环境读取配置，并校验数值范围。
|
||||||
|
func Load() (Config, error) {
|
||||||
|
_ = godotenv.Load()
|
||||||
|
|
||||||
|
databasePort, err := envInt("DATABASE_PORT", 25432)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
redisPort, err := envInt("REDIS_PORT", 26379)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
redisDB, err := envInt("REDIS_DB", 0)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
asynqRedisDB, err := envInt("ASYNQ_REDIS_DB", 1)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
jwtAccessMinutes, err := envInt("JWT_ACCESS_TOKEN_EXPIRE_MINUTES", 60)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
jwtRefreshHours, err := envInt("JWT_REFRESH_TOKEN_EXPIRE_HOURS", 168)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
argon2Time, err := envInt("ARGON2_TIME_COST", 2)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
argon2Memory, err := envInt("ARGON2_MEMORY_COST", 19456)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
argon2Parallelism, err := envInt("ARGON2_PARALLELISM", 1)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
argon2HashLength, err := envInt("ARGON2_HASH_LENGTH", 32)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
argon2SaltLength, err := envInt("ARGON2_SALT_LENGTH", 16)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
cosMaxImageSizeMB, err := envInt("COS_MAX_IMAGE_SIZE_MB", 20)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
cosMaxVideoSizeMB, err := envInt("COS_MAX_VIDEO_SIZE_MB", 100)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
cosMaxAudioSizeMB, err := envInt("COS_MAX_AUDIO_SIZE_MB", 30)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
aiWorkerConcurrency, err := envInt("AI_WORKER_CONCURRENCY", 100)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
aiPollIntervalSeconds, err := envInt("AI_POLL_INTERVAL_SECONDS", 10)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
aiHTTPMaxConnections, err := envInt("AI_HTTP_MAX_CONNECTIONS", 200)
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, err
|
||||||
|
}
|
||||||
|
if cosMaxImageSizeMB < 1 || cosMaxVideoSizeMB < 1 || cosMaxAudioSizeMB < 1 {
|
||||||
|
return Config{}, fmt.Errorf("COS 文件大小限制必须为正整数")
|
||||||
|
}
|
||||||
|
if aiWorkerConcurrency < 1 || aiWorkerConcurrency > 1000 {
|
||||||
|
return Config{}, fmt.Errorf("AI_WORKER_CONCURRENCY 必须在 1 至 1000 之间")
|
||||||
|
}
|
||||||
|
if aiPollIntervalSeconds < 2 || aiPollIntervalSeconds > 300 {
|
||||||
|
return Config{}, fmt.Errorf("AI_POLL_INTERVAL_SECONDS 必须在 2 至 300 之间")
|
||||||
|
}
|
||||||
|
if aiHTTPMaxConnections < 10 || aiHTTPMaxConnections > 5000 {
|
||||||
|
return Config{}, fmt.Errorf("AI_HTTP_MAX_CONNECTIONS 必须在 10 至 5000 之间")
|
||||||
|
}
|
||||||
|
debug, err := strconv.ParseBool(env("DEBUG", "true"))
|
||||||
|
if err != nil {
|
||||||
|
return Config{}, fmt.Errorf("DEBUG 必须是布尔值: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
databaseURL := strings.TrimSpace(os.Getenv("DATABASE_URL"))
|
||||||
|
if databaseURL == "" {
|
||||||
|
databaseURL = buildDatabaseURL(databasePort)
|
||||||
|
} else if parsed, parseErr := url.Parse(databaseURL); parseErr == nil {
|
||||||
|
query := parsed.Query()
|
||||||
|
if query.Get("TimeZone") == "" {
|
||||||
|
query.Set("TimeZone", "Asia/Shanghai")
|
||||||
|
}
|
||||||
|
parsed.RawQuery = query.Encode()
|
||||||
|
databaseURL = parsed.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
return Config{
|
||||||
|
AppEnv: env("APP_ENV", "development"),
|
||||||
|
Debug: debug,
|
||||||
|
ServerAddress: env("SERVER_ADDRESS", ":8900"),
|
||||||
|
DatabaseURL: databaseURL,
|
||||||
|
RedisAddress: fmt.Sprintf("%s:%d", env("REDIS_HOST", "127.0.0.1"), redisPort),
|
||||||
|
RedisPassword: os.Getenv("REDIS_PASSWORD"),
|
||||||
|
RedisDB: redisDB,
|
||||||
|
AsynqRedisDB: asynqRedisDB,
|
||||||
|
CORSOrigins: splitCSV(env("CORS_ORIGINS", "http://localhost:5500,http://localhost:5501")),
|
||||||
|
StaticDirectory: env("STATIC_DIRECTORY", "static"),
|
||||||
|
JWTSecretKey: env("JWT_SECRET_KEY", "replace-with-a-strong-random-secret"),
|
||||||
|
JWTAccessMinutes: jwtAccessMinutes,
|
||||||
|
JWTRefreshHours: jwtRefreshHours,
|
||||||
|
BootstrapAdminUsername: env("ADMIN_BOOTSTRAP_USERNAME", "admin"),
|
||||||
|
BootstrapAdminPassword: os.Getenv("ADMIN_BOOTSTRAP_PASSWORD"),
|
||||||
|
Argon2Time: uint32(argon2Time),
|
||||||
|
Argon2Memory: uint32(argon2Memory),
|
||||||
|
Argon2Parallelism: uint8(argon2Parallelism),
|
||||||
|
Argon2HashLength: uint32(argon2HashLength),
|
||||||
|
Argon2SaltLength: uint32(argon2SaltLength),
|
||||||
|
EncryptionKeyVersion: env("CONFIG_ENCRYPTION_KEY_VERSION", "v1"),
|
||||||
|
EncryptionKey: os.Getenv("CONFIG_ENCRYPTION_KEY"),
|
||||||
|
COSSecretID: os.Getenv("COS_SECRET_ID"),
|
||||||
|
COSSecretKey: os.Getenv("COS_SECRET_KEY"),
|
||||||
|
COSBucket: os.Getenv("COS_BUCKET"),
|
||||||
|
COSRegion: env("COS_REGION", "ap-chengdu"),
|
||||||
|
COSEndpoint: os.Getenv("COS_ENDPOINT"),
|
||||||
|
COSPublicBaseURL: strings.TrimRight(os.Getenv("COS_PUBLIC_BASE_URL"), "/"),
|
||||||
|
COSMaxImageSizeMB: cosMaxImageSizeMB,
|
||||||
|
COSMaxVideoSizeMB: cosMaxVideoSizeMB,
|
||||||
|
COSMaxAudioSizeMB: cosMaxAudioSizeMB,
|
||||||
|
AIWorkerConcurrency: aiWorkerConcurrency,
|
||||||
|
AIPollIntervalSeconds: aiPollIntervalSeconds,
|
||||||
|
AIHTTPMaxConnections: aiHTTPMaxConnections,
|
||||||
|
FFmpegPath: env("FFMPEG_PATH", "ffmpeg"),
|
||||||
|
FFprobePath: env("FFPROBE_PATH", "ffprobe"),
|
||||||
|
PythonPath: env("PYTHON_PATH", "python3"),
|
||||||
|
ASRScriptPath: env("ASR_SCRIPT_PATH", "workers/transcribe.py"),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildDatabaseURL(port int) string {
|
||||||
|
credentials := url.UserPassword(
|
||||||
|
env("DATABASE_USER", "postgres"),
|
||||||
|
env("DATABASE_PASSWORD", "juchuang_dev"),
|
||||||
|
).String()
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"postgres://%s@%s:%d/%s?sslmode=disable&TimeZone=Asia/Shanghai",
|
||||||
|
credentials,
|
||||||
|
env("DATABASE_HOST", "127.0.0.1"),
|
||||||
|
port,
|
||||||
|
url.PathEscape(env("DATABASE_NAME", "juchuang_factory")),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func env(key, fallback string) string {
|
||||||
|
if value := strings.TrimSpace(os.Getenv(key)); value != "" {
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
return fallback
|
||||||
|
}
|
||||||
|
|
||||||
|
func envInt(key string, fallback int) (int, error) {
|
||||||
|
value := strings.TrimSpace(os.Getenv(key))
|
||||||
|
if value == "" {
|
||||||
|
return fallback, nil
|
||||||
|
}
|
||||||
|
parsed, err := strconv.Atoi(value)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("%s 必须是整数: %w", key, err)
|
||||||
|
}
|
||||||
|
return parsed, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func splitCSV(value string) []string {
|
||||||
|
items := strings.Split(value, ",")
|
||||||
|
result := make([]string, 0, len(items))
|
||||||
|
for _, item := range items {
|
||||||
|
if item = strings.TrimSpace(item); item != "" {
|
||||||
|
result = append(result, item)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
package database
|
||||||
|
|
||||||
|
import (
|
||||||
|
_ "embed"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed schema.sql
|
||||||
|
var baselineSchema string
|
||||||
|
|
||||||
|
func bootstrapEmptyDatabase(db *gorm.DB) error {
|
||||||
|
var businessTableCount int64
|
||||||
|
if err := db.Raw(`SELECT count(*) FROM pg_tables
|
||||||
|
WHERE schemaname = 'public' AND tablename NOT IN ('jcf_schema_migrations', 'schema_migrations')`).
|
||||||
|
Scan(&businessTableCount).Error; err != nil {
|
||||||
|
return fmt.Errorf("inspect database schema: %w", err)
|
||||||
|
}
|
||||||
|
if businessTableCount == 0 {
|
||||||
|
baselineDB := db.Session(&gorm.Session{Logger: logger.Default.LogMode(logger.Silent)})
|
||||||
|
if err := baselineDB.Transaction(func(tx *gorm.DB) error {
|
||||||
|
return tx.Exec(baselineSchema).Error
|
||||||
|
}); err != nil {
|
||||||
|
return fmt.Errorf("initialize database baseline: %w", err)
|
||||||
|
}
|
||||||
|
if err := db.Exec(`SET search_path TO public`).Error; err != nil {
|
||||||
|
return fmt.Errorf("restore database search path: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := autoMigrateMissingTables(db, autoMigrateModels()...); err != nil {
|
||||||
|
return fmt.Errorf("auto migrate database models: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func autoMigrateMissingTables(db *gorm.DB, tableModels ...any) error {
|
||||||
|
for _, tableModel := range tableModels {
|
||||||
|
if db.Migrator().HasTable(tableModel) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := db.AutoMigrate(tableModel); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func autoMigrateModels() []any {
|
||||||
|
return []any{
|
||||||
|
&model.AdminUser{},
|
||||||
|
&model.AdminRefreshToken{},
|
||||||
|
&model.AdminAuditLog{},
|
||||||
|
&model.WebUser{},
|
||||||
|
&model.WebRefreshToken{},
|
||||||
|
&model.MediaAsset{},
|
||||||
|
&model.CreativeProject{},
|
||||||
|
&model.ProjectModelConfig{},
|
||||||
|
&model.UserModelConfig{},
|
||||||
|
&model.ScriptAnalysis{},
|
||||||
|
&model.ScriptAnalysisCharacter{},
|
||||||
|
&model.ProjectEpisode{},
|
||||||
|
&model.ProjectAsset{},
|
||||||
|
&model.EpisodeStoryboard{},
|
||||||
|
&model.EpisodeSource{},
|
||||||
|
&model.DramaImportSession{},
|
||||||
|
&model.DramaParseBatch{},
|
||||||
|
&model.DramaParseTask{},
|
||||||
|
&model.GenerationTask{},
|
||||||
|
&model.GenerationOutput{},
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
package database
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
type autoMigrateProbe struct {
|
||||||
|
ID uint `gorm:"primaryKey"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (autoMigrateProbe) TableName() string { return "auto_migrate_probe" }
|
||||||
|
|
||||||
|
func TestBootstrapEmptyDatabase(t *testing.T) {
|
||||||
|
databaseURL := os.Getenv("JCF_BOOTSTRAP_TEST_DATABASE_URL")
|
||||||
|
if databaseURL == "" {
|
||||||
|
t.Skip("JCF_BOOTSTRAP_TEST_DATABASE_URL is not configured")
|
||||||
|
}
|
||||||
|
|
||||||
|
db, sqlDB, err := Open(config.Config{DatabaseURL: databaseURL})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("open empty database: %v", err)
|
||||||
|
}
|
||||||
|
defer sqlDB.Close()
|
||||||
|
|
||||||
|
var tableCount int64
|
||||||
|
if err := db.Raw(`SELECT count(*) FROM pg_tables WHERE schemaname = 'public'`).Scan(&tableCount).Error; err != nil {
|
||||||
|
t.Fatalf("count initialized tables: %v", err)
|
||||||
|
}
|
||||||
|
if tableCount != 33 {
|
||||||
|
t.Fatalf("expected 33 initialized tables, got %d", tableCount)
|
||||||
|
}
|
||||||
|
|
||||||
|
var migrationCount int64
|
||||||
|
if err := db.Table("jcf_schema_migrations").Count(&migrationCount).Error; err != nil {
|
||||||
|
t.Fatalf("count baseline migrations: %v", err)
|
||||||
|
}
|
||||||
|
if migrationCount != 0 {
|
||||||
|
t.Fatalf("expected no baseline migrations, got %d", migrationCount)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := autoMigrateMissingTables(db, &autoMigrateProbe{}); err != nil {
|
||||||
|
t.Fatalf("auto migrate missing table: %v", err)
|
||||||
|
}
|
||||||
|
if !db.Migrator().HasTable(&autoMigrateProbe{}) {
|
||||||
|
t.Fatal("expected AutoMigrate to create missing model table")
|
||||||
|
}
|
||||||
|
if err := db.Migrator().DropTable(&autoMigrateProbe{}); err != nil {
|
||||||
|
t.Fatalf("drop AutoMigrate probe table: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, reopenedSQLDB, err := Open(config.Config{DatabaseURL: databaseURL})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("reopen initialized database: %v", err)
|
||||||
|
}
|
||||||
|
defer reopenedSQLDB.Close()
|
||||||
|
}
|
||||||
@@ -0,0 +1,133 @@
|
|||||||
|
package database
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/config"
|
||||||
|
|
||||||
|
"gorm.io/driver/postgres"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Open(cfg config.Config) (*gorm.DB, *sql.DB, error) {
|
||||||
|
logLevel := logger.Warn
|
||||||
|
if cfg.Debug {
|
||||||
|
logLevel = logger.Info
|
||||||
|
}
|
||||||
|
|
||||||
|
databaseURL := cfg.DatabaseURL
|
||||||
|
if !strings.Contains(strings.ToLower(databaseURL), "timezone=") {
|
||||||
|
separator := "?"
|
||||||
|
if strings.Contains(databaseURL, " ") {
|
||||||
|
separator = " "
|
||||||
|
}
|
||||||
|
if strings.Contains(databaseURL, "?") {
|
||||||
|
separator = "&"
|
||||||
|
}
|
||||||
|
databaseURL += separator + "TimeZone=Asia%2FShanghai"
|
||||||
|
}
|
||||||
|
db, err := gorm.Open(postgres.Open(databaseURL), &gorm.Config{
|
||||||
|
Logger: logger.Default.LogMode(logLevel),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
sqlDB, err := db.DB()
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
sqlDB.SetMaxOpenConns(20)
|
||||||
|
sqlDB.SetMaxIdleConns(5)
|
||||||
|
sqlDB.SetConnMaxLifetime(30 * time.Minute)
|
||||||
|
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
if err := sqlDB.PingContext(ctx); err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`SET TIME ZONE 'Asia/Shanghai'`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := bootstrapEmptyDatabase(db); err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
// Keep older installations compatible when migrations are applied manually.
|
||||||
|
if err := db.Exec(`ALTER TABLE project_model_configs ADD COLUMN IF NOT EXISTS settings jsonb NOT NULL DEFAULT '{}'::jsonb`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`ALTER TABLE creative_projects ADD COLUMN IF NOT EXISTS video_resolution varchar(8) NOT NULL DEFAULT '720p'`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`ALTER TABLE episode_storyboards ADD COLUMN IF NOT EXISTS image_prompt text`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`ALTER TABLE prompts ADD COLUMN IF NOT EXISTS type varchar(50); ALTER TABLE prompts ADD COLUMN IF NOT EXISTS content text; ALTER TABLE prompts ADD COLUMN IF NOT EXISTS scope varchar(20) NOT NULL DEFAULT 'system'; ALTER TABLE prompts ADD COLUMN IF NOT EXISTS owner_user_id uuid REFERENCES web_users(id);`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`ALTER TABLE web_users ADD COLUMN IF NOT EXISTS account citext; UPDATE web_users SET account=username WHERE account IS NULL; CREATE UNIQUE INDEX IF NOT EXISTS idx_web_users_account ON web_users(account);`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`ALTER TABLE redemption_codes ADD COLUMN IF NOT EXISTS code character varying(40);`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`CREATE TABLE IF NOT EXISTS user_model_configs (
|
||||||
|
id uuid PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
user_id uuid NOT NULL REFERENCES web_users(id),
|
||||||
|
purpose varchar(32) NOT NULL CHECK (purpose IN ('prompt_reverse','image_generation','video_generation')),
|
||||||
|
model_id uuid NOT NULL REFERENCES models(id),
|
||||||
|
prompt_id uuid REFERENCES prompts(id),
|
||||||
|
settings jsonb NOT NULL DEFAULT '{}'::jsonb,
|
||||||
|
created_at timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
updated_at timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(user_id, purpose)
|
||||||
|
); CREATE INDEX IF NOT EXISTS idx_user_model_configs_user ON user_model_configs(user_id);
|
||||||
|
DO $$
|
||||||
|
BEGIN
|
||||||
|
IF EXISTS (SELECT 1 FROM information_schema.columns WHERE table_name='user_model_configs' AND column_name='model_type') THEN
|
||||||
|
INSERT INTO user_model_configs(id,user_id,project_type,model_type,purpose,model_id,prompt_id,settings)
|
||||||
|
SELECT DISTINCT ON (project.user_id, config.model_type)
|
||||||
|
gen_random_uuid(), project.user_id, 'video_redraw', config.model_type, config.purpose, config.model_id, config.prompt_id, config.settings
|
||||||
|
FROM project_model_configs config
|
||||||
|
JOIN creative_projects project ON project.id=config.project_id
|
||||||
|
WHERE project.deleted_at IS NULL AND project.project_type='video_redraw'
|
||||||
|
ORDER BY project.user_id, config.model_type, project.updated_at DESC, config.updated_at DESC
|
||||||
|
ON CONFLICT DO NOTHING;
|
||||||
|
ELSE
|
||||||
|
INSERT INTO user_model_configs(id,user_id,purpose,model_id,prompt_id,settings)
|
||||||
|
SELECT DISTINCT ON (project.user_id, config.purpose)
|
||||||
|
gen_random_uuid(), project.user_id, config.purpose, config.model_id, config.prompt_id, config.settings
|
||||||
|
FROM project_model_configs config
|
||||||
|
JOIN creative_projects project ON project.id=config.project_id
|
||||||
|
WHERE project.deleted_at IS NULL
|
||||||
|
ORDER BY project.user_id, config.purpose, project.updated_at DESC, config.updated_at DESC
|
||||||
|
ON CONFLICT DO NOTHING;
|
||||||
|
END IF;
|
||||||
|
END $$;`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if err := db.Exec(`CREATE TABLE IF NOT EXISTS user_prompt_preferences (
|
||||||
|
id uuid PRIMARY KEY DEFAULT gen_random_uuid(), user_id uuid NOT NULL REFERENCES web_users(id) ON DELETE CASCADE,
|
||||||
|
prompt_id uuid NOT NULL REFERENCES prompts(id) ON DELETE CASCADE, prompt_type varchar(64) NOT NULL,
|
||||||
|
created_at timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP, updated_at timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(user_id,prompt_type)
|
||||||
|
); CREATE INDEX IF NOT EXISTS idx_user_prompt_preferences_user ON user_prompt_preferences(user_id);`).Error; err != nil {
|
||||||
|
sqlDB.Close()
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
return db, sqlDB, nil
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,30 @@
|
|||||||
|
// Package datetime centralizes the application's Beijing time policy.
|
||||||
|
package datetime
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
const Zone = "Asia/Shanghai"
|
||||||
|
|
||||||
|
var Location = func() *time.Location {
|
||||||
|
location, err := time.LoadLocation(Zone)
|
||||||
|
if err != nil {
|
||||||
|
return time.FixedZone("CST", 8*60*60)
|
||||||
|
}
|
||||||
|
return location
|
||||||
|
}()
|
||||||
|
|
||||||
|
func Now() time.Time { return time.Now().In(Location) }
|
||||||
|
|
||||||
|
// Normalize converts any source instant (including COS metadata) to Beijing time.
|
||||||
|
func Normalize(value time.Time) time.Time {
|
||||||
|
if value.IsZero() { return value }
|
||||||
|
return value.In(Location)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Parse(value string) (time.Time, error) {
|
||||||
|
parsed, err := time.Parse(time.RFC3339Nano, value)
|
||||||
|
if err != nil { return time.Time{}, err }
|
||||||
|
return Normalize(parsed), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Format(value time.Time) string { return Normalize(value).Format(time.RFC3339Nano) }
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
package drama
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"unicode/utf8"
|
||||||
|
)
|
||||||
|
|
||||||
|
const MaxEpisodeCharacters = 5000
|
||||||
|
|
||||||
|
func ValidateEpisodeContent(content string) error {
|
||||||
|
count := utf8.RuneCountInString(strings.TrimSpace(content))
|
||||||
|
if count == 0 {
|
||||||
|
return fmt.Errorf("剧集原文不能为空")
|
||||||
|
}
|
||||||
|
if count > MaxEpisodeCharacters {
|
||||||
|
return fmt.Errorf("每集原文不能超过%d字，当前%d字", MaxEpisodeCharacters, count)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
package drama
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestValidateEpisodeContent(t *testing.T) {
|
||||||
|
if err := ValidateEpisodeContent(strings.Repeat("字", MaxEpisodeCharacters)); err != nil {
|
||||||
|
t.Fatalf("expected content at limit to pass: %v", err)
|
||||||
|
}
|
||||||
|
if err := ValidateEpisodeContent(strings.Repeat("字", MaxEpisodeCharacters+1)); err == nil {
|
||||||
|
t.Fatal("expected over-limit content to fail")
|
||||||
|
}
|
||||||
|
if err := ValidateEpisodeContent(" \n\t"); err == nil {
|
||||||
|
t.Fatal("expected blank content to fail")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,111 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
const adminContextKey = "admin_user"
|
||||||
|
|
||||||
|
type AdminAuth struct{ service *service.Auth }
|
||||||
|
|
||||||
|
func NewAdminAuth(service *service.Auth) *AdminAuth { return &AdminAuth{service: service} }
|
||||||
|
|
||||||
|
func (h *AdminAuth) Login(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Username string `json:"username" binding:"required"`
|
||||||
|
Password string `json:"password" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请输入管理员账号和密码")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
pair, err := h.service.Login(body.Username, body.Password)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusUnauthorized, "invalid_credentials", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": pair})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminAuth) Refresh(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
RefreshToken string `json:"refresh_token" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "缺少刷新令牌")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
pair, err := h.service.Refresh(body.RefreshToken)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusUnauthorized, "invalid_refresh_token", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": pair})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminAuth) Logout(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
RefreshToken string `json:"refresh_token"`
|
||||||
|
}
|
||||||
|
_ = c.ShouldBindJSON(&body)
|
||||||
|
if err := h.service.Logout(body.RefreshToken); err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "logout_failed", "退出登录失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminAuth) ChangePassword(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
CurrentPassword string `json:"current_password" binding:"required"`
|
||||||
|
NewPassword string `json:"new_password" binding:"required"`
|
||||||
|
ConfirmPassword string `json:"confirm_password" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请完整填写密码")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
admin := currentAdmin(c)
|
||||||
|
if err := h.service.ChangePassword(admin.ID, body.CurrentPassword, body.NewPassword, body.ConfirmPassword); err != nil {
|
||||||
|
if errors.Is(err, service.ErrReauthFailed) || errors.Is(err, service.ErrAdminPasswordConfirmation) || errors.Is(err, service.ErrAdminPasswordLength) {
|
||||||
|
_ = h.service.Audit(admin, "password_change_failed", "admin-auth", admin.ID.String(), err.Error(), c.ClientIP(), c.GetString("trace_id"), nil)
|
||||||
|
fail(c, http.StatusBadRequest, "password_invalid", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusInternalServerError, "password_update_failed", "密码修改失败，请稍后重试")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_ = h.service.Audit(admin, "password_changed", "admin-auth", admin.ID.String(), "", c.ClientIP(), c.GetString("trace_id"), nil)
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminAuth) Middleware() gin.HandlerFunc {
|
||||||
|
return func(c *gin.Context) {
|
||||||
|
raw := strings.TrimSpace(strings.TrimPrefix(c.GetHeader("Authorization"), "Bearer "))
|
||||||
|
admin, err := h.service.Authenticate(raw)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusUnauthorized, "unauthorized", err.Error())
|
||||||
|
c.Abort()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Set(adminContextKey, admin)
|
||||||
|
c.Next()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func currentAdmin(c *gin.Context) *model.AdminUser {
|
||||||
|
value, _ := c.Get(adminContextKey)
|
||||||
|
admin, _ := value.(*model.AdminUser)
|
||||||
|
return admin
|
||||||
|
}
|
||||||
|
|
||||||
|
func fail(c *gin.Context, status int, code, message string) {
|
||||||
|
c.JSON(status, gin.H{"code": code, "message": message, "trace_id": c.GetString("trace_id")})
|
||||||
|
}
|
||||||
@@ -0,0 +1,544 @@
|
|||||||
|
// 管理后台接口处理器，负责解析管理端请求并调用对应业务模块组织响应。
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"image"
|
||||||
|
_ "image/gif"
|
||||||
|
_ "image/jpeg"
|
||||||
|
_ "image/png"
|
||||||
|
"log/slog"
|
||||||
|
"mime/multipart"
|
||||||
|
"net/http"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
mediakey "juhe-factory/api/internal/media"
|
||||||
|
adminmodule "juhe-factory/api/internal/modules/admin"
|
||||||
|
"juhe-factory/api/internal/modules/prompt"
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
"juhe-factory/api/internal/storage"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AdminData struct {
|
||||||
|
data *service.AdminData
|
||||||
|
admin *adminmodule.Service
|
||||||
|
prompts *prompt.Service
|
||||||
|
cos *storage.COS
|
||||||
|
}
|
||||||
|
|
||||||
|
type optionalTime struct{ Time *time.Time }
|
||||||
|
|
||||||
|
func (t *optionalTime) UnmarshalJSON(data []byte) error {
|
||||||
|
if string(data) == "null" || strings.TrimSpace(string(data)) == `""` {
|
||||||
|
t.Time = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var value time.Time
|
||||||
|
if err := json.Unmarshal(data, &value); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.Time = &value
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAdminData(data *service.AdminData, admin *adminmodule.Service, prompts *prompt.Service, cos *storage.COS) *AdminData {
|
||||||
|
return &AdminData{data: data, admin: admin, prompts: prompts, cos: cos}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) ListResource(c *gin.Context) {
|
||||||
|
filters := map[string]string{"enabled": c.Query("enabled"), "payment_type": c.Query("payment_type"), "channel_type": c.Query("channel_type"), "status": c.Query("status"), "pinned": c.Query("pinned")}
|
||||||
|
page, err := h.data.ListResource(c.Request.Context(), c.Param("resource"), c.Query("keyword"), filters, c.Query("page"), c.Query("page_size"))
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": page})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) CreateResource(c *gin.Context) {
|
||||||
|
var body map[string]any
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id, err := h.data.SaveResource(c.Param("resource"), "", body)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "create", c.Param("resource"), id, "", body)
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": gin.H{"id": id}})
|
||||||
|
}
|
||||||
|
func (h *AdminData) UpdateResource(c *gin.Context) {
|
||||||
|
var body map[string]any
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id, err := h.data.SaveResource(c.Param("resource"), c.Param("id"), body)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "update", c.Param("resource"), id, stringValue(body["reason"]), body)
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": gin.H{"id": id}})
|
||||||
|
}
|
||||||
|
func (h *AdminData) ToggleResource(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.data.ToggleResource(c.Param("resource"), c.Param("id"), body.Enabled); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "toggle", c.Param("resource"), c.Param("id"), body.Reason, body)
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
func (h *AdminData) DeleteResource(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
}
|
||||||
|
_ = c.ShouldBindJSON(&body)
|
||||||
|
if err := h.data.DeleteResource(c.Param("resource"), c.Param("id")); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "delete", c.Param("resource"), c.Param("id"), body.Reason, nil)
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) ReorderStyles(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
IDs []string `json:"ids"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.data.ReorderStyles(body.IDs); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "reorder", "styles", strings.Join(body.IDs, ","), "", map[string]any{"count": len(body.IDs)})
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) ListUsers(c *gin.Context) {
|
||||||
|
page, err := h.data.ListUsers(c.Query("keyword"), c.Query("enabled"), c.Query("page"), c.Query("page_size"))
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": page})
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListAuditLogs 返回管理端操作审计记录，不提供修改或删除能力。
|
||||||
|
func (h *AdminData) ListAuditLogs(c *gin.Context) {
|
||||||
|
page, size := parsePage(c)
|
||||||
|
result, err := h.admin.ListAuditLogs(c.Query("keyword"), page, size)
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": result})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) CreateUser(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Account string `json:"account"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
DailyLimit any `json:"daily_limit"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
item, err := h.data.CreateUser(body.Account, body.Password, body.DailyLimit)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "create", "users", stringValue(item["id"]), "", map[string]any{"account": item["account"], "username": item["username"], "daily_limit": body.DailyLimit})
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) BatchCreateUsers(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Prefix string `json:"prefix"`
|
||||||
|
StartSequence string `json:"start_sequence"`
|
||||||
|
EndSequence string `json:"end_sequence"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
DailyLimit any `json:"daily_limit"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
items, err := h.data.BatchCreateUsers(body.Prefix, body.StartSequence, body.EndSequence, body.Password, body.DailyLimit)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "batch_create", "users", "", "", map[string]any{"prefix": body.Prefix, "start_sequence": body.StartSequence, "end_sequence": body.EndSequence, "count": len(items)})
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) BatchUpdateUsers(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
IDs []string `json:"ids"`
|
||||||
|
DailyLimit any `json:"daily_limit"`
|
||||||
|
Enabled *bool `json:"enabled"`
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
updates := map[string]any{}
|
||||||
|
if body.DailyLimit != nil {
|
||||||
|
updates["daily_limit"] = body.DailyLimit
|
||||||
|
}
|
||||||
|
if body.Enabled != nil {
|
||||||
|
updates["enabled"] = *body.Enabled
|
||||||
|
}
|
||||||
|
if err := h.data.UpdateUsers(body.IDs, updates); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "batch_update", "users", strings.Join(body.IDs, ","), body.Reason, updates)
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) BatchDeleteUsers(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
IDs []string `json:"ids"`
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.data.DeleteUsers(body.IDs); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "batch_delete", "users", strings.Join(body.IDs, ","), body.Reason, nil)
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GrantUserPoints 校验管理端积分发放请求，完成入账后记录操作审计。
|
||||||
|
func (h *AdminData) GrantUserPoints(c *gin.Context) {
|
||||||
|
userID, err := uuid.Parse(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("用户标识无效"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var body struct {
|
||||||
|
RequestID string `json:"request_id"`
|
||||||
|
Points any `json:"points"`
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
body.Reason = strings.TrimSpace(body.Reason)
|
||||||
|
if body.Reason == "" {
|
||||||
|
h.bad(c, errors.New("请填写积分发放原因"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len([]rune(body.Reason)) > 200 {
|
||||||
|
h.bad(c, errors.New("积分发放原因不能超过 200 个字符"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
balance, credited, err := h.admin.GrantUserPoints(userID, body.RequestID, body.Points)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if credited {
|
||||||
|
h.audit(c, "grant_points", "users", userID.String(), body.Reason, map[string]any{
|
||||||
|
"points": body.Points, "balance": balance, "request_id": body.RequestID,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": gin.H{"balance": balance, "credited": credited}})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) ListModels(c *gin.Context) {
|
||||||
|
page, size := parsePage(c)
|
||||||
|
result, err := h.admin.ListModels(c.Query("keyword"), c.Query("model_type"), c.Query("channel_id"), page, size)
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": result})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) SaveModel(c *gin.Context) {
|
||||||
|
var body adminmodule.ModelInput
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id, created, normalizedBody, err := h.admin.SaveModel(c.Param("id"), body)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
action := "update"
|
||||||
|
status := http.StatusOK
|
||||||
|
if created {
|
||||||
|
action = "create"
|
||||||
|
status = http.StatusCreated
|
||||||
|
}
|
||||||
|
h.audit(c, action, "models", id, "", normalizedBody)
|
||||||
|
c.JSON(status, gin.H{"data": gin.H{"id": id}})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) ListPrompts(c *gin.Context) {
|
||||||
|
page, size := parsePage(c)
|
||||||
|
items, total, err := h.prompts.ListSystemPrompts(c.Query("keyword"), c.Query("type"), page, size)
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": service.Page{Items: items, Total: total, Page: page, PageSize: size}})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) SavePrompt(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Code string `json:"code"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Category string `json:"category"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Content string `json:"content"`
|
||||||
|
Variables []string `json:"variables"`
|
||||||
|
VersionNote string `json:"version_note"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(body.Category) == "" {
|
||||||
|
body.Category = body.Type
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(body.Type) == "" {
|
||||||
|
body.Type = body.Category
|
||||||
|
}
|
||||||
|
if err := service.ValidatePromptVariables(body.Content, body.Variables); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id, err := h.prompts.SaveVersionedPrompt(currentAdmin(c).ID, c.Param("id"), prompt.VersionedPromptInput{
|
||||||
|
Code: body.Code, Name: body.Name, Category: body.Category, Type: body.Type,
|
||||||
|
Content: body.Content, Variables: body.Variables, VersionNote: body.VersionNote,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "save_draft", "prompts", id, body.VersionNote, nil)
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": gin.H{"id": id}})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SavePromptSimple 保存系统提示词并立即应用，不启用历史版本工作流。
|
||||||
|
func (h *AdminData) SavePromptSimple(c *gin.Context) {
|
||||||
|
var body prompt.SystemPromptInput
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
item, err := h.prompts.SaveSystemPrompt(c.Param("id"), body)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "save", "prompts", item.ID, "", body)
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeletePromptSimple 删除管理端指定的系统提示词。
|
||||||
|
func (h *AdminData) DeletePromptSimple(c *gin.Context) {
|
||||||
|
if err := h.prompts.DeleteSystemPrompt(c.Param("id")); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) PromptAction(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Action string `json:"action"`
|
||||||
|
VersionID string `json:"version_id"`
|
||||||
|
VersionNote string `json:"version_note"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id := c.Param("id")
|
||||||
|
if err := h.prompts.ApplyPromptAction(currentAdmin(c).ID, id, body.Action, body.VersionID, body.VersionNote); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, body.Action, "prompts", id, body.VersionNote, body)
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) PromptHistory(c *gin.Context) {
|
||||||
|
items, err := h.prompts.ListPromptHistory(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) PromptActionSimple(c *gin.Context) { c.Status(http.StatusNoContent) }
|
||||||
|
func (h *AdminData) PromptHistorySimple(c *gin.Context) {
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": []any{}})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) ListRedemptions(c *gin.Context) {
|
||||||
|
page, size := parsePage(c)
|
||||||
|
result, err := h.admin.ListRedemptions(c.Query("keyword"), c.Query("status"), page, size)
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": result})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) CreateRedemptionBatch(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Points any `json:"points"`
|
||||||
|
Quantity int `json:"quantity"`
|
||||||
|
ExpiresAt optionalTime `json:"expires_at"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if body.ExpiresAt.Time == nil {
|
||||||
|
expiresAt := time.Now().AddDate(0, 0, 7)
|
||||||
|
body.ExpiresAt.Time = &expiresAt
|
||||||
|
}
|
||||||
|
admin := currentAdmin(c)
|
||||||
|
batchID, codes, points, err := h.admin.CreateRedemptionBatch(admin.ID, body.Name, body.Points, body.Quantity, *body.ExpiresAt.Time)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.audit(c, "generate", "redemption-codes", batchID, "", map[string]any{"quantity": body.Quantity, "points": points})
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": gin.H{"batch_id": batchID, "codes": codes}})
|
||||||
|
}
|
||||||
|
|
||||||
|
// validateRedemptionBatchLimits 保留 Handler 包内既有测试入口，实际规则由管理后台模块统一实现。
|
||||||
|
func validateRedemptionBatchLimits(value any, quantity int) (string, error) {
|
||||||
|
return adminmodule.ValidateRedemptionBatchLimits(value, quantity)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdminData) UploadStyleImage(c *gin.Context) {
|
||||||
|
if h.cos == nil {
|
||||||
|
fail(c, http.StatusServiceUnavailable, "cos_not_configured", "COS 对象存储未配置")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
file, header, err := c.Request.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("请选择图片文件"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
if header.Size <= 0 || header.Size > h.cos.MaxImageBytes() {
|
||||||
|
h.bad(c, errors.New("图片大小超出限制"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
contentType := header.Header.Get("Content-Type")
|
||||||
|
if contentType != "image/jpeg" && contentType != "image/png" && contentType != "image/gif" {
|
||||||
|
h.bad(c, errors.New("仅支持 JPEG、PNG 或 GIF 图片"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
width, height, err := imageDimensions(file, contentType)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("无法识别图片内容"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if width < 64 || height < 64 || width > 8192 || height > 8192 {
|
||||||
|
h.bad(c, errors.New("图片尺寸必须在 64×64 至 8192×8192 之间"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if _, err = file.Seek(0, 0); err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ext := strings.ToLower(filepath.Ext(header.Filename))
|
||||||
|
if ext == "" {
|
||||||
|
ext = ".img"
|
||||||
|
}
|
||||||
|
key := mediakey.StyleImage(uuid.New(), ext)
|
||||||
|
url, err := h.cos.Put(c, key, contentType, file, header.Size)
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": gin.H{"image_key": key, "image_url": url, "image_mime": contentType, "image_size": header.Size, "image_width": width, "image_height": height}})
|
||||||
|
}
|
||||||
|
|
||||||
|
func imageDimensions(file multipart.File, contentType string) (int, int, error) {
|
||||||
|
if contentType == "image/webp" {
|
||||||
|
return 0, 0, errors.New("WebP 尺寸解析暂不可用")
|
||||||
|
}
|
||||||
|
cfg, _, err := image.DecodeConfig(file)
|
||||||
|
return cfg.Width, cfg.Height, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func parsePage(c *gin.Context) (int, int) {
|
||||||
|
page, _ := strconv.Atoi(c.Query("page"))
|
||||||
|
if page < 1 {
|
||||||
|
page = 1
|
||||||
|
}
|
||||||
|
size, _ := strconv.Atoi(c.Query("page_size"))
|
||||||
|
if size < 1 {
|
||||||
|
size = 20
|
||||||
|
}
|
||||||
|
if size > 100 {
|
||||||
|
size = 100
|
||||||
|
}
|
||||||
|
return page, size
|
||||||
|
}
|
||||||
|
func stringValue(v any) string {
|
||||||
|
if v == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return fmt.Sprint(v)
|
||||||
|
}
|
||||||
|
func (h *AdminData) audit(c *gin.Context, action, resource, id, reason string, detail any) {
|
||||||
|
_ = h.admin.WriteAudit(currentAdmin(c), action, resource, id, reason, c.ClientIP(), c.GetString("trace_id"), detail)
|
||||||
|
}
|
||||||
|
func (h *AdminData) bad(c *gin.Context, err error) {
|
||||||
|
message := err.Error()
|
||||||
|
code := "invalid_request"
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
code = "not_found"
|
||||||
|
}
|
||||||
|
fail(c, http.StatusBadRequest, code, message)
|
||||||
|
}
|
||||||
|
func (h *AdminData) internal(c *gin.Context, err error) {
|
||||||
|
slog.ErrorContext(c.Request.Context(), "admin request failed", "trace_id", c.GetString("trace_id"), "error", err)
|
||||||
|
fail(c, http.StatusInternalServerError, "internal_error", "操作失败，请稍后重试")
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,47 @@
|
|||||||
|
// 本文件负责剧集视频 ZIP 下载请求的参数校验、响应头设置和对象流写入。
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io"
|
||||||
|
"mime"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
"juhe-factory/api/internal/videoarchive"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DownloadEpisodeVideos 下载当前剧集按项目类型筛选后的视频 ZIP。
|
||||||
|
func (h *Creative) DownloadEpisodeVideos(c *gin.Context) {
|
||||||
|
if !h.objectStorageAvailable(c) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
episodeID, err := service.ParseUUID(c.Param("episode_id"), "剧集")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
manifest, err := h.service.EpisodeVideoArchive(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Header("Content-Type", "application/zip")
|
||||||
|
c.Header("Content-Disposition", mime.FormatMediaType("attachment", map[string]string{"filename": manifest.Name}))
|
||||||
|
c.Header("X-Content-Type-Options", "nosniff")
|
||||||
|
c.Status(http.StatusOK)
|
||||||
|
err = videoarchive.Write(c.Request.Context(), c.Writer, manifest.Entries, func(ctx context.Context, objectKey string) (io.ReadCloser, error) {
|
||||||
|
body, _, _, openErr := h.cos.Open(ctx, objectKey)
|
||||||
|
return body, openErr
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
_ = c.Error(err)
|
||||||
|
c.Abort()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,210 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *Creative) PreviewDramaImport(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 4<<20)
|
||||||
|
if err := c.Request.ParseMultipartForm(4 << 20); err != nil {
|
||||||
|
h.bad(c, errors.New("导入内容无效"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if c.Request.MultipartForm != nil {
|
||||||
|
defer c.Request.MultipartForm.RemoveAll()
|
||||||
|
}
|
||||||
|
rawText := strings.TrimSpace(c.PostForm("raw_text"))
|
||||||
|
var filename string
|
||||||
|
var content []byte
|
||||||
|
file, header, err := c.Request.FormFile("file")
|
||||||
|
if err == nil {
|
||||||
|
defer file.Close()
|
||||||
|
filename = header.Filename
|
||||||
|
content, err = io.ReadAll(io.LimitReader(file, 3*1024*1024+1))
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("读取小说文件失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
} else if rawText == "" {
|
||||||
|
h.bad(c, errors.New("请选择小说文件或粘贴文本"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
preview, err := h.service.PreviewDramaImport(currentWebUser(c).ID, projectID, filename, content, rawText)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": preview})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ConfirmDramaImport(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var body struct {
|
||||||
|
ImportToken uuid.UUID `json:"import_token"`
|
||||||
|
ChaptersPerEpisode int `json:"chapters_per_episode"`
|
||||||
|
StartEpisodeNo int `json:"start_episode_no"`
|
||||||
|
TreatAsSingleEpisode bool `json:"treat_as_single_episode"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil || body.ImportToken == uuid.Nil {
|
||||||
|
h.bad(c, errors.New("导入确认参数无效"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
items, err := h.service.ConfirmDramaImport(currentWebUser(c).ID, projectID, body.ImportToken, body.ChaptersPerEpisode, body.StartEpisodeNo, body.TreatAsSingleEpisode)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) GetDramaEpisodeSource(c *gin.Context) {
|
||||||
|
projectID, episodeID, ok := h.dramaEpisodeIDs(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
source, err := h.service.EpisodeSource(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": source})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) SaveDramaEpisodeSource(c *gin.Context) {
|
||||||
|
projectID, episodeID, ok := h.dramaEpisodeIDs(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var body struct {
|
||||||
|
Title string `json:"title"`
|
||||||
|
RawContent string `json:"raw_content"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
source, err := h.service.SaveEpisodeSource(currentWebUser(c).ID, projectID, episodeID, body.RawContent, body.Title)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": source})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) QueueDramaParse(c *gin.Context) {
|
||||||
|
projectID, episodeID, ok := h.dramaEpisodeIDs(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueDramaParse(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ReparseDrama(c *gin.Context) {
|
||||||
|
if !h.objectStorageAvailable(c) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
projectID, episodeID, ok := h.dramaEpisodeIDs(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
userID := currentWebUser(c).ID
|
||||||
|
objectKeys, err := h.service.ResetEpisodeAnalysis(userID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !h.deleteStoredObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueDramaParse(userID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ListDramaParseTasks(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var episodeID *uuid.UUID
|
||||||
|
if raw := strings.TrimSpace(c.Query("episode_id")); raw != "" {
|
||||||
|
parsed, err := service.ParseUUID(raw, "剧集")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
episodeID = &parsed
|
||||||
|
}
|
||||||
|
items, err := h.service.ListDramaParseTasks(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) CancelDramaParse(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
taskID, err := service.ParseUUID(c.Param("task_id"), "解析任务")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = h.service.CancelDramaParse(currentWebUser(c).ID, projectID, taskID); err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) CreateDramaStoryboard(c *gin.Context) {
|
||||||
|
projectID, episodeID, ok := h.dramaEpisodeIDs(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
storyboard, err := h.service.CreateStoryboard(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": storyboard})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) dramaEpisodeIDs(c *gin.Context) (uuid.UUID, uuid.UUID, bool) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return uuid.Nil, uuid.Nil, false
|
||||||
|
}
|
||||||
|
episodeID, err := service.ParseUUID(c.Param("episode_id"), "剧集")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return uuid.Nil, uuid.Nil, false
|
||||||
|
}
|
||||||
|
return projectID, episodeID, true
|
||||||
|
}
|
||||||
@@ -0,0 +1,48 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/redis/go-redis/v9"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Health struct {
|
||||||
|
db *sql.DB
|
||||||
|
redis *redis.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewHealth(db *sql.DB, redisClient *redis.Client) *Health {
|
||||||
|
return &Health{db: db, redis: redisClient}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Health) Check(c *gin.Context) {
|
||||||
|
ctx, cancel := context.WithTimeout(c.Request.Context(), 2*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
services := map[string]string{
|
||||||
|
"postgresql": "ok",
|
||||||
|
"redis": "ok",
|
||||||
|
}
|
||||||
|
status := "ok"
|
||||||
|
statusCode := http.StatusOK
|
||||||
|
|
||||||
|
if err := h.db.PingContext(ctx); err != nil {
|
||||||
|
services["postgresql"] = "unavailable"
|
||||||
|
status = "degraded"
|
||||||
|
statusCode = http.StatusServiceUnavailable
|
||||||
|
}
|
||||||
|
if err := h.redis.Ping(ctx).Err(); err != nil {
|
||||||
|
services["redis"] = "unavailable"
|
||||||
|
status = "degraded"
|
||||||
|
statusCode = http.StatusServiceUnavailable
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(statusCode, gin.H{
|
||||||
|
"status": status,
|
||||||
|
"services": services,
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -0,0 +1,236 @@
|
|||||||
|
// 图片生成接口处理器，负责生图参数校验、参考图上传、历史查询和结果删除。
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/sha256"
|
||||||
|
"encoding/hex"
|
||||||
|
"errors"
|
||||||
|
"image"
|
||||||
|
"io"
|
||||||
|
"mime"
|
||||||
|
"net/http"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
mediakey "juhe-factory/api/internal/media"
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
"juhe-factory/api/internal/modules/productimage"
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
"juhe-factory/api/internal/storage"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ProductImage 组合图片生成生图服务与对象存储依赖。
|
||||||
|
type ProductImage struct {
|
||||||
|
service *productimage.Service
|
||||||
|
cos *storage.COS
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewProductImage 创建图片生成 HTTP 处理器。
|
||||||
|
func NewProductImage(service *productimage.Service, cos *storage.COS) *ProductImage {
|
||||||
|
return &ProductImage{service: service, cos: cos}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate 接收当前页面的临时配置和参考图，并创建图片生成任务。
|
||||||
|
func (h *ProductImage) Generate(c *gin.Context) {
|
||||||
|
if h.cos == nil {
|
||||||
|
fail(c, http.StatusServiceUnavailable, "cos_not_configured", "对象存储未配置")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 48<<20)
|
||||||
|
if err := c.Request.ParseMultipartForm(48 << 20); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "生图请求无效")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if c.Request.MultipartForm != nil {
|
||||||
|
defer c.Request.MultipartForm.RemoveAll()
|
||||||
|
}
|
||||||
|
modelID, err := service.ParseUUID(c.PostForm("model_id"), "图片模型")
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ratio := strings.TrimSpace(c.PostForm("aspect_ratio"))
|
||||||
|
if ratio == "" {
|
||||||
|
ratio = "9:16"
|
||||||
|
}
|
||||||
|
resolution := strings.TrimSpace(c.PostForm("resolution"))
|
||||||
|
if resolution == "" {
|
||||||
|
resolution = "1k"
|
||||||
|
}
|
||||||
|
userID := currentWebUser(c).ID
|
||||||
|
files := c.Request.MultipartForm.File["references"]
|
||||||
|
if len(files) > 4 {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "最多上传 4 张参考图")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
references := make([]productimage.ReferenceUpload, 0, len(files))
|
||||||
|
objectKeys := make([]string, 0, len(files))
|
||||||
|
for index, header := range files {
|
||||||
|
file, openErr := header.Open()
|
||||||
|
if openErr != nil {
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "参考图无法读取")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
contentType := cleanProductImageContentType(header.Header.Get("Content-Type"))
|
||||||
|
if contentType == "" {
|
||||||
|
contentType = cleanProductImageContentType(mime.TypeByExtension(strings.ToLower(filepath.Ext(header.Filename))))
|
||||||
|
}
|
||||||
|
allowed := map[string]bool{"image/jpeg": true, "image/png": true, "image/webp": true}
|
||||||
|
if !allowed[contentType] || header.Size <= 0 || header.Size > h.cos.MaxImageBytes() {
|
||||||
|
file.Close()
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "参考图格式或大小不符合要求")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
config, _, decodeErr := image.DecodeConfig(file)
|
||||||
|
if decodeErr != nil || config.Width <= 0 || config.Height <= 0 {
|
||||||
|
file.Close()
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "参考图内容无法解析")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if _, err = file.Seek(0, io.SeekStart); err != nil {
|
||||||
|
file.Close()
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
mediaID := uuid.New()
|
||||||
|
key := mediakey.ProductImageReference(userID, mediaID, header.Filename, contentType)
|
||||||
|
hash := sha256.New()
|
||||||
|
if _, err = io.Copy(hash, file); err != nil {
|
||||||
|
file.Close()
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if _, err = file.Seek(0, io.SeekStart); err != nil {
|
||||||
|
file.Close()
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
url, putErr := h.cos.Put(c, key, contentType, file, header.Size)
|
||||||
|
file.Close()
|
||||||
|
if putErr != nil {
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.respondError(c, putErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
objectKeys = append(objectKeys, key)
|
||||||
|
name := strings.TrimSpace(filepath.Base(header.Filename))
|
||||||
|
if name == "" {
|
||||||
|
name = "参考图" + string(rune('1'+index))
|
||||||
|
}
|
||||||
|
owner := userID
|
||||||
|
references = append(references, productimage.ReferenceUpload{Asset: &model.MediaAsset{ID: mediaID, OwnerUserID: &owner, StorageProvider: "cos", ObjectKey: key, PublicURL: url, OriginalName: name, DisplayName: "图片生成参考图", MimeType: contentType, SizeBytes: header.Size, SHA256: hex.EncodeToString(hash.Sum(nil)), Width: &config.Width, Height: &config.Height}, Name: name})
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueGeneration(userID, productimage.GenerateInput{Prompt: c.PostForm("prompt"), ModelID: modelID, AspectRatio: ratio, Resolution: resolution, References: references})
|
||||||
|
if err != nil {
|
||||||
|
if !h.cleanupUploadedObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": gin.H{"task_id": task.ID}})
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListGenerations 返回当前用户最近的图片生成历史。
|
||||||
|
func (h *ProductImage) ListGenerations(c *gin.Context) {
|
||||||
|
items, err := h.service.Generations(currentWebUser(c).ID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteGeneration 删除当前用户的一条图片生成记录及其 COS 对象。
|
||||||
|
func (h *ProductImage) DeleteGeneration(c *gin.Context) {
|
||||||
|
taskID, err := service.ParseUUID(c.Param("task_id"), "图片生成任务")
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = h.service.DeleteGeneration(currentWebUser(c).ID, taskID, func(keys []string) error {
|
||||||
|
return h.deleteObjects(c.Request.Context(), keys)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
// deleteObjects 删除一组已经精确解析的 COS 对象键。
|
||||||
|
func (h *ProductImage) deleteObjects(ctx context.Context, keys []string) error {
|
||||||
|
if h.cos == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for _, key := range keys {
|
||||||
|
if err := h.cos.Delete(ctx, key); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// cleanupUploadedObjects 清理提交失败前已经上传的参考图，并统一响应清理错误。
|
||||||
|
func (h *ProductImage) cleanupUploadedObjects(c *gin.Context, keys []string) bool {
|
||||||
|
if err := h.deleteObjects(c.Request.Context(), keys); err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// cleanProductImageContentType 去除图片生成上传类型的参数部分。
|
||||||
|
func cleanProductImageContentType(value string) string {
|
||||||
|
return strings.TrimSpace(strings.Split(value, ";")[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
// respondError 将图片生成模块错误转换为一致的 Web API 响应。
|
||||||
|
func (h *ProductImage) respondError(c *gin.Context, err error) {
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
fail(c, http.StatusNotFound, "not_found", "图片生成记录不存在")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
message := strings.TrimSpace(err.Error())
|
||||||
|
if message == "" {
|
||||||
|
message = "请求处理失败"
|
||||||
|
}
|
||||||
|
if strings.Contains(message, "不能为空") || strings.Contains(message, "不能") || strings.Contains(message, "无效") || strings.Contains(message, "请选择") || strings.Contains(message, "最多") || strings.Contains(message, "不可用") || strings.Contains(message, "未配置") {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", message)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.Contains(message, "积分") {
|
||||||
|
fail(c, http.StatusConflict, "insufficient_points", message)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.Contains(message, "生成中") {
|
||||||
|
fail(c, http.StatusConflict, "generation_active", message)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusInternalServerError, "internal_error", message)
|
||||||
|
}
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestValidateRedemptionBatchLimits(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
points any
|
||||||
|
quantity int
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{name: "minimum", points: 1, quantity: 1},
|
||||||
|
{name: "maximum", points: 100, quantity: 10},
|
||||||
|
{name: "decimal points", points: 1.5, quantity: 2},
|
||||||
|
{name: "points below minimum", points: 0.99, quantity: 1, wantErr: true},
|
||||||
|
{name: "points above maximum", points: 100.01, quantity: 1, wantErr: true},
|
||||||
|
{name: "quantity below minimum", points: 1, quantity: 0, wantErr: true},
|
||||||
|
{name: "quantity above maximum", points: 1, quantity: 11, wantErr: true},
|
||||||
|
{name: "invalid points", points: "invalid", quantity: 1, wantErr: true},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
_, err := validateRedemptionBatchLimits(test.points, test.quantity)
|
||||||
|
if (err != nil) != test.wantErr {
|
||||||
|
t.Fatalf("validateRedemptionBatchLimits(%v, %d) error = %v", test.points, test.quantity, err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,257 @@
|
|||||||
|
// 视频重绘接口处理器，负责重绘工作台、分析任务和源媒体上传请求。
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
mediakey "juhe-factory/api/internal/media"
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *Creative) GetRedrawWorkbench(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
data, err := h.service.GetRedrawWorkbench(currentWebUser(c).ID, projectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": data})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) GetRedrawEpisodeWorkbench(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
episodeID, err := service.ParseUUID(c.Param("episode_id"), "剧集")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
data, err := h.service.GetRedrawEpisodeWorkbench(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": data})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) QueueRedrawEpisodeScript(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
episodeID, err := service.ParseUUID(c.Param("episode_id"), "剧集")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueRedrawEpisodeScript(currentWebUser(c).ID, projectID, episodeID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveRedrawEpisodeScript 保存用户修改后的剧集反推剧本正文。
|
||||||
|
func (h *Creative) SaveRedrawEpisodeScript(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
episodeID, err := service.ParseUUID(c.Param("episode_id"), "剧集")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var body struct {
|
||||||
|
Content string `json:"content"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.SaveRedrawEpisodeScript(currentWebUser(c).ID, projectID, episodeID, body.Content); err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) UpdateRedrawAnalysisSettings(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var body struct {
|
||||||
|
AudioSource string `json:"audio_source"`
|
||||||
|
SourceLanguage string `json:"source_language"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.UpdateRedrawAnalysisSettings(currentWebUser(c).ID, projectID, body.AudioSource, body.SourceLanguage); err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) QueueRedrawAnalysis(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueRedrawAnalysis(currentWebUser(c).ID, projectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ReanalyzeRedraw(c *gin.Context) {
|
||||||
|
if !h.objectStorageAvailable(c) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
objectKeys, err := h.service.ResetRedrawAnalysis(currentWebUser(c).ID, projectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !h.deleteStoredObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueRedrawAnalysis(currentWebUser(c).ID, projectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) QueueRedrawStoryboardAnalysis(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
storyboardID, err := service.ParseUUID(c.Param("storyboard_id"), "分镜")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueRedrawStoryboardAnalysis(currentWebUser(c).ID, projectID, storyboardID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) QueueRedrawScript(c *gin.Context) {
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueRedrawScript(currentWebUser(c).ID, projectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) UploadRedrawSource(c *gin.Context) {
|
||||||
|
h.uploadRedrawMedia(c, "source")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) UploadRedrawSubtitle(c *gin.Context) {
|
||||||
|
h.uploadRedrawMedia(c, "subtitle")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) uploadRedrawMedia(c *gin.Context, kind string) {
|
||||||
|
if h.cos == nil {
|
||||||
|
fail(c, http.StatusServiceUnavailable, "cos_not_configured", "COS 对象存储未配置")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
projectID, ok := h.projectID(c)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
target, err := h.service.PrepareRedrawMediaUpload(currentWebUser(c).ID, projectID, kind)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
project := target.Project
|
||||||
|
file, header, err := c.Request.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("请选择文件"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
contentType := cleanContentType(header.Header.Get("Content-Type"))
|
||||||
|
maxBytes := int64(50 * 1024 * 1024)
|
||||||
|
ext := strings.ToLower(filepath.Ext(header.Filename))
|
||||||
|
allowedVideoTypes := map[string]bool{"video/mp4": true, "video/webm": true, "video/quicktime": true, "video/x-m4v": true}
|
||||||
|
allowed := allowedVideoTypes[contentType] && map[string]bool{".mp4": true, ".webm": true, ".mov": true, ".m4v": true}[ext]
|
||||||
|
if kind == "subtitle" {
|
||||||
|
maxBytes = 2 * 1024 * 1024
|
||||||
|
allowed = ext == ".srt" || ext == ".vtt" || ext == ".txt"
|
||||||
|
if contentType == "" {
|
||||||
|
contentType = "text/plain"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !allowed || header.Size <= 0 || header.Size > maxBytes {
|
||||||
|
h.bad(c, fmt.Errorf("%s文件格式或大小不符合要求", map[bool]string{true: "字幕", false: "视频"}[kind == "subtitle"]))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
oldMediaID := target.OldMediaID
|
||||||
|
if kind == "source" && project.SourceVideoAssetID != nil {
|
||||||
|
objectKeys, err := h.service.ResetRedrawAnalysis(currentWebUser(c).ID, projectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !h.deleteStoredObjects(c, objectKeys) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mediaID := uuid.New()
|
||||||
|
key := mediakey.ProjectSource(projectID, mediaID, header.Filename, contentType)
|
||||||
|
displayName := project.Name + "原视频"
|
||||||
|
if kind == "subtitle" {
|
||||||
|
key = mediakey.ProjectSubtitle(projectID, mediaID, header.Filename, contentType)
|
||||||
|
displayName = project.Name + "字幕"
|
||||||
|
}
|
||||||
|
asset, err := h.storeUpload(c, mediaID, currentWebUser(c).ID, file, header, key, displayName, contentType, nil)
|
||||||
|
if err != nil {
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.PersistRedrawMediaUpload(projectID, kind, project.RedrawStatus, asset, oldMediaID); err != nil {
|
||||||
|
_ = h.cos.Delete(c.Request.Context(), asset.ObjectKey)
|
||||||
|
h.internal(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if target.OldObjectKey != "" && !h.deleteStoredObjects(c, []string{target.OldObjectKey}) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": asset})
|
||||||
|
}
|
||||||
@@ -0,0 +1,166 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *Creative) ListScriptAnalyses(c *gin.Context) {
|
||||||
|
items, err := h.service.ListScriptAnalyses(currentWebUser(c).ID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) CreateScriptAnalysis(c *gin.Context) {
|
||||||
|
item, err := h.service.CreateScriptAnalysis(currentWebUser(c).ID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusCreated, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ListScriptAnalysisImportProjects(c *gin.Context) {
|
||||||
|
items, err := h.service.ListScriptAnalysisImportProjects(currentWebUser(c).ID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ImportScriptAnalysisProject(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var body struct {
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil || body.ProjectID == uuid.Nil {
|
||||||
|
h.bad(c, errors.New("请选择要导入的剧本反推项目"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
item, err := h.service.ImportScriptAnalysisProject(currentWebUser(c).ID, id, body.ProjectID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) ImportScriptAnalysisFile(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 4<<20)
|
||||||
|
if err := c.Request.ParseMultipartForm(4 << 20); err != nil {
|
||||||
|
h.bad(c, errors.New("导入文件无效"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if c.Request.MultipartForm != nil {
|
||||||
|
defer c.Request.MultipartForm.RemoveAll()
|
||||||
|
}
|
||||||
|
file, header, err := c.Request.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("请选择剧本文件"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
content, err := io.ReadAll(io.LimitReader(file, 3*1024*1024+1))
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, errors.New("读取剧本文件失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
item, err := h.service.ImportScriptAnalysisFile(currentWebUser(c).ID, id, header.Filename, content)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) GetScriptAnalysis(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
item, err := h.service.GetScriptAnalysis(currentWebUser(c).ID, id)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) UpdateScriptAnalysis(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var input service.ScriptAnalysisInput
|
||||||
|
if err := c.ShouldBindJSON(&input); err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
item, err := h.service.UpdateScriptAnalysis(currentWebUser(c).ID, id, input)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": item})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) DeleteScriptAnalysis(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.DeleteScriptAnalysis(currentWebUser(c).ID, id); err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) QueueScriptAnalysis(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task, err := h.service.QueueScriptAnalysis(currentWebUser(c).ID, id)
|
||||||
|
if err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusAccepted, gin.H{"data": task})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Creative) CancelScriptAnalysis(c *gin.Context) {
|
||||||
|
id, err := service.ParseUUID(c.Param("script_id"), "剧本")
|
||||||
|
if err != nil {
|
||||||
|
h.bad(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.CancelScriptAnalysis(currentWebUser(c).ID, id); err != nil {
|
||||||
|
h.respondError(c, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
@@ -0,0 +1,226 @@
|
|||||||
|
// WEB 用户接口处理器，负责认证、账户资料、积分查询、头像上传和兑换请求。
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
mediakey "juhe-factory/api/internal/media"
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
"juhe-factory/api/internal/service"
|
||||||
|
"juhe-factory/api/internal/storage"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const webUserContextKey = "web_user"
|
||||||
|
|
||||||
|
type Web struct {
|
||||||
|
service *service.Web
|
||||||
|
cos *storage.COS
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewWeb(service *service.Web, cos *storage.COS) *Web { return &Web{service: service, cos: cos} }
|
||||||
|
|
||||||
|
func (h *Web) Login(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Account string `json:"account" binding:"required"`
|
||||||
|
Password string `json:"password" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请输入账号和密码")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
pair, err := h.service.Login(body.Account, body.Password)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusUnauthorized, "invalid_credentials", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": pair})
|
||||||
|
}
|
||||||
|
func (h *Web) Refresh(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
RefreshToken string `json:"refresh_token" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "缺少刷新令牌")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
pair, err := h.service.Refresh(body.RefreshToken)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusUnauthorized, "invalid_refresh_token", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": pair})
|
||||||
|
}
|
||||||
|
func (h *Web) Logout(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
RefreshToken string `json:"refresh_token"`
|
||||||
|
}
|
||||||
|
_ = c.ShouldBindJSON(&body)
|
||||||
|
if err := h.service.Logout(body.RefreshToken); err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "logout_failed", "退出失败，请重试")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
func (h *Web) Middleware() gin.HandlerFunc {
|
||||||
|
return func(c *gin.Context) {
|
||||||
|
raw := strings.TrimSpace(strings.TrimPrefix(c.GetHeader("Authorization"), "Bearer "))
|
||||||
|
user, err := h.service.Authenticate(raw)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusUnauthorized, "unauthorized", err.Error())
|
||||||
|
c.Abort()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Set(webUserContextKey, user)
|
||||||
|
c.Next()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Account 返回当前用户账户资料、积分额度和头像上传限制。
|
||||||
|
func (h *Web) Account(c *gin.Context) {
|
||||||
|
data, err := h.service.Account(currentWebUser(c).ID)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "account_failed", "账户信息加载失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if h.cos != nil {
|
||||||
|
data["avatar_max_size_bytes"] = h.cos.MaxImageBytes()
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": data})
|
||||||
|
}
|
||||||
|
func (h *Web) UpdateProfile(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Username string `json:"username" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请输入用户名")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
data, err := h.service.UpdateProfile(currentWebUser(c).ID, body.Username)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, service.ErrInvalidUsername) || errors.Is(err, service.ErrUsernameExists) {
|
||||||
|
fail(c, http.StatusBadRequest, "profile_invalid", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusInternalServerError, "profile_update_failed", "资料更新失败，请稍后重试")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": data})
|
||||||
|
}
|
||||||
|
func (h *Web) ChangePassword(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
OriginalPassword string `json:"original_password" binding:"required"`
|
||||||
|
NewPassword string `json:"new_password" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请完整填写密码")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.ChangePassword(currentWebUser(c).ID, body.OriginalPassword, body.NewPassword); err != nil {
|
||||||
|
if errors.Is(err, service.ErrOriginalPassword) || strings.Contains(err.Error(), "密码长度") {
|
||||||
|
fail(c, http.StatusBadRequest, "password_invalid", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fail(c, http.StatusInternalServerError, "password_update_failed", "密码修改失败，请稍后重试")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Status(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
func (h *Web) UploadAvatar(c *gin.Context) {
|
||||||
|
if h.cos == nil {
|
||||||
|
fail(c, http.StatusServiceUnavailable, "cos_not_configured", "对象存储未配置")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
file, header, err := c.Request.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请选择头像图片")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
contentType := strings.TrimSpace(strings.Split(header.Header.Get("Content-Type"), ";")[0])
|
||||||
|
extensions := map[string]string{"image/jpeg": ".jpg", "image/png": ".png", "image/webp": ".webp"}
|
||||||
|
extension, allowed := extensions[contentType]
|
||||||
|
if !allowed || header.Size <= 0 || header.Size > h.cos.MaxImageBytes() {
|
||||||
|
fail(c, http.StatusBadRequest, "avatar_invalid", "头像仅支持 JPEG、PNG 或 WebP，且不得超过图片大小限制")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
headerBytes := make([]byte, 512)
|
||||||
|
read, readErr := file.Read(headerBytes)
|
||||||
|
if readErr != nil && !errors.Is(readErr, io.EOF) {
|
||||||
|
fail(c, http.StatusBadRequest, "avatar_invalid", "头像图片内容无法解析")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if detected := http.DetectContentType(headerBytes[:read]); detected != contentType {
|
||||||
|
fail(c, http.StatusBadRequest, "avatar_invalid", "头像图片内容无法解析")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if _, err := file.Seek(0, io.SeekStart); err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "avatar_upload_failed", "头像上传失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
userID := currentWebUser(c).ID
|
||||||
|
oldKey, err := h.service.AvatarKey(userID)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "avatar_upload_failed", "头像上传失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
key := mediakey.UserAvatar(userID, uuid.New(), extension)
|
||||||
|
url, err := h.cos.Put(c.Request.Context(), key, contentType, file, header.Size)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "avatar_upload_failed", "头像上传失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.service.UpdateAvatar(userID, url, key); err != nil {
|
||||||
|
_ = h.cos.Delete(c.Request.Context(), key)
|
||||||
|
fail(c, http.StatusInternalServerError, "avatar_upload_failed", "头像上传失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if oldKey != "" {
|
||||||
|
_ = h.cos.Delete(c.Request.Context(), oldKey)
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": gin.H{"avatar_url": url}})
|
||||||
|
}
|
||||||
|
func (h *Web) Usage30Days(c *gin.Context) {
|
||||||
|
items, err := h.service.Usage30Days(currentWebUser(c).ID)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "usage_failed", "消耗数据加载失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": items})
|
||||||
|
}
|
||||||
|
|
||||||
|
// ConsumptionRecords 按请求页码返回当前用户的积分变动流水。
|
||||||
|
func (h *Web) ConsumptionRecords(c *gin.Context) {
|
||||||
|
page, pageSize := parsePage(c)
|
||||||
|
result, err := h.service.ConsumptionRecords(currentWebUser(c).ID, page, pageSize)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusInternalServerError, "consumption_records_failed", "消耗记录加载失败")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": result})
|
||||||
|
}
|
||||||
|
func (h *Web) Redeem(c *gin.Context) {
|
||||||
|
var body struct {
|
||||||
|
Code string `json:"code" binding:"required"`
|
||||||
|
}
|
||||||
|
if err := c.ShouldBindJSON(&body); err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "invalid_request", "请输入兑换码")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
result, err := h.service.Redeem(currentWebUser(c).ID, body.Code)
|
||||||
|
if err != nil {
|
||||||
|
fail(c, http.StatusBadRequest, "redeem_failed", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"data": result})
|
||||||
|
}
|
||||||
|
func currentWebUser(c *gin.Context) *model.WebUser {
|
||||||
|
value, _ := c.Get(webUserContextKey)
|
||||||
|
user, _ := value.(*model.WebUser)
|
||||||
|
return user
|
||||||
|
}
|
||||||
@@ -0,0 +1,156 @@
|
|||||||
|
// Package media 统一生成可长期稳定引用的 COS 媒体对象键和扩展名。
|
||||||
|
package media
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const objectRoot = "juyou_ran"
|
||||||
|
|
||||||
|
var unsafeKeyPart = regexp.MustCompile(`[^a-z0-9-]+`)
|
||||||
|
var safeExtension = regexp.MustCompile(`^\.[a-z0-9]+$`)
|
||||||
|
|
||||||
|
// objectKey 为所有媒体对象添加统一的业务根目录，避免不同业务直接污染存储桶根目录。
|
||||||
|
func objectKey(format string, args ...any) string {
|
||||||
|
return fmt.Sprintf(objectRoot+"/"+format, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Slug 将业务名称转换为受控英文路径片段；展示名称保留在数据库中。
|
||||||
|
func Slug(value, fallback string) string {
|
||||||
|
value = strings.ToLower(strings.TrimSpace(value))
|
||||||
|
value = unsafeKeyPart.ReplaceAllString(value, "-")
|
||||||
|
value = strings.Trim(value, "-")
|
||||||
|
if value == "" {
|
||||||
|
value = fallback
|
||||||
|
}
|
||||||
|
if len(value) > 48 {
|
||||||
|
value = strings.Trim(value[:48], "-")
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extension 根据媒体类型或安全的原文件扩展名生成对象扩展名。
|
||||||
|
func Extension(filename, contentType string) string {
|
||||||
|
switch strings.ToLower(contentType) {
|
||||||
|
case "image/jpeg":
|
||||||
|
return ".jpg"
|
||||||
|
case "image/png":
|
||||||
|
return ".png"
|
||||||
|
case "image/webp":
|
||||||
|
return ".webp"
|
||||||
|
case "image/gif":
|
||||||
|
return ".gif"
|
||||||
|
case "video/mp4":
|
||||||
|
return ".mp4"
|
||||||
|
case "video/webm":
|
||||||
|
return ".webm"
|
||||||
|
case "audio/mpeg":
|
||||||
|
return ".mp3"
|
||||||
|
case "audio/wav", "audio/x-wav":
|
||||||
|
return ".wav"
|
||||||
|
case "text/vtt":
|
||||||
|
return ".vtt"
|
||||||
|
}
|
||||||
|
ext := strings.ToLower(filepath.Ext(filename))
|
||||||
|
if len(ext) > 1 && len(ext) <= 8 && safeExtension.MatchString(ext) {
|
||||||
|
return ext
|
||||||
|
}
|
||||||
|
return ".bin"
|
||||||
|
}
|
||||||
|
|
||||||
|
// EpisodeSource 返回剧集原视频的对象键。
|
||||||
|
func EpisodeSource(projectID uuid.UUID, episodeNo int, episodeID, assetID uuid.UUID, filename, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/episodes/%03d-%s/source/%s-source%s", projectID, episodeNo, episodeID, assetID, Extension(filename, contentType))
|
||||||
|
}
|
||||||
|
|
||||||
|
// EpisodeSubtitle 返回剧集字幕的对象键。
|
||||||
|
func EpisodeSubtitle(projectID uuid.UUID, episodeNo int, episodeID, assetID uuid.UUID, filename, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/episodes/%03d-%s/subtitles/%s-subtitle%s", projectID, episodeNo, episodeID, assetID, Extension(filename, contentType))
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProjectSource 返回无剧集项目原视频的对象键。
|
||||||
|
func ProjectSource(projectID, assetID uuid.UUID, filename, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/source/%s-source%s", projectID, assetID, Extension(filename, contentType))
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProjectSubtitle 返回无剧集项目字幕的对象键。
|
||||||
|
func ProjectSubtitle(projectID, assetID uuid.UUID, filename, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/subtitles/%s-subtitle%s", projectID, assetID, Extension(filename, contentType))
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssetReference 返回项目资产参考媒体的对象键。
|
||||||
|
func AssetReference(projectID uuid.UUID, assetType string, assetID, mediaID uuid.UUID, filename, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/assets/%s/%s/%s-reference%s", projectID, Slug(assetType, "custom"), assetID, mediaID, Extension(filename, contentType))
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProductImageReference 返回独立图片生成工具参考图片的稳定对象键。
|
||||||
|
func ProductImageReference(userID, mediaID uuid.UUID, filename, contentType string) string {
|
||||||
|
return objectKey("product-images/users/%s/references/%s%s", userID, mediaID, Extension(filename, contentType))
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssetOutput 返回项目资产生成结果的对象键。
|
||||||
|
func AssetOutput(projectID uuid.UUID, assetType string, assetID, taskID, mediaID uuid.UUID, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/assets/%s/%s/outputs/%s-%s-result%s", projectID, Slug(assetType, "custom"), assetID, taskID, mediaID, Extension("", contentTypeFor("image", contentType)))
|
||||||
|
}
|
||||||
|
|
||||||
|
// StoryboardFrame 返回剧集分镜抽帧的对象键。
|
||||||
|
func StoryboardFrame(projectID, episodeID, storyboardID, mediaID uuid.UUID, sequence int, timestampMS int64) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/episodes/%s/storyboards/%04d-%s/frames/%s-%010d.jpg", projectID, episodeID, sequence, storyboardID, mediaID, timestampMS)
|
||||||
|
}
|
||||||
|
|
||||||
|
// StoryboardAnalysis 返回剧集分镜反推原始响应的对象键。
|
||||||
|
func StoryboardAnalysis(projectID, episodeID, storyboardID, taskID, mediaID uuid.UUID, sequence int) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/episodes/%s/storyboards/%04d-%s/analysis/%s-%s-reverse.json", projectID, episodeID, sequence, storyboardID, taskID, mediaID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TaskOutput 返回剧集分镜生成结果的对象键。
|
||||||
|
func TaskOutput(projectID, episodeID uuid.UUID, sequence int, taskID, mediaID uuid.UUID, outputType, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/episodes/%s/storyboards/%04d/outputs/%s-%s-result%s", projectID, episodeID, sequence, taskID, mediaID, Extension("", contentTypeFor(outputType, contentType)))
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProjectStoryboardFrame 返回无剧集项目分镜抽帧的对象键。
|
||||||
|
func ProjectStoryboardFrame(projectID, storyboardID, mediaID uuid.UUID, sequence int, timestampMS int64) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/storyboards/%04d-%s/frames/%s-%010d.jpg", projectID, sequence, storyboardID, mediaID, timestampMS)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProjectStoryboardAnalysis 返回无剧集项目分镜反推原始响应的对象键。
|
||||||
|
func ProjectStoryboardAnalysis(projectID, storyboardID, taskID, mediaID uuid.UUID, sequence int) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/storyboards/%04d-%s/analysis/%s-%s-reverse.json", projectID, sequence, storyboardID, taskID, mediaID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProjectTaskOutput 返回无剧集项目分镜生成结果的对象键。
|
||||||
|
func ProjectTaskOutput(projectID uuid.UUID, sequence int, taskID, mediaID uuid.UUID, outputType, contentType string) string {
|
||||||
|
return objectKey("video-redraw/projects/%s/storyboards/%04d/outputs/%s-%s-result%s", projectID, sequence, taskID, mediaID, Extension("", contentTypeFor(outputType, contentType)))
|
||||||
|
}
|
||||||
|
|
||||||
|
// StyleImage 返回管理端风格图片的统一对象键。
|
||||||
|
func StyleImage(mediaID uuid.UUID, extension string) string {
|
||||||
|
return objectKey("styles/%s/%s%s", time.Now().Format("2006/01"), mediaID, extension)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UserAvatar 返回用户头像的统一对象键。
|
||||||
|
func UserAvatar(userID, mediaID uuid.UUID, extension string) string {
|
||||||
|
return objectKey("users/%s/avatar/%s%s", userID, mediaID, extension)
|
||||||
|
}
|
||||||
|
|
||||||
|
// StandaloneTaskOutput 返回独立生成任务的结果对象键。
|
||||||
|
func StandaloneTaskOutput(userID, taskID uuid.UUID, outputType, contentType string) string {
|
||||||
|
return objectKey("users/%s/tasks/%s/output-1%s", userID, taskID, Extension("", contentTypeFor(outputType, contentType)))
|
||||||
|
}
|
||||||
|
|
||||||
|
// contentTypeFor 返回已知内容类型，缺省时根据输出类型选择默认媒体类型。
|
||||||
|
func contentTypeFor(outputType, contentType string) string {
|
||||||
|
if contentType != "" {
|
||||||
|
return contentType
|
||||||
|
}
|
||||||
|
if outputType == "image" {
|
||||||
|
return "image/png"
|
||||||
|
}
|
||||||
|
return "video/mp4"
|
||||||
|
}
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
package media
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestAssetKeysUseControlledSegments(t *testing.T) {
|
||||||
|
projectID, assetID, taskID, mediaID := uuid.New(), uuid.New(), uuid.New(), uuid.New()
|
||||||
|
key := AssetOutput(projectID, "角色 中文", assetID, taskID, mediaID, "image/webp")
|
||||||
|
wantPrefix := "juyou_ran/video-redraw/projects/" + projectID.String() + "/assets/custom/" + assetID.String() + "/outputs/"
|
||||||
|
if !strings.HasPrefix(key, wantPrefix) || !strings.HasSuffix(key, ".webp") {
|
||||||
|
t.Fatalf("unexpected key: %s", key)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,46 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AdminUser struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
Username string `json:"username"`
|
||||||
|
PasswordHash string `json:"-"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
LastLoginAt *time.Time `json:"last_login_at"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt gorm.DeletedAt `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AdminRefreshToken struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey"`
|
||||||
|
AdminID uuid.UUID `gorm:"type:uuid"`
|
||||||
|
TokenHash string
|
||||||
|
ExpiresAt time.Time
|
||||||
|
RevokedAt *time.Time
|
||||||
|
CreatedAt time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
type AdminAuditLog struct {
|
||||||
|
ID int64 `json:"id"`
|
||||||
|
AdminID *uuid.UUID `gorm:"type:uuid" json:"admin_id"`
|
||||||
|
AdminUsername string `json:"admin_username"`
|
||||||
|
Action string `json:"action"`
|
||||||
|
ResourceType string `json:"resource_type"`
|
||||||
|
ResourceID string `json:"resource_id"`
|
||||||
|
Reason string `json:"reason"`
|
||||||
|
Detail string `gorm:"type:jsonb" json:"detail"`
|
||||||
|
IPAddress string `gorm:"type:inet" json:"ip_address"`
|
||||||
|
TraceID string `json:"trace_id"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (AdminUser) TableName() string { return "admin_users" }
|
||||||
|
func (AdminRefreshToken) TableName() string { return "admin_refresh_tokens" }
|
||||||
|
func (AdminAuditLog) TableName() string { return "admin_audit_logs" }
|
||||||
@@ -0,0 +1,299 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MediaAsset struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
OwnerUserID *uuid.UUID `gorm:"type:uuid" json:"owner_user_id"`
|
||||||
|
StorageProvider string `json:"storage_provider"`
|
||||||
|
ObjectKey string `json:"object_key"`
|
||||||
|
PublicURL string `json:"public_url"`
|
||||||
|
OriginalName string `json:"original_name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
MimeType string `json:"mime_type"`
|
||||||
|
SizeBytes int64 `json:"size_bytes"`
|
||||||
|
SHA256 string `json:"sha256"`
|
||||||
|
Width *int `json:"width"`
|
||||||
|
Height *int `json:"height"`
|
||||||
|
DurationMS *int64 `json:"duration_ms"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
DeletedAt *time.Time `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CreativeProject struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid" json:"user_id"`
|
||||||
|
ProjectType string `json:"project_type"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
CoverAssetID *uuid.UUID `gorm:"type:uuid" json:"cover_asset_id"`
|
||||||
|
StyleID uuid.UUID `gorm:"type:uuid" json:"style_id"`
|
||||||
|
EraType string `json:"era_type"`
|
||||||
|
CustomEra *string `json:"custom_era"`
|
||||||
|
AspectRatio string `json:"aspect_ratio"`
|
||||||
|
Localization *string `json:"localization"`
|
||||||
|
SourceVideoAssetID *uuid.UUID `gorm:"type:uuid" json:"source_video_asset_id"`
|
||||||
|
SubtitleAssetID *uuid.UUID `gorm:"type:uuid" json:"subtitle_asset_id"`
|
||||||
|
AudioSource string `json:"audio_source"`
|
||||||
|
SourceLanguage *string `json:"source_language"`
|
||||||
|
RedrawStatus string `json:"redraw_status"`
|
||||||
|
AnalysisMessage string `json:"analysis_message"`
|
||||||
|
RedrawScript string `json:"redraw_script"`
|
||||||
|
ShortDramaType *string `json:"short_drama_type"`
|
||||||
|
PlayCount *string `json:"play_count"`
|
||||||
|
AudienceProfile *string `json:"audience_profile"`
|
||||||
|
Producer *string `json:"producer"`
|
||||||
|
CastMembers json.RawMessage `gorm:"type:jsonb;not null;default:'[]'" json:"cast_members"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt *time.Time `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (project *CreativeProject) BeforeCreate(_ *gorm.DB) error {
|
||||||
|
if len(project.CastMembers) == 0 {
|
||||||
|
project.CastMembers = json.RawMessage("[]")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProjectModelConfig struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
ProjectID uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
Purpose string `json:"purpose"`
|
||||||
|
ModelType string `json:"model_type"`
|
||||||
|
ModelID uuid.UUID `gorm:"type:uuid" json:"model_id"`
|
||||||
|
PromptID *uuid.UUID `gorm:"type:uuid" json:"prompt_id"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserModelConfig struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid" json:"user_id"`
|
||||||
|
Purpose string `json:"purpose"`
|
||||||
|
ProjectType string `json:"project_type"`
|
||||||
|
ModelType string `json:"model_type"`
|
||||||
|
ModelID uuid.UUID `gorm:"type:uuid" json:"model_id"`
|
||||||
|
PromptID *uuid.UUID `gorm:"type:uuid" json:"prompt_id"`
|
||||||
|
Settings json.RawMessage `gorm:"type:jsonb" json:"settings"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProjectEpisode struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
ProjectID uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
EpisodeNo int `json:"episode_no"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
CoverAssetID *uuid.UUID `gorm:"type:uuid" json:"cover_asset_id"`
|
||||||
|
SourceVideoAssetID *uuid.UUID `gorm:"type:uuid" json:"source_video_asset_id"`
|
||||||
|
SubtitleAssetID *uuid.UUID `gorm:"type:uuid" json:"subtitle_asset_id"`
|
||||||
|
AudioSource string `json:"audio_source"`
|
||||||
|
SourceLanguage *string `json:"source_language"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
AnalysisMessage string `json:"analysis_message"`
|
||||||
|
RedrawScript string `json:"redraw_script"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt *time.Time `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProjectAsset struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
ProjectID uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
SourceEpisodeID *uuid.UUID `gorm:"type:uuid" json:"source_episode_id"`
|
||||||
|
AssetType string `json:"asset_type"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
ImagePrompt string `json:"image_prompt"`
|
||||||
|
Appearances json.RawMessage `gorm:"type:jsonb" json:"appearances"`
|
||||||
|
ImageAssetID *uuid.UUID `gorm:"type:uuid" json:"image_asset_id"`
|
||||||
|
AudioAssetID *uuid.UUID `gorm:"type:uuid" json:"audio_asset_id"`
|
||||||
|
UserEdited bool `json:"user_edited"`
|
||||||
|
Aliases json.RawMessage `gorm:"type:jsonb" json:"aliases"`
|
||||||
|
Attributes json.RawMessage `gorm:"type:jsonb" json:"attributes"`
|
||||||
|
AIDescription string `gorm:"column:ai_description" json:"ai_description"`
|
||||||
|
SourceParseTaskID *uuid.UUID `gorm:"type:uuid" json:"source_parse_task_id"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt *time.Time `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (asset *ProjectAsset) BeforeCreate(_ *gorm.DB) error {
|
||||||
|
if len(asset.Appearances) == 0 {
|
||||||
|
asset.Appearances = json.RawMessage("[]")
|
||||||
|
}
|
||||||
|
if len(asset.Aliases) == 0 {
|
||||||
|
asset.Aliases = json.RawMessage("[]")
|
||||||
|
}
|
||||||
|
if len(asset.Attributes) == 0 {
|
||||||
|
asset.Attributes = json.RawMessage("{}")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type EpisodeStoryboard struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
EpisodeID *uuid.UUID `gorm:"type:uuid" json:"episode_id"`
|
||||||
|
ProjectID *uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
SequenceNo int `json:"sequence_no"`
|
||||||
|
StableKey string `json:"stable_key"`
|
||||||
|
StartMS int64 `json:"start_ms"`
|
||||||
|
EndMS int64 `json:"end_ms"`
|
||||||
|
DurationSeconds int `json:"duration_seconds"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
ThumbnailAssetID *uuid.UUID `gorm:"type:uuid" json:"thumbnail_asset_id"`
|
||||||
|
PromptContent string `json:"prompt_content"`
|
||||||
|
ImagePrompt string `json:"image_prompt"`
|
||||||
|
ScriptContent string `json:"script_content"`
|
||||||
|
SourceExcerpt string `json:"source_excerpt"`
|
||||||
|
Dialogue json.RawMessage `gorm:"type:jsonb" json:"dialogue"`
|
||||||
|
AssetRefs json.RawMessage `gorm:"type:jsonb" json:"asset_refs"`
|
||||||
|
Locked bool `json:"locked"`
|
||||||
|
ActiveOutputID *uuid.UUID `gorm:"type:uuid" json:"active_output_id"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
SourceParseTaskID *uuid.UUID `gorm:"type:uuid" json:"source_parse_task_id"`
|
||||||
|
UserEdited bool `json:"user_edited"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt *time.Time `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type EpisodeSource struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
EpisodeID uuid.UUID `gorm:"type:uuid;uniqueIndex" json:"episode_id"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
RawContent string `json:"raw_content"`
|
||||||
|
CharCount int `json:"char_count"`
|
||||||
|
SourceType string `json:"source_type"`
|
||||||
|
SourceFilename string `json:"source_filename"`
|
||||||
|
ContentSHA256 string `gorm:"column:content_sha256" json:"content_sha256"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DramaImportSession struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid" json:"user_id"`
|
||||||
|
ProjectID uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
SourceType string `json:"source_type"`
|
||||||
|
SourceFilename string `json:"source_filename"`
|
||||||
|
RawContent string `json:"-"`
|
||||||
|
ContentSHA256 string `gorm:"column:content_sha256" json:"content_sha256"`
|
||||||
|
PreviewData json.RawMessage `gorm:"type:jsonb" json:"preview_data"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
ConsumedAt *time.Time `json:"consumed_at"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DramaParseBatch struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
ProjectID uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid" json:"user_id"`
|
||||||
|
Mode string `json:"mode"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
TotalCount int `json:"total_count"`
|
||||||
|
CompletedCount int `json:"completed_count"`
|
||||||
|
FailedCount int `json:"failed_count"`
|
||||||
|
CurrentEpisodeID *uuid.UUID `gorm:"type:uuid" json:"current_episode_id"`
|
||||||
|
CancelRequestedAt *time.Time `json:"cancel_requested_at"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
FinishedAt *time.Time `json:"finished_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DramaParseTask struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
RequestID string `json:"request_id"`
|
||||||
|
BatchID *uuid.UUID `gorm:"type:uuid" json:"batch_id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid" json:"user_id"`
|
||||||
|
ProjectID uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
EpisodeID uuid.UUID `gorm:"type:uuid" json:"episode_id"`
|
||||||
|
ChannelID uuid.UUID `gorm:"type:uuid" json:"channel_id"`
|
||||||
|
ModelID uuid.UUID `gorm:"type:uuid" json:"model_id"`
|
||||||
|
SourceSHA256 string `gorm:"column:source_sha256" json:"source_sha256"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
AttemptCount int `json:"attempt_count"`
|
||||||
|
ModelSnapshot json.RawMessage `gorm:"type:jsonb" json:"model_snapshot"`
|
||||||
|
PromptSnapshot string `json:"prompt_snapshot"`
|
||||||
|
ContextSnapshot json.RawMessage `gorm:"type:jsonb" json:"context_snapshot"`
|
||||||
|
NormalizedResult json.RawMessage `gorm:"type:jsonb" json:"normalized_result"`
|
||||||
|
InputTokens *int64 `json:"input_tokens"`
|
||||||
|
OutputTokens *int64 `json:"output_tokens"`
|
||||||
|
TotalTokens *int64 `json:"total_tokens"`
|
||||||
|
TokenCountSource string `json:"token_count_source"`
|
||||||
|
UsageRaw json.RawMessage `gorm:"type:jsonb" json:"usage_raw"`
|
||||||
|
BillingSnapshot json.RawMessage `gorm:"type:jsonb;not null;default:'{}'" json:"billing_snapshot"`
|
||||||
|
EstimatedPoints string `gorm:"column:estimated_points" json:"estimated_points"`
|
||||||
|
PrepaidPoints string `gorm:"column:prepaid_points" json:"prepaid_points"`
|
||||||
|
ActualPoints *string `gorm:"column:actual_points" json:"actual_points"`
|
||||||
|
CostRefunded bool `json:"cost_refunded"`
|
||||||
|
RequestCharCount int64 `json:"request_char_count"`
|
||||||
|
ResponseCharCount int64 `json:"response_char_count"`
|
||||||
|
LeaseOwner string `json:"lease_owner"`
|
||||||
|
LeaseUntil *time.Time `json:"lease_until"`
|
||||||
|
CancelRequestedAt *time.Time `json:"cancel_requested_at"`
|
||||||
|
ErrorCode string `json:"error_code"`
|
||||||
|
ErrorMessage string `json:"error_message"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
StartedAt *time.Time `json:"started_at"`
|
||||||
|
FinishedAt *time.Time `json:"finished_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GenerationTask struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
RequestID string `json:"request_id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid" json:"user_id"`
|
||||||
|
ChannelID *uuid.UUID `gorm:"type:uuid" json:"channel_id"`
|
||||||
|
ModelID *uuid.UUID `gorm:"type:uuid" json:"model_id"`
|
||||||
|
ProjectID *uuid.UUID `gorm:"type:uuid" json:"project_id"`
|
||||||
|
EpisodeID *uuid.UUID `gorm:"type:uuid" json:"episode_id"`
|
||||||
|
StoryboardID *uuid.UUID `gorm:"type:uuid" json:"storyboard_id"`
|
||||||
|
ScriptAnalysisID *uuid.UUID `gorm:"type:uuid" json:"script_analysis_id"`
|
||||||
|
TaskType string `json:"task_type"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
UpstreamTaskID string `json:"upstream_task_id"`
|
||||||
|
InputData json.RawMessage `gorm:"type:jsonb" json:"input_data"`
|
||||||
|
BillingSnapshot json.RawMessage `gorm:"type:jsonb;not null;default:'{}'" json:"billing_snapshot"`
|
||||||
|
InputTokens *int64 `json:"input_tokens"`
|
||||||
|
OutputTokens *int64 `json:"output_tokens"`
|
||||||
|
TotalTokens *int64 `json:"total_tokens"`
|
||||||
|
TokenCountSource *string `json:"token_count_source"`
|
||||||
|
UsageRaw json.RawMessage `gorm:"type:jsonb" json:"usage_raw"`
|
||||||
|
ResultData json.RawMessage `gorm:"type:jsonb;not null;default:'{}'" json:"result_data"`
|
||||||
|
PromptSnapshot string `json:"prompt_snapshot"`
|
||||||
|
EstimatedPoints string `gorm:"column:estimated_points" json:"estimated_points"`
|
||||||
|
ActualPoints *string `gorm:"column:actual_points" json:"actual_points"`
|
||||||
|
PrepaidPoints string `gorm:"column:prepaid_points" json:"prepaid_points"`
|
||||||
|
CostRefunded bool `json:"cost_refunded"`
|
||||||
|
SlotReserved bool `json:"slot_reserved"`
|
||||||
|
SlotReservedAt *time.Time `json:"slot_reserved_at"`
|
||||||
|
SlotReleasedAt *time.Time `json:"slot_released_at"`
|
||||||
|
SubmitAttempts int `json:"submit_attempts"`
|
||||||
|
PollAttempts int `json:"poll_attempts"`
|
||||||
|
DownloadAttempts int `json:"download_attempts"`
|
||||||
|
NextSubmitAt *time.Time `json:"next_submit_at"`
|
||||||
|
NextPollAt *time.Time `json:"next_poll_at"`
|
||||||
|
LeaseOwner string `json:"lease_owner"`
|
||||||
|
LeaseUntil *time.Time `json:"lease_until"`
|
||||||
|
UpstreamResultURL string `json:"-"`
|
||||||
|
ErrorCode string `json:"error_code"`
|
||||||
|
ErrorMessage string `json:"error_message"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
SubmittedAt *time.Time `json:"submitted_at"`
|
||||||
|
FinishedAt *time.Time `json:"finished_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GenerationOutput struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
TaskID uuid.UUID `gorm:"type:uuid" json:"task_id"`
|
||||||
|
MediaAssetID uuid.UUID `gorm:"type:uuid" json:"media_asset_id"`
|
||||||
|
OutputType string `json:"output_type"`
|
||||||
|
SequenceNo int `json:"sequence_no"`
|
||||||
|
Metadata json.RawMessage `gorm:"type:jsonb" json:"metadata"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestProjectAssetBeforeCreateInitializesJSONFields(t *testing.T) {
|
||||||
|
asset := &ProjectAsset{}
|
||||||
|
if err := asset.BeforeCreate(nil); err != nil {
|
||||||
|
t.Fatalf("BeforeCreate returned error: %v", err)
|
||||||
|
}
|
||||||
|
if string(asset.Appearances) != "[]" {
|
||||||
|
t.Fatalf("unexpected appearances default: %s", asset.Appearances)
|
||||||
|
}
|
||||||
|
if string(asset.Aliases) != "[]" {
|
||||||
|
t.Fatalf("unexpected aliases default: %s", asset.Aliases)
|
||||||
|
}
|
||||||
|
if string(asset.Attributes) != "{}" {
|
||||||
|
t.Fatalf("unexpected attributes default: %s", asset.Attributes)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,36 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ScriptAnalysis struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid;index" json:"user_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
SourceContent string `json:"source_content"`
|
||||||
|
ResultContent string `json:"result_content"`
|
||||||
|
AnalysisResult json.RawMessage `gorm:"type:jsonb;not null;default:'{}'" json:"analysis_result"`
|
||||||
|
AnalysisStatus string `json:"analysis_status"`
|
||||||
|
AnalysisMessage string `json:"analysis_message"`
|
||||||
|
Characters []ScriptAnalysisCharacter `gorm:"foreignKey:ScriptAnalysisID" json:"characters,omitempty"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt *time.Time `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ScriptAnalysisCharacter struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
ScriptAnalysisID uuid.UUID `gorm:"type:uuid;index" json:"script_analysis_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Faction string `json:"faction"`
|
||||||
|
Biography string `json:"biography"`
|
||||||
|
Motivation string `json:"motivation"`
|
||||||
|
Relationship string `json:"relationship"`
|
||||||
|
SortOrder int `json:"sort_order"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
@@ -0,0 +1,39 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type WebUser struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey" json:"id"`
|
||||||
|
UID string `json:"uid"`
|
||||||
|
Username string `json:"username"`
|
||||||
|
Account string `json:"account"`
|
||||||
|
AvatarURL string `json:"avatar_url"`
|
||||||
|
AvatarKey string `json:"-"`
|
||||||
|
PasswordHash string `json:"-"`
|
||||||
|
PointBalance string `gorm:"column:point_balance" json:"point_balance"`
|
||||||
|
DailyLimit string `json:"daily_limit"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
SessionVersion int64 `json:"-"`
|
||||||
|
LastOnlineAt *time.Time `json:"last_online_at"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt gorm.DeletedAt `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type WebRefreshToken struct {
|
||||||
|
ID uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey"`
|
||||||
|
UserID uuid.UUID `gorm:"type:uuid"`
|
||||||
|
TokenHash string
|
||||||
|
SessionVersion int64
|
||||||
|
ExpiresAt time.Time
|
||||||
|
RevokedAt *time.Time
|
||||||
|
CreatedAt time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
func (WebUser) TableName() string { return "web_users" }
|
||||||
|
func (WebRefreshToken) TableName() string { return "web_refresh_tokens" }
|
||||||
@@ -0,0 +1,306 @@
|
|||||||
|
// 管理后台业务模块，封装模型配置、兑换码、用户积分和操作审计的持久化事务。
|
||||||
|
package admin
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/billing"
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
legacyservice "juhe-factory/api/internal/service"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Service 提供管理后台模块的公开业务接口，并隐藏底层数据库连接。
|
||||||
|
type Service struct {
|
||||||
|
db *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
// ModelPriceInput 表示模型的单项计费配置。
|
||||||
|
type ModelPriceInput struct {
|
||||||
|
PriceKey string `json:"price_key"`
|
||||||
|
Unit string `json:"unit"`
|
||||||
|
Price any `json:"price"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ModelInput 表示管理端新增或修改模型时提交的完整配置。
|
||||||
|
type ModelInput struct {
|
||||||
|
ChannelID string `json:"channel_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
ModelType string `json:"model_type"`
|
||||||
|
Multimodal *bool `json:"multimodal"`
|
||||||
|
TextBillingMode string `json:"text_billing_mode"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
Prices []ModelPriceInput `json:"prices"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewService 创建管理后台业务服务。
|
||||||
|
func NewService(db *gorm.DB) *Service {
|
||||||
|
return &Service{db: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListModels 按管理端筛选条件分页查询模型及价格配置。
|
||||||
|
func (s *Service) ListModels(keyword, modelType, channelID string, page, size int) (legacyservice.Page, error) {
|
||||||
|
query := s.db.Table("models m").Joins("JOIN channels c ON c.id=m.channel_id").Where("m.deleted_at IS NULL")
|
||||||
|
if keyword = strings.TrimSpace(keyword); keyword != "" {
|
||||||
|
query = query.Where("m.name::text ILIKE ?", "%"+keyword+"%")
|
||||||
|
}
|
||||||
|
if modelType = strings.TrimSpace(modelType); modelType != "" {
|
||||||
|
query = query.Where("m.model_type=?", modelType)
|
||||||
|
}
|
||||||
|
if channelID = strings.TrimSpace(channelID); channelID != "" {
|
||||||
|
query = query.Where("m.channel_id=?", channelID)
|
||||||
|
}
|
||||||
|
var total int64
|
||||||
|
if err := query.Count(&total).Error; err != nil {
|
||||||
|
return legacyservice.Page{}, err
|
||||||
|
}
|
||||||
|
items := make([]map[string]any, 0)
|
||||||
|
err := query.Select(`m.id,m.channel_id,c.name AS channel_name,m.name,m.model_type,m.multimodal,m.text_billing_mode,m.enabled,c.enabled AS channel_enabled,
|
||||||
|
coalesce((SELECT jsonb_agg(jsonb_build_object('price_key',p.price_key,'unit',p.unit,'price',p.price) ORDER BY CASE p.price_key WHEN 'default' THEN 0 WHEN '1K' THEN 1 WHEN '2K' THEN 2 WHEN '4K' THEN 3 WHEN '480p' THEN 4 WHEN '720p' THEN 5 WHEN '1080p' THEN 6 ELSE 99 END) FROM model_prices p WHERE p.model_id=m.id),'[]'::jsonb) AS prices`).
|
||||||
|
Order("m.created_at DESC").Offset((page - 1) * size).Limit(size).Find(&items).Error
|
||||||
|
if err != nil {
|
||||||
|
return legacyservice.Page{}, err
|
||||||
|
}
|
||||||
|
for i := range items {
|
||||||
|
items[i]["prices"] = decodePriceList(items[i]["prices"])
|
||||||
|
}
|
||||||
|
return legacyservice.Page{Items: items, Total: total, Page: page, PageSize: size}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveModel 校验模型计费配置，并在同一事务中保存模型及全部价格项。
|
||||||
|
func (s *Service) SaveModel(modelID string, input ModelInput) (string, bool, ModelInput, error) {
|
||||||
|
if err := normalizeModelPrices(&input); err != nil {
|
||||||
|
return "", false, input, err
|
||||||
|
}
|
||||||
|
created := modelID == ""
|
||||||
|
if created {
|
||||||
|
modelID = uuid.NewString()
|
||||||
|
}
|
||||||
|
err := s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
values := map[string]any{"channel_id": input.ChannelID, "name": input.Name, "model_type": input.ModelType, "enabled": input.Enabled, "text_billing_mode": nil}
|
||||||
|
switch input.ModelType {
|
||||||
|
case "text":
|
||||||
|
values["multimodal"] = input.Multimodal
|
||||||
|
values["text_billing_mode"] = input.TextBillingMode
|
||||||
|
default:
|
||||||
|
values["multimodal"] = nil
|
||||||
|
}
|
||||||
|
if created {
|
||||||
|
values["id"] = modelID
|
||||||
|
if err := tx.Table("models").Create(values).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else if err := tx.Table("models").Where("id=? AND deleted_at IS NULL", modelID).Updates(values).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := tx.Exec("DELETE FROM model_prices WHERE model_id=?", modelID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, price := range input.Prices {
|
||||||
|
if err := tx.Exec("INSERT INTO model_prices(id,model_id,price_key,unit,price) VALUES(?,?,?,?,?)", uuid.NewString(), modelID, price.PriceKey, price.Unit, price.Price).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
return modelID, created, input, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListRedemptions 按关键字和状态分页查询兑换码记录。
|
||||||
|
func (s *Service) ListRedemptions(keyword, status string, page, size int) (legacyservice.Page, error) {
|
||||||
|
query := s.db.Table("redemption_codes r").Joins("JOIN redemption_batches b ON b.id=r.batch_id").Joins("LEFT JOIN web_users u ON u.id=r.redeemed_by")
|
||||||
|
if keyword = strings.TrimSpace(keyword); keyword != "" {
|
||||||
|
query = query.Where("r.code ILIKE ? OR r.code_mask ILIKE ? OR b.name::text ILIKE ? OR u.uid=?", "%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%", keyword)
|
||||||
|
}
|
||||||
|
if status = strings.TrimSpace(status); status != "" {
|
||||||
|
query = query.Where("CASE WHEN r.status='unused' AND r.expires_at<CURRENT_TIMESTAMP THEN 'expired' ELSE r.status END=?", status)
|
||||||
|
}
|
||||||
|
var total int64
|
||||||
|
if err := query.Count(&total).Error; err != nil {
|
||||||
|
return legacyservice.Page{}, err
|
||||||
|
}
|
||||||
|
items := make([]map[string]any, 0)
|
||||||
|
err := query.Select("r.id,r.code,b.name AS batch_name,r.points,CASE WHEN r.status='unused' AND r.expires_at<CURRENT_TIMESTAMP THEN 'expired' ELSE r.status END AS status,u.uid AS redeemed_uid,r.redeemed_at,r.expires_at,a.username AS created_by,r.created_at").
|
||||||
|
Joins("JOIN admin_users a ON a.id=b.created_by").Order("r.created_at DESC").Offset((page - 1) * size).Limit(size).Find(&items).Error
|
||||||
|
if err != nil {
|
||||||
|
return legacyservice.Page{}, err
|
||||||
|
}
|
||||||
|
return legacyservice.Page{Items: items, Total: total, Page: page, PageSize: size}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListAuditLogs 按关键字分页查询管理端操作审计记录，仅提供只读访问。
|
||||||
|
func (s *Service) ListAuditLogs(keyword string, page, size int) (legacyservice.Page, error) {
|
||||||
|
query := s.db.Table("admin_audit_logs")
|
||||||
|
if keyword = strings.TrimSpace(keyword); keyword != "" {
|
||||||
|
pattern := "%" + keyword + "%"
|
||||||
|
query = query.Where(`admin_username ILIKE ? OR action::text ILIKE ? OR resource_type::text ILIKE ?
|
||||||
|
OR coalesce(resource_id,'') ILIKE ? OR coalesce(reason,'') ILIKE ? OR coalesce(trace_id,'') ILIKE ?`,
|
||||||
|
pattern, pattern, pattern, pattern, pattern, pattern)
|
||||||
|
if resourceTypes := matchingAuditResourceTypes(keyword); len(resourceTypes) > 0 {
|
||||||
|
query = query.Or("resource_type IN ?", resourceTypes)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var total int64
|
||||||
|
if err := query.Count(&total).Error; err != nil {
|
||||||
|
return legacyservice.Page{}, err
|
||||||
|
}
|
||||||
|
items := make([]map[string]any, 0)
|
||||||
|
err := query.Select(`id,admin_id,admin_username,action,resource_type,coalesce(resource_id,'') AS resource_id,
|
||||||
|
coalesce(reason,'') AS reason,coalesce(trace_id,'') AS trace_id,created_at`).
|
||||||
|
Order("created_at DESC,id DESC").Offset((page - 1) * size).Limit(size).Find(&items).Error
|
||||||
|
return legacyservice.Page{Items: items, Total: total, Page: page, PageSize: size}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// matchingAuditResourceTypes 将中文资源名称转换为数据库内部代码，支持管理端直接使用中文搜索。
|
||||||
|
func matchingAuditResourceTypes(keyword string) []string {
|
||||||
|
keyword = strings.TrimSpace(keyword)
|
||||||
|
if keyword == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
labels := map[string]string{
|
||||||
|
"users": "用户",
|
||||||
|
"redemption-codes": "兑换码",
|
||||||
|
"styles": "风格",
|
||||||
|
"prompts": "提示词",
|
||||||
|
"channels": "渠道",
|
||||||
|
"models": "模型",
|
||||||
|
"admin-auth": "管理员账号",
|
||||||
|
}
|
||||||
|
result := make([]string, 0, len(labels))
|
||||||
|
for resourceType, label := range labels {
|
||||||
|
if strings.Contains(label, keyword) {
|
||||||
|
result = append(result, resourceType)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateRedemptionBatch 校验生成限制，并在事务中创建兑换批次和明文返回值。
|
||||||
|
func (s *Service) CreateRedemptionBatch(adminID uuid.UUID, name string, pointsValue any, quantity int, expiresAt time.Time) (string, []string, string, error) {
|
||||||
|
points, err := ValidateRedemptionBatchLimits(pointsValue, quantity)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, "", err
|
||||||
|
}
|
||||||
|
batchID := uuid.NewString()
|
||||||
|
codes := make([]string, 0, quantity)
|
||||||
|
err = s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
if err := tx.Exec("INSERT INTO redemption_batches(id,name,points,quantity,expires_at,created_by) VALUES(?,?,?,?,?,?)", batchID, name, points, quantity, expiresAt, adminID).Error; err != nil {
|
||||||
|
// 命中 redemption_batches_name_key 唯一约束时返回业务友好提示，避免把底层 SQL 错误透传给前端
|
||||||
|
if strings.Contains(err.Error(), "23505") {
|
||||||
|
return errors.New("批次名称已存在，请更换后重试")
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for i := 0; i < quantity; i++ {
|
||||||
|
plain, hash, mask, err := legacyservice.GenerateRedemptionCode()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := tx.Exec("INSERT INTO redemption_codes(id,batch_id,code_hash,code_mask,code,points,expires_at) VALUES(?,?,?,?,?,?,?)", uuid.NewString(), batchID, hash, mask, plain, points, expiresAt).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
codes = append(codes, plain)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
return batchID, codes, points, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// GrantUserPoints 通过永久积分批次为指定用户增加积分，并使用请求号保证重复提交不会重复入账。
|
||||||
|
func (s *Service) GrantUserPoints(userID uuid.UUID, requestID string, pointsValue any) (string, bool, error) {
|
||||||
|
requestID = strings.TrimSpace(requestID)
|
||||||
|
if _, err := uuid.Parse(requestID); err != nil {
|
||||||
|
return "", false, errors.New("积分发放请求号无效")
|
||||||
|
}
|
||||||
|
points := strings.TrimSpace(fmt.Sprint(pointsValue))
|
||||||
|
sourceID := userID.String() + ":" + requestID
|
||||||
|
var balance string
|
||||||
|
var credited bool
|
||||||
|
err := s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
var err error
|
||||||
|
balance, credited, err = billing.CreditPoints(tx, userID, "admin_grant", sourceID, points, "后台发放积分")
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
return balance, credited, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// WriteAudit 记录管理端操作审计信息，调用方无需接触数据库连接。
|
||||||
|
func (s *Service) WriteAudit(adminUser *model.AdminUser, action, resource, resourceID, reason, ip, traceID string, detail any) error {
|
||||||
|
return legacyservice.WriteAudit(s.db, adminUser, action, resource, resourceID, reason, ip, traceID, detail)
|
||||||
|
}
|
||||||
|
|
||||||
|
// normalizeModelPrices 校验并标准化文本模型的计费键、单位和金额。
|
||||||
|
func normalizeModelPrices(input *ModelInput) error {
|
||||||
|
if input.ModelType != "text" {
|
||||||
|
input.TextBillingMode = ""
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
expected := map[string]string{}
|
||||||
|
switch input.TextBillingMode {
|
||||||
|
case "per_request":
|
||||||
|
expected["default"] = "次"
|
||||||
|
case "per_token":
|
||||||
|
expected["input"] = "M Token"
|
||||||
|
expected["output"] = "M Token"
|
||||||
|
default:
|
||||||
|
return errors.New("文本模型计费模式无效")
|
||||||
|
}
|
||||||
|
if len(input.Prices) != len(expected) {
|
||||||
|
return errors.New("文本模型价格配置不完整")
|
||||||
|
}
|
||||||
|
seen := map[string]bool{}
|
||||||
|
for index := range input.Prices {
|
||||||
|
key := strings.ToLower(strings.TrimSpace(input.Prices[index].PriceKey))
|
||||||
|
unit, ok := expected[key]
|
||||||
|
value, err := strconv.ParseFloat(fmt.Sprint(input.Prices[index].Price), 64)
|
||||||
|
if !ok || seen[key] || err != nil || value < 0 || math.IsNaN(value) || math.IsInf(value, 0) {
|
||||||
|
return errors.New("文本模型价格配置无效")
|
||||||
|
}
|
||||||
|
seen[key] = true
|
||||||
|
input.Prices[index].PriceKey = key
|
||||||
|
input.Prices[index].Unit = unit
|
||||||
|
input.Prices[index].Price = math.Round(value*100) / 100
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// decodePriceList 将 PostgreSQL JSON 聚合结果转换为稳定的价格数组。
|
||||||
|
func decodePriceList(value any) []map[string]any {
|
||||||
|
var raw []byte
|
||||||
|
switch data := value.(type) {
|
||||||
|
case []byte:
|
||||||
|
raw = data
|
||||||
|
case json.RawMessage:
|
||||||
|
raw = data
|
||||||
|
case string:
|
||||||
|
raw = []byte(data)
|
||||||
|
default:
|
||||||
|
return []map[string]any{}
|
||||||
|
}
|
||||||
|
prices := make([]map[string]any, 0)
|
||||||
|
if err := json.Unmarshal(raw, &prices); err != nil {
|
||||||
|
return []map[string]any{}
|
||||||
|
}
|
||||||
|
return prices
|
||||||
|
}
|
||||||
|
|
||||||
|
// ValidateRedemptionBatchLimits 校验单码积分和单批生成数量的安全范围。
|
||||||
|
func ValidateRedemptionBatchLimits(value any, quantity int) (string, error) {
|
||||||
|
points := strings.TrimSpace(fmt.Sprint(value))
|
||||||
|
parsed, err := strconv.ParseFloat(points, 64)
|
||||||
|
if err != nil || math.IsNaN(parsed) || math.IsInf(parsed, 0) || parsed < 1 || parsed > 100 {
|
||||||
|
return "", errors.New("单码积分数量必须为 1 至 100")
|
||||||
|
}
|
||||||
|
if quantity < 1 || quantity > 10 {
|
||||||
|
return "", errors.New("生成数量必须为 1 至 10")
|
||||||
|
}
|
||||||
|
return points, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,67 @@
|
|||||||
|
// 管理后台模块单元测试，验证模型计费、兑换码批次和审计搜索的纯业务规则。
|
||||||
|
package admin
|
||||||
|
|
||||||
|
import (
|
||||||
|
"slices"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestNormalizeModelPrices 验证按 Token 计费时价格键、单位和金额标准化。
|
||||||
|
func TestNormalizeModelPrices(t *testing.T) {
|
||||||
|
input := ModelInput{
|
||||||
|
ModelType: "text",
|
||||||
|
TextBillingMode: "per_token",
|
||||||
|
Prices: []ModelPriceInput{
|
||||||
|
{PriceKey: " INPUT ", Price: "1.236"},
|
||||||
|
{PriceKey: "output", Price: 2},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
if err := normalizeModelPrices(&input); err != nil {
|
||||||
|
t.Fatalf("normalize model prices: %v", err)
|
||||||
|
}
|
||||||
|
if input.Prices[0].PriceKey != "input" || input.Prices[0].Unit != "M Token" || input.Prices[0].Price != 1.24 {
|
||||||
|
t.Fatalf("unexpected normalized input price: %#v", input.Prices[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestNormalizeModelPricesRejectsIncompleteConfig 验证文本模型缺少价格项时拒绝保存。
|
||||||
|
func TestNormalizeModelPricesRejectsIncompleteConfig(t *testing.T) {
|
||||||
|
input := ModelInput{ModelType: "text", TextBillingMode: "per_token", Prices: []ModelPriceInput{{PriceKey: "input", Price: 1}}}
|
||||||
|
if err := normalizeModelPrices(&input); err == nil {
|
||||||
|
t.Fatal("expected incomplete token prices to be rejected")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestValidateRedemptionBatchLimits 验证兑换码单码积分与批次数量边界。
|
||||||
|
func TestValidateRedemptionBatchLimits(t *testing.T) {
|
||||||
|
if points, err := ValidateRedemptionBatchLimits("10.5", 10); err != nil || points != "10.5" {
|
||||||
|
t.Fatalf("expected valid redemption limits, points=%q err=%v", points, err)
|
||||||
|
}
|
||||||
|
for _, test := range []struct {
|
||||||
|
points any
|
||||||
|
quantity int
|
||||||
|
}{{0, 1}, {101, 1}, {10, 0}, {10, 11}} {
|
||||||
|
if _, err := ValidateRedemptionBatchLimits(test.points, test.quantity); err == nil {
|
||||||
|
t.Fatalf("expected limits to be rejected: %#v", test)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestMatchingAuditResourceTypes 验证中文资源名称可转换为审计记录使用的内部代码。
|
||||||
|
func TestMatchingAuditResourceTypes(t *testing.T) {
|
||||||
|
for _, test := range []struct {
|
||||||
|
keyword string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{keyword: "渠道", want: "channels"},
|
||||||
|
{keyword: "兑换", want: "redemption-codes"},
|
||||||
|
{keyword: "管理员账号", want: "admin-auth"},
|
||||||
|
} {
|
||||||
|
if got := matchingAuditResourceTypes(test.keyword); !slices.Contains(got, test.want) {
|
||||||
|
t.Fatalf("keyword %q matched %v, want %q", test.keyword, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if got := matchingAuditResourceTypes("不存在的资源"); len(got) != 0 {
|
||||||
|
t.Fatalf("unexpected resource matches: %v", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,323 @@
|
|||||||
|
// 图片生成生图业务模块，负责提交用户级图片任务、查询历史并清理生成资源。
|
||||||
|
package productimage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
"unicode/utf8"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/billing"
|
||||||
|
"juhe-factory/api/internal/model"
|
||||||
|
queuepkg "juhe-factory/api/internal/queue"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"github.com/hibiken/asynq"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/clause"
|
||||||
|
)
|
||||||
|
|
||||||
|
var activeTaskStatuses = []string{"pending_submission", "submitting", "submitted", "processing", "result_ready", "downloading", "cancel_requested"}
|
||||||
|
|
||||||
|
// ReferenceUpload 表示已经上传到对象存储、等待绑定到生图任务的参考图。
|
||||||
|
type ReferenceUpload struct {
|
||||||
|
Asset *model.MediaAsset
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenerateInput 是图片生成生图接口的配置参数和临时参考图。
|
||||||
|
type GenerateInput struct {
|
||||||
|
Prompt string
|
||||||
|
ModelID uuid.UUID
|
||||||
|
AspectRatio string
|
||||||
|
Resolution string
|
||||||
|
References []ReferenceUpload
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenerationView 汇总任务状态和已经落库的图片结果。
|
||||||
|
type GenerationView struct {
|
||||||
|
TaskID uuid.UUID `json:"task_id"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Prompt string `json:"prompt"`
|
||||||
|
ModelName string `json:"model_name"`
|
||||||
|
ErrorMessage string `json:"error_message"`
|
||||||
|
MediaAssetID *uuid.UUID `json:"media_asset_id"`
|
||||||
|
PublicURL string `json:"public_url"`
|
||||||
|
MimeType string `json:"mime_type"`
|
||||||
|
EstimatedPoints string `json:"estimated_points"`
|
||||||
|
ActualPoints *string `json:"actual_points"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
FinishedAt *time.Time `json:"finished_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Service 组合图片生成模块需要的数据库和任务队列依赖。
|
||||||
|
type Service struct {
|
||||||
|
DB *gorm.DB
|
||||||
|
Queue *asynq.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewService 创建图片生成生图服务。
|
||||||
|
func NewService(db *gorm.DB, queue *asynq.Client) *Service { return &Service{DB: db, Queue: queue} }
|
||||||
|
|
||||||
|
// QueueGeneration 校验配置、保存临时参考图并提交一个独立图片任务。
|
||||||
|
func (s *Service) QueueGeneration(userID uuid.UUID, input GenerateInput) (*model.GenerationTask, error) {
|
||||||
|
if s.Queue == nil {
|
||||||
|
return nil, errors.New("生成任务队列不可用")
|
||||||
|
}
|
||||||
|
prompt := strings.TrimSpace(input.Prompt)
|
||||||
|
if prompt == "" {
|
||||||
|
return nil, errors.New("提示词不能为空")
|
||||||
|
}
|
||||||
|
if utf8.RuneCountInString(prompt) > 5000 {
|
||||||
|
return nil, errors.New("提示词不能超过 5000 个字符")
|
||||||
|
}
|
||||||
|
if len(input.References) > 4 {
|
||||||
|
return nil, errors.New("最多上传 4 张参考图")
|
||||||
|
}
|
||||||
|
if input.ModelID == uuid.Nil {
|
||||||
|
return nil, errors.New("请选择图片模型")
|
||||||
|
}
|
||||||
|
if !validRatio(input.AspectRatio) || !validResolution(input.Resolution) {
|
||||||
|
return nil, errors.New("图片生成参数无效")
|
||||||
|
}
|
||||||
|
var selected struct {
|
||||||
|
ModelID uuid.UUID
|
||||||
|
ChannelID uuid.UUID
|
||||||
|
ModelName string
|
||||||
|
Price string
|
||||||
|
PriceExists bool
|
||||||
|
}
|
||||||
|
if err := s.DB.Raw(`SELECT model.id AS model_id,model.channel_id,model.name AS model_name,
|
||||||
|
coalesce(price.price,0)::text AS price,(price.id IS NOT NULL) AS price_exists
|
||||||
|
FROM models model JOIN channels channel ON channel.id=model.channel_id
|
||||||
|
LEFT JOIN model_prices price ON price.model_id=model.id AND lower(price.price_key)=lower(?)
|
||||||
|
WHERE model.id=? AND model.model_type='image' AND model.enabled=true AND model.deleted_at IS NULL
|
||||||
|
AND channel.enabled=true AND channel.deleted_at IS NULL`, input.Resolution, input.ModelID).Scan(&selected).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if selected.ModelID == uuid.Nil {
|
||||||
|
return nil, errors.New("图片模型不可用")
|
||||||
|
}
|
||||||
|
if !selected.PriceExists {
|
||||||
|
return nil, fmt.Errorf("当前图片模型未配置 %s 价格", input.Resolution)
|
||||||
|
}
|
||||||
|
imageURLs := make([]string, 0, len(input.References))
|
||||||
|
referenceIDs := make([]string, 0, len(input.References))
|
||||||
|
referenceNames := make([]string, 0, len(input.References))
|
||||||
|
seenNames := make(map[string]struct{}, len(input.References))
|
||||||
|
for _, reference := range input.References {
|
||||||
|
if reference.Asset == nil || strings.TrimSpace(reference.Asset.PublicURL) == "" {
|
||||||
|
return nil, errors.New("参考图信息无效")
|
||||||
|
}
|
||||||
|
name := strings.TrimSpace(reference.Name)
|
||||||
|
if name == "" {
|
||||||
|
return nil, errors.New("参考图名称不能为空")
|
||||||
|
}
|
||||||
|
if _, exists := seenNames[name]; exists {
|
||||||
|
return nil, errors.New("参考图名称不能重复")
|
||||||
|
}
|
||||||
|
seenNames[name] = struct{}{}
|
||||||
|
imageURLs = append(imageURLs, reference.Asset.PublicURL)
|
||||||
|
referenceIDs = append(referenceIDs, reference.Asset.ID.String())
|
||||||
|
referenceNames = append(referenceNames, name)
|
||||||
|
}
|
||||||
|
providerPrompt := buildProviderPrompt(prompt, referenceNames)
|
||||||
|
payload, err := json.Marshal(map[string]any{
|
||||||
|
"product_image": true, "prompt": providerPrompt, "display_prompt": prompt,
|
||||||
|
"model": selected.ModelName, "size": input.AspectRatio, "resolution": input.Resolution,
|
||||||
|
"image_urls": imageURLs, "reference_media_asset_ids": referenceIDs, "reference_names": referenceNames, "n": 1,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
task := &model.GenerationTask{ID: uuid.New(), RequestID: "product_image_" + uuid.NewString(), UserID: userID, ChannelID: &selected.ChannelID, ModelID: &selected.ModelID, TaskType: "image_generation", Status: "pending_submission", InputData: payload}
|
||||||
|
err = s.DB.Transaction(func(tx *gorm.DB) error {
|
||||||
|
for _, reference := range input.References {
|
||||||
|
owner := userID
|
||||||
|
reference.Asset.OwnerUserID = &owner
|
||||||
|
if err := tx.Create(reference.Asset).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return billing.PrechargeGenerationTask(tx, task, selected.Price, 1, "图片生成")
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := queuepkg.EnqueueID(s.Queue, queuepkg.TypeDispatchChannel, selected.ChannelID, 0); err != nil {
|
||||||
|
if refundErr := s.failQueuedTask(task.ID, err.Error()); refundErr != nil {
|
||||||
|
return nil, fmt.Errorf("生成任务入队失败且预扣返还失败: %w", refundErr)
|
||||||
|
}
|
||||||
|
return nil, errors.New("生成任务队列暂时不可用，请稍后重试")
|
||||||
|
}
|
||||||
|
return task, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generations 返回当前用户最近五十条图片生成记录。
|
||||||
|
func (s *Service) Generations(userID uuid.UUID) ([]GenerationView, error) {
|
||||||
|
items := make([]GenerationView, 0)
|
||||||
|
err := s.DB.Raw(`SELECT task.id AS task_id,task.status,
|
||||||
|
coalesce(task.input_data->>'display_prompt',task.input_data->>'prompt','') AS prompt,
|
||||||
|
coalesce(model.name,task.input_data->>'model','') AS model_name,task.error_message,
|
||||||
|
output.media_asset_id,coalesce(media.public_url,'') AS public_url,coalesce(media.mime_type,'') AS mime_type,
|
||||||
|
task.estimated_points,task.actual_points,task.created_at,task.finished_at
|
||||||
|
FROM generation_tasks task
|
||||||
|
LEFT JOIN models model ON model.id=task.model_id
|
||||||
|
LEFT JOIN generation_outputs output ON output.task_id=task.id AND output.sequence_no=1
|
||||||
|
LEFT JOIN media_assets media ON media.id=output.media_asset_id AND media.deleted_at IS NULL
|
||||||
|
WHERE task.user_id=? AND task.task_type='image_generation' AND task.input_data->>'product_image'='true'
|
||||||
|
ORDER BY task.created_at DESC LIMIT 50`, userID).Scan(&items).Error
|
||||||
|
return items, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteGeneration 清理当前用户的图片任务、生成媒体和参考媒体；活动任务也允许主动删除。
|
||||||
|
func (s *Service) DeleteGeneration(userID, taskID uuid.UUID, deleteObjects func([]string) error) error {
|
||||||
|
objectKeys := make([]string, 0)
|
||||||
|
var channelID *uuid.UUID
|
||||||
|
err := s.DB.Transaction(func(tx *gorm.DB) error {
|
||||||
|
var task model.GenerationTask
|
||||||
|
if err := tx.Where("id=? AND user_id=? AND task_type='image_generation' AND input_data->>'product_image'='true'", taskID, userID).Take(&task).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !canDeleteGeneration(task.Status, task.ErrorMessage) {
|
||||||
|
return errors.New("图片仍在正常生成中，暂时不能删除")
|
||||||
|
}
|
||||||
|
channelID = task.ChannelID
|
||||||
|
var input struct {
|
||||||
|
ReferenceMediaAssetIDs []uuid.UUID `json:"reference_media_asset_ids"`
|
||||||
|
}
|
||||||
|
_ = json.Unmarshal(task.InputData, &input)
|
||||||
|
mediaIDs := append([]uuid.UUID(nil), input.ReferenceMediaAssetIDs...)
|
||||||
|
var outputs []struct {
|
||||||
|
MediaAssetID uuid.UUID
|
||||||
|
ObjectKey string
|
||||||
|
}
|
||||||
|
if err := tx.Table("generation_outputs output").Select("output.media_asset_id,media.object_key").Joins("JOIN media_assets media ON media.id=output.media_asset_id").Where("output.task_id=?", taskID).Scan(&outputs).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, output := range outputs {
|
||||||
|
mediaIDs = append(mediaIDs, output.MediaAssetID)
|
||||||
|
if output.ObjectKey != "" {
|
||||||
|
objectKeys = append(objectKeys, output.ObjectKey)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(mediaIDs) > 0 {
|
||||||
|
var refs []struct{ ObjectKey string }
|
||||||
|
if err := tx.Table("media_assets").Select("object_key").Where("id IN ?", mediaIDs).Scan(&refs).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, ref := range refs {
|
||||||
|
if ref.ObjectKey != "" {
|
||||||
|
objectKeys = append(objectKeys, ref.ObjectKey)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if deleteObjects != nil {
|
||||||
|
if err := deleteObjects(uniqueStrings(objectKeys)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := tx.Exec("DELETE FROM generation_outputs WHERE task_id=?", taskID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if len(mediaIDs) > 0 {
|
||||||
|
if err := tx.Exec("DELETE FROM channel_asset_cache WHERE media_asset_id IN ?", mediaIDs).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := tx.Exec("DELETE FROM media_assets WHERE id IN ?", mediaIDs).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return tx.Exec("DELETE FROM generation_tasks WHERE id=?", taskID).Error
|
||||||
|
})
|
||||||
|
if err == nil && channelID != nil && s.Queue != nil {
|
||||||
|
_ = queuepkg.EnqueueID(s.Queue, queuepkg.TypeDispatchChannel, *channelID, 0)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// canDeleteGeneration 判断图片任务是否已结束，或因错误停留在活动状态而允许用户清理。
|
||||||
|
func canDeleteGeneration(status, errorMessage string) bool {
|
||||||
|
for _, activeStatus := range activeTaskStatuses {
|
||||||
|
if status == activeStatus {
|
||||||
|
return strings.TrimSpace(errorMessage) != ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// failQueuedTask 在调度队列不可用时结束任务、返还积分并删除参考媒体记录。
|
||||||
|
func (s *Service) failQueuedTask(taskID uuid.UUID, message string) error {
|
||||||
|
return s.DB.Transaction(func(tx *gorm.DB) error {
|
||||||
|
var task model.GenerationTask
|
||||||
|
if err := tx.Clauses(clause.Locking{Strength: "UPDATE"}).Where("id=?", taskID).Take(&task).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
refunded, err := billing.RefundGenerationTask(tx, &task, "图片生成入队失败返还")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var input struct {
|
||||||
|
ReferenceMediaAssetIDs []uuid.UUID `json:"reference_media_asset_ids"`
|
||||||
|
}
|
||||||
|
_ = json.Unmarshal(task.InputData, &input)
|
||||||
|
if len(input.ReferenceMediaAssetIDs) > 0 {
|
||||||
|
if err := tx.Exec("DELETE FROM media_assets WHERE id IN ?", input.ReferenceMediaAssetIDs).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
updates := map[string]any{"status": "failed", "error_code": "queue_unavailable", "error_message": message, "finished_at": time.Now(), "actual_points": "0.00"}
|
||||||
|
if refunded {
|
||||||
|
updates["cost_refunded"] = true
|
||||||
|
}
|
||||||
|
return tx.Model(&model.GenerationTask{}).Where("id=?", taskID).Updates(updates).Error
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// validRatio 判断画幅比例是否属于图片生成工具公开支持的取值。
|
||||||
|
func validRatio(value string) bool {
|
||||||
|
for _, allowed := range []string{"9:21", "1:1", "3:4", "4:3", "9:16", "16:9"} {
|
||||||
|
if value == allowed {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// validResolution 判断分辨率是否属于图片生成工具公开支持的档位。
|
||||||
|
func validResolution(value string) bool {
|
||||||
|
for _, allowed := range []string{"1k", "2k", "4k"} {
|
||||||
|
if value == allowed {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildProviderPrompt 将文件名提及与参考图数组顺序建立明确对应关系。
|
||||||
|
func buildProviderPrompt(prompt string, names []string) string {
|
||||||
|
if len(names) == 0 {
|
||||||
|
return prompt
|
||||||
|
}
|
||||||
|
mappings := make([]string, 0, len(names))
|
||||||
|
for index, name := range names {
|
||||||
|
mappings = append(mappings, fmt.Sprintf("第%d张参考图名称为“%s”", index+1, name))
|
||||||
|
}
|
||||||
|
return "参考图对应关系：" + strings.Join(mappings, "；") + "。\n" + prompt
|
||||||
|
}
|
||||||
|
|
||||||
|
// uniqueStrings 按首次出现顺序去除空对象键和重复对象键。
|
||||||
|
func uniqueStrings(values []string) []string {
|
||||||
|
seen := make(map[string]struct{}, len(values))
|
||||||
|
result := make([]string, 0, len(values))
|
||||||
|
for _, value := range values {
|
||||||
|
if _, ok := seen[value]; ok || value == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
seen[value] = struct{}{}
|
||||||
|
result = append(result, value)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
@@ -0,0 +1,56 @@
|
|||||||
|
// 图片生成服务测试，验证参考图文件名映射和对象键去重规则。
|
||||||
|
package productimage
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
// TestBuildProviderPrompt 验证文件名提及会与参考图数组顺序建立稳定对应关系。
|
||||||
|
func TestBuildProviderPrompt(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
prompt := "让@正面图.png中的商品使用@包装图.webp的包装"
|
||||||
|
got := buildProviderPrompt(prompt, []string{"正面图.png", "包装图.webp"})
|
||||||
|
want := "参考图对应关系：第1张参考图名称为“正面图.png”；第2张参考图名称为“包装图.webp”。\n" + prompt
|
||||||
|
if got != want {
|
||||||
|
t.Fatalf("buildProviderPrompt() = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
if withoutReferences := buildProviderPrompt(prompt, nil); withoutReferences != prompt {
|
||||||
|
t.Fatalf("buildProviderPrompt() without references = %q, want %q", withoutReferences, prompt)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestUniqueStrings 验证重复或空对象键不会触发重复删除请求。
|
||||||
|
func TestUniqueStrings(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
got := uniqueStrings([]string{"a", "", "b", "a", "b", "c"})
|
||||||
|
want := []string{"a", "b", "c"}
|
||||||
|
if len(got) != len(want) {
|
||||||
|
t.Fatalf("uniqueStrings() length = %d, want %d", len(got), len(want))
|
||||||
|
}
|
||||||
|
for index := range want {
|
||||||
|
if got[index] != want[index] {
|
||||||
|
t.Fatalf("uniqueStrings()[%d] = %q, want %q", index, got[index], want[index])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestCanDeleteGeneration 验证失败停滞的活动任务可删除，而正常生成中的任务仍受保护。
|
||||||
|
func TestCanDeleteGeneration(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
status string
|
||||||
|
errorMessage string
|
||||||
|
want bool
|
||||||
|
}{
|
||||||
|
{name: "正常生成", status: "processing", want: false},
|
||||||
|
{name: "连接超时", status: "pending_submission", errorMessage: "dial tcp: i/o timeout", want: true},
|
||||||
|
{name: "生成失败", status: "failed", want: true},
|
||||||
|
{name: "生成成功", status: "succeeded", want: true},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
if got := canDeleteGeneration(test.status, test.errorMessage); got != test.want {
|
||||||
|
t.Fatalf("canDeleteGeneration(%q, %q) = %t, want %t", test.status, test.errorMessage, got, test.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,328 @@
|
|||||||
|
// 提示词业务模块，统一封装用户提示词规则、选择关系和持久化操作。
|
||||||
|
package prompt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Service 提供提示词模块的公开业务接口，并隐藏底层数据库实现。
|
||||||
|
type Service struct {
|
||||||
|
db *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
// CustomPromptInput 表示用户自定义提示词的可编辑内容。
|
||||||
|
type CustomPromptInput struct {
|
||||||
|
Name string
|
||||||
|
Type string
|
||||||
|
Content string
|
||||||
|
}
|
||||||
|
|
||||||
|
// CustomPrompt 表示返回给接口层的用户自定义提示词。
|
||||||
|
type CustomPrompt struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Content string `json:"content"`
|
||||||
|
Scope string `json:"scope"`
|
||||||
|
Editable bool `json:"editable"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemPromptInput 表示管理端可维护的系统提示词内容。
|
||||||
|
type SystemPromptInput struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Content string `json:"content"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// SystemPrompt 表示管理端保存后返回的系统提示词。
|
||||||
|
type SystemPrompt struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Content string `json:"content"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// VersionedPromptInput 表示旧版提示词工作流中的草稿和版本字段。
|
||||||
|
type VersionedPromptInput struct {
|
||||||
|
Code string
|
||||||
|
Name string
|
||||||
|
Category string
|
||||||
|
Type string
|
||||||
|
Content string
|
||||||
|
Variables []string
|
||||||
|
VersionNote string
|
||||||
|
}
|
||||||
|
|
||||||
|
// UserPromptList 表示指定类型的可用提示词及用户当前选择。
|
||||||
|
type UserPromptList struct {
|
||||||
|
Prompts []map[string]any
|
||||||
|
SelectedID *uuid.UUID
|
||||||
|
}
|
||||||
|
|
||||||
|
// PersistenceError 标识应由接口层转换为内部错误的数据库故障。
|
||||||
|
type PersistenceError struct {
|
||||||
|
Err error
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error 返回底层数据库错误文本，供日志与测试定位。
|
||||||
|
func (e PersistenceError) Error() string {
|
||||||
|
return e.Err.Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unwrap 暴露底层错误，保留 errors.Is 和 errors.As 语义。
|
||||||
|
func (e PersistenceError) Unwrap() error {
|
||||||
|
return e.Err
|
||||||
|
}
|
||||||
|
|
||||||
|
var fixedDramaParsePromptTypes = []string{"剧本解析", "角色、场景、道具解析"}
|
||||||
|
|
||||||
|
// NewService 创建提示词服务，数据库连接仅在模块内部使用。
|
||||||
|
func NewService(db *gorm.DB) *Service {
|
||||||
|
return &Service{db: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsFixedDramaParseType 判断提示词类型是否由后端固定配置且禁止用户编辑。
|
||||||
|
func IsFixedDramaParseType(promptType string) bool {
|
||||||
|
for _, fixedType := range fixedDramaParsePromptTypes {
|
||||||
|
if strings.TrimSpace(promptType) == fixedType {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsPersistenceError 判断错误是否属于需要隐藏细节的数据库故障。
|
||||||
|
func IsPersistenceError(err error) bool {
|
||||||
|
var target PersistenceError
|
||||||
|
return errors.As(err, &target)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListSystemPrompts 查询管理端可维护的系统提示词，不返回后端固定配置。
|
||||||
|
func (s *Service) ListSystemPrompts(keyword, promptType string, page, size int) ([]map[string]any, int64, error) {
|
||||||
|
query := s.db.Table("prompts p").Where("p.deleted_at IS NULL AND p.scope='system' AND p.type NOT IN ?", fixedDramaParsePromptTypes)
|
||||||
|
if keyword = strings.TrimSpace(keyword); keyword != "" {
|
||||||
|
query = query.Where("p.name::text ILIKE ? OR p.content ILIKE ?", "%"+keyword+"%", "%"+keyword+"%")
|
||||||
|
}
|
||||||
|
if promptType = strings.TrimSpace(promptType); promptType != "" {
|
||||||
|
query = query.Where("p.type=?", promptType)
|
||||||
|
}
|
||||||
|
var total int64
|
||||||
|
if err := query.Count(&total).Error; err != nil {
|
||||||
|
return nil, 0, err
|
||||||
|
}
|
||||||
|
items := make([]map[string]any, 0)
|
||||||
|
err := query.Select("p.id,p.name,p.type,p.content,p.scope,p.owner_user_id,p.created_at,p.updated_at").
|
||||||
|
Order("p.updated_at DESC").Offset((page - 1) * size).Limit(size).Find(&items).Error
|
||||||
|
return items, total, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveSystemPrompt 校验并新增或更新管理端系统提示词。
|
||||||
|
func (s *Service) SaveSystemPrompt(promptID string, input SystemPromptInput) (SystemPrompt, error) {
|
||||||
|
input.Name = strings.TrimSpace(input.Name)
|
||||||
|
input.Type = strings.TrimSpace(input.Type)
|
||||||
|
input.Content = strings.TrimSpace(input.Content)
|
||||||
|
if input.Name == "" || input.Type == "" || input.Content == "" {
|
||||||
|
return SystemPrompt{}, errors.New("name、type、content不能为空")
|
||||||
|
}
|
||||||
|
if IsFixedDramaParseType(input.Type) {
|
||||||
|
return SystemPrompt{}, errors.New("该提示词已固定在后端代码中，不支持管理")
|
||||||
|
}
|
||||||
|
if promptID == "" {
|
||||||
|
promptID = uuid.NewString()
|
||||||
|
if err := s.db.Exec("INSERT INTO prompts(id,name,type,content,scope,owner_user_id) VALUES(?,?,?,?,'system',NULL)", promptID, input.Name, input.Type, input.Content).Error; err != nil {
|
||||||
|
return SystemPrompt{}, err
|
||||||
|
}
|
||||||
|
} else if err := s.db.Exec("UPDATE prompts SET name=?,type=?,content=?,updated_at=CURRENT_TIMESTAMP WHERE id=? AND scope='system' AND deleted_at IS NULL", input.Name, input.Type, input.Content, promptID).Error; err != nil {
|
||||||
|
return SystemPrompt{}, err
|
||||||
|
}
|
||||||
|
return SystemPrompt{ID: promptID, Name: input.Name, Type: input.Type, Content: input.Content}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteSystemPrompt 删除管理端指定的系统提示词。
|
||||||
|
func (s *Service) DeleteSystemPrompt(promptID string) error {
|
||||||
|
return s.db.Exec("DELETE FROM prompts WHERE id=? AND scope='system'", promptID).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveVersionedPrompt 保存旧版管理接口使用的提示词草稿及版本快照。
|
||||||
|
func (s *Service) SaveVersionedPrompt(adminID uuid.UUID, promptID string, input VersionedPromptInput) (string, error) {
|
||||||
|
if strings.TrimSpace(input.Category) == "" {
|
||||||
|
input.Category = input.Type
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(input.Type) == "" {
|
||||||
|
input.Type = input.Category
|
||||||
|
}
|
||||||
|
err := s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
version := 1
|
||||||
|
if promptID == "" {
|
||||||
|
promptID = uuid.NewString()
|
||||||
|
if err := tx.Exec("INSERT INTO prompts(id,code,name,category,type,content,status) VALUES(?,?,?,?,?,?,'draft')", promptID, input.Code, input.Name, input.Category, input.Type, input.Content).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if err := tx.Raw("SELECT coalesce(max(version),0)+1 FROM prompt_versions WHERE prompt_id=?", promptID).Scan(&version).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := tx.Exec("UPDATE prompts SET code=?,name=?,category=?,type=?,content=? WHERE id=? AND deleted_at IS NULL", input.Code, input.Name, input.Category, input.Type, input.Content, promptID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
variables, _ := json.Marshal(input.Variables)
|
||||||
|
versionID := uuid.NewString()
|
||||||
|
if err := tx.Exec("INSERT INTO prompt_versions(id,prompt_id,version,content,variables,version_note,status,operator_id) VALUES(?,?,?,?,?::jsonb,?,'draft',?)", versionID, promptID, version, input.Content, string(variables), input.VersionNote, adminID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return tx.Exec("UPDATE prompts SET current_version_id=?,status='draft' WHERE id=?", versionID, promptID).Error
|
||||||
|
})
|
||||||
|
return promptID, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ApplyPromptAction 执行旧版提示词接口的发布、停用或版本回滚事务。
|
||||||
|
func (s *Service) ApplyPromptAction(adminID uuid.UUID, promptID, action, versionID, versionNote string) error {
|
||||||
|
return s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
switch action {
|
||||||
|
case "publish":
|
||||||
|
if err := tx.Exec("UPDATE prompt_versions SET status=CASE WHEN id=(SELECT current_version_id FROM prompts WHERE id=?) THEN 'published' ELSE 'retired' END WHERE prompt_id=?", promptID, promptID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return tx.Exec("UPDATE prompts SET status='published' WHERE id=?", promptID).Error
|
||||||
|
case "disable":
|
||||||
|
return tx.Exec("UPDATE prompts SET status='disabled' WHERE id=?", promptID).Error
|
||||||
|
case "rollback":
|
||||||
|
var source struct {
|
||||||
|
Content string
|
||||||
|
Variables string
|
||||||
|
}
|
||||||
|
if err := tx.Raw("SELECT content,variables::text FROM prompt_versions WHERE id=? AND prompt_id=?", versionID, promptID).Scan(&source).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var version int
|
||||||
|
if err := tx.Raw("SELECT coalesce(max(version),0)+1 FROM prompt_versions WHERE prompt_id=?", promptID).Scan(&version).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
newVersionID := uuid.NewString()
|
||||||
|
if err := tx.Exec("INSERT INTO prompt_versions(id,prompt_id,version,content,variables,version_note,status,operator_id) VALUES(?,?,?,?,?::jsonb,?,'draft',?)", newVersionID, promptID, version, source.Content, source.Variables, versionNote, adminID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return tx.Exec("UPDATE prompts SET current_version_id=?,status='draft' WHERE id=?", newVersionID, promptID).Error
|
||||||
|
default:
|
||||||
|
return errors.New("不支持的提示词操作")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListPromptHistory 查询旧版提示词接口保留的版本历史。
|
||||||
|
func (s *Service) ListPromptHistory(promptID string) ([]map[string]any, error) {
|
||||||
|
items := make([]map[string]any, 0)
|
||||||
|
err := s.db.Table("prompt_versions v").
|
||||||
|
Select("v.id,v.version,v.content,v.variables,v.version_note,v.status,v.created_at,a.username AS operator").
|
||||||
|
Joins("LEFT JOIN admin_users a ON a.id=v.operator_id").Where("v.prompt_id=?", promptID).
|
||||||
|
Order("v.version DESC").Find(&items).Error
|
||||||
|
return items, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListUserPrompts 查询系统提示词、当前用户自定义提示词及用户选择。
|
||||||
|
func (s *Service) ListUserPrompts(userID uuid.UUID, promptType string) (UserPromptList, error) {
|
||||||
|
if IsFixedDramaParseType(promptType) {
|
||||||
|
return UserPromptList{Prompts: []map[string]any{}}, nil
|
||||||
|
}
|
||||||
|
var prompts []map[string]any
|
||||||
|
err := s.db.Table("prompts").Select("id,name,type,content,scope,owner_user_id,created_at,updated_at,(scope='user') AS editable").
|
||||||
|
Where("deleted_at IS NULL AND type=? AND (scope='system' OR (scope='user' AND owner_user_id=?))", promptType, userID).
|
||||||
|
Order("CASE WHEN scope='system' THEN 0 ELSE 1 END, updated_at DESC").Find(&prompts).Error
|
||||||
|
if err != nil {
|
||||||
|
return UserPromptList{}, err
|
||||||
|
}
|
||||||
|
var selected struct{ PromptID *uuid.UUID }
|
||||||
|
// 用户未设置偏好时返回空属于正常情况，用 Limit(1).Find 避免触发 GORM 的 record not found 日志
|
||||||
|
s.db.Table("user_prompt_preferences").Select("prompt_id").Where("user_id=? AND prompt_type=?", userID, promptType).Limit(1).Find(&selected)
|
||||||
|
return UserPromptList{Prompts: prompts, SelectedID: selected.PromptID}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SelectUserPrompt 更新用户对指定提示词类型的选择，空值表示关闭选择。
|
||||||
|
func (s *Service) SelectUserPrompt(userID uuid.UUID, promptType string, promptID *uuid.UUID) error {
|
||||||
|
if IsFixedDramaParseType(promptType) {
|
||||||
|
return errors.New("该提示词由平台固定配置，不支持选择")
|
||||||
|
}
|
||||||
|
if promptID == nil {
|
||||||
|
if err := s.db.Exec("DELETE FROM user_prompt_preferences WHERE user_id=? AND prompt_type=?", userID, promptType).Error; err != nil {
|
||||||
|
return PersistenceError{Err: err}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var count int64
|
||||||
|
if err := s.db.Table("prompts").Where("id=? AND type=? AND deleted_at IS NULL AND (scope='system' OR (scope='user' AND owner_user_id=?))", *promptID, promptType, userID).Count(&count).Error; err != nil {
|
||||||
|
return PersistenceError{Err: err}
|
||||||
|
}
|
||||||
|
if count == 0 {
|
||||||
|
return gorm.ErrRecordNotFound
|
||||||
|
}
|
||||||
|
if err := s.db.Exec(`INSERT INTO user_prompt_preferences(id,user_id,prompt_id,prompt_type) VALUES(gen_random_uuid(),?,?,?) ON CONFLICT(user_id,prompt_type) DO UPDATE SET prompt_id=excluded.prompt_id,updated_at=CURRENT_TIMESTAMP`, userID, *promptID, promptType).Error; err != nil {
|
||||||
|
return PersistenceError{Err: err}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateCustomPrompt 校验并创建当前用户拥有的自定义提示词。
|
||||||
|
func (s *Service) CreateCustomPrompt(userID uuid.UUID, input CustomPromptInput) (CustomPrompt, error) {
|
||||||
|
input = normalizeCustomPromptInput(input)
|
||||||
|
if input.Name == "" || input.Type == "" || input.Content == "" {
|
||||||
|
return CustomPrompt{}, errors.New("name、type、content不能为空")
|
||||||
|
}
|
||||||
|
if IsFixedDramaParseType(input.Type) {
|
||||||
|
return CustomPrompt{}, errors.New("该提示词由平台固定配置，不支持自定义")
|
||||||
|
}
|
||||||
|
item := CustomPrompt{ID: uuid.New(), Name: input.Name, Type: input.Type, Content: input.Content, Scope: "user", Editable: true}
|
||||||
|
if err := s.db.Exec("INSERT INTO prompts(id,name,type,content,scope,owner_user_id) VALUES(?,?,?,?,'user',?)", item.ID, item.Name, item.Type, item.Content, userID).Error; err != nil {
|
||||||
|
return CustomPrompt{}, err
|
||||||
|
}
|
||||||
|
return item, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateCustomPrompt 校验并更新当前用户拥有的自定义提示词。
|
||||||
|
func (s *Service) UpdateCustomPrompt(userID, promptID uuid.UUID, input CustomPromptInput) (CustomPrompt, error) {
|
||||||
|
input = normalizeCustomPromptInput(input)
|
||||||
|
if input.Name == "" || input.Content == "" {
|
||||||
|
return CustomPrompt{}, errors.New("name、content不能为空")
|
||||||
|
}
|
||||||
|
if IsFixedDramaParseType(input.Type) {
|
||||||
|
return CustomPrompt{}, errors.New("该提示词由平台固定配置，不支持自定义")
|
||||||
|
}
|
||||||
|
result := s.db.Exec("UPDATE prompts SET name=?,type=COALESCE(NULLIF(?,''),type),content=?,updated_at=CURRENT_TIMESTAMP WHERE id=? AND scope='user' AND owner_user_id=? AND deleted_at IS NULL", input.Name, input.Type, input.Content, promptID, userID)
|
||||||
|
if result.Error != nil {
|
||||||
|
return CustomPrompt{}, result.Error
|
||||||
|
}
|
||||||
|
if result.RowsAffected == 0 {
|
||||||
|
return CustomPrompt{}, gorm.ErrRecordNotFound
|
||||||
|
}
|
||||||
|
return CustomPrompt{ID: promptID, Name: input.Name, Type: input.Type, Content: input.Content, Scope: "user", Editable: true}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteCustomPrompt 在同一事务中删除用户选择关系及其自定义提示词。
|
||||||
|
func (s *Service) DeleteCustomPrompt(userID, promptID uuid.UUID) error {
|
||||||
|
return s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
if err := tx.Exec("DELETE FROM user_prompt_preferences WHERE prompt_id=? AND user_id=?", promptID, userID).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
result := tx.Exec("DELETE FROM prompts WHERE id=? AND scope='user' AND owner_user_id=?", promptID, userID)
|
||||||
|
if result.Error != nil {
|
||||||
|
return result.Error
|
||||||
|
}
|
||||||
|
if result.RowsAffected == 0 {
|
||||||
|
return gorm.ErrRecordNotFound
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// normalizeCustomPromptInput 清理用户输入两端空白，保证校验和写入使用一致值。
|
||||||
|
func normalizeCustomPromptInput(input CustomPromptInput) CustomPromptInput {
|
||||||
|
input.Name = strings.TrimSpace(input.Name)
|
||||||
|
input.Type = strings.TrimSpace(input.Type)
|
||||||
|
input.Content = strings.TrimSpace(input.Content)
|
||||||
|
return input
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
// 提示词模块单元测试，验证固定类型规则和用户输入标准化行为。
|
||||||
|
package prompt
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
// TestIsFixedDramaParseType 验证固定提示词类型及空白清理规则。
|
||||||
|
func TestIsFixedDramaParseType(t *testing.T) {
|
||||||
|
for _, value := range []string{"剧本解析", " 角色、场景、道具解析 "} {
|
||||||
|
if !IsFixedDramaParseType(value) {
|
||||||
|
t.Fatalf("expected %q to be fixed", value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if IsFixedDramaParseType("剧本分析") {
|
||||||
|
t.Fatal("剧本分析不应被识别为固定解析提示词")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestNormalizeCustomPromptInput 验证提示词字段在校验和写入前统一清理两端空白。
|
||||||
|
func TestNormalizeCustomPromptInput(t *testing.T) {
|
||||||
|
input := normalizeCustomPromptInput(CustomPromptInput{Name: " 名称 ", Type: " 类型\n", Content: " 内容 "})
|
||||||
|
if input.Name != "名称" || input.Type != "类型" || input.Content != "内容" {
|
||||||
|
t.Fatalf("unexpected normalized input: %#v", input)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,64 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/provider"
|
||||||
|
)
|
||||||
|
|
||||||
|
type BalanceResult struct {
|
||||||
|
RemainBalance float64 `json:"remain_balance"`
|
||||||
|
RemainCredits float64 `json:"remain_credits"`
|
||||||
|
UsedBalance float64 `json:"used_balance"`
|
||||||
|
UsedCredits float64 `json:"used_credits"`
|
||||||
|
UnlimitedQuota bool `json:"unlimited_quota"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Client) Balance(ctx context.Context, baseURL, apiKey string) (BalanceResult, error) {
|
||||||
|
var result BalanceResult
|
||||||
|
request, err := http.NewRequestWithContext(ctx, http.MethodGet, strings.TrimRight(baseURL, "/")+"/user/balance", nil)
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
request.Header.Set("Authorization", "Bearer "+apiKey)
|
||||||
|
|
||||||
|
client := c.HTTPClient
|
||||||
|
if client == nil {
|
||||||
|
client = http.DefaultClient
|
||||||
|
}
|
||||||
|
response, err := client.Do(request)
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
defer response.Body.Close()
|
||||||
|
|
||||||
|
data, err := io.ReadAll(io.LimitReader(response.Body, 1024*1024))
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
||||||
|
return result, &provider.HTTPError{StatusCode: response.StatusCode, Body: safeBody(data), RetryAfter: response.Header.Get("Retry-After")}
|
||||||
|
}
|
||||||
|
|
||||||
|
var decoded struct {
|
||||||
|
Success bool `json:"success"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
BalanceResult
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
return result, fmt.Errorf("余额响应不是有效 JSON: %w", err)
|
||||||
|
}
|
||||||
|
if !decoded.Success {
|
||||||
|
if decoded.Message == "" {
|
||||||
|
return result, errors.New("获取渠道余额失败")
|
||||||
|
}
|
||||||
|
return result, errors.New(decoded.Message)
|
||||||
|
}
|
||||||
|
return decoded.BalanceResult, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestBalanceUsesConfiguredBaseURLAndAPIKey(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
|
||||||
|
if request.URL.Path != "/v1/user/balance" {
|
||||||
|
t.Fatalf("unexpected path: %s", request.URL.Path)
|
||||||
|
}
|
||||||
|
if request.Header.Get("Authorization") != "Bearer test-key" {
|
||||||
|
t.Fatalf("unexpected authorization header: %s", request.Header.Get("Authorization"))
|
||||||
|
}
|
||||||
|
fmt.Fprint(response, `{"success":true,"remain_balance":10.5,"remain_credits":105.25,"used_balance":2.3,"used_credits":2005.21388,"unlimited_quota":false}`)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
result, err := NewClient(server.Client()).Balance(context.Background(), server.URL+"/v1", "test-key")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Balance returned error: %v", err)
|
||||||
|
}
|
||||||
|
if result.RemainBalance != 10.5 || result.RemainCredits != 105.25 || result.UsedBalance != 2.3 || result.UsedCredits != 2005.21388 || result.UnlimitedQuota {
|
||||||
|
t.Fatalf("unexpected result: %#v", result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBalanceReturnsUpstreamMessage(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, _ *http.Request) {
|
||||||
|
fmt.Fprint(response, `{"success":false,"message":"record not found"}`)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
_, err := NewClient(server.Client()).Balance(context.Background(), server.URL, "test-key")
|
||||||
|
if err == nil || err.Error() != "record not found" {
|
||||||
|
t.Fatalf("expected upstream message, got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,141 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/provider"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ChatResult struct {
|
||||||
|
Content string
|
||||||
|
Raw json.RawMessage
|
||||||
|
FinishReason string
|
||||||
|
InputTokens int64
|
||||||
|
OutputTokens int64
|
||||||
|
TotalTokens int64
|
||||||
|
UsageRaw json.RawMessage
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Client) Chat(ctx context.Context, baseURL, apiKey string, payload map[string]any) (string, json.RawMessage, error) {
|
||||||
|
result, err := c.ChatWithUsage(ctx, baseURL, apiKey, payload)
|
||||||
|
return result.Content, result.Raw, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Client) ChatWithUsage(ctx context.Context, baseURL, apiKey string, payload map[string]any) (ChatResult, error) {
|
||||||
|
var result ChatResult
|
||||||
|
requestPayload := make(map[string]any, len(payload)+1)
|
||||||
|
for key, value := range payload {
|
||||||
|
requestPayload[key] = value
|
||||||
|
}
|
||||||
|
requestPayload["stream"] = false
|
||||||
|
body, err := json.Marshal(requestPayload)
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
request, err := http.NewRequestWithContext(ctx, http.MethodPost, strings.TrimRight(baseURL, "/")+"/chat/completions", bytes.NewReader(body))
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
request.Header.Set("Authorization", "Bearer "+apiKey)
|
||||||
|
request.Header.Set("Content-Type", "application/json")
|
||||||
|
response, err := c.HTTPClient.Do(request)
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
defer response.Body.Close()
|
||||||
|
data, err := io.ReadAll(io.LimitReader(response.Body, 16*1024*1024))
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
||||||
|
result.Raw = append(json.RawMessage(nil), data...)
|
||||||
|
return result, &provider.HTTPError{StatusCode: response.StatusCode, Body: safeBody(data), RetryAfter: response.Header.Get("Retry-After")}
|
||||||
|
}
|
||||||
|
type chatChoice struct {
|
||||||
|
Message struct {
|
||||||
|
Content any `json:"content"`
|
||||||
|
} `json:"message"`
|
||||||
|
FinishReason string `json:"finish_reason"`
|
||||||
|
}
|
||||||
|
var decoded struct {
|
||||||
|
Choices []chatChoice `json:"choices"`
|
||||||
|
Usage json.RawMessage `json:"usage"`
|
||||||
|
Data struct {
|
||||||
|
Choices []chatChoice `json:"choices"`
|
||||||
|
Usage json.RawMessage `json:"usage"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
result.Raw = append(json.RawMessage(nil), data...)
|
||||||
|
return result, fmt.Errorf("文本模型响应不是有效 JSON: %w；上游响应: %s", err, safeBody(data))
|
||||||
|
}
|
||||||
|
result.Raw = append(json.RawMessage(nil), data...)
|
||||||
|
result.UsageRaw = decoded.Usage
|
||||||
|
choices := decoded.Choices
|
||||||
|
if len(choices) == 0 {
|
||||||
|
choices = decoded.Data.Choices
|
||||||
|
if len(result.UsageRaw) == 0 {
|
||||||
|
result.UsageRaw = decoded.Data.Usage
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(choices) == 0 {
|
||||||
|
return result, fmt.Errorf("文本模型响应缺少 choices；上游响应: %s", safeBody(data))
|
||||||
|
}
|
||||||
|
result.FinishReason = choices[0].FinishReason
|
||||||
|
if len(result.UsageRaw) > 0 {
|
||||||
|
var usage map[string]any
|
||||||
|
if json.Unmarshal(result.UsageRaw, &usage) == nil {
|
||||||
|
result.InputTokens = intValue(usage["prompt_tokens"], usage["input_tokens"])
|
||||||
|
result.OutputTokens = intValue(usage["completion_tokens"], usage["output_tokens"])
|
||||||
|
result.TotalTokens = intValue(usage["total_tokens"])
|
||||||
|
if result.TotalTokens == 0 {
|
||||||
|
result.TotalTokens = result.InputTokens + result.OutputTokens
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
switch content := choices[0].Message.Content.(type) {
|
||||||
|
case string:
|
||||||
|
result.Content = content
|
||||||
|
return result, nil
|
||||||
|
case []any:
|
||||||
|
parts := make([]string, 0)
|
||||||
|
for _, item := range content {
|
||||||
|
if block, ok := item.(map[string]any); ok {
|
||||||
|
if text, ok := block["text"].(string); ok {
|
||||||
|
parts = append(parts, text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
result.Content = strings.Join(parts, "\n")
|
||||||
|
return result, nil
|
||||||
|
default:
|
||||||
|
return result, errors.New("文本模型响应内容无效")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func intValue(values ...any) int64 {
|
||||||
|
for _, value := range values {
|
||||||
|
switch number := value.(type) {
|
||||||
|
case float64:
|
||||||
|
if number > 0 {
|
||||||
|
return int64(number)
|
||||||
|
}
|
||||||
|
case int64:
|
||||||
|
if number > 0 {
|
||||||
|
return number
|
||||||
|
}
|
||||||
|
case json.Number:
|
||||||
|
if parsed, err := number.Int64(); err == nil && parsed > 0 {
|
||||||
|
return parsed
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
@@ -0,0 +1,58 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/provider"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestChatWithUsageReadsOpenAICompatibleAndNestedUsage(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
name string
|
||||||
|
response string
|
||||||
|
content string
|
||||||
|
input int64
|
||||||
|
output int64
|
||||||
|
total int64
|
||||||
|
finish string
|
||||||
|
}{
|
||||||
|
{name: "top level", response: `{"choices":[{"message":{"content":"result"},"finish_reason":"stop"}],"usage":{"prompt_tokens":120,"completion_tokens":30,"total_tokens":150}}`, content: "result", input: 120, output: 30, total: 150, finish: "stop"},
|
||||||
|
{name: "nested alternate names", response: `{"data":{"choices":[{"message":{"content":[{"type":"text","text":"part one"},{"type":"text","text":"part two"}]}}],"usage":{"input_tokens":44,"output_tokens":11}}}`, content: "part one\npart two", input: 44, output: 11, total: 55},
|
||||||
|
}
|
||||||
|
for _, test := range cases {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
|
||||||
|
if request.URL.Path != "/chat/completions" {
|
||||||
|
t.Fatalf("unexpected path: %s", request.URL.Path)
|
||||||
|
}
|
||||||
|
response.Header().Set("Content-Type", "application/json")
|
||||||
|
fmt.Fprint(response, test.response)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
result, err := NewClient(server.Client()).ChatWithUsage(context.Background(), server.URL, "key", map[string]any{"model": "text-model"})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ChatWithUsage returned error: %v", err)
|
||||||
|
}
|
||||||
|
if result.Content != test.content || result.InputTokens != test.input || result.OutputTokens != test.output || result.TotalTokens != test.total || result.FinishReason != test.finish {
|
||||||
|
t.Fatalf("unexpected result: %#v", result)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChatWithUsageReturnsExplicitUpstreamFailure(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, _ *http.Request) {
|
||||||
|
response.WriteHeader(http.StatusBadGateway)
|
||||||
|
fmt.Fprint(response, `{"error":"relay unavailable"}`)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
_, err := NewClient(server.Client()).ChatWithUsage(context.Background(), server.URL, "key", map[string]any{"model": "text-model"})
|
||||||
|
httpErr, ok := err.(*provider.HTTPError)
|
||||||
|
if !ok || httpErr.StatusCode != http.StatusBadGateway {
|
||||||
|
t.Fatalf("expected HTTPError with relay status, got %T %v", err, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,64 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/provider"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Client struct {
|
||||||
|
HTTPClient *http.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewClient(client *http.Client) Client {
|
||||||
|
return Client{HTTPClient: client}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Client) requestJSON(ctx context.Context, method, baseURL, path, apiKey, requestID string, payload map[string]any) ([]byte, error) {
|
||||||
|
var body io.Reader
|
||||||
|
if payload != nil {
|
||||||
|
encoded, err := json.Marshal(payload)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
body = bytes.NewReader(encoded)
|
||||||
|
}
|
||||||
|
request, err := http.NewRequestWithContext(ctx, method, strings.TrimRight(baseURL, "/")+path, body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
request.Header.Set("Authorization", "Bearer "+apiKey)
|
||||||
|
if payload != nil {
|
||||||
|
request.Header.Set("Content-Type", "application/json")
|
||||||
|
}
|
||||||
|
if requestID != "" {
|
||||||
|
request.Header.Set("Idempotency-Key", requestID)
|
||||||
|
request.Header.Set("X-Request-ID", requestID)
|
||||||
|
}
|
||||||
|
response, err := c.HTTPClient.Do(request)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer response.Body.Close()
|
||||||
|
data, err := io.ReadAll(io.LimitReader(response.Body, 4*1024*1024))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
||||||
|
return nil, &provider.HTTPError{StatusCode: response.StatusCode, Body: safeBody(data), RetryAfter: response.Header.Get("Retry-After")}
|
||||||
|
}
|
||||||
|
return data, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func safeBody(data []byte) string {
|
||||||
|
value := strings.TrimSpace(string(data))
|
||||||
|
if len(value) > 2000 {
|
||||||
|
value = value[:2000]
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
@@ -0,0 +1,103 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/provider"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c Client) SubmitImage(ctx context.Context, baseURL, apiKey, requestID string, payload map[string]any) (provider.SubmitResult, error) {
|
||||||
|
data, err := c.requestJSON(ctx, http.MethodPost, baseURL, "/images/generations", apiKey, requestID, payload)
|
||||||
|
if err != nil {
|
||||||
|
return provider.SubmitResult{}, err
|
||||||
|
}
|
||||||
|
result := provider.SubmitResult{Raw: append(json.RawMessage(nil), data...)}
|
||||||
|
var decoded any
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
return provider.SubmitResult{}, errors.New("中转站提交响应不是有效 JSON")
|
||||||
|
}
|
||||||
|
result.TaskID = findString(decoded, "task_id", "id", "request_id")
|
||||||
|
result.URL = findURL(decoded)
|
||||||
|
if result.TaskID == "" && result.URL == "" {
|
||||||
|
return provider.SubmitResult{}, errors.New("中转站提交响应缺少任务 ID 或结果 URL")
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// PollImage 通过 APIMart 统一任务接口查询图片生成状态和结果。
|
||||||
|
func (c Client) PollImage(ctx context.Context, baseURL, apiKey, taskID string) (provider.PollResult, error) {
|
||||||
|
data, err := c.requestJSON(ctx, http.MethodGet, baseURL, "/tasks/"+taskID, apiKey, "", nil)
|
||||||
|
if err != nil {
|
||||||
|
return provider.PollResult{}, err
|
||||||
|
}
|
||||||
|
var decoded any
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
return provider.PollResult{}, errors.New("中转站轮询响应不是有效 JSON")
|
||||||
|
}
|
||||||
|
status := strings.ToLower(findString(decoded, "task_status", "status", "state"))
|
||||||
|
if status == "" {
|
||||||
|
status = "processing"
|
||||||
|
}
|
||||||
|
return provider.PollResult{Status: status, URL: findURL(decoded), Error: findString(decoded, "error_message", "error", "message", "detail"), Raw: append(json.RawMessage(nil), data...)}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func findURL(value any) string {
|
||||||
|
var walk func(any, string) string
|
||||||
|
walk = func(node any, parentKey string) string {
|
||||||
|
switch data := node.(type) {
|
||||||
|
case string:
|
||||||
|
if (strings.Contains(strings.ToLower(parentKey), "url") || strings.HasPrefix(data, "http://") || strings.HasPrefix(data, "https://")) && (strings.HasPrefix(data, "https://") || strings.HasPrefix(data, "http://")) {
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
case map[string]any:
|
||||||
|
for key, child := range data {
|
||||||
|
if found := walk(child, key); found != "" {
|
||||||
|
return found
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case []any:
|
||||||
|
for _, child := range data {
|
||||||
|
if found := walk(child, parentKey); found != "" {
|
||||||
|
return found
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return walk(value, "")
|
||||||
|
}
|
||||||
|
|
||||||
|
func findString(value any, keys ...string) string {
|
||||||
|
wanted := map[string]bool{}
|
||||||
|
for _, key := range keys {
|
||||||
|
wanted[strings.ToLower(key)] = true
|
||||||
|
}
|
||||||
|
return findMatchingString(value, func(key, _ string) bool { return wanted[strings.ToLower(key)] })
|
||||||
|
}
|
||||||
|
|
||||||
|
func findMatchingString(value any, match func(key, value string) bool) string {
|
||||||
|
switch data := value.(type) {
|
||||||
|
case map[string]any:
|
||||||
|
for key, child := range data {
|
||||||
|
if text, ok := child.(string); ok && match(key, text) {
|
||||||
|
return text
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, child := range data {
|
||||||
|
if found := findMatchingString(child, match); found != "" {
|
||||||
|
return found
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case []any:
|
||||||
|
for _, child := range data {
|
||||||
|
if found := findMatchingString(child, match); found != "" {
|
||||||
|
return found
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
@@ -0,0 +1,48 @@
|
|||||||
|
// 图片生成中转站适配器测试，验证任务提交、状态轮询及成功结果解析契约。
|
||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestSubmitImageUsesGenerationEndpoint 验证图片生成使用配置渠道下的标准提交接口。
|
||||||
|
func TestSubmitImageUsesGenerationEndpoint(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
|
||||||
|
if request.Method != http.MethodPost || request.URL.Path != "/v1/images/generations" {
|
||||||
|
t.Fatalf("unexpected request: %s %s", request.Method, request.URL.Path)
|
||||||
|
}
|
||||||
|
fmt.Fprint(response, `{"code":200,"data":[{"status":"submitted","task_id":"task-image-1"}]}`)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
result, err := NewClient(server.Client()).SubmitImage(context.Background(), server.URL+"/v1", "test-key", "request-1", map[string]any{"model": "gpt-image-2"})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SubmitImage returned error: %v", err)
|
||||||
|
}
|
||||||
|
if result.TaskID != "task-image-1" {
|
||||||
|
t.Fatalf("unexpected task id: %s", result.TaskID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestPollImageUsesTaskEndpoint 验证图片任务使用 APIMart 统一任务接口查询并解析完成结果。
|
||||||
|
func TestPollImageUsesTaskEndpoint(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
|
||||||
|
if request.Method != http.MethodGet || request.URL.Path != "/v1/tasks/task-image-1" {
|
||||||
|
t.Fatalf("unexpected request: %s %s", request.Method, request.URL.Path)
|
||||||
|
}
|
||||||
|
fmt.Fprint(response, `{"code":200,"data":{"id":"task-image-1","status":"completed","result":{"images":[{"url":["https://example.com/result.png"]}]}}}`)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
result, err := NewClient(server.Client()).PollImage(context.Background(), server.URL+"/v1", "test-key", "task-image-1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("PollImage returned error: %v", err)
|
||||||
|
}
|
||||||
|
if result.Status != "completed" || result.URL != "https://example.com/result.png" {
|
||||||
|
t.Fatalf("unexpected result: %#v", result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,225 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"juhe-factory/api/internal/provider"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Seedance2 struct {
|
||||||
|
Gateway Client
|
||||||
|
}
|
||||||
|
|
||||||
|
type ReferenceCapability struct {
|
||||||
|
Supported bool `json:"supported"`
|
||||||
|
MaxCount int `json:"max_count"`
|
||||||
|
MinTotalDurationMS int64 `json:"min_total_duration_ms,omitempty"`
|
||||||
|
MaxTotalDurationMS int64 `json:"max_total_duration_ms,omitempty"`
|
||||||
|
MinShortSidePixels int `json:"min_short_side_pixels,omitempty"`
|
||||||
|
MaxShortSidePixels int `json:"max_short_side_pixels,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type VideoModelCapabilities struct {
|
||||||
|
// DurationMinSeconds / DurationMaxSeconds 表示视频模型支持的生成时长范围（秒），供前端可视化选择与后端校验。
|
||||||
|
DurationMinSeconds int `json:"duration_min_seconds"`
|
||||||
|
DurationMaxSeconds int `json:"duration_max_seconds"`
|
||||||
|
ReferenceImages ReferenceCapability `json:"reference_images"`
|
||||||
|
ReferenceAudios ReferenceCapability `json:"reference_audios"`
|
||||||
|
ReferenceVideos ReferenceCapability `json:"reference_videos"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var seedance2Capabilities = map[string]VideoModelCapabilities{
|
||||||
|
"doubao-seedance-2.0": seedance2ReferenceCapabilities(),
|
||||||
|
"doubao-seedance-2.0-fast": seedance2ReferenceCapabilities(),
|
||||||
|
"doubao-seedance-2.0-mini": seedance2ReferenceCapabilities(),
|
||||||
|
}
|
||||||
|
|
||||||
|
func seedance2ReferenceCapabilities() VideoModelCapabilities {
|
||||||
|
return VideoModelCapabilities{
|
||||||
|
DurationMinSeconds: 5,
|
||||||
|
DurationMaxSeconds: 15,
|
||||||
|
ReferenceImages: ReferenceCapability{Supported: true, MaxCount: 9},
|
||||||
|
ReferenceAudios: ReferenceCapability{Supported: true, MaxCount: 3, MaxTotalDurationMS: 15_000},
|
||||||
|
ReferenceVideos: ReferenceCapability{Supported: true, MaxCount: 3, MinTotalDurationMS: 1_800, MaxTotalDurationMS: 15_200, MinShortSidePixels: 480, MaxShortSidePixels: 720},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func VideoCapabilities(modelName string) (VideoModelCapabilities, bool) {
|
||||||
|
capabilities, ok := seedance2Capabilities[strings.TrimSpace(modelName)]
|
||||||
|
return capabilities, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSeedance2(client *http.Client) Seedance2 {
|
||||||
|
return Seedance2{Gateway: NewClient(client)}
|
||||||
|
}
|
||||||
|
|
||||||
|
func BuildSeedance2Payload(modelName string, inputData json.RawMessage) (map[string]any, error) {
|
||||||
|
modelName = strings.TrimSpace(modelName)
|
||||||
|
capabilities, supported := VideoCapabilities(modelName)
|
||||||
|
if !supported {
|
||||||
|
return nil, fmt.Errorf("暂不支持视频模型 %q", modelName)
|
||||||
|
}
|
||||||
|
input := map[string]any{}
|
||||||
|
if err := json.Unmarshal(inputData, &input); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
prompt := strings.TrimSpace(fmt.Sprint(input["prompt"]))
|
||||||
|
if prompt == "" || prompt == "<nil>" {
|
||||||
|
return nil, errors.New("生成提示词不能为空")
|
||||||
|
}
|
||||||
|
size := strings.TrimSpace(fmt.Sprint(input["size"]))
|
||||||
|
if size != "16:9" && size != "9:16" {
|
||||||
|
return nil, errors.New("项目画面比例不受 Seedance 2 支持")
|
||||||
|
}
|
||||||
|
duration, err := strconv.Atoi(fmt.Sprint(input["duration"]))
|
||||||
|
if err != nil || duration < 5 || duration > 15 {
|
||||||
|
return nil, errors.New("Seedance 2 视频时长必须为 5 到 15 秒")
|
||||||
|
}
|
||||||
|
images, err := seedanceURLList(input["image_urls"])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Seedance 2 参考图无效: %w", err)
|
||||||
|
}
|
||||||
|
if len(images) > capabilities.ReferenceImages.MaxCount {
|
||||||
|
return nil, fmt.Errorf("Seedance 2 最多支持 %d 张参考图", capabilities.ReferenceImages.MaxCount)
|
||||||
|
}
|
||||||
|
audios, err := seedanceURLList(input["audio_urls"])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Seedance 2 参考音频无效: %w", err)
|
||||||
|
}
|
||||||
|
if len(audios) > capabilities.ReferenceAudios.MaxCount {
|
||||||
|
return nil, fmt.Errorf("Seedance 2 最多支持 %d 个参考音频", capabilities.ReferenceAudios.MaxCount)
|
||||||
|
}
|
||||||
|
videos, err := seedanceURLList(input["video_urls"])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Seedance 2 参考视频无效: %w", err)
|
||||||
|
}
|
||||||
|
if len(videos) > capabilities.ReferenceVideos.MaxCount {
|
||||||
|
return nil, fmt.Errorf("Seedance 2 最多支持 %d 个参考视频", capabilities.ReferenceVideos.MaxCount)
|
||||||
|
}
|
||||||
|
if len(audios) > 0 && len(images) == 0 && len(videos) == 0 {
|
||||||
|
return nil, errors.New("Seedance 2 参考音频必须与参考图片或参考视频同时使用")
|
||||||
|
}
|
||||||
|
payload := map[string]any{
|
||||||
|
"model": modelName, "prompt": prompt, "size": size, "duration": duration,
|
||||||
|
"resolution": seedanceString(input["resolution"], "480p"),
|
||||||
|
}
|
||||||
|
if len(images) > 0 {
|
||||||
|
payload["image_urls"] = images
|
||||||
|
}
|
||||||
|
if len(audios) > 0 {
|
||||||
|
payload["audio_urls"] = audios
|
||||||
|
}
|
||||||
|
if len(videos) > 0 {
|
||||||
|
payload["video_urls"] = videos
|
||||||
|
}
|
||||||
|
if value, ok := input["generate_audio"].(bool); ok {
|
||||||
|
payload["generate_audio"] = value
|
||||||
|
}
|
||||||
|
if value, ok := input["return_last_frame"].(bool); ok {
|
||||||
|
payload["return_last_frame"] = value
|
||||||
|
}
|
||||||
|
return payload, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s Seedance2) Submit(ctx context.Context, baseURL, apiKey, requestID string, payload map[string]any) (provider.SubmitResult, error) {
|
||||||
|
data, err := s.Gateway.requestJSON(ctx, http.MethodPost, baseURL, "/videos/generations", apiKey, requestID, payload)
|
||||||
|
if err != nil {
|
||||||
|
return provider.SubmitResult{}, err
|
||||||
|
}
|
||||||
|
var decoded any
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
return provider.SubmitResult{}, errors.New("Seedance 2 提交响应不是有效 JSON")
|
||||||
|
}
|
||||||
|
result := provider.SubmitResult{TaskID: findString(decoded, "task_id", "id"), URL: findSeedanceVideoURL(decoded), Raw: append(json.RawMessage(nil), data...)}
|
||||||
|
if result.TaskID == "" && result.URL == "" {
|
||||||
|
return provider.SubmitResult{}, errors.New("Seedance 2 提交响应缺少任务 ID 或视频地址")
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s Seedance2) Poll(ctx context.Context, baseURL, apiKey, taskID string) (provider.PollResult, error) {
|
||||||
|
data, err := s.Gateway.requestJSON(ctx, http.MethodGet, baseURL, "/tasks/"+taskID, apiKey, "", nil)
|
||||||
|
if err != nil {
|
||||||
|
return provider.PollResult{}, err
|
||||||
|
}
|
||||||
|
var decoded any
|
||||||
|
if err := json.Unmarshal(data, &decoded); err != nil {
|
||||||
|
return provider.PollResult{}, errors.New("Seedance 2 轮询响应不是有效 JSON")
|
||||||
|
}
|
||||||
|
status := strings.ToLower(findString(decoded, "task_status", "status", "state"))
|
||||||
|
if status == "" {
|
||||||
|
status = "processing"
|
||||||
|
}
|
||||||
|
return provider.PollResult{
|
||||||
|
Status: status, URL: findSeedanceVideoURL(decoded),
|
||||||
|
Error: findString(decoded, "error_message", "message", "detail"), Raw: append(json.RawMessage(nil), data...),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func seedanceURLList(value any) ([]string, error) {
|
||||||
|
if value == nil {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
raw, ok := value.([]any)
|
||||||
|
if !ok {
|
||||||
|
return nil, errors.New("必须是 URL 数组")
|
||||||
|
}
|
||||||
|
result := make([]string, 0, len(raw))
|
||||||
|
for _, item := range raw {
|
||||||
|
url := strings.TrimSpace(fmt.Sprint(item))
|
||||||
|
if !strings.HasPrefix(url, "https://") && !strings.HasPrefix(url, "http://") && !strings.HasPrefix(url, "asset://") {
|
||||||
|
return nil, errors.New("包含不可访问的 URL")
|
||||||
|
}
|
||||||
|
result = append(result, url)
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func seedanceString(value any, fallback string) string {
|
||||||
|
text := strings.TrimSpace(fmt.Sprint(value))
|
||||||
|
if text == "" || text == "<nil>" {
|
||||||
|
return fallback
|
||||||
|
}
|
||||||
|
return text
|
||||||
|
}
|
||||||
|
|
||||||
|
func findSeedanceVideoURL(value any) string {
|
||||||
|
if url := findURLInSeedanceVideoField(value); url != "" {
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
return findURL(value)
|
||||||
|
}
|
||||||
|
|
||||||
|
func findURLInSeedanceVideoField(value any) string {
|
||||||
|
switch data := value.(type) {
|
||||||
|
case map[string]any:
|
||||||
|
for key, child := range data {
|
||||||
|
if strings.Contains(strings.ToLower(key), "video") {
|
||||||
|
if text, ok := child.(string); ok && (strings.HasPrefix(text, "https://") || strings.HasPrefix(text, "http://")) {
|
||||||
|
return text
|
||||||
|
}
|
||||||
|
if url := findURL(child); url != "" {
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, child := range data {
|
||||||
|
if url := findURLInSeedanceVideoField(child); url != "" {
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case []any:
|
||||||
|
for _, child := range data {
|
||||||
|
if url := findURLInSeedanceVideoField(child); url != "" {
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
@@ -0,0 +1,48 @@
|
|||||||
|
package apimart
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestVideoCapabilities(t *testing.T) {
|
||||||
|
capabilities, ok := VideoCapabilities("doubao-seedance-2.0")
|
||||||
|
if !ok {
|
||||||
|
t.Fatal("expected Seedance 2 capabilities")
|
||||||
|
}
|
||||||
|
if capabilities.ReferenceImages.MaxCount != 9 || capabilities.ReferenceAudios.MaxCount != 3 || capabilities.ReferenceVideos.MaxCount != 3 {
|
||||||
|
t.Fatalf("unexpected reference limits: %+v", capabilities)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildSeedance2PayloadSupportsVideoAndAudioReferences(t *testing.T) {
|
||||||
|
input := json.RawMessage(`{
|
||||||
|
"prompt":"test",
|
||||||
|
"size":"16:9",
|
||||||
|
"duration":5,
|
||||||
|
"video_urls":["https://example.com/reference.mp4"],
|
||||||
|
"audio_urls":["https://example.com/reference.mp3"]
|
||||||
|
}`)
|
||||||
|
payload, err := BuildSeedance2Payload("doubao-seedance-2.0", input)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("BuildSeedance2Payload() error = %v", err)
|
||||||
|
}
|
||||||
|
if videos, ok := payload["video_urls"].([]string); !ok || len(videos) != 1 {
|
||||||
|
t.Fatalf("video_urls = %#v", payload["video_urls"])
|
||||||
|
}
|
||||||
|
if audios, ok := payload["audio_urls"].([]string); !ok || len(audios) != 1 {
|
||||||
|
t.Fatalf("audio_urls = %#v", payload["audio_urls"])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildSeedance2PayloadRejectsTooManyVideos(t *testing.T) {
|
||||||
|
input := json.RawMessage(`{
|
||||||
|
"prompt":"test",
|
||||||
|
"size":"16:9",
|
||||||
|
"duration":5,
|
||||||
|
"video_urls":["https://example.com/1.mp4","https://example.com/2.mp4","https://example.com/3.mp4","https://example.com/4.mp4"]
|
||||||
|
}`)
|
||||||
|
if _, err := BuildSeedance2Payload("doubao-seedance-2.0", input); err == nil {
|
||||||
|
t.Fatal("expected too many reference videos to be rejected")
|
||||||
|
}
|
||||||
|
}
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user