From b38b1d8b592e0cd0937660afd23cdd2a32cc33d2 Mon Sep 17 00:00:00 2001
From: Grant <grant@sc07.company>
Date: Tue, 4 Jun 2024 19:31:59 -0600
Subject: [PATCH] virginmap overlay (fixes #32)

---
 packages/client/src/board.scss                | 10 +++
 .../client/src/components/CanvasWrapper.tsx   |  2 +
 .../components/Overlay/OverlaySettings.tsx    | 22 +++++
 .../src/components/Overlay/VirginOverlay.tsx  | 81 +++++++++++++++++++
 .../components/Settings/SettingsSidebar.tsx   |  2 +
 packages/client/src/components/Template.scss  |  3 -
 packages/client/src/components/Template.tsx   |  1 +
 packages/client/src/contexts/AppContext.tsx   |  4 +
 packages/client/src/lib/canvas.ts             | 21 +++++
 packages/client/src/lib/keybinds.ts           |  5 ++
 packages/client/src/lib/template.ts           |  3 +-
 packages/lib/src/net.ts                       |  3 +
 12 files changed, 153 insertions(+), 4 deletions(-)
 create mode 100644 packages/client/src/components/Overlay/OverlaySettings.tsx
 create mode 100644 packages/client/src/components/Overlay/VirginOverlay.tsx

diff --git a/packages/client/src/board.scss b/packages/client/src/board.scss
index 1f17dd9..43c3cf8 100644
--- a/packages/client/src/board.scss
+++ b/packages/client/src/board.scss
@@ -23,3 +23,13 @@
   pointer-events: none;
   touch-action: none;
 }
+
+.board-overlay {
+  position: absolute;
+  top: 0;
+  left: 0;
+}
+
+.no-interact {
+  pointer-events: none;
+}
diff --git a/packages/client/src/components/CanvasWrapper.tsx b/packages/client/src/components/CanvasWrapper.tsx
index b7b890a..1421517 100644
--- a/packages/client/src/components/CanvasWrapper.tsx
+++ b/packages/client/src/components/CanvasWrapper.tsx
@@ -9,6 +9,7 @@ import { IPosition } from "@sc07-canvas/lib/src/net";
 import { Template } from "./Template";
 import { IRouterData, Router } from "../lib/router";
 import { KeybindManager } from "../lib/keybinds";
