node-ldapjs/bin/ldapjs-add

#!/usr/bin/env node
// -*- mode: js -*-
// Copyright 2011 Mark Cavage.  All rights reserved.

var fs = require('fs');
var path = require('path');
var url = require('url');

var nopt = require('nopt');

var ldap = require('../lib/index');
var Logger = require('bunyan');


///--- Globals

nopt.typeDefs.DN = {
  type: ldap.DN,
  validate: function(data, k, val) {
    data[k] = ldap.parseDN(val);
  }
};

var opts = {
  'debug': Number,
  'binddn': ldap.DN,
  'file': String,
  'password': String,
  'url': url
};

var shortOpts = {
  'd': ['--debug'],
  'D': ['--binddn'],
  'f': ['--file'],
  'w': ['--password'],
  'u': ['--url']
};



///--- Helpers

function usage(code, message) {
  var _opts = '';
  Object.keys(shortOpts).forEach(function(k) {
    if (!Array.isArray(shortOpts[k]))
      return;
    var longOpt = shortOpts[k][0].replace('--', '');
    var type = opts[longOpt].name || 'string';
    if (type && type === 'boolean') type = '';
    type = type.toLowerCase();

    _opts += ' [--' + longOpt + ' ' + type + ']';
  });
  _opts += ' [JSON]';

  var msg = (message ? message + '\n' : '') +
    'usage: ' + path.basename(process.argv[1]) + _opts;

  process.stderr.write(msg + '\n');
  process.exit(code);
}


function perror(err) {
  if (parsed.debug) {
    process.stderr.write(err.stack + '\n');
  } else {
    process.stderr.write(err.message + '\n');
  }
  process.exit(1);
}



///--- Mainline

var logLevel = 'info';
var parsed;

try {
  parsed = nopt(opts, shortOpts, process.argv, 2);
  if (parsed.file) {
    parsed.file = JSON.parse(fs.readFileSync(parsed.file, 'utf8'));
    if (!Array.isArray(parsed.file))
      parsed.file = [parsed.file];
  }
} catch (e) {
  usage(1, e.toString());
}

if (parsed.help)
  usage(0);
if (!parsed.file) {
  parsed.file = [];
  parsed.argv.remain.forEach(function(a) {
    var o =JSON.parse(a);
    if (Array.isArray(o)) {
      o.forEach(function(i) {
        parsed.file.push(o);
      });
      return;
    }
    parsed.file.push(o);
  });
}

if (!parsed.file)
  usage(1, 'either -f or stdin must be used for adding objects');

if (parsed.debug)
  logLevel = (parsed.debug > 1 ? 'trace' : 'debug');
if (!parsed.url)
  parsed.url = 'ldap://127.0.0.1:389';
if (!parsed.binddn)
  parsed.binddn = '';
if (!parsed.password)
  parsed.password = '';

var log = new Logger({
    name: 'ldapjs',
    component: 'client',
    stream: process.stderr,
    level: logLevel
});

var client = ldap.createClient({
  url: parsed.url,
  log: log
});

client.on('error', function(err) {
  perror(err);
});

client.bind(parsed.binddn, parsed.password, function(err, res) {
  if (err)
    perror(err);

  var finished = 0;
  function callback(err) {
    if (err)
      perror(err);

    if (++finished === parsed.file.length)
      client.unbind(function () { return; });
  }

  parsed.file.forEach(function(entry) {
    var dn = entry.dn;
    delete entry.dn;
    client.add(dn, entry, callback);
  });
});