com.krrrr38.jabot.plugin.handler.ReplaceHandler Maven / Gradle / Ivy

Go to download
package com.krrrr38.jabot.plugin.handler;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

public class ReplaceHandler extends Handler {

    @Override
    List build(Map options) {
        return Arrays.asList(listRule(), deleteRule(), deleteAllRule(), registerRule(), replaceRule());
    }

    private Rule listRule() {
        return new Rule(
                Pattern.compile("\\Alist patterns\\z", Pattern.CASE_INSENSITIVE),
                "list patterns",
                "Show registered patterns to replace",
                "replace patterns",
                false,
                strings -> brainGuard(() -> {
                    Map patterns = getAll();
                    if (patterns.isEmpty()) {
                        send("No registry replace patterns");
                    } else {
                        String header = "=== Replace Patterns ===\n";
                        String message = patterns.entrySet().stream().map(e -> {
                            return String.format("%s → %s", e.getKey(), e.getValue());
                        }).collect(Collectors.joining("\n"));
                        send(header + message);
                    }
                    return Optional.empty();
                })
        );
    }

    private Rule deleteRule() {
        return new Rule(
                Pattern.compile("\\Adelete pattern (.+)\\z", Pattern.CASE_INSENSITIVE),
                "delete replate pattern",
                "Delete replace pattern with key",
                "delete pattern ",
                false,
                strings -> brainGuard(() -> {
                    if (delete(strings[0].trim())) {
                        send("Deleted");
                    } else {
                        send("NotFound");
                    }
                    return Optional.empty();
                })
        );
    }

    private Rule deleteAllRule() {
        return new Rule(
                Pattern.compile("\\Adelete all patterns\\z", Pattern.CASE_INSENSITIVE),
                "delete all replace patterns",
                "Delete all replace patterns",
                "delete all patterns",
                false,
                strings -> brainGuard(() -> {
                    if (clear()) {
                        send("Deleted all patterns");
                    } else {
                        send("Failed to clear patterns");
                    }
                    return Optional.empty();
                })
        );
    }

    private Rule registerRule() {
        return new Rule(
                Pattern.compile("\\Areplace (.+) with (.+)\\z", Pattern.CASE_INSENSITIVE),
                "register replace pattern",
                "Register replace pattern",
                "replace  with ",
                false,
                strings -> brainGuard(() -> {
                    String from = strings[0];
                    String to = strings[1];
                    if (store(from, to)) {
                        send(String.format("Registered pattern: %s → %s", from, to));
                    } else {
                        send(String.format("Failed to registry pattern: %s → %s", from, to));
                    }
                    return Optional.empty();
                })
        );
    }

    private Rule replaceRule() {
        return new Rule(
                Pattern.compile("\\A(.+)\\z"),
                "replace all message",
                "Reply your message based on registered patterns",
                "*",
                false,
                strings -> brainGuard(() -> {
                    String message = strings[0];
                    for (Map.Entry entry : getAll().entrySet()) {
                        message = message.replace(entry.getKey(), entry.getValue());
                    }
                    return Optional.of(message);
                })
        );
    }

    @Override
    public void afterRegister(List handlers) {

    }
}