use crabrolls::prelude::*;
use std::error::Error;

struct EchoApp;

impl EchoApp {
	fn new() -> Self {
		Self
	}
}

impl Application for EchoApp {
	async fn advance(
		&self,
		env: &impl Environment,
		metadata: Metadata,
		payload: &[u8],
		_deposit: Option<Deposit>,
	) -> Result<FinishStatus, Box<dyn Error>> {
		println!(
			"Advance method called with payload: {:?}",
			String::from_utf8_lossy(payload)
		);
		env.send_notice(payload).await?;
		env.send_report(payload).await?;
		env.send_voucher(metadata.sender, payload).await?;
		Ok(FinishStatus::Accept)
	}

	async fn inspect(&self, env: &impl Environment, payload: &[u8]) -> Result<FinishStatus, Box<dyn Error>> {
		println!(
			"Inspect method called with payload: {:?}",
			String::from_utf8_lossy(payload)
		);
		env.send_report(payload).await?;
		Ok(FinishStatus::Accept)
	}
}

#[async_std::main]
async fn main() {
	let app = EchoApp::new();
	let options = RunOptions::default();
	if let Err(e) = Supervisor::run(app, options).await {
		eprintln!("Error: {}", e);
	}
}

#[cfg(test)]
mod tests {
	use super::EchoApp;
	use crabrolls::prelude::*;
	use ethabi::Address;

	#[async_std::test]
	async fn test_echo() {
		let app = EchoApp::new();
		let tester = Tester::new(app, MockupOptions::default());

		let address = Address::default();

		let payload = b"Hi Crabrolls!";
		let result = tester.advance(address, payload).await;

		assert!(result.is_accepted(), "Expected Accept status");

		assert!(!result.is_errored(), "Expected no error");

		assert_eq!(
			result.get_outputs().len(),
			3,
			"Expected 3 outputs, got {}",
			result.get_outputs().len()
		);

		assert_eq!(
			result.get_outputs(),
			vec![
				Output::Notice {
					payload: payload.to_vec()
				},
				Output::Report {
					payload: payload.to_vec()
				},
				Output::Voucher {
					destination: address,
					payload: payload.to_vec()
				}
			],
			"Expected outputs to match"
		);

		assert_eq!(
			result.get_metadata().sender,
			Address::default(),
			"Unexpected sender address"
		);
	}
}