import { defineStore } from 'pinia'
import { ref } from 'vue'
import { MQTTClient } from '@/utils/mqttClient'
import { useUserStore } from './user'
import { useOrderPositionStore } from '@/stores/modules/orderPosition.ts'

// ============================================================
// 类型声明
// ============================================================

/** 持仓单项数据结构（根据实际业务接口调整字段） */
export interface PositionItem {
  id: string
  code: string
  name?: string
  volume?: number
  direction?: number // 1-做多, 2-做空
  open_price?: number
  current_price?: number
  profit?: number
  take_profit?: number
  stop_loss?: number
  open_time?: string
  [key: string]: any
}

/** 用户信息最小化类型（用于类型断言） */
interface UserInfo {
  redis_key: string
  token: string
  account_id: string
}

/** Vite 环境变量扩展（如项目已有 env.d.ts 可忽略） */
declare const __VITE_MQTT_TRANSACTION_HOST__: string

// ============================================================
// Store 定义
// ============================================================

export const useTransactionStore = defineStore('transaction', () => {
  // ------------------ 依赖注入 ------------------
  const mqttClient = new MQTTClient()
  const userStore = useUserStore()
  const orderStore = useOrderPositionStore()

  // ------------------ 状态 ------------------
  /** 持仓列表 */
  const positionList = ref<PositionItem[]>([])

  // ------------------ 初始化 ------------------

  /**
   * 初始化行情 MQTT 连接
   * 建立连接 → 订阅主题 → 注册消息回调
   */
  const initMarketConditionsMQTT = async (): Promise<void> => {
    const { redis_key, token } = userStore.userInfo as UserInfo

    await mqttClient.connect(import.meta.env.VITE_MQTT_TRANSACTION_HOST as string, {
      username: `${redis_key}`,
      password: `${token}`,
    })

    await subscribeTheme()
  }

  /**
   * 订阅主题并注册消息处理器
   * 仅保留持仓列表推送（POSITION_FLOATING_BACK）
   */
  const subscribeTheme = async (): Promise<void> => {
    const { id } = userStore.userInfo as UserInfo

    if (mqttClient.isConnected) {
      const topic = `push/${id}`
      await mqttClient.subscribe([topic])

      mqttClient.onTopicMessages({
        [topic]: (_topic: string, msg: string): void => {
          console.log(_topic, msg)

          const [type, data] = msg.split('|')
          switch (type) {
            case 'POSITION_FLOATING_BACK': {
              // 持仓列表回调：交由 orderStore 处理
              orderStore.handlePositionOrderList(data)
              break
            }
            default:
              break
          }
        },
      })
    } else {
      setTimeout(subscribeTheme, 30)
    }
  }

  // ------------------ 持仓列表 ------------------

  /**
   * 向服务端请求持仓列表
   * 服务端通过 POSITION_FLOATING_BACK 推送结果
   */
  const getPositionList = (account_id: string): void => {

    if (mqttClient.isConnected) {
      console.log('持仓 POSITION 持仓列表数据', `POSITION::${account_id}`)
      mqttClient.publish('server/trade', `POSITION::${account_id}`)
    } else {
      setTimeout(getPositionList, 30)
    }
  }

  /**
   * 关闭订阅，清空数组
   * */
  const disconnect = () => {
    mqttClient.disconnect()
    positionList.value = []
  }

  // ------------------ 暴露 ------------------
  return {
    mqttClient,
    positionList,
    initMarketConditionsMQTT,
    getPositionList,
  }
})