1 files changed, 112 insertions, 0 deletions

diff --git a/users/ryan/modules/simple-bar/simple-bar-source/lib/components/yabai-context.jsx b/users/ryan/modules/simple-bar/simple-bar-source/lib/components/yabai-context.jsx
new file mode 100755
index 0000000..245977b
--- /dev/null
+++ b/users/ryan/modules/simple-bar/simple-bar-source/lib/components/yabai-context.jsx
@@ -0,0 +1,112 @@
+import * as Uebersicht from "uebersicht";
+import { useSimpleBarContext } from "./simple-bar-context.jsx";
+import useServerSocket from "../hooks/use-server-socket.js";
+import * as Yabai from "../yabai.js";
+import * as Skhd from "../skhd.js";
+
+const { React } = Uebersicht;
+
+// Create a context with default values
+const YabaiContext = React.createContext({
+  spaces: [],
+  windows: [],
+  skhdMode: "",
+});
+
+/**
+ * Custom hook to use the YabaiContext
+ * @returns {Object} The context value
+ */
+export function useYabaiContext() {
+  return React.useContext(YabaiContext);
+}
+
+export default React.memo(YabaiContextProvider);
+
+/**
+ * YabaiContextProvider component provides context for Yabai spaces, windows, and skhd mode.
+ * It uses server sockets to fetch and update the context values.
+ *
+ * @param {Object} props - The component props.
+ * @param {Array} props.spaces - Initial spaces data.
+ * @param {Array} props.windows - Initial windows data.
+ * @param {Object} props.skhdMode - Initial skhd mode data.
+ * @param {React.ReactNode} props.children - The child components to be wrapped by the provider.
+ * @returns {JSX.Element} The YabaiContext provider component.
+ */
+function YabaiContextProvider({ spaces, windows, skhdMode, children }) {
+  const { settings, setDisplays } = useSimpleBarContext();
+  const { enableServer, yabaiServerRefresh } = settings.global;
+  const { displaySkhdMode } = settings.process;
+  const serverEnabled = enableServer && yabaiServerRefresh;
+
+  const [yabaiSpaces, setYabaiSpaces] = React.useState(spaces);
+  const [yabaiWindows, setYabaiWindows] = React.useState(windows);
+  const [currentSkhdMode, setCurrentSkhdMode] = React.useState(skhdMode);
+
+  // Resets the spaces state to an empty array.
+  const resetSpaces = () => {
+    setYabaiSpaces([]);
+  };
+
+  // Resets the windows state to an empty array.
+  const resetWindows = () => {
+    setYabaiWindows([]);
+  };
+
+  // Resets the displays state to an empty array.
+  const resetDisplays = () => {
+    setDisplays([]);
+  };
+
+  // Resets the skhd mode state to an empty object.
+  const resetSkhdMode = () => {
+    setCurrentSkhdMode({});
+  };
+
+  // Fetches the current spaces from Yabai and updates the state.
+  const getSpaces = async () => {
+    const newSpaces = await Yabai.getSpaces();
+    setYabaiSpaces(newSpaces);
+  };
+
+  // Fetches the current windows from Yabai and updates the state.
+  const getWindows = async () => {
+    const newWindows = await Yabai.getWindows();
+    setYabaiWindows(newWindows);
+  };
+
+  // Fetches the current displays from Yabai and updates the state.
+  const getDisplays = async () => {
+    const newDisplays = await Yabai.getDisplays();
+    setDisplays(newDisplays);
+  };
+
+  // Fetches the current skhd mode from Skhd and updates the state.
+  const getSkhdMode = async () => {
+    const newSkhdMode = await Skhd.getMode();
+    setCurrentSkhdMode(newSkhdMode);
+  };
+
+  useServerSocket("spaces", serverEnabled, getSpaces, resetSpaces);
+  useServerSocket("windows", serverEnabled, getWindows, resetWindows);
+  useServerSocket("displays", serverEnabled, getDisplays, resetDisplays);
+  useServerSocket("mode", displaySkhdMode, getSkhdMode, resetSkhdMode);
+
+  const usedWindows = serverEnabled ? yabaiWindows : windows;
+  // Windows with an empty subrole should not be displayed in the UI
+  // See https://github.com/Jean-Tinland/simple-bar/issues/495 for reference
+  const filteredWindows = usedWindows.filter((window) => window.subrole);
+
+  return (
+    <YabaiContext.Provider
+      value={{
+        spaces: serverEnabled ? yabaiSpaces : spaces,
+        windows: filteredWindows,
+        skhdMode: serverEnabled ? currentSkhdMode : skhdMode,
+      }}
+    >
+      {children}
+    </YabaiContext.Provider>
+  );
+}