<?php
require_once __DIR__ . '/../vendor/autoload.php';

use Utopia\Database\Document;
use Utopia\Database\Database;
use Utopia\Database\Validator\Authorization;
use Utopia\Database\Validator\UID;
use Appwrite\Event\Event;
use Appwrite\Utopia\Response\Model\Execution;
use Appwrite\Messaging\Adapter\Realtime;
use Appwrite\Stats\Stats;
use Utopia\App;
use Utopia\Swoole\Request;
use Appwrite\Utopia\Response;
use Utopia\CLI\Console;
use Swoole\Process;
use Swoole\Http\Server;
use Swoole\Http\Request as SwooleRequest;
use Swoole\Http\Response as SwooleResponse;
use Utopia\Orchestration\Orchestration;
use Utopia\Database\Adapter\MariaDB;
use Utopia\Cache\Adapter\Redis as RedisCache;
use Utopia\Config\Config;
use Utopia\Validator\ArrayList;
use Utopia\Validator\JSON;
use Utopia\Validator\Text;
use Cron\CronExpression;
use Utopia\Storage\Device\Local;
use Utopia\Storage\Storage;
use Swoole\Coroutine as Co;
use Utopia\Cache\Cache;
use Utopia\Database\Query;
use Utopia\Orchestration\Adapter\DockerCLI;
use Utopia\Validator\Boolean;
use Utopia\Logger\Log;

require_once __DIR__ . '/init.php';

global $register;
$logError = function(Throwable $error, string $action, Utopia\Route $route = null) use ($register) {
    $logger = $register->get('logger');

    if($logger) {
        $version = App::getEnv('_APP_VERSION', 'UNKNOWN');

        $log = new Log();
        $log->setNamespace("executor");
        $log->setServer(\gethostname());
        $log->setVersion($version);
        $log->setType(Log::TYPE_ERROR);
        $log->setMessage($error->getMessage());

        if($route) {
            $log->addTag('method', $route->getMethod());
            $log->addTag('url',  $route->getPath());
        }

        $log->addTag('code', $error->getCode());
        $log->addTag('verboseType', get_class($error));

        $log->addExtra('file', $error->getFile());
        $log->addExtra('line', $error->getLine());
        $log->addExtra('trace', $error->getTraceAsString());

        $log->setAction($action);

        $isProduction = App::getEnv('_APP_ENV', 'development') === 'production';
        $log->setEnvironment($isProduction ? Log::ENVIRONMENT_PRODUCTION : Log::ENVIRONMENT_STAGING);

        $responseCode = $logger->addLog($log);
        Console::info('Executor log pushed with status code: '.$responseCode);
    }

    Console::error('[Error] Type: ' . get_class($error));
    Console::error('[Error] Message: ' . $error->getMessage());
    Console::error('[Error] File: ' . $error->getFile());
    Console::error('[Error] Line: ' . $error->getLine());
};

try {
    $dockerUser = App::getEnv('DOCKERHUB_PULL_USERNAME', null);
    $dockerPass = App::getEnv('DOCKERHUB_PULL_PASSWORD', null);
    $dockerEmail = App::getEnv('DOCKERHUB_PULL_EMAIL', null);
    $orchestration = new Orchestration(new DockerCLI($dockerUser, $dockerPass));

    $runtimes = Config::getParam('runtimes');

    Swoole\Runtime::enableCoroutine(true, SWOOLE_HOOK_ALL);

    // Warmup: make sure images are ready to run fast 🚀
    Co\run(function () use ($runtimes, $orchestration) {
        foreach ($runtimes as $runtime) {
            go(function () use ($runtime, $orchestration) {
                Console::info('Warming up ' . $runtime['name'] . ' ' . $runtime['version'] . ' environment...');

                $response = $orchestration->pull($runtime['image']);

                if ($response) {
                    Console::success("Successfully Warmed up {$runtime['name']} {$runtime['version']}!");
                } else {
                    Console::warning("Failed to Warmup {$runtime['name']} {$runtime['version']}!");
                }
            });
        }
    });

    $activeFunctions = new Swoole\Table(1024);
    $activeFunctions->column('id', Swoole\Table::TYPE_STRING, 512);
    $activeFunctions->column('name', Swoole\Table::TYPE_STRING, 512);
    $activeFunctions->column('status', Swoole\Table::TYPE_STRING, 512);
    $activeFunctions->column('key', Swoole\Table::TYPE_STRING, 4096);
    $activeFunctions->create();

    Co\run(function () use ($orchestration, $activeFunctions) {
        $executionStart = \microtime(true);

        $residueList = $orchestration->list(['label' => 'appwrite-type=function']);

        foreach ($residueList as $value) {
            $activeFunctions->set($value->getName(), [
                'id' => $value->getId(),
                'name' => $value->getName(),
                'status' => $value->getStatus(),
                'private-key' => ''
            ]);
        }

        $executionEnd = \microtime(true);

        Console::info(count($activeFunctions) . ' functions listed in ' . ($executionEnd - $executionStart) . ' seconds');
    });
} catch (\Throwable $error) {
    call_user_func($logError, $error, "startupError");
}

