etherpad-lite/src/node/db/AuthorManager.js

/**
 * The AuthorManager controlls all information about the Pad authors
 */

/*
 * 2011 Peter 'Pita' Martischka (Primary Technology Ltd)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS-IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

var ERR = require("async-stacktrace");
var db = require("./DB").db;
var customError = require("../utils/customError");
var randomString = require('ep_etherpad-lite/static/js/pad_utils').randomString;
const thenify = require("thenify").withCallback;

exports.getColorPalette = function() {
  return [
    "#ffc7c7", "#fff1c7", "#e3ffc7", "#c7ffd5", "#c7ffff", "#c7d5ff", "#e3c7ff", "#ffc7f1",
    "#ffa8a8", "#ffe699", "#cfff9e", "#99ffb3", "#a3ffff", "#99b3ff", "#cc99ff", "#ff99e5",
    "#e7b1b1", "#e9dcAf", "#cde9af", "#bfedcc", "#b1e7e7", "#c3cdee", "#d2b8ea", "#eec3e6",
    "#e9cece", "#e7e0ca", "#d3e5c7", "#bce1c5", "#c1e2e2", "#c1c9e2", "#cfc1e2", "#e0bdd9",
    "#baded3", "#a0f8eb", "#b1e7e0", "#c3c8e4", "#cec5e2", "#b1d5e7", "#cda8f0", "#f0f0a8",
    "#f2f2a6", "#f5a8eb", "#c5f9a9", "#ececbb", "#e7c4bc", "#daf0b2", "#b0a0fd", "#bce2e7",
    "#cce2bb", "#ec9afe", "#edabbd", "#aeaeea", "#c4e7b1", "#d722bb", "#f3a5e7", "#ffa8a8",
    "#d8c0c5", "#eaaedd", "#adc6eb", "#bedad1", "#dee9af", "#e9afc2", "#f8d2a0", "#b3b3e6"
  ];
};

/**
 * Checks if the author exists
 */
exports.doesAuthorExists = thenify(function(authorID, callback)
{
  // check if the database entry of this author exists
  db.get("globalAuthor:" + authorID, function(err, author) {
    if (ERR(err, callback)) return;

    callback(null, author !== null);
  });
});

/**
 * Returns the AuthorID for a token.
 * @param {String} token The token
 * @param {Function} callback callback (err, author)
 */
exports.getAuthor4Token = thenify(function(token, callback)
{
  mapAuthorWithDBKey("token2author", token, function(err, author) {
    if (ERR(err, callback)) return;

    // return only the sub value authorID
    callback(null, author ? author.authorID : author);
  });
});

/**
 * Returns the AuthorID for a mapper.
 * @param {String} token The mapper
 * @param {String} name The name of the author (optional)
 * @param {Function} callback callback (err, author)
 */
exports.createAuthorIfNotExistsFor = thenify(function(authorMapper, name, callback)
{
  mapAuthorWithDBKey("mapper2author", authorMapper, function(err, author) {
    if (ERR(err, callback)) return;

    if (name) {
      // set the name of this author
      exports.setAuthorName(author.authorID, name);
    }

    // return the authorID
    callback(null, author);
  });
});

/**
 * Returns the AuthorID for a mapper. We can map using a mapperkey,
 * so far this is token2author and mapper2author
 * @param {String} mapperkey The database key name for this mapper
 * @param {String} mapper The mapper
 * @param {Function} callback callback (err, author)
 */
function mapAuthorWithDBKey (mapperkey, mapper, callback)
{
  // try to map to an author
  db.get(mapperkey + ":" + mapper, function(err, author) {
    if (ERR(err, callback)) return;

    if (author === null) {
      // there is no author with this mapper, so create one
      exports.createAuthor(null, function(err, author) {
        if (ERR(err, callback)) return;

        // create the token2author relation
        db.set(mapperkey + ":" + mapper, author.authorID);

        // return the author
        callback(null, author);
      });

      return;
    }

    // there is an author with this mapper
    // update the timestamp of this author
    db.setSub("globalAuthor:" + author, ["timestamp"], Date.now());

    // return the author
    callback(null, {authorID: author});
  });
}

