<?php

global $cli, $register;

require_once __DIR__ . '/../init.php';

use Utopia\App;
use Utopia\Cache\Adapter\Redis;
use Utopia\Cache\Cache;
use Utopia\CLI\Console;
use Utopia\Database\Adapter\MariaDB;
use Utopia\Database\Database;
use Utopia\Database\Document;
use Utopia\Database\Query;
use Utopia\Database\Validator\Authorization;

$cli
    ->task('usage')
    ->desc('Schedules syncing data from influxdb to Appwrite console db')
    ->action(function () use ($register) {
        Console::title('Usage Sync V1');
        Console::success(APP_NAME . ' usage sync process v1 has started');

        $interval = (int) App::getEnv('_APP_USAGE_SYNC_INTERVAL', '30'); //30 seconds

        $cacheAdapter = new Cache(new Redis($register->get('cache')));
        $dbForConsole = new Database(new MariaDB($register->get('db')), $cacheAdapter);
        $dbForConsole->setNamespace('project_console_internal');
        $dbForProject = new Database(new MariaDB($register->get('db')), $cacheAdapter);

        Authorization::disable();
        $projectIds = [];
        $latestProject = null;
        $latestData = [];
        do {
            $projects = $dbForConsole->find('projects', [], 100, orderAfter:$latestProject);
            if (!empty($projects)) {
                $latestProject = $projects[array_key_last($projects)];
                $latestData = getLatestData($projects, $latestData, $dbForProject);
            }
        } while (!empty($projects));

        $projects = null;

        $firstRun = true;
        Console::loop(function () use ($interval, $register, &$projectIds, &$latestData, $dbForProject, $dbForConsole, &$firstRun, &$latestProject) {
            $time = date('d-m-Y H:i:s', time());
            Console::info("[{$time}] Syncing usage data from influxdb to Appwrite Console DB every {$interval} seconds");

            if (!$firstRun) {
                $projects = $dbForConsole->find('projects', limit:100, orderAfter:$latestProject);
                if (!empty($projects)) {
                    $latestProject = $projects[array_key_last($projects)];
                    $latestData = getLatestData($projects, $latestData, $dbForProject);
                }
            }

            $client = $register->get('influxdb');
            if ($client) {
                foreach ($projectIds as $id => $value) {
                    syncData($client, $id, '30m', $latestData, $dbForProject);
                    syncData($client, $id, '1d', $latestData, $dbForProject);
                }
            }
            $firstRun = false;
        }, $interval);
    });

function getLatestData(&$projects, &$latestData, $dbForProject)
{
    foreach ($projects as $project) {
        $id = $project->getId();
        $projectIds[$id] = true;
        $dbForProject->setNamespace("project_{$id}_internal");
        $doc = $dbForProject->findOne('stats', [new Query("period", Query::TYPE_EQUAL, ["1d"])], 0, ['time'], [Database::ORDER_DESC]);
        $latestData[$id]["1d"] = $doc ? $doc->getAttribute('time') : null;
        $doc = $dbForProject->findOne('stats', [new Query("period", Query::TYPE_EQUAL, ["30m"])], 0, ['time'], [Database::ORDER_DESC]);
        $latestData[$id]["30m"] = $doc ? $doc->getAttribute('time') : null;
    }
    $projects = null;
    return $latestData;
}

