Web-Frontend/client/src/components/layout/Canvas/views/DashboardCanvas.jsx

import { HStack, Flex, Tabs } from "@chakra-ui/react";
import { useState } from "react";
import WidgetCanvas from "./DashboardCanvasNav/WidgetCanvas";
import { SearchBar } from "../components/SearchBar";
import { SettingsButton } from "../components/SettingsButton";
import { DailyActivity } from "./DashboardCanvasNav/DailyActivity";
import { Gas } from "./DashboardCanvasNav/Gas";
import { Seismic } from "./DashboardCanvasNav/Seismic";
import { RemoteSensing } from "./DashboardCanvasNav/RemoteSensing";
import { CancelButton } from "../components/CancelButton";
import { SaveButton } from "../components/SaveButton";
import DeleteAllButton from "../components/DeleteAllButton";

export function DashboardCanvas({ view, layout, setLayout, onLayoutChange, onCancel, isDelete, setIsDelete, isEditable, setIsEditable, onChangeView, setOriginalLayout }) {
  const [widgetArray, setWidgetArray] = useState([]);

  const onAddWidget = (ID) => {
    setLayout(prev => [
      ...prev,
      {
        i: ID,
        x: (prev.length * 3) % 12,
        y: Infinity, 
        w: 3, 
        h: 3,
        static: false
      }
    ]);
    onEditWidgets();
  };

  const onEditWidgets = () => {
    console.log("widgets are now editable");
    setOriginalLayout(layout); // Backup layout before editing
    // Set all widgets to editable, static: false
    setLayout(prevLayout =>
      prevLayout.map(item => ({
        ...item, 
        static: false
      }))
    );
    setIsEditable(true); // Enable edit mode
  }

  const onSave = () => {
    console.log("saving widgets...");
    // Set all widgets back to static: true
    setLayout(prevLayout => 
      prevLayout.map(item => ({
        ...item,
        static: true
    }))
  );
    setIsEditable(false);
    setIsDelete(false);
    setOriginalLayout([]); // Clear backup layout
  }

  const onDeleteWidgets = () => {
    console.log("Start deleting widgets...");
    setOriginalLayout(layout);
    setIsEditable(true);
    setIsDelete(true);
  }

  const DeleteWidget = (id) => {
    console.log("Deleting widget with id: ", id);
    setLayout(layout => layout.filter(w => w.i !== id));
    // add removal for widgetArray here
  }

  const DeleteAll = () => {
    const confirm = window.confirm(
      "Are you sure you want to delete all widgets?"
    );
    if (confirm) {
      setLayout([]);
      setWidgetArray([]);
      onSave();
    }
    else return;
  }

  return (
    /* HEADER */
    <Tabs.Root 
      lazyMount 
      value={ view }

      // Prevent navigation away from 'My Dashboard' while in Edit mode
      onValueChange={(e) => {
        const nextView = e.value;
        onChangeView(nextView); 
      }}
    >
      <Flex 
        as="header"
				position="sticky"
				top="0"
        zIndex="dropdown" // zIndex: 1100
				bg="base200"
				px={6}
				pt={3}
				align="center"
				justify="space-between"
      >
        <Tabs.List>
          <Tabs.Trigger value="widget" fontSize="lg">
            My Dashboard
          </Tabs.Trigger>
          <Tabs.Trigger value="gas" fontSize="lg">
            Gas
          </Tabs.Trigger>
          <Tabs.Trigger value="seismic" fontSize="lg">
            Seismic
          </Tabs.Trigger>
          <Tabs.Trigger value="remote" fontSize="lg">
            Remote Sensing
          </Tabs.Trigger>
          <Tabs.Trigger value="daily" fontSize="lg">
            Daily Activity
          </Tabs.Trigger>
        </Tabs.List>

        <HStack >
          {isEditable ? (
            <>
              <SaveButton onSave={ onSave } />
              <CancelButton onCancel={ onCancel } />
              {isDelete ? (
                <DeleteAllButton onDeleteAll={ DeleteAll } />
              ):(null)
              }
            </>
          ):(
            <>
              <SearchBar />
              <SettingsButton
                onAddWidget={ onAddWidget }
                onEditWidgets={ onEditWidgets }
                onDeleteWidgets={ onDeleteWidgets }
                setWidgetArray={ setWidgetArray }
                widgetArray={ widgetArray }
              />
            </>
          )}
        </HStack>
      </Flex>

      {/* CANVAS */}
      <Tabs.Content value="widget">
        <WidgetCanvas 
          layout={ layout }
          onLayoutChange={ onLayoutChange }
          isEditable={ isEditable }
          isDelete={ isDelete }
          DeleteWidget={ DeleteWidget }
          setWidgetArray={ setWidgetArray }
          widgetArray={ widgetArray }
        />
      </Tabs.Content>
      <Tabs.Content value="gas">
        <Gas />
      </Tabs.Content>
      <Tabs.Content value="seismic">
        <Seismic />
      </Tabs.Content>
      <Tabs.Content value="remote">
        <RemoteSensing />
      </Tabs.Content>
      <Tabs.Content value="daily">
        <DailyActivity />
      </Tabs.Content>
    </Tabs.Root>
  );
}