$createRuntimeServer = function(string $functionId, string $projectId, string $tagId, Database $database) use($logError): void
{
    global $orchestration;
    global $runtimes;
    global $activeFunctions;

    // Grab Function Document
    $function = Authorization::skip(function () use ($database, $functionId) {
        return $database->getDocument('functions', $functionId);
    });

    $tag = Authorization::skip(function () use ($database, $tagId) {
        return $database->getDocument('tags', $tagId);
    });

    if ($tag->getAttribute('buildId') === null) {
        throw new Exception('Tag has no buildId');
    }

    // Grab Build Document
    $build = Authorization::skip(function () use ($database, $tag) {
        return $database->getDocument('builds', $tag->getAttribute('buildId'));
    });

    // Check if function isn't already created
    $functions = $orchestration->list(['label' => 'appwrite-type=function', 'name' => 'appwrite-function-' . $tag->getId()]);

    if (\count($functions) > 0) {
        return;
    }

    // Generate random secret key
    $secret = \bin2hex(\random_bytes(16));

    // Check if runtime is active
    $runtime = (isset($runtimes[$function->getAttribute('runtime', '')]))
        ? $runtimes[$function->getAttribute('runtime', '')]
        : null;

    if ($tag->getAttribute('functionId') !== $function->getId()) {
        throw new Exception('Tag not found', 404);
    }

    if (\is_null($runtime)) {
        throw new Exception('Runtime "' . $function->getAttribute('runtime', '') . '" is not supported');
    }

    // Process environment variables
    $vars = \array_merge($function->getAttribute('vars', []), [
        'APPWRITE_FUNCTION_ID' => $function->getId(),
        'APPWRITE_FUNCTION_NAME' => $function->getAttribute('name', ''),
        'APPWRITE_FUNCTION_TAG' => $tag->getId(),
        'APPWRITE_FUNCTION_RUNTIME_NAME' => $runtime['name'],
        'APPWRITE_FUNCTION_RUNTIME_VERSION' => $runtime['version'],
        'APPWRITE_FUNCTION_PROJECT_ID' => $projectId,
        'INTERNAL_RUNTIME_KEY' => $secret
    ]);

    $vars = \array_merge($vars, $build->getAttribute('envVars', [])); // for gettng endpoint.

    $container = 'appwrite-function-' . $tag->getId();

    if ($activeFunctions->exists($container) && !(\substr($activeFunctions->get($container)['status'], 0, 2) === 'Up')) { // Remove container if not online
        // If container is online then stop and remove it
        try {
            $orchestration->remove($container, true);
        } catch (Exception $e) {
            try {
                throw new Exception('Failed to remove container: ' . $e->getMessage());
            } catch (Throwable $error) {
                call_user_func($logError, $error, "createRuntimeServer");
            }
        }

        $activeFunctions->del($container);
    }

    // Check if tag hasn't failed
    if ($build->getAttribute('status') == 'failed') {
        throw new Exception('Tag build failed, please check your logs.', 500);
    }

    // Check if tag is built yet.
    if ($build->getAttribute('status') !== 'ready') {
        throw new Exception('Tag is not built yet', 500);
    }

    // Grab Tag Files
    $tagPath = $build->getAttribute('outputPath', '');

    $tagPathTarget = '/tmp/project-' . $projectId . '/' . $build->getId() . '/builtCode/code.tar.gz';
    $tagPathTargetDir = \pathinfo($tagPathTarget, PATHINFO_DIRNAME);
    $container = 'appwrite-function-' . $tag->getId();

    $device = Storage::getDevice('builds');

    if (!\file_exists($tagPathTargetDir)) {
        if (!\mkdir($tagPathTargetDir, 0777, true)) {
            throw new Exception('Can\'t create directory ' . $tagPathTargetDir);
        }
    }

    if (!\file_exists($tagPathTarget)) {
        if (App::getEnv('_APP_STORAGE_DEVICE', Storage::DEVICE_LOCAL) === Storage::DEVICE_LOCAL) {
            if (!\copy($tagPath, $tagPathTarget)) {
                throw new Exception('Can\'t create temporary code file ' . $tagPathTarget);
            }
        } else {
            $buffer = $device->read($tagPath);
            \file_put_contents($tagPathTarget, $buffer);
        }
    };

    /**
     * Limit CPU Usage - DONE
     * Limit Memory Usage - DONE
     * Limit Network Usage
     * Limit Storage Usage (//--storage-opt size=120m \)
     * Make sure no access to redis, mariadb, influxdb or other system services
     * Make sure no access to NFS server / storage volumes
     * Access Appwrite REST from internal network for improved performance
     */
    if (!$activeFunctions->exists($container)) { // Create contianer if not ready
        $executionStart = \microtime(true);
        $executionTime = \time();

        $orchestration->setCpus(App::getEnv('_APP_FUNCTIONS_CPUS', '1'));
        $orchestration->setMemory(App::getEnv('_APP_FUNCTIONS_MEMORY', '256'));
        $orchestration->setSwap(App::getEnv('_APP_FUNCTIONS_MEMORY_SWAP', '256'));

        foreach ($vars as $key => $value) {
            $vars[$key] = strval($value);
        }

        // Launch runtime server
        $id = $orchestration->run(
            image: $runtime['image'],
            name: $container,
            vars: $vars,
            labels: [
                'appwrite-type' => 'function',
                'appwrite-created' => strval($executionTime),
                'appwrite-runtime' => $function->getAttribute('runtime', ''),
                'appwrite-project' => $projectId,
                'appwrite-tag' => $tag->getId(),
            ],
            hostname: $container,
            mountFolder: $tagPathTargetDir,
        );

        if (empty($id)) {
            throw new Exception('Failed to create container');
        }

        // Add to network
        $orchestration->networkConnect($container, 'appwrite_runtimes');

        $executionEnd = \microtime(true);

        $activeFunctions->set($container, [
            'id' => $id,
            'name' => $container,
            'status' => 'Up ' . \round($executionEnd - $executionStart, 2) . 's',
            'key' => $secret,
        ]);

        Console::info('Runtime Server created in ' . ($executionEnd - $executionStart) . ' seconds');
    } else {
        Console::info('Runtime server is ready to run');
    }
};

