Fix notebook inventory clicks and ALL button

gradle.properties

@@ -35,7 +35,7 @@ mod_name=Rivencraft
 # The license of the mod. Review your options at https://choosealicense.com/. All Rights Reserved is the default.
 mod_license=LGPL-3.0-or-later
 # The mod version. Rivencraft uses the x.y.z.b policy documented in README.md.
-mod_version=0.3.3.0
+mod_version=0.3.3.3
 immersive_portals_version=6.0.7
 # The group ID for the mod. It is only important when publishing as an artifact to a Maven repository.
 # This should match the base package used for the mod sources.

src/main/java/net/dnorberg/rivencraft/client/screen/NotebookScreen.java

@@ -1,104 +1,178 @@
 package net.dnorberg.rivencraft.client.screen;
 
-import net.dnorberg.rivencraft.Rivencraft;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Locale;
 import net.dnorberg.rivencraft.menu.NotebookMenu;
 import net.dnorberg.rivencraft.network.RenameNotebookPagePayload;
+import net.dnorberg.rivencraft.registry.ModItems;
 import net.minecraft.client.gui.GuiGraphics;
 import net.minecraft.client.gui.components.Button;
 import net.minecraft.client.gui.components.EditBox;
 import net.minecraft.client.gui.components.Tooltip;
 import net.minecraft.client.gui.screens.inventory.AbstractContainerScreen;
 import net.minecraft.network.chat.Component;
-import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.entity.player.Inventory;
 import net.minecraft.world.inventory.ClickType;
-import net.minecraft.world.inventory.Slot;
 import net.minecraft.world.item.ItemStack;
 import net.neoforged.neoforge.network.PacketDistributor;
 
