// Copyright 2022 Parity Technologies (UK) Ltd.
// This file is part of Polkadot.

// Polkadot 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.

// Polkadot 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 Polkadot.  If not, see <http://www.gnu.org/licenses/>.

use fatality::{fatality, Fatality};

#[derive(Debug, thiserror::Error)]
#[error("We tried")]
struct FatalX;

impl Fatality for FatalX {
	fn is_fatal(&self) -> bool {
		true
	}
}

#[derive(Debug, thiserror::Error)]
#[error("Get a dinosaur bandaid")]
struct Bobo;

#[fatality(splitable)]
enum Kaboom {
	#[fatal(forward)]
	#[error(transparent)]
	Iffy{#[from] x: FatalX},

	#[error(transparent)]
	Bobo{#[from] b: Bobo},
}

fn iffy() -> Result<(), Kaboom> {
	Err(FatalX)?
}

fn bobo() -> Result<(), Kaboom> {
	Err(Bobo)?
}

fn main() {
	assert!(iffy().unwrap_err().is_fatal());
	assert!(!bobo().unwrap_err().is_fatal());
}