use std::collections::BTreeMap;

use geom::{Angle, Circle, Distance, Speed, Time};
use map_gui::render::DrawPedestrian;
use map_model::{BuildingID, LaneID, OffstreetParking, Traversable, SIDEWALK_THICKNESS};
use sim::{DrawPedestrianInput, PedestrianID, PersonID, TripMode, TripResult, VehicleType};
use widgetry::{Color, EventCtx, Line, Text, TextExt, Widget};

use crate::app::App;
use crate::info::{header_btns, make_table, make_tabs, Details, Tab};

pub fn info(ctx: &mut EventCtx, app: &App, details: &mut Details, id: BuildingID) -> Widget {
    Widget::custom_col(vec![
        header(ctx, app, details, id, Tab::BldgInfo(id)),
        info_body(ctx, app, details, id).tab_body(ctx),
    ])
}

fn info_body(ctx: &mut EventCtx, app: &App, details: &mut Details, id: BuildingID) -> Widget {
    let mut rows = vec![];

    let b = app.primary.map.get_b(id);

    let mut kv = vec![("Address", b.address.clone())];
    if let Some(ref names) = b.name {
        kv.push(("Name", names.get(app.opts.language.as_ref()).to_string()));
    }
    if app.opts.dev {
        kv.push(("OSM ID", format!("{}", b.orig_id.inner())));
    }

    let num_spots = b.num_parking_spots();
    if app.primary.sim.infinite_parking() {
        kv.push((
            "Parking",
            format!(
                "Unlimited, currently {} cars inside",
                app.primary.sim.bldg_to_parked_cars(b.id).len()
            ),
        ));
    } else if num_spots > 0 {
        let free = app.primary.sim.get_free_offstreet_spots(b.id).len();
        if let OffstreetParking::PublicGarage(ref n, _) = b.parking {
            kv.push((
                "Parking",
                format!("{} / {} public spots available via {}", free, num_spots, n),
            ));
        } else {
            kv.push((
                "Parking",
                format!("{} / {} private spots available", free, num_spots),
            ));
        }
    } else {
        kv.push(("Parking", "None".to_string()));
    }

    rows.extend(make_table(ctx, kv));

    let mut txt = Text::new();

    if !b.amenities.is_empty() {
        txt.add_line("");
        if b.amenities.len() == 1 {
            txt.add_line("1 amenity:");
        } else {
            txt.add_line(format!("{} amenities:", b.amenities.len()));
        }
        for a in &b.amenities {
            txt.add_line(format!(
                "  {} ({})",
                a.names.get(app.opts.language.as_ref()),
                a.amenity_type
            ));
        }
    }

    if !app.primary.sim.infinite_parking() {
        txt.add_line("");
        if let Some(pl) = app
            .primary
            .sim
            .walking_path_to_nearest_parking_spot(&app.primary.map, id)
            .and_then(|path| path.trace(&app.primary.map))
        {
            let color = app.cs.parking_trip;
            // TODO But this color doesn't show up well against the info panel...
            txt.add_line(Line("Nearest parking").fg(color));
            txt.append(Line(format!(
                " is ~{} away by foot",
                pl.length() / Speed::miles_per_hour(3.0)
            )));

            details
                .unzoomed
                .push(color, pl.make_polygons(Distance::meters(10.0)));
            details.zoomed.extend(
                color,
                pl.dashed_lines(
                    Distance::meters(0.75),
                    Distance::meters(1.0),
                    Distance::meters(0.4),
                ),
            );
        } else {
            txt.add_line("No nearby parking available")
        }
    }

    if !txt.is_empty() {
        rows.push(txt.into_widget(ctx))
    }

    if app.opts.dev {
        rows.push(
            ctx.style()
                .btn_outline
                .text("Open OSM")
                .build_widget(ctx, format!("open {}", b.orig_id)),
        );

        if !b.osm_tags.is_empty() {
            rows.push("Raw OpenStreetMap data".text_widget(ctx));
            rows.extend(make_table(
                ctx,
                b.osm_tags
                    .inner()
                    .iter()
                    .map(|(k, v)| (k, v.to_string()))
                    .collect(),
            ));
        }
    }

    Widget::col(rows)
}

pub fn people(ctx: &mut EventCtx, app: &App, details: &mut Details, id: BuildingID) -> Widget {
    Widget::custom_col(vec![
        header(ctx, app, details, id, Tab::BldgPeople(id)),
        people_body(ctx, app, details, id).tab_body(ctx),
    ])
}