+/** A page surface modelled on Legacy's folder GUI, backed by the Notebook menu. */
 public final class NotebookScreen extends AbstractContainerScreen<NotebookMenu> {
-    private static final ResourceLocation NOTEBOOK_BACKGROUND = ResourceLocation.fromNamespaceAndPath(
+    private static final int SURFACE_TOP = 20;
-            Rivencraft.MOD_ID, "textures/gui/notebook.png");
+    private static final int SURFACE_BOTTOM = 132;
-    private static final ResourceLocation INVENTORY_BACKGROUND = ResourceLocation.withDefaultNamespace(
+    private static final int SURFACE_RIGHT = 156;
-            "textures/gui/container/generic_54.png");
+    private static final int PAGE_WIDTH = 30;
-    private EditBox nameInput;
+    private static final int PAGE_HEIGHT = 40;
-    private Button previousButton;
+    private static final int PAGE_STEP_X = 31;
-    private Button nextButton;
+    private static final int PAGE_STEP_Y = 41;
+    private static final int PAGE_COLUMNS = 5;
+    private EditBox textInput;
+    private boolean renaming;
+    private boolean draggingScrollbar;
+    private boolean surfaceClickPendingRelease;
     private int selectedSlot = -1;
+    private int scroll;
 
     public NotebookScreen(NotebookMenu menu, Inventory inventory, Component title) {
         super(menu, inventory, title);
         imageWidth = 176;
-        imageHeight = 183;
+        imageHeight = 213;
-        inventoryLabelY = 89;
+        inventoryLabelY = 123;
-        titleLabelY = 7;
     }
 
     @Override
     protected void init() {
         super.init();
-        nameInput = new EditBox(font, leftPos + 47, topPos + 7, 86, 18,
+        Button sortButton = Button.builder(Component.literal("AZ"), button -> {})
-                Component.translatable("screen.rivencraft.notebook.rename"));
+                .bounds(leftPos, topPos, 18, 18).build();
-        nameInput.setMaxLength(32);
+        Button allButton = Button.builder(Component.literal("ALL"), button -> {})
-        nameInput.setResponder(ignored -> updateButtons());
+                .bounds(leftPos + 20, topPos, 26, 18).build();
-        addRenderableWidget(nameInput);
+        sortButton.active = false;
-
+        allButton.active = false;
-        previousButton = Button.builder(Component.literal("←"), ignored -> moveSelected(false))
+        addRenderableWidget(sortButton);
-                .bounds(leftPos + 136, topPos + 7, 16, 18)
+        addRenderableWidget(allButton);
-                .tooltip(Tooltip.create(Component.translatable("screen.rivencraft.notebook.move_previous")))
+        textInput = new EditBox(font, leftPos + 48, topPos, 86, 18,
-                .build();
+                Component.translatable("screen.rivencraft.notebook.search"));
-        nextButton = Button.builder(Component.literal("→"), ignored -> moveSelected(true))
+        textInput.setMaxLength(32);
-                .bounds(leftPos + 153, topPos + 7, 16, 18)
+        textInput.setHint(Component.translatable("screen.rivencraft.notebook.search"));
-                .tooltip(Tooltip.create(Component.translatable("screen.rivencraft.notebook.move_next")))
+        textInput.setResponder(value -> {
-                .build();
+            if (!renaming) scroll = 0;
-        addRenderableWidget(previousButton);
+        });
-        addRenderableWidget(nextButton);
+        addRenderableWidget(textInput);
-        updateButtons();
+        addRenderableWidget(Button.builder(Component.literal("R"), button -> toggleRename())
+                .bounds(leftPos + 136, topPos, 18, 18)
+                .tooltip(Tooltip.create(Component.translatable("screen.rivencraft.notebook.rename_toggle")))
+                .build());
     }
 
     @Override
     public void render(GuiGraphics graphics, int mouseX, int mouseY, float partialTick) {
         super.render(graphics, mouseX, mouseY, partialTick);
-        if (selectedSlot >= 0) {
+        List<Integer> pages = visiblePages();
-            Slot slot = menu.slots.get(selectedSlot);
+        scroll = Math.min(scroll, maxScroll(pages.size()));
-            graphics.renderOutline(leftPos + slot.x - 1, topPos + slot.y - 1, 18, 18, 0xFFE0C36A);
+        graphics.enableScissor(leftPos, topPos + SURFACE_TOP, leftPos + SURFACE_RIGHT,
+                topPos + SURFACE_BOTTOM);
+        for (int position = 0; position < pages.size(); position++) {
+            int x = leftPos + (position % PAGE_COLUMNS) * PAGE_STEP_X;
+            int y = topPos + SURFACE_TOP + (position / PAGE_COLUMNS) * PAGE_STEP_Y - scroll;
+            if (y + PAGE_HEIGHT > topPos + SURFACE_TOP && y < topPos + SURFACE_BOTTOM) {
+                drawPage(graphics, pages.get(position), x, y);
+            }
+        }
+        graphics.disableScissor();
+        drawScrollbar(graphics, pages.size());
+        int hovered = pageAt(mouseX, mouseY, pages);
+        if (!menu.getCarried().isEmpty() && inSurface(mouseX, mouseY)) {
+            graphics.renderItem(menu.getCarried(), mouseX - 8, mouseY - 8);
+        } else if (hovered >= 0) {
+            ItemStack page = menu.getSlot(hovered).getItem();
+            if (!page.isEmpty() && matchesSearch(page)) graphics.renderTooltip(font, page, mouseX, mouseY);
+        } else {
+            renderTooltip(graphics, mouseX, mouseY);
         }
-        renderTooltip(graphics, mouseX, mouseY);
     }
 
     @Override
     protected void renderBg(GuiGraphics graphics, float partialTick, int mouseX, int mouseY) {
-        graphics.blit(NOTEBOOK_BACKGROUND, leftPos, topPos, 0, 0, imageWidth, 88);
+        graphics.fill(leftPos, topPos + SURFACE_TOP, leftPos + SURFACE_RIGHT,
-        graphics.blit(INVENTORY_BACKGROUND, leftPos, topPos + 87, 0, 126, imageWidth, 96);
+                topPos + SURFACE_BOTTOM, 0xB0000000);
+        drawBeveledPanel(graphics, leftPos, topPos + 132, imageWidth, 81);
+        for (int row = 0; row < 3; row++) {
+            for (int column = 0; column < 9; column++) drawSlot(graphics, 7 + column * 18, 134 + row * 18);
+        }
+        for (int column = 0; column < 9; column++) drawSlot(graphics, 7 + column * 18, 192);
     }
 
     @Override
     protected void renderLabels(GuiGraphics graphics, int mouseX, int mouseY) {
-        graphics.drawString(font, title, titleLabelX, titleLabelY, 0x3F2A18, false);
         graphics.drawString(font, playerInventoryTitle, inventoryLabelX, inventoryLabelY, 0x404040, false);
     }
 
     @Override
-    protected void slotClicked(Slot slot, int slotId, int mouseButton, ClickType type) {
+    public boolean mouseClicked(double mouseX, double mouseY, int button) {
-        if (slotId >= 0 && slotId < NotebookMenu.NOTEBOOK_SLOTS) {
+        if (inScrollbar(mouseX, mouseY) && button == 0) {
-            selectedSlot = slotId;
+            draggingScrollbar = true;
+            setScrollFromMouse(mouseY, visiblePages().size());
+            return true;
         }
-        super.slotClicked(slot, slotId, mouseButton, type);
+        int index = pageAt(mouseX, mouseY, visiblePages());
-        refreshName();
+        if (index >= 0 && (button == 0 || button == 1)) {
+            ItemStack page = menu.getSlot(index).getItem();
+            if (!page.isEmpty() && !matchesSearch(page)) return true;
+            if (renaming) finishRename();
+            setFocused(null);
+            textInput.setFocused(false);
+            selectedSlot = index;
+            slotClicked(menu.getSlot(index), index, button, ClickType.PICKUP);
+            surfaceClickPendingRelease = true;
+            return true;
+        }
+        return super.mouseClicked(mouseX, mouseY, button);
+    }
+
+    @Override
+    public boolean mouseDragged(double mouseX, double mouseY, int button, double dragX, double dragY) {
+        if (draggingScrollbar && button == 0) {
+            setScrollFromMouse(mouseY, visiblePages().size());
+            return true;
+        }
+        return super.mouseDragged(mouseX, mouseY, button, dragX, dragY);
+    }
+
+    @Override
+    public boolean mouseReleased(double mouseX, double mouseY, int button) {
+        if (draggingScrollbar || surfaceClickPendingRelease) {
+            draggingScrollbar = false;
+            surfaceClickPendingRelease = false;
+            return true;
+        }
+        draggingScrollbar = false;
+        return super.mouseReleased(mouseX, mouseY, button);
+    }
+
+    @Override
+    public boolean mouseScrolled(double mouseX, double mouseY, double scrollX, double scrollY) {
+        if (inSurface(mouseX, mouseY) || inScrollbar(mouseX, mouseY)) {
+            scroll = Math.clamp(scroll - (int)Math.signum(scrollY) * 20,
+                    0, maxScroll(visiblePages().size()));
+            return true;
+        }
+        return super.mouseScrolled(mouseX, mouseY, scrollX, scrollY);
     }
 
     @Override
     public boolean keyPressed(int keyCode, int scanCode, int modifiers) {
-        if (nameInput.isFocused() && (keyCode == 257 || keyCode == 335)) {
+        if (renaming && textInput.isFocused() && (keyCode == 257 || keyCode == 335)) {
-            renameSelected();
+            finishRename();
             return true;
         }
-        if (!nameInput.isFocused() && keyCode == 263) {
+        if (!textInput.isFocused() && keyCode == 82 && selectedSlot >= 0) {
-            moveSelected(false);
+            toggleRename();
             return true;
         }
-        if (!nameInput.isFocused() && keyCode == 262) {
+        if (!textInput.isFocused() && selectedSlot >= 0 && (keyCode == 263 || keyCode == 262)) {
-            moveSelected(true);
+            moveSelected(keyCode == 262);
             return true;
         }
         return super.keyPressed(keyCode, scanCode, modifiers);
@@ -106,51 +180,149 @@ public final class NotebookScreen extends AbstractContainerScreen<NotebookMenu>
 
     @Override
     public void onClose() {
-        renameSelected();
+        commitRename();
         super.onClose();
     }
 
-    private void refreshName() {
+    private List<Integer> visiblePages() {
-        ItemStack selected = selectedPage();
+        List<Integer> pages = new ArrayList<>();
-        nameInput.setValue(selected.isEmpty() ? "" : selected.getHoverName().getString());
+        int firstEmpty = -1;
-        updateButtons();
+        for (int index = 0; index < NotebookMenu.NOTEBOOK_SLOTS; index++) {
+            ItemStack stack = menu.getSlot(index).getItem();
+            if (stack.isEmpty()) {
+                if (firstEmpty < 0) firstEmpty = index;
+            } else {
+                pages.add(index);
+            }
+        }
+        if (firstEmpty >= 0 && menu.getCarried().is(ModItems.PAGE.get())) {
+            pages.add(firstEmpty);
+        }
+        return pages;
     }
 
-    private void renameSelected() {
+    private boolean matchesSearch(ItemStack page) {
-        if (selectedSlot < 0 || selectedPage().isEmpty()) {
+        return renaming || textInput.getValue().isBlank()
-            return;
+                || page.getHoverName().getString().toLowerCase(Locale.ROOT)
+                        .contains(textInput.getValue().toLowerCase(Locale.ROOT));
+    }
+
+    private int pageAt(double mouseX, double mouseY, List<Integer> pages) {
+        if (!inSurface(mouseX, mouseY)) return -1;
+        int localX = (int)mouseX - leftPos;
+        int localY = (int)mouseY - topPos - SURFACE_TOP + scroll;
+        int column = localX / PAGE_STEP_X;
+        int position = (localY / PAGE_STEP_Y) * PAGE_COLUMNS + column;
+        return column < PAGE_COLUMNS && localX % PAGE_STEP_X < PAGE_WIDTH
+                && localY % PAGE_STEP_Y < PAGE_HEIGHT && position < pages.size()
+                ? pages.get(position) : -1;
+    }
+
+    private boolean inSurface(double mouseX, double mouseY) {
+        return mouseX >= leftPos && mouseX < leftPos + SURFACE_RIGHT
+                && mouseY >= topPos + SURFACE_TOP && mouseY < topPos + SURFACE_BOTTOM;
+    }
+
+    private boolean inScrollbar(double mouseX, double mouseY) {
+        return mouseX >= leftPos + SURFACE_RIGHT && mouseX < leftPos + imageWidth
+                && mouseY >= topPos + SURFACE_TOP && mouseY < topPos + SURFACE_BOTTOM;
+    }
+
+    private int maxScroll(int pageCount) {
+        int rows = (pageCount + PAGE_COLUMNS - 1) / PAGE_COLUMNS;
+        return Math.max(0, rows * PAGE_STEP_Y - (SURFACE_BOTTOM - SURFACE_TOP));
+    }
+
+    private void setScrollFromMouse(double mouseY, int pageCount) {
+        int maximum = maxScroll(pageCount);
+        double fraction = (mouseY - topPos - SURFACE_TOP - 10.0)
+                / (SURFACE_BOTTOM - SURFACE_TOP - 20.0);
+        scroll = Math.clamp((int)(Math.clamp(fraction, 0.0, 1.0) * maximum), 0, maximum);
+    }
+
+    private void drawScrollbar(GuiGraphics graphics, int pageCount) {
+        int x = leftPos + SURFACE_RIGHT;
+        int y = topPos + SURFACE_TOP;
+        int height = SURFACE_BOTTOM - SURFACE_TOP;
+        graphics.fill(x, y, x + 20, y + height, 0xFF858585);
+        graphics.renderOutline(x, y, 20, height, 0xFF262626);
+        for (int line = y + 5; line < y + height - 4; line += 5) {
+            graphics.fill(x + 4, line, x + 16, line + 1, 0xFFB7B7B7);
+        }
+        int maximum = maxScroll(pageCount);
+        int thumbY = y + 4 + (maximum == 0 ? 0 : scroll * (height - 23) / maximum);
+        graphics.fill(x + 4, thumbY, x + 16, thumbY + 15, 0xFFC9C9C9);
+        graphics.renderOutline(x + 4, thumbY, 12, 15, 0xFF414141);
+    }
+
+    private void drawPage(GuiGraphics graphics, int index, int x, int y) {
+        ItemStack page = menu.getSlot(index).getItem();
+        boolean visible = !page.isEmpty() && matchesSearch(page);
+        graphics.fill(x, y, x + PAGE_WIDTH, y + PAGE_HEIGHT,
+                visible ? 0xFFE5D4A7 : 0x704B4945);
+        graphics.fill(x + 2, y + 2, x + PAGE_WIDTH - 2, y + PAGE_HEIGHT - 2,
+                visible ? 0xFFF4E5C3 : 0x603B3934);
+        graphics.renderOutline(x, y, PAGE_WIDTH, PAGE_HEIGHT,
+                index == selectedSlot ? 0xFFFFD061 : 0xFF564A34);
+        if (visible) {
+            graphics.pose().pushPose();
+            graphics.pose().translate(x + 3, y + 8, 0);
+            graphics.pose().scale(1.5F, 1.5F, 1.0F);
+            graphics.renderItem(page, 0, 0);
+            graphics.pose().popPose();
+        }
+    }
+
+    private void toggleRename() {
+        if (renaming) {
+            finishRename();
+        } else if (selectedSlot >= 0 && !menu.getSlot(selectedSlot).getItem().isEmpty()) {
+            renaming = true;
+            textInput.setValue(menu.getSlot(selectedSlot).getItem().getHoverName().getString());
+            textInput.setHint(Component.translatable("screen.rivencraft.notebook.rename"));
+            setFocused(textInput);
+            textInput.setFocused(true);
+        }
+    }
+
+    private void finishRename() {
+        commitRename();
+        renaming = false;
+        textInput.setValue("");
+        textInput.setHint(Component.translatable("screen.rivencraft.notebook.search"));
+    }
+
+    private void commitRename() {
+        if (renaming && selectedSlot >= 0 && !menu.getSlot(selectedSlot).getItem().isEmpty()) {
+            PacketDistributor.sendToServer(new RenameNotebookPagePayload(
+                    menu.containerId, selectedSlot, textInput.getValue()));
         }
-        PacketDistributor.sendToServer(new RenameNotebookPagePayload(
-                menu.containerId, selectedSlot, nameInput.getValue()));
     }
 
     private void moveSelected(boolean next) {
         int target = selectedSlot + (next ? 1 : -1);
-        if (selectedSlot < 0 || selectedPage().isEmpty()
+        if (selectedSlot < 0 || target < 0 || target >= NotebookMenu.NOTEBOOK_SLOTS
-                || target < 0 || target >= NotebookMenu.NOTEBOOK_SLOTS) {
+                || menu.getSlot(selectedSlot).getItem().isEmpty()) return;
-            return;
-        }
-        renameSelected();
         int buttonId = (next ? NotebookMenu.MOVE_NEXT_BASE : NotebookMenu.MOVE_PREVIOUS_BASE) + selectedSlot;
         if (menu.clickMenuButton(minecraft.player, buttonId)) {
             minecraft.gameMode.handleInventoryButtonClick(menu.containerId, buttonId);
             selectedSlot = target;
-            refreshName();
         }
     }
 
-    private ItemStack selectedPage() {
+    private static void drawBeveledPanel(GuiGraphics graphics, int x, int y, int width, int height) {
-        return selectedSlot >= 0 && selectedSlot < NotebookMenu.NOTEBOOK_SLOTS
+        graphics.fill(x, y, x + width, y + height, 0xFFC6C6C6);
-                ? menu.getSlot(selectedSlot).getItem()
+        graphics.fill(x, y, x + width, y + 2, 0xFFFFFFFF);
-                : ItemStack.EMPTY;
+        graphics.fill(x, y, x + 2, y + height, 0xFFFFFFFF);
+        graphics.fill(x, y + height - 2, x + width, y + height, 0xFF555555);
+        graphics.fill(x + width - 2, y, x + width, y + height, 0xFF555555);
     }
 
-    private void updateButtons() {
+    private void drawSlot(GuiGraphics graphics, int x, int y) {
-        if (previousButton == null || nextButton == null) {
+        graphics.fill(leftPos + x, topPos + y, leftPos + x + 18, topPos + y + 18, 0xFF8B8B8B);
-            return;
+        graphics.fill(leftPos + x, topPos + y, leftPos + x + 18, topPos + y + 1, 0xFF373737);
-        }
+        graphics.fill(leftPos + x, topPos + y, leftPos + x + 1, topPos + y + 18, 0xFF373737);
-        boolean hasPage = !selectedPage().isEmpty();
+        graphics.fill(leftPos + x + 17, topPos + y, leftPos + x + 18, topPos + y + 18, 0xFFFFFFFF);
-        previousButton.active = hasPage && selectedSlot > 0;
+        graphics.fill(leftPos + x, topPos + y + 17, leftPos + x + 18, topPos + y + 18, 0xFFFFFFFF);
-        nextButton.active = hasPage && selectedSlot >= 0 && selectedSlot < NotebookMenu.NOTEBOOK_SLOTS - 1;
     }
 }

src/main/java/net/dnorberg/rivencraft/gametest/WritingDataGameTests.java

@@ -22,6 +22,7 @@ import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.item.Items;
 import net.minecraft.world.InteractionHand;
+import net.minecraft.world.inventory.ClickType;
 import net.neoforged.neoforge.gametest.GameTestHolder;
 import net.neoforged.neoforge.gametest.PrefixGameTestTemplate;
 
@@ -176,6 +177,29 @@ public final class WritingDataGameTests {
         helper.succeed();
     }
 
+    @GameTest(template = "empty")
+    public static void notebookMenuSurvivesInventoryStackReplacementAndClicks(GameTestHelper helper) {
+        var player = helper.makeMockServerPlayerInLevel();
+        player.setItemInHand(InteractionHand.MAIN_HAND, new ItemStack(ModItems.NOTEBOOK.get()));
+        NotebookMenu menu = new NotebookMenu(8, player.getInventory(), InteractionHand.MAIN_HAND);
+
+        // Slot synchronization may replace the ItemStack object while the Notebook remains held.
+        player.setItemInHand(InteractionHand.MAIN_HAND, player.getMainHandItem().copy());
+        helper.assertTrue(menu.stillValid(player),
+                "Replacing the held stack object must not close an open Notebook menu");
+        menu.getSlot(0).set(PageItem.createSymbolPage(reference("rivencraft:terrain/standard")));
+        helper.assertTrue(player.getMainHandItem().has(ModDataComponents.NOTEBOOK_CONTENTS.get()),
+                "Page edits must persist to the current held Notebook stack");
+
+        player.getInventory().setItem(9, new ItemStack(Items.DIRT));
+        menu.clicked(NotebookMenu.NOTEBOOK_SLOTS, 0, ClickType.PICKUP, player);
+        helper.assertTrue(menu.getCarried().is(Items.DIRT),
+                "Clicking a player inventory item must pick it up without closing the Notebook");
+        helper.assertTrue(menu.stillValid(player),
+                "Normal inventory clicks must leave the Notebook menu valid");
+        helper.succeed();
+    }
+
     private static SymbolReference reference(String id) {
         return new SymbolReference(ResourceLocation.parse(id));
     }

src/main/java/net/dnorberg/rivencraft/menu/NotebookMenu.java

@@ -26,7 +26,7 @@ public final class NotebookMenu extends AbstractContainerMenu {
     private static final int PLAYER_INVENTORY_START = NOTEBOOK_SLOTS;
     private final Inventory playerInventory;
     private final InteractionHand hand;
-    private final ItemStack notebook;
+    private final int openedHotbarSlot;
     private final NotebookContainer contents;
 
     public NotebookMenu(int containerId, Inventory inventory, RegistryFriendlyByteBuf data) {
@@ -37,34 +37,37 @@ public final class NotebookMenu extends AbstractContainerMenu {
         super(ModMenus.NOTEBOOK.get(), containerId);
         this.playerInventory = inventory;
         this.hand = hand;
-        this.notebook = inventory.player.getItemInHand(hand);
+        this.openedHotbarSlot = inventory.selected;
-        this.contents = new NotebookContainer(notebook);
+        this.contents = new NotebookContainer(inventory, hand);
 
         for (int row = 0; row < 3; row++) {
             for (int column = 0; column < 9; column++) {
-                addSlot(new PageSlot(contents, column + row * 9, 8 + column * 18, 32 + row * 18));
+                // The client draws these pages on a filtered, scrollable surface instead.
+                addSlot(new PageSlot(contents, column + row * 9, -1000, -1000));
             }
         }
         for (int row = 0; row < 3; row++) {
             for (int column = 0; column < 9; column++) {
                 int inventoryIndex = column + row * 9 + 9;
-                addSlot(playerSlot(inventoryIndex, 8 + column * 18, 101 + row * 18));
+                addSlot(playerSlot(inventoryIndex, 8 + column * 18, 135 + row * 18));
             }
         }
         for (int column = 0; column < 9; column++) {
-            addSlot(playerSlot(column, 8 + column * 18, 159));
+            addSlot(playerSlot(column, 8 + column * 18, 193));
         }
     }
 
     private Slot playerSlot(int inventoryIndex, int x, int y) {
-        boolean locked = hand == InteractionHand.MAIN_HAND && inventoryIndex == playerInventory.selected;
+        boolean locked = hand == InteractionHand.MAIN_HAND && inventoryIndex == openedHotbarSlot;
         return locked ? new LockedSlot(playerInventory, inventoryIndex, x, y)
                 : new Slot(playerInventory, inventoryIndex, x, y);
     }
 
     @Override
     public boolean stillValid(Player player) {
-        return player.isAlive() && player.getItemInHand(hand) == notebook && notebook.is(ModItems.NOTEBOOK.get());
+        return player.isAlive()
+                && (hand == InteractionHand.OFF_HAND || player.getInventory().selected == openedHotbarSlot)
+                && player.getItemInHand(hand).is(ModItems.NOTEBOOK.get());
     }
 
     @Override
@@ -134,10 +137,16 @@ public final class NotebookMenu extends AbstractContainerMenu {
 
     @Override
     public void clicked(int slotId, int button, ClickType clickType, Player player) {
-        super.clicked(slotId, button, clickType, player);
         if (!stillValid(player)) {
             player.closeContainer();
+            return;
         }
+        if (clickType == ClickType.SWAP
+                && (button == openedHotbarSlot && hand == InteractionHand.MAIN_HAND
+                        || button == 40 && hand == InteractionHand.OFF_HAND)) {
+            return;
+        }
+        super.clicked(slotId, button, clickType, player);
     }
 
     private static final class PageSlot extends Slot {
@@ -159,12 +168,15 @@ public final class NotebookMenu extends AbstractContainerMenu {
     }
 
     private static final class NotebookContainer extends SimpleContainer {
-        private final ItemStack notebook;
+        private final Inventory inventory;
+        private final InteractionHand hand;
         private boolean loading;
 
-        private NotebookContainer(ItemStack notebook) {
+        private NotebookContainer(Inventory inventory, InteractionHand hand) {
             super(NOTEBOOK_SLOTS);
-            this.notebook = notebook;
+            this.inventory = inventory;
+            this.hand = hand;
+            ItemStack notebook = inventory.player.getItemInHand(hand);
             loading = true;
             List<ItemStack> saved = notebook.getOrDefault(
                     ModDataComponents.NOTEBOOK_CONTENTS.get(), NotebookContents.EMPTY).pages();
@@ -186,7 +198,8 @@ public final class NotebookMenu extends AbstractContainerMenu {
         @Override
         public void setChanged() {
             super.setChanged();
-            if (loading || notebook.isEmpty()) {
+            ItemStack notebook = inventory.player.getItemInHand(hand);
+            if (loading || !notebook.is(ModItems.NOTEBOOK.get())) {
                 return;
             }
             NonNullList<ItemStack> saved = NonNullList.withSize(NOTEBOOK_SLOTS, ItemStack.EMPTY);

src/main/resources/assets/rivencraft/lang/en_us.json

@@ -16,6 +16,8 @@
   "item.rivencraft.notebook.pages": "Pages: %s/%s",
   "item.rivencraft.notebook.unresolved": "Unresolved pages: %s",
   "screen.rivencraft.notebook.rename": "Rename selected page",
+  "screen.rivencraft.notebook.search": "Search pages",
+  "screen.rivencraft.notebook.rename_toggle": "Rename selected page (R); use left/right arrows to reorder",
   "screen.rivencraft.notebook.move_previous": "Move selected page left",
   "screen.rivencraft.notebook.move_next": "Move selected page right",
   "message.rivencraft.linking_book.bind_hint": "Sneak-use to bind this book to your current location",