smartshell/ts/classes.shelllog.ts

import * as plugins from './plugins.js';

/**
 * a log handler for spawned logs
 * making sure the process doesn't run out of memory
 */
export class ShellLog {
  private chunks: Buffer[] = [];
  private totalLength = 0;

  /**
   * Get the accumulated log as a single Buffer.
   * Concatenation happens lazily only when accessed.
   */
  public get logStore(): Buffer {
    if (this.chunks.length === 0) {
      return Buffer.alloc(0);
    }
    if (this.chunks.length === 1) {
      return this.chunks[0];
    }
    // Flatten chunks into a single buffer
    const combined = Buffer.concat(this.chunks, this.totalLength);
    // Replace chunks array with the single combined buffer for future access
    this.chunks = [combined];
    return combined;
  }

  /**
   * Set the log store directly (used for truncation).
   */
  public set logStore(value: Buffer) {
    this.chunks = [value];
    this.totalLength = value.length;
  }

  /**
   * Get the current total length of buffered data without concatenating.
   */
  public get logLength(): number {
    return this.totalLength;
  }

  /**
   * log data to console
   * @param dataArg
   */
  public writeToConsole(dataArg: string | Buffer): void {
    // make sure we have the data as string
    process.stdout.write(dataArg);
  }

  /**
   * add data to Buffer for later consumption
   * @param dataArg
   */
  public addToBuffer(dataArg: string | Buffer): void {
    // make sure we have the data as Buffer
    const dataBuffer: Buffer = Buffer.isBuffer(dataArg) ? dataArg : Buffer.from(dataArg);
    this.chunks.push(dataBuffer);
    this.totalLength += dataBuffer.length;
  }

  public logAndAdd(dataArg: string | Buffer): void {
    this.writeToConsole(dataArg);
    this.addToBuffer(dataArg);
  }
}