Function getDatabaseFunctions

test/unit/persisters/common/databases.ts:486–581

(
  cmd: (
    db: Database,
    sql: string,
    args?: any[],
  ) => Promise<{[id: string]: any}[]>,
  isPostgres = false,
  jsonValues = false,
)

};

export const getDatabaseFunctions = <Database>(
  cmd: (
    db: Database,
    sql: string,
    args?: any[],
  ) => Promise<{[id: string]: any}[]>,
  isPostgres = false,
  jsonValues = false,
): [
  (db: Database) => Promise<DumpOut>,
  (db: Database, dump: DumpIn) => Promise<void>,
] => {
  const getDatabase = async (db: Database): Promise<DumpOut> => {
    const dump: DumpOut = {};
    (
      await cmd(
        db,
        isPostgres
          ? 'SELECT table_name tn, column_name cn, data_type ty ' +
              'FROM information_schema.columns ' +
              `WHERE table_schema='public' ` +
              'AND table_name NOT LIKE $1 ' +
              'AND table_name NOT LIKE $2'
          : 'SELECT t.name tn, c.name cn, LOWER(c.type) ty ' +
              'FROM pragma_table_list() t, ' +
              'pragma_table_info(t.name) c ' +
              `WHERE t.schema='main' AND t.type = 'table' ` +
              'AND t.name NOT LIKE $1 ' +
              'AND t.name NOT LIKE $2',
        ['%sql%', '%electric%'],
      )
    ).forEach(({tn, cn, ty}) => {
      if (!dump[tn]) {
        dump[tn] = [{}, [{}]];
      }
      dump[tn][0][cn] = ty;
    });
    await Promise.all(
      Object.keys(dump).map(async (tn) => {
        const rows = await cmd(
          db,
          'SELECT * FROM ' + escapeId(tn) + ' ORDER BY 1',
        );
        rows.forEach((row) => {
          Object.entries(row).forEach(([column, value], index) => {
            if (index == 0 || !jsonValues) {
              row[column] = value;
            } else {
              row[column] = JSON.parse(value);
            }
          });
        });
        dump[tn][1] = [...rows];
      }),
    );
    return dump;
  };