use std::{
    fs::File,
    io::{BufRead, BufReader},
};
use tempfile::Builder;

#[test]
pub fn osm_fixture_stoppoints() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let stops = osm_transit_extractor::get_stop_points_from_osm(&mut parsed_pbf);
    assert_eq!(stops[0].id, "node:260743996");
    assert_eq!(stops.len(), 77);
}

#[test]
pub fn osm_fixture_stoppoints_categorization() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let mut stop_points = osm_transit_extractor::get_stop_points_from_osm(&mut parsed_pbf);
    let routes = osm_transit_extractor::get_routes_from_osm(&mut parsed_pbf);
    osm_transit_extractor::update_stop_points_type(&mut stop_points, &routes);
    let stop_points_unknown = stop_points
        .iter()
        .filter(|s| s.stop_point_type == osm_transit_extractor::StopPointType::Unknown);
    let stop_points_platform = stop_points
        .iter()
        .filter(|s| s.stop_point_type == osm_transit_extractor::StopPointType::Platform);
    assert_eq!(stop_points_unknown.count(), 52);
    assert_eq!(stop_points_platform.count(), 12);
}

#[test]
pub fn osm_fixture_stopareas() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let stop_areas = osm_transit_extractor::get_stop_areas_from_osm(&mut parsed_pbf);
    assert_eq!(stop_areas.len(), 1);
}

#[test]
pub fn osm_fixture_stationaccesses() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/station.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let accesses = osm_transit_extractor::get_station_accesses_from_osm(&mut parsed_pbf);
    assert_eq!(accesses.len(), 5);
    assert!(accesses
        .iter()
        .map(|access| &access.id)
        .any(|x| x == "node:6024574088"));
}

#[test]
pub fn osm_fixture_stopareas_stationaccesses() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/station.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let stop_areas = osm_transit_extractor::get_stop_areas_from_osm(&mut parsed_pbf);
    assert_eq!(stop_areas.len(), 1);
    assert!(stop_areas[0]
        .station_access_ids
        .contains(&String::from("node:6024574088")));
}

#[test]
pub fn osm_fixture_routes_count() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let routes = osm_transit_extractor::get_routes_from_osm(&mut parsed_pbf);
    assert_eq!(routes.len(), 3);
    for r in routes {
        if r.id == "relation:1257168" {
            assert_eq!(r.ordered_route_points.len(), 34);
            assert_eq!(r.ordered_route_points[0].stop_point_id, "node:3270784465");
            assert_eq!(r.ordered_route_points[30].stop_point_id, "node:1577028157");
        }
    }
}

#[test]
pub fn osm_fixture_lines_count() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let lines = osm_transit_extractor::get_lines_from_osm(&mut parsed_pbf);
    assert_eq!(lines.len(), 1);
    assert_eq!(lines[0].routes_id.len(), 2);
}

#[test]
pub fn osm_fixture_routes_tags() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let routes = osm_transit_extractor::get_routes_from_osm(&mut parsed_pbf);
    for r in routes {
        match r.id.as_ref() {
            "relation:1257168" => {
                assert_eq!(r.colour, String::new());
                assert_eq!(r.operator, "RATP".to_string());
                assert_eq!(r.network, "RATP".to_string());
                assert_eq!(r.mode, "bus".to_string());
                assert_eq!(r.code, "57".to_string());
                assert_eq!(r.origin, "Arcueil - Laplace".to_string());
            }
            "relation:1257174" => {
                assert_eq!(r.destination, "Arcueil - Laplace".to_string());
            }
            _ => {}
        }
    }
}

#[test]
pub fn osm_fixture_lines_tags() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let lines = osm_transit_extractor::get_lines_from_osm(&mut parsed_pbf);
    assert_eq!(lines[0].colour, "#9C983A".to_string());
    assert_eq!(lines[0].operator, "RATP".to_string());
    assert_eq!(lines[0].network, "RATP".to_string());
    assert_eq!(lines[0].mode, "bus".to_string());
    assert_eq!(lines[0].code, "57".to_string());
}

