Genarrative/apps/ai-game-creator-shell/src/features/app-shell/useProjectCreationDirectory.ts

import { useCallback, useRef, useState } from 'react';

import { resolveTauriInvoke } from '../../app/tauri';
import {
  readProjectCreationDirectory,
  writeProjectCreationDirectory,
} from './model';

/**
 * 「项目创建目录」用户偏好。
 *
 * 默认沿用 AGC 管理的应用数据目录（`<app_data>/projects`）；用户改选时必须走原生目录
 * 选择器，因为只有它构成"用户显式选择"边界：选择结果当场按用户选择范围加固，后续建项
 * 再由 Rust 侧私有路径门禁复核一次。
 */
export function useProjectCreationDirectory() {
  const [projectCreationDirectory, setProjectCreationDirectory] = useState(
    readProjectCreationDirectory,
  );
  const [projectCreationDirectoryBusy, setProjectCreationDirectoryBusy] =
    useState(false);
  const [projectCreationDirectoryStatus, setProjectCreationDirectoryStatus] =
    useState('');
  const pickInFlightRef = useRef(false);

  const pickProjectCreationDirectory = useCallback(async () => {
    if (pickInFlightRef.current) {
      return;
    }
    const invoke = resolveTauriInvoke();
    if (!invoke) {
      setProjectCreationDirectoryStatus('需要在陶泥儿客户端内运行');
      return;
    }
    pickInFlightRef.current = true;
    setProjectCreationDirectoryBusy(true);
    setProjectCreationDirectoryStatus('正在选择项目创建目录');
    try {
      const selected = await invoke<string | null>(
        'pick_local_project_directory',
        {
          title: '选择项目创建目录',
          ...(projectCreationDirectory
            ? { initialPath: projectCreationDirectory }
            : {}),
        },
      );
      if (!selected) {
        setProjectCreationDirectoryStatus('已取消');
        return;
      }
      setProjectCreationDirectory(writeProjectCreationDirectory(selected));
      setProjectCreationDirectoryStatus('已更新项目创建目录');
    } catch (error) {
      setProjectCreationDirectoryStatus(
        error instanceof Error ? error.message : String(error),
      );
    } finally {
      pickInFlightRef.current = false;
      setProjectCreationDirectoryBusy(false);
    }
  }, [projectCreationDirectory]);

  const resetProjectCreationDirectory = useCallback(() => {
    writeProjectCreationDirectory('');
    setProjectCreationDirectory('');
    setProjectCreationDirectoryStatus('已恢复默认位置');
  }, []);

  return {
    projectCreationDirectory,
    projectCreationDirectoryBusy,
    projectCreationDirectoryStatus,
    pickProjectCreationDirectory,
    resetProjectCreationDirectory,
  };
}

export type ProjectCreationDirectoryController = ReturnType<
  typeof useProjectCreationDirectory
>;