node-ldapjs/bin/ldapjs-add

156 lines
2.9 KiB
JavaScript
Executable File

#!/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);
});
});