$execute = function(string $trigger, string $projectId, string $executionId, string $functionId, Database $database, string $event = '', string $eventData = '', string $data = '', array $webhooks = [], string $userId = '', string $jwt = '') use($logError, $createRuntimeServer): array
{
    Console::info('Executing function: ' . $functionId);

    global $activeFunctions;
    global $runtimes;
    global $register;

    // Grab Tag Document
    $function = Authorization::skip(function () use ($database, $functionId) {
        return $database->getDocument('functions', $functionId);
    });

    $tag = Authorization::skip(function () use ($database, $function) {
        return $database->getDocument('tags', $function->getAttribute('tag', ''));
    });

    // Grab Build Document
    $build = Authorization::skip(function () use ($database, $tag) {
        return $database->getDocument('builds', $tag->getAttribute('buildId', ''));
    });

    if ($tag->getAttribute('functionId') !== $function->getId()) {
        throw new Exception('Tag not found', 404);
    }

    // Grab execution document if exists
    // It it doesn't exist, create a new one.
    $execution = Authorization::skip(function () use ($database, $executionId, $userId, $function, $tag, $trigger, $functionId) {
        return (!empty($executionId)) ? $database->getDocument('executions', $executionId) : $database->createDocument('executions', new Document([
            '$id' => $executionId,
            '$read' => (!$userId == '') ? ['user:' . $userId] : [],
            '$write' => [],
            'dateCreated' => time(),
            'functionId' => $function->getId(),
            'tagId' => $tag->getId(),
            'trigger' => $trigger, // http / schedule / event
            'status' => 'processing', // waiting / processing / completed / failed
            'statusCode' => 0,
            'stdout' => '',
            'stderr' => '',
            'time' => 0.0,
            'search' => implode(' ', [$functionId, $executionId]),
        ]));
    });

    if (false === $execution || ($execution instanceof Document && $execution->isEmpty())) {
        throw new Exception('Failed to create or read execution');
    }


    if ($build->getAttribute('status') == 'building') {

        $execution->setAttribute('status', 'failed')
            ->setAttribute('statusCode', 500)
            ->setAttribute('stderr', 'Tag is still being built.')
            ->setAttribute('time', 0);

        Authorization::skip(function () use ($database, $execution) {
            return $database->updateDocument('executions', $execution->getId(), $execution);
        });

        throw new Exception('Execution Failed. Reason: Tag is still being built.');
    }

    // Check if runtime is active
    $runtime = (isset($runtimes[$function->getAttribute('runtime', '')]))
        ? $runtimes[$function->getAttribute('runtime', '')]
        : null;

    if (\is_null($runtime)) {
        throw new Exception('Runtime "' . $function->getAttribute('runtime', '') . '" is not supported');
    }

    // Process environment variables
    $vars = \array_merge($function->getAttribute('vars', []), [
        'APPWRITE_FUNCTION_ID' => $function->getId(),
        'APPWRITE_FUNCTION_NAME' => $function->getAttribute('name', ''),
        'APPWRITE_FUNCTION_TAG' => $tag->getId(),
        'APPWRITE_FUNCTION_TRIGGER' => $trigger,
        'APPWRITE_FUNCTION_RUNTIME_NAME' => $runtime['name'],
        'APPWRITE_FUNCTION_RUNTIME_VERSION' => $runtime['version'],
        'APPWRITE_FUNCTION_EVENT' => $event,
        'APPWRITE_FUNCTION_EVENT_DATA' => $eventData,
        'APPWRITE_FUNCTION_DATA' => $data,
        'APPWRITE_FUNCTION_USER_ID' => $userId,
        'APPWRITE_FUNCTION_JWT' => $jwt,
        'APPWRITE_FUNCTION_PROJECT_ID' => $projectId,
    ]);

    $vars = \array_merge($vars, $build->getAttribute('envVars', []));

    $container = 'appwrite-function-' . $tag->getId();

    try {
        if ($build->getAttribute('status') !== 'ready') {
            // Create a new build entry
            $buildId = $database->getId();
            Authorization::skip(function () use ($buildId, $database, $tag, $userId, $runtime, $function, $projectId) {
                $database->createDocument('builds', new Document([
                    '$id' => $buildId,
                    '$read' => (!$userId == '') ? ['user:' . $userId] : [],
                    '$write' => [],
                    'dateCreated' => time(),
                    'status' => 'processing',
                    'outputPath' => '',
                    'runtime' => $function->getAttribute('runtime', ''),
                    'source' => $tag->getAttribute('path'),
                    'sourceType' => Storage::DEVICE_LOCAL,
                    'stdout' => '',
                    'stderr' => '',
                    'buildTime' => 0,
                    'envVars' => [
                        'ENTRYPOINT_NAME' => $tag->getAttribute('entrypoint'),
                        'APPWRITE_FUNCTION_ID' => $function->getId(),
                        'APPWRITE_FUNCTION_NAME' => $function->getAttribute('name', ''),
                        'APPWRITE_FUNCTION_RUNTIME_NAME' => $runtime['name'],
                        'APPWRITE_FUNCTION_RUNTIME_VERSION' => $runtime['version'],
                        'APPWRITE_FUNCTION_PROJECT_ID' => $projectId,
                    ]
                ]));

                $tag->setAttribute('buildId', $buildId);

                $database->updateDocument('tags', $tag->getId(), $tag);
            });

            runBuildStage($buildId, $projectId, $database);
            sleep(1);
        }
    } catch (Exception $e) {
        $execution->setAttribute('status', 'failed')
            ->setAttribute('statusCode', 500)
            ->setAttribute('stderr', \utf8_encode(\mb_substr($e->getMessage(), -4000))) // log last 4000 chars output
            ->setAttribute('time', 0);

        Authorization::skip(function () use ($database, $execution) {
            return $database->updateDocument('executions', $execution->getId(), $execution);
        });

        throw new Error('Something went wrong building the code. ' . $e->getMessage());
    }

    try {
        if (!$activeFunctions->exists($container)) { // Create contianer if not ready
            $createRuntimeServer($functionId, $projectId, $tag->getId(), $database);
        } else if ($activeFunctions->get($container)['status'] === 'Down') {
            sleep(1);
        } else {
            Console::info('Container is ready to run');
        }
    } catch (Exception $e) {
        $execution->setAttribute('status', 'failed')
            ->setAttribute('statusCode', 500)
            ->setAttribute('stderr', \utf8_encode(\mb_substr($e->getMessage(), -4000))) // log last 4000 chars output
            ->setAttribute('time', 0);

        $execution = Authorization::skip(function () use ($database, $execution) {
            return $database->updateDocument('executions', $execution->getId(), $execution);
        });

        try {
            throw new Exception('Something went wrong building the runtime server. ' . $e->getMessage());
        } catch (\Exception $error) {
            call_user_func($logError, $error, "execution");
        }

        return [
            'status' => 'failed',
            'response' => \utf8_encode(\mb_substr($e->getMessage(), -4000)), // log last 4000 chars output
            'time' => 0
        ];
    }

    $internalFunction = $activeFunctions->get('appwrite-function-' . $tag->getId());
    $key = $internalFunction['key'];

    // Process environment variables
    $vars = \array_merge($function->getAttribute('vars', []), [
        'APPWRITE_FUNCTION_ID' => $function->getId(),
        'APPWRITE_FUNCTION_NAME' => $function->getAttribute('name', ''),
        'APPWRITE_FUNCTION_TAG' => $tag->getId(),
        'APPWRITE_FUNCTION_TRIGGER' => $trigger,
        'APPWRITE_FUNCTION_RUNTIME_NAME' => $runtime['name'],
        'APPWRITE_FUNCTION_RUNTIME_VERSION' => $runtime['version'],
        'APPWRITE_FUNCTION_EVENT' => $event,
        'APPWRITE_FUNCTION_EVENT_DATA' => $eventData,
        'APPWRITE_FUNCTION_DATA' => $data,
        'APPWRITE_FUNCTION_USER_ID' => $userId,
        'APPWRITE_FUNCTION_JWT' => $jwt,
        'APPWRITE_FUNCTION_PROJECT_ID' => $projectId
    ]);

    $vars = \array_merge($vars, $build->getAttribute('envVars', []));

    $stdout = '';
    $stderr = '';

    $executionStart = \microtime(true);

    $statusCode = 0;

    $errNo = -1;
    $attempts = 0;
    $max = 5;

    $executorResponse = '';

    // cURL request to runtime
    do {
        $attempts++;
        $ch = \curl_init();

        $body = \json_encode([
            'path' => '/usr/code',
            'file' => $build->getAttribute('envVars', [])['ENTRYPOINT_NAME'],
            'env' => $vars,
            'payload' => $data,
            'timeout' => $function->getAttribute('timeout', (int) App::getEnv('_APP_FUNCTIONS_TIMEOUT', 900))
        ]);

        \curl_setopt($ch, CURLOPT_URL, "http://" . $container . ":3000/");
        \curl_setopt($ch, CURLOPT_POST, true);
        \curl_setopt($ch, CURLOPT_POSTFIELDS, $body);

        \curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
        \curl_setopt($ch, CURLOPT_TIMEOUT, $function->getAttribute('timeout', (int) App::getEnv('_APP_FUNCTIONS_TIMEOUT', 900)));
        \curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 10);

        \curl_setopt($ch, CURLOPT_HTTPHEADER, [
            'Content-Type: application/json',
            'Content-Length: ' . \strlen($body),
            'x-internal-challenge: ' . $key,
            'host: null'
        ]);

        $executorResponse = \curl_exec($ch);

        $statusCode = \curl_getinfo($ch, CURLINFO_HTTP_CODE);

        $error = \curl_error($ch);

        $errNo = \curl_errno($ch);

        \curl_close($ch);
        if ($errNo != CURLE_COULDNT_CONNECT && $errNo != 111) {
            break;
        }

        sleep(1);
    } while ($attempts < $max);

    if ($attempts >= 5) {
        $stderr = 'Failed to connect to executor runtime after 5 attempts.';
        $statusCode = 124;
    }

    // If timeout error
    if ($errNo == CURLE_OPERATION_TIMEDOUT || $errNo == 110) {
        $statusCode = 124;
    }

    // 110 is the Swoole error code for timeout, see: https://www.swoole.co.uk/docs/swoole-error-code
    if ($errNo !== 0 && $errNo != CURLE_COULDNT_CONNECT && $errNo != CURLE_OPERATION_TIMEDOUT && $errNo != 110) {
        throw new Exception('An internal curl error has occurred within the executor! Error Msg: ' . $error, 500);
    }

    $executionData = [];

    if (!empty($executorResponse)) {
        $executionData = json_decode($executorResponse, true);
    }

    if (isset($executionData['code'])) {
        $statusCode = $executionData['code'];
    }

    if ($statusCode === 500) {
        if (isset($executionData['message'])) {
            $stderr = $executionData['message'];
        } else {
            $stderr = 'Internal Runtime error';
        }
    } else if ($statusCode === 124) {
        $stderr = 'Execution timed out.';
    } else if ($statusCode === 0) {
        $stderr = 'Execution failed.';
    } else if ($statusCode >= 200 && $statusCode < 300) {
        $stdout = $executorResponse;
    } else {
        $stderr = 'Execution failed.';
    }

    $executionEnd = \microtime(true);
    $executionTime = ($executionEnd - $executionStart);
    $functionStatus = ($statusCode >= 200 && $statusCode < 300) ? 'completed' : 'failed';

    Console::info('Function executed in ' . ($executionEnd - $executionStart) . ' seconds, status: ' . $functionStatus);

    $execution->setAttribute('tagId', $tag->getId())
        ->setAttribute('status', $functionStatus)
        ->setAttribute('statusCode', $statusCode)
        ->setAttribute('stdout', \utf8_encode(\mb_substr($stdout, -8000)))
        ->setAttribute('stderr', \utf8_encode(\mb_substr($stderr, -8000)))
        ->setAttribute('time', $executionTime);

    $execution = Authorization::skip(function () use ($database, $execution) {
        return $database->updateDocument('executions', $execution->getId(), $execution);
    });

    $executionModel = new Execution();
    $executionUpdate = new Event('v1-webhooks', 'WebhooksV1');

    $executionUpdate
        ->setParam('projectId', $projectId)
        ->setParam('userId', $userId)
        ->setParam('webhooks', $webhooks)
        ->setParam('event', 'functions.executions.update')
        ->setParam('eventData', $execution->getArrayCopy(array_keys($executionModel->getRules())));

    $executionUpdate->trigger();

    $target = Realtime::fromPayload('functions.executions.update', $execution);

    Realtime::send(
        projectId: $projectId,
        payload: $execution->getArrayCopy(),
        event: 'functions.executions.update',
        channels: $target['channels'],
        roles: $target['roles']
    );

    if (App::getEnv('_APP_USAGE_STATS', 'enabled') == 'enabled') {
        $statsd = $register->get('statsd');

        $usage = new Stats($statsd);

        $usage
            ->setParam('projectId', $projectId)
            ->setParam('functionId', $function->getId())
            ->setParam('functionExecution', 1)
            ->setParam('functionStatus', $functionStatus)
            ->setParam('functionExecutionTime', $executionTime * 1000) // ms
            ->setParam('networkRequestSize', 0)
            ->setParam('networkResponseSize', 0)
            ->submit();

        $usage->submit();
    }

    return [
        'status' => $functionStatus,
        'response' => ($functionStatus !== 'completed') ? $stderr : $stdout,
        'time' => $executionTime
    ];
};

