diff options
| -rw-r--r-- | files/couch-doc-update | 116 | ||||
| -rw-r--r-- | manifests/add_user.pp | 7 | ||||
| -rw-r--r-- | manifests/update.pp | 3 |
3 files changed, 126 insertions, 0 deletions
diff --git a/files/couch-doc-update b/files/couch-doc-update new file mode 100644 index 0000000..9ba6db7 --- /dev/null +++ b/files/couch-doc-update @@ -0,0 +1,116 @@ +#!/usr/bin/ruby + +# +# This script will update the values of a particular couchdb document. The benefit of this little script over +# using a simple curl command for updating a document is this: +# +# * exit non-zero status if document was not updated. +# * updates existing documents easily, taking care of the _rev id for you. +# * if document doesn't exist, it is created +# +# REQUIREMENTS +# +# gem 'couchrest' +# +# USAGE +# +# couch-doc-update --db <db> --id <doc_id> --data <json> +# +# EXAMPLE +# +# create a new user: +# couch-doc-update --db _users --id org.couchdb.user:ca_daemon --data '{"type": "user", "name": "ca_daemon", "roles": ["certs"], "password": "sshhhh"}' +# +# update a user: +# couch-doc-update --db _users --id org.couchdb.user:ca_daemon --data '{"password":"sssshhh"}' +# + +begin; require 'rubygems'; rescue LoadError; end # optionally load rubygems +require 'couchrest' +require 'json' + +def main + # + # parse options + # + @db_name = nil + @doc_id = nil + @new_data = nil + loop do + case ARGV[0] + when '--db' then ARGV.shift; @db_name = ARGV.shift + when '--id' then ARGV.shift; @doc_id = ARGV.shift + when '--data' then ARGV.shift; @new_data = ARGV.shift + when /^-/ then usage("Unknown option: #{ARGV[0].inspect}") + else break + end + end + usage("Missing required option") unless @db_name && @doc_id && @new_data + @new_data = JSON.parse(@new_data) + + # + # update document + # + begin + @db = CouchRest.database(connection_string(@db_name)) + @doc = get_document(@db, @doc_id) + result = if @doc + update_document(@db, @doc, @new_data) + else + create_document(@db, @doc_id, @new_data) + end + exit 0 if result['ok'] + raise StandardError.new(result.inspect) + rescue StandardError => exc + puts "ERROR: " + exc.to_s + exit 1 + end +end + +def get_document(db, doc_id) + begin + db.get(doc_id) + rescue RestClient::ResourceNotFound + nil + end +end + +def update_document(db, doc, data) + data.each do |key, value| + doc[key] = value + end + db.save_doc(doc) +end + +def create_document(db, doc_id, data) + data["_id"] = doc_id + db.save_doc(data) +end + +def connection_string(database) + protocol = "http" + hostname = "127.0.0.1" + port = "5984" + username = "admin" + password = "" + + netrc = File.read('/etc/couchdb/couchdb.netrc') + netrc.scan(/\w+ [\w\.]+/).each do |key_value| + key, value = key_value.split ' ' + case key + when "machine" then hostname = value + when "login" then username = value + when "password" then password = value + end + end + + "%s://%s:%s@%s:%s/%s" % [protocol, username, password, hostname, port, database] +end + +def usage(s) + $stderr.puts(s) + $stderr.puts("Usage: #{File.basename($0)} --db <db> --id <doc_id> --data <json>") + exit(2) +end + +main() diff --git a/manifests/add_user.pp b/manifests/add_user.pp index acadcd0..560a6a8 100644 --- a/manifests/add_user.pp +++ b/manifests/add_user.pp @@ -5,5 +5,12 @@ define couchdb::add_user ($host='127.0.0.1:5984', $roles, $pw ) { url => "_users/org.couchdb.user:$name", data => "{ \"_id\": \"org.couchdb.user:$name\", \"type\": \"user\", \"name\": \"$name\", \"roles\": $roles, \"password\": \"$pw\"}", } + # + # couchdb::update { "update_user_$name": + # db => "_users", + # id => "org.couchdb.user:$name", + # data => "{\"type\": \"user\", \"name\": \"$name\", \"roles\": $roles, \"password\": \"$pw\"}", + # } + # } diff --git a/manifests/update.pp b/manifests/update.pp new file mode 100644 index 0000000..129d875 --- /dev/null +++ b/manifests/update.pp @@ -0,0 +1,3 @@ +define couchdb::update ($db, $id, $data) { + exec { "couch-doc-update --db $db --id $id --data \'$data\'": } +} |