use std::{collections::HashSet, fmt::Display};

#[derive(Debug, Hash, PartialEq, Eq)]
pub struct Point(pub isize, pub isize);

#[derive(Debug)]
pub enum Fold {
  Horizontal(isize),
  Vertical(isize),
}

#[derive(Debug)]
pub struct Canvas {
  pub width: isize,
  pub height: isize,
  pub points: HashSet<Point>,
}

impl Canvas {
  pub fn fold(self, fold: &Fold) -> Self {
    let (width, height) = match fold {
      Fold::Horizontal(amount) => (self.width, *amount),
      Fold::Vertical(amount) => (*amount, self.height),
    };

    let mut points = HashSet::new();
    for Point(x, y) in self.points {
      if y > height {
        points.insert(Point(x, height - (y - height)));
        continue;
      }

      if x > width {
        points.insert(Point(width - (x - width), y));
        continue;
      }

      points.insert(Point(x, y));
    }

    Self {
      width,
      height,
      points,
    }
  }
}

impl Display for Canvas {
  fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
    for y in 0..self.height {
      for x in 0..self.width {
        if x % (self.width / 8) == 0 {
          write!(f, " ")?;
        }

        if self.points.contains(&Point(x, y)) {
          write!(f, "█")?;
        } else {
          write!(f, " ")?;
        }
      }

      writeln!(f)?;
    }

    Ok(())
  }
}