<?php

use WHMCS\User\User;

if (!defined("WHMCS")) {
    die("This file cannot be accessed directly");
}

/**
 * You'll need to configure the following constrants in your configuration.php file.
 * This is only temporary and will be replaced with pulling the token from the database in the future.
 *
 * const VIRTFUSION_API_URL = "https://your-virtfusion-url.com/api/v1";
 *
 * You can create a token in the VirtFusion control panel under System > API.
 *
 * const VIRT_TOKEN = "your-virtfusion-token";
 */

/**
 * If the constants are not defined, return null to prevent errors.
 */
if (!defined("VIRTFUSION_API_URL") || !defined("VIRT_TOKEN")) {
    return null;
}

if (!function_exists('fetchPackageId')) {
    /**
     * @param string $packageName
     * @return int|null
     * @throws JsonException
     */
    function fetchPackageId(string $packageName): ?int
    {
        $url = sprintf("%s/packages", VIRTFUSION_API_URL);

        $packages = makeRequest($url);

        foreach ($packages['data'] as $package) {
            if ($package['name'] === $packageName && $package['enabled'] === true) {
                return $package['id'];
            }
        }

        return null;
    }
}

if (!function_exists('fetchTemplates')) {
    /**
     * @param int $serverPackageId
     * @return array|null
     * @throws JsonException
     */
    function fetchTemplates(int $serverPackageId): ?array
    {
        $url = sprintf("%s/media/templates/fromServerPackageSpec/%d", VIRTFUSION_API_URL, $serverPackageId);

        return makeRequest($url);
    }
}

if (!function_exists('custom_os_templates_hook')) {
    /**
     * @param array $vars
     * @return array|null[]
     */
    function custom_os_templates_hook(array $vars): array
    {
        try {
            $serverPackageId = fetchPackageId($vars['productinfo']['name']); // Replace with the appropriate server package ID

            if ($serverPackageId === null) {
                return [
                    'templates' => null,
                ];
            }

            $templates = fetchTemplates($serverPackageId);

            // Assign the generated dropdown menu to a Smarty template variable
            return [
                'templates' => $templates,
            ];
        } catch (JsonException $e) {
            return [
                'templates' => null,
            ];
        }
    }
}

if (!function_exists('get_vf_user_details')) {
    /**
     * @param int $id
     * @return array
     * @throws JsonException
     */
    function get_vf_user_details(int $id): ?array {
        $url = sprintf("%s/users/%s/byExtRelation", VIRTFUSION_API_URL, $id);
        $response = makeRequest($url);

        if (isset($response['msg']) && $response['msg'] === "ext_relation_id not found") {
            return null;
        }

        return $response['data'];
    }
}

if (!function_exists('makeRequest')) {
    /**
     * @param string $url
     * @return mixed
     * @throws JsonException
     * @throws Exception
     */
    function makeRequest(string $url): array
    {
        $curl = curl_init();
        curl_setopt_array($curl, array(
            CURLOPT_URL => $url,
            CURLOPT_RETURNTRANSFER => true,
            CURLOPT_HTTPHEADER => array(
                sprintf("Authorization: Bearer %s", VIRT_TOKEN)
            )
        ));

        $response = curl_exec($curl);
        $err = curl_error($curl);
        curl_close($curl);

        if ($err) {
            throw new Exception("cURL Error: " . $err);
        }

        return json_decode($response, true, 512, JSON_THROW_ON_ERROR);
    }
}

if (!function_exists('get_users_ssh_keys')) {
    /**
     * @throws JsonException
     */
    function get_users_ssh_keys(?User $user): ?array {
        if (is_null($user)) {
            return null;
        }

        $vfUser = get_vf_user_details($user['id']);

        $url = sprintf("%s/ssh_keys/user/%s", VIRTFUSION_API_URL, $vfUser['id']);

        $response = makeRequest($url);

        return $response;
    }
}

if (!function_exists('add_hook_os_templates')) {
    /**
     * @param array $vars
     * @return array|null
     * @throws JsonException
     */
    function add_hook_os_templates(array $vars): ?array
    {
        if (!isset($vars['productinfo']['module']) || $vars['productinfo']['module'] !== 'VirtFusionDirect') {
            return null;
        }

        $templates_data = custom_os_templates_hook($vars)['templates'];

        if (empty($templates_data)) {
            return null;
        }

        $dropdownOptions = [];
        foreach ($templates_data['data'] as $osCategory) {
            foreach ($osCategory['templates'] as $template) {
                $optionValue = $template['id'];
                $optionLabel = $template['name'] . " " . $template['version'] . " " . $template['variant'];
                $dropdownOptions[] = ['id' => $optionValue, 'name' => $optionLabel];
            }
        }

        // Sort dropdownOptions alphabetically by the 'name' key
        usort($dropdownOptions, function ($a, $b) {
            return strcmp($a['name'], $b['name']);
        });

        $osTemplates = [
            'id' => 'os_template',
            'optionname' => 'Initial Operating System',
            'optiontype' => 1,
            'options' => $dropdownOptions,
            'selectedvalue' => ''
        ];

        $sshKeys = get_users_ssh_keys($vars['loggedinuser']);

        $sshKeysOptions = [
            'id' => 'ssh_key',
            'optionname' => 'Initial SSH Key',
            'optiontype' => 1,
            'options' => array_map(function ($sshKey) {
                if ($sshKey['enabled'] === false) {
                    return null;
                }

                return [
                    'id' => $sshKey['id'],
                    'name' => $sshKey['name']
                ];
            }, $sshKeys['data']),
            'selectedvalue' => ''
        ];

        $configurableoptions = $vars['configurableoptions'];
        array_push(
            $configurableoptions,
            $osTemplates,
            $sshKeysOptions
        );

        return [
            'configurableoptions' => $configurableoptions,
        ];
    }

    add_hook('ClientAreaPageCart', 1, 'add_hook_os_templates');
}