/*
 * Copyright 2018 John Grosh (john.a.grosh@gmail.com).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.jagrosh.vortex.automod;

import com.jagrosh.vortex.Action;
import com.jagrosh.vortex.Constants;
import com.jagrosh.vortex.Vortex;
import com.jagrosh.vortex.database.managers.PunishmentManager.Punishment;
import com.jagrosh.vortex.utils.FormatUtil;
import com.jagrosh.vortex.utils.LogUtil;
import com.jagrosh.vortex.utils.OtherUtil;
import java.time.Instant;
import java.time.OffsetDateTime;
import java.time.temporal.ChronoUnit;
import java.util.List;
import java.util.concurrent.TimeUnit;
import net.dv8tion.jda.api.Permission;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.Role;
import net.dv8tion.jda.api.entities.User;

/**
 *
 * @author John Grosh (john.a.grosh@gmail.com)
 */
public class StrikeHandler
{
    private final static String STRIKE_FORMAT = Constants.WARNING+" You have received `%d` strikes in **%s** for: `%s`";
    private final static String PARDON_FORMAT = Constants.SUCCESS+" You have been pardoned `%d` strikes in **%s** for: `%s`";
    private final static String PUNISH_FORMAT = "\n%s You have been **%s** from **%s**";
    private final static String PUNISH_FORMAT_TIME = "\n%s You have been **%s** for %s from **%s**";
    
    private final Vortex vortex;
    
    public StrikeHandler(Vortex vortex)
    {
        this.vortex = vortex;
    }
    
    public void pardonStrikes(Member moderator, OffsetDateTime nowo, long targetId, int number, String reason)
    {
        int[] counts = vortex.getDatabase().strikes.removeStrikes(moderator.getGuild(), targetId, number);
        Member member = moderator.getGuild().getMemberById(targetId);
        User user = member == null ? vortex.getShardManager().getUserById(targetId) : member.getUser();
        if(user==null)
        {
            moderator.getJDA().retrieveUserById(targetId).queue(u -> vortex.getModLogger().postPardonCase(moderator, nowo, number, counts[0], counts[1], u, reason));
        }
        else
        {
            String dmmsg = String.format(PARDON_FORMAT, number, moderator.getGuild().getName(), reason);
            vortex.getModLogger().postPardonCase(moderator, nowo, number, counts[0], counts[1], user, reason);
            OtherUtil.safeDM(user, dmmsg, moderator.getGuild().getMemberById(targetId)!=null, ()->{});
        }
    }
    
