DownQuest.js

const oculusStoreAccessToken = "OC|752908224809889|";
let accessToken = "";
let applicationID = null;
let dlcs = [];
let channelDataCache = null;
let dlcDataCache = null;
let versionDataCache = {};
let currentUrl = "";

document.addEventListener("DOMContentLoaded", () => {
  handleURLChange(window.location.href);
});

chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
  if (request.message === "urlChanged") {
    console.log("New URL:", request.url);
    handleURLChange(request.url);
  }
});

function handleURLChange(newUrl) {
  if (newUrl !== currentUrl) {
    currentUrl = newUrl;
    channelDataCache = null;
    dlcDataCache = null;
    versionDataCache = {};
  }

  clearExistingButtons();

  const pathSegments = window.location.pathname.split("/").filter(Boolean);
  if (!shouldShowButtons(pathSegments)) return;

  createFloatingButtons();

  applicationID = determineApplicationID(pathSegments);
  if (applicationID) {
    if (!dlcDataCache) fetchDLCData();
    if (!channelDataCache) fetchChannelData();
  }
}

function shouldShowButtons(pathSegments) {
  if (pathSegments.length <= 3) return false;
  const thirdSegment = pathSegments[2];
  if (["view", "section"].includes(thirdSegment)) return false;
  if (thirdSegment === "pcvr" && ["view", "section"].includes(pathSegments[3]))
    return false;
  return true;
}

function determineApplicationID(pathSegments) {
  if (pathSegments.length > 4 && pathSegments[2] === "pcvr")
    return pathSegments[4];
  if (pathSegments.length > 3) return pathSegments[3];
  return null;
}

function clearExistingButtons() {
  document
    .querySelectorAll(".custom-floating-button")
    .forEach((button) => button.remove());
}

function createFloatingButtons() {
  const buttonsConfig = [
    {
      text: "DLC",
      clickHandler: handleDLCButtonClick,
      modalTitle: "DLC Downloads",
      contentFunction: displayDLCs,
    },
    {
      text: "Downgrade",
      clickHandler: handleDowngradeButtonClick,
      modalTitle: "App Downgrade Options",
      contentFunction: displayDowngradeOptions,
      style: { right: "120px" },
    },
  ];

  buttonsConfig.forEach(
    ({ text, clickHandler, modalTitle, contentFunction, style }) => {
      const button = document.createElement("button");
      button.className = "custom-button custom-floating-button";
      button.innerText = text;
      if (style) Object.assign(button.style, style);
      button.addEventListener("click", async () => {
        await clickHandler();
        showModal(modalTitle, contentFunction, true);
      });
      document.body.appendChild(button);
    },
  );
}

async function handleDLCButtonClick() {
  if (!dlcDataCache) await fetchDLCData();
}

async function handleDowngradeButtonClick() {
  if (!channelDataCache) await fetchChannelData();
}

async function fetchChannelData() {
  if (channelDataCache) return channelDataCache;

  const requestData = buildGraphQLRequestData({
    access_token: oculusStoreAccessToken,
    variables: JSON.stringify({ applicationID }),
    doc_id: "3828663700542720",
  });

  try {
    const response = await sendGraphQLRequest(requestData);
    channelDataCache = response.data.node.release_channels.nodes.map(
      (node) => ({
        id: node.id,
        name: node.channel_name,
      }),
    );

    await Promise.all(
      channelDataCache.map((channel) => fetchVersions(channel.id)),
    );

    return channelDataCache;
  } catch (error) {
    console.error("Failed to fetch channel data:", error);
    return [];
  }
}

