import org.osbot.rs07.api.Client;
import org.osbot.rs07.listener.LoginResponseCodeListener;
import org.osbot.rs07.script.Script;
import org.osbot.rs07.script.ScriptManifest;
import org.osbot.rs07.utility.ConditionalSleep;
import util.LoginEvent;

import java.sql.*;

@ScriptManifest(author = "Sibbernski", name = "sibbernskiAccountLocker", info = "hekkoi", version = 0.1, logo = "")

public class sibbernskiLocker extends Script {

    private boolean bannedOrLocked = false;

    private String[] params;
    private String username;
    private String password;
    private String proxyId;
    private LoginEvent loginEvent;


    @Override
    public void onStart() {
        params = getParameters().split("/");
        username = params[0];
        password = params[1];
        proxyId = params[2];
        loginEvent = new LoginEvent(username, password);
    }

    @Override
    public int onLoop() throws InterruptedException {
        if (!loginEvent.hasFinished() || !loginEvent.hasFailed()) {
            loginCharacter();
        }

        return 1000;
    }

    @Override
    public void onResponseCode(int code) {
        log("Response code: " + code);
        if (code != 2 && code != 9 && code != 5 && code != 21) {
            log("login failed with code " + code);
            stop();
        }
        switch (code) {
            case 4:
                updateLockStatus();
                this.stop(true);
                System.exit(0);
                break;
            case 18:
                updateBanStatus();
                this.stop(true);
                System.exit(0);
                break;
        }
    }

    private void updateLockStatus() {
        try {
            String host = "localhost:3306";
            String db = "muling";
            String user = "root";
            String pass = "";
            String connStr = String.format("jdbc:mysql://%s/%s?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=UTC", host, db);
            Connection conn = DriverManager.getConnection(connStr, user, pass);
            ResultSet result = conn.createStatement().executeQuery("SELECT * FROM newaccs WHERE AccountName='" + username + "'");
            if (result.next()) {
                String query = "UPDATE newaccs set AccountLocked='yes', AssignedProxy='" + proxyId + "' WHERE AccountName='" + username + "'";
                PreparedStatement preparedStmt = conn.prepareStatement(query);
                preparedStmt.executeUpdate();
                log("Account was locked, updated db.");
                bannedOrLocked = true;
            }
        } catch (SQLException e) {
            log(e);
        }
    }

    private void updateBanStatus() {
        try {
            String host = "localhost:3306";
            String db = "muling";
            String user = "root";
            String pass = "";
            String connStr = String.format("jdbc:mysql://%s/%s?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=UTC", host, db);
            Connection conn = DriverManager.getConnection(connStr, user, pass);
            ResultSet result = conn.createStatement().executeQuery("SELECT * FROM newaccs WHERE AccountName='" + username + "'");
            if (result.next()) {
                String query = "UPDATE newaccs set AccountBanned='yes', AssignedProxy='" + proxyId + "' WHERE AccountName='" + username + "'";
                PreparedStatement preparedStmt = conn.prepareStatement(query);
                preparedStmt.executeUpdate();
                log("Account was banned, updated db.");
                bannedOrLocked = true;
            }
        } catch (SQLException e) {
            log(e);
        }
    }

    private void loginCharacter() throws InterruptedException {
        log("Login...");
        sleep(2000);
        if (getClient().getLoginState() != null && getClient().getLoginState() != Client.LoginState.LOADING_MAP && !getClient().isLoggedIn()) {
            if (loginEvent != null && loginEvent.isWorking()) {
                return;
            }
            log("Logging: " + username);
            loginEvent = new LoginEvent(username, password);
            execute(loginEvent);
        }
    }

}