// Define the ChatStreamController class for writing messages to a readable
// stream which will be decoded by the react/ai Chat API. There does not seem
// to be functionality exported from the associated packages to do this so
// for now we do it manually after reverse engineering the protocol.

export interface ChatFileChange {
  filePath: string;
  contents: string;
}

export class ChatStreamController {
  private controller: ReadableStreamDefaultController;

  constructor(controller: ReadableStreamDefaultController) {
    this.controller = controller;
  }

  writeText(text: string) {
    this.controller.enqueue(`0:${JSON.stringify(text)}\n`);
  }

  writeFileChanges(title: string, fileChanges: ChatFileChange[]) {
    let text = `<boltArtifact title="${title}">`;
    for (const fileChange of fileChanges) {
      text += `<boltAction type="file" filePath="${fileChange.filePath}">${fileChange.contents}</boltAction>`;
    }
    text += "</boltArtifact>";
    this.writeText(text);
  }

  writeAnnotation(type: string, value: any) {
    this.controller.enqueue(`8:[{"type":"${type}","value":${JSON.stringify(value)}}]\n`);
  }
}