cherry-studio/src/main/apiServer/routes/agents/middleware/common.ts

import { Request, Response } from 'express'

import { agentService } from '../../../../services/agents'
import { loggerService } from '../../../../services/LoggerService'

const logger = loggerService.withContext('ApiServerMiddleware')

// Since Zod validators handle their own errors, this is now a pass-through
export const handleValidationErrors = (_req: Request, _res: Response, next: any): void => {
  next()
}

// Middleware to check if agent exists
export const checkAgentExists = async (req: Request, res: Response, next: any): Promise<void> => {
  try {
    const { agentId } = req.params
    const exists = await agentService.agentExists(agentId)

    if (!exists) {
      res.status(404).json({
        error: {
          message: 'Agent not found',
          type: 'not_found',
          code: 'agent_not_found'
        }
      })
      return
    }

    next()
  } catch (error) {
    logger.error('Error checking agent existence:', error as Error)
    res.status(500).json({
      error: {
        message: 'Failed to validate agent',
        type: 'internal_error',
        code: 'agent_validation_failed'
      }
    })
  }
}