npmx.dev / app / composables / useFileTreeState.ts

export function useFileTreeState(baseUrl: string) {
  const expanded = useState<Set<string>>(`npmx-file-tree${baseUrl}`, () => new Set<string>())

  function toggleDir(path: string) {
    if (expanded.value.has(path)) {
      expanded.value.delete(path)
    } else {
      expanded.value.add(path)
    }
  }

  function isExpanded(path: string) {
    return expanded.value.has(path)
  }

  function autoExpandAncestors(path: string) {
    if (!path) return
    const parts = path.split('/').filter(Boolean)
    let prefix = ''
    for (const part of parts) {
      prefix = prefix ? `${prefix}/${part}` : part
      expanded.value.add(prefix)
    }
  }

  return {
    toggleDir,
    isExpanded,
    autoExpandAncestors,
  }
}