import { defineStore } from 'pinia'
import { ref } from 'vue'

export const usePowerStationStore = defineStore('powerStation', () => {
  const currentNodeId = ref<string>('')
  const modelHierarchy = ref<any[]>([])
  const isModelLoading = ref(false)
  const loadMode = ref<boolean>(false) // false: 分批加载, true: 完整加载
  const modelCacheEnabled = ref<boolean>(false) // false: 不使用缓存, true: 使用缓存

  // 聚焦事件监听器
  const focusListeners = ref<((nodeId: string) => void)[]>([])

  // 操作方法 - 仅处理节点选择
  const selectNode = (node: any) => {
    currentNodeId.value = node.id
  }

  // 触发聚焦事件
  const triggerFocus = (nodeId: string) => {
    focusListeners.value.forEach(listener => listener(nodeId))
  }

  // 添加聚焦监听器
  const addFocusListener = (listener: (nodeId: string) => void) => {
    focusListeners.value.push(listener)
  }

  // 移除聚焦监听器
  const removeFocusListener = (listener: (nodeId: string) => void) => {
    const index = focusListeners.value.indexOf(listener)
    if (index > -1) {
      focusListeners.value.splice(index, 1)
    }
  }

  return {
    currentNodeId,
    modelHierarchy,
    isModelLoading,
    loadMode,
    modelCacheEnabled,
    selectNode,
    triggerFocus,
    addFocusListener,
    removeFocusListener,
  }
})