#![no_std]
#![no_main]

mod gpio;

use esp_backtrace as _;
use esp_println::println;
use hal::{clock::ClockControl, peripherals::Peripherals, prelude::*, timer::TimerGroup, Rtc, Delay};
#[entry]
fn main() -> ! {
  unsafe 
  {
    for i in 4..10 
    {
      gpio::output_enable(i);
    }
  }
  
  let peripherals = Peripherals::take();
  let mut system = peripherals.SYSTEM.split();
  let clocks = ClockControl::boot_defaults(system.clock_control).freeze();

  // Disable the RTC and TIMG watchdog timers
  let mut rtc = Rtc::new(peripherals.RTC_CNTL);

  let timer_group0 = TimerGroup::new(
      peripherals.TIMG0,
      &clocks,
      &mut system.peripheral_clock_control,
  );

  let mut wdt0 = timer_group0.wdt;

  let timer_group1 = TimerGroup::new(
      peripherals.TIMG1,
      &clocks,
      &mut system.peripheral_clock_control,
  );

  let mut wdt1 = timer_group1.wdt;
  rtc.swd.disable();
  rtc.rwdt.disable();
  wdt0.disable();
  wdt1.disable();

  let mut btn_state = 1;
  let mut delay = Delay::new(&clocks);
  
  loop 
  {
    unsafe 
    {
      let btn = gpio::read(3);
      if btn != btn_state && btn == 0 {
        for i in 4..10 
        {
          gpio::write(i, 1);
          delay.delay_ms(32u8);
          gpio::write(i, 0);
          delay.delay_ms(32u8);
        }

        let mut j = 8;

        for i in 0..5
        {
          gpio::write(j, 1);
          delay.delay_ms(32u8);
          gpio::write(j, 0);
          delay.delay_ms(32u8);
          j -= 1;
        }

        btn_state = !btn_state;
      }
    }
  }
}