/**
 * Internal function that creates the database entry for an author
 * @param {String} name The name of the author
 */
exports.createAuthor = thenify(function(name, callback)
{
  // create the new author name
  var author = "a." + randomString(16);

  // create the globalAuthors db entry
  var authorObj = {
    "colorId": Math.floor(Math.random() * (exports.getColorPalette().length)),
    "name": name,
    "timestamp": Date.now()
  };

  // set the global author db entry
  db.set("globalAuthor:" + author, authorObj);

  callback(null, {authorID: author});
});

/**
 * Returns the Author Obj of the author
 * @param {String} author The id of the author
 * @param {Function} callback callback(err, authorObj)
 */
exports.getAuthor = thenify(function(author, callback)
{
  db.get("globalAuthor:" + author, callback);
});

/**
 * Returns the color Id of the author
 * @param {String} author The id of the author
 * @param {Function} callback callback(err, colorId)
 */
exports.getAuthorColorId = thenify(function(author, callback)
{
  db.getSub("globalAuthor:" + author, ["colorId"], callback);
});

/**
 * Sets the color Id of the author
 * @param {String} author The id of the author
 * @param {String} colorId The color id of the author
 * @param {Function} callback (optional)
 */
exports.setAuthorColorId = thenify(function(author, colorId, callback)
{
  db.setSub("globalAuthor:" + author, ["colorId"], colorId, callback);
});

/**
 * Returns the name of the author
 * @param {String} author The id of the author
 * @param {Function} callback callback(err, name)
 */
exports.getAuthorName = thenify(function(author, callback)
{
  db.getSub("globalAuthor:" + author, ["name"], callback);
});

/**
 * Sets the name of the author
 * @param {String} author The id of the author
 * @param {String} name The name of the author
 * @param {Function} callback (optional)
 */
exports.setAuthorName = thenify(function(author, name, callback)
{
  db.setSub("globalAuthor:" + author, ["name"], name, callback);
});

/**
 * Returns an array of all pads this author contributed to
 * @param {String} author The id of the author
 * @param {Function} callback (optional)
 */
exports.listPadsOfAuthor = thenify(function(authorID, callback)
{
  /* There are two other places where this array is manipulated:
   * (1) When the author is added to a pad, the author object is also updated
   * (2) When a pad is deleted, each author of that pad is also updated
   */

  // get the globalAuthor
  db.get("globalAuthor:" + authorID, function(err, author) {
    if (ERR(err, callback)) return;

    if (author === null) {
      // author does not exist
      callback(new customError("authorID does not exist", "apierror"));

      return;
    }

    // everything is fine, return the pad IDs
    var pads = [];

    if (author.padIDs != null) {
      for (var padId in author.padIDs) {
        pads.push(padId);
      }
    }

    callback(null, {padIDs: pads});
  });
});

/**
 * Adds a new pad to the list of contributions
 * @param {String} author The id of the author
 * @param {String} padID The id of the pad the author contributes to
 */
exports.addPad = function(authorID, padID)
{
  // get the entry
  db.get("globalAuthor:" + authorID, function(err, author) {
    if (ERR(err)) return;
    if (author === null) return;

    if (author.padIDs == null) {
      // the entry doesn't exist so far, let's create it
      author.padIDs = {};
    }

    // add the entry for this pad
    author.padIDs[padID] = 1; // anything, because value is not used

    // save the new element back
    db.set("globalAuthor:" + authorID, author);
  });
}

/**
 * Removes a pad from the list of contributions
 * @param {String} author The id of the author
 * @param {String} padID The id of the pad the author contributes to
 */
exports.removePad = function(authorID, padID)
{
  db.get("globalAuthor:" + authorID, function(err, author) {
    if (ERR(err)) return;
    if (author === null) return;

    if (author.padIDs !== null) {
      // remove pad from author
      delete author.padIDs[padID];
      db.set("globalAuthor:" + authorID, author);
    }
  });
}