App::post('/v1/execute') // Define Route
    ->desc('Execute a function')
    ->inject('request')
    ->param('trigger', '', new Text(1024))
    ->param('projectId', '', new Text(1024))
    ->param('executionId', '', new Text(1024), '', true)
    ->param('functionId', '', new Text(1024))
    ->param('event', '', new Text(1024), '', true)
    ->param('eventData', '', new Text(10240), '', true)
    ->param('data', '', new Text(1024), '', true)
    ->param('webhooks', [], new ArrayList(new JSON()), '', true)
    ->param('userId', '', new Text(1024), '', true)
    ->param('jwt', '', new Text(1024), '', true)
    ->inject('response')
    ->inject('dbForProject')
    ->action(
        function ($trigger, $projectId, $executionId, $functionId, $event, $eventData, $data, $webhooks, $userId, $jwt, $request, $response, $dbForProject) use ($execute, $logError) {
            try {
                $data = $execute($trigger, $projectId, $executionId, $functionId, $dbForProject, $event, $eventData, $data, $webhooks, $userId, $jwt);
                $response->json($data);
            } catch (Exception $e) {
                call_user_func($logError, $e, "executeEndpoint");

                $response
                    ->addHeader('Cache-Control', 'no-cache, no-store, must-revalidate')
                    ->addHeader('Expires', '0')
                    ->addHeader('Pragma', 'no-cache')
                    ->json(['error' => $e->getMessage()]);
            }
        }
    );


