use crate::constants;
use crate::devices::device::{Device, DeviceId, DeviceRelayID};
use crate::energy::consumption::EnergyConsumption;
use crate::utils::time_utils::time_secs;
use prettytable::{row, Table};
use std::collections::HashMap;

#[derive(Default)]
pub struct DeviceState {
    pub last_ping: u64,
}

impl DeviceState {
    pub fn record_ping(&mut self) {
        self.last_ping = time_secs();
    }

    pub fn is_online(&self) -> bool {
        (time_secs() - self.last_ping) < constants::DEVICE_MAX_PING_TIME
    }
}

#[derive(Default, Clone)]
struct RelayState {
    on: bool,
    since: usize,
}

#[derive(Default)]
pub struct EnergyEngine {
    devices_state: HashMap<DeviceId, DeviceState>,
    relays_state: HashMap<DeviceRelayID, RelayState>,
}

impl EnergyEngine {
    pub fn device_state(&mut self, dev_id: &DeviceId) -> &mut DeviceState {
        if !self.devices_state.contains_key(dev_id) {
            self.devices_state
                .insert(dev_id.clone(), Default::default());
        }

        self.devices_state.get_mut(dev_id).unwrap()
    }

    pub fn relay_state(&mut self, relay_id: DeviceRelayID) -> &mut RelayState {
        if !self.relays_state.contains_key(&relay_id) {
            self.relays_state.insert(relay_id, Default::default());
        }

        self.relays_state.get_mut(&relay_id).unwrap()
    }

    fn print_summary(&mut self, curr_consumption: EnergyConsumption, devices: &[Device]) {
        log::info!("Current consumption: {curr_consumption}");

        let mut table = Table::new();
        table.add_row(row!["Device", "Relay", "On", "Since"]);
        for d in devices {
            for r in &d.relays {
                let status = self.relay_state(r.id);
                table.add_row(row![d.name, r.name, status.on.to_string(), status.since]);
            }
        }
        table.printstd();
    }

    pub fn refresh(&mut self, curr_consumption: EnergyConsumption, devices: &[Device]) {
        let new_relays_state = self.relays_state.clone();

        // Forcefully turn off relays that belongs to offline devices

        // Forcefully turn off relays with dependency conflicts

        // Virtually turn off all relays that can be stopped

        // Turn on relays based on priority / dependencies

        // Turn on relays with running constraints

        // Commit changes
        self.print_summary(curr_consumption, devices); // TODO :replace
    }
}