com.github.twitch4j.chat.flag.FlagParser Maven / Gradle / Ivy

Go to download
package com.github.twitch4j.chat.flag;

import com.github.twitch4j.chat.events.channel.IRCMessageEvent;
import lombok.NonNull;
import lombok.experimental.UtilityClass;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

@UtilityClass
public class FlagParser {

    private final String IRC_TAG_NAME = "flags";

    @NonNull
    public List parseFlags(@NonNull String rawFlags) {
        final String[] splitFlags = rawFlags.split(",");
        final List flags = new ArrayList<>(splitFlags.length);

        for (String rawFlag : splitFlags) {
            String[] parts = rawFlag.split(":", -1);
            if (parts.length != 2)
                continue;

            String[] indices = parts[0].split("-");
            if (indices.length != 2)
                continue;
            int start = Integer.parseInt(indices[0]);
            int end = Integer.parseInt(indices[1]);

            AutoModFlag.AutoModFlagBuilder builder = AutoModFlag.builder()
                .startIndex(start)
                .endIndex(end);

            String[] scores = parts[1].split("/");
            for (String score : scores) {
                String[] scoreParts = score.split("\\.");
                if (scoreParts.length != 2)
                    continue;
                FlagType type = FlagType.parse(scoreParts[0]);
                if (type != null)
                    builder.score(type, Integer.parseInt(scoreParts[1]));
            }

            flags.add(builder.build());
        }

        return Collections.unmodifiableList(flags);
    }

    @NonNull
    public List parseFlags(@NonNull IRCMessageEvent event) {
        return event.getTagValue(IRC_TAG_NAME).map(FlagParser::parseFlags).orElse(Collections.emptyList());
    }

}