#include <Stepper.h>

/////////////SET THE BELOW PINS FOR YOUR HARDWARE SETUP//////////////
const int increaseButton = 5;       //PIN # BUTTON CONNECTED TO
const int decreaseButton = 6;       //PIN # BUTTON CONNECTED TO
const int enableButton = 7;         //PIN # BUTTON CONNECTED TO
const int buttonA = A4;             //PIN # BUTTON CONNECTED TO
const int buttonB = A3;             //PIN # BUTTON CONNECTED TO
const int sonoff = A2;              //PIN # BUTTON CONNECTED TO
const int redLed = 2;               //PIN # LED CONNECTED TO
const int greenLed = 3;             //PIN # LED CONNECTED TO
const int disablemotor = HIGH;      //SET TO AUTOMATICALLY DISABLE MOTOR TO SAVE POWER

const int stepsPerRevolution = 2048;                            // change this to fit the number of steps per revolution
Stepper myStepper = Stepper(stepsPerRevolution, 9, 11, 10, 12); // Pins entered in sequence IN1-IN3-IN2-IN4 for proper step sequence

//////////////////NO NEED TO CHANGE ANYTHING BELOW///////////////////

long posA = 0;                     //position A in steps
long posB = 0;                     //position B in steps
long currentPos = 0;               //current position
long rotationAmount = 1024;         // steps to jump during a move
int buttonAState = HIGH;
int buttonBState = HIGH;
int enabledState = HIGH;          // A variable for saving the enabled state
int triggered = LOW;              // a trigger for a mode
int alreadymovedtopositiona = LOW;
int alreadymovedtopositionb = LOW;
int alreadyDisabledTheMotor = LOW;

void setup() {
  myStepper.setSpeed(14);  
  Serial.begin(9600);
  
  pinMode(increaseButton, INPUT_PULLUP);
  pinMode(decreaseButton, INPUT_PULLUP);
  pinMode(enableButton, INPUT_PULLUP);
  pinMode(redLed, OUTPUT);
  pinMode(greenLed, OUTPUT);
  pinMode(buttonA , INPUT_PULLUP);
  pinMode(buttonB , INPUT_PULLUP);
  pinMode(sonoff , INPUT_PULLUP);
}

void loop() {

//READ STATUS OF THE BUTTONS
  int increasingState = digitalRead(increaseButton);
  int decreasingState = digitalRead(decreaseButton);
  int triggered = digitalRead(enableButton);
  int buttonAState = digitalRead(buttonA);
  int buttonBState = digitalRead(buttonB);
  int sonoffState = digitalRead(sonoff);

  
//TRIGGER ENABLED STATE PERMANENTLY 
  if(enabledState == HIGH && triggered == LOW) {
    enabledState = LOW;
    Serial.println("ENABLED");
    delay (1000);     
  }

  if (increasingState == LOW && decreasingState == LOW){
    enabledState = HIGH;
    Serial.println("DISABLED");
  }

//CHECK IF ENABLED FOR LIGHTS
  if(enabledState == LOW){
    digitalWrite(greenLed, HIGH);
    digitalWrite(redLed, LOW);
  }

  if (enabledState == HIGH){
    digitalWrite(greenLed, LOW);
    digitalWrite(redLed, HIGH);
  }

//CHECK FOR ENABLED STATE BEFORE DOING ANYTHING
  if (enabledState == LOW){

  //INCREASING
    if(increasingState == LOW){
      delay(10);
      myStepper.step(rotationAmount); 
      currentPos = currentPos + rotationAmount;
      increasingState = digitalRead(increaseButton); 
      Serial.println(currentPos);
    }

  //DECREASING
    if(decreasingState == LOW){
     delay(10);
     myStepper.step(-rotationAmount); 
     currentPos = currentPos - rotationAmount;
     decreasingState = digitalRead(decreaseButton); 
     Serial.println(currentPos);
    }
  
  //SAVE POSITION A
    if(triggered == LOW && buttonAState == LOW) {
      posA = currentPos;
      Serial.println("Position A Saved");
      Serial.println(posA);
      delay(1000);
    }
  
  //MOVE TO SAVED POSITION A

    if (triggered == HIGH && sonoffState == HIGH){
      if(sonoffState == HIGH && currentPos - posA > 0){
        myStepper.step(-rotationAmount);
        currentPos = currentPos - rotationAmount;
        Serial.println("Decreasing to position A");
        Serial.println(currentPos);
        if (currentPos - posA == 0){
        alreadyDisabledTheMotor = LOW;
        }
      }

    

      if(sonoffState == HIGH && currentPos - posA < 0){
        myStepper.step(rotationAmount);
        currentPos = currentPos + rotationAmount;
        Serial.println("Increasing to position A");
        Serial.println(currentPos);
        if (currentPos - posA == 0){
          alreadyDisabledTheMotor = LOW;
        }
      }
    }  


    //SAVE POSITION B
    if(triggered == LOW && buttonBState == LOW) {
      posB = currentPos;
      Serial.println("Position B Saved");
      Serial.println(posB);
      delay(1000);
    }
  
  //MOVE TO SAVED POSITION B

    if(buttonBState == LOW && sonoffState == LOW){
      if (currentPos - posB > 0){
        myStepper.step(-rotationAmount);
        currentPos = currentPos - rotationAmount;
        Serial.println("Decreasing to position B");
        Serial.println(currentPos);
        if (currentPos - posB == 0){
          alreadyDisabledTheMotor = LOW;
        }
      }
    }

    if(buttonBState == LOW || sonoffState == LOW){
      if (currentPos - posB < 0){
        myStepper.step(rotationAmount);
        currentPos = currentPos + rotationAmount;
        Serial.println("Increasing to position B");
        Serial.println(currentPos);
        if (currentPos - posB == 0){
          alreadyDisabledTheMotor = LOW;
        }
      }
    }    
  //CODE TO DISABLE THE MOTOR FOR POWER SAVING
    if (disablemotor == HIGH && alreadyDisabledTheMotor == LOW){
      digitalWrite(9,LOW);
      digitalWrite(10,LOW);
      digitalWrite(11,LOW);
      digitalWrite(12,LOW); 
      Serial.println("POWER SAVING - MOTOR DE-ENERGIZED") ;
      alreadyDisabledTheMotor = HIGH;
    }

  } //enable state
} //void loop