public class Main extends WebsitePlugin{
	
	private SQLSaver saver;
	
	private static Main INSTANCE;
	
	@Override
	public void onEnable() {
		INSTANCE = this;
		// Save the default config, anchor from child
		saveDefaultConfig(this.getClass());
		String sqlHost = getConfig().getString("sqlHost");
		String sqlUser = getConfig().getString("sqlUser");
		String sqlPassword = getConfig().getString("sqlPassword");
		String sqlDatabase = getConfig().getString("sqlDatabase");
		int sqlPort = getConfig().getInt("sqlPort");
		
		saver = new SQLSaver(sqlHost, sqlDatabase, sqlUser, sqlPassword, sqlPort);
		
		this.getAPILoader().registerAPIClass(user.class);
		
	}
	
	public SQLSaver getSQLSaver() {
		return saver;
	}
	
	public static Main getInstance() {
		return INSTANCE;
	}

}

public class user {

	public static String register(String[] args) {
		try {
			Map<String, Object> params = paramify(args);
			String username = String.valueOf(params.get("username"));
			String password = String.valueOf(params.get("password"));
			String email = String.valueOf(params.get("email"));
			PasswordAuthentication auth = new PasswordAuthentication();
			long seconds = TimeUnit.MILLISECONDS.toSeconds(System.currentTimeMillis());
			password = auth.hash(password);

			if (username.matches("/[\\'^£$%&*()}{@#~?><>,|=_+¬-]/")) {
				return JSONObjects.getStatusFailure("Username can only contain a-z 0-9");
			}

			if (username.contains(" ")) {
				return JSONObjects.getStatusFailure("Username cannot contain spaces");
			}

			ResultSet result = Main.getInstance().getSQLSaver().getConnection().createStatement()
					.executeQuery("SELECT email FROM users WHERE email = '" + params.get("email") + "'");
			if (result.first()) {
				return JSONObjects.getStatusFailure("An account with that E-Mail already exists");
			}

			result = Main.getInstance().getSQLSaver().getConnection().createStatement()
					.executeQuery("SELECT username FROM users WHERE username = '" + params.get("username") + "'");
			if (result.first()) {
				return JSONObjects.getStatusFailure("An account with that username already exists");
			}
			UUID uid = UUID.randomUUID();
			Integer rst = Main.getInstance().getSQLSaver().getConnection().createStatement()
					.executeUpdate("INSERT INTO users (username, token, password, avatar, email, timestamp) VALUES ('"
							+ username + "', '" + uid.toString() + "', '" + password + "', 'NULL', '" + email + "', '"
							+ seconds + "')");
			if (rst == 1) {
				JSONObject data = new JSONObject();
				data.put("tfa", false);
				data.put("token", uid.toString());
				data.put("username", username);
				return JSONObjects.getStatusOk(data.toJSONString());
			} else {
				return JSONObjects.getStatusFailure("SQL returned satus 0");
			}

		} catch (NumberFormatException | SQLException e) {
			System.out.println("Error in get: " + e.toString());
			return JSONObjects.getStatusError("Error in get: " + e.toString());
		}
	}

	public static String login(String[] args) {
		try {
			Map<String, Object> params = paramify(args);
			String username = String.valueOf(params.get("username"));
			String password = String.valueOf(params.get("password"));
			PasswordAuthentication auth = new PasswordAuthentication();

			ResultSet result = Main.getInstance().getSQLSaver().getConnection().createStatement().executeQuery(
					"SELECT id,username,password,email FROM users WHERE username = '" + params.get("username") + "';");
			if (!result.first()) {
				System.out.println("Null");
			}
			String password2 = result.getString("password");

			if (auth.authenticate(password, password2)) {
				// Logged in
				UUID uid = UUID.randomUUID();

				Integer rst = Main.getInstance().getSQLSaver().getConnection().createStatement().executeUpdate(
						"UPDATE users SET token = '" + uid.toString() + "' WHERE username = '" + username + "';");

				if (rst == 1) {
					JSONObject data = new JSONObject();
					data.put("tfa", false);
					data.put("token", uid.toString());
					data.put("username", username);
					System.out.println("JSON:" + JSONObjects.getStatusOk(data));
					return JSONObjects.getStatusOk(data);
				} else {
					return JSONObjects.getStatusFailure("SQL returned satus 0");
				}

			} else {
				// Incorrect password
				return JSONObjects.getStatusFailure("Incorrect Password");
			}
		} catch (NumberFormatException | SQLException e) {
			System.out.println("Error in login: " + e.toString());
			return JSONObjects.getStatusError("Error in login: " + e.toString());
		}
	}

	public static String verifylogin(String[] args) {
		try {
			Map<String, Object> params = paramify(args);
			String token = String.valueOf(params.get("_t"));
			System.out.println("Token: " + token);

			ResultSet result = Main.getInstance().getSQLSaver().getConnection().createStatement().executeQuery(
					"SELECT username,token,avatar,timestamp,id FROM users WHERE token = '" + token + "';");
			if (result.first()) {
				System.out.println("Logged in:");
				JSONObject obj = new JSONObject();
				obj.put("tfa", false);
				obj.put("token", token);
				obj.put("username", result.getString("username"));
				obj.put("id", String.valueOf(result.getInt("id")));
				// TODO: Return banned, avatar, isAdmin, etc.
				return JSONObjects.getStatusOk(obj);
			} else {
				System.out.println("No token found");
				return JSONObjects.getStatusFailure();

			}

		} catch (NumberFormatException | SQLException e) {
			System.out.println("Error in login: " + e.toString());
			e.printStackTrace();
			return JSONObjects.getStatusError("Error in login: " + e.toString());
		}
	}

	public static String logout(String[] args) {
		try {
			Map<String, Object> params = paramify(args);
			Integer userid = (Integer) params.get("userid");

			Integer rst = Main.getInstance().getSQLSaver().getConnection().createStatement()
					.executeUpdate("UPDATE users SET token = NULL WHERE id = '" + userid + "';");
			if (rst == 1) {
				return JSONObjects.getStatusOk();
			} else {
				return JSONObjects.getStatusFailure("SQL returned status 0.");
			}
		} catch (NumberFormatException | SQLException e) {
			System.out.println("Error in logout: " + e.toString());
			e.printStackTrace();
			return JSONObjects.getStatusError("Error in login: " + e.toString());
		}
	}


	// Converts params sent in as String E.g (users=10) to the proper Data Dype
	// variable
	// in the format of Map<String, Object>
	public static Map<String, Object> paramify(String[] args) throws NumberFormatException {
		Map<String, Object> localParams = new HashMap<>();
		for (String arg : args) {
			String param = StringUtils.substringBefore(arg, "=");
			if (param.equalsIgnoreCase("userid")) {
				System.out.println("UserID");
				Integer value = Integer.parseInt(StringUtils.substringAfter(arg, "="));
				localParams.put(param, value);
			} else if (param.equalsIgnoreCase("scriptid")) {
				Integer value = Integer.parseInt(StringUtils.substringAfter(arg, "="));
				localParams.put(param, value);
			} else {
				String value = StringUtils.substringAfter(arg, "=");
				value = value.replace("%20", " ");
				localParams.put(param, value);
			}
		}
		return localParams;
	}

}