Focus dashboard on Agent Network for netbird.ai signups

src/app/(dashboard)/agent-network/layout.tsx

@@ -1,19 +1,26 @@
 "use client";
 
-import { isAgentNetworkEnabled } from "@utils/netbird";
 import { notFound } from "next/navigation";
 import * as React from "react";
+import { useAgentNetworkMode } from "@/modules/agent-network/useAgentNetworkMode";
 
 // Gates the entire Agent Network route tree behind the NETBIRD_AGENT_NETWORK
-// flag. When disabled, these routes don't exist as far as the user is
-// concerned — the dashboard behaves exactly as it did without the feature.
+// flag or the account-level agent_network_only setting. When disabled, these
+// routes don't exist as far as the user is concerned — the dashboard behaves
+// exactly as it did without the feature. Rendering waits for the account to
+// load so accounts enabled via settings don't get a redirect flash.
 export default function AgentNetworkLayout({
   children,
 }: {
   children: React.ReactNode;
 }) {
-  if (!isAgentNetworkEnabled()) {
-    notFound();
+  const { enabled, loading } = useAgentNetworkMode();
+
+  if (enabled) {
+    return <>{children}</>;
+  }
+  if (loading) {
+    return null;
   }
-  return <>{children}</>;
-}
+  notFound();
+}

src/app/(dashboard)/control-center/page.tsx

@@ -12,7 +12,6 @@ import SquareIcon from "@components/SquareIcon";
 import GetStartedTest from "@components/ui/GetStartedTest";
 import { SmallBadge } from "@components/ui/SmallBadge";
 import useFetchApi from "@utils/api";