async function fetchDLCData() {
  if (dlcDataCache) {
    dlcs = dlcDataCache;
    return;
  }

  const requestData = buildGraphQLRequestData({
    access_token: oculusStoreAccessToken,
    variables: JSON.stringify({
      id: applicationID,
      first: 10000,
    }),
    doc_id: "3853229151363174",
  });

  try {
    const response = await sendGraphQLRequest(requestData);
    dlcs = response.data.node.latest_supported_binary.firstIapItems.edges.map(
      (edge) => {
        const node = edge.node;
        if (node.latest_supported_asset_file) {
          return [
            "asset",
            node.id,
            node.display_name,
            node.latest_supported_asset_file.id,
          ];
        } else if (node.bundle_items) {
          const bundle = node.bundle_items.edges.map((edge) => edge.node.id);
          return ["bundle", node.id, node.display_name, bundle];
        } else {
          return [null, node.id, node.display_name];
        }
      },
    );
    dlcDataCache = dlcs;
  } catch (error) {
    console.error("Failed to fetch DLC data:", error);
  }
}

function buildGraphQLRequestData({ access_token, variables, doc_id }) {
  return `access_token=${access_token}&variables=${variables}&doc_id=${doc_id}`;
}

function displayDLCs(header, content) {
  const dlcSection = document.createElement("div");
  dlcSection.className = "dlc-list";

  if (dlcs.length === 0) {
    const noDLCMessage = document.createElement("div");
    noDLCMessage.className = "no-dlc-message";
    noDLCMessage.innerText = "No DLCs available for this game.";
    dlcSection.appendChild(noDLCMessage);
  } else {
    dlcs.forEach((dlc) => {
      const dlcItem = document.createElement("div");
      dlcItem.className = "dlc-item";

      const dlcName = document.createElement("span");
      dlcName.innerText = dlc[2];
      dlcItem.appendChild(dlcName);

      const downloadButton = createDownloadButton(dlc);
      if (downloadButton) {
        dlcItem.appendChild(downloadButton);
      }

      dlcSection.appendChild(dlcItem);
    });
  }

  content.appendChild(dlcSection);
}

function createDownloadButton(dlc) {
  const downloadButton = document.createElement("button");
  downloadButton.className = "custom-button";
  downloadButton.style.marginLeft = "10px";
  downloadButton.innerText = "Download";

  if (dlc[0] === "asset") {
    downloadButton.addEventListener("click", () => downloadBuild(dlc[3]));
    return downloadButton;
  } else if (dlc[0] === "bundle") {
    downloadButton.addEventListener("click", () => {
      showModal("Download DLC from bundle!", (header, versions) => {
        const headerText1 = document.createTextNode("Selected bundle: ");
        const headerText2 = document.createTextNode(dlc[2]);
        const boldText = document.createElement("b");
        boldText.appendChild(headerText2);
        header.appendChild(headerText1);
        header.appendChild(boldText);

        dlc[3].forEach((id) => {
          dlcs.forEach((tmpDLC) => {
            if (tmpDLC[1] === id && tmpDLC[0] === "asset") {
              createVersion(versions, null, tmpDLC[2], tmpDLC[3]);
            }
          });
        });
      });
    });
    return downloadButton;
  } else {
    return null;
  }
}

function displayDowngradeOptions(header, versions) {
  header.innerHTML = "";
  const headerText1 = document.createTextNode("Choose the ");
  const headerText2 = document.createTextNode("release channel ");
  const headerText3 = document.createTextNode("you want to use: ");
  const boldText = document.createElement("b");
  boldText.appendChild(headerText2);
  header.appendChild(headerText1);
  header.appendChild(boldText);
  header.appendChild(headerText3);

  createChannelDropdownToggle(header, versions);
}

function createChannelDropdownToggle(header, versions) {
  fetchChannelData().then((channels) => {
    const dropdown = document.createElement("select");
    dropdown.className = "dropdown";

    channels.forEach((channel) => {
      const option = document.createElement("option");
      option.value = channel.id;
      option.textContent = channel.name;
      dropdown.appendChild(option);
    });

    if (channels.length > 0) {
      fetchVersions(channels[0].id, versions);
    }

    dropdown.addEventListener("change", () => {
      const selectedChannel = channels.find(
        (channel) => channel.id === dropdown.value,
      );
      if (selectedChannel) {
        versions.innerHTML = "";
        fetchVersions(selectedChannel.id, versions);
      }
    });

    header.appendChild(dropdown);
  });
}

const zIndexBase = 1000;

