use super::{Message, ResultMessage};
use std::collections::{HashMap, HashSet};
use std::sync::mpsc::{channel, Receiver, Sender};
use std::thread::JoinHandle;

struct InBuffer {
    receiver: Receiver<Message>,
    row_requests: HashMap<u64, Vec<Vec<u32>>>,
    results_requests: HashMap<u64, ResultMessage>,
}

impl InBuffer {
    fn new(receiver: Receiver<Message>) -> Self {
        Self {
            receiver,
            row_requests: HashMap::new(),
            results_requests: HashMap::new(),
        }
    }
    fn read(&mut self) -> Option<Vec<RowResult>> {
        loop {
            match self
                .receiver
                .recv()
                .expect("Channel to Output Daemon broke")
            {
                Message::ResultMessage(results) => {
                    if let Some(result_walls) = self.row_requests.get(&results.id) {
                        return Some(Self::calc_results(results.valid_walls(), result_walls));
                    } else {
                        self.results_requests.insert(results.id, results);
                    }
                }
                Message::OutputMessage((id, output)) => {
                    if let Some(results) = self.results_requests.get(&id) {
                        return Some(Self::calc_results(results.valid_walls(), output.as_ref()));
                    } else {
                        self.row_requests.insert(id, output);
                    }
                }
                Message::Terminate => {
                    return None;
                }
                _ => {
                    println!("Invalid MessageType");
                }
            }
        }
    }
    fn calc_results(res_req: &[Vec<u32>], row_req: &[Vec<u32>]) -> Vec<RowResult> {
        let out = Vec::new();
        for (rows, perms) in row_req.iter().zip(res_req.iter()) {
            for p in perms {
                let new = rows.clone();
                new.push(*p);
                out.push(RowResult::new(new));
            }
        }
        out
    }
}

#[derive(PartialEq, Eq, Hash)]
pub struct RowResult {
    rows: Vec<u32>,
}

impl RowResult {
    fn new(rows: Vec<u32>) -> Self {
        rows.push(0);
        Self { rows }
    }
    fn output(&self) {
        println!("{:?}", self.rows);
    }
}

pub struct Output {
    input: InBuffer,
    permutations: Vec<Vec<u32>>,
    permutations_mask: Vec<u64>,
    results: HashSet<RowResult>,
}

impl Output {
    pub fn launch_sevice(
        permutations: &[Vec<u32>],
        permutations_mask: &[u64],
    ) -> (Sender<Message>, JoinHandle<()>) {
        let (sender, receiver) = channel();
        let input = InBuffer::new(receiver);

        let output = Self {
            input,
            permutations: permutations.into(),
            permutations_mask: permutations_mask.into(),
            results: HashSet::new(),
        };
        (
            sender,
            std::thread::Builder::new()
                .name("GPU Manager Deamon".into())
                .spawn(move || {
                    output.run();
                })
                .unwrap(),
        )
    }

    fn run(mut self) {
        loop {
            if let Some(walls) = self.input.read() {
                for wall in walls {
                    self.results.insert(wall);
                }
            } else {
                for wall in self.results {
                    wall.output()
                }
            }
        }
    }
}