allow setting custom netrc in query and document defines

We need to perform some actions as the replication user for plain couch.

4 files changed, 24 insertions, 13 deletions

diff --git a/files/couch-doc-update b/files/couch-doc-update
index 5ceed5e..fea6456 100644
--- a/files/couch-doc-update
+++ b/files/couch-doc-update
@@ -62,6 +62,7 @@ def process_options
   doc_id     = nil
   new_data   = nil
   filename   = nil
+  netrc_file = nil
   loop do
     case ARGV[0]
       when '--host' then ARGV.shift; host     = ARGV.shift
@@ -69,6 +70,7 @@ def process_options
       when '--id'   then ARGV.shift; doc_id   = ARGV.shift
       when '--data' then ARGV.shift; new_data = ARGV.shift
       when '--file' then ARGV.shift; filename = ARGV.shift
+      when '--netrc-file' then ARGV.shift; netrc_file = ARGV.shift
       when /^-/     then usage("Unknown option: #{ARGV[0].inspect}")
       else break
     end
@@ -76,7 +78,7 @@ def process_options
   usage("Missing required option") unless db_name && doc_id && new_data
   new_data = MultiJson.load(new_data)
   new_data.merge!(read_file(filename)) if filename
-  db  = CouchRest.database(connection_string(db_name, host))
+  db  = CouchRest.database(connection_string(db_name, host, netrc_file))
   return db, doc_id, new_data
 end
 
@@ -124,14 +126,14 @@ def create_document(db, doc_id, data)
   db.save_doc(data)
 end
 
-def connection_string(database, host)
+def connection_string(database, host, netrc_file = nil)
   protocol  = "http"
   #hostname  = "127.0.0.1"
   port      = "5984"
   username  = "admin"
   password  = ""
 
-  netrc = File.read('/etc/couchdb/couchdb.netrc')
+  netrc = File.read(netrc_file || '/etc/couchdb/couchdb.netrc')
   netrc.scan(/\w+ [\w\.]+/).each do |key_value|
     key, value = key_value.split ' '
     case key
@@ -148,7 +150,7 @@ end
 
 def usage(s)
   $stderr.puts(s)
-  $stderr.puts("Usage: #{File.basename($0)} --host <host> --db <db> --id <doc_id> --data <json> [--file <file>]")
+  $stderr.puts("Usage: #{File.basename($0)} --host <host> --db <db> --id <doc_id> --data <json> [--file <file>] [--netrc-file <netrc-file>]")
   exit(2)
 end
 
diff --git a/manifests/document.pp b/manifests/document.pp
index 1153296..3ecc05d 100644
--- a/manifests/document.pp
+++ b/manifests/document.pp
@@ -10,6 +10,7 @@ define couchdb::document(
   $id,
   $host   = '127.0.0.1:5984',
   $data   = '{}',
+  $netrc  = '/etc/couchdb/couchdb.netrc',
   $ensure = 'content') {
 
   $url = "${host}/${db}/${id}"
@@ -17,7 +18,7 @@ define couchdb::document(
   case $ensure {
     default: { err ( "unknown ensure value '${ensure}'" ) }
     content: {
-      exec { "couch-doc-update --host ${host} --db ${db} --id ${id} --data \'${data}\'":
+      exec { "couch-doc-update --netrc-file ${netrc} --host ${host} --db ${db} --id ${id} --data \'${data}\'":
         require => Exec['wait_for_couchdb'],
         unless  => "couch-doc-diff $url '$data'"
       }
@@ -28,7 +29,7 @@ define couchdb::document(
         cmd    => 'PUT',
         host   => $host,
         path   => "${db}/${id}",
-        unless => "/usr/bin/curl -s -f --netrc-file /etc/couchdb/couchdb.netrc ${url}"
+        unless => "/usr/bin/curl -s -f --netrc-file ${netrc} ${url}"
       }
     }
 
@@ -37,7 +38,7 @@ define couchdb::document(
         cmd    => 'DELETE',
         host   => $host,
         path   => "${db}/${id}",
-        unless => "/usr/bin/curl -s -f --netrc-file /etc/couchdb/couchdb.netrc ${url}"
+        unless => "/usr/bin/curl -s -f --netrc-file ${netrc} ${url}"
       }
     }
   }
diff --git a/manifests/mirror_db.pp b/manifests/mirror_db.pp
index c2e5b21..06007e0 100644
--- a/manifests/mirror_db.pp
+++ b/manifests/mirror_db.pp
@@ -3,14 +3,18 @@ define couchdb::mirror_db (
   $from='',
   $to='' )
 {
-  $source = "$from$name"
-  $target = "$to$name"
+  $source = "$from/$name"
+  if $to == '' { $target = $name }
+  else { $target = "$to/$name" }
+
+  $replication_user = "replication"
+  $replication_role = "replication"
 
   couchdb::document { "${name}_replication":
-    db   => "_replicate",
+    db   => "_replicator",
     id   => "${name}_replication",
     host => $host,
-    data   => "{ \"source\": ${source}, \"target\": ${target}, \"continuous\": true }",
+    data   => "{ \"source\": \"${source}\", \"target\": \"${target}\", \"continuous\": true, \"user_ctx\": { \"name\": \"${replication_user}\", \"roles\": [\"${replication_role}\"] } }",
     require => Couchdb::Query["create_db_${name}"]
   }
 }
diff --git a/manifests/query.pp b/manifests/query.pp
index 72ae4d2..9507ca1 100644
--- a/manifests/query.pp
+++ b/manifests/query.pp
@@ -1,7 +1,11 @@
 define couchdb::query (
-  $cmd, $path, $host='127.0.0.1:5984', $data = '{}', $unless = undef) {
+  $cmd, $path,
+  $netrc='/etc/couchdb/couchdb.netrc',
+  $host='127.0.0.1:5984',
+  $data = '{}',
+  $unless = undef) {
 
-  exec { "/usr/bin/curl -s --netrc-file /etc/couchdb/couchdb.netrc -X ${cmd} ${host}/${path} --data \'${data}\'":
+  exec { "/usr/bin/curl -s --netrc-file ${netrc} -X ${cmd} ${host}/${path} --data \'${data}\'":
     require => [ Package['curl'], Exec['wait_for_couchdb'] ],
     unless  => $unless
   }