QMAI/src/lib/project-file-tree-refresh.ts

import { listDirectory } from "@/commands/fs"
import { useWikiStore } from "@/stores/wiki-store"
import { normalizePath } from "@/lib/path-utils"

export interface RefreshProjectFileTreeOptions {
  projectId?: string
  clearDisplayTreeFirst?: boolean
  bumpDataVersion?: boolean
  refreshDisplayTree?: boolean
  refreshPathIndex?: boolean
}

async function listDirectoryWithRetry(
  path: string,
  options: Parameters<typeof listDirectory>[1],
  attempts = 3,
): Promise<Awaited<ReturnType<typeof listDirectory>>> {
  let lastError: unknown
  for (let i = 0; i < attempts; i += 1) {
    try {
      return await listDirectory(path, options)
    } catch (err) {
      lastError = err
      if (i < attempts - 1) {
        await new Promise((resolve) => setTimeout(resolve, 250))
      }
    }
  }
  throw lastError
}

function isStillCurrentProject(projectId: string | undefined, projectPath: string): boolean {
  const current = useWikiStore.getState().project
  if (!current) return false
  if (projectId && current.id !== projectId) return false
  return normalizePath(current.path) === projectPath
}

/**
 * Reload the project file tree and optionally rebuild the path index.
 * QMAI keeps a full tree for KnowledgeTree (flatten/sort/chapter batch),
 * unlike llm-wiki's shallow display tree + lazy children.
 */
export async function refreshProjectFileTree(
  projectPath: string,
  options: RefreshProjectFileTreeOptions = {},
): Promise<void> {
  const normalizedProjectPath = normalizePath(projectPath)
  const currentProjectId = options.projectId ?? useWikiStore.getState().project?.id
  const refreshDisplayTree = options.refreshDisplayTree ?? true
  const refreshPathIndex = options.refreshPathIndex ?? true

  if (options.clearDisplayTreeFirst && isStillCurrentProject(currentProjectId, normalizedProjectPath)) {
    useWikiStore.getState().setFileTree([], { syncPathIndex: false })
  }

  if (refreshDisplayTree) {
    try {
      const tree = await listDirectoryWithRetry(normalizedProjectPath, {}, 3)
      if (isStillCurrentProject(currentProjectId, normalizedProjectPath)) {
        useWikiStore.getState().setFileTree(tree, { syncPathIndex: refreshPathIndex })
      }
    } catch (err) {
      console.error("Failed to refresh project file tree:", err)
    }
  } else if (refreshPathIndex) {
    if (!isStillCurrentProject(currentProjectId, normalizedProjectPath)) return
    listDirectoryWithRetry(normalizedProjectPath, {}, 3)
      .then((fullTree) => {
        if (!isStillCurrentProject(currentProjectId, normalizedProjectPath)) return
        useWikiStore.getState().setProjectPathIndexFromTree(fullTree)
      })
      .catch((err) => {
        console.error("Failed to refresh project path index:", err)
      })
  }

  if (options.bumpDataVersion && isStillCurrentProject(currentProjectId, normalizedProjectPath)) {
    useWikiStore.getState().bumpDataVersion()
  }
}