// Cleanup Endpoints used internally by appwrite when a function or tag gets deleted to also clean up their containers
App::post('/v1/cleanup/function')
    ->param('functionId', '', new UID())
    ->inject('response')
    ->inject('dbForProject')
    ->inject('projectID')
    ->action(function ($functionId, $response, $dbForProject, $projectID) use($logError) {
        /** @var string $functionId */
        /** @var Appwrite\Utopia\Response $response */
        /** @var Utopia\Database\Database $dbForProject */
        /** @var string $projectID */

        global $orchestration;

        try {
            // Get function document
            $function = Authorization::skip(function () use ($dbForProject, $functionId) {
                return $dbForProject->getDocument('functions', $functionId);
            });

            // Check if function exists
            if ($function->isEmpty()) {
                throw new Exception('Function not found', 404);
            }

            $results = Authorization::skip(function () use ($dbForProject, $functionId) {
                return $dbForProject->find('tags', [new Query('functionId', Query::TYPE_EQUAL, [$functionId])], 999);
            });

            // If amount is 0 then we simply return true
            if (count($results) === 0) {
                return $response->json(['success' => true]);
            }

            // Delete the containers of all tags
            foreach ($results as $tag) {
                try {
                    // Remove any ongoing builds
                    if ($tag->getAttribute('buildId')) {
                        $build = Authorization::skip(function () use ($dbForProject, $tag) {
                            return $dbForProject->getDocument('builds', $tag->getAttribute('buildId'));
                        });

                        if ($build->getAttribute('status') == 'building') {
                            // Remove the build
                            $orchestration->remove('build-stage-' . $tag->getAttribute('buildId'), true);
                            Console::info('Removed build for tag ' . $tag['$id']);
                        }
                    }

                    $orchestration->remove('appwrite-function-' . $tag['$id'], true);
                    Console::info('Removed container for tag ' . $tag['$id']);
                } catch (Exception $e) {
                    // Do nothing, we don't care that much if it fails
                }
            }

            return $response->json(['success' => true]);
        } catch (Exception $e) {
            call_user_func($logError, $e, "cleanupFunction");

            return $response->json(['error' => $e->getMessage()]);
        }
    });

App::post('/v1/cleanup/tag')
    ->param('tagId', '', new UID(), 'Tag unique ID.')
    ->inject('response')
    ->inject('dbForProject')
    ->inject('projectID')
    ->action(function ($tagId, $response, $dbForProject, $projectID) use($logError) {
        /** @var string $tagId */
        /** @var Appwrite\Utopia\Response $response */
        /** @var Appwrite\Database\Database $dbForProject */
        /** @var string $projectID */

        global $orchestration;

        try {
            // Get tag document
            $tag = Authorization::skip(function () use ($dbForProject, $tagId) {
                return $dbForProject->getDocument('tags', $tagId);
            });

            // Check if tag exists
            if ($tag->isEmpty()) {
                throw new Exception('Tag not found', 404);
            }

            try {
                // Remove any ongoing builds
                if ($tag->getAttribute('buildId')) {
                    $build = Authorization::skip(function () use ($dbForProject, $tag) {
                        return $dbForProject->getDocument('builds', $tag->getAttribute('buildId'));
                    });
            
                    if ($build->getAttribute('status') == 'building') {
                        // Remove the build
                        $orchestration->remove('build-stage-' . $tag->getAttribute('buildId'), true);
                        Console::info('Removed build for tag ' . $tag['$id']);
                    }
                }

                // Remove the container of the tag
                $orchestration->remove('appwrite-function-' . $tag['$id'], true);
                Console::info('Removed container for tag ' . $tag['$id']);
            } catch (Exception $e) {
                // Do nothing, we don't care that much if it fails
            }
        } catch (Exception $e) {
            call_user_func($logError, $e, "cleanupFunction");
            return $response->json(['error' => $e->getMessage()]);
        }

        return $response->json(['success' => true]);
    });

