<?php
use phpseclib\Net\SFTP;

require __DIR__ . '/vendor/autoload.php';

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

// Connect to the database
$dsn = "{$c['db_type']}:host={$c['db_host']};dbname={$c['db_database']}";
$options = [
    PDO::ATTR_ERRMODE            => PDO::ERRMODE_EXCEPTION,
    PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
    PDO::ATTR_EMULATE_PREPARES   => false,
];

try {
    $dbh = new PDO($dsn, $c['db_username'], $c['db_password'], $options);
} catch (PDOException $e) {
    die("Connection failed: " . $e->getMessage());
}

$domainCount = fetchCount($dbh, 'domain');
$hostCount = fetchCount($dbh, 'host');
$contactCount = fetchCount($dbh, 'contact');
$registrarCount = fetchCount($dbh, 'registrar');

// Fetching TLDs
$stmt = $dbh->query("SELECT id,tld FROM domain_tld;");
$tlds = $stmt->fetchAll();

// Fetching details from rde_escrow_deposits table
$stmt = $dbh->prepare("SELECT deposit_id, revision FROM rde_escrow_deposits;");
$stmt->execute();
$deposit_id = $stmt->fetch();

// Determine the next revision number
$nextRevisionNumber = is_null($deposit_id['deposit_id']) ? 1 : ($deposit_id['revision'] + 1);

// Format the revision number (001, 002, 003, ..., 010, ...)
$finalDepositId = str_pad($nextRevisionNumber, 3, '0', STR_PAD_LEFT);

