CDRM-Extension/frontend/src/components/topnav.jsx

import { useEffect, useState } from "react";
import hamburgerIcon from "../assets/hamburger.svg";

function TopNav({ onMenuClick }) {
  const [injectionType, setInjectionType] = useState("LICENSE");
  const [drmOverride, setDrmOverride] = useState("DISABLED");

  useEffect(() => {
    chrome.storage.local.get(["injection_type", "drm_override"], (result) => {
      if (result.injection_type !== undefined) {
        setInjectionType(result.injection_type);
      }
      if (result.drm_override !== undefined) {
        setDrmOverride(result.drm_override);
      }
    });
  }, []);

  const handleInjectionTypeChange = (type) => {
    chrome.storage.local.set({ injection_type: type }, () => {
      if (chrome.runtime.lastError) {
        console.error(
          "Error updating injection_type:",
          chrome.runtime.lastError
        );
      } else {
        setInjectionType(type);
        console.log(`Injection type updated to ${type}`);
      }
    });
  };

  const handleDrmOverrideChange = (type) => {
    chrome.storage.local.set({ drm_override: type }, () => {
      if (chrome.runtime.lastError) {
        console.error("Error updating drm_override:", chrome.runtime.lastError);
      } else {
        setDrmOverride(type);
        console.log(`DRM Override updated to ${type}`);
      }
    });
  };

  return (
    <div className="w-full h-full flex flex-row overflow-x-hidden">
      <img
        src={hamburgerIcon}
        alt="Menu"
        className="h-full w-16 p-2 flex items-center cursor-pointer"
        onClick={onMenuClick}
      />
      <div className="flex flex-row h-full justify-center items-center ml-auto mr-2">
        <p className="text-white text-lg p-2 mr-2 border-r-2 border-r-white text-nowrap">
          DRM Override:
        </p>
        <button
          onClick={() => handleDrmOverrideChange("WIDEVINE")}
          className={`text-white text-lg p-2 rounded-md m-1 cursor-pointer ${
            drmOverride === "WIDEVINE" ? "bg-green-500/70" : "bg-black"
          }`}
        >
          Widevine
        </button>
        <button
          onClick={() => handleDrmOverrideChange("PLAYREADY")}
          className={`text-white text-lg p-2 rounded-md m-1 cursor-pointer ${
            drmOverride === "PLAYREADY" ? "bg-sky-500/70" : "bg-black"
          }`}
        >
          PlayReady
        </button>
        <button
          onClick={() => handleDrmOverrideChange("DISABLED")}
          className={`text-white text-lg p-2 rounded-md m-1 cursor-pointer ${
            drmOverride === "DISABLED" ? "bg-red-500/70" : "bg-black"
          }`}
        >
          Disabled
        </button>
      </div>
      <div className="flex flex-row h-full justify-center items-center ml-auto mr-2">
        <p className="text-white text-lg p-2 mr-2 border-r-2 border-r-white text-nowrap">
          Injection Type:
        </p>
        <button
          onClick={() => handleInjectionTypeChange("LICENSE")}
          className={`text-white text-lg p-2 rounded-md m-1 cursor-pointer ${
            injectionType === "LICENSE" ? "bg-sky-500/70" : "bg-black"
          }`}
        >
          License
        </button>
        <button
          onClick={() => handleInjectionTypeChange("EME")}
          className={`text-white text-lg p-2 rounded-md m-1 cursor-pointer ${
            injectionType === "EME" ? "bg-green-500/70" : "bg-black"
          }`}
        >
          EME
        </button>
        <button
          onClick={() => handleInjectionTypeChange("DISABLED")}
          className={`text-white text-lg p-2 rounded-md m-1 cursor-pointer ${
            injectionType === "DISABLED" ? "bg-red-500/70" : "bg-black"
          }`}
        >
          Disabled
        </button>
      </div>
    </div>
  );
}

export default TopNav;