#include <Servo.h>
#include <LiquidCrystal_I2C.h>

#define ECHO_PIN 2
#define TRIG_PIN 3

Servo myservo;  // create servo object to control a servo
#define I2C_ADDR    0x27
#define LCD_COLUMNS 16
#define LCD_LINES   2
LiquidCrystal_I2C lcd(I2C_ADDR, LCD_COLUMNS, LCD_LINES);

int pos = 0;    // variable to store the servo position

float readDistanceCM() {
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);
  int duration = pulseIn(ECHO_PIN, HIGH);
  return duration * 0.034 / 2;
}

void setup() {
  Serial.begin(115200);
  pinMode(LED_BUILTIN, OUTPUT);
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);

  myservo.attach(9);  // attaches the servo on pin 9 to the servo object
  // Init LCD
  lcd.init();
  lcd.backlight();

  // Print something
  lcd.setCursor(0, 0);
  lcd.print("Distance: ");
}

void loop() {
  for (int i = 0; i < 4; ++i) {
    lcd.clear();
    lcd.setCursor(0, 1);  // Set cursor to the second row
    lcd.print("Loading");

    for (int j = 0; j < i; ++j) {
      lcd.print('.');
    }

    delay(500);
  }

  float distance = readDistanceCM();

  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Distance: ");
  lcd.setCursor(10, 0);
  lcd.print(distance);

  bool isNearby = distance < 100;
  digitalWrite(LED_BUILTIN, isNearby);

  Serial.print("Measured distance: ");
  Serial.println(distance);

  if (distance < 50) {
    myservo.write(180);  // Set servo position to 180
    lcd.setCursor(0, 1);
    lcd.print("Servo: 180       ");
  } else if (distance >= 50 && distance < 100) {
    myservo.write(90);   // Set servo position to 90
    lcd.setCursor(0, 1);
    lcd.print("Servo: 90        ");
  } else {
    myservo.write(0);    // Set servo position to 0
    lcd.setCursor(0, 1);
    lcd.print("Servo: 0         ");
  }

  delay(2000);
}