fn people_body(ctx: &mut EventCtx, app: &App, details: &mut Details, id: BuildingID) -> Widget {
    let mut rows = vec![];

    // Two caveats about these counts:
    // 1) A person might use multiple modes through the day, but this just picks a single category.
    // 2) Only people currently in the building currently are counted, whether or not that's their
    //    home.
    let mut drivers = 0;
    let mut cyclists = 0;
    let mut others = 0;

    let mut ppl: Vec<(Time, Widget)> = Vec::new();
    for p in app.primary.sim.bldg_to_people(id) {
        let person = app.primary.sim.get_person(p);

        let mut has_car = false;
        let mut has_bike = false;
        for vehicle in &person.vehicles {
            if vehicle.vehicle_type == VehicleType::Car {
                has_car = true;
            } else if vehicle.vehicle_type == VehicleType::Bike {
                has_bike = true;
            }
        }
        if has_car {
            drivers += 1;
        } else if has_bike {
            cyclists += 1;
        } else {
            others += 1;
        }

        let mut next_trip: Option<(Time, TripMode)> = None;
        for t in &person.trips {
            match app.primary.sim.trip_to_agent(*t) {
                TripResult::TripNotStarted => {
                    let trip = app.primary.sim.trip_info(*t);
                    next_trip = Some((trip.departure, trip.mode));
                    break;
                }
                TripResult::Ok(_) | TripResult::ModeChange => {
                    // TODO What to do here? This is meant for building callers right now
                    break;
                }
                TripResult::TripDone | TripResult::TripCancelled => {}
                TripResult::TripDoesntExist => unreachable!(),
            }
        }

        details
            .hyperlinks
            .insert(p.to_string(), Tab::PersonTrips(p, BTreeMap::new()));
        let widget = Widget::row(vec![
            ctx.style().btn_outline.text(p.to_string()).build_def(ctx),
            if let Some((t, mode)) = next_trip {
                format!(
                    "Leaving in {} to {}",
                    t - app.primary.sim.time(),
                    mode.verb()
                )
                .text_widget(ctx)
            } else {
                "Staying inside".text_widget(ctx)
            },
        ]);
        ppl.push((
            next_trip
                .map(|(t, _)| t)
                .unwrap_or_else(|| app.primary.sim.get_end_of_day()),
            widget,
        ));
    }

    // Sort by time to next trip
    ppl.sort_by_key(|(t, _)| *t);
    if ppl.is_empty() {
        rows.push("Nobody's inside right now".text_widget(ctx));
    } else {
        rows.push(
            format!(
                "{} drivers, {} cyclists, {} others",
                drivers, cyclists, others
            )
            .text_widget(ctx),
        );

        for (_, w) in ppl {
            rows.push(w);
        }
    }

    Widget::col(rows)
}

fn header(ctx: &EventCtx, app: &App, details: &mut Details, id: BuildingID, tab: Tab) -> Widget {
    let rows = vec![
        Widget::row(vec![
            Line(id.to_string()).small_heading().into_widget(ctx),
            header_btns(ctx),
        ]),
        make_tabs(
            ctx,
            &mut details.hyperlinks,
            tab,
            vec![("Info", Tab::BldgInfo(id)), ("People", Tab::BldgPeople(id))],
        ),
    ];

    draw_occupants(details, app, id, None);
    // TODO Draw cars parked inside?

    Widget::custom_col(rows)
}

pub fn draw_occupants(details: &mut Details, app: &App, id: BuildingID, focus: Option<PersonID>) {
    // TODO Lots of fun ideas here. Have a deterministic simulation based on building ID and time
    // to have people "realistically" move around. Draw little floor plans.

    let mut ppl = app.primary.sim.bldg_to_people(id);
    let num_rows_cols = (ppl.len() as f64).sqrt().ceil() as usize;

    let ped_len = SIDEWALK_THICKNESS.inner_meters() / 2.0;
    let separation = ped_len * 1.5;

    let total_width_height = (num_rows_cols as f64) * (ped_len + separation);
    let top_left = app
        .primary
        .map
        .get_b(id)
        .label_center
        .offset(-total_width_height / 2.0, -total_width_height / 2.0);

    // TODO Current thing is inefficient and can easily wind up outside the building.

    'OUTER: for x in 0..num_rows_cols {
        for y in 0..num_rows_cols {
            let person = if let Some(p) = ppl.pop() {
                p
            } else {
                break 'OUTER;
            };
            let pos = top_left.offset(
                (x as f64) * (ped_len + separation),
                (y as f64) * (ped_len + separation),
            );

            if Some(person) == focus {
                details.zoomed.push(
                    Color::YELLOW.alpha(0.8),
                    Circle::new(pos, SIDEWALK_THICKNESS).to_polygon(),
                );
            }

            DrawPedestrian::geometry(
                &mut details.zoomed,
                &app.primary.sim,
                &app.cs,
                &DrawPedestrianInput {
                    // Lies
                    id: PedestrianID(person.0),
                    person,
                    pos,
                    facing: Angle::degrees(90.0),
                    waiting_for_turn: None,
                    preparing_bike: false,
                    // Both hands and feet!
                    waiting_for_bus: true,
                    on: Traversable::Lane(LaneID(0)),
                },
                0,
            );
        }
    }
}