/*
   This file is part of Fennix Kernel.

   Fennix Kernel is free software: you can redistribute it and/or
   modify it under the terms of the GNU General Public License as
   published by the Free Software Foundation, either version 3 of
   the License, or (at your option) any later version.

   Fennix Kernel is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with Fennix Kernel. If not, see <https://www.gnu.org/licenses/>.
*/

#include <power.hpp>

#include <memory.hpp>
#include <acpi.hpp>
#include <debug.h>
#include <io.h>

#include "../kernel.h"

namespace Power
{
	void Power::Reboot()
	{
		if (((ACPI::ACPI *)this->acpi)->FADT)
			if (((ACPI::DSDT *)this->dsdt)->ACPIShutdownSupported)
				((ACPI::DSDT *)this->dsdt)->Reboot();

		uint8_t val = 0x02;
		while (val & 0x02)
			val = inb(0x64);
		outb(0x64, 0xFE);

		warn("Executing the second attempt to reboot...");

		// https://wiki.osdev.org/Reboot
		/* Second attempt to reboot */
		asmv("cli");
		uint8_t temp;
		do
		{
			temp = inb(0x64);
			if (((temp) & (1 << (0))) != 0)
				inb(0x60);
		} while (((temp) & (1 << (1))) != 0);
		outb(0x64, 0xFE);

		CPU::Stop();
	}

	void Power::Shutdown()
	{
		if (((ACPI::ACPI *)this->acpi)->FADT)
			if (((ACPI::DSDT *)this->dsdt)->ACPIShutdownSupported)
				((ACPI::DSDT *)this->dsdt)->Shutdown();
				/* TODO: If no ACPI, display "It is now safe to turn off your computer" */

				/* FIXME: Detect emulators and use their shutdown methods */
#ifdef DEBUG
		outl(0xB004, 0x2000); // for qemu
		outl(0x604, 0x2000);  // if qemu not working, bochs and older versions of qemu
		outl(0x4004, 0x3400); // virtual box
#endif
		CPU::Stop();
	}

	void Power::InitDSDT()
	{
		if (((ACPI::ACPI *)this->acpi)->FADT)
			this->dsdt = new ACPI::DSDT((ACPI::ACPI *)acpi);
	}

	Power::Power()
	{
		this->acpi = new ACPI::ACPI;
		this->madt = new ACPI::MADT(((ACPI::ACPI *)acpi)->MADT);
		trace("Power manager initialized");
	}
}