package net.kuraczyk.banhammer.cmds;

import net.kuraczyk.banhammer.BanHammer;
import net.kuraczyk.banhammer.utils.IPBan;
import net.kuraczyk.banhammer.utils.PlayerBan;
import net.kuraczyk.banhammer.utils.TextUtils;
import net.kyori.adventure.text.Component;
import org.bukkit.Bukkit;
import org.bukkit.command.Command;
import org.bukkit.command.CommandExecutor;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;

import java.net.InetAddress;
import java.net.UnknownHostException;
import java.sql.*;
import java.util.Objects;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.Gson;

public class BanIP implements CommandExecutor {

    BanHammer plugin;

    public BanIP(BanHammer p){
        plugin = p;
        p.getCommand("banip").setExecutor(this);
    }

    @Override
    public boolean onCommand(CommandSender sender, Command cmd, String lab, String[] args) {
        if(sender instanceof Player) return banIpCommand((Player)sender, args);
        else return banIpCommand(args);
    }

    private boolean banIpCommand(Player p, String[] args) {
        if(!p.hasPermission("banhammer.banip")) {
            p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.noPermission"))));
            return false;
        }
        if(args.length==0) p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.usage"))));
        else if(args.length>=1) {
            Timestamp banEndDate=null;
            String scope = "*", reason = plugin.translations.getString("default.reason"), duration="-1";

            if(args.length>=2) scope=args[1];
            if(args.length>=3) {
                duration=args[2];
                String regex = "^(perm)|(-1)|\\d+:(s|m|h|d)$";
                String regexOnlyNum = "^\\d+$";
                if(!(duration.matches(regex) || duration.matches(regexOnlyNum))) {
                    p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.durationFormatError"))));
                    return false;
                }
                char type = plugin.config.getString("defaultDurationType").charAt(0);
                Integer time=0;
                boolean chk = false;
                for(int i = 0; i < duration.length(); i++){
                    if(duration.charAt(i)==':'){
                        chk = true;
                        time = Integer.parseInt(duration.substring(0, i));
                        if(i+1 < duration.length()) type = duration.charAt(i+1);
                    }
                }
                if(!chk) time = Integer.parseInt(duration);
                switch (type){
                    case 'd':
                        banEndDate = new Timestamp(System.currentTimeMillis()+ TimeUnit.DAYS.toMillis(time));
                        break;
                    case 'h':
                        banEndDate = new Timestamp(System.currentTimeMillis()+TimeUnit.HOURS.toMillis(time));
                        break;
                    case 'm':
                        banEndDate = new Timestamp(System.currentTimeMillis()+TimeUnit.MINUTES.toMillis(time));
                        break;
                    case 's':
                        banEndDate = new Timestamp(System.currentTimeMillis()+TimeUnit.SECONDS.toMillis(time));
                        break;
                    default:
                        p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.durationFormatError"))));
                        return false;
                }
            }

            StringBuilder newReason = new StringBuilder();
            for(int i = 3; i < args.length; i++){
                newReason.append(args[i]).append(" ");
            }
            if(!newReason.isEmpty()) reason=newReason.toString();

            if(Objects.equals(duration, "-1") || Objects.equals(duration, "perm")) banEndDate = null;

            InetAddress addr;
            try {
                addr = InetAddress.getByName(args[0]);
            } catch (UnknownHostException e){
                plugin.getLogger().severe(e.toString());
                p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.unexpectedError"))));
                return false;
            }

            try {
                Connection conn = plugin.db.getConnection();
                PreparedStatement stmt = conn.prepareStatement("SELECT * FROM ip_bans WHERE (ip_address = ?::inet) AND scope = ?");
                stmt.setObject(1, addr.getHostAddress());
                stmt.setString(2, scope);
                ResultSet rs = stmt.executeQuery();
                if(rs.next()){
                    p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.alreadyBannedInScopeError"))));
                    return false;
                }
            } catch (SQLException e){
                plugin.getLogger().severe(e.toString());
                p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.unexpectedError"))));
                return false;
            }

            IPBan b = banIp(addr, reason, scope, banEndDate, p.getName(), p.getUniqueId());

            if(b != null) {
                if(banEndDate == null) {
                    createIPKickJob(b, scope);
                    p.sendMessage(TextUtils.colorString(resolveIpBanMessagePlaceholders(addPrefix(plugin.translations.getString("command.banip.successfulBanMessagePerm")), args[0], scope, reason, p.getName())));
                    String bcMess = plugin.translations.getString("command.banip.banPublicAnnouncementPerm");
                    if(bcMess!=null) Bukkit.broadcast(TextUtils.colorString(resolveIpBanMessagePlaceholders(bcMess, args[0], scope, reason, p.getName())));
                }
                else {
                    long timeLeft = banEndDate.getTime() - new Timestamp(System.currentTimeMillis()).getTime();
                    long timeLeftDays = TimeUnit.MILLISECONDS.toDays(timeLeft);
                    timeLeft-=TimeUnit.DAYS.toMillis(timeLeftDays);
                    long timeLeftHours = TimeUnit.MILLISECONDS.toHours(timeLeft);
                    timeLeft-=TimeUnit.HOURS.toMillis(timeLeftHours);
                    long timeLeftMinutes = TimeUnit.MILLISECONDS.toMinutes(timeLeft);
                    timeLeft-=TimeUnit.MINUTES.toMillis(timeLeftMinutes);
                    long timeLeftSeconds = TimeUnit.MILLISECONDS.toSeconds(timeLeft);
                    String parsedBanDuration = TextUtils.parsePolishDate(timeLeftDays, timeLeftHours, timeLeftMinutes, timeLeftSeconds);

                    createIPKickJob(b, scope);
                    p.sendMessage(TextUtils.colorString(resolveIpBanMessagePlaceholders(addPrefix(plugin.translations.getString("command.banip.successfulBanMessageTemp")), args[0], scope, reason, p.getName(), parsedBanDuration)));
                    String bcMess = plugin.translations.getString("command.banip.banPublicAnnouncementTemp");
                    if(bcMess!=null) Bukkit.broadcast(TextUtils.colorString(resolveIpBanMessagePlaceholders(bcMess, args[0], scope, reason, p.getName(), parsedBanDuration)));

                }
                return true;
            }
            else {
                plugin.getLogger().severe("BanHammer: Something went wrong while user used ban command... Maybe DB or Technician has skill issue???");
                p.sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.unexpectedError"))));
                return false;
            }

        }
        return false;
    }
    private boolean banIpCommand(String[] args) {
        if(args.length==0) Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.usage"))));
        else if(args.length>=1) {
            Timestamp banEndDate=null;
            String scope = "*", reason = plugin.translations.getString("default.reason"), duration="-1";

            if(args.length>=2) scope=args[1];
            if(args.length>=3) {
                duration=args[2];
                String regex = "^(perm)|(-1)|\\d+:(s|m|h|d)$";
                String regexOnlyNum = "^\\d+$";
                if(!(duration.matches(regex) || duration.matches(regexOnlyNum))) {
                    Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.durationFormatError"))));
                    return false;
                }
                char type = plugin.config.getString("defaultDurationType").charAt(0);
                Integer time=0;
                boolean chk = false;
                for(int i = 0; i < duration.length(); i++){
                    if(duration.charAt(i)==':'){
                        chk = true;
                        time = Integer.parseInt(duration.substring(0, i));
                        if(i+1 < duration.length()) type = duration.charAt(i+1);
                    }
                }
                if(!chk) time = Integer.parseInt(duration);
                switch (type){
                    case 'd':
                        banEndDate = new Timestamp(System.currentTimeMillis()+ TimeUnit.DAYS.toMillis(time));
                        break;
                    case 'h':
                        banEndDate = new Timestamp(System.currentTimeMillis()+TimeUnit.HOURS.toMillis(time));
                        break;
                    case 'm':
                        banEndDate = new Timestamp(System.currentTimeMillis()+TimeUnit.MINUTES.toMillis(time));
                        break;
                    case 's':
                        banEndDate = new Timestamp(System.currentTimeMillis()+TimeUnit.SECONDS.toMillis(time));
                        break;
                    default:
                        Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.durationFormatError"))));
                        return false;
                }
            }

            StringBuilder newReason = new StringBuilder();
            for(int i = 3; i < args.length; i++){
                newReason.append(args[i]).append(" ");
            }
            if(!newReason.isEmpty()) reason=newReason.toString();

            if(Objects.equals(duration, "-1") || Objects.equals(duration, "perm")) banEndDate = null;

            InetAddress addr;
            try {
                addr = InetAddress.getByName(args[0]);
            } catch (UnknownHostException e){
                plugin.getLogger().severe(e.toString());
                Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.unexpectedError"))));
                return false;
            }

            try {
                Connection conn = plugin.db.getConnection();
                PreparedStatement stmt = conn.prepareStatement("SELECT * FROM ip_bans WHERE (ip_address = ?::inet) AND scope = ?");
                stmt.setObject(1, addr.getHostAddress());
                stmt.setString(2, scope);
                ResultSet rs = stmt.executeQuery();
                if(rs.next()){
                    Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.alreadyBannedInScopeError"))));
                    return false;
                }
            } catch (SQLException e){
                plugin.getLogger().severe(e.toString());
                Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.unexpectedError"))));
                return false;
            }

            IPBan b = banIp(addr, reason, scope, banEndDate, "CONSOLE", null);

            if(b != null) {
                if(banEndDate == null) {
                    createIPKickJob(b, scope);
                    Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(resolveIpBanMessagePlaceholders(addPrefix(plugin.translations.getString("command.banip.successfulBanMessagePerm")), args[0], scope, reason, "CONSOLE")));
                    String bcMess = plugin.translations.getString("command.banip.banPublicAnnouncementPerm");
                    if(bcMess!=null) Bukkit.broadcast(TextUtils.colorString(resolveIpBanMessagePlaceholders(bcMess, args[0], scope, reason, "CONSOLE")));
                }
                else {
                    long timeLeft = banEndDate.getTime() - new Timestamp(System.currentTimeMillis()).getTime();
                    long timeLeftDays = TimeUnit.MILLISECONDS.toDays(timeLeft);
                    timeLeft-=TimeUnit.DAYS.toMillis(timeLeftDays);
                    long timeLeftHours = TimeUnit.MILLISECONDS.toHours(timeLeft);
                    timeLeft-=TimeUnit.HOURS.toMillis(timeLeftHours);
                    long timeLeftMinutes = TimeUnit.MILLISECONDS.toMinutes(timeLeft);
                    timeLeft-=TimeUnit.MINUTES.toMillis(timeLeftMinutes);
                    long timeLeftSeconds = TimeUnit.MILLISECONDS.toSeconds(timeLeft);
                    String parsedBanDuration = TextUtils.parsePolishDate(timeLeftDays, timeLeftHours, timeLeftMinutes, timeLeftSeconds);

                    createIPKickJob(b, scope);
                    Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(resolveIpBanMessagePlaceholders(addPrefix(plugin.translations.getString("command.banip.successfulBanMessageTemp")), args[0], scope, reason, "CONSOLE", parsedBanDuration)));
                    String bcMess = plugin.translations.getString("command.banip.banPublicAnnouncementTemp");
                    if(bcMess!=null) Bukkit.broadcast(TextUtils.colorString(resolveIpBanMessagePlaceholders(bcMess, args[0], scope, reason, "CONSOLE", parsedBanDuration)));

                }
                return true;
            }
            else {
                plugin.getLogger().severe("BanHammer: Something went wrong while user used ban command... Maybe DB or Technician has skill issue???");
                Bukkit.getConsoleSender().sendMessage(TextUtils.colorString(addPrefix(plugin.translations.getString("command.banip.unexpectedError"))));
                return false;
            }

        }
        return false;
    }

    private String addPrefix(String s) {
        return (String)(plugin.translations.getString("prefix")+s);
    }

    private IPBan banIp(InetAddress ipAddr, String reason, String scope, Timestamp endDate, String op, UUID opUuid) {
        try {
            Connection conn = plugin.db.getConnection();
            PreparedStatement ps = conn.prepareStatement("INSERT INTO ip_bans (ip_address, reason, scope, operator_name, operator_uuid, end_time) VALUES (?, ?, ?, ?, ?, ?)", Statement.RETURN_GENERATED_KEYS);
            ps.setObject(1, ipAddr.getHostAddress(), Types.OTHER);
            ps.setString(2, reason);
            ps.setString(3, scope);
            ps.setString(4, op);
            ps.setObject(5, opUuid);
            ps.setObject(6, endDate);
            if(ps.executeUpdate() == 0) return null;

            ResultSet rs = ps.getGeneratedKeys();
            rs.next();

            Timestamp ts = (Timestamp)rs.getObject("end_time");
            String parsedDate = null;

            if(ts!=null) {
                long timeLeft = (ts).getTime() - new Timestamp(System.currentTimeMillis()).getTime();
                long timeLeftDays = TimeUnit.MILLISECONDS.toDays(timeLeft);
                timeLeft-=TimeUnit.DAYS.toMillis(timeLeftDays);
                long timeLeftHours = TimeUnit.MILLISECONDS.toHours(timeLeft);
                timeLeft-=TimeUnit.HOURS.toMillis(timeLeftHours);
                long timeLeftMinutes = TimeUnit.MILLISECONDS.toMinutes(timeLeft);
                timeLeft-=TimeUnit.MINUTES.toMillis(timeLeftMinutes);
                long timeLeftSeconds = TimeUnit.MILLISECONDS.toSeconds(timeLeft);
                parsedDate = TextUtils.parsePolishDate(timeLeftDays, timeLeftHours, timeLeftMinutes, timeLeftSeconds);
            }

            IPBan b = new IPBan(ipAddr, reason, parsedDate, op, rs.getObject("id", UUID.class));
            return b;
        }
        catch (SQLException e) {
            plugin.getLogger().severe(e.toString());
            return null;
        }
    }

    private void createIPKickJob(IPBan b, String scope){
        final Gson gson = new Gson();
        try {
            Connection conn = plugin.db.getConnection();

            JsonElement banObject = gson.toJsonTree(b);
            JsonObject wrapper = new JsonObject();
            wrapper.addProperty("ban_type", "ip");
            wrapper.addProperty("ban_scope", scope);
            wrapper.add("ban_data", banObject);
            String json = gson.toJson(wrapper);

            PreparedStatement ps = conn.prepareStatement("SELECT pg_notify('ban_jobs', ?)");
            ps.setString(1, json);
            ps.execute();
        }
        catch (SQLException e) {
            plugin.getLogger().severe(e.toString());
        }
    }

    private String resolveIpBanMessagePlaceholders(String s, String ipAddr, String scope, String reason, String operator) {
        return s.replace("{ipAddr}", ipAddr)
                .replace("{scope}", scope)
                .replace("{reason}", reason)
                .replace("{operator}", operator)
                .replace("{prefix}", plugin.translations.getString("prefix"));
    }
    private String resolveIpBanMessagePlaceholders(String s, String player, String scope, String reason, String operator, String duration) {
        return resolveIpBanMessagePlaceholders(s, player, scope, reason, operator).replace("{duration}", duration);
    }

}