#!/usr/bin/env sh

#Here is a sample custom api script.
#This file name is "dns_dnsever.sh"
#So, here must be a method   dns_dnsever_add()
#Which will be called by acme.sh to add the txt record to your api system.
#returns 0 means success, otherwise error.
#
#Author: hiska
#Report Bugs here: https://github.com/hiskang/acme.sh
#
########  Public functions #####################

#Usage: dns_myapi_add   _acme-challenge.www.domain.com   "XKrxpRBosdIKFzxW_CT3KLZNf6q0HG9i01zxXp5CPBs"
dns_dnsever_add() {
  fulldomain=$1
  txtvalue=$2
  _info "Using dnsever"
  _debug fulldomain "$fulldomain"
  _debug txtvalue "$txtvalue"

  if [ -z "$DNSEVER_ID" ] || [ -z "$DNSEVER_PW" ]; then
    DNSEVER_ID=""
    DNSEVER_PW=""
    _err "You don't specify dnsever.com ID or PW yet."
    _err "Please create you key and try again."
    return 1
  fi

  #save the api key and email to the account conf file.
  _saveaccountconf DNSEVER_ID "$DNSEVER_ID"
  _saveaccountconf DNSEVER_PW "$DNSEVER_PW"

  dnsever_txt "add" "$DNSEVER_ID" "$DNSEVER_PW" "$fulldomain" "$txtvalue"
  return $?
}

#Usage: fulldomain txtvalue
#Remove the txt record after validation.
dns_dnsever_rm() {
  fulldomain=$1
  txtvalue=$2
  _info "Using dnsever"
  _debug fulldomain "$fulldomain"
  _debug txtvalue "$txtvalue"

  if [ -z "$DNSEVER_ID" ] || [ -z "$DNSEVER_PW" ]; then
    DNSEVER_ID=""
    DNSEVER_PW=""
    _err "You don't specify dnsever.com ID or PW yet."
    _err "Please create you key and try again."
    return 1
  fi

  #save the api key and email to the account conf file.
  _saveaccountconf DNSEVER_ID "$DNSEVER_ID"
  _saveaccountconf DNSEVER_PW "$DNSEVER_PW"

  dnsever_txt "delete" "$DNSEVER_ID" "$DNSEVER_PW" "$fulldomain" "$txtvalue"
  return $?
}

####################  Private functions below ##################################

