Heimdall-Apps/Traefik/Traefik.php

<?php

namespace App\SupportedApps\Traefik;

use GuzzleHttp\Exception\GuzzleException;
use GuzzleHttp\Client;
use Illuminate\Support\Collection;

class Traefik extends \App\SupportedApps implements \App\EnhancedApps
{
    public function getRequestAttrs()
    {
        $username = $this->getConfigValue("username", null);
        $password = $this->getConfigValue("password", null);
        $ignoreTls = $this->getConfigValue("ignore_tls", false);

        $isLoginNeeded =
            !is_null($username) &&
            !empty($username) &&
            !is_null($password) &&
            !empty($password);

        $attrs["headers"] = ["Accept" => "application/json"];
        if ($isLoginNeeded) {
            $attrs["auth"] = [$username, $password];
        }
        if ($ignoreTls) {
            $attrs["verify"] = false;
        }

        return $attrs;
    }

    public function test()
    {
        $attrs = $this->getRequestAttrs();
        $test = parent::appTest($this->url("api/version"), $attrs);

        echo $test->status;
    }

    public function livestats()
    {
        $apiEndpoints = collect([
            "httpRouters" => "api/http/routers",
            "httpServices" => "api/http/services",
            "tcpRouters" => "api/tcp/routers",
            "tcpServices" => "api/tcp/services",
        ]);

        $status = "active";
        $attrs = $this->getRequestAttrs();

        $responses = $apiEndpoints->mapWithKeys(function (
            $endpoint,
            $key
        ) use (
            $attrs
        ) {
            $response = parent::execute($this->url($endpoint), $attrs);
            $body = json_decode($response->getBody());
            $bodyCollection = collect($body);

            return [
                $key => [
                    "data" => $bodyCollection
                        ->filter(function ($value, $key) {
                            return $value->status === "enabled";
                        })
                        ->count(),
                    "total" => $bodyCollection->count(),
                ],
            ];
        });

        $data = $this->getViewData(
            $this->getConfigValue("fields", "E"),
            $responses
        );
        return parent::getLiveStats($status, $data);
    }

    public function getViewData($config, $responses)
    {
        $nullValue = ["data" => 0, "total" => 0];

        switch ($config) {
            /* HTTP routers/services only */
            case "H":
                return [
                    "left" => [
                        "title" => "Routers",
                        "data" => $responses->get("httpRouters", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("httpRouters", $nullValue)[
                            "total"
                        ],
                    ],
                    "right" => [
                        "title" => "Services",
                        "data" => $responses->get("httpServices", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("httpServices", $nullValue)[
                            "total"
                        ],
                    ],
                ];
                break;

            /* TCP routers/services only */
            case "T":
                return [
                    "left" => [
                        "title" => "Routers",
                        "data" => $responses->get("tcpRouters", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("tcpRouters", $nullValue)[
                            "total"
                        ],
                    ],
                    "right" => [
                        "title" => "Services",
                        "data" => $responses->get("tcpServices", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("tcpServices", $nullValue)[
                            "total"
                        ],
                    ],
                ];

            /* Routers only */
            case "R":
                return [
                    "left" => [
                        "title" => "HTTP",
                        "data" => $responses->get("httpRouters", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("httpRouters", $nullValue)[
                            "total"
                        ],
                    ],
                    "right" => [
                        "title" => "TCP",
                        "data" => $responses->get("tcpRouters", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("tcpRouters", $nullValue)[
                            "total"
                        ],
                    ],
                ];

            /* Services only */
            case "S":
                return [
                    "left" => [
                        "title" => "HTTP",
                        "data" => $responses->get("httpServices", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("httpServices", $nullValue)[
                            "total"
                        ],
                    ],
                    "right" => [
                        "title" => "TCP",
                        "data" => $responses->get("tcpServices", $nullValue)[
                            "data"
                        ],
                        "total" => $responses->get("tcpServices", $nullValue)[
                            "total"
                        ],
                    ],
                ];

            /* Everything */
            default:
                return [
                    "left" => [
                        "title" => "Routers",
                        "data" =>
                            $responses->get("httpRouters", $nullValue)["data"] +
                            $responses->get("tcpRouters", $nullValue)["data"],
                        "total" =>
                            $responses->get("httpRouters", $nullValue)[
                                "total"
                            ] +
                            $responses->get("tcpRouters", $nullValue)["total"],
                    ],
                    "right" => [
                        "title" => "Services",
                        "data" =>
                            $responses->get("httpServices", $nullValue)[
                                "data"
                            ] +
                            $responses->get("tcpServices", $nullValue)["data"],
                        "total" =>
                            $responses->get("httpServices", $nullValue)[
                                "total"
                            ] +
                            $responses->get("tcpServices", $nullValue)["total"],
                    ],
                ];
        }
    }

    public function url($endpoint)
    {
        $api_url = parent::normaliseurl($this->config->url) . $endpoint;
        return $api_url;
    }

    public function getConfigValue($key, $default = null)
    {
        return isset($this->config) && isset($this->config->$key)
            ? $this->config->$key
            : $default;
    }
}