PrefabPage.java

package com.hypixel.hytale.builtin.buildertools.prefablist;

import com.hypixel.hytale.builtin.buildertools.BuilderToolsPlugin;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.protocol.GameMode;
import com.hypixel.hytale.protocol.packets.interface_.CustomPageLifetime;
import com.hypixel.hytale.protocol.packets.interface_.CustomUIEventBindingType;
import com.hypixel.hytale.protocol.packets.interface_.Page;
import com.hypixel.hytale.protocol.packets.inventory.SetActiveSlot;
import com.hypixel.hytale.server.core.entity.entities.Player;
import com.hypixel.hytale.server.core.entity.entities.player.pages.InteractiveCustomUIPage;
import com.hypixel.hytale.server.core.inventory.Inventory;
import com.hypixel.hytale.server.core.inventory.ItemStack;
import com.hypixel.hytale.server.core.inventory.container.ItemContainer;
import com.hypixel.hytale.server.core.prefab.PrefabStore;
import com.hypixel.hytale.server.core.prefab.selection.standard.BlockSelection;
import com.hypixel.hytale.server.core.ui.Value;
import com.hypixel.hytale.server.core.ui.browser.FileBrowserConfig;
import com.hypixel.hytale.server.core.ui.browser.FileBrowserEventData;
import com.hypixel.hytale.server.core.ui.browser.FileListProvider;
import com.hypixel.hytale.server.core.ui.browser.ServerFileBrowser;
import com.hypixel.hytale.server.core.ui.builder.EventData;
import com.hypixel.hytale.server.core.ui.builder.UICommandBuilder;
import com.hypixel.hytale.server.core.ui.builder.UIEventBuilder;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import java.nio.file.Files;
import java.nio.file.LinkOption;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;
import javax.annotation.Nonnull;

public class PrefabPage extends InteractiveCustomUIPage<FileBrowserEventData> {
   private static final String PASTE_TOOL_ID = "EditorTool_Paste";
   private static final String ASSETS_ROOT_KEY = "Assets";
   @Nonnull
   private final ServerFileBrowser browser;
   @Nonnull
   private final BuilderToolsPlugin.BuilderState builderState;
   @Nonnull
   private final AssetPrefabFileProvider assetProvider;
   private boolean inAssetsRoot = true;
   @Nonnull
   private Path assetsCurrentDir = Paths.get("");

   public PrefabPage(@Nonnull PlayerRef playerRef, Path defaultRoot, @Nonnull BuilderToolsPlugin.BuilderState builderState) {
      super(playerRef, CustomPageLifetime.CanDismiss, FileBrowserEventData.CODEC);
      this.builderState = builderState;
      this.assetProvider = new AssetPrefabFileProvider();
      PrefabStore prefabStore = PrefabStore.get();
      List<FileBrowserConfig.RootEntry> roots = buildRootEntries(prefabStore);
      FileBrowserConfig config = FileBrowserConfig.builder().listElementId("#FileList").rootSelectorId("#RootSelector").searchInputId("#SearchInput").roots(roots).allowedExtensions(".prefab.json").enableRootSelector(true).enableSearch(true).enableDirectoryNav(true).maxResults(50).build();
      String savedSearchQuery = builderState.getPrefabListSearchQuery();
      Path initialRoot = ((FileBrowserConfig.RootEntry)roots.get(0)).path();
      this.browser = new ServerFileBrowser(config, initialRoot, (Path)null);
      if (savedSearchQuery != null && !savedSearchQuery.isEmpty()) {
         this.browser.setSearchQuery(savedSearchQuery);
      }

   }

   @Nonnull
   private static List<FileBrowserConfig.RootEntry> buildRootEntries(@Nonnull PrefabStore prefabStore) {
      List<FileBrowserConfig.RootEntry> roots = new ObjectArrayList<FileBrowserConfig.RootEntry>();
      roots.add(new FileBrowserConfig.RootEntry("Assets", Paths.get("Assets")));
      roots.add(new FileBrowserConfig.RootEntry("Server", prefabStore.getServerPrefabsPath()));
      return roots;
   }

