autoload colors; colors;

function secure_source () {
  if [ -e $1 ]; then
    source $1
  fi
}
function contains() {
  local n=$#
  local value=${!n}
  for ((i=1;i < $#;i++)) {
    if [ "${!i}" == "${value}" ]; then
      echo "y"
      return 0
    fi
  }
  echo "n"
  return 1
}

function getIP () {
  echo $(grep $1 ~/.ssh/config -A 1 | tail -1 | tr -s ' ' | cut -d ' ' -f 3)
}

function getServerName () {
  echo $(grep $1 ~/.ssh/config -B 1 | head -1 | tr -s ' ' | cut -d ' ' -f 3)
}

function ssh_exec_by_file () {
  ssh -t $1 "bash -s" -- < $2
}

function cp_container() {
  sourceName=$1
  targetName=$2
  cp $sourceName $targetName -r
  cd $targetName
  mv ${sourceName}.scss ${targetName}.scss
  mv ${sourceName}.js ${targetName}.js
  cd ..
}
function set_proxy() {
  export http_proxy=http://127.0.0.1:8119;export https_proxy=http://127.0.0.1:8119;
}
function set_vagrant_proxy() {
  export http_proxy=http://127.0.0.1:9119;export https_proxy=http://127.0.0.1:9119;
}
function set_ss_proxy() {
  export https_proxy=socks5://127.0.0.1:1080/
  export http_proxy=socks5://127.0.0.1:1080/
}
function unset_proxy() {
  unset all_proxy
}
function post {
  curl -H "Content-Type: application/json" -X POST -d $1 $2
}
function git-set-remote {
  git remote rm origin
  git remote rm upstream
  git remote add origin $1
  git remote add upstream $1
}
function search_installed_packages {
  dpkg --get-selections | grep $1
}
function delete_packages {
  sudo apt remove `_search_installed_packages $1 | cut -f 1 | tr "\n" " "`
}
function gem_source_to_taobao {
  gem source -r https://rubygems.org/
  gem source -a https://ruby.taobao.org
}

function set_anonymous {
  git filter-branch --env-filter '
  if [ "$GIT_AUTHOR_NAME" = "ralletstellar" ]; then \
    export GIT_AUTHOR_NAME="Anonymous" GIT_AUTHOR_EMAIL="anonymous@anonymous.com"; \
  fi
  '
}
function genAndroidKeystore {
  keytool -genkey -v -keystore /tmp/.keystore -alias rallets -keyalg RSA -keysize 2048 -validity 10000
}
function delete_except_latest {
  total=`ll | wc -l`
  num_to_delete=$((total-$1))
  ls -t | tail -n $num_to_delete | xargs rm
}
function strip_color() {
  sed "s,$(printf '\033')\\[[0-9;]*[a-zA-Z],,g"
}
function docker_rm_all() {
  docker rm -f `docker ps --no-trunc -aq`
}
function Replace () {
  if [[ "$(uname)" == "Darwin" ]]; then
    if [ "$#" -eq 3 ]; then
      ag $2 -l -G "$1" | xargs sed -i '' s/$2/$3/g
    elif [ "$#" -eq 2 ]; then
      ag $1 -l | xargs sed -i '' s/$1/$2/g
    fi
  elif [[ "$(expr substr $(uname -s) 1 5)" == "Linux" ]]; then
    if [ "$#" -eq 3 ]; then
      ag $2 -l -G $1 | xargs sed -i s/$2/$3/g
    elif [ "$#" -eq 2 ]; then
      ag $1 -l | xargs sed -i s/$1/$2/g
    fi
  fi
}
function git-change-module-remote() {
  git config --file=.gitmodules submodule.$1.url $2
  git config --file=.gitmodules submodule.$1.branch $3
  git submodule sync
  git submodule update --init --recursive --remote
}
function replace() {
  ag -l -G $1 | xargs sed -ri.bak -e "s/$2/$3/g"
}
function markdown-preview() {
  cat $1 | instant-markdown-d > /dev/null 2>&1
}
function swap() {
  if [ $# -ne 2 ]; then
    echo "Usage: swap file1 file2"
  else
    local TMPFILE=$(mktemp)
    mv "$1" $TMPFILE
    mv "$2" "$1"
    mv $TMPFILE "$2"
  fi
}

function init_db() {
  SERVICE=server
  while getopts ":s:" o; do
    case "${o}" in
      s)
        SERVICE=${OPTARG}
        ;;
      *)
        echo "Usage: init_db [-s SERVICE]"
        return
        ;;
    esac
  done
  dc exec $SERVICE rails db:drop db:create db:migrate db:seed
}
function cmd_exists() {
  $* &> /dev/null
  if [[ $? == 0 ]]; then
    echo Y
  else
    echo N
  fi
}
function qshell_setup() {
  if [[ $(cmd_exists qshell) == 'N' ]]; then
    curl https://raw.githubusercontent.com/paiyou-network/scripts/master/install-qshell.sh | bash
  fi
  ~/Projects/paiyou-hub/bin/setup-qshell-account.sh
}
function qshell_upload() {
  qshell_setup
  bucket=assets
  while getopts "b:f:k:t" o; do
    case "${o}" in
      b)
        bucket=${OPTARG}
        ;;
      f)
        filepath=${OPTARG}
        ;;
      k)
        key=${OPTARG}
        ;;
      t)
        timestamp=true
        ;;
      *)
        echo "Usage: qshell_upload [-b BUCKET] -f FILEPATH [-k KEY] [-t]"
        return
        ;;
    esac
  done
  if [[ $key == '' ]]; then
    key=$(basename $filepath)
  fi
  if [[ $timestamp == 'true' ]];then
    key=`date +%Y%m%dT%H%M%S`_${key}
  fi
  qshell rput $bucket $key $filepath
}

