import {
	AgentProviderModelsQueryDto,
	isAgentModelProvider,
	type AgentProviderModelsResponse,
	type ChatIntegrationDescriptor,
} from '@n8n/api-types';
import type { AuthenticatedRequest } from '@n8n/db';
import { Get, Param, ProjectScope, Query, RestController } from '@n8n/decorators';
import type { Response } from 'express';

import { BadRequestError } from '@/errors/response-errors/bad-request.error';

import { AgentIntegrationPersistenceService } from './agent-integration-persistence.service';
import { AgentModelCatalogService } from './agent-model-catalog.service';
import { filterOfferedAgentModelProviders } from './model-catalog';

@RestController('/projects/:projectId/agents/v2')
export class AgentsCatalogController {
	constructor(
		private readonly agentIntegrationPersistenceService: AgentIntegrationPersistenceService,
		private readonly agentModelCatalogService: AgentModelCatalogService,
	) {}

	@Get('/catalog/models')
	@ProjectScope('agent:read')
	async getModelCatalog() {
		const { fetchProviderCatalog } = await import('@n8n/agents');
		return filterOfferedAgentModelProviders(await fetchProviderCatalog());
	}

	@Get('/catalog/models/:provider')
	@ProjectScope('agent:read')
	async getProviderModels(
		req: AuthenticatedRequest<{ projectId: string }>,
		_res: Response,
		@Param('provider') provider: string,
		@Query query: AgentProviderModelsQueryDto,
	): Promise<AgentProviderModelsResponse> {
		if (!isAgentModelProvider(provider)) {
			throw new BadRequestError(`Unknown model provider "${provider}"`);
		}
		return await this.agentModelCatalogService.getProviderModels(
			req.user,
			req.params.projectId,
			provider,
			query.credentialId,
		);
	}

	@Get('/catalog/integrations')
	@ProjectScope('agent:read')
	listIntegrations(): ChatIntegrationDescriptor[] {
		return this.agentIntegrationPersistenceService.listChatIntegrations();
	}
}