zed/crates/prettier/src/prettier_server.js

const { Buffer } = require("buffer");
const fs = require("fs");
const path = require("path");
const { once } = require("events");

const prettierContainerPath = process.argv[2];
if (prettierContainerPath == null || prettierContainerPath.length == 0) {
  process.stderr.write(
    `Prettier path argument was not specified or empty.\nUsage: ${process.argv[0]} ${process.argv[1]} prettier/path\n`,
  );
  process.exit(1);
}
fs.stat(prettierContainerPath, (err, stats) => {
  if (err) {
    process.stderr.write(`Path '${prettierContainerPath}' does not exist\n`);
    process.exit(1);
  }

  if (!stats.isDirectory()) {
    process.stderr.write(`Path '${prettierContainerPath}' exists but is not a directory\n`);
    process.exit(1);
  }
});
const prettierPath = path.join(prettierContainerPath, "node_modules/prettier");

class Prettier {
  constructor(path, prettier, config) {
    this.path = path;
    this.prettier = prettier;
    this.config = config;
  }
}

(async () => {
  let prettier;
  let config;
  try {
    prettier = await loadPrettier(prettierPath);
    config = (await prettier.resolveConfig(prettierPath)) || {};
  } catch (e) {
    process.stderr.write(`Failed to load prettier: ${e}\n`);
    process.exit(1);
  }
  process.stderr.write(`Prettier at path '${prettierPath}' loaded successfully, config: ${JSON.stringify(config)}\n`);
  process.stdin.resume();
  handleBuffer(new Prettier(prettierPath, prettier, config));
})();

async function handleBuffer(prettier) {
  for await (const messageText of readStdin()) {
    let message;
    try {
      message = JSON.parse(messageText);
    } catch (e) {
      sendResponse(makeError(`Parse error in request message: ${e}\nMessage: ${messageText}`));
      continue;
    }
    // allow concurrent request handling by not `await`ing the message handling promise (async function)
    handleMessage(message, prettier).catch((e) => {
      if ((message.params || {}).text !== undefined) {
        message.params.text = "..snip..";
      }
      sendResponse({
        id: message.id,
        ...makeError(`${e}\nWhile handling prettier request: ${JSON.stringify(message)}`),
      });
    });
  }
}

const headerSeparator = "\r\n";
const contentLengthHeaderName = "Content-Length";

async function* readStdin() {
  let buffer = Buffer.alloc(0);
  let streamEnded = false;
  process.stdin.on("end", () => {
    streamEnded = true;
  });
  process.stdin.on("data", (data) => {
    buffer = Buffer.concat([buffer, data]);
  });

  async function handleStreamEnded(errorMessage) {
    sendResponse(makeError(errorMessage));
    buffer = Buffer.alloc(0);
    messageLength = null;
    await once(process.stdin, "readable");
    streamEnded = false;
  }

  try {
    let headersLength = null;
    let messageLength = null;
    main_loop: while (true) {
      if (messageLength === null) {
        while (buffer.indexOf(`${headerSeparator}${headerSeparator}`) === -1) {
          if (streamEnded) {
            await handleStreamEnded("Unexpected end of stream: headers not found");
            continue main_loop;
          } else if (buffer.length > contentLengthHeaderName.length * 10) {
            await handleStreamEnded(
              `Unexpected stream of bytes: no headers end found after ${buffer.length} bytes of input`,
            );
            continue main_loop;
          }
          await once(process.stdin, "readable");
        }
        const headers = buffer.subarray(0, buffer.indexOf(`${headerSeparator}${headerSeparator}`)).toString("ascii");
        const contentLengthHeader = headers
          .split(headerSeparator)
          .map((header) => header.split(":"))
          .filter((header) => header[2] === undefined)
          .filter((header) => (header[1] || "").length > 0)
          .find((header) => (header[0] || "").trim() === contentLengthHeaderName);
        const contentLength = (contentLengthHeader || [])[1];
        if (contentLength === undefined) {
          await handleStreamEnded(`Missing or incorrect ${contentLengthHeaderName} header: ${headers}`);
          continue main_loop;
        }
        headersLength = headers.length + headerSeparator.length * 2;
        messageLength = parseInt(contentLength, 10);
      }

      while (buffer.length < headersLength + messageLength) {
        if (streamEnded) {
          await handleStreamEnded(
            `Unexpected end of stream: buffer length ${buffer.length} does not match expected header length ${headersLength} + body length ${messageLength}`,
          );
          continue main_loop;
        }
        await once(process.stdin, "readable");
      }

      const messageEnd = headersLength + messageLength;
      const message = buffer.subarray(headersLength, messageEnd);
      buffer = buffer.subarray(messageEnd);
      headersLength = null;
      messageLength = null;
      yield message.toString("utf8");
    }
  } catch (e) {
    sendResponse(makeError(`Error reading stdin: ${e}`));
  } finally {
    process.stdin.off("data", () => {});
  }
}

