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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

#define PADDLE_WIDTH 4
#define PADDLE_HEIGHT 20
#define BALL_SIZE 4

#define BUTTON_LEFT 12
#define BUTTON_RIGHT 13
#define LED_PIN_1 11
#define LED_PIN_2 10

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

int paddle1Y;
int paddle2Y;
int ballX, ballY;
int ballSpeedX = 4; // Vitesse horizontale de la balle (augmentée)
int ballSpeedY = 4; // Vitesse verticale de la balle (augmentée)
int scorePlayer1 = 0;
int scorePlayer2 = 0;
bool moveUp = false;
bool moveDown = false;
bool gameOver = false; // Variable pour détecter la fin de partie

unsigned long gameOverTime = 0; // Variable pour stocker le moment où la partie s'est terminée

void setup() {
  Wire.begin();
  pinMode(BUTTON_LEFT, INPUT_PULLUP);
  pinMode(BUTTON_RIGHT, INPUT_PULLUP);
  pinMode(LED_PIN_1, OUTPUT);
  pinMode(LED_PIN_2, OUTPUT);

  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  display.clearDisplay();
  paddle1Y = SCREEN_HEIGHT / 2 - PADDLE_HEIGHT / 2;
  paddle2Y = SCREEN_HEIGHT / 2 - PADDLE_HEIGHT / 2;
  ballX = SCREEN_WIDTH / 2 - BALL_SIZE / 2;
  ballY = SCREEN_HEIGHT / 2 - BALL_SIZE / 2;
}

void loop() {
  // Déplacement de la raquette
  if (digitalRead(BUTTON_LEFT) == LOW) {
    moveUp = true;
  } else {
    moveUp = false;
  }
  if (digitalRead(BUTTON_RIGHT) == LOW) {
    moveDown = true;
  } else {
    moveDown = false;
  }
  
  if (moveUp && paddle1Y > 0) {
    paddle1Y -= 2;
  }
  if (moveDown && paddle1Y < SCREEN_HEIGHT - PADDLE_HEIGHT) {
    paddle1Y += 2;
  }

  // Si la partie est terminée et le délai de 4 secondes est écoulé, réinitialiser la partie
  if (gameOver && millis() - gameOverTime >= 4000) {
    resetGame();
  }

  // Si la partie n'est pas terminée, mettre à jour le jeu normalement
  if (!gameOver) {
    // Mouvement de la balle
    ballX += ballSpeedX;
    ballY += ballSpeedY;

    // Rebondissement de la balle sur les bords de l'écran
    if (ballY <= 0 || ballY >= SCREEN_HEIGHT - BALL_SIZE) {
      ballSpeedY = -ballSpeedY;
    }

    // Rebondissement de la balle sur les raquettes
    if ((ballX <= PADDLE_WIDTH && ballY + BALL_SIZE >= paddle1Y && ballY <= paddle1Y + PADDLE_HEIGHT) ||
        (ballX >= SCREEN_WIDTH - PADDLE_WIDTH - BALL_SIZE && ballY + BALL_SIZE >= paddle2Y && ballY <= paddle2Y + PADDLE_HEIGHT)) {
      ballSpeedX = -ballSpeedX;
      ballSpeedX += (ballSpeedX > 0) ? 1 : -1; // Augmente la vitesse à chaque rebond
    }

    // Marquage des points si la balle atteint les bords de l'écran
    if (ballX <= 0) {
      scorePlayer2++;
      digitalWrite(LED_PIN_2, HIGH); // Allume la LED 2
      resetBall();
    } else if (ballX >= SCREEN_WIDTH - BALL_SIZE) {
      scorePlayer1++;
      digitalWrite(LED_PIN_1, HIGH); // Allume la LED 1
      resetBall();
    }

    // Vérification si un joueur a gagné
    if (scorePlayer1 >= 2 || scorePlayer2 >= 2) {
      endGame();
    }

    // Affichage des éléments du jeu sur l'écran
    display.clearDisplay();
    display.fillRect(0, paddle1Y, PADDLE_WIDTH, PADDLE_HEIGHT, WHITE);
    display.fillRect(SCREEN_WIDTH - PADDLE_WIDTH, paddle2Y, PADDLE_WIDTH, PADDLE_HEIGHT, WHITE);
    display.fillCircle(ballX, ballY, BALL_SIZE, WHITE);

    // Affichage des scores pour chaque joueur
    display.setTextSize(1);
    display.setTextColor(WHITE);
    display.setCursor(SCREEN_WIDTH / 2 - 30, 0);
    display.print(scorePlayer1);
    display.print(" - ");
    display.print(scorePlayer2);

    display.display();
  } else {
    // Affichage du message de fin de partie
    display.clearDisplay();
    display.setTextSize(2);
    display.setTextColor(WHITE);
    display.setCursor(0, 0);
    display.println("Bien joue,");
    display.setCursor(0, 20);
    display.println("tu as gagne!");
    display.display();
  }

  delay(20); // Pause pour rafraîchir l'affichage
}

void resetBall() {
  ballX = SCREEN_WIDTH / 2 - BALL_SIZE / 2;
  ballY = SCREEN_HEIGHT / 2 - BALL_SIZE / 2;
  ballSpeedX = 4; // Réinitialisation de la vitesse de la balle
  ballSpeedY = 4; // Réinitialisation de la vitesse de la balle
}

void resetGame() {
  scorePlayer1 = 0;
  scorePlayer2 = 0;
  gameOver = false;
}

void endGame() {
  gameOver = true;
  gameOverTime = millis(); // Enregistrer le moment où la partie s'est terminée
}