1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use bevy::prelude::*;

mod bootstrap;
mod data;
mod fuzz_input;
mod input;
mod math;
mod output;
mod runner;
mod window;

pub use bootstrap::FuzzTarget;
pub use bootstrap::{bin_bootstrap, fuzz_bootstrap};
pub use output::{parse_commands, EventOutputPlugin};
pub use runner::fuzz_runner;

pub mod prelude {
    pub use crate::{
        bootstrap::{bin_bootstrap, fuzz_bootstrap, FuzzTarget},
        data::FuzzData,
        fuzz_input::FuzzInput,
        FuzzPlugin,
    };
}

#[derive(Default)]
pub struct FuzzPlugin {}

impl FuzzPlugin {
    pub fn new() -> Self {
        Self {}
    }
}

const CLEAN_STAGE_LABEL: &str = "fuzz_clean_startup_system";

impl Plugin for FuzzPlugin {
    fn build(&self, app: &mut App) {
        app.set_runner(|_| {
            panic!("use bevy_fuzz::runner::fuzz_runner(&mut app) instead");
        })
        .add_startup_stage_before(
            StartupStage::PreStartup,
            CLEAN_STAGE_LABEL,
            SystemStage::parallel(),
        )
        .add_startup_system_to_stage(CLEAN_STAGE_LABEL, clean_all_system);
    }
}

fn clean_all_system(mut commands: Commands, all_entities: Query<Entity>) {
    for entity in all_entities.iter() {
        commands.entity(entity).despawn_recursive();
    }
}