use actix::{Actor, Context, Handler, Message};
use emgauwa_lib::errors::EmgauwaError;
use emgauwa_lib::models::Controller;
use futures::executor::block_on;
use sqlx::{Pool, Sqlite};

#[derive(Message)]
#[rtype(result = "Result<(), EmgauwaError>")]
pub struct Reload {}

#[derive(Message)]
#[rtype(result = "Controller")]
pub struct GetThis {}

pub struct AppState {
	pub pool: Pool<Sqlite>,
	pub this: Controller,
}

impl AppState {
	pub fn new(pool: Pool<Sqlite>, this: Controller) -> AppState {
		AppState { pool, this }
	}
}

impl Actor for AppState {
	type Context = Context<Self>;
}

impl Handler<Reload> for AppState {
	type Result = Result<(), EmgauwaError>;

	fn handle(&mut self, _msg: Reload, _ctx: &mut Self::Context) -> Self::Result {
		let mut pool_conn = block_on(self.pool.acquire())?;

		self.this.reload(&mut pool_conn)?;

		Ok(())
	}
}

impl Handler<GetThis> for AppState {
	type Result = Controller;

	fn handle(&mut self, _msg: GetThis, _ctx: &mut Self::Context) -> Self::Result {
		self.this.clone()
	}
}