package com.minestudio.replica.command; import com.minestudio.replica.capture.CaptureManager; import com.minestudio.replica.data.CaptureSession; import com.mojang.brigadier.CommandDispatcher; import com.mojang.brigadier.arguments.StringArgumentType; import com.mojang.brigadier.context.CommandContext; import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource; import net.minecraft.text.Text; import net.minecraft.util.Formatting; import java.util.List; import static net.fabricmc.fabric.api.client.command.v2.ClientCommandManager.argument; import static net.fabricmc.fabric.api.client.command.v2.ClientCommandManager.literal; public class ReplicaCommand { public static void register(CommandDispatcher dispatcher) { dispatcher.register(literal("replica") .then(literal("lobby") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureLobby(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("scoreboard") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureScoreboard(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("tab") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureTab(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("entities") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureEntities(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("holograms") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureHolograms(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("titles") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureTitles(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("build") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureBuild(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("ui") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureUI(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("full") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> captureFull(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("stop") .executes(ReplicaCommand::stopCapture)) .then(literal("pause") .executes(ReplicaCommand::pauseCapture)) .then(literal("resume") .executes(ReplicaCommand::resumeCapture)) .then(literal("status") .executes(ReplicaCommand::showStatus)) .then(literal("list") .executes(ctx -> listSessions(ctx, "all")) .then(argument("type", StringArgumentType.word()) .executes(ctx -> listSessions(ctx, StringArgumentType.getString(ctx, "type"))))) .then(literal("view") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> viewSession(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("export") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> exportSession(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("delete") .then(argument("name", StringArgumentType.greedyString()) .executes(ctx -> deleteSession(ctx, StringArgumentType.getString(ctx, "name"))))) .then(literal("clear") .executes(ReplicaCommand::clearAllSessions)) .then(literal("help") .executes(ReplicaCommand::showHelp)) .executes(ReplicaCommand::showHelp) ); } private static int captureLobby(CommandContext ctx, String name) { return startCapture(ctx, name, "LOBBY", "lobby visual elements"); } private static int captureScoreboard(CommandContext ctx, String name) { return startCapture(ctx, name, "SCOREBOARD", "scoreboard data"); } private static int captureTab(CommandContext ctx, String name) { return startCapture(ctx, name, "TAB", "tab list"); } private static int captureEntities(CommandContext ctx, String name) { return startCapture(ctx, name, "ENTITIES", "entities and NPCs"); } private static int captureHolograms(CommandContext ctx, String name) { return startCapture(ctx, name, "HOLOGRAMS", "holograms"); } private static int captureTitles(CommandContext ctx, String name) { return startCapture(ctx, name, "TITLES", "titles and subtitles"); } private static int captureBuild(CommandContext ctx, String name) { return startCapture(ctx, name, "BUILD", "world build data"); } private static int captureUI(CommandContext ctx, String name) { return startCapture(ctx, name, "UI", "UI elements"); } private static int captureFull(CommandContext ctx, String name) { return startCapture(ctx, name, "FULL", "all visual elements"); } private static int startCapture(CommandContext ctx, String name, String type, String description) { CaptureManager manager = CaptureManager.getInstance(); if (manager.isCapturing()) { ctx.getSource().sendFeedback(Text.literal("⚠ Already capturing! Stop current session first.") .formatted(Formatting.YELLOW)); return 0; } manager.startCapture(name, type); ctx.getSource().sendFeedback(Text.literal("▶ Started capturing ") .formatted(Formatting.AQUA) .append(Text.literal(description).formatted(Formatting.LIGHT_PURPLE)) .append(Text.literal(" as \"").formatted(Formatting.AQUA)) .append(Text.literal(name).formatted(Formatting.WHITE, Formatting.BOLD)) .append(Text.literal("\"").formatted(Formatting.AQUA))); ctx.getSource().sendFeedback(Text.literal("💾 Recording " + type + " mode...").formatted(Formatting.GRAY)); return 1; } private static int stopCapture(CommandContext ctx) { CaptureManager manager = CaptureManager.getInstance(); if (!manager.isCapturing()) { ctx.getSource().sendFeedback(Text.literal("⚠ No active capture session.") .formatted(Formatting.YELLOW)); return 0; } CaptureSession session = manager.stopCaptureAndReturn(); ctx.getSource().sendFeedback(Text.literal("■ Stopped capture: ") .formatted(Formatting.RED) .append(Text.literal(session.getName()).formatted(Formatting.WHITE, Formatting.BOLD))); ctx.getSource().sendFeedback(Text.literal("📊 Recorded: ") .formatted(Formatting.GRAY) .append(Text.literal(session.getScoreboards().size() + " scoreboards, ") .formatted(Formatting.AQUA)) .append(Text.literal(session.getTabLists().size() + " tab updates, ") .formatted(Formatting.AQUA)) .append(Text.literal(session.getEntities().size() + " entities") .formatted(Formatting.AQUA))); return 1; } private static int pauseCapture(CommandContext ctx) { CaptureManager manager = CaptureManager.getInstance(); if (!manager.isCapturing()) { ctx.getSource().sendFeedback(Text.literal("⚠ No active capture session.") .formatted(Formatting.YELLOW)); return 0; } manager.pauseCapture(); ctx.getSource().sendFeedback(Text.literal("⏸ Capture paused. Use /replica resume to continue.") .formatted(Formatting.GOLD)); return 1; } private static int resumeCapture(CommandContext ctx) { CaptureManager manager = CaptureManager.getInstance(); if (!manager.isPaused()) { ctx.getSource().sendFeedback(Text.literal("⚠ No paused capture session.") .formatted(Formatting.YELLOW)); return 0; } manager.resumeCapture(); ctx.getSource().sendFeedback(Text.literal("▶ Capture resumed.") .formatted(Formatting.GREEN)); return 1; } private static int showStatus(CommandContext ctx) { CaptureManager manager = CaptureManager.getInstance(); if (!manager.isCapturing()) { ctx.getSource().sendFeedback(Text.literal("⦿ No active capture") .formatted(Formatting.GRAY) .append(Text.literal(" | Total sessions: ") .formatted(Formatting.DARK_GRAY)) .append(Text.literal(String.valueOf(manager.getAllSessions().size())) .formatted(Formatting.WHITE))); return 1; } CaptureSession current = manager.getCurrentSession(); long duration = (System.currentTimeMillis() - current.getStartTime()) / 1000; ctx.getSource().sendFeedback(Text.literal("━━━━━━━━━━ REPLICA STATUS ━━━━━━━━━━") .formatted(Formatting.AQUA, Formatting.BOLD)); ctx.getSource().sendFeedback(Text.literal("📹 Capturing: ") .formatted(Formatting.GRAY) .append(Text.literal(current.getName()).formatted(Formatting.WHITE, Formatting.BOLD)) .append(Text.literal(" [" + current.getCaptureType() + "]") .formatted(Formatting.LIGHT_PURPLE))); ctx.getSource().sendFeedback(Text.literal("⏱ Duration: ") .formatted(Formatting.GRAY) .append(Text.literal(duration + "s").formatted(Formatting.WHITE))); ctx.getSource().sendFeedback(Text.literal("📊 Data: ") .formatted(Formatting.GRAY) .append(Text.literal(current.getScoreboards().size() + " scoreboards, ") .formatted(Formatting.AQUA)) .append(Text.literal(current.getTabLists().size() + " tab, ") .formatted(Formatting.AQUA)) .append(Text.literal(current.getEntities().size() + " entities, ") .formatted(Formatting.AQUA)) .append(Text.literal(current.getTitles().size() + " titles") .formatted(Formatting.AQUA))); ctx.getSource().sendFeedback(Text.literal("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") .formatted(Formatting.AQUA)); return 1; } private static int listSessions(CommandContext ctx, String type) { CaptureManager manager = CaptureManager.getInstance(); List sessions = manager.getAllSessions(); if (sessions.isEmpty()) { ctx.getSource().sendFeedback(Text.literal("📁 No saved sessions.") .formatted(Formatting.GRAY)); return 0; } ctx.getSource().sendFeedback(Text.literal("━━━━━━━━━ SAVED SESSIONS (" + sessions.size() + ") ━━━━━━━━━") .formatted(Formatting.AQUA, Formatting.BOLD)); for (CaptureSession session : sessions) { if (!type.equals("all") && !session.getCaptureType().equalsIgnoreCase(type)) { continue; } ctx.getSource().sendFeedback(Text.literal("▪ ") .formatted(Formatting.DARK_AQUA) .append(Text.literal(session.getName()).formatted(Formatting.WHITE, Formatting.BOLD)) .append(Text.literal(" [" + session.getCaptureType() + "]").formatted(Formatting.LIGHT_PURPLE)) .append(Text.literal(" - " + session.getFormattedTimestamp()).formatted(Formatting.GRAY))); } ctx.getSource().sendFeedback(Text.literal("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") .formatted(Formatting.AQUA)); return 1; } private static int viewSession(CommandContext ctx, String name) { CaptureManager manager = CaptureManager.getInstance(); CaptureSession session = manager.getSession(name); if (session == null) { ctx.getSource().sendFeedback(Text.literal("⚠ Session not found: " + name) .formatted(Formatting.RED)); return 0; } ctx.getSource().sendFeedback(Text.literal("━━━━━━━━ " + session.getName() + " [" + session.getCaptureType() + "] ━━━━━━━━") .formatted(Formatting.AQUA, Formatting.BOLD)); ctx.getSource().sendFeedback(Text.literal("📅 Recorded: ") .formatted(Formatting.GRAY) .append(Text.literal(session.getFormattedTimestamp()).formatted(Formatting.WHITE))); ctx.getSource().sendFeedback(Text.literal("\n📊 SCOREBOARD DATA (" + session.getScoreboards().size() + "):") .formatted(Formatting.YELLOW, Formatting.BOLD)); session.getScoreboards().stream().limit(5).forEach(sb -> { ctx.getSource().sendFeedback(Text.literal(" ▸ " + sb.getTitle() + " (" + sb.getLines().size() + " lines)") .formatted(Formatting.AQUA)); }); ctx.getSource().sendFeedback(Text.literal("\n👥 TAB LIST DATA (" + session.getTabLists().size() + "):") .formatted(Formatting.YELLOW, Formatting.BOLD)); session.getTabLists().stream().limit(5).forEach(tab -> { ctx.getSource().sendFeedback(Text.literal(" ▸ " + tab.getPlayers().size() + " players") .formatted(Formatting.AQUA)); }); ctx.getSource().sendFeedback(Text.literal("\n🎭 ENTITIES (" + session.getEntities().size() + "):") .formatted(Formatting.YELLOW, Formatting.BOLD)); session.getEntities().stream().limit(5).forEach(entity -> { ctx.getSource().sendFeedback(Text.literal(" ▸ " + entity.getName() + " [" + entity.getType() + "]") .formatted(Formatting.AQUA)); }); ctx.getSource().sendFeedback(Text.literal("\n📝 TITLES (" + session.getTitles().size() + "):") .formatted(Formatting.YELLOW, Formatting.BOLD)); session.getTitles().stream().limit(5).forEach(title -> { ctx.getSource().sendFeedback(Text.literal(" ▸ " + title.getTitle()) .formatted(Formatting.AQUA)); }); ctx.getSource().sendFeedback(Text.literal("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") .formatted(Formatting.AQUA)); return 1; } private static int exportSession(CommandContext ctx, String name) { CaptureManager manager = CaptureManager.getInstance(); CaptureSession session = manager.getSession(name); if (session == null) { ctx.getSource().sendFeedback(Text.literal("⚠ Session not found: " + name) .formatted(Formatting.RED)); return 0; } String exportData = manager.exportSession(session); ctx.getSource().sendFeedback(Text.literal("📤 Exported " + session.getName() + " to game directory") .formatted(Formatting.GREEN)); ctx.getSource().sendFeedback(Text.literal("📁 Location: .minecraft/replica/exports/" + session.getName() + ".json") .formatted(Formatting.GRAY)); return 1; } private static int deleteSession(CommandContext ctx, String name) { CaptureManager manager = CaptureManager.getInstance(); if (manager.deleteSession(name)) { ctx.getSource().sendFeedback(Text.literal("🗑 Deleted session: " + name) .formatted(Formatting.RED)); return 1; } else { ctx.getSource().sendFeedback(Text.literal("⚠ Session not found: " + name) .formatted(Formatting.YELLOW)); return 0; } } private static int clearAllSessions(CommandContext ctx) { CaptureManager manager = CaptureManager.getInstance(); int count = manager.getAllSessions().size(); manager.clearAllSessions(); ctx.getSource().sendFeedback(Text.literal("🗑 Cleared " + count + " session(s)") .formatted(Formatting.RED)); return 1; } private static int showHelp(CommandContext ctx) { ctx.getSource().sendFeedback(Text.literal("━━━━━━━━━━ REPLICA COMMANDS ━━━━━━━━━━") .formatted(Formatting.AQUA, Formatting.BOLD)); ctx.getSource().sendFeedback(Text.literal("\n🎯 CAPTURE MODES:").formatted(Formatting.GOLD, Formatting.BOLD)); ctx.getSource().sendFeedback(Text.literal("/replica lobby ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture lobby visuals").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica scoreboard ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture scoreboard only").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica tab ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture tab list").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica entities ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture entities & NPCs").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica holograms ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture holograms").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica titles ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture titles/subtitles").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica build ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture build data").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica ui ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture UI elements").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica full ").formatted(Formatting.AQUA) .append(Text.literal(" - Capture everything").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("\n⚙️ CONTROLS:").formatted(Formatting.GOLD, Formatting.BOLD)); ctx.getSource().sendFeedback(Text.literal("/replica stop").formatted(Formatting.AQUA) .append(Text.literal(" - Stop current capture").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica pause").formatted(Formatting.AQUA) .append(Text.literal(" - Pause capture").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica resume").formatted(Formatting.AQUA) .append(Text.literal(" - Resume capture").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica status").formatted(Formatting.AQUA) .append(Text.literal(" - Show capture status").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("\n📁 MANAGEMENT:").formatted(Formatting.GOLD, Formatting.BOLD)); ctx.getSource().sendFeedback(Text.literal("/replica list [type]").formatted(Formatting.AQUA) .append(Text.literal(" - List saved sessions").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica view ").formatted(Formatting.AQUA) .append(Text.literal(" - View session details").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica export ").formatted(Formatting.AQUA) .append(Text.literal(" - Export to JSON").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica delete ").formatted(Formatting.AQUA) .append(Text.literal(" - Delete session").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("/replica clear").formatted(Formatting.AQUA) .append(Text.literal(" - Clear all sessions").formatted(Formatting.GRAY))); ctx.getSource().sendFeedback(Text.literal("\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") .formatted(Formatting.AQUA)); ctx.getSource().sendFeedback(Text.literal("By BetterRs & YashDev | Mine Studio") .formatted(Formatting.GRAY, Formatting.ITALIC)); return 1; } }