+import { VirginOverlay } from "./Overlay/VirginOverlay";
 
 export const CanvasWrapper = () => {
   const { config } = useAppContext();
@@ -17,6 +18,7 @@ export const CanvasWrapper = () => {
   return (
     <main>
       <PanZoomWrapper>
+        <VirginOverlay />
         {config && <Template />}
         <CanvasInner />
       </PanZoomWrapper>
diff --git a/packages/client/src/components/Overlay/OverlaySettings.tsx b/packages/client/src/components/Overlay/OverlaySettings.tsx
new file mode 100644
index 0000000..0cebb22
--- /dev/null
+++ b/packages/client/src/components/Overlay/OverlaySettings.tsx
@@ -0,0 +1,22 @@
+import { Switch } from "@nextui-org/react";
+import { useAppContext } from "../../contexts/AppContext";
+
+export const OverlaySettings = () => {
+  const { showVirginOverlay, setShowVirginOverlay } = useAppContext();
+
+  return (
+    <>
+      <header>
+        <h2>Overlays</h2>
+      </header>
+      <section>
+        <Switch
+          isSelected={showVirginOverlay}
+          onValueChange={setShowVirginOverlay}
+        >
+          Virgin Map Overlay
+        </Switch>
+      </section>
+    </>
+  );
+};
diff --git a/packages/client/src/components/Overlay/VirginOverlay.tsx b/packages/client/src/components/Overlay/VirginOverlay.tsx
new file mode 100644
index 0000000..d3eb35c
--- /dev/null
+++ b/packages/client/src/components/Overlay/VirginOverlay.tsx
@@ -0,0 +1,81 @@
+import { useEffect, useRef } from "react";
+import { useAppContext } from "../../contexts/AppContext";
+import { Canvas } from "../../lib/canvas";
+import { KeybindManager } from "../../lib/keybinds";
+
+export const VirginOverlay = () => {
+  const { config, showVirginOverlay, setShowVirginOverlay } = useAppContext();
+  const canvasRef = useRef<HTMLCanvasElement | null>(null);
+
+  useEffect(() => {
+    const handleKeybind = () => {
+      setShowVirginOverlay((v) => !v);
+    };
+
+    KeybindManager.on("TOGGLE_VIRGIN", handleKeybind);
+
+    return () => {
+      KeybindManager.off("TOGGLE_VIRGIN", handleKeybind);
+    };
+  }, [setShowVirginOverlay]);
+
+  useEffect(() => {
+    if (!config) {
+      console.warn("[VirginOverlay] config is not defined");
+      return;
+    }
+    if (!canvasRef.current) {
+      console.warn("[VirginOverlay] canvasRef is not defined");
+      return;
+    }
+
+    const [width, height] = config.canvas.size;
+
+    canvasRef.current.width = width;
+    canvasRef.current.height = height;
+  }, [config]);
+
+  useEffect(() => {
+    if (!canvasRef.current) {
+      console.warn("[VirginOverlay] canvasRef is not defined");
+      return;
+    }
+
+    const updateVirginmap = () => {
+      const ctx = canvasRef.current!.getContext("2d");
+      if (!ctx) {
+        console.warn("[VirginOverlay] canvas context cannot be aquired");
+        return;
+      }
+
+      ctx.clearRect(0, 0, canvasRef.current!.width, canvasRef.current!.height);
+
+      const pixels = Canvas.instance!.getAllPixels();
+      for (const pixel of pixels) {
+        if (pixel.color !== -1) continue;
+
+        ctx.fillStyle = "rgba(0,140,0,0.5)";
+        ctx.fillRect(pixel.x, pixel.y, 1, 1);
+      }
+    };
+
+    var updateInterval = setInterval(updateVirginmap, 1000);
+
+    return () => {
+      clearInterval(updateInterval);
+    };
+  }, [canvasRef]);
+
+  return (
+    <canvas
+      id="virgin-overlay"
+      className="board-overlay no-interact pixelate"
+      ref={(r) => (canvasRef.current = r)}
+      width="1000"
+      height="1000"
+      style={{
+        display: showVirginOverlay ? "block" : "none",
+      }}
+    />
+  );
+};
diff --git a/packages/client/src/components/Settings/SettingsSidebar.tsx b/packages/client/src/components/Settings/SettingsSidebar.tsx
index d2720ab..96546e2 100644
--- a/packages/client/src/components/Settings/SettingsSidebar.tsx
+++ b/packages/client/src/components/Settings/SettingsSidebar.tsx
@@ -4,6 +4,7 @@ import { TemplateSettings } from "./TemplateSettings";
 import { FontAwesomeIcon } from "@fortawesome/react-fontawesome";
 import { faXmark } from "@fortawesome/free-solid-svg-icons/faXmark";
 import { ChatSettings } from "./ChatSettings";
+import { OverlaySettings } from "../Overlay/OverlaySettings";
 
 export const SettingsSidebar = () => {
   const { settingsSidebar, setSettingsSidebar, setShowKeybinds } =
@@ -34,6 +35,7 @@ export const SettingsSidebar = () => {
       </section>
       <TemplateSettings />
       <ChatSettings />
+      <OverlaySettings />
     </div>
   );
 };
diff --git a/packages/client/src/components/Template.scss b/packages/client/src/components/Template.scss
index d9b0434..a056bd3 100644
--- a/packages/client/src/components/Template.scss
+++ b/packages/client/src/components/Template.scss
@@ -1,7 +1,4 @@
 #template {
-  position: absolute;
-  top: 0;
-  left: 0;
   width: 100px;
 
   > * {
diff --git a/packages/client/src/components/Template.tsx b/packages/client/src/components/Template.tsx
index 4223446..840ca6e 100644
--- a/packages/client/src/components/Template.tsx
+++ b/packages/client/src/components/Template.tsx
@@ -136,6 +136,7 @@ export const Template = () => {
   return (
     <div
       id="template"
+      className="board-overlay"
       ref={templateHolder}
       style={{
         top: y,
diff --git a/packages/client/src/contexts/AppContext.tsx b/packages/client/src/contexts/AppContext.tsx
index cb153a9..69aff8d 100644
--- a/packages/client/src/contexts/AppContext.tsx
+++ b/packages/client/src/contexts/AppContext.tsx
@@ -42,6 +42,8 @@ export const AppContext = ({ children }: PropsWithChildren) => {
   }>();
   const [showKeybinds, setShowKeybinds] = useState(false);
 
+  const [showVirginOverlay, setShowVirginOverlay] = useState(false);
+
   const [hasAdmin, setHasAdmin] = useState(false);
 
   useEffect(() => {
@@ -140,6 +142,8 @@ export const AppContext = ({ children }: PropsWithChildren) => {
         setPixelWhois,
         showKeybinds,
         setShowKeybinds,
+        showVirginOverlay,
+        setShowVirginOverlay,
       }}
     >
       {!config && (
diff --git a/packages/client/src/lib/canvas.ts b/packages/client/src/lib/canvas.ts
index 30bab06..a1ea2e4 100644
--- a/packages/client/src/lib/canvas.ts
+++ b/packages/client/src/lib/canvas.ts
@@ -99,6 +99,27 @@ export class Canvas extends EventEmitter<CanvasEvents> {
     return this.PanZoom;
   }
 
+  getAllPixels() {
+    let pixels: {
+      x: number;
+      y: number;
+      color: number;
+    }[] = [];
+
+    for (const [x_y, value] of Object.entries(this.pixels)) {
+      if (value.type === "pending") continue;
+
+      const [x, y] = x_y.split("_").map((v) => parseInt(v));
+      pixels.push({
+        x,
+        y,
+        color: value.color,
+      });
+    }
+
+    return pixels;
+  }
+
   /**
    * Get nearby pixels
    * @param x
diff --git a/packages/client/src/lib/keybinds.ts b/packages/client/src/lib/keybinds.ts
index 4857306..258f257 100644
--- a/packages/client/src/lib/keybinds.ts
+++ b/packages/client/src/lib/keybinds.ts
@@ -31,6 +31,11 @@ const KEYBINDS = enforceObjectType({
       alt: true,
     },
   ],
+  TOGGLE_VIRGIN: [
+    {
+      key: "KeyV",
+    },
+  ],
 });
 
 class KeybindManager_ extends EventEmitter<{
diff --git a/packages/client/src/lib/template.ts b/packages/client/src/lib/template.ts
index aae095f..deabade 100644
--- a/packages/client/src/lib/template.ts
+++ b/packages/client/src/lib/template.ts
@@ -50,6 +50,7 @@ export class Template extends EventEmitter<TemplateEvents> {
     this.$wrapper = templateHolder;
 
     this.$imageLoader = document.createElement("img");
+    this.$imageLoader.style.setProperty("display", "none");
     this.$imageLoader.setAttribute("crossorigin", "");
     this.$imageLoader.addEventListener("load", () => {
       console.log("imageLoader loaded image");
@@ -101,7 +102,7 @@ export class Template extends EventEmitter<TemplateEvents> {
 
     switch (key) {
       case "enable":
-        this.setElementVisible([this.$canvas, this.$imageLoader], !!value);
+        this.setElementVisible([this.$canvas], !!value);
         break;
     }
 
diff --git a/packages/lib/src/net.ts b/packages/lib/src/net.ts
index 26fb300..f11c1c9 100644
--- a/packages/lib/src/net.ts
+++ b/packages/lib/src/net.ts
@@ -53,6 +53,9 @@ export interface IAppContext {
   showKeybinds: boolean;
   setShowKeybinds: (v: boolean) => void;
 
+  showVirginOverlay: boolean;
+  setShowVirginOverlay: React.Dispatch<React.SetStateAction<boolean>>;
+
   hasAdmin: boolean;
 }