<?php

namespace Pterodactyl\Http\Controllers\Admin;

use Illuminate\Http\Request;
use Illuminate\Support\Facades\DB;
use JavaScript;
use Pterodactyl\Http\Controllers\Controller;
use Pterodactyl\Models\Allocation;
use Pterodactyl\Models\Database;
use Pterodactyl\Models\Egg;
use Pterodactyl\Models\Node;
use Pterodactyl\Models\Server;
use Pterodactyl\Models\User;
use Pterodactyl\Services\DaemonKeys\DaemonKeyProviderService;

class StatisticsController extends Controller
{

    private $keyProviderService;

    function __construct(DaemonKeyProviderService $keyProviderService)
    {
        $this->keyProviderService = $keyProviderService;
    }

    public function index(Request $request)
    {
        $servers = Server::all();
        $nodes = Node::all();
        $serversCount = count($servers);
        $nodesCount = count($nodes);
        $usersCount = User::count();
        $eggsCount = Egg::count();
        $databasesCount = Database::count();
        $totalServerRam = DB::table('servers')->sum('memory');
        $totalNodeRam = DB::table('nodes')->sum('memory');
        $totalServerDisk = DB::table('servers')->sum('disk');
        $totalNodeDisk = DB::table('nodes')->sum('disk');
        $totalAllocations = Allocation::count();

        $suspendedServersCount = Server::where('suspended', true)->count();

        $tokens = [];
        foreach ($nodes as $node) {
            $server = Server::where('node_id', $node->id)->first();
            if ($server == null)
                continue;

            $tokens[$node->id] = $this->keyProviderService->handle($server, $request->user());
        }

        Javascript::put([
            'servers' => $servers,
            'serverCount' => $serversCount,
            'suspendedServers' => $suspendedServersCount,
            'totalServerRam' => $totalServerRam,
            'totalNodeRam' => $totalNodeRam,
            'totalServerDisk' => $totalServerDisk,
            'totalNodeDisk' => $totalNodeDisk,
            'nodes' => $nodes,
            'tokens' => $tokens,
        ]);

        return view('admin.statistics', [
            'serversCount' => $serversCount,
            'nodesCount' => $nodesCount,
            'usersCount' => $usersCount,
            'eggsCount' => $eggsCount,
            'totalServerRam' => $totalServerRam,
            'databasesCount' => $databasesCount,
            'totalNodeRam' => $totalNodeRam,
            'totalNodeDisk' => $totalNodeDisk,
            'totalServerDisk' => $totalServerDisk,
            'totalAllocations' => $totalAllocations,
        ]);
    }

}