✨ handle labels in resources too

server/routers/resource/listResources.ts

@@ -1,7 +1,9 @@
 import {
     db,
+    labels,
     resourceHeaderAuth,
     resourceHeaderAuthExtendedCompatibility,
+    resourceLabels,
     resourcePassword,
     resourcePincode,
     resources,
@@ -9,8 +11,11 @@ import {
     sites,
     targetHealthCheck,
     targets,
-    userResources
+    userResources,
+    type Label
 } from "@server/db";
+import { isLicensedOrSubscribed } from "#dynamic/lib/isLicencedOrSubscribed";
+import { tierMatrix } from "@server/lib/billing/tierMatrix";
 import response from "@server/lib/response";
 import logger from "@server/logger";
 import { OpenAPITags, registry } from "@server/openApi";
@@ -154,10 +159,11 @@ export type ResourceWithTargets = {
         siteNiceId: string;
         online?: boolean; // undefined for local sites
     }>;
+    labels?: Array<Pick<Label, "color" | "labelId" | "name">>;
 };
 
-function queryResourcesBase() {
+function queryResourcesBase(isLabelFeatureEnabled: boolean) {
-    return db
+    let query = db
         .select({
             resourceId: resources.resourceId,
             name: resources.name,
@@ -203,8 +209,18 @@ function queryResourcesBase() {
         .leftJoin(
             targetHealthCheck,
             eq(targetHealthCheck.targetId, targets.targetId)
+        );
+
+    if (isLabelFeatureEnabled) {
+        query = query
+            .leftJoin(
+                resourceLabels,
+                eq(resourceLabels.resourceId, resources.resourceId)
             )
-        .groupBy(
+            .leftJoin(labels, eq(labels.labelId, resourceLabels.labelId));
+    }
+
+    return query.groupBy(
         resources.resourceId,
         resourcePassword.passwordId,
         resourcePincode.pincodeId,
@@ -288,6 +304,11 @@ export async function listResources(
             );
         }
 
+        const isLabelFeatureEnabled = await isLicensedOrSubscribed(
+            orgId,
+            tierMatrix.labels
+        );
+
         let accessibleResources: Array<{ resourceId: number }>;
         if (req.user) {
             accessibleResources = await db
@@ -369,8 +390,7 @@ export async function listResources(
             conditions.push(inArray(resources.resourceId, resourcesWithSite));
         }
         if (query) {
-            conditions.push(
+            const queryList = [
-                or(
                 like(
                     sql`LOWER(${resources.name})`,
                     "%" + query.toLowerCase() + "%"
@@ -383,11 +403,21 @@ export async function listResources(
                     sql`LOWER(${resources.fullDomain})`,
                     "%" + query.toLowerCase() + "%"
                 )
+            ];
+
+            if (isLabelFeatureEnabled) {
+                queryList.push(
+                    like(
+                        sql`LOWER(${labels.name})`,
+                        "%" + query.toLowerCase() + "%"
                     )
                 );
             }
 
-        const baseQuery = queryResourcesBase().where(and(...conditions));
+            conditions.push(or(...queryList));
+        }
+
+        const baseQuery = queryResourcesBase(isLabelFeatureEnabled).where(and(...conditions));
 
         // we need to add `as` so that drizzle filters the result as a subquery
         const countQuery = db.$count(baseQuery.as("filtered_resources"));
@@ -407,6 +437,35 @@ export async function listResources(
         ]);
 
         const resourceIdList = rows.map((row) => row.resourceId);
+
+        let labelsForResources: Array<{
+            labelId: number;
+            name: string;
+            color: string;
+            resourceId: number;
+        }> = [];
+
+        if (isLabelFeatureEnabled) {
+            labelsForResources =
+                resourceIdList.length === 0
+                    ? []
+                    : await db
+                          .select({
+                              labelId: labels.labelId,
+                              name: labels.name,
+                              color: labels.color,
+                              resourceId: resourceLabels.resourceId
+                          })
+                          .from(labels)
+                          .innerJoin(
+                              resourceLabels,
+                              eq(resourceLabels.labelId, labels.labelId)
+                          )
+                          .where(
+                              inArray(resourceLabels.resourceId, resourceIdList)
+                          );
+        }
+
         const allResourceTargets =
             resourceIdList.length === 0
                 ? []
@@ -458,7 +517,10 @@ export async function listResources(
                     headerAuthId: row.headerAuthId,
                     health: row.health ?? null,
                     targets: [],
-                    sites: []
+                    sites: [],
+                    labels: labelsForResources.filter(
+                        (l) => l.resourceId === row.resourceId
+                    )
                 };
                 map.set(row.resourceId, entry);
             }

src/app/[orgId]/settings/resources/proxy/page.tsx

@@ -111,6 +111,7 @@ export default async function ProxyResourcesPage(
             protocol: resource.protocol,
             proxyPort: resource.proxyPort,
             http: resource.http,
+            labels: resource.labels,
             authState: !resource.http
                 ? "none"
                 : resource.sso ||