First version of the hooks.

2023-09-10 14:04:08 -04:00
parent 9558142277
commit 28b9761223
1 changed files with 176 additions and 0 deletions
--- a/hooks.php
+++ b/hooks.php
@@ -1 +1,177 @@
 <?php
+
+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);
+
+        $curl = curl_init();
+        curl_setopt_array($curl, [
+            CURLOPT_URL => $url,
+            CURLOPT_RETURNTRANSFER => true,
+            CURLOPT_HTTPHEADER => [
+                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);
+        }
+
+        $packages = json_decode($response, true, 512, JSON_THROW_ON_ERROR);
+
+        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
+     * @throws Exception
+     */
+    function fetchTemplates(int $serverPackageId): ?array
+    {
+        $url = sprintf("%s/media/templates/fromServerPackageSpec/%d", VIRTFUSION_API_URL, $serverPackageId);
+
+        $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('custom_os_templates_hook')) {
+    /**
+     * @param array $vars
+     * @return array|null[]
+     * @throws Exception
+     */
+    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('add_hook_os_templates')) {
+    /**
+     * @param array $vars
+     * @return array|null
+     * @throws Exception
+     */
+    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']);
+        });
+
+        $newOption = [
+            'id' => 'os_template',
+            'optionname' => 'Initial Operating System',
+            'optiontype' => 1,
+            'options' => $dropdownOptions,
+            'selectedvalue' => ''
+        ];
+        $configurableoptions = $vars['configurableoptions'];
+        $configurableoptions[] = $newOption;
+
+        return [
+            'configurableoptions' => $configurableoptions,
+        ];
+    }
+
+    add_hook('ClientAreaPageCart', 1, 'add_hook_os_templates');
+}
+
+