feat: add STRM tree dry-run preview

This commit is contained in:
ShukeBta
2026-07-02 21:15:11 +08:00
parent c07290619c
commit d11f784438
8 changed files with 113 additions and 14 deletions
+2
View File
@@ -17,6 +17,7 @@ export type GenerateSTRMResult = {
updated: number
skipped: number
cleaned: number
previewed?: number
errors?: string[]
items?: Array<{
media_id: string
@@ -38,6 +39,7 @@ export type GenerateSTRMTreeInput = {
base_url?: string
overwrite?: boolean
cleanup?: boolean
dry_run?: boolean
}
export const strmAPI = {
@@ -271,6 +271,7 @@ export function StrmGenerateResultPanel({ result }: { result: GenerateSTRMResult
输出目录:{result.output_dir}
</div>
<div className="mt-1">
{result.previewed ? `预检 ${result.previewed} · ` : ''}
新增 {result.generated} · 更新 {result.updated} · 跳过 {result.skipped} · 清理 {result.cleaned || 0}
</div>
{result.errors && result.errors.length > 0 && (
+14 -3
View File
@@ -32,6 +32,7 @@ export function StrmTreeGenerateSection({
cleanup,
generating,
onGenerate,
onPreview,
onImportTreeFile,
outputDir,
outputPrefix,
@@ -39,6 +40,7 @@ export function StrmTreeGenerateSection({
pathsText,
provider,
result,
runningMode,
setBaseURL,
setCleanup,
setOutputDir,
@@ -167,13 +169,22 @@ export function StrmTreeGenerateSection({
<input type="checkbox" checked={cleanup} onChange={(e) => setCleanup(e.target.checked)} />
清理不在当前目录树中的旧 STRM
</label>
<button
type="button"
disabled={generating || (!treeText.trim() && !pathsText.trim()) || !outputDir.trim()}
className="inline-flex min-h-10 items-center justify-center gap-2 rounded-2xl border border-primary-400/40 px-3 py-2 text-sm font-medium text-brand-500 transition hover:bg-primary-400/10 disabled:cursor-not-allowed disabled:opacity-50 md:col-span-2"
onClick={onPreview}
>
{runningMode === 'preview' ? <Loader2 size={16} className="animate-spin" /> : <FolderTree size={16} />}
{runningMode === 'preview' ? '预检中...' : '预检目录树'}
</button>
<button
type="submit"
disabled={generating || (!treeText.trim() && !pathsText.trim()) || !outputDir.trim()}
className="neon-button md:col-span-4"
className="neon-button md:col-span-2"
>
{generating ? <Loader2 size={16} className="animate-spin" /> : <FolderTree size={16} />}
{generating ? '生成中...' : '按目录树生成'}
{runningMode === 'generate' ? <Loader2 size={16} className="animate-spin" /> : <FolderTree size={16} />}
{runningMode === 'generate' ? '生成中...' : '按目录树生成'}
</button>
</form>
<StrmGenerateResultPanel result={result} />
+23 -8
View File
@@ -16,7 +16,7 @@ export function useStrmTreeGenerateForm() {
const [pathsText, setPathsText] = useState('')
const [overwrite, setOverwrite] = useState(false)
const [cleanup, setCleanup] = useState(false)
const [generating, setGenerating] = useState(false)
const [runningMode, setRunningMode] = useState<'generate' | 'preview' | null>(null)
const [result, setResult] = useState<GenerateSTRMResult | null>(null)
const onImportTreeFile = async (event: ChangeEvent<HTMLInputElement>) => {
@@ -38,8 +38,7 @@ export function useStrmTreeGenerateForm() {
}
}
const onGenerate = async (event: FormEvent) => {
event.preventDefault()
const runGenerate = async (dryRun: boolean) => {
const trimmedBaseURL = baseURL.trim()
const paths = parsePathList(pathsText)
if (!treeText.trim() && paths.length === 0) {
@@ -50,7 +49,7 @@ export function useStrmTreeGenerateForm() {
toast.error('域名必须以 http:// 或 https:// 开头')
return
}
setGenerating(true)
setRunningMode(dryRun ? 'preview' : 'generate')
try {
const next = await strmAPI.generateFromTree({
provider,
@@ -62,28 +61,44 @@ export function useStrmTreeGenerateForm() {
paths: paths.length > 0 ? paths : undefined,
overwrite,
cleanup,
dry_run: dryRun,
})
setResult(next)
toast.success(`生成完成:新增 ${next.generated} · 更新 ${next.updated} · 跳过 ${next.skipped}`)
if (dryRun) {
toast.success(`预检完成:将处理 ${next.previewed ?? 0} 个视频 · 跳过 ${next.skipped}`)
} else {
toast.success(`生成完成:新增 ${next.generated} · 更新 ${next.updated} · 跳过 ${next.skipped}`)
}
} catch (err: unknown) {
toast.error(apiErrorMessage(err, '目录树生成失败'))
toast.error(apiErrorMessage(err, dryRun ? '目录树预检失败' : '目录树生成失败'))
} finally {
setGenerating(false)
setRunningMode(null)
}
}
const onGenerate = async (event: FormEvent) => {
event.preventDefault()
await runGenerate(false)
}
const onPreview = async () => {
await runGenerate(true)
}
return {
baseURL,
cleanup,
generating,
generating: runningMode !== null,
onImportTreeFile,
onGenerate,
onPreview,
outputDir,
outputPrefix,
overwrite,
pathsText,
provider,
result,
runningMode,
setBaseURL,
setCleanup,
setOutputDir,