create_db () {
  local url=$1
  local db=$2
  $CURL -X PUT "${url}/${db}"
}

db_exists () {
  local url=$1
  local db=$2
  $CURL -X GET "${url}/${db}" | grep -q -v '{"error":"not_found"'
  return $?
}

delete_db () {
  local url=$1
  local db=$2
  $CURL -X DELETE "${url}/${db}"
}

delete_doc () {
  local url=$1
  local db=$2
  local doc=$3
  local latest_rev=`get_latest_rev $url $db $doc`
  $CURL -X DELETE "${url}/${db}/${doc}?rev=$latest_rev"
}

doc_exists () {
  local url=$1
  local db=$2
  local doc=$3
  $CURL -X GET "${url}/${db}/${doc}" | grep -q -v '{"error":"not_found"'
  return $?
}


dump_db_old () {
  # couchdb-dump cmd is VERY slow
  # we don't use this method in production, 
  # only left here for speed evaluation

  # dump_db() and restore_db() rely on python-couchdb package,
  #  python-couchdb =< 0.8-1 needs to be patched, see
  #  http://code.google.com/p/couchdb-python/issues/detail?id=194


  local url=$1
  local db=$2
  local user=$3
  local pw=$4
  local dumpdir=$5

  [ -z $dumpdir ] && dumpdir='/var/backups/couchdb'
  
  echo "Dumping db \"$db\" to ${dumpdir}/$db"
  couchdb-dump -u $user -p $pw ${url}/$db > ${dumpdir}/$db #2>/dev/null

  echo "Dumping _security to ${DUMPDIR}/${db}_security"
  $CURL -X GET "${URL}/${db}/_security" > ${DUMPDIR}/${db}_security

  chmod 600 ${dumpdir}/${db}*
}

dump_db () {
  local url=$1
  local db=$2

  # old curl options, just for re-evaluating speed
  # curl_opts='--netrc-file /etc/couchdb/couchdb.netrc'

  #read -a docs <<< `curl $curl_opts -sS 127.0.0.1:5984/$db/_all_docs | json_pp | sed -ne 's/"id" : "//p' | sed -e 's/",//'`
  read -a docs <<< `wget $WGET_OPTS 127.0.0.1:5984/$db/_all_docs | json_pp | sed -ne 's/"id" : "//p' | sed -e 's/",//;s/\//%2F/'`

  count=${#docs[*]}
  if [[ $count != 0 ]]
  then
    last=${docs[$(( count - 1 ))]}
  fi

  echo '{'
  echo '"new_edits":false,'
  echo '"docs": ['
  for id in "${docs[@]}"
  do
    # curl $curl_opts -sS 127.0.0.1:5984/$db/$id
    wget $WGET_OPTS 127.0.0.1:5984/$db/$id?attachments=true 
    if [[ $id != $last ]]
    then
      echo ','
    fi
  done

  echo ']'
  echo '}'
}

dump_db_to_file () {
  local url=$1
  local db=$2
  local dumpdir=$3

  [ -z $dumpdir ] && dumpdir='/var/backups/couchdb'
  echo "Dumping db \"$db\" to ${dumpdir}/$db"

  dump_db $url $db $dumpdir > ${dumpdir}/$db
  chmod 600 ${dumpdir}/${db}

  dump_db_security $url $db $dumpdir
}

dump_db_security () {


  local url=$1
  local db=$2
  local dumpdir=$3

  [ -z $dumpdir ] && dumpdir='/var/backups/couchdb'

  echo "Dumping \"$db\" _security to ${DUMPDIR}/${db}_security"
  $CURL -X GET "${URL}/${db}/_security" > ${DUMPDIR}/${db}_security

  chmod 600 ${dumpdir}/${db}_security
} 


get_dbs () { 
  local url=$1
  local dbs="`$CURL -X GET "${url}/_all_dbs" | sed 's/[\[",]/ /g' | sed 's/]//'`"
  echo "$dbs"
}

get_security () {
  local url=$1
  local db=$2
  local security=`$CURL -X GET "${url}/${db}/_security"`
  echo $security
}

get_latest_rev () {
  local url=$1
  local db=$2
  local doc=$3
  local latest_rev=''
  latest_rev=`$CURL -X GET "${url}/${db}/${doc}" | sed 's/^.*"_rev":"//'|cut -d'"' -f 1`
  echo $latest_rev
}

replicate_db () {
  local url=$1
  local backend_url=$2
  local source_db=$3
  local target_db=$4
  local additional_opts=$5
  local task=''
  # old style replication, no status can be queried
  # -X POST http://localhost:5984/_replicate -d ' {"source": "http://admin:zyMM7LZMjGE2aUvJ5sH_8SraPuxB2H5L@localhost:5984/users", "target": "http://admin:zyMM7LZMjGE2aUvJ5sH_8SraPuxB2H5L@localhost:5984/users_replicated", "create_target": true }
  # netrcfile doesn't work with replicate, we need to use username + pw here
  #echo "Getting _security from $db"
  local security=`get_security $url $source_db` 

  create_db $url $target_db

  #echo "Set security for $tmpdb"
  set_security $url $target_db $security

  #echo "Replicating $db to $tmpdb"
  #$CURL -X POST ${BACKEND_URL}/_replicator -d " { \"_id\":  \"${source_db}_${target_db}\", \"source\": \"$auth_url/${source_db}\", \"target\": \"$auth_url/${target_db}\", \"create_target\": true $additional_opts }"
  task="${source_db}_${target_db}"
  $CURL -X POST ${BACKEND_URL}/_replicator -d " { \"_id\":  \"${task}\", \"source\": \"$auth_url/${source_db}\", \"target\": \"$auth_url/${target_db}\" $additional_opts }"
  
  #echo -e "\nGetting replication status of task \"$task\":"
  #$CURL -X GET ${BACKEND_URL}/_replicator/$task

  wait_for_complete_replication $backend_url
}

restore_db () {
  local url=$1
  local db=$2
  local dumpdir=$3
  [ -z $dumpdir ] && local dumpdir='/var/backups/couchdb'

  # restore with couchdb-load only works with an empty db
  db_exists $url $db && delete_db $url $db
  create_db $url $db
  
  # security 
  $CURL -d @${dumpdir}/${db}_security -X PUT "${url}/${db}/_security"
  
  # content
  $CURL -d @${dumpdir}/${db} -X POST "${url}/${db}/_bulk_docs"
  
  # old style
  # couchdb-load -u $couchdb_user -p $couchdb_pw ${url}/$db <  ${dumpdir}/$db 
  # very old style
  # $CURL -d @${dumpdir}/$db -X POST "${url}/${db}/_bulk_docs"
}

set_security () {
  local url=$1
  local db=$2
  local security=$3
  $CURL -X PUT "${url}/${db}/_security" -d $security
}

wait_for_complete_replication () {
  local backend_url=$1
  local uncomplete_tasks=1
  echo -e "\nWaiting for uncompleted replication tasks"
  while [ $uncomplete_tasks -gt 0 ]
  do
    uncomplete_tasks=`$CURL -X GET "${backend_url}/_replicator/_all_docs/?include_docs=true" | egrep -v '("_replication_state":"completed"|"id":"_design/_replicator"|{"total_rows":|^]}$)'|wc -l`
    sleep 1
  done
}