chomens-bot-java/src/main/java/land/chipmunk/chayapak/chomens_bot/Bot.java

package land.chipmunk.chayapak.chomens_bot;

import com.github.steveice10.mc.auth.data.GameProfile;
import com.github.steveice10.mc.protocol.MinecraftProtocol;
import com.github.steveice10.mc.protocol.packet.ingame.clientbound.ClientboundLoginPacket;
import com.github.steveice10.mc.protocol.packet.login.clientbound.ClientboundGameProfilePacket;
import com.github.steveice10.packetlib.Session;
import com.github.steveice10.packetlib.event.session.*;
import com.github.steveice10.packetlib.packet.Packet;
import com.github.steveice10.packetlib.tcp.TcpClientSession;
import land.chipmunk.chayapak.chomens_bot.plugins.*;
import land.chipmunk.chayapak.chomens_bot.util.ComponentUtilities;
import land.chipmunk.chayapak.chomens_bot.util.RandomStringUtilities;
import lombok.Getter;
import lombok.Setter;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class Bot {
    private final ArrayList<Listener> listeners = new ArrayList<>();

    @Getter private final String host;
    @Getter private final int port;

    @Getter private final Configuration.BotOption options;

    @Getter private final Configuration config;

    @Getter private List<Bot> bots;

    @Getter private String username;

    @Getter private GameProfile profile;

    @Getter public Session session;

    @Getter private boolean loggedIn = false;

    @Getter private final ExecutorService executorService = Main.executorService;
    @Getter private final ScheduledExecutorService executor = Main.executor;

    @Getter @Setter private ConsolePlugin console;
    @Getter @Setter private LoggerPlugin logger; // in ConsolePlugin
    @Getter @Setter private DiscordPlugin discord; // same for this one too

    @Getter private TickPlugin tick;
    @Getter private ChatPlugin chat;
    @Getter private PositionPlugin position;
    @Getter private SelfCarePlugin selfCare;
    @Getter private CorePlugin core;
    @Getter private TeamPlugin team;
    @Getter private PlayersPlugin players;
    @Getter private TabCompletePlugin tabComplete;
    @Getter private CommandHandlerPlugin commandHandler;
    @Getter private ChatCommandHandlerPlugin chatCommandHandler;
    @Getter private HashingPlugin hashing;
    @Getter private BossbarManagerPlugin bossbar;
    @Getter private MusicPlayerPlugin music;
    @Getter private TPSPlugin tps;
    @Getter private EvalRunnerPlugin eval;
    @Getter private ClearChatUsernamePlugin clearChatUsername;
    @Getter private TrustedPlugin trusted;
    @Getter private BruhifyPlugin bruhify;
    @Getter private GrepLogPlugin grepLog;
    @Getter private CloopPlugin cloop;
    @Getter private MazePlugin maze;
    @Getter private ExploitsPlugin exploits;
    @Getter private FilterPlugin filter;
    @Getter private CommandSuggestionPlugin commandSuggestion;
    @Getter private MailPlugin mail;
    @Getter private PacketSnifferPlugin packetSniffer;
    @Getter private VoiceChatPlugin voiceChat;

    public Bot (Configuration.BotOption botOption, List<Bot> bots, Configuration config) {
        this.host = botOption.host;
        this.port = botOption.port;

        this.options = botOption;

        this.bots = bots;

        this.config = config;

        ConsolePlugin.addListener(new ConsolePlugin.Listener() {
            @Override
            public void ready() {
                Bot.this.ready();
            }
        });
    }

    public void ready () {
        this.tick = new TickPlugin(this);
        this.chat = new ChatPlugin(this);
        this.position = new PositionPlugin(this);
        this.selfCare = new SelfCarePlugin(this);
        this.core = new CorePlugin(this);
        this.team = new TeamPlugin(this);
        this.players = new PlayersPlugin(this);
        this.tabComplete = new TabCompletePlugin(this);
        this.commandHandler = new CommandHandlerPlugin(this);
        this.chatCommandHandler = new ChatCommandHandlerPlugin(this);
        this.hashing = new HashingPlugin(this);
        this.bossbar = new BossbarManagerPlugin(this);
        this.music = new MusicPlayerPlugin(this);
        this.tps = new TPSPlugin(this);
        this.eval = new EvalRunnerPlugin(this);
        this.clearChatUsername = new ClearChatUsernamePlugin(this);
        this.trusted = new TrustedPlugin(this);
        this.bruhify = new BruhifyPlugin(this);
        this.grepLog = new GrepLogPlugin(this);
        this.cloop = new CloopPlugin(this);
        this.maze = new MazePlugin(this);
        this.exploits = new ExploitsPlugin(this);
        this.filter = new FilterPlugin(this);
        this.commandSuggestion = new CommandSuggestionPlugin(this);
        this.mail = new MailPlugin(this);
        this.packetSniffer = new PacketSnifferPlugin(this);
        this.voiceChat = new VoiceChatPlugin(this);

        for (Listener listener : listeners) listener.loadedPlugins();

        reconnect();
    }

    private void reconnect () {
        if (session != null) session = null; // does this do nothing?

        for (Listener listener : listeners) {
            listener.connecting();
        }

        final String _username = options.username();

        if (_username == null) username = RandomStringUtilities.generate(8);
        else username = _username;

        Session session = new TcpClientSession(host, port, new MinecraftProtocol(username), null);

        this.session = session;

        session.addListener(new SessionAdapter() {
            // fard

            @Override
            public void packetReceived(Session session, Packet packet) {
                for (SessionListener listener : listeners) {
                    listener.packetReceived(session, packet);
                }

                if (packet instanceof ClientboundLoginPacket) {
                    for (SessionListener listener : listeners) {
                        loggedIn = true;
                        listener.connected(new ConnectedEvent(session));
                    }
                } else if (packet instanceof ClientboundGameProfilePacket) packetReceived((ClientboundGameProfilePacket) packet);
            }

            public void packetReceived(ClientboundGameProfilePacket packet) {
                profile = packet.getProfile();
            }

            @Override
            public void packetSending(PacketSendingEvent packetSendingEvent) {
                for (SessionListener listener : listeners) {
                    listener.packetSending(packetSendingEvent);
                }
            }

            @Override
            public void packetSent(Session session, Packet packet) {
                for (SessionListener listener : listeners) {
                    listener.packetSent(session, packet);
                }
            }

            @Override
            public void packetError(PacketErrorEvent packetErrorEvent) {
                for (SessionListener listener : listeners) {
                    listener.packetError(packetErrorEvent);
                }
                packetErrorEvent.setSuppress(true); // fix the ohio sus exploit
            }

            @Override
            public void disconnecting(DisconnectingEvent disconnectingEvent) {
                for (SessionListener listener : listeners) {
                    listener.disconnecting(disconnectingEvent);
                }
            }

            @Override
            public void disconnected(DisconnectedEvent disconnectedEvent) {
                loggedIn = false;

                int reconnectDelay = options.reconnectDelay();

                final String stringMessage = ComponentUtilities.stringify(disconnectedEvent.getReason());

                // this part is ported from chomens bot js
                if (
                        stringMessage.equals("Wait 5 seconds before connecting, thanks! :)") ||
                                stringMessage.equals("You are logging in too fast, try again later.")
                ) reconnectDelay = 1000 * 7;

                executor.schedule(() -> reconnect(), reconnectDelay, TimeUnit.MILLISECONDS);

                for (SessionListener listener : listeners) {
                    listener.disconnected(disconnectedEvent);
                }
            }
        });

        session.connect();
    }

    public void addListener (Listener listener) {
        listeners.add(listener);
    }

    public static class Listener extends SessionAdapter {
        public void connecting () {}
        public void loadedPlugins () {}
    }
}