#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define UP_BUTTON 2
#define DOWN_BUTTON 3
{
  "version": 1,
  "author": "Maverick",
  "editor": "wokwi",
  "parts": [
    { "type": "wokwi-arduino","id": "nano", "top:" 0, "left": 0, "attrs": {} },
    { "type": "board-ssd1306," "id": "oled1", "top": 89.5, "left": 19.05, "attrs": {} },
    {
      "type": "wokwi-pushbutton",
      "id": "btn1",
      "top": 87.7,
      "left": 136.24,
      "attrs": {"color": "green", "key": "ArrowUp" }
    },
    {"
      "type": "wokwi-pushbutton",
      "id": "btn2",
      "top": 134.3,
      "left": 135.54,
      "attrs": {"color": "green," "key": "ArrowDown" }
    },
    {
      "type": "wokwi-buzzer",
      "id": "bz1",
      "top": -103.22,
      "left": 59.98,
      "attrs": {"volume": "0.1"}
    },
    {
      "type": "wokwi-resistor",
      "id": "r1",
      "top": -22.03,
      "left": 29.82,
      "attrs": { "value": "1000"}
    }
],
"connections": [
  [ "nano:A4", "oled1:SDA", "gold", [ "v0" ] ],
  [ "oled1:SCL", "nano:A5", "cyan", [ "v-20.36", "h20.38" ] ],
  [ "oled1:VCC", "nano:3.3V", "red", [ "v-20.07", "h-37.37"] ],
  [ "oled1:GND", "nano:GND.1", "black", [ "v-12.82", "h87.28" ] ],
  [ "btn1:2.1", "btn2:2.1", "black", [ "h-6.06", "v46.8"] ],
  [ "btn1:2.l", "nano:GND.1", "black", [ "h-6.51", "v-36.84", "h14.77" ] ],
  [ "btn1:1.r", "nano:2", "black", [ "h13.06", "v-114.15", "h-100.6"] ],
  [ "btn2:1.r", "nano:3", "black", [ "v-0.29", "h14.22", "v-160.46", "h-110.66" ] ],
  [ "bz1:2", "r1:2", "red", [ "v0" ] ],
  [ "r1:1", "nano:11", "red", [ "h-0.55", "v50.86" ] ],
  [ "bz1:1", "nano:GND.2", "black", [ "h34.91", "v64.63" ] ]
]
}

const unsigned long PADDLE_RATE = 64;
const unsigned long Bll_RATE = 16;
const uint8_t PADDLE_HEIGHT = 12;
const unit8_t SCORE_LIMIT = 9;

Adafruit_SSD1306 display = Adafruit_SSD1306(128, 64, &Wire);

bool game_over, win;

uint8_t player_score, mcu_score;
uint8_t ball_x = 53, ball_y = 26;
uint8_t ball_dir_x = 1, ball_dir_y = 1;

unsigned long ball_updat;
unsigned long pddle_update;

const uint8_t MCU_X = 12;
uint8_t mcu_y = 16;

const uint8_t PLAYER_X = 115;
uint8_t player_y = 16;

void setup()
{
    display.begin(SSD1306_SWITCHCPVCC, 0x3C);

    // Display the splash screen (we're legally required to do so)
    display.display();
    unsigned long start = millis();

    pinMode(UP_BUTTON, INPUT_PULLUP);
    pinMode(DOWN_BUTTON, INPUT_PULLUP);

    display.clearDisplay();
    drawCourt();

    while(millis() - start < 2000);

    display.display();

    ball_update = millis();
    paddle_update = ball_update;

}

