Initial commit

src/augmentationRegistry.ts

@@ -0,0 +1,122 @@
+/**
+ * Augmentation Registry
+ *
+ * This module provides a registry for augmentations that are loaded at build time.
+ * It replaces the dynamic loading mechanism in pluginLoader.ts.
+ */
+
+import { IPipeline } from './types/pipelineTypes.js'
+import { AugmentationType, IAugmentation } from './types/augmentations.js'
+
+// Forward declaration of the pipeline instance to avoid circular dependency
+// The actual pipeline will be provided when initializeAugmentationPipeline is called
+let defaultPipeline: IPipeline | null = null
+
+/**
+ * Sets the default pipeline instance
+ * This function should be called from pipeline.ts after the pipeline is created
+ */
+export function setDefaultPipeline(pipeline: IPipeline): void {
+  defaultPipeline = pipeline
+}
+
+/**
+ * Registry of all available augmentations
+ */
+export const availableAugmentations: IAugmentation[] = []
+
+/**
+ * Registers an augmentation with the registry
+ *
+ * @param augmentation The augmentation to register
+ * @returns The augmentation that was registered
+ */
+export function registerAugmentation<T extends IAugmentation>(augmentation: T): T {
+  // Set enabled to true by default if not specified
+  if (augmentation.enabled === undefined) {
+    augmentation.enabled = true
+  }
+
+  // Add to the registry
+  availableAugmentations.push(augmentation)
+
+  return augmentation
+}
+
+/**
+ * Initializes the augmentation pipeline with all registered augmentations
+ *
+ * @param pipeline Optional custom pipeline to use instead of the default
+ * @returns The pipeline that was initialized
+ * @throws Error if no pipeline is provided and the default pipeline hasn't been set
+ */
+export function initializeAugmentationPipeline(
+  pipelineInstance?: IPipeline
+): IPipeline {
+  // Use the provided pipeline or fall back to the default
+  const pipeline = pipelineInstance || defaultPipeline
+
+  if (!pipeline) {
+    throw new Error('No pipeline provided and default pipeline not set. Call setDefaultPipeline first.')
+  }
+
+  // Register all augmentations with the pipeline
+  for (const augmentation of availableAugmentations) {
+    if (augmentation.enabled) {
+      pipeline.register(augmentation)
+    }
+  }
+
+  return pipeline
+}
+
+/**
+ * Enables or disables an augmentation by name
+ *
+ * @param name The name of the augmentation to enable/disable
+ * @param enabled Whether to enable or disable the augmentation
+ * @returns True if the augmentation was found and updated, false otherwise
+ */
+export function setAugmentationEnabled(name: string, enabled: boolean): boolean {
+  const augmentation = availableAugmentations.find(aug => aug.name === name)
+
+  if (augmentation) {
+    augmentation.enabled = enabled
+    return true
+  }
+
+  return false
+}
+
+/**
+ * Gets all augmentations of a specific type
+ *
+ * @param type The type of augmentation to get
+ * @returns An array of all augmentations of the specified type
+ */
+export function getAugmentationsByType(type: AugmentationType): IAugmentation[] {
+  return availableAugmentations.filter(aug => {
+    // Check if the augmentation is of the specified type
+    // This is a simplified check and may need to be updated based on how types are determined
+    switch (type) {
+      case AugmentationType.SENSE:
+        return 'processRawData' in aug && 'listenToFeed' in aug
+      case AugmentationType.CONDUIT:
+        return 'establishConnection' in aug && 'readData' in aug && 'writeData' in aug
+      case AugmentationType.COGNITION:
+        return 'reason' in aug && 'infer' in aug && 'executeLogic' in aug
+      case AugmentationType.MEMORY:
+        return 'storeData' in aug && 'retrieveData' in aug && 'updateData' in aug
+      case AugmentationType.PERCEPTION:
+        return 'interpret' in aug && 'organize' in aug && 'generateVisualization' in aug
+      case AugmentationType.DIALOG:
+        return 'processUserInput' in aug && 'generateResponse' in aug && 'manageContext' in aug
+      case AugmentationType.ACTIVATION:
+        return 'triggerAction' in aug && 'generateOutput' in aug && 'interactExternal' in aug
+      case AugmentationType.WEBSOCKET:
+        return 'connectWebSocket' in aug && 'sendWebSocketMessage' in aug && 'onWebSocketMessage' in aug
+      default:
+        return false
+    }
+  })
+}