me.rkfg.xmpp.bot.plugins.FaggotOfTheDayPlugin.java Source code

Introduction

Here is the source code for me.rkfg.xmpp.bot.plugins.FaggotOfTheDayPlugin.java

Source

/**
 * Plugin for Faggot of the Day game in MUCs.
 * Copyright (C) 2017 Kona-chan
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package me.rkfg.xmpp.bot.plugins;

import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.ZoneId;
import java.util.Arrays;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Random;
import java.util.Set;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.regex.Matcher;
import java.util.stream.Collectors;

import org.apache.commons.lang3.StringUtils;
import org.jivesoftware.smack.packet.Message;
import org.jivesoftware.smackx.muc.MultiUserChat;
import org.jivesoftware.smackx.muc.Occupant;
import org.jxmpp.util.XmppStringUtils;

import ru.ppsrk.gwt.client.ClientAuthException;
import ru.ppsrk.gwt.client.LogicException;

/**
 * Lets users of MUCs play Faggot of the Day.
 *
 * @author Kona-chan
 * @version 0.3.0
 */
public final class FaggotOfTheDayPlugin extends CommandPlugin {

    private static final List<String> ALL_COMMANDS = Arrays.asList("pidor", "");
    private static final String INFO_FAGGOT_IS = "??  ?  ";
    private static final String INFO_FAGGOT_IS_YOU = INFO_FAGGOT_IS + "!";

    private static final long PERIOD = TimeUnit.DAYS.toMillis(1);

    private final Set<Occupant> occupants = new HashSet<>();
    private Occupant faggot;

    private final AtomicBoolean startedListener = new AtomicBoolean(false);
    private final Timer timer = new Timer();
    private final Random random = new Random();

    @Override
    public void init() {
        startTimer();
    }

    @Override
    public String processCommand(Message message, Matcher matcher) throws LogicException, ClientAuthException {
        startListening();

        if (faggot == null) {
            calculateFaggot();
        }

        final Occupant sender = getMUCManager().getMUCOccupant(message.getFrom());
        final String senderJid = XmppStringUtils.parseBareJid(sender.getJid());
        final String faggotJid = XmppStringUtils.parseBareJid(faggot.getJid());
        if (senderJid.equals(faggotJid)) {
            return INFO_FAGGOT_IS_YOU;
        }

        final String roomJid = XmppStringUtils.parseBareJid(message.getFrom());
        final Occupant occupant = Optional.ofNullable(getMUCManager().listMUCOccupantsByJID(roomJid).get(faggotJid))
                .orElse(faggot);

        return INFO_FAGGOT_IS + occupant.getNick() + ".";
    }

    @Override
    public List<String> getCommand() {
        return ALL_COMMANDS;
    }

    @Override
    public String getManual() {
        final String commands = StringUtils.join(ALL_COMMANDS, "|");
        final String sampleCommand = PREFIX + ALL_COMMANDS.get(0);

        return ",  ??  ?.\n" + ": <" + commands + ">\n"
                + ": " + sampleCommand;
    }

    private void startTimer() {
        timer.scheduleAtFixedRate(new TimerTask() {

            @Override
            public void run() {
                log.info(" A new day starts in the Empire ?");

                calculateFaggot();
            }

        }, getFirstTime(), PERIOD);
    }

    private Date getFirstTime() {
        final LocalTime midnight = LocalTime.MIDNIGHT;
        final LocalDate today = LocalDate.now();
        final LocalDateTime todayMidnight = LocalDateTime.of(today, midnight);
        return Date.from(todayMidnight.atZone(ZoneId.systemDefault()).toInstant());
    }

    private void startListening() {
        if (startedListener.getAndSet(true)) {
            return;
        }

        getMUCManager().listMUCs().forEach(muc -> {
            muc.addMessageListener(message -> {
                occupants.add(muc.getOccupant(message.getFrom()));
            });

            occupants.addAll(getAllOccupants(muc));
        });
    }

    private Set<Occupant> getAllOccupants(MultiUserChat muc) {
        return muc.getOccupants().stream().map(muc::getOccupant)
                .filter(occupant -> !occupant.getNick().equals(getBotNick())).collect(Collectors.toSet());
    }

    private void calculateFaggot() {
        occupants.removeIf(occupant -> occupant.getNick().equals(getBotNick()));

        if (occupants.isEmpty()) {
            log.info("No contenders for Faggot of the Day today.");
            if (faggot != null) {
                log.info("{} remains Faggot of the Day!", faggot.getNick());
            }
            return;
        }

        final Set<String> uniqueJids = occupants.stream().map(Occupant::getJid).map(XmppStringUtils::parseBareJid)
                .collect(Collectors.toSet());
        log.info("Contenders for todays Faggot of the Day title: {}", uniqueJids);

        final int i = random.nextInt(uniqueJids.size());
        final String faggotJid = uniqueJids.stream().skip(i).findFirst().get();
        faggot = occupants.stream()
                .filter(occupant -> faggotJid.equals(XmppStringUtils.parseBareJid(occupant.getJid()))).findFirst()
                .get();
        log.info("{} becomes Faggot of the Day!", faggot.getNick());

        occupants.clear();
    }

}