arangodb/js/apps/system/_system/users/APP/storage.js

/*global applicationContext */
'use strict';
const _ = require('underscore');
const joi = require('joi');
const arangodb = require('org/arangodb');
const db = arangodb.db;
const Foxx = require('org/arangodb/foxx');
const errors = require('./errors');
const refreshUserCache = require('org/arangodb/users').reload;
const User = Foxx.Model.extend({
  schema: {
    user: joi.string().required(),
    authData: joi.object().required(),
    userData: joi.object().required()
  }
});
const users = new Foxx.Repository(
  db._collection('_users'),
  {model: User}
);

function resolve(username) {
  const user = users.firstExample({user: username});
  if (!user.get('_key')) {
    return null;
  }
  return user;
}

function listUsers() {
  return users.collection.all().toArray().map(function (user) {
    return user.user;
  }).filter(Boolean);
}

function createUser(username, userData, authData) {
  if (!userData) {
    userData = {};
  }
  if (!authData) {
    authData = {};
  }
  if (
    applicationContext.mount.indexOf('/_system/') === 0
      && !authData.hasOwnProperty('active')
  ) {
    authData.active = true;
  }

  if (!username) {
    throw new Error('Must provide username!');
  }
  let user;
  db._executeTransaction({
    collections: {
      read: [users.collection.name()],
      write: [users.collection.name()]
    },
    action: function () {
      if (resolve(username)) {
        throw new errors.UsernameNotAvailable(username);
      }
      user = new User({
        user: username,
        userData: userData,
        authData: authData
      });
      users.save(user);
    }
  });
  refreshUserCache();
  return user;
}

function getUser(uid) {
  let user;
  try {
    user = users.byId(uid);
  } catch (err) {
    if (
      err instanceof arangodb.ArangoError
        && err.errorNum === arangodb.ERROR_ARANGO_DOCUMENT_NOT_FOUND
    ) {
      throw new errors.UserNotFound(uid);
    }
    throw err;
  }
  return user;
}

function deleteUser(uid) {
  try {
    users.removeById(uid);
  } catch (err) {
    if (
      err instanceof arangodb.ArangoError
        && err.errorNum === arangodb.ERROR_ARANGO_DOCUMENT_NOT_FOUND
    ) {
      throw new errors.UserNotFound(uid);
    }
    throw err;
  }
  refreshUserCache();
  return null;
}

_.extend(User.prototype, {
  save: function () {
    const user = this;
    users.replace(user);
    refreshUserCache();
    return user;
  },
  delete: function () {
    try {
      deleteUser(this.get('_key'));
    } catch (e) {
      if (e instanceof errors.UserNotFound) {
        return false;
      }
      throw e;
    }
    refreshUserCache();
    return true;
  }
});

exports.resolve = resolve;
exports.list = listUsers;
exports.create = createUser;
exports.get = getUser;
exports.delete = deleteUser;
exports.errors = errors;
exports.repository = users;