feat: registerXrdModel

Author: devthejo

packages/sdk/src/Model/index.ts

@@ -13,10 +13,79 @@ type Status = {
   [key: string]: any
 }
 
+// XRD Model Registry types
+export interface XrdModelRegistryEntry {
+  modelClass: typeof Model
+  group: string
+  kind: string
+}
+
+export interface XrdModelRegistry {
+  [key: string]: XrdModelRegistryEntry
+}
+
+// Global registry for XRD models
+const xrdModelRegistry: XrdModelRegistry = {}
+
+/**
+ * Register an XRD model class with the global registry
+ * @param group - The API group (e.g., "workspace.fabrique.social.gouv.fr")
+ * @param kind - The resource kind (e.g., "XRedis")
+ * @returns Class decorator function
+ */
+export function registerXrdModel(group: string, kind: string) {
+  return function <T extends new (...args: any[]) => Model<any>>(target: T): T {
+    const registryKey = `${group}/${kind}`
+
+    xrdModelRegistry[registryKey] = {
+      modelClass: target as any,
+      group,
+      kind,
+    }
+
+    return target
+  }
+}
+
+/**
+ * Get a registered XRD model class by group and kind
+ * @param group - The API group
+ * @param kind - The resource kind
+ * @returns The registered model class or undefined if not found
+ */
+export function getRegisteredXrdModel(group: string, kind: string): typeof Model | undefined {
+  const registryKey = `${group}/${kind}`
+  return xrdModelRegistry[registryKey]?.modelClass
+}
+
+/**
+ * Get a registered XRD model class by apiVersion and kind
+ * @param apiVersion - The full API version (e.g., "workspace.fabrique.social.gouv.fr/v1alpha1")
+ * @param kind - The resource kind
+ * @returns The registered model class or undefined if not found
+ */
+export function getRegisteredXrdModelByApiVersion(
+  apiVersion: string,
+  kind: string
+): typeof Model | undefined {
+  // Extract group from apiVersion
+  // If no '/' is present, use "core" as convention for core resources
+  const group = apiVersion.includes("/") ? apiVersion.split("/")[0] : "core"
+  return getRegisteredXrdModel(group, kind)
+}
+
+/**
+ * Get all registered XRD models
+ * @returns The complete registry
+ */
+export function getXrdModelRegistry(): XrdModelRegistry {
+  return { ...xrdModelRegistry }
+}
+
 export class Model<T> extends BaseModel<T> {
   getMetadata(): IObjectMeta {
     const self = this as any
-    if (!self.metadatata) {
+    if (!self.metadata) {
       throw new Error("No metadata found")
     }
     return self.metadata

packages/server/src/executor.ts

@@ -1,6 +1,7 @@
 import fs from "fs"
 
 import { createLogger } from "@crossplane-js/libs"
+import { getRegisteredXrdModelByApiVersion } from "@crossplane-js/sdk"
 
 import type { NodeResponse, NodeError, FunctionInput } from "./types"
 
@@ -68,7 +69,37 @@ export async function executeCode(
 
       let result
       try {
-        result = await module.default(input)
+        let composite: any = input
+
+        // Try to instantiate using registered XRD model if available
+        try {
+          const inputData = input as any
+          const compositeResource = inputData?.observed?.composite?.resource
+
+          if (compositeResource?.apiVersion && compositeResource?.kind) {
+            const RegisteredModelClass = getRegisteredXrdModelByApiVersion(
+              compositeResource.apiVersion,
+              compositeResource.kind
+            )
+
+            if (RegisteredModelClass) {
+              moduleLogger.debug(`Using registered XRD model for ${compositeResource.kind}`)
+              composite = new RegisteredModelClass(compositeResource)
+            } else {
+              moduleLogger.debug(
+                `No registered XRD model found for ${compositeResource.kind}, using raw input`
+              )
+            }
+          }
+        } catch (modelErr) {
+          moduleLogger.debug(
+            `Failed to instantiate XRD model, falling back to raw input: ${(modelErr as Error).message}`
+          )
+          // Fall back to raw input if model instantiation fails
+          composite = input
+        }
+
+        result = await module.default(composite)
         moduleLogger.debug("Function execution completed")
       } catch (execErr) {
         moduleLogger.error(`Error during function execution: ${(execErr as Error).message}`)