   public void build(@Nonnull Ref<EntityStore> ref, @Nonnull UICommandBuilder commandBuilder, @Nonnull UIEventBuilder eventBuilder, @Nonnull Store<EntityStore> store) {
      commandBuilder.append("Pages/PrefabListPage.ui");
      this.browser.buildRootSelector(commandBuilder, eventBuilder);
      this.browser.buildSearchInput(commandBuilder, eventBuilder);
      this.buildCurrentPath(commandBuilder);
      this.buildFileList(commandBuilder, eventBuilder);
   }

   public void handleDataEvent(@Nonnull Ref<EntityStore> ref, @Nonnull Store<EntityStore> store, @Nonnull FileBrowserEventData data) {
      if (data.getRoot() != null) {
         this.inAssetsRoot = "Assets".equals(data.getRoot());
         this.assetsCurrentDir = Paths.get("");
         this.browser.handleEvent(data);
         UICommandBuilder commandBuilder = new UICommandBuilder();
         UIEventBuilder eventBuilder = new UIEventBuilder();
         this.buildCurrentPath(commandBuilder);
         this.buildFileList(commandBuilder, eventBuilder);
         this.sendUpdate(commandBuilder, eventBuilder, false);
      } else if (data.getSearchQuery() != null) {
         this.browser.handleEvent(data);
         UICommandBuilder commandBuilder = new UICommandBuilder();
         UIEventBuilder eventBuilder = new UIEventBuilder();
         this.buildCurrentPath(commandBuilder);
         this.buildFileList(commandBuilder, eventBuilder);
         this.sendUpdate(commandBuilder, eventBuilder, false);
      } else {
         String selectedPath = data.getSearchResult() != null ? data.getSearchResult() : data.getFile();
         if (selectedPath != null) {
            if (this.inAssetsRoot) {
               this.handleAssetsNavigation(ref, store, selectedPath, data.getSearchResult() != null);
            } else {
               this.handleRegularNavigation(ref, store, selectedPath, data.getSearchResult() != null);
            }
         }

      }
   }

   private void handleAssetsNavigation(@Nonnull Ref<EntityStore> ref, @Nonnull Store<EntityStore> store, @Nonnull String selectedPath, boolean isSearchResult) {
      if ("..".equals(selectedPath)) {
         if (!this.assetsCurrentDir.toString().isEmpty()) {
            Path parent = this.assetsCurrentDir.getParent();
            this.assetsCurrentDir = parent != null ? parent : Paths.get("");
            UICommandBuilder commandBuilder = new UICommandBuilder();
            UIEventBuilder eventBuilder = new UIEventBuilder();
            this.buildCurrentPath(commandBuilder);
            this.buildFileList(commandBuilder, eventBuilder);
            this.sendUpdate(commandBuilder, eventBuilder, false);
         }

      } else {
         String targetVirtualPath;
         if (isSearchResult) {
            targetVirtualPath = selectedPath;
         } else {
            targetVirtualPath = this.assetsCurrentDir.toString().isEmpty() ? selectedPath : this.assetsCurrentDir.toString().replace('\\', '/') + "/" + selectedPath;
         }

         Path resolvedPath = this.assetProvider.resolveVirtualPath(targetVirtualPath);
         if (resolvedPath == null) {
            this.sendUpdate();
         } else {
            if (Files.isDirectory(resolvedPath, new LinkOption[0])) {
               this.assetsCurrentDir = Paths.get(targetVirtualPath);
               UICommandBuilder commandBuilder = new UICommandBuilder();
               UIEventBuilder eventBuilder = new UIEventBuilder();
               this.buildCurrentPath(commandBuilder);
               this.buildFileList(commandBuilder, eventBuilder);
               this.sendUpdate(commandBuilder, eventBuilder, false);
            } else {
               this.handlePrefabSelection(ref, store, resolvedPath, targetVirtualPath);
            }

         }
      }
   }

