import { logger } from './logger'
import { closeDb } from '../db'
import { getGatewayManagerInstance } from './gateway-bootstrap'

function shouldStopGatewaysOnShutdown(signal: string): boolean {
  // 总是停止网关，无论是开发环境还是生产环境
  // 这样可以避免 nodemon 重启时的孤儿进程问题
  const override = process.env.HERMES_WEB_UI_STOP_GATEWAYS_ON_SHUTDOWN?.trim()
  if (override === '0' || override === 'false') return false
  if (override === '1' || override === 'true') return true

  return signal !== 'SIGUSR2'
}

export function bindShutdown(server: any, groupChatServer?: any, chatRunServer?: any): void {
  let isShuttingDown = false

  const shutdown = async (signal: string) => {
    if (isShuttingDown) return
    isShuttingDown = true

    // Force exit after 3s no matter what
    setTimeout(() => process.exit(0), 3000)

    logger.info('Shutting down (%s)...', signal)

    try {
      if (shouldStopGatewaysOnShutdown(signal)) {
        // Stop gateway processes owned by this Web UI instance first.
        try {
          const gatewayManager = getGatewayManagerInstance()
          if (gatewayManager) {
            await gatewayManager.stopAll()
            logger.info('All gateways stopped')
          }
        } catch (err) {
          logger.warn(err, 'Failed to stop gateways (non-fatal)')
        }
      } else {
        logger.info('Skipping gateway shutdown for %s', signal)
      }

      // Close ChatRunSocket first to abort all active runs and close EventSource connections
      if (chatRunServer) {
        chatRunServer.close()
        logger.info('ChatRunSocket closed')
      }

      // Disconnect Socket.IO before HTTP server to prevent hanging
      if (groupChatServer) {
        groupChatServer.agentClients.disconnectAll()
        groupChatServer.getIO().close()
        logger.info('Socket.IO closed')
      }

      const servers = Array.isArray(server) ? server : [server].filter(Boolean)
      if (servers.length) {
        await Promise.all(servers.map((httpServer) => (
          new Promise<void>((resolve) => {
            httpServer.close(() => {
              logger.info('HTTP server closed')
              resolve()
            })
          })
        )))
      }
    } catch (err) {
      logger.error(err, 'Shutdown error')
    }

    closeDb()
    process.exit(0)
  }

  process.once('SIGUSR2', shutdown)
  process.on('SIGINT', shutdown)
  process.on('SIGTERM', shutdown)
}