RenderService.cpp

//
// Created by epiphyllum on 22/04/22.
//

#include <iostream>
#include <algorithm>
#include <utility>
#include "RenderService.h"
#include "Utility/Util.h"
#include "Utility/UiUtil.h"
#include "HacknetApplication.h"
#include "AsciiArt.h"

const AsciiArt serverLogos[] = {
        AsciiArt("ASCII/laptop.ascii"),
        AsciiArt("ASCII/server.ascii"),
};

void RenderService::RenderTick()
{

    if (requireUpdate)
    {
        Util::clearScreen();
        UIUtil::drawFramework();
    }
    RenderMemory();
    RenderStatusBar();
    RenderTerminal();

    requireUpdate = false;
}

void RenderService::RenderTerminal()
{
    // Area buffer
    if (requireUpdate)
    {
        int buffSize = std::min(UIUtil::SIZE_TERMINAL.height - 2, static_cast<int>(ref->commandBuffer.size()));
        int h = 0;
        for (auto it = ref->commandBuffer.end() - buffSize; it != ref->commandBuffer.end(); ++it)
        {
            Util::setCursorPos(UIUtil::START_TERMINAL + Coord(0, h++));
            std::cout << *it;
        }
    }
    else
    {
        // only update prompt
        Util::clearLine(UIUtil::START_TERMINAL + Coord(0, UIUtil::SIZE_TERMINAL.height - 1),
                        UIUtil::SIZE_TERMINAL.width);
    }

    // Area prompt
    ref->inputService.renderCMD(ref->getPrompt(), UIUtil::START_TERMINAL + Coord(0, UIUtil::SIZE_TERMINAL.height - 1));
}

void RenderService::RenderStatusBar()
{
    // write time

    if (ref->CurrentConnected == nullptr)
    {
        // disconnected
        Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(84, 2));
        std::cout << "已断开连接";
    }
    else
    {
        // connected
        serverLogos[ref->CurrentConnected->getName().size() % 2].draw(UIUtil::START_STATUSPANEL);
        Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(11, 0));
        std::cout << "连接到";
        Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(11, 1));
        std::cout << ref->CurrentConnected->getName();
        Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(11, 2));
        std::cout << "@" << ref->CurrentConnected->getIp();

        if (ref->CurrentConnected->isAccessible())
        {
            // :: 您是本系统的管理员 ::
            Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(11, 4));
            Util::setColorAttr(Util::BG_LIGHT_BLUE);
            Util::setColorAttr(Util::FG_BLACK);
            // 72
            for (int i = 0; i < 72; ++i)
            {
                std::cout << ' ';
            }
            std::cout << ":: 您是本系统的管理员 ::";
            for (int i = 0; i < 72; ++i)
            {
                std::cout << ' ';
            }
            Util::setColorAttr(Util::ATTR_NORMAL);
        }
        else if (ref->nmapDetail)
        {
            // show nmap detail
            Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(100, 0));
            Util::setColorAttr(ref->CurrentConnected->isHttpLocked() ? Util::FG_RED : Util::FG_GREEN);
            std::cout << "#80: HTTP";
            Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(100, 1));
            Util::setColorAttr(ref->CurrentConnected->isSmtpLocked() ? Util::FG_RED : Util::FG_GREEN);
            std::cout << "#25: SMTP";
            Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(100, 2));
            Util::setColorAttr(ref->CurrentConnected->isFtpLocked() ? Util::FG_RED : Util::FG_GREEN);
            std::cout << "#21:  FTP";
            Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(100, 3));
            Util::setColorAttr(ref->CurrentConnected->isSshLocked() ? Util::FG_RED : Util::FG_GREEN);
            std::cout << "#22:  SSH";
            Util::setColorAttr(Util::ATTR_NORMAL);

            if (ref->CurrentConnected->getShellLife() != 0)
            {
                Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(110, 1));
                // detected shell
                int prog = std::min(std::max(ref->shellProgress, 0), ref->getCurrentConnected()->getShellLife()) * 30 /
                           ref->CurrentConnected->getShellLife();
                std::string text = ref->shellProgress == -1 ? "Proxy detected" : (ref->shellProgress >
                                                                                  ref->CurrentConnected->getShellLife()
                                                                                  ? "Proxy failed"
                                                                                  : "Proxy overloading...");
                Util::setColorAttr(Util::FG_WHITE);
                Util::setColorAttr(Util::BG_LIGHT_GREEN);
                for (int i = 0; i < 30; ++i)
                {
                    if (i == prog)
                    {
                        Util::setColorAttr(Util::BG_LIGHT_RED);
                    }
                    std::cout << (i >= text.size() ? ' ' : text[i]);
                }

            }

            Util::setCursorPos(UIUtil::START_STATUSPANEL + Coord(110, 2));
            Util::setColorAttr(
                    ref->CurrentConnected->checkIfSecureBroken(ref->shellProgress) ? Util::BG_GREEN : Util::BG_RED);
            Util::setColorAttr(Util::FG_WHITE);
            std::cout << "Open ports required: " << ref->CurrentConnected->getMinRequired();
            Util::clearLine(Util::getCursorPos(),
                            (UIUtil::START_STATUSPANEL + Coord(140, 2)).x - Util::getCursorPos().x);
            Util::setColorAttr(Util::ATTR_NORMAL);

        }
    }
}

RenderService::RenderService(HacknetApplication *ref) : ref(ref)
{

}

void RenderService::RenderMemory()
{
    int totalHeight = 0;
    for (auto &item: ref->backgroundTasks)
    {
        if (item->isStopped())
        {
            continue;
        }
        // Draw Title
        Util::setCursorPos(UIUtil::START_MEMORYPANEL + Coord(0, totalHeight));
        Util::setColorAttr(Util::BG_LIGHT_BLUE);
        Util::setColorAttr(Util::FG_BLACK);
        std::cout << "App: " << item->getThreadName();
        Util::clearLine(Util::getCursorPos(),
                        UIUtil::SIZE_MEMORYPANEL.width - Util::getCursorPos().x + UIUtil::START_MEMORYPANEL.x);
        Util::setColorAttr(Util::ATTR_NORMAL);
        item->renderMemory(UIUtil::START_MEMORYPANEL + Coord(0, totalHeight + 1));
        totalHeight += item->getMemorySize() + 1;
    }
    // clear empty area
    Util::clearArea(UIUtil::START_MEMORYPANEL + Coord(0, totalHeight),
                    Size2D(UIUtil::SIZE_MEMORYPANEL.width, UIUtil::SIZE_MEMORYPANEL.height - totalHeight));
}

bool RenderService::isRequireUpdate() const
{
    return requireUpdate;
}

void RenderService::setRequireUpdate(bool requireUpdate)
{
    RenderService::requireUpdate = requireUpdate;
}