foreach ($tlds as $tld) {
    $tldname = strtoupper(ltrim($tld['tld'], '.'));
    
    // Skip subdomains
    if (strpos($tldname, '.') !== false) {
        continue;
    }
    
    // Starting the XML for this TLD
    // Initializing XMLWriter
    $xml = new XMLWriter();
    $xml->openMemory();
    $xml->startDocument('1.0', 'UTF-8');
    $xml->startElementNS('rde', 'deposit', 'urn:ietf:params:xml:ns:rde-1.0');
    $xml->writeAttributeNS('xmlns', 'rdeHeader', null, 'urn:ietf:params:xml:ns:rdeHeader-1.0');
    $xml->writeAttributeNS('xmlns', 'rdeDom', null, 'urn:ietf:params:xml:ns:rdeDomain-1.0');
    $xml->writeAttributeNS('xmlns', 'rdeContact', null, 'urn:ietf:params:xml:ns:rdeContact-1.0');
    $xml->writeAttributeNS('xmlns', 'rdeHost', null, 'urn:ietf:params:xml:ns:rdeHost-1.0');
    $xml->writeAttributeNS('xmlns', 'rdeRegistrar', null, 'urn:ietf:params:xml:ns:rdeRegistrar-1.0');

    // Fetch domain details for this TLD
    $stmt = $dbh->prepare("SELECT * FROM domain WHERE tldid = :tldid;");
    $stmt->bindParam(':tldid', $tld['id']);
    $stmt->execute();
    $domains = $stmt->fetchAll();

    foreach ($domains as $domain) {
        $xml->startElement('rdeDom:domain');
        $xml->writeElement('rdeDom:name', $domain['name']);
        $xml->writeElement('rdeDom:roid', $domain['id']);
        $xml->writeElement('rdeDom:uName', $domain['name']);
        $xml->writeElement('rdeDom:idnTableId', 'Latn');

        // Fetch domain status
        $stmt = $dbh->prepare("SELECT * FROM domain_status WHERE domain_id = :domain_id;");
        $stmt->bindParam(':domain_id', $domain['id']);
        $stmt->execute();
        $status = $stmt->fetch();
        $xml->writeElement('rdeDom:status', $status['status'] ?? 'okk');

        $xml->writeElement('rdeDom:registrant', $domain['registrant']);

        // Fetch domain contacts
        $stmt = $dbh->prepare("SELECT * FROM domain_contact_map WHERE domain_id = :domain_id;");
        $stmt->bindParam(':domain_id', $domain['id']);
        $stmt->execute();
        $domain_contacts = $stmt->fetchAll();
        foreach ($domain_contacts as $contact) {
            $xml->startElement('rdeDom:contact');
            $xml->writeAttribute('type', $contact['type']);
            $xml->text($contact['contact_id']);
            $xml->endElement();  // Closing rdeDom:contact
        }

        // Fetch domain hosts and incorporate into XML
        $stmt = $dbh->prepare("SELECT host.name FROM domain_host_map JOIN host ON domain_host_map.host_id = host.id WHERE domain_host_map.domain_id = :domain_id;");
        $stmt->bindParam(':domain_id', $domain['id']);
        $stmt->execute();
        $domain_hosts = $stmt->fetchAll();
        $xml->startElement('rdeDom:ns');
        foreach ($domain_hosts as $host) {
            $xml->writeElement('domain:hostObj', $host['name']);
        }
        $xml->endElement();  // Closing rdeDom:ns

        $xml->writeElement('rdeDom:clID', $domain['clid']);
        $xml->writeElement('rdeDom:crRr', $domain['crid']);
        $xml->writeElement('rdeDom:crDate', date("Y-m-d\\TH:i:s.0\\Z", strtotime($domain['crdate'])));
        $xml->writeElement('rdeDom:exDate', date("Y-m-d\\TH:i:s.0\\Z", strtotime($domain['exdate'])));
        
        $xml->endElement();  // Closing rdeDom:domain
    }

    // Fetch and incorporate registrar details
    $stmt = $dbh->prepare("SELECT * FROM registrar;");
    $stmt->execute();
    $registrars = $stmt->fetchAll();

    $xml->startElement('rdeRegistrar:registrar');
    foreach ($registrars as $registrar) {
        $xml->writeElement('rdeRegistrar:id', $registrar['clid']);
        $xml->writeElement('rdeRegistrar:name', $registrar['name']);
        $xml->writeElement('rdeRegistrar:gurid', $registrar['iana_id']);
        $xml->writeElement('rdeRegistrar:status', 'ok');

        // Fetch and incorporate registrar contact details
        $stmt = $dbh->prepare("SELECT * FROM registrar_contact WHERE registrar_id = :registrar_id;");
        $stmt->bindParam(':registrar_id', $registrar['id']);
        $stmt->execute();
        $registrar_contacts = $stmt->fetchAll();

        foreach ($registrar_contacts as $contact) {
            $xml->startElement('rdeRegistrar:postalInfo');
            $xml->writeAttribute('type', 'int');
            $xml->startElement('rdeRegistrar:addr');
            $xml->writeElement('rdeRegistrar:street', $contact['street1']);
            $xml->writeElement('rdeRegistrar:city', $contact['city']);
            $xml->writeElement('rdeRegistrar:pc', $contact['pc']);
            $xml->writeElement('rdeRegistrar:cc', $contact['cc']);
            $xml->endElement();  // Closing rdeRegistrar:addr
            $xml->endElement();  // Closing rdeRegistrar:postalInfo
            
            $xml->writeElement('rdeRegistrar:voice', $contact['voice']);
            $xml->writeElement('rdeRegistrar:fax', $contact['fax']);
            $xml->writeElement('rdeRegistrar:email', $contact['email']);
        }

        $xml->writeElement('rdeRegistrar:url', $registrar['url']);
        $xml->startElement('rdeRegistrar:whoisInfo');
        $xml->writeElement('rdeRegistrar:name', $registrar['whois_server']);
        $xml->writeElement('rdeRegistrar:url', $registrar['whois_server']);
        $xml->endElement();  // Closing rdeRegistrar:whoisInfo

        $xml->writeElement('rdeRegistrar:crDate', date("Y-m-d\\TH:i:s.0\\Z", strtotime($registrar['crdate'])));
    }
    $xml->endElement();  // Closing rdeRegistrar:registrar

    // Fetch and incorporate host details
    $stmt = $dbh->prepare("SELECT * FROM host;");
    $stmt->execute();
    $hosts = $stmt->fetchAll();

    foreach ($hosts as $host) {
        $xml->startElement('rdeHost:host');
        $xml->writeElement('rdeHost:name', $host['name']);
        $xml->writeElement('rdeHost:roid', $host['id']);
        
        $xml->startElement('rdeHost:status');
        $xml->writeAttribute('s', 'ok');
        $xml->text('ok');
        $xml->endElement();  // Closing rdeHost:status
        
        $xml->writeElement('rdeHost:clID', $host['clid']);
        $xml->writeElement('rdeHost:crRr', $host['crid']);
        $xml->writeElement('rdeHost:crDate', date("Y-m-d\\TH:i:s.0\\Z", strtotime($host['crdate'])));
        $xml->endElement();  // Closing rdeHost:host
    }

    // Fetch and incorporate contact details
    $stmt = $dbh->prepare("SELECT * FROM contact;");
    $stmt->execute();
    $contacts = $stmt->fetchAll();

    foreach ($contacts as $contact) {
        $xml->startElement('rdeContact:contact');
        $xml->writeElement('rdeContact:id', $contact['identifier']);
        $xml->writeElement('rdeContact:roid', $contact['id']);
        $xml->startElement('rdeContact:status');
        $xml->writeAttribute('s', 'ok');
        $xml->text('ok');
        $xml->endElement();  // Closing rdeContact:status

        // Fetch postalInfo for the current contact
        $stmtPostal = $dbh->prepare("SELECT * FROM contact_postalInfo WHERE contact_id = :contact_id;");
        $stmtPostal->bindParam(':contact_id', $contact['id']);
        $stmtPostal->execute();
        $postalInfo = $stmtPostal->fetch();

        if ($postalInfo) {
            $xml->startElement('rdeContact:postalInfo');
            $xml->writeAttribute('type', 'int');
            $xml->writeElement('contact:name', $postalInfo['name']);
            $xml->writeElement('contact:org', $postalInfo['org']);
            $xml->startElement('contact:addr');
            $xml->writeElement('contact:street', $postalInfo['street1']);
            $xml->writeElement('contact:city', $postalInfo['city']);
            $xml->writeElement('contact:pc', $postalInfo['pc']);
            $xml->writeElement('contact:cc', $postalInfo['cc']);
            $xml->endElement();  // Closing contact:addr
            $xml->endElement();  // Closing rdeContact:postalInfo
        }

        $xml->writeElement('rdeContact:voice', $contact['voice']);
        $xml->writeElement('rdeContact:fax', $contact['fax']);
        $xml->writeElement('rdeContact:email', $contact['email']);
        $xml->writeElement('rdeContact:clID', $contact['clid']);
        $xml->writeElement('rdeContact:crRr', $contact['crid']);
        $xml->writeElement('rdeContact:crDate', date("Y-m-d\\TH:i:s.0\\Z", strtotime($contact['crdate'])));
        if (!empty($contact['upid'])) {
            $xml->writeElement('rdeContact:upRr', $contact['upid']);
            $xml->writeElement('rdeContact:upDate', date("Y-m-d\\TH:i:s.0\\Z", strtotime($contact['update'])));
        }
        $xml->endElement();  // Closing rdeContact:contact
    }
    
    // Writing the rdeHeader section to XML
    $xml->startElement('rdeIDN:idnTableRef');
    $xml->writeAttribute('id', 'Latn');
    $xml->writeElement('rdeIDN:url', 'https://namingo.org');
    $xml->writeElement('rdeIDN:urlPolicy', 'https://namingo.org');
    $xml->endElement();  // Closing rdeIDN:idnTableRef
    
    $xml->startElement('rdeHeader:header');
    $xml->writeElement('rdeHeader:tld', $tld['tld']);
    
    $xml->startElement('rdeHeader:count');
    $xml->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeDomain-1.0');
    $xml->text($domainCount);
    $xml->endElement();

    $xml->startElement('rdeHeader:count');
    $xml->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeHost-1.0');
    $xml->text($hostCount);
    $xml->endElement();

    $xml->startElement('rdeHeader:count');
    $xml->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeRegistrar-1.0');
    $xml->text($registrarCount);
    $xml->endElement();

    $xml->startElement('rdeHeader:count');
    $xml->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeContact-1.0');
    $xml->text($contactCount);
    $xml->endElement();

    $xml->startElement('rdeHeader:count');
    $xml->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeNNDN-1.0');
    $xml->text('0');
    $xml->endElement();

    $xml->startElement('rdeHeader:count');
    $xml->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeIDN-1.0');
    $xml->text('0');
    $xml->endElement();

    $xml->endElement();  // Closing rdeHeader:header
    
    $xml->endElement();  // Closing the 'rde:deposit' element
    $es = $xml->outputMemory();

    // Define the base name without the extension
    $baseFileName = "{$tldname}_".date('Y-m-d')."_full_S1_R{$finalDepositId}";

    // XML, tar, and gzip filenames
    $xmlFileName = $baseFileName . ".xml";
    $tarFileName = $baseFileName . ".tar";
    $gzipFileName = $baseFileName . ".tar.gz";

    // Save the main XML file
    file_put_contents($c['escrow_deposit_path']."/".$xmlFileName, $es, LOCK_EX);

    // Compress the XML file using tar
    $phar = new PharData($c['escrow_deposit_path']."/".$tarFileName);
    $phar->addFile($c['escrow_deposit_path']."/".$xmlFileName, $xmlFileName);

    // Compress the tar archive using gzip
    $phar->compress(Phar::GZ);

    // Delete the original tar file
    unlink($c['escrow_deposit_path']."/".$tarFileName);

    // Check if the $c['escrow_deleteXML'] variable is set to true and delete the original XML file
    if ($c['escrow_deleteXML']) {
        unlink($c['escrow_deposit_path']."/".$xmlFileName);
    }
    
    // Initialize a GnuPG instance
    $res = gnupg_init();

    // Get information about the public key from its content
    $publicKeyInfo = gnupg_import($res, file_get_contents($c['escrow_keyPath']));
    $fingerprint = $publicKeyInfo['fingerprint'];

    // Check if the key is already in the keyring
    $existingKeys = gnupg_keyinfo($res, $fingerprint);

    if (!$existingKeys) {
        // If not, import the public key
        gnupg_import($res, file_get_contents($c['escrow_keyPath']));
    }

    // Read the .tar.gz file contents
    $fileData = file_get_contents($c['escrow_deposit_path'] . "/" . $gzipFileName);
    
    // Add the encryption key
    gnupg_addencryptkey($res, $fingerprint);

    // Encrypt the file data using the public key
    $encryptedData = gnupg_encrypt($res, $fileData);

    if (!$encryptedData) {
        die("Error encrypting data: " . gnupg_geterror($res));
    }

    // Save the encrypted data to a new file
    file_put_contents($c['escrow_deposit_path'] . "/" . $baseFileName . ".ryde", $encryptedData);

    // Delete the original .tar.gz file
    unlink($c['escrow_deposit_path'] . "/" . $gzipFileName);
    
    $encryptedFilePath = $c['escrow_deposit_path'] . "/" . $baseFileName . ".ryde";
    
    // Initialize the GnuPG extension
    $gpg = new gnupg();
    $gpg->seterrormode(gnupg::ERROR_EXCEPTION); // throw exceptions on errors

    // Import your private key (if it's not already in the keyring)
    $privateKeyData = file_get_contents($c['escrow_privateKey']);
    $importResult = $gpg->import($privateKeyData);

    // Set the key to be used for signing
    $privateKeyId = $importResult['fingerprint'];
    $gpg->addsignkey($privateKeyId);
    
    // Specify the detached signature mode
    $gpg->setsignmode(GNUPG_SIG_MODE_DETACH);

    // Sign the encrypted data
    $encryptedData = file_get_contents($encryptedFilePath);
    $signature = $gpg->sign($encryptedData);

    // Save the signature to a .sig file
    $signatureFilePath = $c['escrow_deposit_path'] . '/' . pathinfo($encryptedFilePath, PATHINFO_FILENAME) . '.sig';
    file_put_contents($signatureFilePath, $signature);

    // Optionally, delete the encrypted file if you don't need it anymore
    // unlink($encryptedFilePath);
    
    // Start XMLWriter for the report
    $reportXML = new XMLWriter();
    $reportXML->openMemory();
    $reportXML->startDocument('1.0', 'UTF-8');

    $reportXML->startElement('rdeReport:report');
    $reportXML->writeAttribute('xmlns:rdeReport', 'urn:ietf:params:xml:ns:rdeReport-1.0');
    $reportXML->writeAttribute('xmlns:rdeHeader', 'urn:ietf:params:xml:ns:rdeHeader-1.0');

    $reportXML->writeElement('rdeReport:id', $finalDepositId);
    $reportXML->writeElement('rdeReport:version', '1');
    $reportXML->writeElement('rdeReport:rydeSpecEscrow', 'RFC8909');
    $reportXML->writeElement('rdeReport:rydeSpecMapping', 'RFC9022');
    $reportXML->writeElement('rdeReport:resend', '0');
    $reportXML->writeElement('rdeReport:crDate', date("Y-m-d\\TH:i:s.0\\Z"));
    $reportXML->writeElement('rdeReport:kind', 'FULL');
    $reportXML->writeElement('rdeReport:watermark', date('Y-m-d\\T00:00:00\\Z'));

    $reportXML->startElement('rdeHeader:header');
    $reportXML->writeElement('rdeHeader:tld', $tld['tld']);
    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeDomain-1.0');
    $reportXML->text($domainCount);
    $reportXML->endElement();

    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeHost-1.0');
    $reportXML->text($hostCount);
    $reportXML->endElement();

    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeContact-1.0');
    $reportXML->text($contactCount);
    $reportXML->endElement();

    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeRegistrar-1.0');
    $reportXML->text($registrarCount);
    $reportXML->endElement();

    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeIDN-1.0');
    $reportXML->text('0');
    $reportXML->endElement();

    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeNNDN-1.0');
    $reportXML->text('0');
    $reportXML->endElement();

    $reportXML->startElement('rdeHeader:count');
    $reportXML->writeAttribute('uri', 'urn:ietf:params:xml:ns:rdeEppParams-1.0');
    $reportXML->text('0');
    $reportXML->endElement();
    
    $reportXML->endElement(); // Closing rdeHeader:header
    $reportXML->endElement(); // Closing rdeReport:report

    $reps = $reportXML->outputMemory();

    // Save the report file
    $reportFilePath = $c['escrow_deposit_path']."/{$tldname}_".date('Y-m-d')."_full_R{$finalDepositId}.rep";
    file_put_contents($reportFilePath, $reps, LOCK_EX);

    if ($c['escrow_RDEupload']) {
        // Connect to the SFTP server
        $sftp = new SFTP($c['escrow_sftp_host']);

        // Login with username and password
        if (!$sftp->login($c['escrow_sftp_username'], $c['escrow_sftp_password'])) {
            die('Login failed');
        }

        // Define the remote directory where you want to upload the files
        $remoteDir = $c['escrow_sftp_remotepath'];

        // Upload the files
        $filesToUpload = [
            $encryptedFilePath,
            $signatureFilePath,
            $reportFilePath
        ];

        foreach ($filesToUpload as $filePath) {
            $remoteFile = $remoteDir . basename($filePath);
            if (!$sftp->put($remoteFile, $filePath, SFTP::SOURCE_LOCAL_FILE)) {
                echo "Failed to upload " . basename($filePath) . "\n";
            } else {
                echo "Successfully uploaded " . basename($filePath) . "\n";
            }
        }
        
        $reportFileData = file_get_contents($reportFilePath);

        $ch = curl_init();

        // Set cURL options
        curl_setopt($ch, CURLOPT_URL, $c['escrow_report_url']);
        curl_setopt($ch, CURLOPT_USERPWD, $c['escrow_report_username'].':'.$c['escrow_report_password']);
        curl_setopt($ch, CURLOPT_CUSTOMREQUEST, 'PUT');
        curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
        curl_setopt($ch, CURLOPT_POSTFIELDS, $reportFileData);
        curl_setopt($ch, CURLOPT_HTTPHEADER, array(
            'Content-Type: application/octet-stream',
            'Content-Length: ' . strlen($reportFileData)
        ));

        $response = curl_exec($ch);

        if ($response === false) {
            die('Error occurred: ' . curl_error($ch));
        }

        $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
        if ($httpCode >= 200 && $httpCode < 300) {
            echo "File uploaded successfully.\n";
        } else {
            echo "Failed to upload file. HTTP Status Code: " . $httpCode . "\n";
        }

        curl_close($ch);
        
    }

}