package launcher;

import launcher.Fancy.MainWindow;
import launcher.Gameupdater.ClientUpdater;
import launcher.Utils.Defaults;
import launcher.Utils.Logger;

import javax.swing.*;
import java.awt.*;
import java.io.*;
import java.net.URL;
import java.net.URLConnection;

public class Launcher extends Component {
  public static ImageIcon icon = null;
  private JProgressBar m_progressBar;
  public static ClientUpdater updater;

  public void initializeLauncher() {
    Settings.loadSettings();

    // Add progress bar
    m_progressBar = new JProgressBar();

    if (Main.disabledUpdate) {
      Settings.autoUpdate = false;
      Settings.saveSettings();
    } else if (Settings.firstRun) {
      int response = JOptionPane.showConfirmDialog(
          this,
          "The OpenRSC Launcher has an automatic update feature.\n"
              + "\n"
              + "Would you like to enable this feature?\n",
          "OpenRSC",
          JOptionPane.YES_NO_OPTION,
          JOptionPane.INFORMATION_MESSAGE,
          icon);
      if (response == JOptionPane.YES_OPTION || response == JOptionPane.CLOSED_OPTION) {
        Settings.autoUpdate = true;
        JOptionPane.showMessageDialog(
            this,
            "The OpenRSC Launcher is set to check for updates at every launch!",
            "OpenRSC",
            JOptionPane.INFORMATION_MESSAGE,
            icon);
      } else if (response == JOptionPane.NO_OPTION) {
        Settings.autoUpdate = false;
        JOptionPane.showMessageDialog(
            this,
            "The OpenRSC launcher will not check for updates automatically.\n"
                + "\n"
                + "You will not get notified when new releases are available. To update the launcher, you\n"
                + "will need to do it manually by replacing 'OpenRSC.jar'.\n"
                + "\n"
                + "You can enable automatic updates again in the localSettings.conf file.",
            "OpenRSC",
            JOptionPane.INFORMATION_MESSAGE,
            icon);
      }
      Settings.saveSettings();
    }

    if (Settings.autoUpdate) {
      setStatus("Checking for launcher update...");
      double latestVersion = fetchLatestVersionNumber();
      if (Defaults._CURRENT_VERSION < latestVersion) {
        setStatus("Launcher update is available");
        int response = JOptionPane.showConfirmDialog(
            this,
            "A launcher update is available!\n"
                + "\n"
                + "Latest: "
                + String.format("%8.6f", latestVersion)
                + "\n"
                + "Installed: "
                + String.format("%8.6f", Defaults._CURRENT_VERSION)
                + "\n"
                + "\n"
                + "Would you like to update now?",
            "OpenRSC",
            JOptionPane.YES_NO_OPTION,
            JOptionPane.INFORMATION_MESSAGE,
            icon);
        if (response == JOptionPane.YES_OPTION) {
          if (updateJar()) {
            JOptionPane.showMessageDialog(
                this,
                "The launcher has been updated successfully!\n"
                    + "\n"
                    + "The launcher requires a restart, and will now exit.",
                "OpenRSC",
                JOptionPane.INFORMATION_MESSAGE,
                icon);
            System.exit(0);
          } else {
            response = JOptionPane.showConfirmDialog(
                this,
                "The launcher has failed to update, please try again later.\n"
                    + "\n"
                    + "Would you like to continue without updating?",
                "OpenRSC",
                JOptionPane.YES_NO_OPTION,
                JOptionPane.ERROR_MESSAGE,
                icon);
            if (response == JOptionPane.NO_OPTION || response == JOptionPane.CLOSED_OPTION) {
              System.exit(0);
            }
          }
        }
      }
    }

    // Initialize UI
    final MainWindow frame = new MainWindow();
    frame.build();

    // Fetch OpenRSC client jar and cache updates; also init progress bar
    updater = new ClientUpdater(Main.configFileLocation);
    updater.updateOpenRSCClient();
  }

  public static Double fetchLatestVersionNumber() {
    try {
      double currentVersion = 0.0;
      URL updateURL = new URL(Defaults._VERSION_UPDATE_URL);

      // Open connection
      URLConnection connection = updateURL.openConnection();
      connection.setConnectTimeout(3000);
      connection.setReadTimeout(3000);
      BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
      String line;
      while ((line = in.readLine()) != null) {
        if (line.contains("_CURRENT_VERSION")) {
          currentVersion = Double.parseDouble(line.substring(line.indexOf('=') + 1, line.indexOf(';')));
          Logger.Info("Current Version: " + currentVersion);
          break;
        }
      }

      // Close connection
      in.close();
      return currentVersion;
    } catch (Exception e) {
      Logger.Error("Error checking latest version");
      return Defaults._CURRENT_VERSION;
    }
  }

  public boolean updateJar() {
    setStatus("Starting launcher update...");
    setProgress(0, 1);

    try {
      URL url = new URL(Defaults._GAME_FILES_SERVER + Defaults._LAUNCHER_FILENAME);

      // Open connection
      URLConnection connection = url.openConnection();
      connection.setConnectTimeout(3000);
      connection.setReadTimeout(3000);

      int size = connection.getContentLength();
      int offset = 0;
      byte[] data = new byte[size];

      InputStream input = url.openStream();

      int readSize;
      while ((readSize = input.read(data, offset, size - offset)) != -1) {
        offset += readSize;
        setStatus("Updating launcher (" + (offset / 1024) + "KiB / " + (size / 1024) + "KiB)");
        setProgress(offset, size);
      }

      if (offset == size) {
        File file = new File("./" + Defaults._LAUNCHER_FILENAME);
        FileOutputStream output = new FileOutputStream(file);
        output.write(data);
        output.close();
      }
    } catch (Exception ignored) {
    }
    return true;
  }

  /**
   * Sets the progress value of the launcher progress bar.
   *
   * @param value the number of tasks that have been completed
   * @param total the total number of tasks to complete
   */
  public void setProgress(final int value, final int total) {
    SwingUtilities.invokeLater(
        new Runnable() {
          @Override
          public void run() {
            if (total == 0) {
              m_progressBar.setValue(0);
              return;
            }

            m_progressBar.setValue(value * 100 / total);
          }
        });
  }

  /**
   * Changes the launcher progress bar text and pauses the thread for 5 seconds.
   *
   * @param text the text to change the progress bar text to
   */
  public void error(String text) {
    setStatus("Error: " + text);
    try {
      Thread.sleep(5000);
      System.exit(0);
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  public void setStatus(final String text) {
    SwingUtilities.invokeLater(
        new Runnable() {
          @Override
          public void run() {
            m_progressBar.setString(text);
          }
        });
  }
}