MoviePilot-Frontend/src/utils/federationLoader.ts

import api from '@/api'
import {
  __federation_method_setRemote,
  __federation_method_getRemote,
  __federation_method_unwrapDefault,
  // @ts-ignore
} from 'virtual:__federation__'

// 定义远程模块接口
interface RemoteModule {
  id: string
  url: string
}

/**
 * 获取单个远程模块信息
 * @param id 远程模块ID
 */
async function fetchSingleRemoteModule(id: string): Promise<RemoteModule | null> {
  try {
    const modules = await fetchRemoteModules()
    return modules.find(module => module.id === id) || null
  } catch (error) {
    console.error(`获取远程模块信息失败: ${id}`, error)
    return null
  }
}

/**
 * 加载远程组件
 * @param id 远程模块ID
 * @param componentName 组件名称 (如 'Page')
 */
export async function loadRemoteComponent(id: string, componentName: string = 'Page') {
  try {
    const module = await __federation_method_getRemote(id, `./${componentName}`)
    return __federation_method_unwrapDefault(module)
  } catch (error) {
    // 组件未注册，尝试重新注册
    try {
      const moduleInfo = await fetchSingleRemoteModule(id)
      if (moduleInfo) {
        console.log(`组件未注册，正在重新注册: ${id}`)
        injectRemoteModule(moduleInfo)

        // 重新尝试加载组件
        const module = await __federation_method_getRemote(id, `./${componentName}`)
        return __federation_method_unwrapDefault(module)
      } else {
        console.error(`无法找到远程模块信息: ${id}`)
        throw new Error(`无法找到远程模块信息: ${id}`)
      }
    } catch (retryError) {
      console.error(`重新注册并加载组件失败: ${id}/${componentName}`, retryError)
      throw retryError
    }
  }
}

/**
 * 从API获取远程模块列表
 */
async function fetchRemoteModules(): Promise<RemoteModule[]> {
  try {
    const response = await api.get('plugin/remotes?token=moviepilot')
    return (response as any) || []
  } catch (error) {
    console.error('获取远程模块列表失败:', error)
    return []
  }
}

/**
 * 动态注入Federation Remote模块
 * @param modules 远程模块列表
 */
function injectRemoteModule(module: RemoteModule): void {
  // 从浏览器地址栏获取当前地址前缀
  const baseUrl = new URL(window.location.href)
  // 环境变量
  let apiBase = import.meta.env.VITE_API_BASE_URL
  if (apiBase.startsWith('/')) {
    apiBase = apiBase.slice(1)
  }
  if (apiBase.endsWith('/')) {
    apiBase = apiBase.slice(0, -1)
  }
  __federation_method_setRemote(module.id, {
    url: () => Promise.resolve(`${baseUrl.origin}/${apiBase}${module.url}`),
    format: 'esm',
    from: 'vite',
  })
  console.log('已注入远程模块:', module)
}

/**
 * 初始化并加载所有远程组件
 */
export async function loadRemoteComponents(): Promise<void> {
  try {
    // 获取远程模块列表
    const modules = await fetchRemoteModules()

    // 确保有模块才注入
    if (modules && modules.length > 0) {
      // 注入远程模块
      modules.forEach(module => {
        injectRemoteModule(module)
      })
    } else {
      console.log('没有发现可用的远程模块')
    }
  } catch (error) {
    console.error('加载远程组件失败:', error)
  }
}