From 560ca4e429b0ad18a4e7fbb57fda2b99c2b4095f Mon Sep 17 00:00:00 2001
From: trafficlunar <trafficlunar@proton.me>
Date: Fri, 20 Dec 2024 22:02:27 +0000
Subject: [PATCH] feat: exporting to .litematic

this took like 2 days

todo:
- fix other blocks
- block properties
---
 src/components/dialogs/SaveLitematic.tsx | 149 +++++++++++++++++++++++
 src/components/menubar/FileMenu.tsx      |   2 +-
 2 files changed, 150 insertions(+), 1 deletion(-)
 create mode 100644 src/components/dialogs/SaveLitematic.tsx

diff --git a/src/components/dialogs/SaveLitematic.tsx b/src/components/dialogs/SaveLitematic.tsx
new file mode 100644
index 0000000..75dcac4
--- /dev/null
+++ b/src/components/dialogs/SaveLitematic.tsx
@@ -0,0 +1,149 @@
+import { useContext, useState } from "react";
+import * as nbt from "nbtify";
+
+import { CanvasContext } from "@/context/Canvas";
+
+import { DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogTitle } from "@/components/ui/dialog";
+import { Button } from "@/components/ui/button";
+import { Input } from "@/components/ui/input";
+
+function SaveLitematic({ close }: DialogProps) {
+	const { canvasSize, blocks } = useContext(CanvasContext);
+
+	const [fileName, setFileName] = useState("blockmatic");
+
+	const onSubmit = async () => {
+		// todo: check if file name input is empty/valid
+		const width = canvasSize.maxX - canvasSize.minX;
+		const height = canvasSize.maxY - canvasSize.minY;
+
+		// Fill in empty blocks with "air"
+		const filledBlocks: Block[] = [];
+
+		for (let x = canvasSize.minX; x <= canvasSize.maxX - 1; x++) {
+			for (let y = canvasSize.minY; y <= canvasSize.maxY - 1; y++) {
+				const existingBlock = blocks.find((block) => block.x === x && block.y === y);
+				if (existingBlock) {
+					filledBlocks.push(existingBlock);
+				} else {
+					filledBlocks.push({ name: "air", x, y });
+				}
+			}
+		}
+
+		// Generate the block pallete
+		const blockStatePallete = [
+			{ Name: "minecraft:air" },
+			...Array.from(new Set(blocks.map((block) => `minecraft:${block.name}`))).map((name) => ({ Name: name })),
+		];
+
+		// Get the block states
+		const requiredBits = Math.max(Math.ceil(Math.log2(blockStatePallete.length)), 2);
+		const blockStates = new BigInt64Array(Math.ceil((filledBlocks.length * requiredBits) / 64));
+
+		filledBlocks.forEach((block) => {
+			const blockId = blockStatePallete.findIndex((entry) => entry.Name === `minecraft:${block.name}`);
+
+			const reversedY = height - 1 - block.y;
+			const index = reversedY * width + block.x;
+			const startOffset = index * requiredBits;
+			const startArrayIndex = Math.floor(startOffset / 64);
+			const endArrayIndex = ((index + 1) * requiredBits - 1) >> 6;
+			const bitOffset = startOffset % 64;
+			const mask = (1 << requiredBits) - 1;
+
+			blockStates[startArrayIndex] =
+				(blockStates[startArrayIndex] & ~(BigInt(1) << BigInt(bitOffset))) | (BigInt(blockId & mask) << BigInt(bitOffset));
+
+			if (startArrayIndex !== endArrayIndex) {
+				const endOffset = 64 - bitOffset;
+				const j1 = requiredBits - endOffset;
+
+				blockStates[endArrayIndex] = ((blockStates[endArrayIndex] >> BigInt(j1)) << BigInt(j1)) | (BigInt(blockId & mask) >> BigInt(endOffset));
+			}
+		});
+
+		// Generate NBT data
+		const data = {
+			MinecraftDataVersion: new nbt.Int32(3955),
+			Version: new nbt.Int32(7),
+			SubVersion: new nbt.Int32(1),
+			Metadata: {
+				Name: fileName,
+				Author: "blockmatic",
+				Description: "Created with blockmatic",
+				TimeCreated: new Date().getTime(),
+				TimeModified: new Date().getTime(),
+				EnclosingSize: {
+					x: new nbt.Int32(width),
+					y: new nbt.Int32(height),
+					z: new nbt.Int32(1),
+				},
+				RegionCount: new nbt.Int32(1),
+				TotalBlocks: new nbt.Int32(blocks.length),
+				TotalVolume: new nbt.Int32(width * height),
+			},
+			Regions: {
+				Image: {
+					Position: {
+						x: new nbt.Int32(0),
+						y: new nbt.Int32(0),
+						z: new nbt.Int32(0),
+					},
+					Size: {
+						x: new nbt.Int32(width),
+						y: new nbt.Int32(height),
+						z: new nbt.Int32(1),
+					},
+					BlockStates: blockStates,
+					BlockStatePalette: blockStatePallete,
+					TileEntities: [],
+					Entities: [],
+					PendingBlockTicks: [],
+					PendingFluidTicks: [],
+				},
+			},
+		};
+
+		// Write to file
+		const bytes = await nbt.write(data);
+		const compressed = await nbt.compress(bytes, "gzip");
+
+		const blob = new Blob([compressed], { type: "application/x-gzip" });
+		const url = URL.createObjectURL(blob);
+
+		const a = document.createElement("a");
+		a.href = url;
+		a.download = `${fileName}.litematic`;
+
+		document.body.appendChild(a);
+		a.click();
+		document.body.removeChild(a);
+
+		URL.revokeObjectURL(url);
+
+		close();
+	};
+
+	return (
+		<DialogContent>
+			<DialogHeader>
+				<DialogTitle>Save as .litematic</DialogTitle>
+				<DialogDescription>Save your image as a litematic</DialogDescription>
+			</DialogHeader>
+
+			<div className="flex items-center gap-2">
+				<Input value={fileName} onChange={(e) => setFileName(e.target.value)} />
+				<span>.litematic</span>
+			</div>
+
+			<DialogFooter>
+				<Button type="submit" onClick={onSubmit}>
+					Download
+				</Button>
+			</DialogFooter>
+		</DialogContent>
+	);
+}
+
+export default SaveLitematic;
diff --git a/src/components/menubar/FileMenu.tsx b/src/components/menubar/FileMenu.tsx
index a90fd00..b1d1422 100644
--- a/src/components/menubar/FileMenu.tsx
+++ b/src/components/menubar/FileMenu.tsx
@@ -28,7 +28,7 @@ function FileMenu() {
 					<MenubarSubTrigger>Export to...</MenubarSubTrigger>
 					<MenubarSubContent>
 						<MenubarItem>.schematic</MenubarItem>
-						<MenubarItem>.litematic</MenubarItem>
+						<MenubarItem onClick={() => openDialog("SaveLitematic")}>.litematic</MenubarItem>
 						<MenubarItem>image</MenubarItem>
 					</MenubarSubContent>
 				</MenubarSub>