   private void handleRegularNavigation(@Nonnull Ref<EntityStore> ref, @Nonnull Store<EntityStore> store, @Nonnull String selectedPath, boolean isSearchResult) {
      if (this.browser.handleEvent(FileBrowserEventData.file(selectedPath))) {
         UICommandBuilder commandBuilder = new UICommandBuilder();
         UIEventBuilder eventBuilder = new UIEventBuilder();
         this.buildCurrentPath(commandBuilder);
         this.buildFileList(commandBuilder, eventBuilder);
         this.sendUpdate(commandBuilder, eventBuilder, false);
      } else {
         Path file;
         if (isSearchResult) {
            file = this.browser.resolveSecure(selectedPath);
         } else {
            file = this.browser.resolveFromCurrent(selectedPath);
         }

         if (file != null && !Files.isDirectory(file, new LinkOption[0])) {
            this.handlePrefabSelection(ref, store, file, selectedPath);
         } else {
            this.sendUpdate();
         }
      }
   }

   private void handlePrefabSelection(@Nonnull Ref<EntityStore> ref, @Nonnull Store<EntityStore> store, @Nonnull Path file, @Nonnull String displayPath) {
      Player playerComponent = (Player)store.getComponent(ref, Player.getComponentType());

      assert playerComponent != null;

      if (playerComponent.getGameMode() != GameMode.Creative) {
         playerComponent.getPageManager().setPage(ref, store, Page.None);
      } else {
         PlayerRef playerRefComponent = (PlayerRef)store.getComponent(ref, PlayerRef.getComponentType());

         assert playerRefComponent != null;

         playerComponent.getPageManager().setPage(ref, store, Page.None);
         BlockSelection prefab = PrefabStore.get().getPrefab(file);
         BuilderToolsPlugin.addToQueue(playerComponent, playerRefComponent, (r, s, componentAccessor) -> s.load(displayPath, prefab, componentAccessor));
         this.switchToPasteTool(playerComponent, playerRefComponent);
      }
   }

   private void buildCurrentPath(@Nonnull UICommandBuilder commandBuilder) {
      String displayPath;
      if (this.inAssetsRoot) {
         String currentDirStr = this.assetsCurrentDir.toString().replace('\\', '/');
         displayPath = currentDirStr.isEmpty() ? "Assets" : "Assets/" + currentDirStr;
      } else {
         Path root = this.browser.getRoot();
         String rootDisplay = this.getRootDisplayName(root);
         String currentPath = this.browser.getCurrentDir().toString().replace('\\', '/');
         displayPath = currentPath.isEmpty() ? rootDisplay : rootDisplay + "/" + currentPath;
      }

      commandBuilder.set("#CurrentPath.Text", displayPath);
   }

   @Nonnull
   private String getRootDisplayName(@Nonnull Path root) {
      PrefabStore prefabStore = PrefabStore.get();
      String rootStr = root.toString();
      if ("Assets".equals(rootStr)) {
         return "Assets";
      } else if (rootStr.equals(prefabStore.getServerPrefabsPath().toString())) {
         return "Server";
      } else if (rootStr.equals(prefabStore.getWorldGenPrefabsPath().toString())) {
         return "WorldGen";
      } else {
         return root.getFileName() != null ? root.getFileName().toString() : rootStr;
      }
   }

   private void buildFileList(@Nonnull UICommandBuilder commandBuilder, @Nonnull UIEventBuilder eventBuilder) {
      if (this.inAssetsRoot) {
         this.buildAssetsFileList(commandBuilder, eventBuilder);
      } else {
         this.browser.buildFileList(commandBuilder, eventBuilder);
      }

   }

