package com.jsql.util.bruter;
   
   import com.jsql.util.LogLevelUtil;
   import org.apache.logging.log4j.LogManager;
   import org.apache.logging.log4j.Logger;
   
   import java.util.ArrayList;
   import java.util.List;
   
  public class Bruter {
      
      /**
       * Log4j logger sent to view.
       */
      private static final Logger LOGGER = LogManager.getRootLogger();
  
      protected final List<String> characters = new ArrayList<>();
       
      protected boolean found = false;
       
      protected int maxLength;
      protected int minLength;
       
      protected int count;
       
      protected long starttime;
      protected long endtime;
       
      private static final char[] specialCharacters = {
          '~', '`', '!', '@', '#', '$', '%', '^',
          '&', '*', '(', ')', '_', '-', '+', '=', '{', '}', '[', ']', '|', '\\',
          ';', ':', '\'', '"', '<', '.', ',', '>', '/', '?', ' '
      };
       
      protected boolean done = false;
  
      public long getRemainder() {
          return this.getNumberOfPossibilities() - this.count;
      }
  
      public long getNumberOfPossibilities() {
          long possibilities = 0;
          for (int i = this.minLength; i <= this.maxLength; i++) {
              possibilities += (long) Math.pow(this.characters.size(), i);
          }
          return possibilities;
      }
  
      public void addLowerCaseLetters() {
          for (var c = 'a'; c <= 'z'; c++) {
              this.characters.add(String.valueOf(c));
          }
      }
  
      public void addDigits() {
          for (var c = 0; c <= 9; c++) {
              this.characters.add(String.valueOf(c));
          }
      }
  
      public void addUpperCaseLetters() {
          for (var c = 'A'; c <= 'Z'; c++) {
              this.characters.add(String.valueOf(c));
          }
      }
  
      public void addSpecialCharacters() {
          for (char c: Bruter.specialCharacters) {
              this.characters.add(String.valueOf(c));
          }
      }
  
      public void excludeChars(String s) {
          char[] arrayChars = s.toCharArray();
          for (char arrayChar: arrayChars) {
              this.characters.remove(Character.toString(arrayChar));
          }
      }
  
      public int getPerSecond() {
          int i;
          try {
              i = (int) (this.count / this.calculateTimeDifference());
          } catch (Exception e) {
              LOGGER.log(LogLevelUtil.IGNORE, e);
              return 0;  // Ignore division by zero
          }
          return i;
      }
  
      public String calculateTimeElapsed() {
          long timeTaken = this.calculateTimeDifference();
          
          int seconds = (int) timeTaken;
          
          var minutes = 0;
          var hours = 0;
          var days = 0;
          
         minutes = seconds / 60;
         seconds = seconds % 60;
         hours = minutes / 60;
         minutes = minutes % 60;
         days = hours / 24;
         hours = hours % 24;
         
         return String.format(
             "Time elapsed: %sdays %sh %smin %ss",
             days,
             hours,
             minutes,
             seconds
         );
     }
 
     private long calculateTimeDifference() {
         return (long) ((this.endtime - this.starttime) * (1 * Math.pow(10, -9)));
     }
     
     
     // Getter and setter
 
     public synchronized void setEndtime(long endtime) {
         this.endtime = endtime;
     }
 
     public void setMaxLength(int maxLength) {
         this.maxLength = maxLength;
     }
 
     public void setMinLength(int minLength) {
         this.minLength = minLength;
     }
 
     public boolean isFound() {
         return this.found;
     }
 
     public void setFound(boolean found) {
         this.found = found;
     }
 
     public int getCounter() {
         return this.count;
     }
 
     public void setIsDone(Boolean done) {
         this.done = done;
     }
 
     public boolean isDone() {
         return this.done;
     }
 }