package com.jsql.util.bruter;
   
   import com.jsql.util.I18nUtil;
   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 {
      
      private static final Logger LOGGER = LogManager.getRootLogger();
      public static final String PATTERN_PERIOD = "%s: %s %s %s %s %s %s %s %s";
  
      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 = seconds / 60;
          seconds = seconds % 60;
          var hours = minutes / 60;
          minutes = minutes % 60;
          var days = hours / 24;
          hours = hours % 24;
 
         return String.format(
             Bruter.PATTERN_PERIOD,
             "Time elapsed",
             days, I18nUtil.valueByKey("BRUTEFORCE_DAYS"),
             hours, I18nUtil.valueByKey("BRUTEFORCE_HOURS"),
             minutes, I18nUtil.valueByKey("BRUTEFORCE_MINUTES"),
             seconds, I18nUtil.valueByKey("BRUTEFORCE_SECONDS")
         );
     }
 
     private long calculateTimeDifference() {
         return (long) ((this.endtime - this.starttime) * 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;
     }
 }