starlight-view-modes/middleware.ts

Starlight plugin adding view mode capabilities to your documentation website.

import { defineRouteMiddleware } from "@astrojs/starlight/route-data";
import type { APIContext } from "astro";

import { getRouteData } from "./libs/routeData";
import { modifySidebarAndPagination } from "./libs/sidebar";
import { insertModePathname } from "./libs/utils";

export const onRequest = defineRouteMiddleware(async (context) => {
  const { starlightRoute } = context.locals;
  const { id, sidebar, pagination, siteTitleHref } = starlightRoute;

  await modifySidebarAndPagination(starlightRoute, id, sidebar, pagination);
  await attachRouteData(context);

  const currentMode = context.locals.starlightViewModes.modes.find(
    (mode) => mode.isCurrent
  );

  if (currentMode && currentMode.name !== "default")
    starlightRoute.siteTitleHref = insertModePathname(
      siteTitleHref,
      currentMode.name
    );
});

async function attachRouteData(context: APIContext) {
  context.locals.starlightViewModes = await getRouteData(context);
}