functions.sh

# ===================================
# bash-backup
#  Functions file
#
# author: yusuke@newsdict.xyz
# LICENSE: MIT License
# github: newsdict/bash-backup
# ===================================

# current date time
now=$(date +%Y%m%d%H%M)
serial=$(( ( RANDOM % 100 )  + 1 ))

# Require environment file
get_require_environments()
{
  if [ ! -z "${options[0]}" ] && [ -f $current_path/.env-${options[0]} ] ; then
    echo $current_path/.env-${options[0]}
  elif [ ! -z "${options[0]}" ] && [ -f $HOME/.env-${options[0]} ] ; then
    echo $HOME/.env-${options[0]}
  elif [ -f $current_path/.env ]; then
    echo $current_path/.env
  elif [ -f $HOME/.env ]; then
    echo $HOME/.env
  fi
}
function upload_storage()
{
  log::info "####"
  log::info "# start to upload storage"
  log::info "####"

  _backup_path="s3://${s3_conf["bucket"]}/${s3_conf["path"]}"
  export AWS_ACCESS_KEY_ID=${s3_conf["access_key"]}
  export AWS_SECRET_ACCESS_KEY=${s3_conf["secret"]}
  _option="--region ${s3_conf["region"]}"

  aws $_option s3 cp --quiet $compress_name $_backup_path/
  log::info "uploaded: $_backup_path/$(basename $compress_name)"

  for _file in $(aws $_option s3 ls $_backup_path/ | awk '{print $4}' ); do
  	if ! grep $_file <(aws $_option s3 ls $_backup_path/ | awk '{print $4}' |tail -n ${s3_conf["keep"]}) >/dev/null ;then
  		aws $_option s3 rm --quiet $_backup_path/$_file && log::info "deleted file: $_backup_path/$_file"
  	fi
  done

  log::info "####"
  log::info "# uploaded storage"
  log::info "####"
}
function compress()
{
  log::info "####"
  log::info "# start compress"
  log::info "####"

  pushd $working_directory > /dev/null

  _compress_name=$temporary_directory/$backup_filename.$now

  if [ "$commpression_type" = "gzip" ]; then

    tar -zcf $_compress_name.tar.gz ./*
    compress_name="$_compress_name.tar.gz"
    log::info "compress: $compress_name"

  elif [ "$commpression_type" = "zip" ]; then

    zip -rq $archive_file.zip ./*
    compress_name="$_compress_name.zip"
    log::info "compress: $compress_name"

  elif [ "$commpression_type" = "bzip2" ]; then

    tar -jcf $archive_file.tar.bz2 ./*
    compress_name="$_compress_name.tar.bz2"
    log::info "compress: $compress_name"

  fi

  popd > /dev/null
  log::info "####"
  log::info "# end compress"
  log::info "####"
}
function dump_database()
{
  log::info "####"
  log::info "# start database dump"
  log::info "####"
  if [ ${database_conf["engine"]} = "mysql" ]; then

    _dump_file=$working_directory/mysql.dump
    echo "[client]
      user = ${database_conf["username"]}
      password = ${database_conf["password"]}
      port = ${database_conf["port"]}
      host = ${database_conf["host"]}" > $temporary_directory/my.cnf.$serial
    mysqldump \
      --defaults-file=$temporary_directory/my.cnf.$serial \
      ${database_conf["name"]} > $_dump_file
    rm $temporary_directory/my.cnf.$serial
    log::info "dump file: $_dump_file"

  fi
  log::info "####"
  log::info "# end database dump"
  log::info "####"
}
function archive()
{
  log::info "####"
  log::info "# start archive"
  log::info "####"
  archive_file=$working_directory/archive

  if [ "$commpression_type" = "gzip" ]; then

    # Suppress `tar: Removing leading `/' from member names`
    declare -a _archive_path
    for ((i = 0; i < ${#archive_paths[@]}; i++)) {
      _archive_path[i]=$(echo ${archive_paths[i]} | sed 's#^/##')
    }
    tar -zcf $archive_file.tar.gz -C / ${_archive_path[@]}
    log::info "archive: $archive_file.tar.gz"

  elif [ "$commpression_type" = "zip" ]; then

    zip -rq $archive_file.zip ${archive_paths[@]}
    log::info "archive: $archive_file.zip"

  elif [ "$commpression_type" = "bzip2" ]; then

    # Suppress `tar: Removing leading `/' from member names`
    declare -a _archive_path
    for ((i = 0; i < ${#archive_paths[@]}; i++)) {
      _archive_path[i]=$(echo ${archive_paths[i]} | sed 's#^/##')
    }
    tar -jcf $archive_file.tar.bz2 -C / ${_archive_path[@]}
    log::info "archive: $archive_file.tar.bz2"

  fi

  log::info "####"
  log::info "# end archive"
  log::info "####"
}
# parse arguments
#  usage)
#   declare -A options
#   pasrse_arguments $@
function pasrse_arguments()
{
  local kv_regex='^--?(.+)=(.+)$'
  local k_regex='^--?([^=]+)$'
  local argc=0
  for arg in $@
  do
    if [[ $arg =~ $kv_regex ]]; then
      options[${BASH_REMATCH[1]}]=${BASH_REMATCH[2]}
    elif [[ $arg =~ $k_regex ]]; then
      options[${BASH_REMATCH[1]}]=1
    else
      options[$argc]=$arg
      argc=$(( $argc + 1 ))
    fi
  done
}
function initialize()
{
  # default parameters
  temporary_directory=/tmp
  log_name=/var/log/backup.$now.log
  backup_filename=backup
  display_messages=1
  archive=0
  database=0
  commpression_type=gzip
}
function create_temporary_directory()
{
  working_directory=$temporary_directory/$backup_filename.$now.$serial
  if [ ! -d $working_directory ]; then
    mkdir $working_directory
  fi
}
function version()
{
  echo $(cat $current_path/VERSION)
}
function clean()
{
  log::info "script log: $log_name"
  log::info "Complete script"
  rm -rf $working_directory
  rm $compress_name
}
function log::info(){
  echo "["$(date +"%Y-%m-%d %H:%M:%S")"] "$1 2>&1 >> $log_name
  if [ $display_messages -eq 1 ]; then
    echo -e "\e[1;32m["$(date +"%Y-%m-%d %H:%M:%S")"] "$1"\e[0m"
  fi
}
function log::error(){
  echo "["$(date +"%Y-%m-%d %H:%M:%S")"] [ERROR] "$1 2>&1 >> $log_name
  if [ $display_messages -eq 1 ]; then
    echo -e "\e[1;31m["$(date +"%Y-%m-%d %H:%M:%S")"] [ERROR] "$1"\e[0m"
  fi
  exit 1
}