import {
  Controller,
  Get,
  Inject,
  Param,
  Query,
  Request,
  UseGuards,
} from '@nestjs/common';
import { JwtAuthGuard } from '../../auth/guards/jwt.guard';
import { EntityRelationService } from '../service/entity-relation.service';

@UseGuards(JwtAuthGuard)
@Controller('entity-relation')
export class EntityRelationController {
  constructor(
    @Inject('EntityRelationService')
    private readonly entityRelationService: EntityRelationService,
  ) {}

  @Get(':source_entity_type')
  async getEntityRelations(
    @Param('source_entity_type') sourceEntityType: string,
    @Query('include') include: string, // <-- new param
    @Request() req: any,
  ): Promise<any> {
    const loggedInUser = req.user.userData;
    const shouldIncludeSelf = include === 'true';

    return this.entityRelationService.getEntityRelations(
      sourceEntityType,
      loggedInUser,
      shouldIncludeSelf,
    );
  }
}