#!/bin/bash

function _ezfio_usage()
{
      echo "
Usage:
  ezfio set_file    EZFIO_DIRECTORY
  ezfio unset_file 

  ezfio has         DIRECTORY   ITEM
  ezfio get         DIRECTORY   ITEM
  ezfio set         DIRECTORY   ITEM  VALUE  : Scalar values
  ezfio set         DIRECTORY   ITEM         : Array values read from stdin

  ezfio set_verbose
  ezfio unset_verbose
"
}

function _require_ezfio_file()
{
  if [[ -z ${EZFIO_FILE} ]]
  then
    echo "EZFIO directory not set: try 'ezfio set_file'"
    return 1
  fi

  if [[ ! -d ${EZFIO_FILE} ]]
  then
    echo "EZFIO directory not accessible: try 'ezfio set_file'"
    return 1
  fi
}

function _require_first_argument()
{
  if [[ -z $1 ]]
  then
    _ezfio_usage
    echo "Error: First argument not set"
    return 1
  else
    return 0
  fi
}

function _require_second_argument()
{
  if [[ -z $2 ]]
  then
    _ezfio_usage
    echo "Error: Second argument not set"
    return 1
  fi
}

function _ezfio_set_verbose()
{
  function _ezfio_info()
  {
    echo "$1"
  }
  return 0
}

function _ezfio_unset_verbose()
{
  function _ezfio_info()
  {
    :
  }
  return 0
}

function _ezfio_set_file()
{
  _require_first_argument $@ || return 1

  if [[ ! -d $1 ]]
  then
    echo "No such directory: $1"
    return 1
  fi
  export EZFIO_FILE=$1
  _ezfio_info "Set file ${EZFIO_FILE}"
  return 0
}

function _ezfio_unset_file()
{
  _require_ezfio_file || return 1
  _ezfio_info "Unset file ${EZFIO_FILE}"
  unset EZFIO_FILE
  return 0
}

function _ezfio_has()
{
  _require_ezfio_file        || return 1
  _require_first_argument $@ || return 1
  
  if [[ ! -d ${EZFIO_FILE}/${1,,} ]]
  then
      return 1
  fi
  
  if [[ -z $2 ]]
  then
      return 0
  fi

  if [[ -f  ${EZFIO_FILE}/${1,,}/${2,,} ]]
  then
      _ezfio_info "EZFIO has $1 / $2"
      return 0
  elif [[ -f  ${EZFIO_FILE}/${1,,}/${2,,}.gz ]]
  then
      _ezfio_info "EZFIO has $1 / $2"
      return 0
  else
      _ezfio_info "EZFIO doesn't have $1 / $2"
      return 1
  fi
}

function _ezfio_get()
{
  _require_ezfio_file        || return 1

  if [[ -z $1 ]] 
  then
    ls ${EZFIO_FILE}
    return 0
  fi

  _ezfio_has $@ || return 1

  if [[ -z $2 ]]
  then
      ls ${EZFIO_FILE}/${1,,}
      return 0
  fi

  if [[ -f ${EZFIO_FILE}/${1,,}/${2,,} ]]
  then
    cat ${EZFIO_FILE}/${1,,}/${2,,}
  elif [[ -f ${EZFIO_FILE}/${1,,}/${2,,}.gz ]]
  then
    zcat ${EZFIO_FILE}/${1,,}/${2,,}.gz
  else
    _ezfio_info "EZFIO doesn't have $1 / $2"
    return 1
  fi
  return 0
}

function _ezfio_set()
{
  _require_ezfio_file         || return 1
  _require_first_argument  $@ || return 1
  _require_second_argument $@ || return 2

  if [[ ! -d ${EZFIO_FILE}/${1,,} ]]
  then
    mkdir -p ${EZFIO_FILE}/${1,,}
  fi
  if [[ ! -z "$3" ]]
  then
    echo "$3" > ${EZFIO_FILE}/${1,,}/${2,,}
  else
    zcat > ${EZFIO_FILE}/${1,,}/${2,,}.gz
  fi
  return 0
}

#---

function ezfio()
{
  case $1 in 
    "has")
      shift
      _ezfio_has $@
      ;;

    "set")
      shift
      _ezfio_set $@
      ;;

    "get")
      shift
      _ezfio_get $@
      ;;

    "set_file")
      shift
      _ezfio_set_file $@
      ;;

    "unset_file")
      shift
      _ezfio_unset_file 
      ;;

    "set_verbose")
      _ezfio_set_verbose
      ;;

    "unset_verbose")
      _ezfio_unset_verbose
      ;;

    *)
      _ezfio_usage
      ;;
  esac

}


_ezfio_unset_verbose


# Completion


_Complete()
{
  local cur

  COMPREPLY=()
  cur="${COMP_WORDS[COMP_CWORD]}"
  prev="${COMP_WORDS[COMP_CWORD-1]}"
  prev2="${COMP_WORDS[COMP_CWORD-2]}"

  case "${prev2}" in
    set|has|get)
      COMPREPLY=( $(compgen -W "$(cd ${EZFIO_FILE}/${prev} ; ls | sed 's/\.gz//' )" -- $cur ) )
      return 0
      ;;
  esac

  case "${prev}" in
    unset_file|set_verbose|unset_verbose)
      COMPREPLY=()
      return 0
      ;;
    set_file)
      COMPREPLY=( $(compgen -W "$(\ls -d */ | sed 's|/||g')" -- ${cur} ) )
      return 0
      ;;
    set|has|get)
      COMPREPLY=( $(compgen -W "$(cd ${EZFIO_FILE} ; \ls -d */ | sed 's|/||g')" -- $cur ) )
      return 0
      ;;
    *)
      if [[ -z ${EZFIO_FILE} ]]
      then
        COMPREPLY=( $(compgen -W 'set_file \
                                  set_verbose unset_verbose -h' -- $cur ) )
      else
        COMPREPLY=( $(compgen -W 'has get set unset_file \
                                  set_verbose unset_verbose -h' -- $cur ) )
      fi
      return 0
      ;;
  esac

}

complete -F _Complete ezfio