feat(clover): Ensure Clover RDS Assets can connect to subnets and sgs

systeminit · Feb 25, 2025 · 4a05e27 · 4a05e27
1 parent b4028e6
commit 4a05e27
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/bin/clover/src/pipeline-steps/assetSpecificOverrides.ts b/bin/clover/src/pipeline-steps/assetSpecificOverrides.ts
@@ -144,6 +144,54 @@ const overrides = new Map<string, OverrideFn>([
       setAnnotationOnSocket(socket, { tokens: ["Project Identifier"] });
     },
   ],
+  [
+    "AWS::RDS::DBInstance",
+    (spec: ExpandedPkgSpec) => {
+      const variant = spec.schemas[0].variants[0];
+
+      const prop = variant.domain.entries.find((p: ExpandedPropSpec) =>
+        p.name === "VPCSecurityGroups"
+      );
+
+      if (!prop) return;
+      const socket = createInputSocketFromProp(prop);
+
+      setAnnotationOnSocket(socket, { tokens: ["Group Id"] });
+      variant.sockets.push(socket);
+    },
+  ],
+  [
+    "AWS::RDS::DBCluster",
+    (spec: ExpandedPkgSpec) => {
+      const variant = spec.schemas[0].variants[0];
+
+      const prop = variant.domain.entries.find((p: ExpandedPropSpec) =>
+        p.name === "VpcSecurityGroupIds"
+      );
+
+      if (!prop) return;
+      const socket = createInputSocketFromProp(prop);
+
+      setAnnotationOnSocket(socket, { tokens: ["Group Id"] });
+      variant.sockets.push(socket);
+    },
+  ],
+  [
+    "AWS::RDS::DBSubnetGroup",
+    (spec: ExpandedPkgSpec) => {
+      const variant = spec.schemas[0].variants[0];
+
+      const prop = variant.domain.entries.find((p: ExpandedPropSpec) =>
+        p.name === "SubnetIds"
+      );
+
+      if (!prop) return;
+      const socket = createInputSocketFromProp(prop);
+
+      setAnnotationOnSocket(socket, { tokens: ["Subnet Id"] });
+      variant.sockets.push(socket);
+    },
+  ],
 ]);
 
 function addSecretProp(