devbury.dewey.plugins.RemindMe.java Source code

Introduction

Here is the source code for devbury.dewey.plugins.RemindMe.java

Source

/*
 * Copyright 2015 devbury LLC
 *
 *    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 devbury.dewey.plugins;

import devbury.dewey.core.event.MessageEvent;
import devbury.dewey.core.event.MessageEventListener;
import devbury.dewey.core.model.Address;
import devbury.dewey.core.model.Message;
import devbury.dewey.core.model.User;
import devbury.dewey.core.server.ChatServer;
import devbury.dewey.core.server.Plugin;
import devbury.dewey.plugins.help.Usage;
import devbury.dewey.plugins.help.UsageSummary;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.TaskScheduler;

import java.time.Instant;
import java.time.temporal.ChronoUnit;
import java.util.Date;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@Plugin
public class RemindMe implements MessageEventListener, Usage {

    private static final Logger logger = LoggerFactory.getLogger(RemindMe.class);

    private final Pattern pattern = Pattern.compile(
            "^remind +(me|us) +in +(\\d+) +(second|seconds|minute|minutes|hour|hours|day|days|week|weeks|month|months) +to +(.*)$");

    @Autowired
    private ChatServer chatServer;

    @Autowired
    private TaskScheduler taskScheduler;

    @Override
    public void onEvent(MessageEvent event) {
        Message message = event.getMessage();
        if (message.isCommand()) {
            Matcher matcher = pattern.matcher(message.getCommand());
            if (matcher.matches()) {
                String notify = matcher.group(1);
                long timeAmount = Long.parseLong(matcher.group(2));
                String timeUnits = matcher.group(3);
                String reminderMessage = matcher.group(4);

                User from = message.getFrom();

                boolean groupNotify = message.isToGroup() && notify.equals("us");
                String mention = message.isToGroup() ? message.getFrom().getMentionName() : "";
                Address replyTo = message.isToGroup() ? message.getGroup() : from;
                if (groupNotify) {
                    notify = "@All, " + mention + " asked me to remind everyone to '" + reminderMessage + "'";
                } else {
                    notify = mention + " You asked me to remind you to " + matcher.group(4)
                            .replaceAll(" my ", " your ").replaceAll(" me ", " you ").replaceAll(" [Ii] ", " you ");
                }

                String body = notify;

                scheduleMessage(replyTo, body, notifyAt(timeAmount, timeUnits));

                if (groupNotify) {
                    chatServer.sendMessage(replyTo, mention + " Sure, I'll remind everyone");
                } else {
                    chatServer.sendMessage(replyTo, mention + " Sure, I'll remind you");
                }
                logger.debug("processedMessage");
                return;
            }
        }

        logger.debug("did not process message {}", message.getBody());
    }

    @Override
    public UsageSummary usageSummary() {
        return new UsageSummary("Remind the user to do something in the future").addCommand("remind");
    }

    @Override
    public String usageDetails(String command, String args) {
        return "remind (me|us) in X (second|seconds|minute|minutes|hour|hours|day|days|week|weeks|month|months) to Y";
    }

    protected void scheduleMessage(Address replyTo, String message, Date notifyAt) {
        taskScheduler.schedule(() -> chatServer.sendMessage(replyTo, message), notifyAt);
    }

    private Date notifyAt(long amount, String units) {
        ChronoUnit chronoUnit = ChronoUnit.SECONDS;
        switch (units) {
        case "weeks":
        case "week":
            chronoUnit = ChronoUnit.WEEKS;
            break;
        case "months":
        case "month":
            chronoUnit = ChronoUnit.MONTHS;
            break;
        case "days":
        case "day":
            chronoUnit = ChronoUnit.DAYS;
            break;
        case "hours":
        case "hour":
            chronoUnit = ChronoUnit.HOURS;
            break;
        case "minutes":
        case "minute":
            chronoUnit = ChronoUnit.MINUTES;
            break;
        }
        return Date.from(Instant.now().plus(amount, chronoUnit));
    }
}