refactor: split strm page hooks

2026-09-29 11:36:36 +08:00 · 2026-06-24 12:31:51 +08:00
parent aec5914846
commit ea34d7578e
5 changed files with 331 additions and 288 deletions
@@ -0,0 +1,174 @@
+import type { FormEvent } from 'react'
+import { useEffect, useState } from 'react'
+import toast from 'react-hot-toast'
+
+import { adminAPI } from '../api/admin'
+import { strmAPI, type GenerateSTRMResult } from '../api/strm'
+import type { Library } from '../types'
+import {
+  inferSTRMOutputRoot,
+  playbackStatusText,
+  preferredSTRMBaseURL,
+  suggestedSTRMOutputDir,
+  type CloudPlaybackMode,
+} from './strmPageModel'
+import { apiErrorMessage, isHTTPURL } from './strmPageUtils'
+
+export function useStrmGenerateForm(libraries: Library[]) {
+  const [generateLibraryID, setGenerateLibraryID] = useState('')
+  const [baseURL, setBaseURL] = useState('')
+  const [outputDir, setOutputDir] = useState('')
+  const [outputRoot, setOutputRoot] = useState('')
+  const [outputScope, setOutputScope] = useState('')
+  const [outputDirTouched, setOutputDirTouched] = useState(false)
+  const [settingsLoaded, setSettingsLoaded] = useState(false)
+  const [cloudPlaybackMode, setCloudPlaybackMode] = useState<CloudPlaybackMode>('redirect_proxy')
+  const [strmPlaybackEnabled, setStrmPlaybackEnabled] = useState(false)
+  const [redirectProxyEnabled, setRedirectProxyEnabled] = useState(true)
+  const [autoGenerate, setAutoGenerate] = useState(false)
+  const [savingSettings, setSavingSettings] = useState(false)
+  const [overwrite, setOverwrite] = useState(false)
+  const [generating, setGenerating] = useState(false)
+  const [generateResult, setGenerateResult] = useState<GenerateSTRMResult | null>(null)
+
+  useEffect(() => {
+    adminAPI
+      .listSettings()
+      .then((rows) => {
+        const settings = Object.fromEntries(rows.map((row) => [row.key, row.value]))
+        const savedOutputDir = settings['strm.output_dir'] || ''
+        const mode = settings['cloud.playback_mode']
+        const nextMode =
+          mode === 'strm' || mode === 'redirect_proxy'
+            ? mode
+            : settings['strm.enabled'] === 'true'
+              ? 'strm'
+              : 'redirect_proxy'
+
+        setBaseURL(preferredSTRMBaseURL(settings['strm.base_url'] || settings['app.server_url'] || ''))
+        setOutputDir(savedOutputDir)
+        setOutputRoot(savedOutputDir)
+        setOutputScope(settings['strm.output_scope'] || '')
+        setOutputDirTouched(false)
+        setCloudPlaybackMode(nextMode)
+        setStrmPlaybackEnabled(
+          settings['cloud.playback_strm_enabled'] != null
+            ? settings['cloud.playback_strm_enabled'] === 'true'
+            : settings['strm.enabled'] === 'true' || nextMode === 'strm',
+        )
+        setRedirectProxyEnabled(settings['cloud.playback_redirect_proxy_enabled'] !== 'false')
+        setAutoGenerate(settings['strm.auto_generate_enabled'] === 'true')
+        setSettingsLoaded(true)
+      })
+      .catch(() => setSettingsLoaded(true))
+  }, [])
+
+  useEffect(() => {
+    if (!generateLibraryID && libraries[0]) setGenerateLibraryID(libraries[0].id)
+  }, [libraries, generateLibraryID])
+
+  useEffect(() => {
+    if (!settingsLoaded || outputDirTouched || !generateLibraryID) return
+    const root = outputScope === 'library' ? inferSTRMOutputRoot(outputRoot, libraries) : outputRoot
+    if (generateLibraryID === '*') {
+      setOutputDir(root)
+      return
+    }
+    const library = libraries.find((item) => item.id === generateLibraryID)
+    if (library) setOutputDir(suggestedSTRMOutputDir(root, library))
+  }, [settingsLoaded, outputDirTouched, outputRoot, outputScope, generateLibraryID, libraries])
+
+  const onOutputDirChange = (value: string) => {
+    setOutputDir(value)
+    setOutputRoot(value)
+    setOutputScope('')
+    setOutputDirTouched(true)
+  }
+
+  const onGenerate = async (event: FormEvent) => {
+    event.preventDefault()
+    const trimmedBaseURL = baseURL.trim()
+    if (!generateLibraryID || !trimmedBaseURL) return
+    if (!isHTTPURL(trimmedBaseURL)) {
+      toast.error('域名必须以 http:// 或 https:// 开头')
+      return
+    }
+
+    setGenerating(true)
+    try {
+      const result = await strmAPI.generate({
+        library_id: generateLibraryID,
+        base_url: trimmedBaseURL.replace(/\/+$/, ''),
+        output_dir: outputDir.trim(),
+        overwrite,
+        enabled: autoGenerate,
+        include_local: true,
+      })
+      const nextOutputDir = result.output_dir || outputDir
+      setGenerateResult(result)
+      setOutputDir(nextOutputDir)
+      setOutputRoot(inferSTRMOutputRoot(nextOutputDir, libraries))
+      setOutputScope(generateLibraryID === '*' ? 'all' : 'library')
+      setOutputDirTouched(false)
+      toast.success(`生成完成：新增 ${result.generated} · 更新 ${result.updated} · 跳过 ${result.skipped}`)
+    } catch (err: unknown) {
+      toast.error(apiErrorMessage(err, '生成失败'))
+    } finally {
+      setGenerating(false)
+    }
+  }
+
+  const saveSTRMSettings = async () => {
+    setSavingSettings(true)
+    try {
+      const effectiveMode: CloudPlaybackMode =
+        strmPlaybackEnabled && !redirectProxyEnabled
+          ? 'strm'
+          : !strmPlaybackEnabled && redirectProxyEnabled
+            ? 'redirect_proxy'
+            : cloudPlaybackMode
+      await Promise.all([
+        adminAPI.updateSetting('cloud.playback_mode', effectiveMode),
+        adminAPI.updateSetting('cloud.playback_strm_enabled', String(strmPlaybackEnabled)),
+        adminAPI.updateSetting('cloud.playback_redirect_proxy_enabled', String(redirectProxyEnabled)),
+        adminAPI.updateSetting('strm.enabled', String(strmPlaybackEnabled)),
+        adminAPI.updateSetting('strm.auto_generate_enabled', String(autoGenerate)),
+      ])
+      setCloudPlaybackMode(effectiveMode)
+      toast.success(
+        strmPlaybackEnabled || redirectProxyEnabled
+          ? `播放设置已保存：优先 ${effectiveMode === 'strm' ? 'STRMURL' : '302/反代'}`
+          : '播放设置已保存：云盘第三方播放已关闭',
+      )
+    } catch (err: unknown) {
+      toast.error(apiErrorMessage(err, '保存 STRM 开关失败'))
+    } finally {
+      setSavingSettings(false)
+    }
+  }
+
+  return {
+    autoGenerate,
+    baseURL,
+    cloudPlaybackMode,
+    generateLibraryID,
+    generateResult,
+    generating,
+    onGenerate,
+    outputDir,
+    overwrite,
+    playbackStatus: playbackStatusText(strmPlaybackEnabled, redirectProxyEnabled, cloudPlaybackMode),
+    redirectProxyEnabled,
+    saveSTRMSettings,
+    savingSettings,
+    setAutoGenerate,
+    setBaseURL,
+    setCloudPlaybackMode,
+    setGenerateLibraryID,
+    setOutputDir: onOutputDirChange,
+    setOverwrite,
+    setRedirectProxyEnabled,
+    setStrmPlaybackEnabled,
+    strmPlaybackEnabled,
+  }
+}