async function handleMessage(message, prettier) {
  const { method, id, params } = message;
  if (method === undefined) {
    throw new Error(`Message method is undefined: ${JSON.stringify(message)}`);
  } else if (method == "initialized") {
    return;
  } else if (method === "shutdown") {
    sendResponse({ result: {} });
  } else if (method == "exit") {
    process.exit(0);
  }

  if (id === undefined) {
    throw new Error(`Message id is undefined: ${JSON.stringify(message)}`);
  }

  if (method === "prettier/format") {
    if (params === undefined || params.text === undefined) {
      throw new Error(`Message params.text is undefined: ${JSON.stringify(message)}`);
    }
    if (params.options === undefined) {
      throw new Error(`Message params.options is undefined: ${JSON.stringify(message)}`);
    }

    let resolvedConfig = {};
    if (params.options.filepath) {
      resolvedConfig = (await prettier.prettier.resolveConfig(params.options.filepath)) || {};

      if (params.options.ignorePath) {
        const fileInfo = await prettier.prettier.getFileInfo(params.options.filepath, {
          ignorePath: params.options.ignorePath,
        });
        if (fileInfo.ignored) {
          process.stderr.write(
            `Ignoring file '${params.options.filepath}' based on rules in '${params.options.ignorePath}'\n`,
          );
          sendResponse({ id, result: { text: params.text } });
          return;
        }
      }
    }

    // Marking the params.options.filepath as undefined makes
    // prettier.format() work even if no filepath is set.
    if (params.options.filepath === null) {
      params.options.filepath = undefined;
    }

    const plugins =
      Array.isArray(resolvedConfig?.plugins) && resolvedConfig.plugins.length > 0
        ? resolvedConfig.plugins
        : params.options.plugins;

    const rangeOptions = {};
    if (params.options.rangeStart != null) {
      rangeOptions.rangeStart = params.options.rangeStart;
    }
    if (params.options.rangeEnd != null) {
      rangeOptions.rangeEnd = params.options.rangeEnd;
    }

    const options = {
      ...(params.options.prettierOptions || prettier.config),
      ...resolvedConfig,
      plugins,
      parser: params.options.parser,
      filepath: params.options.filepath,
      ...rangeOptions
    };
    process.stderr.write(
      `Resolved config: ${JSON.stringify(resolvedConfig)}, will format file '${
        params.options.filepath || ""
      }' with options: ${JSON.stringify(options)}\n`,
    );
    const formattedText = await prettier.prettier.format(params.text, options);
    sendResponse({ id, result: { text: formattedText } });
  } else if (method === "prettier/clear_cache") {
    prettier.prettier.clearConfigCache();
    prettier.config = (await prettier.prettier.resolveConfig(prettier.path)) || {};
    sendResponse({ id, result: null });
  } else if (method === "initialize") {
    sendResponse({
      id,
      result: {
        capabilities: {},
      },
    });
  } else {
    throw new Error(`Unknown method: ${method}`);
  }
}

function makeError(message) {
  return {
    error: {
      code: -32600, // invalid request code
      message,
    },
  };
}

function sendResponse(response) {
  const responsePayloadString = JSON.stringify({
    jsonrpc: "2.0",
    ...response,
  });
  const headers = `${contentLengthHeaderName}: ${Buffer.byteLength(
    responsePayloadString,
  )}${headerSeparator}${headerSeparator}`;
  process.stdout.write(headers + responsePayloadString);
}

function loadPrettier(prettierPath) {
  return new Promise((resolve, reject) => {
    fs.access(prettierPath, fs.constants.F_OK, (err) => {
      if (err) {
        reject(`Path '${prettierPath}' does not exist.Error: ${err}`);
      } else {
        try {
          resolve(require(prettierPath));
        } catch (err) {
          reject(`Error requiring prettier module from path '${prettierPath}'.Error: ${err}`);
        }
      }
    });
  });
}