tom: add support for ip/port

tombburnell · Sep 8, 2014 · cc2d636 · cc2d636
1 parent e8369af
commit cc2d636
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/bin/etcd-dump.js b/bin/etcd-dump.js
@@ -11,18 +11,21 @@ var prog = require('commander');
 // etcd-dump's package.json file
 var pkg = require('../package.json');
 
-// Dumper class
-var dumper = require('../')();
 
 // General options
 prog
 .version(pkg.version)
-.option('-f, --file [json_file]', 'Path to JSON dump file for dumping/storing', './etcd_dump.json');
+.option('-f, --file [json_file]', 'Path to JSON dump file for dumping/storing', './etcd_dump.json')
+.option('-h, --host [host]', 'Hostname to connect to', 'localhost')
+.option('-p, --port [port]', 'Port to connect to', '4001')
+;
 
 // Dump command
 prog
 .command('dump')
 .action(function() {
+    var dumper = require('../')(prog.host, prog.port);
+
     return dumper.dump()
     .then(function(data) {
         // Write file to disk
@@ -34,6 +37,7 @@ prog
 prog
 .command('restore')
 .action(function() {
+    var dumper = require('../')(prog.host, prog.port);
     var entries = JSON.parse(fs.readFileSync(prog.file));
 
     return dumper.restore(entries)

diff --git a/index.js b/index.js
@@ -22,9 +22,9 @@ function normalize(obj) {
 }
 
 
-function Dumper(etcd) {
+function Dumper(host,port) {
     // ETCD client
-    this.store = new Etcd();
+    this.store = new Etcd(host, port);
 
     _.bindAll(this);
 }
@@ -47,8 +47,8 @@ Dumper.prototype.restore = function(entries) {
 };
 
 // Restore the database from input data
-function createDumper() {
-    return new Dumper();
+function createDumper(host,port) {
+    return new Dumper(host,port);
 }
 
 // Exports