Function promptForQuery

src/commands/dataconnect-sql-shell.ts:39–64

()

];

async function promptForQuery(): Promise<string> {
  let query = "";
  const line = "";

  do {
    let line = await input({
      message: query ? "> " : "Enter your SQL query (or '.exit'):",
      transformer: (input: string) => {
        // Highlight SQL keywords
        return input
          .split(" ")
          .map((word) => (sqlKeywords.includes(word.toUpperCase()) ? clc.cyan(word) : word))
          .join(" ");
      },
      nonInteractive: false,
    });
    line = line.trimEnd();

    if (line.toLowerCase() === ".exit") {
      return ".exit";
    }

    query += (query ? "\n" : "") + line;
  } while (line !== "" && !query.endsWith(";"));
  return query;
}

async function mainShellLoop(conn: pg.PoolClient) {
  while (true) {