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   if [ -z "$LUA_Key" ] || [ -z "$LUA_Email" ]; then
  21     LUA_Key=""
  22     LUA_Email=""
  23     _err "You don't specify luadns api key and email yet."
  24     _err "Please create you key and try again."
  25     return 1
  26   fi
  27
  28   #save the api key and email to the account conf file.
  29   _saveaccountconf LUA_Key "$LUA_Key"
  30   _saveaccountconf LUA_Email "$LUA_Email"
  31
  32   _debug "First detect the root zone"
  33   if ! _get_root "$fulldomain"; then
  34     _err "invalid domain"
  35     return 1
  36   fi
  37   _debug _domain_id "$_domain_id"
  38   _debug _sub_domain "$_sub_domain"
  39   _debug _domain "$_domain"
  40
  41   _debug "Getting txt records"
  42   _LUA_rest GET "zones/${_domain_id}/records"
  43
  44   if ! _contains "$response" "\"id\":"; then
  45     _err "Error"
  46     return 1
  47   fi
  48
  49   count=$(printf "%s\n" "$response" | _egrep_o "\"name\":\"$fulldomain.\",\"type\":\"TXT\"" | wc -l | tr -d " ")
  50   _debug count "$count"
  51   if [ "$count" = "0" ]; then
  52     _info "Adding record"
  53     if _LUA_rest POST "zones/$_domain_id/records" "{\"type\":\"TXT\",\"name\":\"$fulldomain.\",\"content\":\"$txtvalue\",\"ttl\":120}"; then
  54       if _contains "$response" "$fulldomain"; then
  55         _info "Added"
  56         #todo: check if the record takes effect
  57         return 0
  58       else
  59         _err "Add txt record error."
  60         return 1
  61       fi
  62     fi
  63     _err "Add txt record error."
  64   else
  65     _info "Updating record"
  66     record_id=$(printf "%s\n" "$response" | _egrep_o "\"id\":[^,]*,\"name\":\"$fulldomain.\",\"type\":\"TXT\"" | _head_n 1 | cut -d: -f2 | cut -d, -f1)
  67     _debug "record_id" "$record_id"
  68
  69     _LUA_rest PUT "zones/$_domain_id/records/$record_id" "{\"id\":$record_id,\"type\":\"TXT\",\"name\":\"$fulldomain.\",\"content\":\"$txtvalue\",\"zone_id\":$_domain_id,\"ttl\":120}"
  70     if [ "$?" = "0" ] && _contains "$response" "updated_at"; then
  71       _info "Updated!"
  72       #todo: check if the record takes effect
  73       return 0
  74     fi
  75     _err "Update error"
  76     return 1
  77   fi
  78
  79 }
  80
  81 #fulldomain
  82 dns_lua_rm() {
  83   fulldomain=$1
  84   txtvalue=$2
  85   _debug "First detect the root zone"
  86   if ! _get_root "$fulldomain"; then
  87     _err "invalid domain"
  88     return 1
  89   fi
  90   _debug _domain_id "$_domain_id"
  91   _debug _sub_domain "$_sub_domain"
  92   _debug _domain "$_domain"
  93
  94   _debug "Getting txt records"
  95   _LUA_rest GET "zones/${_domain_id}/records"
  96
  97   count=$(printf "%s\n" "$response" | _egrep_o "\"name\":\"$fulldomain.\",\"type\":\"TXT\"" | wc -l | tr -d " ")
  98   _debug count "$count"
  99   if [ "$count" = "0" ]; then
 100     _info "Don't need to remove."
 101   else
 102     record_id=$(printf "%s\n" "$response" | _egrep_o "\"id\":[^,]*,\"name\":\"$fulldomain.\",\"type\":\"TXT\"" | _head_n 1 | cut -d: -f2 | cut -d, -f1)
 103     _debug "record_id" "$record_id"
 104     if [ -z "$record_id" ]; then
 105       _err "Can not get record id to remove."
 106       return 1
 107     fi
 108     if ! _LUA_rest DELETE "/zones/$_domain_id/records/$record_id"; then
 109       _err "Delete record error."
 110       return 1
 111     fi
 112     _contains "$response" "$record_id"
 113   fi
 114 }
 115
 116 ####################  Private functions below ##################################
 117 #_acme-challenge.www.domain.com
 118 #returns
 119 # _sub_domain=_acme-challenge.www
 120 # _domain=domain.com
 121 # _domain_id=sdjkglgdfewsdfg
 122 _get_root() {
 123   domain=$1
 124   i=2
 125   p=1
 126   if ! _LUA_rest GET "zones"; then
 127     return 1
 128   fi
 129   while true; do
 130     h=$(printf "%s" "$domain" | cut -d . -f $i-100)
 131     _debug h "$h"
 132     if [ -z "$h" ]; then
 133       #not valid
 134       return 1
 135     fi
 136
 137     if _contains "$response" "\"name\":\"$h\""; then
 138       _domain_id=$(printf "%s\n" "$response" | _egrep_o "\"id\":[^,]*,\"name\":\"$h\"" | cut -d : -f 2 | cut -d , -f 1)
 139       _debug _domain_id "$_domain_id"
 140       if [ "$_domain_id" ]; then
 141         _sub_domain=$(printf "%s" "$domain" | cut -d . -f 1-$p)
 142         _domain="$h"
 143         return 0
 144       fi
 145       return 1
 146     fi
 147     p=$i
 148     i=$(_math "$i" + 1)
 149   done
 150   return 1
 151 }
 152
 153 _LUA_rest() {
 154   m=$1
 155   ep="$2"
 156   data="$3"
 157   _debug "$ep"
 158
 159   export _H1="Accept: application/json"
 160   export _H2="Authorization: Basic $LUA_auth"
 161   if [ "$m" != "GET" ]; then
 162     _debug data "$data"
 163     response="$(_post "$data" "$LUA_Api/$ep" "" "$m")"
 164   else
 165     response="$(_get "$LUA_Api/$ep")"
 166   fi
 167
 168   if [ "$?" != "0" ]; then
 169     _err "error $ep"
 170     return 1
 171   fi
 172   _debug2 response "$response"
 173   return 0
 174 }