consulkv

#!/bin/bash

# Borrowed from: https://gist.github.com/progrium/b45a9fe697dd68c3ea0f
# All credit to @progrium.

if [ -f /etc/default/octohost ]; then
  . /etc/default/octohost
else
  CONSUL_SERVER="localhost:8500"
fi

# Adding anonymous token by default.
if [ -z $CONSUL_TOKEN ];
then
  CONSUL_TOKEN="anonymous"
fi

main() {
  case "$1" in
    info)
      curl -s "$CONSUL_SERVER/v1/kv/$2?token=$CONSUL_TOKEN" | jq -r .[]
      ;;
    get)
      curl -s "$CONSUL_SERVER/v1/kv/$2?token=$CONSUL_TOKEN" | jq -r .[].Value | base64 -d
      ;;
    set)
      curl -s -X PUT -d "$3" "$CONSUL_SERVER/v1/kv/$2?token=$CONSUL_TOKEN" > /dev/null
      ;;
    del)
      curl -s -X DELETE -d "$3" "$CONSUL_SERVER/v1/kv/$2?token=$CONSUL_TOKEN" > /dev/null
      ;;
    ls)
      if [[ "$2" == "" ]]; then
        curl -s "$CONSUL_SERVER/v1/kv/?keys&token=$CONSUL_TOKEN" | jq -r .[]
      else
        curl -s "$CONSUL_SERVER/v1/kv/$2/?keys&token=$CONSUL_TOKEN" | jq -r .[] | sed "s|$2/||"
      fi
      ;;
    service:set)
      curl -s -X PUT -d "$2" "$CONSUL_SERVER/v1/agent/service/register?token=$CONSUL_TOKEN"
      ;;
    service:del)
      curl -s "$CONSUL_SERVER/v1/agent/service/deregister/$2?token=$CONSUL_TOKEN"
      ;;
    service:health)
      STATUS=$(curl -s "$CONSUL_SERVER/v1/health/service/$2?token=$CONSUL_TOKEN" | jq '.[].Checks[0].Status' | cut -d '"' -f 2)
      echo $STATUS
      ;;
    services)
      curl -s "$CONSUL_SERVER/v1/agent/services?token=$CONSUL_TOKEN" | jq -c '.[]'
      ;;
    services:catalog)
      for service in $( curl -s "$CONSUL_SERVER/v1/catalog/services?token=$CONSUL_TOKEN" | jq 'keys' | jq '.[]' ); do
        SERVICE=$(echo $service | cut -d '"' -f 2)
        curl -s "$CONSUL_SERVER/v1/catalog/service/$SERVICE?token=$CONSUL_TOKEN" | jq -c '.[]'
      done
      ;;
  esac
}

main "$@"