api-server/src/modules/ai-runtime/internal/runtime-internal.controller.ts

import { Controller, Get, Post, Param, Body, Req, UseGuards, HttpCode, HttpStatus } from '@nestjs/common';
import { InternalAuthGuard } from '../../../common/guards/internal-auth.guard';
import { RuntimeInternalService } from './runtime-internal.service';
import {
  RuntimePollJobsRequestDto, RuntimeLockJobRequestDto, RuntimeHeartbeatRequestDto,
  RuntimeResolveCredentialRequestDto, RuntimeSubmitResultRequestDto,
  RuntimeSubmitFailureRequestDto, RuntimeSubmitInvocationLogsRequestDto,
} from './dto/runtime-internal.dto';

@Controller('internal/runtime')
@UseGuards(InternalAuthGuard)
export class RuntimeInternalController {
  constructor(private readonly service: RuntimeInternalService) {}

  private instanceId(req: any): string {
    return (req.headers['x-runtime-instance-id'] as string) || 'unknown';
  }

  // ── Poll ──

  @Post('jobs/poll')
  async pollJobs(@Req() req: any, @Body() dto: RuntimePollJobsRequestDto) {
    return this.service.pollJobs(this.instanceId(req), dto.supportedJobTypes, dto.limit ?? 5, dto.capabilities);
  }

  // ── Lock ──

  @Post('jobs/:jobId/lock')
  async lockJob(@Req() req: any, @Param('jobId') jobId: string, @Body() dto: RuntimeLockJobRequestDto) {
    return this.service.lockJob(jobId, dto.runtimeInstanceId || this.instanceId(req));
  }

  // ── Heartbeat ──

  @Post('jobs/:jobId/heartbeat')
  async heartbeatJob(@Req() req: any, @Param('jobId') jobId: string, @Body() dto: RuntimeHeartbeatRequestDto) {
    return this.service.heartbeatJob(jobId, dto.runtimeInstanceId || this.instanceId(req));
  }

  // ── Snapshot ──

  @Get('jobs/:jobId/snapshot')
  async getSnapshot(@Param('jobId') jobId: string) {
    return this.service.getSnapshot(jobId);
  }

  // ── Credential Resolve ──

  @Post('model-credentials/resolve')
  async resolveCredential(@Body() dto: RuntimeResolveCredentialRequestDto) {
    return this.service.resolveCredential(dto.jobId, dto.apiKeyMode, dto.provider, dto.credentialId);
  }

  // ── Submit Result ──

  @Post('jobs/:jobId/result')
  @HttpCode(HttpStatus.CREATED)
  async submitResult(@Param('jobId') jobId: string, @Body() dto: RuntimeSubmitResultRequestDto) {
    return this.service.submitResult(jobId, dto);
  }

  // ── Submit Failure ──

  @Post('jobs/:jobId/fail')
  async submitFailure(@Param('jobId') jobId: string, @Body() dto: RuntimeSubmitFailureRequestDto) {
    return this.service.submitFailure(jobId, dto);
  }

  // ── Submit Invocation Logs ──

  @Post('invocation-logs')
  @HttpCode(HttpStatus.CREATED)
  async submitInvocationLogs(@Body() dto: RuntimeSubmitInvocationLogsRequestDto) {
    return this.service.submitInvocationLogs(dto.logs);
  }
}