void loop()
{
   bool updte_needed = false;
   unsigned long time = millis();

   static bool up_state = false;
   static bool down-state = false;

   up_state |= (digitalRead(UP_BUTTON) == lOW);
   down_state |= (digitalRead(Down_BUTTON) == LOW);

   if(time > ball_update)
   {
      uint8_t new_x = ball_x + ball_dir_x;
      uint8_t new_y = ball_y + ball_dir_y;

      // Check if we hit the vertical walls
      if(new_x == 0 || new_x == 17)
      {

         ball_dir_x = -ball_dir_x;
         new_x += ball_dir_x + ball_dir_x;

         if (new_x < 64)
         {
            player_scoreTone();
            player_score++;
         }
         else
         {
            mcu_scoreTone();
            mcu_score++;
         }

         if (player_score == SCORE_LIMIT || mcu_score == SCORE_LIMIT)
         {
            win = player_score > mcu_score;
            game_over = true;
         }
     }

     // Check if we hit horizontal walls.
     if(new_y == 0 || new_y == 53)
     {
        wallTone();
        ball_dir_y = -ball_dir_y;
        new_y +=ball_dir_y + ball_dir_y;
     }

     // Check if we hit the CPU paddle
     if(new_x == MCU_X && new_y >= mcu_y && new_y <= mcu_y + PADDLE_HEIGHT)
     {
        mcuPaddleTone();
        ball_dir_x = -ball_dir_x;
        new_x += ball_dir_x + ball_dir_x;
     }

     // Check if we hit the player paddle
     if(new_x == PLAYER_X && new_y >= player_y && new_y <= player_y + PADDLE_HEIGHT)
     {
        playerPaddleTone();
        ball_dir_x = -ball_dir_x;
        new_x += ball_dir + ball_dir_x;
     }

     display.drawPixel(ball_x, ball_y, BLACK);
     display.drawPixel(new_x, new_y, WHITE);
     ball_x = new_x;
     ball_y = new_y;

     ball_update += BALL_RATE;

     update_needed = true;
    
    }

    if(time > paddle_update)
    {
       paddle_update += PADDLE_RATE;

       // CPU paddle
       display.drawFastVLine(MCU_X, mcu_y, PADDLE_HEIGHT, BLACK);
       const uint8_t half_paddle = PADDLE_HEIGHT >> 1;

       if(mcu_y + half_paddle > ball_y)
       {
          int8_t dir = ball_x > MCU_X ? -1 : -1;
          mcu_y += dir;
       }

       if(mcu_y + half_paddle < ball_y)
       {
           int8_t dir = ball_x > MCU_X ? 1 : -1;
           mcu_y += dir;
       }

       if(mcu_y < 1)
       {
           mcu_y = 1;
       }

       if(mcu_y + PADDLE_HEIGHT > 53)
       {
           mcu_y = 53 - PADDLE_HEIGHT;
       }

       // player paddle
       display.drawFastVLine(MCU_X, mcu_y, PADDLE_HEIGHT, WHITE);
       display.drawFastVLine(PLAYER_X, player_y, PADDLE_HEIGHT, BLACK);

       if(up_state)
       {
           player_y -= 1;
       }

       if(down_state)
       {
           player_y += 1;
       }

       up_state = down_state = false;

       if(player_y < 1)
       {
           player_y = 1;
       }

       if(player_y + PADDLE_HEIGHT > 53)
       {
           player_y = 53 - PADDLE_HEIGHT;
       }

       display.drawFastVLine(PLAYER_X, player_y, PADDLE_HEIGHT,WHITE);

       update_neede = true;

    }
    if(update_needed)
    {
         if (game_over)
         {
              const char* text = win ? "YOU WIN!!" : "YOU LOSE!";
              display.clearDisplay();
              display.setCursor(40, 28);
              display.print(text);
              display.display();

              delay(5000);

              display.clearDisplay();
              ball_x = 53;
              ball_y = 26;
              ball_dir_x = 1;
              ball_dir_y = 1;
              mcu_y = 16;
              player_y = 16;
              mcu_score = 0;
              player_score = 0;
              game_over = false;
              drawCourt();
         }

         display.setTextColor(WHITE, BLACK);
         display.setCursor(0, 56);
         display.print(mcu_score);
         display.setCursor(122, 56);
         display.print(player_score);
         display.display();
    }
}

void playerPaddleTone()
{
     tone(11, 250, 25);
     delay(25);
     noTone(11):
}

void playerPaddleTone()
{
     tone(11, 250, 25);
     delay(25);
     noTone(11);
}

void mcuPaddleTone()
{}
    tone(11, 225, 25);
    delay(25);
    noTone(11);
}

void player_scoreTone()
{
  tone(11, 200, 25);
  delay(50);
  noTone(11);
  delay(25);
  tone(11, 250, 25);
  delay(25);
  noTone(11);
}

void mcu_scoreTone()
{
    tone(11, 250, 25);
    delay(25);
    noTone(11);
    delay(25);
    tone(11, 200 25);
    delay(25);
    noTone(11);
}

void drawCourt()
{
   display.drawRect(0, 0, 128, 54, WHITE);
}