App::post('/v1/tag')
    ->param('functionId', '', new UID(), 'Function unique ID.')
    ->param('tagId', '', new UID(), 'Tag unique ID.')
    ->param('userId', '', new UID(), 'User unique ID.', true)
    ->param('autoDeploy', false, new Boolean(), '', true)
    ->inject('response')
    ->inject('dbForProject')
    ->inject('projectID')
    ->action(function ($functionId, $tagId, $userId, $autoDeploy, $response, $dbForProject, $projectID) use ($createRuntimeServer) {
        global $runtimes;

        // Get function document
        $function = Authorization::skip(function () use ($functionId, $dbForProject) {
            return $dbForProject->getDocument('functions', $functionId);
        });

        // Get tag document
        $tag = Authorization::skip(function () use ($tagId, $dbForProject) {
            return $dbForProject->getDocument('tags', $tagId);
        });

        // Check if both documents exist
        if ($function->isEmpty()) {
            throw new Exception('Function not found', 404);
        }

        if ($tag->isEmpty()) {
            throw new Exception('Tag not found', 404);
        }

        $runtime = (isset($runtimes[$function->getAttribute('runtime')]))
            ? $runtimes[$function->getAttribute('runtime')]
            : null;

        // Create a new build entry
        $buildId = $dbForProject->getId();

        if ($tag->getAttribute('buildId')) {
            $buildId = $tag->getAttribute('buildId');   
        } else {
            Authorization::skip(function () use ($buildId, $dbForProject, $tag, $userId, $function, $projectID, $runtime) {
                $dbForProject->createDocument('builds', new Document([
                    '$id' => $buildId,
                    '$read' => (!empty($userId)) ? ['user:' . $userId] : [],
                    '$write' => [],
                    'dateCreated' => time(),
                    'status' => 'processing',
                    'runtime' => $function->getAttribute('runtime'),
                    'outputPath' => '',
                    'source' => $tag->getAttribute('path'),
                    'sourceType' => Storage::DEVICE_LOCAL,
                    'stdout' => '',
                    'stderr' => '',
                    'buildTime' => 0,
                    'envVars' => [
                        'ENTRYPOINT_NAME' => $tag->getAttribute('entrypoint'),
                        'APPWRITE_FUNCTION_ID' => $function->getId(),
                        'APPWRITE_FUNCTION_NAME' => $function->getAttribute('name', ''),
                        'APPWRITE_FUNCTION_RUNTIME_NAME' => $runtime['name'],
                        'APPWRITE_FUNCTION_RUNTIME_VERSION' => $runtime['version'],
                        'APPWRITE_FUNCTION_PROJECT_ID' => $projectID,
                    ]
                ]));

                $tag->setAttribute('buildId', $buildId);

                $dbForProject->updateDocument('tags', $tag->getId(), $tag);
            });
        }

        // Build Code
        go(function () use ($dbForProject, $projectID, $tagId, $buildId, $functionId, $function, $createRuntimeServer) {
            // Build Code
            runBuildStage($buildId, $projectID, $dbForProject);

            // Update the schedule
            $schedule = $function->getAttribute('schedule', '');
            $cron = (empty($function->getAttribute('tag')) && !empty($schedule)) ? new CronExpression($schedule) : null;
            $next = (empty($function->getAttribute('tag')) && !empty($schedule)) ? $cron->getNextRunDate()->format('U') : 0;

            // Grab tag
            $tag = Authorization::skip(function () use ($dbForProject, $tagId, $next, $buildId) {
                return $dbForProject->getDocument('tags', $tagId);
            });

            // Grab build
            $build = Authorization::skip(function () use ($dbForProject, $buildId) {
                return $dbForProject->getDocument('builds', $buildId);
            });

            // If the build failed, it won't be possible to deploy
            if ($build->getAttribute('status') !== 'ready') {
                return;
            }

            if ($tag->getAttribute('automaticDeploy') === true) {
                // Update the function document setting the tag as the active one
                $function = Authorization::skip(function () use ($function, $dbForProject, $tag, $next) {
                    return $function = $dbForProject->updateDocument('functions', $function->getId(), new Document(array_merge($function->getArrayCopy(), [
                        'tag' => $tag->getId(),
                        'scheduleNext' => (int)$next,
                    ])));
                });
            }

            // Deploy Runtime Server
            $createRuntimeServer($functionId, $projectID, $tagId, $dbForProject);
        });

        if (false === $function) {
            throw new Exception('Failed saving function to DB', 500);
        }

        $response->dynamic($function, Response::MODEL_FUNCTION);
    });

App::get('/v1/')
    ->inject('request')
    ->inject('response')
    ->action(function ($request, $response) {
            $response
                ->addHeader('Cache-Control', 'no-cache, no-store, must-revalidate')
                ->addHeader('Expires', '0')
                ->addHeader('Pragma', 'no-cache')
                ->json(['status' => 'online']);
        }
    );

// Build Endpoints
App::post('/v1/build/:buildId') // Start a Build
    ->param('buildId', '', new UID(), 'Build unique ID.', false)
    ->inject('response')
    ->inject('dbForProject')
    ->inject('projectID')
    ->action(function ($buildId, $response, $dbForProject, $projectID) use ($logError) {
        /** @var string $buildId */
        /** @var Appwrite\Utopia\Response $response */
        /** @var Utopia\Database\Database $dbForProject */
        /** @var string $projectID */

        try {
            // Get build document
            $build = Authorization::skip(function () use ($buildId, $dbForProject) {
                return $dbForProject->getDocument('builds', $buildId);
            });

            // Check if build exists
            if ($build->isEmpty()) {
                throw new Exception('Build not found', 404);
            }

            // Check if build is already running
            if ($build->getAttribute('status') === 'running') {
                throw new Exception('Build is already running', 409);
            }

            // Check if build is already finished
            if ($build->getAttribute('status') === 'finished') {
                throw new Exception('Build is already finished', 409);
            }

            go(function () use ($buildId, $dbForProject, $projectID) {
                // Build Code
                runBuildStage($buildId, $projectID, $dbForProject);
            });

            // return success
            return $response->json(['success' => true]);
        } catch (Exception $e) {
            call_user_func($logError, $e, "buildEndpoint");

            $response
                ->addHeader('Cache-Control', 'no-cache, no-store, must-revalidate')
                ->addHeader('Expires', '0')
                ->addHeader('Pragma', 'no-cache')
                ->json(['error' => $e->getMessage()]);
        }
    });