#[test]
pub fn osm_fixture_stoppoints_csv() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let stops = osm_transit_extractor::get_stop_points_from_osm(&mut parsed_pbf);
    let tmp_dir = Builder::new()
        .prefix("osm_transit_extractor")
        .tempdir()
        .expect("create temp dir");
    osm_transit_extractor::write_stop_points_to_csv(&stops, &tmp_dir, false);
    let file_path = tmp_dir.path().join("osm-transit-extractor_stop_points.csv");
    assert!(file_path.is_file());
    let file = File::open(file_path).unwrap();
    let reader = BufReader::new(file);
    assert_eq!(78, reader.lines().count());

    tmp_dir.close().expect("delete temp dir");
}

#[test]
pub fn osm_fixture_stationaccesses_csv() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/station.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let accesses = osm_transit_extractor::get_station_accesses_from_osm(&mut parsed_pbf);
    let tmp_dir = Builder::new()
        .prefix("osm_transit_extractor")
        .tempdir()
        .expect("create temp dir");
    osm_transit_extractor::write_station_accesses_to_csv(&accesses, &tmp_dir, false);
    let file_path = tmp_dir
        .path()
        .join("osm-transit-extractor_station_accesses.csv");
    assert!(file_path.is_file());
    let file = File::open(file_path).unwrap();
    let reader = BufReader::new(file);
    assert_eq!(6, reader.lines().count());

    tmp_dir.close().expect("delete temp dir");
}

#[test]
pub fn osm_fixture_stopareas_stoppoints_csv() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let stop_areas = osm_transit_extractor::get_stop_areas_from_osm(&mut parsed_pbf);
    let tmp_dir = Builder::new()
        .prefix("osm_transit_extractor")
        .tempdir()
        .expect("create temp dir");
    osm_transit_extractor::write_stop_areas_stop_point_to_csv(&stop_areas, &tmp_dir);
    let file_path = tmp_dir
        .path()
        .join("osm-transit-extractor_stop_areas_stop_point.csv");
    assert!(file_path.is_file());
    let file = File::open(file_path).unwrap();
    let reader = BufReader::new(file);
    assert_eq!(3, reader.lines().count());
    tmp_dir.close().expect("delete temp dir");
}

#[test]
pub fn osm_fixture_stopareas_stationaccesses_csv() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/station.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let stop_areas = osm_transit_extractor::get_stop_areas_from_osm(&mut parsed_pbf);
    let tmp_dir = Builder::new()
        .prefix("osm_transit_extractor")
        .tempdir()
        .expect("create temp dir");
    osm_transit_extractor::write_stop_areas_station_accesses_to_csv(&stop_areas, &tmp_dir);
    let file_path = tmp_dir
        .path()
        .join("osm-transit-extractor_stop_areas_station_accesses.csv");
    assert!(file_path.is_file());
    let file = File::open(file_path).unwrap();
    let reader = BufReader::new(file);
    assert_eq!(6, reader.lines().count());
    tmp_dir.close().expect("delete temp dir");
}

#[test]
pub fn osm_fixture_routes_csv() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let routes = osm_transit_extractor::get_routes_from_osm(&mut parsed_pbf);
    let tmp_dir = Builder::new()
        .prefix("osm_transit_extractor")
        .tempdir()
        .expect("create temp dir");
    osm_transit_extractor::write_routes_to_csv(routes, &tmp_dir, true);
    let file_path = tmp_dir.path().join("osm-transit-extractor_routes.csv");
    assert!(file_path.is_file());
    let file = File::open(file_path).unwrap();
    let reader = BufReader::new(file);
    assert_eq!(4, reader.lines().count());
    tmp_dir.close().expect("delete temp dir");
}

#[test]
pub fn osm_fixture_lines_csv() {
    let osm_path = std::env::current_dir()
        .unwrap()
        .join("tests/fixtures/osm_fixture.osm.pbf");
    let mut parsed_pbf = osmpbfreader::OsmPbfReader::new(std::fs::File::open(osm_path).unwrap());
    let lines = osm_transit_extractor::get_lines_from_osm(&mut parsed_pbf);
    let tmp_dir = Builder::new()
        .prefix("osm_transit_extractor")
        .tempdir()
        .expect("create temp dir");
    osm_transit_extractor::write_lines_to_csv(lines, &tmp_dir, false);
    let file_path = tmp_dir.path().join("osm-transit-extractor_lines.csv");
    assert!(file_path.is_file());
    let file = File::open(file_path).unwrap();
    let reader = BufReader::new(file);
    assert_eq!(2, reader.lines().count());

    tmp_dir.close().expect("delete temp dir");
}