#![allow(dead_code)]
use std::collections::HashMap;
use std::fs::File;
use std::io;
use std::io::BufRead;
use std::path::Path;
use std::time::Instant;

#[derive(Debug, Clone)]
struct Folder {
    folders: HashMap<String, u32>,
    size: u32,
}

fn main() {
    let now = Instant::now();
    part_one();
    let part_one_duration = now.elapsed();
    part_two();
    let part_two_duration = now.elapsed();
    println!(
        "p1: {}ms, p2: {}ms",
        part_one_duration.as_millis(),
        (part_two_duration - part_one_duration).as_millis()
    );
}

fn read_lines<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>>
where
    P: AsRef<Path>,
{
    // note that this discards a final newline
    let file = File::open(filename)?;
    Ok(io::BufReader::new(file).lines())
}

fn part_one() {
    if let Ok(lines) = read_lines("./inputs/input") {
        let mut sum = 0;
        let mut stack: Vec<u32> = vec![0];
        for line in lines {
            if let Ok(command) = line {
                match command.as_str() {
                    "$ cd /" => continue,                   // skip start
                    "$ ls" => continue, // we can assume this if we're not moving between directories
                    s if s.starts_with("dir ") => continue, // we'll get there when we `cd`
                    "$ cd .." => {
                        let f = stack.pop().unwrap();
                        if f < 100000 {
                            sum += f
                        }
                        let mut p = stack.pop().unwrap();
                        p += f;
                        stack.push(p);
                    }
                    s if s.starts_with("$ cd ") => {
                        stack.push(0);
                    }
                    s => {
                        // otherwise it must be a file
                        let mut f = stack.pop().unwrap();
                        let mut sp = s.split(" ");
                        let size = sp.next().unwrap().parse::<u32>().unwrap();
                        f += size;
                        stack.push(f);
                    }
                };
            }
        }
        println!("{:?}", sum);
    }
}

fn part_two() {}