function runBuildStage(string $buildId, string $projectID, Database $database): Document
{
    global $runtimes;
    global $orchestration;

    $buildStdout = '';
    $buildStderr = '';

    // Check if build has already been run
    $build = Authorization::skip(function () use ($buildId, $database) {
        return $database->getDocument('builds', $buildId);
    });

    try {
        // If we already have a built package ready there is no need to rebuild.
        if ($build->getAttribute('status') === 'ready' && \file_exists($build->getAttribute('outputPath'))) {
            return $build;
        }

        // Update Tag Status
        $build->setAttribute('status', 'building');

        Authorization::skip(function () use ($build, $database) {
            $database->updateDocument('builds', $build->getId(), $build);
        });

        // Check if runtime is active
        $runtime = (isset($runtimes[$build->getAttribute('runtime', '')]))
            ? $runtimes[$build->getAttribute('runtime', '')]
            : null;

        if (\is_null($runtime)) {
            throw new Exception('Runtime "' . $build->getAttribute('runtime', '') . '" is not supported');
        }

        // Grab Tag Files
        $tagPath = $build->getAttribute('source', '');
        $sourceType = $build->getAttribute('sourceType', '');

        $device = Storage::getDevice('builds');

        $tagPathTarget = '/tmp/project-' . $projectID . '/' . $build->getId() . '/code.tar.gz';
        $tagPathTargetDir = \pathinfo($tagPathTarget, PATHINFO_DIRNAME);

        $container = 'build-stage-' . $build->getId();

        // Perform various checks
        if (!\file_exists($tagPathTargetDir)) {
            if (!\mkdir($tagPathTargetDir, 0777, true)) {
                throw new Exception('Can\'t create directory ' . $tagPathTargetDir);
            }
        }

        if (!\file_exists($tagPathTarget)) {
            if (App::getEnv('_APP_STORAGE_DEVICE', Storage::DEVICE_LOCAL) === Storage::DEVICE_LOCAL) {
                if (!\copy($tagPath, $tagPathTarget)) {
                    throw new Exception('Can\'t create temporary code file ' . $tagPathTarget);
                }
            } else {
                $buffer = $device->read($tagPath);
                \file_put_contents($tagPathTarget, $buffer);
            }
        }

        if (!$device->exists($tagPath)) {
            throw new Exception('Code is not readable: ' . $build->getAttribute('source', ''));
        }

        $vars = $build->getAttribute('envVars', []);

        // Start tracking time
        $buildStart = \microtime(true);
        $buildTime = \time();

        $orchestration->setCpus(App::getEnv('_APP_FUNCTIONS_CPUS', 0));
        $orchestration->setMemory(App::getEnv('_APP_FUNCTIONS_MEMORY', 256));
        $orchestration->setSwap(App::getEnv('_APP_FUNCTIONS_MEMORY_SWAP', 256));

        foreach ($vars as &$value) {
            $value = strval($value);
        }

        if (!\file_exists('/tmp/project-' . $projectID . '/' . $build->getId() . '/builtCode')) {
            if (!\mkdir('/tmp/project-' . $projectID . '/' . $build->getId() . '/builtCode', 0777, true)) {
                throw new Exception('Can\'t create directory /tmp/project-' . $projectID . '/' . $build->getId() . '/builtCode');
            }
        };

        // Launch build container
        $id = $orchestration->run(
            image: $runtime['base'],
            name: $container,
            vars: $vars,
            workdir: '/usr/code',
            labels: [
                'appwrite-type' => 'function',
                'appwrite-created' => strval($buildTime),
                'appwrite-runtime' => $build->getAttribute('runtime', ''),
                'appwrite-project' => $projectID,
                'appwrite-build' => $build->getId(),
            ],
            command: [
                'tail',
                '-f',
                '/dev/null'
            ],
            hostname: $container,
            mountFolder: $tagPathTargetDir,
            volumes: [
                '/tmp/project-' . $projectID . '/' . $build->getId() . '/builtCode' . ':/usr/builtCode:rw'
            ]
        );

        if (empty($id)) {
            throw new Exception('Failed to start build container');
        }

        // Extract user code into build container
        $untarStdout = '';
        $untarStderr = '';

        $untarSuccess = $orchestration->execute(
            name: $container,
            command: [
                'sh',
                '-c',
                'mkdir -p /usr/code && cp /tmp/code.tar.gz /usr/workspace/code.tar.gz && cd /usr/workspace/ && tar -zxf /usr/workspace/code.tar.gz -C /usr/code && rm /usr/workspace/code.tar.gz'
            ],
            stdout: $untarStdout,
            stderr: $untarStderr,
            timeout: 60
        );

        if (!$untarSuccess) {
            throw new Exception('Failed to extract tar: ' . $untarStderr);
        }

        // Build Code / Install Dependencies
        $buildSuccess = $orchestration->execute(
            name: $container,
            command: ['sh', '-c', 'cd /usr/local/src && ./build.sh'],
            stdout: $buildStdout,
            stderr: $buildStderr,
            timeout: App::getEnv('_APP_FUNCTIONS_BUILD_TIMEOUT', 900)
        );

        if (!$buildSuccess) {
            throw new Exception('Failed to build dependencies: ' . $buildStderr);
        }

        // Repackage Code and Save.
        $compressStdout = '';
        $compressStderr = '';

        $builtCodePath = '/tmp/project-' . $projectID . '/' . $build->getId() . '/builtCode/code.tar.gz';

        $compressSuccess = $orchestration->execute(
            name: $container,
            command: [
                'tar', '-C', '/usr/code', '-czvf', '/usr/builtCode/code.tar.gz', './'
            ],
            stdout: $compressStdout,
            stderr: $compressStderr,
            timeout: 60
        );

        if (!$compressSuccess) {
            throw new Exception('Failed to compress built code: ' . $compressStderr);
        }

        // Remove Container
        $orchestration->remove($id, true);

        // Check if the build was successful by checking if file exists
        if (!\file_exists($builtCodePath)) {
            throw new Exception('Something went wrong during the build process.');
        }

        // Upload new code
        $device = Storage::getDevice('builds');

        $path = $device->getPath(\uniqid() . '.' . \pathinfo('code.tar.gz', PATHINFO_EXTENSION));

        if (!\file_exists(\dirname($path))) { // Checks if directory path to file exists
            if (!@\mkdir(\dirname($path), 0777, true)) {
                throw new Exception('Can\'t create directory: ' . \dirname($path));
            }
        }

        if (App::getEnv('_APP_STORAGE_DEVICE', Storage::DEVICE_LOCAL) === Storage::DEVICE_LOCAL) {
            if (!$device->move($builtCodePath, $path)) {
                throw new Exception('Failed to upload built code upload to storage', 500);
            }
        } else {
            if (!$device->upload($builtCodePath, $path)) {
                throw new Exception('Failed to upload built code upload to storage', 500);
            }
        }

        if ($buildStdout == '') {
            $buildStdout = 'Build Successful!';
        }

        $build->setAttribute('outputPath', $path)
            ->setAttribute('status', 'ready')
            ->setAttribute('stdout',  \utf8_encode(\mb_substr($buildStdout, -4096)))
            ->setAttribute('stderr', \utf8_encode(\mb_substr($buildStderr, -4096)))
            ->setAttribute('buildTime', $buildTime);

        // Update build with built code attribute
        $build = Authorization::skip(function () use ($build, $buildId, $database) {
            return $database->updateDocument('builds', $buildId, $build);
        });

        $buildEnd = \microtime(true);

        Console::info('Build Stage Ran in ' . ($buildEnd - $buildStart) . ' seconds');
    } catch (Exception $e) {
        $build->setAttribute('status', 'failed')
            ->setAttribute('stdout',  \utf8_encode(\mb_substr($buildStdout, -4096)))
            ->setAttribute('stderr', \utf8_encode(\mb_substr($e->getMessage(), -4096)));

        $build = Authorization::skip(function () use ($build, $buildId, $database) {
            return $database->updateDocument('builds', $buildId, $build);
        });

        // also remove the container if it exists
        if ($id) {
            $orchestration->remove($id, true);
        }

        throw new Exception('Build failed: ' . $e->getMessage());
    }

    return $build;
}

