#!/bin/bash
#
# pdns-tools
# https://git.stack-source.com/msb/pdns-tools
# Copyright (c) 2022 Matthew Saunders Brown <matthewsaundersbrown@gmail.com>
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)

# load include file
source $(dirname $0)/pdns.sh

help()
{
  echo "Replace Resource Record set in zone."
  echo ""
  echo "usage: $thisfilename -z <zone> -n <name> -t <type> -r <record> [-l <ttl>] [-s <status>] [-c <comment>] [-a <account>] [-h]"
  echo ""
  echo "  -h            Print this help."
  echo "  -z <zone>     Zone (domain name) to modify TTL."
  echo "  -l <ttl>      TTL to set all records to."
  echo
  echo "                Adjust TTL for all records of the given zone (domain name)."
  echo "                Typical ranges are from 300 (5 minutes) to 86400 (1 day)"
  echo "                with 'normal settings being 3600 (1 hour)."
}

/usr/bin/curl --silent --output /dev/null --write-out "%{http_code}" -H "X-API-Key: $api_key" "$api_base_url/zones/$zone?rrsets=false"

/usr/bin/curl --silent --output /dev/null --write-out "%{http_code}" -H "X-API-Key: aejahluadoolap6zootoochoib9Xooyi" "https://pdnsapi.jcihosting.net/api/v1/servers/localhost/zones/lamphost.com?rrsets=true"

 GET /servers/{server_id}/zones/{zone_id}¶



pdns:getoptions "$@"

# check for zone, make sure it ends with a .
if [[ -z $zone ]]; then
  echo "zone is required"
  exit
elif [[ $zone != *\. ]]; then
  zone="$zone."
fi

# check for name, make sure it ends with a .
if [[ -z $name ]]; then
  echo "name is required"
  exit
elif [[ $name = "@" ]]; then
  name=$zone
elif [[ $name != *\. ]]; then
  name="$name."
fi

# make sure name is equal to or part of zone
if [[ $name != $zone ]] && [[ $name != *\.$zone ]]; then
  name="$name$zone"
fi

# check for type
if [[ -z $type ]]; then
  echo "type is required"
  exit
fi

# check for record data
if [[ -z $record ]]; then
  echo "record is required"
  exit
fi

# check for ttl
if [[ -z $ttl ]]; then
  ttl=$zone_defaults_ttl
fi

# first query to see if zone already exists
zone_status=$(/usr/bin/curl --silent --output /dev/null --write-out "%{http_code}" -H "X-API-Key: $api_key" "$api_base_url/zones/$zone?rrsets=false")

if [[ $zone_status = 200 ]]; then
  # verified zone exists, add record(s)

  data="{\"rrsets\":[{\"name\":\"$name\",\"type\":\"$type\",\"ttl\":$ttl,\"changetype\":\"REPLACE\",\"records\":["

  # turn record in to array of records
  orig_ifs="$IFS"
  IFS='|'
  read -r -a resourcerecords <<< "$record"
  IFS="$orig_ifs"

  # get number of records in set
  resourcerecords_records_count=${#resourcerecords[@]}
  records_count=0

  for resourcerecord in "${resourcerecords[@]}"; do

    records_count=$((records_count+1))

    # make sure hostnames end in a .
    declare -a types_that_require_dot=("CNAME MX NS PTR SRV")
    if [[ " ${types_that_require_dot[*]} " =~ " ${type} " ]]; then
      if [[ $resourcerecord != *\. ]]; then
        resourcerecord="$resourcerecord."
      fi
    fi

    # quote TXT records
    if [[ $type = "TXT" ]]; then
      resourcerecord="\\\"$resourcerecord\\\""
    fi

    # set disabled status
    if [[ $status = 1 ]]; then
      disabled=true
    else
      disabled=false
    fi

    data="$data{\"content\":\"$resourcerecord\",\"disabled\":$disabled}"

    if [[ $records_count < $resourcerecords_records_count ]]; then
      data="$data,"
    else
      data="$data]"
    fi

  done

  # add comment, if set
  if [[ -n $comment ]]; then
    # set account to hostname if not specified with -a option
    if [[ -z $account ]]; then
      account=$(/usr/bin/hostname -f)
    fi
    data= "$data,\"comments\":[{\"content\":\"$comment\",\"account\":\"$account\"}]"
  fi

  # close out json string
  data="$data}]}"

  # add record(s)
  zone_status=$(/usr/bin/curl --silent --request PATCH --output /dev/null --write-out "%{http_code}" --header "X-API-Key: $api_key" --data "$data" "$api_base_url/zones/$zone")

  if [[ $zone_status = 204 ]]; then
    echo "Success. Record(s) for $zone created/updated."
  else
    echo "Error. http response updating record(s) for $zone was: $zone_status"
  fi

elif [[ $zone_status = 404 ]]; then

  echo "Zone $zone does not exist, can't update records."

else
  echo "Unexpected http response checking for Zone $zone: $zone_status"
fi