   private void buildAssetsFileList(@Nonnull UICommandBuilder commandBuilder, @Nonnull UIEventBuilder eventBuilder) {
      commandBuilder.clear("#FileList");
      List<FileListProvider.FileEntry> entries = this.assetProvider.getFiles(this.assetsCurrentDir, this.browser.getSearchQuery());
      int buttonIndex = 0;
      if (!this.assetsCurrentDir.toString().isEmpty() && this.browser.getSearchQuery().isEmpty()) {
         commandBuilder.append("#FileList", "Pages/BasicTextButton.ui");
         commandBuilder.set("#FileList[0].Text", "../");
         eventBuilder.addEventBinding(CustomUIEventBindingType.Activating, "#FileList[0]", EventData.of("File", ".."));
         ++buttonIndex;
      }

      for(FileListProvider.FileEntry entry : entries) {
         String displayText = entry.isDirectory() ? entry.displayName() + "/" : entry.displayName();
         commandBuilder.append("#FileList", "Pages/BasicTextButton.ui");
         commandBuilder.set("#FileList[" + buttonIndex + "].Text", displayText);
         if (!entry.isDirectory()) {
            commandBuilder.set("#FileList[" + buttonIndex + "].Style", Value.ref("Pages/BasicTextButton.ui", "SelectedLabelStyle"));
         }

         String eventKey = !this.browser.getSearchQuery().isEmpty() && !entry.isDirectory() ? "SearchResult" : "File";
         eventBuilder.addEventBinding(CustomUIEventBindingType.Activating, "#FileList[" + buttonIndex + "]", EventData.of(eventKey, entry.name()));
         ++buttonIndex;
      }

   }

   private void switchToPasteTool(@Nonnull Player playerComponent, @Nonnull PlayerRef playerRef) {
      Inventory inventory = playerComponent.getInventory();
      ItemContainer hotbar = inventory.getHotbar();
      ItemContainer storage = inventory.getStorage();
      ItemContainer tools = inventory.getTools();
      int hotbarSize = hotbar.getCapacity();

      for(short slot = 0; slot < hotbarSize; ++slot) {
         ItemStack itemStack = hotbar.getItemStack(slot);
         if (itemStack != null && !itemStack.isEmpty() && "EditorTool_Paste".equals(itemStack.getItemId())) {
            inventory.setActiveHotbarSlot((byte)slot);
            playerRef.getPacketHandler().writeNoCache(new SetActiveSlot(-1, (byte)slot));
            return;
         }
      }

      short emptySlot = -1;

      for(short slot = 0; slot < hotbarSize; ++slot) {
         ItemStack itemStack = hotbar.getItemStack(slot);
         if (itemStack == null || itemStack.isEmpty()) {
            emptySlot = slot;
            break;
         }
      }

      if (emptySlot != -1) {
         for(short slot = 0; slot < storage.getCapacity(); ++slot) {
            ItemStack itemStack = storage.getItemStack(slot);
            if (itemStack != null && !itemStack.isEmpty() && "EditorTool_Paste".equals(itemStack.getItemId())) {
               storage.moveItemStackFromSlotToSlot(slot, 1, hotbar, emptySlot);
               inventory.setActiveHotbarSlot((byte)emptySlot);
               playerRef.getPacketHandler().writeNoCache(new SetActiveSlot(-1, (byte)emptySlot));
               return;
            }
         }

         ItemStack pasteToolStack = null;

         for(short slot = 0; slot < tools.getCapacity(); ++slot) {
            ItemStack itemStack = tools.getItemStack(slot);
            if (itemStack != null && !itemStack.isEmpty() && "EditorTool_Paste".equals(itemStack.getItemId())) {
               pasteToolStack = itemStack;
               break;
            }
         }

         if (pasteToolStack != null) {
            hotbar.setItemStackForSlot(emptySlot, new ItemStack(pasteToolStack.getItemId()));
            inventory.setActiveHotbarSlot((byte)emptySlot);
            playerRef.getPacketHandler().writeNoCache(new SetActiveSlot(-1, (byte)emptySlot));
         }
      }
   }
}