function syncData($client, $projectId, $period, &$latestData, $dbForProject)
{
    $requests = [];
    $network = [];
    $functions = [];
    $start = DateTime::createFromFormat('U', \strtotime($period == '1d' ? '-90 days' : '-1 days'))->format(DateTime::RFC3339);
    if (!empty($latestData[$projectId][$period])) {
        $start = DateTime::createFromFormat('U', $latestData[$projectId][$period])->format(DateTime::RFC3339);
    }
    $end = DateTime::createFromFormat('U', \strtotime('now'))->format(DateTime::RFC3339);
    $database = $client->selectDB('telegraf');

    // Requests
    $result = $database->query('SELECT sum(value) AS "value" FROM "appwrite_usage_requests_all" WHERE time > \'' . $start . '\' AND time < \'' . $end . '\' AND "metric_type"=\'counter\' AND "project"=\'' . $projectId . '\' GROUP BY time(' . $period . ') FILL(null)');
    $points = $result->getPoints();

    $dbForProject->setNamespace("project_{$projectId}_internal");
    foreach ($points as $point) {
        $requests[] = [
            'value' => (!empty($point['value'])) ? $point['value'] : 0,
            'date' => \strtotime($point['time']),
            'dateStr' => $point['time'],
            'point' => $point,
        ];
        $time = \strtotime($point['time']);
        $id = \md5($time . '_' . $period . '_requests');
        $value = (!empty($point['value'])) ? $point['value'] : 0;
        $document = $dbForProject->getDocument('stats', $id);
        if ($document->isEmpty()) {
            $dbForProject->createDocument('stats', new Document([
                '$id' => $id,
                'period' => $period,
                'time' => $time,
                'metric' => 'requests',
                'value' => $value,
                'type' => 0,
            ]));
        } else {
            $dbForProject->updateDocument('stats', $document->getId(),
                $document->setAttribute('value', $value));
        }
        $latestData[$id]["30m"] = $time;
    }

    // Network
    $result = $database->query('SELECT sum(value) AS "value" FROM "appwrite_usage_network_all" WHERE time > \'' . $start . '\' AND time < \'' . $end . '\' AND "metric_type"=\'counter\' AND "project"=\'' . $projectId . '\'GROUP BY time(' . $period . ') FILL(null)');
    $points = $result->getPoints();

    foreach ($points as $point) {
        $network[] = [
            'value' => (!empty($point['value'])) ? $point['value'] : 0,
            'date' => \strtotime($point['time']),
            'dateStr' => $point['time'],
            'point' => $point,
        ];
        $time = \strtotime($point['time']);
        $id = \md5($time . '_' . $period . '_network');
        $value = (!empty($point['value'])) ? $point['value'] : 0;
        $document = $dbForProject->getDocument('stats', $id);
        if ($document->isEmpty()) {
            $dbForProject->createDocument('stats', new Document([
                '$id' => $id,
                'period' => $period,
                'time' => $time,
                'metric' => 'network',
                'value' => $value,
                'type' => 0,
            ]));
        } else {
            $dbForProject->updateDocument('stats', $document->getId(),
                $document->setAttribute('value', $value));
        }
        $latestData[$id]["30m"] = $time;
    }

    // Functions
    $result = $database->query('SELECT sum(value) AS "value" FROM "appwrite_usage_executions_all" WHERE time > \'' . $start . '\' AND time < \'' . $end . '\' AND "metric_type"=\'counter\' AND "project"=\'' . $id . '\'GROUP BY time(' . $period . ') FILL(null)');
    $points = $result->getPoints();

    foreach ($points as $point) {
        $functions[] = [
            'value' => (!empty($point['value'])) ? $point['value'] : 0,
            'date' => \strtotime($point['time']),
            'dateStr' => $point['time'],
            'point' => $point,
        ];
        $time = \strtotime($point['time']);
        $id = \md5($time . '_' . $period . '_functions');
        $value = (!empty($point['value'])) ? $point['value'] : 0;
        $document = $dbForProject->getDocument('stats', $id);
        if ($document->isEmpty()) {
            $dbForProject->createDocument('stats', new Document([
                '$id' => $id,
                'period' => $period,
                'time' => $time,
                'metric' => 'functions',
                'value' => $value,
                'type' => 0,
            ]));
        } else {
            $dbForProject->updateDocument('stats', $document->getId(),
                $document->setAttribute('value', $value));
        }
        $latestData[$id]["30m"] = $time;
    }
}