teams/teams-db.js

The team application allows teams to be created and maintained. When teams are created or modified, the teams application will generate events.

'use strict';
var Pool = require('pg').Pool;
var lib = require('http-helper-functions');
var pge = require('pg-event-producer');

var config = {
  host: process.env.PG_HOST,
  user: process.env.PG_USER,
  password: process.env.PG_PASSWORD,
  database: process.env.PG_DATABASE
};

var pool = new Pool(config);
var eventProducer = new pge.eventProducer(pool);

function createTeamThen(req, res, id, selfURL, team, callback) {
  lib.internalizeURLs(team, req.headers.host);
  var query = `INSERT INTO teams (id, data) values('${id}', '${JSON.stringify(team)}') RETURNING etag`;
  function eventData(pgResult) {
    return {id: selfURL, action: 'create', etag: pgResult.rows[0].etag, team: team}
  }
  pge.queryAndStoreEvent(req, res, pool, query, 'teams', eventData, eventProducer, function(pgResult, pgEventResult) {
    callback(pgResult.rows[0].etag);
  });
}

function withTeamDo(req, res, id, callback) {
  lib.ifAllowedThen(req, res, '_resource', 'read', function() {
    pool.query('SELECT etag, data FROM teams WHERE id = $1', [id], function (err, pg_res) {
      if (err) {
        lib.internalError(res, err);
      }
      else {
        if (pg_res.rowCount === 0) { 
          lib.notFound(req, res);
        }
        else {
          var row = pg_res.rows[0];
          callback(row.data, row.etag);
        }
      }
    });
  });
}

function withTeamsForUserDo(req, res, user, callback) {
  var query = "SELECT id FROM teams, jsonb_array_elements(teams.data->'members') AS member WHERE member = $1"
  pool.query(query, [JSON.stringify(user)], function (err, pg_res) {
    if (err) {
      lib.internalError(res, err);
    }
    else {
      callback(pg_res.rows.map(row => row.id));
    }
  });
}
    
function deleteTeamThen(req, res, id, callback) {
  var query = `DELETE FROM teams WHERE id = '${id}' RETURNING *`;
  function eventData(pgResult) {
    return {id: id, action: 'delete', etag: pgResult.rows[0].etag, team: pgResult.rows[0].data}
  }
  pge.queryAndStoreEvent(req, res, pool, query, 'teams', eventData, eventProducer, function(pgResult, pgEventResult) {
    callback(pgResult.rows[0].data, pgResult.rows[0].etag);
  });

}

function updateTeamThen(req, res, id, team, patchedTeam, etag, callback) {
  lib.internalizeURLs(patchedTeam, req.headers.host);
  var key = lib.internalizeURL(id, req.headers.host);
  var query = `UPDATE teams SET data = ('${JSON.stringify(patchedTeam)}') WHERE subject = '${key}' AND etag = ${etag} RETURNING etag`;
  function eventData(pgResult) {
    return {id: id, action: 'update', etag: pgResult.rows[0].etag, before: team, after: patchedTeam}
  }
  pge.queryAndStoreEvent(req, res, pool, query, 'teams', eventData, eventProducer, function(pgResult, pgEventResult) {
    callback(pgResult.rows[0].etag);
  });
}

function init(callback) {
  var query = 'CREATE TABLE IF NOT EXISTS teams (id text primary key, etag serial, data jsonb);'
  pool.query(query, function(err, pgResult) {
    if(err) {
      console.error('error creating permissions table', err);
    } else {
      console.log(`connected to PG at ${config.host}`);
      eventProducer.init(callback);
    }
  });    
}

process.on('unhandledRejection', function(e) {
  console.log(e.message, e.stack)
})

exports.createTeamThen = createTeamThen;
exports.updateTeamThen = updateTeamThen;
exports.deleteTeamThen = deleteTeamThen;
exports.withTeamDo = withTeamDo;
exports.withTeamsForUserDo = withTeamsForUserDo;
exports.init = init;