app.py

import os
import CloudFlare
import waitress
import flask
from logger import Logger


app = flask.Flask(__name__)

logger = Logger(__file__, __name__).get_logger()

logger.info('Starting Cloudflare DynDNS server...')
@app.route('/', methods=['GET'])
def main():
    token = flask.request.args.get('token')
    zone = flask.request.args.get('zone')
    record = flask.request.args.get('record')
    ipv4 = flask.request.args.get('ipv4')
    ipv6 = flask.request.args.get('ipv6')
    ipv6prefix = flask.request.args.get('ipv6prefix')
    cf = CloudFlare.CloudFlare(token=token)
    logger.info(f"request received: {flask.request.args}")

    if not token:
        logger.error('Missing token URL parameter.')
        return flask.jsonify({'status': 'error', 'message': 'Missing token URL parameter.'}), 400
    if not zone:
        logger.error('Missing zone URL parameter.')
        return flask.jsonify({'status': 'error', 'message': 'Missing zone URL parameter.'}), 400
    if not record:
        logger.error('Missing record URL parameter.')
        return flask.jsonify({'status': 'error', 'message': 'Missing record URL parameter.'}), 400
    if not ipv4 and not ipv6:
        logger.error('Missing ipv4 or ipv6 URL parameter.')
        return flask.jsonify({'status': 'error', 'message': 'Missing ipv4 or ipv6 URL parameter.'}), 400

    try:
        zones = cf.zones.get(params={'name': zone})

        if not zones:
            logger.error('Zone {} does not exist.'.format(zone))
            return flask.jsonify({'status': 'error', 'message': 'Zone {} does not exist.'.format(zone)}), 404

        a_record = cf.zones.dns_records.get(zones[0]['id'], params={
                                            'name': '{}.{}'.format(record, zone), 'match': 'all', 'type': 'A'})
        aaaa_record = cf.zones.dns_records.get(zones[0]['id'], params={
                                               'name': '{}.{}'.format(record, zone), 'match': 'all', 'type': 'AAAA'})
        txt_record = cf.zones.dns_records.get(zones[0]['id'], params={
                                                'name': '{}.{}'.format(record, zone), 'match': 'all', 'type': 'TXT'})

        if ipv4 is not None and not a_record:
            logger.error(f"A record for {record}.{zone} does not exist.")
            return flask.jsonify({'status': 'error', 'message': 'A record for {}.{} does not exist.'.format(record, zone)}), 404

        if ipv6 is not None and not aaaa_record:
            logger.error(f"AAAA record for {record}.{zone} does not exist.")
            return flask.jsonify({'status': 'error', 'message': 'AAAA record for {}.{} does not exist.'.format(record, zone)}), 404

        if ipv4 is not None and a_record[0]['content'] != ipv4:
            cf.zones.dns_records.put(zones[0]['id'], a_record[0]['id'], data={
                                     'name': a_record[0]['name'], 'type': 'A', 'content': ipv4, 'proxied': a_record[0]['proxied'], 'ttl': a_record[0]['ttl']})

        if ipv6 is not None and aaaa_record[0]['content'] != ipv6:
            cf.zones.dns_records.put(zones[0]['id'], aaaa_record[0]['id'], data={
                                     'name': aaaa_record[0]['name'], 'type': 'AAAA', 'content': ipv6, 'proxied': aaaa_record[0]['proxied'], 'ttl': aaaa_record[0]['ttl']})

        if ipv6prefix is not None:
            txt_record = cf.zones.dns_records.get(zones[0]['id'], params={
                                                  'name': '{}.{}'.format(record, zone), 'match': 'all', 'type': 'TXT'})
            if not txt_record:
                cf.zones.dns_records.post(zones[0]['id'], data={
                                          'name': '{}.{}'.format(record, zone), 'type': 'TXT', 'content': ipv6prefix, 'ttl': 120})
            elif txt_record[0]['content'] != ipv6prefix:
                cf.zones.dns_records.put(zones[0]['id'], txt_record[0]['id'], data={
                                         'name': txt_record[0]['name'], 'type': 'TXT', 'content': ipv6prefix, 'ttl': txt_record[0]['ttl']})

    except CloudFlare.exceptions.CloudFlareAPIError as e:
        logger.error(e)
        return flask.jsonify({'status': 'error', 'message': str(e)}), 500
    return flask.jsonify({'status': 'success', 'message': 'Update successful.'}), 200


@app.route('/healthz', methods=['GET'])
def healthz():
    return flask.jsonify({'status': 'success', 'message': 'OK'}), 200

app.secret_key = os.urandom(24)
waitress.serve(app, host='0.0.0.0', port=80)