#!/bin/bash
################################################
# Title      : Ceph RBD Scripts                #
# Date       : 2017-01-20                      #
# Maintainer : Yongbok Kim (ruo91@yongbok.net) #
################################################
#HOSTNAME_GET="$(hostname | cut -d '.' -f '1')"
IMG_POOL_NAME="rbd"
IMG_NAME="$(hostname)"
MOUNT_POINT="/var/lib/docker"

CEPH_RBD_SIZE="40960"
CEPH_CLUSTER_NAME="ybkim-ceph-cluster"
CEPH_CLUSTER_CONF="/etc/ceph/$CEPH_CLUSTER_NAME.conf"
CEPH_KEYRING_CONF="/etc/ceph/$CEPH_CLUSTER_NAME.client.admin.keyring"

function f_rbd_map {

  if [[ -d "$MOUNT_POINT" ]]; then
      rbd create $IMG_NAME --size $CEPH_RBD_SIZE --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1
      rbd map $IMG_NAME --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1

  else
      mkdir $MOUNT_POINT
      rbd create $IMG_NAME --size $CEPH_RBD_SIZE --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1
      rbd map $IMG_NAME --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1
  fi
}

function f_rbd_unmap {
  umount $MOUNT_POINT > /dev/null 2>&1
  rbd unmap /dev/rbd/$IMG_POOL_NAME/$IMG_NAME --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1
}

function f_rbd_rm {
  umount $MOUNT_POINT > /dev/null 2>&1
  rbd unmap /dev/rbd/$IMG_POOL_NAME/$IMG_NAME --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1
  rbd rm $IMG_NAME --pool $IMG_POOL_NAME -k $CEPH_KEYRING_CONF -c $CEPH_CLUSTER_CONF > /dev/null 2>&1
}

function f_help {
  echo "Usage: $ARG_0 [Arguments]"
  echo
  echo "- Arguments"
  echo "m, map       : rbd map"
  echo "u, unmap     : rbd unmap"
  ecoh "r, remove    : rbd remove"
  echo
}

# Main
ARG_0="$0"
ARG_1="$1"
ARG_2="$2"

case ${ARG_1} in
    m|M|map|MAP)
        f_rbd_map
    ;;

    u|U|unmap|UNAMP)
        f_rbd_unmap
    ;;

    r|R|rm|RM)
        f_rbd_rm
    ;;

    *)
      f_help
    ;;
esac