ceo-api/src/segments/segments.service.ts

import { Injectable, NotFoundException } from '@nestjs/common';
import { and, desc, eq } from 'drizzle-orm';
import { db } from '../db/client';
import { savedSegments } from '../db/schema';
import { AuditService } from '../audit/audit.service';
import { IntelligenceService } from '../intelligence/intelligence.service';
import type { AuthenticatedSession } from '../auth/tenant.guard';
import type { CreateSegmentDto } from './dto';

@Injectable()
export class SegmentsService {
  constructor(
    private readonly audit: AuditService,
    private readonly intelligence: IntelligenceService,
  ) {}

  async list(session: AuthenticatedSession) {
    return db.query.savedSegments.findMany({
      where: eq(savedSegments.tenantId, session.tenantId),
      orderBy: desc(savedSegments.createdAt),
    });
  }

  async create(session: AuthenticatedSession, dto: CreateSegmentDto) {
    const [row] = await db
      .insert(savedSegments)
      .values({
        tenantId: session.tenantId,
        createdByUserId: session.userId,
        name: dto.name,
        query: dto.query,
        resultLimit: dto.resultLimit,
      })
      .returning();
    await this.audit.record(null, {
      tenantId: session.tenantId,
      actorId: session.userId,
      action: 'segment.created',
      resource: `segment:${row.id}`,
    });
    return row;
  }

  /** Re-ruleaza cautarea salvata contra intelligence-api si intoarce rezultate proaspete. */
  async run(session: AuthenticatedSession, id: string) {
    const segment = await this.getById(session, id);
    const results = await this.intelligence.searchCompanies(segment.query, segment.resultLimit);
    return { segment, ...results };
  }

  async remove(session: AuthenticatedSession, id: string) {
    const segment = await this.getById(session, id);
    await db.delete(savedSegments).where(eq(savedSegments.id, segment.id));
    await this.audit.record(null, {
      tenantId: session.tenantId,
      actorId: session.userId,
      action: 'segment.deleted',
      resource: `segment:${id}`,
    });
    return { deleted: true };
  }

  private async getById(session: AuthenticatedSession, id: string) {
    const segment = await db.query.savedSegments.findFirst({
      where: and(eq(savedSegments.id, id), eq(savedSegments.tenantId, session.tenantId)),
    });
    if (!segment) {
      throw new NotFoundException('Segment not found');
    }
    return segment;
  }
}