dnsever_txt() {
  action="$1"
  login_id="$2"
  login_password="$3"
  fulldomain="$4"
  txt="$5"

  response=$(_post "login_id=$login_id&login_password=$login_password" "https://kr.dnsever.com/index.html")
  if [ $? != 0 ] || [ -z "$response" ]; then
    _err "dnsever_txt:$action ERROR login failed. Please check https://kr.dnsever.com/index.html with login_id=$login_id login_password=$login_password"
    return 1
  fi

  _H1="$(grep PHPSESSID "$HTTP_HEADER" | sed s/^Set-//)"
  export _H1

  response=$(_post "" "https://kr.dnsever.com/start.html")
  if [ $? != 0 ] || [ -z "$response" ]; then
    _err "dnsever_txt:$action ERROR login failed. Please check https://kr.dnsever.com/start.html after login"
    return 1
  fi

  if printf "%s\n" "$response" | grep "/confirm_email.html" >/dev/null; then
    response=$(_post "command=skipemail" "https://kr.dnsever.com/confirm_email.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR skipemail"
      return 1
    fi
    response=$(_post "" "https://kr.dnsever.com/start.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR login failed. Please check https://kr.dnsever.com/start.html after login"
      return 1
    fi
  fi

  skey=$(printf "%s\n" "$response" | _egrep_o "name=\"skey\" value=\".+\"" | cut -f3 -d= | tr -d \")
  _debug skey "$skey"

  if [ -z "$skey" ]; then
    _err "dnsever_txt:$action ERROR login failed with login_id=$login_id login_password=$login_password"
    response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
    return 1
  fi

  user_domain=$(dnsever_select_user_domain "$fulldomain" "$response")
  _debug user_domain "$user_domain"

  if [ -z "$user_domain" ]; then
    _err "dnsever_txt:$action ERROR no matching domain in DNSEver"
    response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
    return 1
  fi

  if [ "$action" = "add" ]; then

    subname=$(echo "$fulldomain" | sed "s/\.$user_domain\$//")
    _debug subname "$subname"

    if [ -z "$subname" ] || [ -z "$txt" ]; then
      _err "dnsever_txt ERROR subname=$subname or txt=$txt is empty"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    _info "dnsever_txt:$action skey=$skey user_domain=$user_domain selected_menu=edittxt command=add_txt subname=$subname txt=$txt"

    response=$(_post "skey=$skey&user_domain=$user_domain&selected_menu=edittxt" "https://kr.dnsever.com/start.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR failed to get TXT records from DNSEver"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    check=$(dnsever_check "$fulldomain" "$txt" "$response")
    if [ $? = 0 ] || [ -n "$check" ]; then
      _err "dnsever_txt:$action ERROR $fulldomain=$txt already exists"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    response=$(_post "skey=$skey&user_domain=$user_domain&selected_menu=edittxt&command=add_txt&subname=$subname&new_txt=$txt" "https://kr.dnsever.com/start.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR failed to add_text $fulldomain=$txt"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    check=$(dnsever_check "$fulldomain" "$txt" "$response")
    if [ $? != 0 ] || [ -z "$check" ]; then
      _err "dnsever_txt:$action ERROR failed to get newly added $fulldomain=$txt"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

  elif [ "$action" = "delete" ]; then

    response=$(_post "skey=$skey&user_domain=$user_domain&selected_menu=edittxt" "https://kr.dnsever.com/start.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR failed to get TXT records from DNSEver"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    check=$(dnsever_check "$fulldomain" "$txt" "$response")
    if [ $? != 0 ] || [ -z "$check" ]; then
      _err "dnsever_txt:$action ERROR $fulldomain=$txt does not exists"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    _info "dnsever_txt:$action skey=$skey user_domain=$user_domain selected_menu=edittxt command=delete_txt$(echo "$check" | sed 's/\&/ /g')"

    response=$(_post "skey=$skey&user_domain=$user_domain&selected_menu=edittxt&command=delete_txt&$check" "https://kr.dnsever.com/start.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR failed to delete $fulldomain=$txt from DNSEver"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    response=$(_post "skey=$skey&user_domain=$user_domain&selected_menu=edittxt" "https://kr.dnsever.com/start.html")
    if [ $? != 0 ] || [ -z "$response" ]; then
      _err "dnsever_txt:$action ERROR failed to get $fulldomain=$txt from DNSEver"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

    check=$(dnsever_check "$fulldomain" "$txt" "$response")
    if [ $? = 0 ] && [ -n "$check" ]; then
      _err "dnsever_txt:$action ERROR $fulldomain=$txt still exists"
      response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
      return 1
    fi

  else
    _err "dnsever_txt:$action action should be add or delete"
  fi

  response=$(_post "skey=$skey" "https://kr.dnsever.com/logout.php")
  return 0
}

dnsever_select_user_domain() {
  fulldomain="$1"
  response="$2"

  domains=$(printf "%s\n" "$response" | _egrep_o "<OPTION value=\".+\"" | cut -f2 -d= | tr -d \")
  _debug domains "$domains"

  nmax=0
  selected=""
  for domain in $domains; do
    if echo "$fulldomain" | grep -q "$domain\$"; then
      n=${#domain}
      if [ "$n" -gt $nmax ]; then
        nmax=$n
        selected="$domain"
      fi
    fi
  done
  echo "$selected"
}

dnsever_check() {
  fulldomain="$1"
  txtvalue="$2"
  response="$3"

  matched=$(printf "%s\n" "$response" | grep "$fulldomain" | _egrep_o "name=\"domain_for_txt_.+\" " | sed s/name=\"domain_for_txt_// | tr -d "\" ")
  _debug matched "$matched"

  check=""
  for n in $matched; do
    seq=$(printf "%s\n" "$response" | _egrep_o "name=\"seq_$n\" value=\".+\"" | cut -f3 -d= | tr -d \")
    old_txt=$(printf "%s\n" "$response" | _egrep_o "name=\"old_txt_$n\" value=\".+\" " | cut -f3 -d= | tr -d "\" ")
    if [ "$txtvalue" != "$old_txt" ]; then
      _info "dnsever_check skip seq=$seq fulldomain=$fulldomain due to old_txt=$old_txt is different from txtvalue=$txtvalue skip"
      continue
    fi
    check="${check}&check[]=$n&domain_for_txt_$n=$fulldomain&seq_$n=$seq&old_txt_$n=$old_txt"
  done

  _debug check "$check"
  if [ -z "$check" ]; then
    return 1
  fi

  echo "$check"
  return 0
}