<?php

require_once 'vendor/autoload.php';

use Badcow\DNS\Zone;
use Badcow\DNS\Rdata\Factory;
use Badcow\DNS\ResourceRecord;
use Badcow\DNS\Classes;
use Badcow\DNS\ZoneBuilder;
use Badcow\DNS\AlignedBuilder;

$c = require_once 'config.php';
require_once 'helpers.php';

$logFilePath = '/var/log/namingo/write_zone.log';
$log = setupLogger($logFilePath, 'Zone_Generator');
$log->info('job started.');

use Swoole\Coroutine;

// Initialize the PDO connection pool
$pool = new Swoole\Database\PDOPool(
    (new Swoole\Database\PDOConfig())
        ->withDriver($c['db_type'])
        ->withHost($c['db_host'])
        ->withPort($c['db_port'])
        ->withDbName($c['db_database'])
        ->withUsername($c['db_username'])
        ->withPassword($c['db_password'])
        ->withCharset('utf8mb4')
);

Swoole\Runtime::enableCoroutine();

// Creating first coroutine
Coroutine::create(function () use ($pool, $log, $c) {
    try {
        $pdo = $pool->get();
        $sth = $pdo->prepare('SELECT id, tld FROM domain_tld');
        $sth->execute();
        $serial = generateSerial($c['dns_serial'] ?? 1);

        $tlds = [];

        while (list($id, $tld) = $sth->fetch(PDO::FETCH_NUM)) {
            $tldRE = preg_quote($tld, '/');
            $cleanedTld = ltrim(strtolower($tld), '.');
            $zone = new Zone($cleanedTld.'.');
            $zone->setDefaultTtl(3600);

            $soa = new ResourceRecord;
            $soa->setName('@');
            $soa->setClass(Classes::INTERNET);
            $soa->setRdata(Factory::Soa(
                $c['ns']['ns1'] . '.',
                $c['dns_soa'] . '.',
                $serial, 
                900, 
                1800, 
                3600000, 
                3600
            ));
            $zone->addResourceRecord($soa);

            foreach ($c['ns'] as $ns) {
                $nsRecord = new ResourceRecord;
                $nsRecord->setName($cleanedTld . '.');
                $nsRecord->setClass(Classes::INTERNET);
                $nsRecord->setRdata(Factory::Ns($ns . '.'));
                $zone->addResourceRecord($nsRecord);
            }

            // Include custom records if the file exists
            $customRecordsFile = "/opt/registry/automation/{$cleanedTld}.php";

            if (file_exists($customRecordsFile)) {
                $log->info("Loading custom records for {$cleanedTld} from {$customRecordsFile}.");
                $customRecords = require $customRecordsFile;

                foreach ($customRecords as $record) {
                    try {
                        $customRecord = new ResourceRecord;
                        $customRecord->setName($record['name']);
                        $customRecord->setClass(Classes::INTERNET);
                        $customRecord->setRdata(Factory::{$record['type']}(...$record['parameters']));
                        $zone->addResourceRecord($customRecord);
                    } catch (Throwable $e) {
                        $log->error("Failed to add custom record for {$cleanedTld}: " . $e->getMessage());
                    }
                }
            }

            // Fetch domains for this TLD
            $sthDomains = $pdo->prepare('SELECT DISTINCT domain.id, domain.name FROM domain WHERE tldid = :id AND (exdate > CURRENT_TIMESTAMP OR rgpstatus = \'pendingRestore\') ORDER BY domain.name');

            $domainIds = [];
            $sthDomains->execute([':id' => $id]);
            while ($row = $sthDomains->fetch(PDO::FETCH_ASSOC)) {
                $domainIds[] = $row['id'];
            }

            $statuses = [];
            if (count($domainIds) > 0) {
                $placeholders = implode(',', array_fill(0, count($domainIds), '?'));
                $sthStatus = $pdo->prepare("SELECT domain_id, id FROM domain_status WHERE domain_id IN ($placeholders) AND status LIKE '%Hold'");
                $sthStatus->execute($domainIds);
                while ($row = $sthStatus->fetch(PDO::FETCH_ASSOC)) {
                    $statuses[$row['domain_id']] = $row['id'];
                }
            }

            $sthDomains->execute([':id' => $id]);

            while (list($did, $dname) = $sthDomains->fetch(PDO::FETCH_NUM)) {
                if (isset($statuses[$did])) continue;

                $dname_clean = $dname;
                $dname_clean = ($dname_clean == "$tld.") ? '@' : $dname_clean;

                // NS records for the domain
                $sthNsRecords = $pdo->prepare('SELECT DISTINCT host.name FROM domain_host_map INNER JOIN host ON domain_host_map.host_id = host.id WHERE domain_host_map.domain_id = :did');
                $sthNsRecords->execute([':did' => $did]);
                while (list($hname) = $sthNsRecords->fetch(PDO::FETCH_NUM)) {
                    $nsRecord = new ResourceRecord;
                    $nsRecord->setName($dname_clean . '.');
                    $nsRecord->setClass(Classes::INTERNET);
                    $nsRecord->setRdata(Factory::Ns($hname . '.'));
                    $zone->addResourceRecord($nsRecord);
                }

                // A/AAAA records for the domain
                $sthHostRecords = $pdo->prepare("SELECT host.name, host_addr.ip, host_addr.addr FROM host INNER JOIN host_addr ON host.id = host_addr.host_id WHERE host.domain_id = :did ORDER BY host.name");
                $sthHostRecords->execute([':did' => $did]);
                while (list($hname, $type, $addr) = $sthHostRecords->fetch(PDO::FETCH_NUM)) {
                    $hname_clean = $hname;
                    $hname_clean = ($hname_clean == "$tld.") ? '@' : $hname_clean;
                    $record = new ResourceRecord;
                    $record->setName($hname_clean . '.');
                    $record->setClass(Classes::INTERNET);

                    if ($type == 'v4') {
                        $record->setRdata(Factory::A($addr));
                    } else {
                        $record->setRdata(Factory::AAAA($addr));
                    }

                    $zone->addResourceRecord($record);
                }

                // DS records for the domain
                $sthDS = $pdo->prepare("SELECT keytag, alg, digesttype, digest FROM secdns WHERE domain_id = :did");
                $sthDS->execute([':did' => $did]);
                while (list($keytag, $alg, $digesttype, $digest) = $sthDS->fetch(PDO::FETCH_NUM)) {
                    $dsRecord = new ResourceRecord;
                    $dsRecord->setName($dname_clean . '.');
                    $dsRecord->setClass(Classes::INTERNET);
                    $dsRecord->setRdata(Factory::Ds($keytag, $alg, hex2bin($digest), $digesttype));
                    $zone->addResourceRecord($dsRecord);
                }
            }

            if (isset($c['zone_mode']) && $c['zone_mode'] === 'nice') {
                $builder = new AlignedBuilder();
            } else {
                $builder = new ZoneBuilder();
            }
            $completed_zone = $builder->build($zone);

            if ($c['dns_server'] == 'bind') {
                $basePath = '/var/lib/bind';
            } elseif ($c['dns_server'] == 'nsd') {
                $basePath = '/etc/nsd';
            } elseif ($c['dns_server'] == 'knot') {
                $basePath = '/etc/knot/zones';
            } else {
                // Default path
                $basePath = '/var/lib/bind';
            }

            file_put_contents("{$basePath}/{$cleanedTld}.zone", $completed_zone);

            $tlds[] = $cleanedTld;
        }

        foreach ($tlds as $cleanedTld) {
            if ($c['dns_server'] == 'bind') {
                exec("rndc reload {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to reload BIND for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }

                exec("rndc notify {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to notify secondary servers for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }
            } elseif ($c['dns_server'] == 'nsd') {
                exec("nsd-control reload 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to reload NSD. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }
            } elseif ($c['dns_server'] == 'knot') {
                exec("knotc reload 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to reload Knot DNS. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }

                exec("knotc zone-notify {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to notify secondary servers for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }
            } elseif ($c['dns_server'] == 'opendnssec') {
                chown("{$basePath}/{$cleanedTld}.zone", 'opendnssec');
                chgrp("{$basePath}/{$cleanedTld}.zone", 'opendnssec');

                exec("ods-signer sign {$cleanedTld} 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to sign zone with OpenDNSSEC for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }
                sleep(1);
                copy("/var/lib/opendnssec/signed/{$cleanedTld}", "/var/lib/bind/{$cleanedTld}.zone.signed");

                exec("rndc reload {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to reload BIND for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }

                exec("rndc notify {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to notify secondary servers for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }
            } else {
                // Default
                exec("rndc reload {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to reload BIND for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }

                exec("rndc notify {$cleanedTld}. 2>&1", $output, $return_var);
                if ($return_var != 0) {
                    $log->error("Failed to notify secondary servers for {$cleanedTld}. Output: " . implode(" ", $output) . " Code: " . $return_var);
                }
            }
        }

        $log->info('job finished successfully.');
    } catch (PDOException $e) {
        $log->error('Database error: ' . $e->getMessage());
    } catch (Throwable $e) {
        $log->error('Error: ' . $e->getMessage());
    } finally {
        // Return the connection to the pool
        $pool->put($pdo);
    }
});