playwright/src/cli/codegen/outputs.ts

/**
 * Copyright (c) Microsoft Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import * as fs from 'fs';
import * as querystring from 'querystring';
import { Writable } from 'stream';
import { highlight } from 'highlight.js';
import { CodeGeneratorOutput } from './codeGenerator';

export class OutputMultiplexer implements CodeGeneratorOutput {
  private _outputs: CodeGeneratorOutput[]
  constructor(outputs: CodeGeneratorOutput[]) {
    this._outputs = outputs;
  }

  printLn(text: string) {
    for (const output of this._outputs)
      output.printLn(text);
  }

  popLn(text: string) {
    for (const output of this._outputs)
      output.popLn(text);
  }

  flush() {
    for (const output of this._outputs)
      output.flush();
  }
}

export class FileOutput implements CodeGeneratorOutput {
  private _fileName: string;
  private _lines: string[];
  constructor(fileName: string) {
    this._fileName = fileName;
    this._lines = [];
  }

  printLn(text: string) {
    this._lines.push(...text.trimEnd().split('\n'));
  }

  popLn(text: string) {
    this._lines.length -= text.trimEnd().split('\n').length;
  }

  flush() {
    fs.writeFileSync(this._fileName, this._lines.join('\n'));
  }
}

export class TerminalOutput implements CodeGeneratorOutput {
  private _output: Writable
  private _language: string;

  constructor(output: Writable, language: string) {
    this._output = output;
    this._language = language;
  }

  private _highlight(text: string) {
    let highlightedCode = highlight(this._language, text).value;
    highlightedCode = querystring.unescape(highlightedCode);
    highlightedCode = highlightedCode.replace(/<span class="hljs-keyword">/g, '\x1b[38;5;205m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-built_in">/g, '\x1b[38;5;220m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-literal">/g, '\x1b[38;5;159m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-title">/g, '');
    highlightedCode = highlightedCode.replace(/<span class="hljs-number">/g, '\x1b[38;5;78m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-string">/g, '\x1b[38;5;130m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-comment">/g, '\x1b[38;5;23m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-subst">/g, '\x1b[38;5;242m');
    highlightedCode = highlightedCode.replace(/<span class="hljs-function">/g, '');
    highlightedCode = highlightedCode.replace(/<span class="hljs-params">/g, '');
    highlightedCode = highlightedCode.replace(/<span class="hljs-attr">/g, '');
    highlightedCode = highlightedCode.replace(/<\/span>/g, '\x1b[0m');
    highlightedCode = highlightedCode.replace(/&#x27;/g, "'");
    highlightedCode = highlightedCode.replace(/&quot;/g, '"');
    highlightedCode = highlightedCode.replace(/&gt;/g, '>');
    highlightedCode = highlightedCode.replace(/&lt;/g, '<');
    highlightedCode = highlightedCode.replace(/&amp;/g, '&');
    return highlightedCode;
  }

  printLn(text: string) {
    // Split into lines for highlighter to not fail.
    for (const line of text.split('\n'))
      this._output.write(this._highlight(line) + '\n');
  }

  popLn(text: string) {
    const terminalWidth = process.stdout.columns || 80;
    for (const line of text.split('\n')) {
      const terminalLines = ((line.length - 1) / terminalWidth | 0) + 1;
      for (let i = 0; i < terminalLines; ++i)
        this._output.write('\u001B[1A\u001B[2K');
    }
  }

  flush() {}
}