pg-to-dbml/src/funcs/getColumnDefault.js

CLI tool to scan your postgres database, and output DBML.

module.exports = function getColumnDefault(columnDefault, dataType) {
  if (!columnDefault || columnDefault.includes('::')) return '';
  const isFuncRegEx = /\(/;
  const isFunc = isFuncRegEx.test(columnDefault);

  if (isFunc) {
    return `default: \`${columnDefault}\``;
  }

  const useQuotes =
    ['varchar', 'character', 'char', 'text', 'timestamp'].findIndex(type => type === dataType) > -1;

  return useQuotes ? `default: '${columnDefault}'` : `default: ${columnDefault}`;
};