dnsapi/dns_lua.sh

   1 #!/usr/bin/env sh
   2
   3 # bug reports to dev@1e.ca
   4
   5 #
   6 #LUA_Key="sdfsdfsdfljlbjkljlkjsdfoiwje"
   7 #
   8 #LUA_Email="user@luadns.net"
   9
  10 LUA_Api="https://api.luadns.com/v1"
  11 LUA_auth=$(printf "%s" "$LUA_Email:$LUA_Key" | _base64)
  12
  13 ########  Public functions #####################
  14
  15 #Usage: add  _acme-challenge.www.domain.com   "XKrxpRBosdIKFzxW_CT3KLZNf6q0HG9i01zxXp5CPBs"
  16 dns_lua_add() {
  17   fulldomain=$1
  18   txtvalue=$2
  19
  20   LUA_Key="${LUA_Key:-$(_readaccountconf_mutable LUA_Key)}"
  21   LUA_Email="${LUA_Email:-$(_readaccountconf_mutable LUA_Email)}"
  22   if [ -z "$LUA_Key" ] || [ -z "$LUA_Email" ]; then
  23     LUA_Key=""
  24     LUA_Email=""
  25     _err "You don't specify luadns api key and email yet."
  26     _err "Please create you key and try again."
  27     return 1
  28   fi
  29
  30   #save the api key and email to the account conf file.
  31   _saveaccountconf_mutable LUA_Key "$LUA_Key"
  32   _saveaccountconf_mutable LUA_Email "$LUA_Email"
  33
  34   _debug "First detect the root zone"
  35   if ! _get_root "$fulldomain"; then
  36     _err "invalid domain"
  37     return 1
  38   fi
  39   _debug _domain_id "$_domain_id"
  40   _debug _sub_domain "$_sub_domain"
  41   _debug _domain "$_domain"
  42
  43   _info "Adding record"
  44   if _LUA_rest POST "zones/$_domain_id/records" "{\"type\":\"TXT\",\"name\":\"$fulldomain.\",\"content\":\"$txtvalue\",\"ttl\":120}"; then
  45     if _contains "$response" "$fulldomain"; then
  46       _info "Added"
  47       #todo: check if the record takes effect
  48       return 0
  49     else
  50       _err "Add txt record error."
  51       return 1
  52     fi
  53   fi
  54 }
  55
  56 #fulldomain
  57 dns_lua_rm() {
  58   fulldomain=$1
  59   txtvalue=$2
  60
  61   LUA_Key="${LUA_Key:-$(_readaccountconf_mutable LUA_Key)}"
  62   LUA_Email="${LUA_Email:-$(_readaccountconf_mutable LUA_Email)}"
  63   _debug "First detect the root zone"
  64   if ! _get_root "$fulldomain"; then
  65     _err "invalid domain"
  66     return 1
  67   fi
  68   _debug _domain_id "$_domain_id"
  69   _debug _sub_domain "$_sub_domain"
  70   _debug _domain "$_domain"
  71
  72   _debug "Getting txt records"
  73   _LUA_rest GET "zones/${_domain_id}/records"
  74
  75   count=$(printf "%s\n" "$response" | _egrep_o "\"name\":\"$fulldomain.\",\"type\":\"TXT\"" | wc -l | tr -d " ")
  76   _debug count "$count"
  77   if [ "$count" = "0" ]; then
  78     _info "Don't need to remove."
  79   else
  80     record_id=$(printf "%s\n" "$response" | _egrep_o "\"id\":[^,]*,\"name\":\"$fulldomain.\",\"type\":\"TXT\"" | _head_n 1 | cut -d: -f2 | cut -d, -f1)
  81     _debug "record_id" "$record_id"
  82     if [ -z "$record_id" ]; then
  83       _err "Can not get record id to remove."
  84       return 1
  85     fi
  86     if ! _LUA_rest DELETE "/zones/$_domain_id/records/$record_id"; then
  87       _err "Delete record error."
  88       return 1
  89     fi
  90     _contains "$response" "$record_id"
  91   fi
  92 }
  93
  94 ####################  Private functions below ##################################
  95 #_acme-challenge.www.domain.com
  96 #returns
  97 # _sub_domain=_acme-challenge.www
  98 # _domain=domain.com
  99 # _domain_id=sdjkglgdfewsdfg
 100 _get_root() {
 101   domain=$1
 102   i=2
 103   p=1
 104   if ! _LUA_rest GET "zones"; then
 105     return 1
 106   fi
 107   while true; do
 108     h=$(printf "%s" "$domain" | cut -d . -f $i-100)
 109     _debug h "$h"
 110     if [ -z "$h" ]; then
 111       #not valid
 112       return 1
 113     fi
 114
 115     if _contains "$response" "\"name\":\"$h\""; then
 116       _domain_id=$(printf "%s\n" "$response" | _egrep_o "\"id\":[^,]*,\"name\":\"$h\"" | cut -d : -f 2 | cut -d , -f 1)
 117       _debug _domain_id "$_domain_id"
 118       if [ "$_domain_id" ]; then
 119         _sub_domain=$(printf "%s" "$domain" | cut -d . -f 1-$p)
 120         _domain="$h"
 121         return 0
 122       fi
 123       return 1
 124     fi
 125     p=$i
 126     i=$(_math "$i" + 1)
 127   done
 128   return 1
 129 }
 130
 131 _LUA_rest() {
 132   m=$1
 133   ep="$2"
 134   data="$3"
 135   _debug "$ep"
 136
 137   export _H1="Accept: application/json"
 138   export _H2="Authorization: Basic $LUA_auth"
 139   if [ "$m" != "GET" ]; then
 140     _debug data "$data"
 141     response="$(_post "$data" "$LUA_Api/$ep" "" "$m")"
 142   else
 143     response="$(_get "$LUA_Api/$ep")"
 144   fi
 145
 146   if [ "$?" != "0" ]; then
 147     _err "error $ep"
 148     return 1
 149   fi
 150   _debug2 response "$response"
 151   return 0
 152 }