From 8ee07bbf0ae8c871f0f63e2ec090b80d03678d75 Mon Sep 17 00:00:00 2001
From: Lokilife <lokilife0002@gmail.com>
Date: Fri, 30 Aug 2024 14:05:57 +0300
Subject: [PATCH] Add createmap command

---
 .../_Exodus/Mapping/CreateMapCommand.cs       | 126 ++++++++++++++++++
 1 file changed, 126 insertions(+)
 create mode 100644 Content.Server/_Exodus/Mapping/CreateMapCommand.cs

diff --git a/Content.Server/_Exodus/Mapping/CreateMapCommand.cs b/Content.Server/_Exodus/Mapping/CreateMapCommand.cs
new file mode 100644
index 0000000000..a66b8f845a
--- /dev/null
+++ b/Content.Server/_Exodus/Mapping/CreateMapCommand.cs
@@ -0,0 +1,126 @@
+using System.Linq;
+using Content.Server.Administration;
+using Content.Server.GameTicking;
+using Content.Shared.Administration;
+using Content.Shared.CCVar;
+using Robust.Server.GameObjects;
+using Robust.Server.Maps;
+using Robust.Shared.Configuration;
+using Robust.Shared.Console;
+using Robust.Shared.ContentPack;
+using Robust.Shared.Map;
+
+namespace Content.Server.Exodus.Mapping
+{
+    [AdminCommand(AdminFlags.Server | AdminFlags.Mapping)]
+    sealed class CreteMapCommand : IConsoleCommand
+    {
+        [Dependency] private readonly IEntityManager _entities = default!;
+        [Dependency] private readonly IMapManager _map = default!;
+        [Dependency] private readonly IConfigurationManager _cfg = default!;
+
+        public string Command => "createmap";
+        public string Description => Loc.GetString("cmd-mapping-desc");
+        public string Help => Loc.GetString("cmd-mapping-help");
+
+        public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
+        {
+            switch (args.Length)
+            {
+                case 1:
+                    return CompletionResult.FromHint(Loc.GetString("cmd-hint-mapping-id"));
+                case 2:
+                    var res = IoCManager.Resolve<IResourceManager>();
+                    var opts = CompletionHelper.UserFilePath(args[1], res.UserData)
+                        .Concat(CompletionHelper.ContentFilePath(args[1], res));
+                    return CompletionResult.FromHintOptions(opts, Loc.GetString("cmd-hint-mapping-path"));
+            }
+            return CompletionResult.Empty;
+        }
+
+        public void Execute(IConsoleShell shell, string argStr, string[] args)
+        {
+            if (shell.Player is not { } player)
+            {
+                shell.WriteError(Loc.GetString("shell-cannot-run-command-from-server"));
+                return;
+            }
+
+            if (args.Length > 2)
+            {
+                shell.WriteLine(Help);
+                return;
+            }
+
+#if DEBUG
+            shell.WriteLine(Loc.GetString("cmd-mapping-warning"));
+#endif
+
+            MapId mapId;
+            string? toLoad = null;
+            var mapSys = _entities.System<SharedMapSystem>();
+
+            // Get the map ID to use
+            if (args.Length is 1 or 2)
+            {
+                if (!int.TryParse(args[0], out var intMapId))
+                {
+                    shell.WriteError(Loc.GetString("cmd-mapping-failure-integer", ("arg", args[0])));
+                    return;
+                }
+
+                mapId = new MapId(intMapId);
+
+                // no loading null space
+                if (mapId == MapId.Nullspace)
+                {
+                    shell.WriteError(Loc.GetString("cmd-mapping-nullspace"));
+                    return;
+                }
+
+                if (_map.MapExists(mapId))
+                {
+                    shell.WriteError(Loc.GetString("cmd-mapping-exists", ("mapId", mapId)));
+                    return;
+                }
+
+                // either load a map or create a new one.
+                if (args.Length <= 1)
+                {
+                    mapSys.CreateMap(mapId, runMapInit: false);
+                }
+                else
+                {
+                    var loadOptions = new MapLoadOptions { StoreMapUids = true };
+                    _entities.System<MapLoaderSystem>().TryLoad(mapId, args[1], out _, loadOptions);
+                }
+
+                // was the map actually created or did it fail somehow?
+                if (!_map.MapExists(mapId))
+                {
+                    shell.WriteError(Loc.GetString("cmd-mapping-error"));
+                    return;
+                }
+            }
+            else
+            {
+                mapSys.CreateMap(out mapId, runMapInit: false);
+            }
+
+            // map successfully created. run misc helpful mapping commands
+            if (player.AttachedEntity is { Valid: true } playerEntity &&
+                _entities.GetComponent<MetaDataComponent>(playerEntity).EntityPrototype?.ID != GameTicker.AdminObserverPrototypeName)
+            {
+                shell.ExecuteCommand("aghost");
+            }
+
+            shell.ExecuteCommand($"tp 0 0 {mapId}");
+            _map.SetMapPaused(mapId, true);
+
+            if (args.Length == 2)
+                shell.WriteLine(Loc.GetString("cmd-mapping-success-load", ("mapId", mapId), ("path", args[1])));
+            else
+                shell.WriteLine(Loc.GetString("cmd-mapping-success", ("mapId", mapId)));
+        }
+    }
+}