ComfyUI_frontend/src/composables/nodePack/usePacksSelection.ts

import { type Ref, computed } from 'vue'

import type { components } from '@/types/comfyRegistryTypes'
import { useComfyManagerStore } from '@/workbench/extensions/manager/stores/comfyManagerStore'

type NodePack = components['schemas']['Node']

type SelectionState = 'all-installed' | 'none-installed' | 'mixed'

/**
 * Composable for managing multi-package selection states
 * Handles installation status tracking and selection state determination
 */
export function usePacksSelection(nodePacks: Ref<NodePack[]>) {
  const managerStore = useComfyManagerStore()

  const installedPacks = computed(() =>
    nodePacks.value.filter((pack) => managerStore.isPackInstalled(pack.id))
  )

  const notInstalledPacks = computed(() =>
    nodePacks.value.filter((pack) => !managerStore.isPackInstalled(pack.id))
  )

  const isAllInstalled = computed(
    () => installedPacks.value.length === nodePacks.value.length
  )

  const isNoneInstalled = computed(
    () => notInstalledPacks.value.length === nodePacks.value.length
  )

  const isMixed = computed(
    () => installedPacks.value.length > 0 && notInstalledPacks.value.length > 0
  )

  const selectionState = computed<SelectionState>(() => {
    if (isAllInstalled.value) return 'all-installed'
    if (isNoneInstalled.value) return 'none-installed'
    return 'mixed'
  })

  return {
    installedPacks,
    notInstalledPacks,
    isAllInstalled,
    isNoneInstalled,
    isMixed,
    selectionState
  }
}