feat(private-link): Add private links UI

Author: 0ski

.changeset/private-networking-dequeue.md

@@ -0,0 +1,5 @@
+---
+"@trigger.dev/core": patch
+---
+
+Add optional `hasPrivateLink` field to the dequeue message organization object for private networking support

.server-changes/private-networking.md

@@ -0,0 +1,6 @@
+---
+area: webapp
+type: feature
+---
+
+Add private networking support via AWS PrivateLink. Includes BillingClient methods for managing private connections, org settings UI pages for connection management, and supervisor changes to apply `privatelink` pod labels for CiliumNetworkPolicy matching.

apps/supervisor/src/index.ts

@@ -267,6 +267,7 @@ class ManagedSupervisor {
           snapshotId: message.snapshot.id,
           snapshotFriendlyId: message.snapshot.friendlyId,
           placementTags: message.placementTags,
+          hasPrivateLink: message.organization.hasPrivateLink,
         });
 
         // Disabled for now

apps/supervisor/src/workloadManager/kubernetes.ts

@@ -336,13 +336,20 @@ export class KubernetesWorkloadManager implements WorkloadManager {
   }
 
   #getSharedLabels(opts: WorkloadManagerCreateOptions): Record<string, string> {
-    return {
+    const labels: Record<string, string> = {
       env: opts.envId,
       envtype: this.#envTypeToLabelValue(opts.envType),
       org: opts.orgId,
       project: opts.projectId,
       machine: opts.machine.name,
     };
+
+    // Add privatelink label for CiliumNetworkPolicy matching
+    if (opts.hasPrivateLink) {
+      labels.privatelink = opts.orgId;
+    }
+
+    return labels;
   }
 
   #getResourceRequestsForMachine(preset: MachinePreset): ResourceQuantities {

apps/supervisor/src/workloadManager/types.ts

@@ -35,4 +35,6 @@ export interface WorkloadManagerCreateOptions {
   runFriendlyId: string;
   snapshotId: string;
   snapshotFriendlyId: string;
+  // private networking
+  hasPrivateLink?: boolean;
 }

apps/webapp/app/components/navigation/OrganizationSettingsSideMenu.tsx

@@ -3,6 +3,7 @@ import {
   ChartBarIcon,
   Cog8ToothIcon,
   CreditCardIcon,
+  LockClosedIcon,
   UserGroupIcon,
 } from "@heroicons/react/20/solid";
 import { ArrowLeftIcon } from "@heroicons/react/24/solid";
@@ -19,6 +20,7 @@ import {
   rootPath,
   v3BillingAlertsPath,
   v3BillingPath,
+  v3PrivateConnectionsPath,
   v3UsagePath,
 } from "~/utils/pathBuilder";
 import { LinkButton } from "../primitives/Buttons";
@@ -46,7 +48,7 @@ export function OrganizationSettingsSideMenu({
   organization: MatchedOrganization;
   buildInfo: BuildInfo;
 }) {
-  const { isManagedCloud } = useFeatures();
+  const { isManagedCloud, hasPrivateConnections } = useFeatures();
   const currentPlan = useCurrentPlan();
   const isAdmin = useHasAdminAccess();
   const showBuildInfo = isAdmin || !isManagedCloud;
@@ -103,6 +105,15 @@ export function OrganizationSettingsSideMenu({
               />
             </>
           )}
+          {hasPrivateConnections && (
+            <SideMenuItem
+              name="Private Connections"
+              icon={LockClosedIcon}
+              activeIconColor="text-purple-500"
+              to={v3PrivateConnectionsPath(organization)}
+              data-action="private-connections"
+            />
+          )}
           <SideMenuItem
             name="Team"
             icon={UserGroupIcon}

apps/webapp/app/features.server.ts

@@ -2,6 +2,7 @@ import { requestUrl } from "./utils/requestUrl.server";
 
 export type TriggerFeatures = {
   isManagedCloud: boolean;
+  hasPrivateConnections: boolean;
 };
 
 function isManagedCloud(host: string): boolean {
@@ -13,9 +14,17 @@ function isManagedCloud(host: string): boolean {
   );
 }
 
+function hasPrivateConnections(host: string): boolean {
+  if (process.env.PRIVATE_CONNECTIONS_ENABLED === "1") {
+    return isManagedCloud(host);
+  }
+  return false;
+}
+
 function featuresForHost(host: string): TriggerFeatures {
   return {
     isManagedCloud: isManagedCloud(host),
+    hasPrivateConnections: hasPrivateConnections(host),
   };
 }
 

apps/webapp/app/hooks/useFeatures.ts

@@ -5,5 +5,5 @@ import type { TriggerFeatures } from "~/features.server";
 export function useFeatures(): TriggerFeatures {
   const routeMatch = useTypedRouteLoaderData<typeof loader>("root");
 
-  return routeMatch?.features ?? { isManagedCloud: false };
+  return routeMatch?.features ?? { isManagedCloud: false, hasPrivateConnections: false };
 }