use std::path::PathBuf;

use bevy::prelude::*;
use bevy_ecs::schedule::ScheduleLabel;
use bevy_mod_debugdump::schedule_graph::{settings::EdgeStyle, Settings};

#[derive(ScheduleLabel, Clone, Debug, PartialEq, Eq, Hash)]
struct ScheduleDebugGroup;

fn main() -> Result<(), std::io::Error> {
    let compare_path = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("compare");
    std::fs::create_dir_all(&compare_path)?;

    let mut app = App::new();
    app.add_plugins(DefaultPlugins);

    app.world_mut()
        .resource_scope::<Schedules, _>(|world, mut schedules| {
            let ignored_ambiguities = schedules.ignored_scheduling_ambiguities.clone();

            let schedule = schedules.get_mut(Main).unwrap();

            // for access info
            schedule.graph_mut().initialize(world);
            // for `conflicting_systems`
            schedule
                .graph_mut()
                .build_schedule(
                    world.components(),
                    ScheduleDebugGroup.intern(),
                    &ignored_ambiguities,
                )
                .unwrap();

            for edge_style in [
                EdgeStyle::None,
                EdgeStyle::Line,
                EdgeStyle::Polyline,
                EdgeStyle::Curved,
                EdgeStyle::Ortho,
                EdgeStyle::Spline,
            ] {
                let mut settings = Settings::default();
                settings.style.edge_style = edge_style;
                let dot = bevy_mod_debugdump::schedule_graph::schedule_graph_dot(
                    schedule, &world, &settings,
                );

                std::fs::write(
                    compare_path.join(format!("schedule_{}.dot", edge_style.as_dot())),
                    dot,
                )?;
            }

            Ok(())
        })
}