Improve comments

Author: brookslybrand

app/actions/menu-collapse/route.ts

@@ -16,6 +16,8 @@ export async function action({ request, context }: Route.ActionArgs) {
   let parsedOpen = open === "true";
 
   let menuCollapse = menuCollapseContext(context);
+  // This action is responsible for setting the menu collapse state.
+  // Middleware is responsible for writing it to the cookie.
   menuCollapse.set(category, parsedOpen);
 
   return parsedOpen;

app/actions/menu-collapse/server.ts

@@ -11,6 +11,7 @@ let cookie = createCookie("menu-collapse", {
   sameSite: "lax",
 });
 
+// Default behavior: missing categories are treated as "open" (true)
 type MenuCollapseState = Record<string, boolean>;
 
 let menuCollapseStateContext = unstable_createContext<MenuCollapseState>({});

app/components/docs-menu/menu.tsx

@@ -126,7 +126,7 @@ function MenuCategoryDetails({
 function useMenuCollapse(category?: string) {
   const menuCollapseState = useDocsLayoutRouteLoaderData()?.menuCollapseState;
   const [isOpen, setIsOpen] = React.useState(
-    () => menuCollapseState?.[category ?? ""] ?? true,
+    menuCollapseState?.[category ?? ""] ?? true,
   );
   const submit = useSubmit();
 
@@ -149,7 +149,7 @@ function useMenuCollapse(category?: string) {
     [category, submit],
   );
 
-  // Auto open the details element, necessary when navigating from the index page or a document request on a page in a collapsed menu
+  // Auto-expand when navigating to a page within this category
   let { isActive } = useNavigation(category);
   React.useEffect(() => {
     if (isActive) {