# All files / src/db/repositories/drizzle conversation.ts

70% Statements 7/10
75% Branches 6/8
60% Functions 3/5
70% Lines 7/10

Press n or j to go to the next uncovered block, b, p or k for the previous block.

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63
    64
    65
    66
    67
    68

    84x

    4x
    4x
    4x

    4x

    5x

    5x

     /**
     * Drizzle Conversation Repository Adapter
     *
     * Implements ConversationRepository interface using Drizzle ORM.
     */

    import { eq, and, asc } from 'drizzle-orm';
    import { nanoid } from 'nanoid';
    import type { DrizzleDatabase } from '../../index.js';
    import { conversations, type Conversation } from '../../schema.js';
    import type { ConversationRepository, CreateConversationData } from '../conversation-repository.js';

    export class DrizzleConversationRepository implements ConversationRepository {
      constructor(private db: DrizzleDatabase) {}

      async create(data: CreateConversationData): Promise<Conversation> {
        const now = new Date();
        const id = nanoid();
        await this.db.insert(conversations).values({
          id,
          fromAgentId: data.fromAgentId,
          toAgentId: data.toAgentId,
          initiativeId: data.initiativeId ?? null,
          phaseId: data.phaseId ?? null,
          taskId: data.taskId ?? null,
          question: data.question,
          status: 'pending',
          createdAt: now,
          updatedAt: now,
        });
        return this.findById(id) as Promise<Conversation>;
      }

      async findById(id: string): Promise<Conversation | null> {
        const rows = await this.db
          .select()
          .from(conversations)
          .where(eq(conversations.id, id))
          .limit(1);
        return rows[0] ?? null;
      }

      async findPendingForAgent(toAgentId: string): Promise<Conversation[]> {
        return this.db
          .select()
          .from(conversations)
          .where(
            and(
              eq(conversations.toAgentId, toAgentId),
              eq(conversations.status, 'pending' as 'pending' | 'answered'),
            ),
          )
          .orderBy(asc(conversations.createdAt));
      }

      async answer(id: string, answer: string): Promise<Conversation | null> {
        await this.db
          .update(conversations)
          .set({
            answer,
            status: 'answered' as 'pending' | 'answered',
            updatedAt: new Date(),
          })
          .where(eq(conversations.id, id));
        return this.findById(id);
      }
    }