function gitr() {
  for dir in `ls`; do
    if [[ -d "$dir" && -d "$dir/.git" ]]; then
      pushd .
      echo "$fg[green]$(basename $dir)$reset_color"
      cd $dir
      git $*
      popd
    fi
  done
}

function gitcopy() {
  n=1
  while getopts "c:n:t" o; do
    case "${o}" in
      c)
        commit=${OPTARG}
        ;;
      n)
        n=${OPTARG}
        ;;
      t)
        trelloCardName=${OPTARG}
        ;;
      *)
        usage
        ;;
    esac
  done
  prefix=`git remote get-url origin | sed -E 's/git@github.com:/https:\/\/github.com\//g' | sed -E 's/(.*)\.git/\1/'`
  project_name=`echo $prefix | sed -E 's/.*\/(.*)/\1/'`
  trelloCardName=`git log $commit -n $n --pretty="✔️  %s"`
  commits=`git log $commit -n $n --stat --pretty="
* [$project_name]($prefix/commit/%H) %an: **%s**" | sed 's/^[^*]/> /'`
  echo $commits
  which pbcopy &> /dev/null
  if [[ $? == '0' ]]; then
    echo $commits | pbcopy
  fi
  if [[ "$trelloCardName" != '' ]]; then
    ruby ~/Projects/paiyou-hub/bin/trello_action.rb -n $trelloCardName -d "$commits" -u `git config user.name`
  fi
}
function kubectl() {
  KCONTEXT=${KCONTEXT:-gcloud}
  DEBUG=false
  finalopts=()
  while [[ $@ != "" ]] do
    case $1 in
      -c)
        KCONTEXT="$2"
        shift; shift
        ;;
      --context=*)
        KCONTEXT="${i#*=}"
        shift
        ;;
      --debug)
        DEBUG=true
        shift
        ;;
      *)
        finalopts+=($1)
        shift
        ;;
    esac
  done
  [[ $DEBUG == "true" ]] && echo "kubectl $finalopts --kubeconfig=$HOME/.kube/${KCONTEXT}_config"
  command kubectl $finalopts --kubeconfig=$HOME/.kube/${KCONTEXT}_config
}
function stern {
  KCONTEXT=${KCONTEXT:-gcloud}
  finalopts=()
  while [[ $@ != "" ]] do
    case $1 in
      -c)
        KCONTEXT="$2"
        shift; shift
        ;;
      --context=*)
        KCONTEXT="${i#*=}"
        shift
        ;;
      *)
        finalopts+=($1)
        shift
        ;;
    esac
  done
  echo "stern $finalopts --kubeconfig=$HOME/.kube/${KCONTEXT}_config"
  command stern $finalopts -t --since 10m --kubeconfig=$HOME/.kube/${KCONTEXT}_config
}
function helm() {
  KCONTEXT=${KCONTEXT:-gcloud}
  DEBUG=false
  finalopts=()
  while [[ $@ != "" ]] do
    case $1 in
      -c)
        KCONTEXT="$2"
        shift; shift
        ;;
      --context=*)
        KCONTEXT="${i#*=}"
        shift
        ;;
      --debug)
        DEBUG=true
        finalopts+=($1)
        shift
        ;;
      *)
        finalopts+=($1)
        shift
        ;;
    esac
  done
  TLS=""
  case $KCONTEXT in
    gcloud)
      TLS="--tls"
  esac
  [[ $DEBUG == "true" ]] && echo "helm $finalopts $TLS --kubeconfig=$HOME/.kube/${KCONTEXT}_config"
  command helm $finalopts $TLS --kubeconfig=$HOME/.kube/${KCONTEXT}_config
}
function kexec {
  RAN=false
  KCONTEXT=${KCONTEXT:-gcloud}
  NAMESPACE=default
  finalopts=()
  while [[ $@ != "" ]] do
    case $1 in
      -c)
        KCONTEXT="$2"
        shift; shift
        ;;
      --context=*)
        KCONTEXT="${i#*=}"
        shift
        ;;
      -r)
        RAN=true
        shift
        ;;
      -p)
        PROJECT="$2"
        shift; shift
        ;;
      -n)
        NAMESPACE="$2"
        shift; shift
        ;;
      *)
        finalopts+=($1)
        shift
        ;;
    esac
  done

  RUNNING_POD_INDEX=-1
  while true; do
    ALL_PODS=$(kubectl -c $KCONTEXT -n $NAMESPACE get pods | grep "$PROJECT")
    echo $fg[green]"All Pods:"$reset_color
    echo $ALL_PODS
    if  [[ ${#ALL_PODS[@]} == 0 ]]; then
      echo $fg[red]"Pod not found for $PROJECT"$reset_color
      break
    fi
    RUNNING_PODS=($(echo $ALL_PODS | egrep "$PROJECT.* ?1/[0-9]? *Running" | awk '{print $1}'))
    if [[ `echo $ALL_PODS | wc -l` != ${#RUNNING_PODS[@]} ]]; then
      sleep 2
      echo $fg[red]'Pods are not ready, wait...'$reset_color
      continue
    fi
    if [[ ${#RUNNING_PODS[@]} == 0 ]]; then
      echo "Pod not found for $PROJECT"
      break
    fi
    if [[ $RAN == 'true' ]];then
      RUNNING_POD_INDEX=`shuf -i 1-${#RUNNING_PODS[@]} -n 1`
      break
    fi
    if [ ${#RUNNING_PODS[@]} -eq 1 ];then
      RUNNING_POD_INDEX=1
      break
    elif [ ${#RUNNING_PODS[@]} -gt 1 ];then
      echo $fg[green]'Running Pods:'$reset_color
      INDEX=1
      for i in $RUNNING_PODS;do
        echo "[$INDEX] $i"
        let INDEX=${INDEX}+1
      done
      echo $fg[green]'Select option of pod to execute:'$reset_color
      while true;do
        read RUNNING_POD_INDEX
        if [[ $RUNNING_POD_INDEX -gt 0 && $RUNNING_POD_INDEX -le ${#RUNNING_PODS[@]} ]];then
          break
        else
          echo 'invalid option...'
        fi
      done
      break
    fi
  done
  if [[ $RUNNING_POD_INDEX != -1 ]]; then
    echo "kubectl -c $KCONTEXT exec -it $RUNNING_PODS[$RUNNING_POD_INDEX] $finalopts"
    kubectl -n $NAMESPACE -c $KCONTEXT exec -it $RUNNING_PODS[$RUNNING_POD_INDEX] $finalopts
  fi
}

function klogs {
  KCONTEXT=${KCONTEXT:-gcloud}
  finalopts=()
  while [[ $@ != "" ]] do
    case $1 in
      -c)
        KCONTEXT="$2"
        shift; shift
        ;;
      --context=*)
        KCONTEXT="${i#*=}"
        shift
        ;;
      -p)
        PROJECT="$2"
        shift; shift
        ;;
      *)
        finalopts+=($1)
        shift
        ;;
    esac
  done

  kubectl -c $KCONTEXT logs -f deployment/$PROJECT --all-containers=true --since=5s --pod-running-timeout=2s $finalopts
}

function rgm {
  args=("${(@s/,/)1}")
  regex=${(j:.*\n.*:)args}
  echo "Rip Search with $regex..."
  rg -U $regex
}

function dc {
  if [ -e ./docker-compose-dev.yml ]; then
    docker_file_path=./docker-compose-dev.yml
  elif [ -e ./docker-compose.yml ]; then
    docker_file_path=./docker-compose.yml
  fi
  docker-compose -f $docker_file_path $@
}
function get_ip_of_ssh_hostname {
  ssh -G $1 | awk '/^hostname / { print $2  }'
}
unalias gc
unalias gcm
function gc {
  while true;do
    users=($GIT_USERS)
    for ((i=1; i<=${#users[@]}; i++)) do
      echo "$i | $users[$i]"
    done
    echo 'please input your number or name:'
    read input
    for ((i=1; i<=${#users[@]}; i++)) do
      if [[ $input == $users[$i] ]];then
        name=$input
        index=$i
        break 2
      elif [[ "$input" =~ '^[0-9]+$' ]];then
        if [[ $input -gt 0 && $input -le ${#users[@]} ]];then
          name=$users[$input]
          index=$input
          break 2
        fi
      fi
    done
    echo 'invalid option...'
  done
  emails=($GIT_EMAILS)
  git config user.name $name
  git config user.email $emails[$index]
  (git commit --verbose $*) || return
  (git commit --amend --author="$name <$emails[$index]>") || return
}
function gcm {
  (gc --message $*) || return
}