-import { isAgentNetworkEnabled, isAgentNetworkOnly } from "@utils/netbird";
 import {
   Background,
   Edge,
@@ -54,6 +53,7 @@ import AIProvidersProvider, {
   useAIProviders,
 } from "@/modules/agent-network/AIProvidersProvider";
 import { AIProviderId } from "@/modules/agent-network/data/mockData";
+import { useAgentNetworkMode } from "@/modules/agent-network/useAgentNetworkMode";
 import { FlowSelector, FlowView } from "@/modules/control-center/FlowSelector";
 import { NetworkRoutingPeerCount } from "@/modules/control-center/NetworkRoutingPeerCount";
 import { ControlCenterCurrentUserBadge } from "@/modules/control-center/user/ControlCenterCurrentUserBadge";
@@ -90,6 +90,8 @@ function ControlCenterView() {
   const [layoutInitialized, setLayoutInitialized] = useState(false);
   const [forceLayoutChange, setForceLayoutChange] = useState(false);
   const { loggedInUser } = useLoggedInUser();
+  const { only: agentNetworkOnly, enabled: agentNetworkEnabled } =
+    useAgentNetworkMode();
 
   const queryParams = useSearchParams();
   const queryTab = queryParams.get("tab");
@@ -125,13 +127,13 @@ function ControlCenterView() {
     "/agent-network/providers",
     true,
     true,
-    isAgentNetworkEnabled(),
+    agentNetworkEnabled,
   );
   const { data: agentPolicies } = useFetchApi<APIPolicy[]>(
     "/agent-network/policies",
     true,
     true,
-    isAgentNetworkEnabled(),
+    agentNetworkEnabled,
   );
 
   // providerById lets the overlay look up a Provider's display payload
@@ -1995,7 +1997,7 @@ function ControlCenterView() {
               {/* Networks is dropped as a top-level pivot in the
                   agent-network repackaging — keep the dropdown + per-network
                   chrome for everyone else so flag-off behaviour is unchanged. */}
-              {!isAgentNetworkOnly() && currentView === "networks" && (
+              {!agentNetworkOnly && currentView === "networks" && (
                 <div className={"w-64"}>
                   <SelectDropdown
                     variant={"secondary"}
@@ -2011,7 +2013,7 @@ function ControlCenterView() {
                 </div>
               )}
 
-              {!isAgentNetworkOnly() && selectedNetwork && currentNetwork && (
+              {!agentNetworkOnly && selectedNetwork && currentNetwork && (
                 <NetworkRoutingPeerCount network={currentNetwork} />
               )}
             </div>

src/cloud/contexts/NetBirdCloudProvider.tsx

@@ -1,8 +1,9 @@
+import { notify } from "@components/Notification";
 import { useApiCall } from "@utils/api";
 import loadConfig from "@utils/config";
 import { isNetBirdCloud } from "@utils/netbird";
 import * as React from "react";
-import { useEffect, useState } from "react";
+import { useEffect, useRef, useState } from "react";
 import { useSWRConfig } from "swr";
 import { Hubspot, submitHubspotForm } from "@/cloud/analytics/Hubspot";
 import { AWSChoosePlan } from "@/cloud/aws/AWSChoosePlan";
@@ -11,8 +12,14 @@ import { useDomainCategory } from "@/cloud/cloud-hooks/useDomainCategory";
 import HowDidYouHearAboutUs from "@/cloud/survey/HowDidYouHearAboutUs";
 import { useAnalytics } from "@/contexts/AnalyticsProvider";
 import { useBilling } from "@/contexts/BillingProvider";
+import {
+  AGENT_NETWORK_SIGNUP_SOURCE,
+  SIGNUP_SOURCE_LOCAL_STORAGE_KEY,
+} from "@/hooks/useSignupSource";
+import type { Account } from "@/interfaces/Account";
 import type { Group } from "@/interfaces/Group";
 import { PlanTier } from "@/interfaces/Subscription";
+import { useAccount } from "@/modules/account/useAccount";
 import { OnboardingProvider } from "@/modules/onboarding/OnboardingProvider";
 
 export const NetBirdCloudProvider = () => {
@@ -25,6 +32,9 @@ export const NetBirdCloudProvider = () => {
     "/integrations/billing/aws/marketplace/enrich",
     true,
   ).post;
+  const account = useAccount();
+  const accountRequest = useApiCall<Account>("/accounts", true).put;
+  const signupSourceApplied = useRef(false);
 
   useEffect(() => {
     try {
@@ -41,6 +51,42 @@ export const NetBirdCloudProvider = () => {
     } catch (e) {}
   }, []);
 
+  // Apply the netbird.ai signup source once the account is available. Only
+  // new accounts (signup form still pending) are switched to the Agent
+  // Network focused view — a stale flag from an existing user is discarded.
+  useEffect(() => {
+    if (!account || signupSourceApplied.current) return;
+    try {
+      const source = localStorage.getItem(SIGNUP_SOURCE_LOCAL_STORAGE_KEY);
+      if (source !== AGENT_NETWORK_SIGNUP_SOURCE) return;
+
+      if (
+        account.onboarding?.signup_form_pending !== true ||
+        account.settings?.agent_network_only === true
+      ) {
+        localStorage.removeItem(SIGNUP_SOURCE_LOCAL_STORAGE_KEY);
+        return;
+      }
+
+      signupSourceApplied.current = true;
+      notify({
+        title: "Agent Network",
+        description: "Agent Network focused view enabled for your account.",
+        promise: accountRequest(
+          {
+            id: account.id,
+            settings: { ...account.settings, agent_network_only: true },
+          },
+          "/" + account.id,
+        ).then(() => {
+          mutate("/accounts");
+          localStorage.removeItem(SIGNUP_SOURCE_LOCAL_STORAGE_KEY);
+        }),
+        loadingMessage: "Enabling Agent Network focused view...",
+      });
+    } catch (e) {}
+  }, [account]);
+
   const hasFreeOrTrialAWSPlan =
     subscription?.plan_tier === PlanTier.FREE ||
     subscription?.plan_tier === PlanTier.TRIAL;

src/hooks/useSignupSource.ts

@@ -0,0 +1,24 @@
+import { useEffect } from "react";
+
+export const SIGNUP_SOURCE_QUERY_PARAM = "signup_source";
+export const SIGNUP_SOURCE_LOCAL_STORAGE_KEY = "netbird-signup-source";
+export const AGENT_NETWORK_SIGNUP_SOURCE = "netbird.ai";
+
+/**
+ * Store the signup_source query parameter into localStorage so it survives
+ * the OIDC redirect and can be applied once the account is available.
+ */
+export function useSignupSource() {
+  useEffect(() => {
+    if (typeof window === "undefined") return;
+
+    const params = new URLSearchParams(window.location.search);
+    const source = params.get(SIGNUP_SOURCE_QUERY_PARAM);
+
+    if (source === AGENT_NETWORK_SIGNUP_SOURCE) {
+      try {
+        localStorage.setItem(SIGNUP_SOURCE_LOCAL_STORAGE_KEY, source);
+      } catch (e) {}
+    }
+  }, []);
+}

src/interfaces/Account.ts

@@ -35,6 +35,7 @@ export interface Account {
     local_mfa_enabled?: boolean;
     ipv6_enabled_groups?: string[];
     network_range_v6?: string;
+    agent_network_only?: boolean;
   };
   onboarding?: AccountOnboarding;
 }

src/layouts/AppLayout.tsx

@@ -21,6 +21,7 @@ import ErrorBoundaryProvider from "@/contexts/ErrorBoundary";
 import { GlobalThemeProvider } from "@/contexts/GlobalThemeProvider";
 import InstanceSetupProvider from "@/contexts/InstanceSetupProvider";
 import { NavigationEvents } from "@/contexts/NavigationEvents";
+import { useSignupSource } from "@/hooks/useSignupSource";
 
 const inter = localFont({
   src: "../assets/fonts/Inter.ttf",
@@ -39,6 +40,7 @@ export default function AppLayout({
   children,
 }: Readonly<{ children: React.ReactNode }>) {
   useAWSMarketplace();
+  useSignupSource();
 
   return (
     <html lang="en">

src/layouts/Navigation.tsx

@@ -2,11 +2,7 @@
 
 import { ScrollArea } from "@components/ScrollArea";
 import { cn } from "@utils/helpers";
-import {
-  isAgentNetworkEnabled,
-  isAgentNetworkOnly,
-  isNetBirdCloud,
-} from "@utils/netbird";
+import { isNetBirdCloud } from "@utils/netbird";
 import AccessControlIcon from "@/assets/icons/AccessControlIcon";
 import AgentNetworkIcon from "@/assets/icons/AgentNetworkIcon";
 import ControlCenterIcon from "@/assets/icons/ControlCenterIcon";
@@ -24,6 +20,7 @@ import { useAnnouncement } from "@/contexts/AnnouncementProvider";
 import { useApplicationContext } from "@/contexts/ApplicationProvider";
 import { usePermissions } from "@/contexts/PermissionsProvider";
 import { headerHeight } from "@/layouts/Header";
+import { useAgentNetworkMode } from "@/modules/agent-network/useAgentNetworkMode";
 import { NavigationUsageInfo } from "@/modules/billing/NavigationUsageInfo";
 import { NetworkNavigation } from "@/modules/networks/misc/NetworkNavigation";
 import { SmallBadge } from "@components/ui/SmallBadge";
@@ -43,6 +40,8 @@ export default function Navigation({
   const { bannerHeight } = useAnnouncement();
   const { isNavigationCollapsed } = useApplicationContext();
   const { permission, isRestricted } = usePermissions();
+  const { only: agentNetworkOnly, enabled: agentNetworkEnabled } =
+    useAgentNetworkMode();
 
   return (
     <div
@@ -126,7 +125,7 @@ export default function Navigation({
                   />
                 </SidebarItem>
 
-                {!isAgentNetworkOnly() && <NetworkNavigation />}
+                {!agentNetworkOnly && <NetworkNavigation />}
 
                 <SidebarItem
                   icon={<ReverseProxyIcon size={16} />}
@@ -145,7 +144,7 @@ export default function Navigation({
                   href={"/reverse-proxy"}
                   collapsible
                   exactPathMatch={false}
-                  visible={permission?.services?.read && !isAgentNetworkOnly()}
+                  visible={permission?.services?.read && !agentNetworkOnly}
                 >
                   <SidebarItem
                     label="Services"
@@ -183,7 +182,7 @@ export default function Navigation({
                   label={
                     <div className={"flex items-center gap-2"}>
                       Agent Network
-                      {!isAgentNetworkOnly() && (
+                      {!agentNetworkOnly && (
                         <SmallBadge
                           text={"Beta"}
                           variant={"sky"}
@@ -200,44 +199,36 @@ export default function Navigation({
                   exactPathMatch={false}
                   // Parent is visible when at least one child is permitted. All
                   // Agent Network pages guard on services.read, so the section
-                  // tracks that (plus the feature flag).
-                  visible={isAgentNetworkEnabled() && permission?.services?.read}
+                  // tracks that (plus the feature gating).
+                  visible={agentNetworkEnabled && permission?.services?.read}
                 >
                   <SidebarItem
                     label="Providers"
                     isChild
                     href={"/agent-network/providers"}
                     exactPathMatch={true}
-                    visible={
-                      isAgentNetworkEnabled() && permission?.services?.read
-                    }
+                    visible={agentNetworkEnabled && permission?.services?.read}
                   />
                   <SidebarItem
                     label="Policies"
                     isChild
                     href={"/agent-network/policies"}
                     exactPathMatch={true}
-                    visible={
-                      isAgentNetworkEnabled() && permission?.services?.read
-                    }
+                    visible={agentNetworkEnabled && permission?.services?.read}
                   />
                   <SidebarItem
                     label="Usage & Logs"
                     isChild
                     href={"/agent-network/usage"}
                     exactPathMatch={true}
-                    visible={
-                      isAgentNetworkEnabled() && permission?.services?.read
-                    }
+                    visible={agentNetworkEnabled && permission?.services?.read}
                   />
                   <SidebarItem
                     label="Configuration"
                     isChild
                     href={"/agent-network/configuration"}
                     exactPathMatch={true}
-                    visible={
-                      isAgentNetworkEnabled() && permission?.services?.read
-                    }
+                    visible={agentNetworkEnabled && permission?.services?.read}
                   />
                 </SidebarItem>
 
@@ -249,7 +240,7 @@ export default function Navigation({
                   exactPathMatch={true}
                   visible={
                     (permission.dns.read || permission.nameservers.read) &&
-                    !isAgentNetworkOnly()
+                    !agentNetworkOnly
                   }
                 >
                   <SidebarItem
@@ -351,14 +342,15 @@ export function SidebarItemGroup({ children }: SidebarItemGroupProps) {
 
 const ActivityNavigationItem = () => {
   const { permission } = usePermissions();
+  const { only: agentNetworkOnly } = useAgentNetworkMode();
 
   return (
     <SidebarItem
       icon={<ActivityIcon />}
       label="Activity"
       href={"/events"}
       collapsible
-      visible={permission.events.read && !isAgentNetworkOnly()}
+      visible={permission.events.read && !agentNetworkOnly}
     >
       <SidebarItem
         label="Audit Events"

src/modules/agent-network/AIProvidersProvider.tsx

@@ -2,7 +2,6 @@
 
 import { notify } from "@components/Notification";
 import useFetchApi, { useApiCall } from "@utils/api";
-import { isAgentNetworkEnabled } from "@utils/netbird";
 import React, {
   createContext,
   useCallback,
@@ -22,6 +21,7 @@ import {
   PolicyLimits,
   ProviderModel,
 } from "@/modules/agent-network/data/mockData";
+import { useAgentNetworkMode } from "@/modules/agent-network/useAgentNetworkMode";
 
 export type APIProviderModel = {
   id: string;
@@ -501,12 +501,13 @@ export function useAIProviders() {
 // which we still tolerate via ignoreError so older deploys don't surface
 // a spurious error in the empty state.
 export function useAgentNetworkSettings() {
+  const { enabled: agentNetworkEnabled } = useAgentNetworkMode();
   const { data, error, isLoading, mutate } =
     useFetchApi<APIAgentNetworkSettings>(
       "/agent-network/settings",
       true,
       true,
-      isAgentNetworkEnabled(),
+      agentNetworkEnabled,
     );
   const settings = useMemo<AgentNetworkSettings | null>(
     () => (data ? settingsFromAPI(data) : null),
@@ -528,7 +529,7 @@ export default function AIProvidersProvider({ children }: Readonly<Props>) {
   // disabled — it can safely wrap surfaces like the Control Center
   // without hitting agent-network endpoints in deployments that don't
   // have the feature.
-  const agentNetworkEnabled = isAgentNetworkEnabled();
+  const { enabled: agentNetworkEnabled } = useAgentNetworkMode();
 
   const {
     data: apiProviders,