function showModal(title, contentFunction, closeable = true) {
  const modalCount = document.querySelectorAll(".custom-modal").length;

  const modalZIndex = zIndexBase + modalCount * 2;
  const backdropZIndex = modalZIndex - 1;

  const modal = document.createElement("div");
  modal.className = "custom-modal";
  modal.style.zIndex = modalZIndex;
  document.body.appendChild(modal);

  const backdrop = document.createElement("div");
  backdrop.className = "custom-backdrop";
  backdrop.style.zIndex = backdropZIndex;
  if (closeable) {
    backdrop.addEventListener("click", closeModal);
  }
  document.body.appendChild(backdrop);

  const titleElement = document.createElement("h2");
  titleElement.innerText = title;
  modal.appendChild(titleElement);

  const header = document.createElement("div");
  const content = document.createElement("div");

  function closeModal() {
    modal.remove();
    backdrop.remove();
  }

  contentFunction(header, content, closeModal);
  modal.appendChild(header);
  modal.appendChild(content);

  if (closeable) {
    const closeButton = document.createElement("button");
    closeButton.className = "custom-close-button";
    closeButton.innerText = "×";
    closeButton.addEventListener("click", closeModal);
    modal.appendChild(closeButton);
  }
}

async function fetchVersions(channelId, versionsContainer = null) {
  if (versionDataCache[channelId]) {
    if (versionsContainer) {
      displayVersions(versionDataCache[channelId], versionsContainer);
    }
    return versionDataCache[channelId];
  }

  const requestData = buildGraphQLRequestData({
    access_token: oculusStoreAccessToken,
    variables: JSON.stringify({ releaseChannelID: channelId }),
    doc_id: "3973666182694273",
  });

  try {
    const response = await sendGraphQLRequest(requestData);
    const versionList = response.data.node.binaries.edges.map((version) => ({
      version: version.node.version,
      changeLog: version.node.change_log,
      id: version.node.id,
      obb: version.node.obb_binary,
      versionCode: version.node.version_code,
    }));

    versionDataCache[channelId] = versionList;

    if (versionsContainer) {
      displayVersions(versionList, versionsContainer);
    }

    return versionList;
  } catch (error) {
    console.error("Failed to fetch versions:", error);
    if (versionsContainer) {
      const errorMessage = document.createElement("div");
      errorMessage.textContent = "Error fetching versions.";
      versionsContainer.appendChild(errorMessage);
    }
    return [];
  }
}

function displayVersions(versionList, versionsContainer) {
  if (versionList.length === 0) {
    const noVersionsMessage = document.createElement("div");
    noVersionsMessage.textContent = "No versions available for this channel.";
    versionsContainer.appendChild(noVersionsMessage);
  } else {
    versionList.forEach((version) => {
      createVersion(
        versionsContainer,
        version.version,
        version.changeLog,
        version.id,
        version.obb,
        version.versionCode,
      );
    });
  }
}

function sendGraphQLRequest(requestData) {
  return new Promise((resolve, reject) => {
    const xhr = new XMLHttpRequest();
    const graphURI = "https://graph.oculus.com/graphql";
    xhr.open("POST", graphURI, true);
    xhr.setRequestHeader("Content-Type", "application/x-www-form-urlencoded");
    xhr.onreadystatechange = function () {
      if (xhr.readyState === 4) {
        if (xhr.status === 200) {
          const response = JSON.parse(xhr.responseText);
          resolve(response);
        } else {
          try {
            const errorResponse = JSON.parse(xhr.responseText);
            if (errorResponse.error && errorResponse.error.code === 190) {
              showModal("Session Error", (header, _) => {
                const headerText1 = document.createTextNode(
                  "Error validating access token. Please log into ",
                );
                const headerText3 = document.createTextNode(" and try again.");
                const link = document.createElement("a");
                link.className = "custom-link";
                link.href = "https://secure.oculus.com";
                link.target = "_blank";
                link.textContent = "secure.oculus.com";
                header.appendChild(headerText1);
                header.appendChild(link);
                header.appendChild(headerText3);
              });
            } else {
              console.error(
                "Request failed with status:",
                xhr.status,
                xhr.responseText,
              );
            }
          } catch (e) {
            console.error("Failed to parse error response:", e);
          }
          reject(xhr.status);
        }
      }
    };
    xhr.send(requestData);
  });
}