App::setMode(App::MODE_TYPE_PRODUCTION); // Define Mode

$http = new Server("0.0.0.0", 8080);

$handleShutdown = function() use($logError)
{
    try {
        Console::info('Cleaning up containers before shutdown...');

        // Remove all containers.
        global $orchestration;

        global $register;

        $functionsToRemove = $orchestration->list(['label' => 'appwrite-type=function']);

        foreach ($functionsToRemove as $container) {
            $orchestration->remove($container->getId(), true);

            // Get a database instance
            $db = $register->get('dbPool')->get();
            $cache = $register->get('redisPool')->get();

            $cache = new Cache(new RedisCache($cache));

            $database = new Database(new MariaDB($db), $cache);
            $database->setNamespace('project_'.$container->getLabels()["appwrite-project"].'_internal');

            // Get list of all processing executions
            $executions = Authorization::skip(function () use ($database, $container) {
                return $database->find('executions', [
                    new Query('tagId', Query::TYPE_EQUAL, [$container->getLabels()["appwrite-tag"]]),
                    new Query('status', Query::TYPE_EQUAL, ['waiting'])
                ]);
            });

            // Mark all processing executions as failed
            foreach ($executions as $execution) {
                $execution->setAttribute('status', 'failed')
                    ->setAttribute('statusCode', 1)
                    ->setAttribute('stderr', 'Appwrite was shutdown during execution');

                Authorization::skip(function () use ($database, $execution) {
                    $database->updateDocument('executions', $execution->getId(), $execution);
                });
            }

            Console::info('Removed container ' . $container->getName());
        }
    } catch(\Throwable $error) {
        call_user_func($logError, $error, "shutdownError");
    }
};

$http->on('start', function ($http) use($handleShutdown) {
    Process::signal(SIGINT, function () use ($http, $handleShutdown) {
        $handleShutdown();
        $http->shutdown();
    });

    Process::signal(SIGQUIT, function () use ($http, $handleShutdown) {
        $handleShutdown();
        $http->shutdown();
    });

    Process::signal(SIGKILL, function () use ($http, $handleShutdown) {
        $handleShutdown();
        $http->shutdown();
    });

    Process::signal(SIGTERM, function () use ($http, $handleShutdown) {
        $handleShutdown();
        $http->shutdown();
    });
});

$http->on('request', function (SwooleRequest $swooleRequest, SwooleResponse $swooleResponse) use($logError) {
    global $register;

    $request = new Request($swooleRequest);
    $response = new Response($swooleResponse);
    $app = new App('UTC');

    $db = $register->get('dbPool')->get();

    App::setResource('db', function () use (&$db) {
        return $db;
    });

    $redis = $register->get('redisPool')->get();

    App::setResource('cache', function () use (&$redis) {
        return $redis;
    });

    $projectId = $request->getHeader('x-appwrite-project', '');

    Storage::setDevice('functions', new Local(APP_STORAGE_FUNCTIONS . '/app-' . $projectId));
    Storage::setDevice('builds', new Local(APP_STORAGE_BUILDS . '/app-' . $projectId));

    // Check environment variable key
    $secretKey = $request->getHeader('x-appwrite-executor-key', '');

    if (empty($secretKey)) {
        $swooleResponse->status(401);
        return $swooleResponse->end('401: Authentication Error');
    }

    if ($secretKey !== App::getEnv('_APP_EXECUTOR_SECRET', '')) {
        $swooleResponse->status(401);
        return $swooleResponse->end('401: Authentication Error');
    }

    App::setResource('dbForProject', function ($db, $cache) use ($projectId) {
        $cache = new Cache(new RedisCache($cache));

        $database = new Database(new MariaDB($db), $cache);
        $database->setDefaultDatabase(App::getEnv('_APP_DB_SCHEMA', 'appwrite'));
        $database->setNamespace('_project_'.$projectId);
    
        return $database;
    }, ['db', 'cache']);

    App::error(function ($error, $utopia, $request, $response) use ($logError) {
        /** @var Exception $error */
        /** @var Utopia\App $utopia */
        /** @var Utopia\Swoole\Request $request */
        /** @var Appwrite\Utopia\Response $response */

        if ($error instanceof PDOException) {
            throw $error;
        }

        $route = $utopia->match($request);
        call_user_func($logError, $error, "httpError", $route);

        $version = App::getEnv('_APP_VERSION', 'UNKNOWN');

        $code = $error->getCode();
        $message = $error->getMessage();

        $output = ((App::isDevelopment())) ? [
            'message' => $error->getMessage(),
            'code' => $error->getCode(),
            'file' => $error->getFile(),
            'line' => $error->getLine(),
            'trace' => $error->getTrace(),
            'version' => $version,
        ] : [
            'message' => $message,
            'code' => $code,
            'version' => $version,
        ];

        $response
            ->addHeader('Cache-Control', 'no-cache, no-store, must-revalidate')
            ->addHeader('Expires', '0')
            ->addHeader('Pragma', 'no-cache')
            ->setStatusCode($code);

        $response->dynamic(
            new Document($output),
            $utopia->isDevelopment() ? Response::MODEL_ERROR_DEV : Response::MODEL_ERROR
        );
    }, ['error', 'utopia', 'request', 'response']);

    App::setResource('projectID', function () use ($projectId) {
        return $projectId;
    });

    try {
        $app->run($request, $response);
    } catch (Exception $e) {
        call_user_func($logError, $e, "serverError");
        $swooleResponse->end('500: Server Error');
    } finally {
        /** @var PDOPool $dbPool */
        $dbPool = $register->get('dbPool');
        $dbPool->put($db);

        /** @var RedisPool $redisPool */
        $redisPool = $register->get('redisPool');
        $redisPool->put($redis);
    }
});

$http->start();