feature: create roles and tables from endpoints

Author: soedirgo

src/api/roles.ts

@@ -36,6 +36,43 @@ FROM
     res.status(500).json({ error: 'Database error', status: 500 })
   }
 })
+router.post('/', async (req, res) => {
+  try {
+    const {
+      name,
+      is_super_user = false,
+      has_create_db_privileges = false,
+      has_replication_privileges = false,
+      can_bypass_rls = false,
+      connections = -1,
+      valid_until,
+    } = req.body as {
+      name: string
+      is_super_user?: boolean
+      has_create_db_privileges?: boolean
+      has_replication_privileges?: boolean
+      can_bypass_rls?: boolean
+      connections?: number
+      valid_until?: string
+    }
+    const sql = `
+CREATE ROLE ${name}
+WITH
+  ${is_super_user ? 'SUPERUSER' : 'NOSUPERUSER'}
+  ${has_create_db_privileges ? 'CREATEDB' : 'NOCREATEDB'}
+  ${has_replication_privileges ? 'REPLICATION' : 'NOREPLICATION'}
+  ${can_bypass_rls ? 'BYPASSRLS' : 'NOBYPASSRLS'}
+  CONNECTION LIMIT ${connections}
+  ${valid_until === undefined ? '' : `VALID UNTIL '${valid_until}'`}`
+    console.log(sql)
+    const { data } = await RunQuery(req.headers.pg, sql)
+    return res.status(200).json(data)
+  } catch (error) {
+    // For this one, we always want to give back the error to the customer
+    console.log('Soft error!', error)
+    res.status(200).json([{ error: error.toString() }])
+  }
+})
 
 const removeSystemSchemas = (data: Roles.Role[]) => {
   return data.map((role) => {

src/api/tables.ts

@@ -2,7 +2,7 @@ import { Router } from 'express'
 
 import sql = require('../lib/sql')
 const { columns, grants, primary_keys, relationships, tables } = sql
-import { coalesceRowsToArray } from '../lib/helpers'
+import { coalesceRowsToArray, formatColumns } from '../lib/helpers'
 import { RunQuery } from '../lib/connectionPool'
 import { DEFAULT_SYSTEM_SCHEMAS } from '../lib/constants/schemas'
 import { Tables } from '../lib/interfaces'
@@ -49,6 +49,26 @@ FROM
     res.status(500).json({ error: 'Database error', status: 500 })
   }
 })
+router.post('/', async (req, res) => {
+  try {
+    const { schema = 'public', name, columns, primary_keys = [] } = req.body as {
+      schema?: string
+      name: string
+      columns: Tables.Column[]
+      primary_keys?: Tables.PrimaryKey[]
+    }
+    const sql = `
+CREATE TABLE ${schema}.${name} (
+  ${formatColumns({ columns, primary_keys })}
+)`
+    const { data } = await RunQuery(req.headers.pg, sql)
+    return res.status(200).json(data)
+  } catch (error) {
+    // For this one, we always want to give back the error to the customer
+    console.log('Soft error!', error)
+    res.status(200).json([{ error: error.toString() }])
+  }
+})
 
 export = router
 

src/lib/connectionPool.ts

@@ -5,7 +5,7 @@ import pg = require('pg')
 pg.types.setTypeParser(20, 'text', parseInt)
 const { Pool } = pg
 
-export const RunQuery = async (connectionString, sql) => {
+export const RunQuery = async (connectionString: any, sql: string) => {
   const pool = new Pool({ connectionString })
   try {
     const results = await pool.query(sql)

src/lib/helpers.ts

@@ -1,3 +1,5 @@
+import { Tables } from '../lib/interfaces'
+
 export const coalesceRowsToArray = (source: string, joinQuery: string) => {
   return `
 COALESCE(
@@ -10,3 +12,36 @@ COALESCE(
   '[]'
 ) AS ${source}`
 }
+
+export const formatColumns = ({
+  columns,
+  primary_keys,
+}: {
+  columns: Tables.Column[]
+  primary_keys: Tables.PrimaryKey[]
+}) => {
+  const pkey_columns = primary_keys.map((primary_key) => primary_key.name)
+  return columns
+    .map((column) => {
+      const {
+        name,
+        default_value,
+        is_identity = false,
+        is_nullable = true,
+        data_type,
+      } = column as {
+        name: string
+        default_value?: string
+        is_identity?: boolean
+        is_nullable?: boolean
+        data_type: string
+      }
+      return `
+${name} ${data_type}
+${default_value === undefined ? '' : `DEFAULT ${default_value}`}
+${is_identity ? 'GENERATED BY DEFAULT AS IDENTITY' : ''}
+${is_nullable ? '' : 'NOT NULL'}
+${pkey_columns.includes(name) ? 'PRIMARY KEY' : ''}`
+    })
+    .join(',')
+}