function createVersion(versions, version, changeLog, id, obb, versionCode) {
  const versionRow = document.createElement("div");
  versionRow.className = "custom-version-row";
  versionRow.style.marginBottom = "10px";
  versions.appendChild(versionRow);

  const descriptionButton = document.createElement("button");
  descriptionButton.className = "custom-button";
  descriptionButton.style.width = "100%";
  descriptionButton.style.textAlign = "left";
  descriptionButton.addEventListener("click", () => {
    if (obb) {
      downloadBuild(id, versionCode, obb.id);
    } else {
      downloadBuild(id, versionCode, null);
    }
  });
  descriptionButton.innerText = version
    ? `${version} | ${changeLog || "–"}`
    : changeLog || "–";
  versionRow.appendChild(descriptionButton);
}

async function downloadBuild(binaryId, versionCode, obbId) {
  chrome.runtime.sendMessage("Me want cookie!", async (response) => {
    accessToken = response;
    if (accessToken) {
      try {
        const manifestResponse = await fetch(getManifestURI(binaryId));
        if (manifestResponse.status === 200) {
          const arrayBuffer = await manifestResponse.arrayBuffer();
          const unzip = new Zlib.Unzip(new Uint8Array(arrayBuffer));
          const manifest = unzip.decompress("manifest.json");
          const manifestJSON = JSON.parse(new TextDecoder().decode(manifest));
          const files = manifestJSON.files;

          window.onbeforeunload = () =>
            "You are about to leave this page. Your download will not complete.";

          const customContent = async (header, versions, closeModal) => {
            const headerText1 = document.createTextNode(
              "Currently downloading: ",
            );
            const headerText2 = document.createTextNode(
              manifestJSON.canonicalName,
            );
            const boldText = document.createElement("b");
            boldText.appendChild(headerText2);
            header.appendChild(headerText1);
            header.appendChild(boldText);

            try {
              const progressIndicator = createProgressIndicator(versions);
              let dirHandle;
              try {
                dirHandle = await self.showDirectoryPicker({
                  startIn: "downloads",
                });
                while (
                  (await dirHandle.queryPermission({ mode: "readwrite" })) !==
                  "granted"
                ) {
                  await dirHandle.requestPermission({ mode: "readwrite" });
                }
              } catch (e) {
                closeModal();
                return;
              }

              for (let i = 0; true; i++) {
                let directoryName =
                  i === 0
                    ? manifestJSON.canonicalName
                    : `${manifestJSON.canonicalName} (${i})`;
                try {
                  await dirHandle.getDirectoryHandle(directoryName);
                } catch {
                  dirHandle = await dirHandle.getDirectoryHandle(
                    directoryName,
                    { create: true },
                  );
                  break;
                }
              }

              const totalSize = Object.values(files).reduce(
                (total, file) => total + file.size,
                0,
              );
              let currentSize = 0;

              const downloadFile = async (key) => {
                let path = key.split(/\\|\//);
                let currentDir = dirHandle;
                while (path.length > 1) {
                  currentDir = await currentDir.getDirectoryHandle(
                    path.shift(),
                    { create: true },
                  );
                }

                const newFileHandle = await currentDir.getFileHandle(path, {
                  create: true,
                });
                const writable = await newFileHandle.createWritable();

                for (const segment of files[key].segments) {
                  const response = await fetch(
                    getSegmentURI(binaryId, segment[1]),
                  );
                  const arrayBuffer = await response.arrayBuffer();
                  const inflate = new Zlib.Inflate(new Uint8Array(arrayBuffer));
                  const inflated = inflate.decompress();
                  await writable.write(inflated);
                  currentSize += segment[2];
                  progressIndicator.style.width =
                    (currentSize / totalSize) * 100 + "%";
                }
                await writable.close();
              };

              await Promise.all(Object.keys(files).map(downloadFile));

              closeModal();
              window.onbeforeunload = null;
            } catch (e) {
              console.error(e);
              const errorMSG = document.createElement("div");
              const error = document.createTextNode(e);
              errorMSG.appendChild(error);
              errorMSG.style.color = "red";
              versions.appendChild(errorMSG);
              window.onbeforeunload = null;
            }
          };
          showModal("Downloading...", customContent, false);
        } else {
          console.info("Probably a mobile build (or not purchased)");
          if (obbId != null) {
            openURI(getDownloadURI(obbId), true);
          }
          const requestData = buildAppBinaryInfoRequestData(
            applicationID,
            versionCode,
          );
          sendGraphQLRequest(requestData)
            .then((response) => {
              const binary = response.data.app_binary_info.info[0].binary;
              if (binary != null) {
                const edges = binary.asset_files.edges;
                if (edges.length < 1) {
                  openURI(getDownloadURI(binaryId), true);
                } else {
                  openURI(getDownloadURI(binary.id), true);
                  edges.forEach((edge) => {
                    openURI(
                      `${edge.node.uri}&access_token=${accessToken}`,
                      false,
                    );
                  });
                }
              } else {
                openURI(getDownloadURI(binaryId), true);
              }
            })
            .catch();
        }
      } catch (error) {
        console.error("Network error:", error);
      }
    } else {
      const customContent = (header, versions) => {
        const headerText1 = document.createTextNode(
          "No oculus.com login found, please log into ",
        );
        const headerText2 = document.createTextNode("secure.oculus.com");
        const headerText3 = document.createTextNode(" and try again");
        const link = document.createElement("a");
        link.className = "custom-link";
        link.href = "https://secure.oculus.com";
        link.target = "_blank";
        link.appendChild(headerText2);
        header.appendChild(headerText1);
        header.appendChild(link);
        header.appendChild(headerText3);
      };

      showModal("No access_token found", customContent);
    }
  });
}

function buildAppBinaryInfoRequestData(appId, versionCode) {
  return `access_token=${accessToken}&doc=query ($params: AppBinaryInfoArgs!) { app_binary_info(args: $params) { info { binary { ... on AndroidBinary { id package_name version_code asset_files { edges { node { ... on AssetFile { file_name uri size } } } } } } } }}&variables={\"params\":{\"app_params\":[{\"app_id\":\"${appId}\",\"version_code\":\"${versionCode}\"}]}}`;
}

async function openURI(uri, important) {
  try {
    const response = await fetch(uri, {
      method: "HEAD",
      headers: { "Content-Type": "application/x-www-form-urlencoded" },
    });

    if (response.ok) {
      window.open(uri, "_blank");
    } else {
      console.info(
        "Couldn't access " + uri + " because you probably haven't purchased it",
      );
      if (important) {
        const customContent = (header, _) => {
          const headerText = document.createTextNode(
            "It seems like DownQuest wasn't able to generate all required download links. This is most likely the case because you don't own the content or have been logged out.",
          );
          header.appendChild(headerText);
        };
        showModal("Info", customContent);
      }
    }
  } catch (error) {
    console.error("Network error:", error);
  }
}

function createProgressIndicator(parent) {
  const progressIndicatorBackground = document.createElement("div");
  progressIndicatorBackground.className = "progress-indicator-background";
  parent.appendChild(progressIndicatorBackground);

  const progressIndicator = document.createElement("div");
  progressIndicator.className = "progress-indicator";
  progressIndicatorBackground.appendChild(progressIndicator);

  return progressIndicator;
}

function getManifestURI(id) {
  return `https://securecdn.oculus.com/binaries/download/?access_token=${accessToken}&id=${id}&get_manifest=1`;
}

function getDownloadURI(id) {
  return `https://securecdn.oculus.com/binaries/download/?id=${id}&access_token=${accessToken}`;
}

function getSegmentURI(binaryId, segmentSha256) {
  return `https://securecdn.oculus.com/binaries/segment/?access_token=${accessToken}&binary_id=${binaryId}&segment_sha256=${segmentSha256}`;
}