    public void applyStrikes(Member moderator, OffsetDateTime nowo, User target, int number, String reason)
    {
        boolean isMember = moderator.getGuild().getMemberById(target.getIdLong())!=null;
        Instant now = nowo.toInstant();
        int[] counts = vortex.getDatabase().strikes.addStrikes(moderator.getGuild(), target.getIdLong(), number);
        if(counts[0]>110 && counts[1]>110)
            return;
        List<Punishment> punishments = vortex.getDatabase().actions.getPunishments(moderator.getGuild(), counts[0], counts[1]);
        String dmmsg = String.format(STRIKE_FORMAT, number, moderator.getGuild().getName(), reason);
        if(punishments.isEmpty())
        {
            vortex.getModLogger().postStrikeCase(moderator, nowo, number, counts[0], counts[1], target, reason);
            OtherUtil.safeDM(target, dmmsg, isMember, ()->{});
        }
        else
        {
            String notimeaudit = LogUtil.auditStrikeReasonFormat(moderator, 0, counts[0], counts[1], reason);
            boolean canBan = moderator.getGuild().getSelfMember().hasPermission(Permission.BAN_MEMBERS);
            if(punishments.stream().anyMatch(p -> p.action==Action.BAN) && canBan)
            {
                OtherUtil.safeDM(target, dmmsg + punish(Action.BAN, moderator.getGuild()), isMember, 
                        () -> moderator.getGuild().ban(target, 7, notimeaudit).queue());
                vortex.getDatabase().tempbans.clearBan(moderator.getGuild(), target.getIdLong());
                return;
            }
            if(punishments.stream().anyMatch(p -> p.action==Action.SILENTBAN) && canBan)
            {
                OtherUtil.safeDM(target, dmmsg + punish(Action.SILENTBAN, moderator.getGuild()), isMember,
                        () -> moderator.getGuild().ban(target, 0, notimeaudit).queue());
                vortex.getDatabase().tempbans.clearBan(moderator.getGuild(), target.getIdLong());
                return;
            }
            int muteDuration = 0;
            int banDuration = 0;
            for(Punishment p: punishments)
            {
                if(p.action==Action.MUTE)
                    muteDuration = Integer.MAX_VALUE;
                else if(p.action==Action.TEMPMUTE && p.time>muteDuration)
                    muteDuration = p.time;
                else if(p.action==Action.TEMPBAN && p.time>banDuration)
                    banDuration = p.time;
            }
            if(banDuration>0 && canBan)
            {
                int finalBanDuration = banDuration;
                OtherUtil.safeDM(target, dmmsg + punishTime(Action.TEMPBAN, moderator.getGuild(), banDuration), isMember, 
                        () -> moderator.getGuild().ban(target, 7, LogUtil.auditStrikeReasonFormat(moderator, finalBanDuration, counts[0], counts[1], reason)).queue());
                vortex.getDatabase().tempbans.setBan(moderator.getGuild(), target.getIdLong(), now.plus(banDuration, ChronoUnit.MINUTES));
                if(muteDuration>0)
                    vortex.getDatabase().tempmutes.setMute(moderator.getGuild(), target.getIdLong(), muteTime(now, muteDuration));
                return;
            }
            if(punishments.stream().anyMatch(p -> p.action==Action.SOFTBAN) && canBan)
            {
                OtherUtil.safeDM(target, dmmsg + punish(Action.SOFTBAN, moderator.getGuild()), isMember, 
                        () -> moderator.getGuild().ban(target, 7, notimeaudit).queue(
                                s -> moderator.getGuild().unban(target).reason(notimeaudit).queueAfter(4, TimeUnit.SECONDS)));
                if(muteDuration>0)
                    vortex.getDatabase().tempmutes.setMute(moderator.getGuild(), target.getIdLong(), muteTime(now, muteDuration));
                return;
            }
            boolean canKick = moderator.getGuild().getSelfMember().hasPermission(Permission.KICK_MEMBERS);
            if(punishments.stream().anyMatch(p -> p.action==Action.KICK) && canKick)
            {
                if(moderator.getGuild().isMember(target))
                {
                    OtherUtil.safeDM(target, dmmsg + punish(Action.KICK, moderator.getGuild()), isMember, 
                        () -> moderator.getGuild().kick(target.getId(), notimeaudit).queue());
                }
                else
                {
                    vortex.getModLogger().postStrikeCase(moderator, nowo, number, counts[0], counts[1], target, reason);
                }
                if(muteDuration>0)
                    vortex.getDatabase().tempmutes.setMute(moderator.getGuild(), target.getIdLong(), muteTime(now, muteDuration));
                return;
            }
            boolean canMute = moderator.getGuild().getSelfMember().hasPermission(Permission.MANAGE_ROLES);
            if(muteDuration>0 && canMute)
            {
                vortex.getDatabase().tempmutes.setMute(moderator.getGuild(), target.getIdLong(), muteTime(now, muteDuration));
                Role muted = vortex.getDatabase().settings.getSettings(moderator.getGuild()).getMutedRole(moderator.getGuild());
                Member mem = moderator.getGuild().getMember(target);
                if(muted==null || mem==null || !moderator.getGuild().getSelfMember().canInteract(muted))
                {
                    vortex.getModLogger().postStrikeCase(moderator, nowo, number, counts[0], counts[1], target, reason);
                    OtherUtil.safeDM(target, dmmsg, isMember, ()->{});
                    return;
                }
                if(mem.getRoles().contains(muted))
                {
                    vortex.getModLogger().postPseudoCase(moderator, nowo, 
                            muteDuration==Integer.MAX_VALUE ? Action.MUTE : Action.TEMPMUTE, target, 
                            muteDuration==Integer.MAX_VALUE ? 0 : muteDuration, "["+counts[1]+" strikes] "+reason);
                }
                else
                {
                    moderator.getGuild().addRoleToMember(mem, muted)
                        .reason(muteDuration==Integer.MAX_VALUE ? notimeaudit : LogUtil.auditStrikeReasonFormat(moderator, muteDuration, counts[0], counts[1], reason))
                        .queue();
                }
                OtherUtil.safeDM(target, dmmsg + (muteDuration==Integer.MAX_VALUE ? punish(Action.MUTE, moderator.getGuild()) 
                        : punishTime(Action.TEMPMUTE, moderator.getGuild(), muteDuration)), isMember, ()->{});
            }
        }
    }
    
    private static String punish(Action action, Guild guild)
    {
        return String.format(PUNISH_FORMAT, action.getEmoji(), action.getVerb(), guild.getName());
    }
    
    private static String punishTime(Action action, Guild guild, int minutes)
    {
        return String.format(PUNISH_FORMAT_TIME, action.getEmoji(), action.getVerb(), FormatUtil.secondsToTime(minutes*60), guild.getName());
    }
    
    private static Instant muteTime(Instant now, int minutes)
    {
        return minutes==Integer.MAX_VALUE ? Instant.MAX : now.plus(minutes, ChronoUnit.MINUTES);
    }
}