enigma-bbs/core/user_group.js

/* jslint node: true */
'use strict';

const userDb		= require('./database.js').dbs.user;

const async			= require('async');
const _				= require('lodash');

exports.getGroupsForUser	 	= getGroupsForUser;
exports.addUserToGroup			= addUserToGroup;
exports.addUserToGroups			= addUserToGroups;
exports.removeUserFromGroup		= removeUserFromGroup;

function getGroupsForUser(userId, cb) {
    const sql =
		`SELECT group_name
		FROM user_group_member
		WHERE user_id=?;`;

    const groups = [];

    userDb.each(sql, [ userId ], (err, row) => {
        if(err) {
            return cb(err);
        }

        groups.push(row.group_name);
    },
    () => {
        return cb(null, groups);
    });
}

function addUserToGroup(userId, groupName, transOrDb, cb) {
    if(!_.isFunction(cb) && _.isFunction(transOrDb)) {
        cb = transOrDb;
        transOrDb = userDb;
    }

    transOrDb.run(
        `REPLACE INTO user_group_member (group_name, user_id)
		VALUES(?, ?);`,
        [ groupName, userId ],
        err => {
            return cb(err);
        }
    );
}

function addUserToGroups(userId, groups, transOrDb, cb) {

    async.each(groups, (groupName, nextGroupName) => {
        return addUserToGroup(userId, groupName, transOrDb, nextGroupName);
    }, err => {
        return cb(err);
    });
}

function removeUserFromGroup(userId, groupName, cb) {
    userDb.run(
        `DELETE FROM user_group_member
		WHERE group_name=? AND user_id=?;`,
        [ groupName, userId ],
        err => {
            return cb(err);
        }
    );
}