Emit Signal method and delegate to provide a hook into events

Author: George Ndungu

powershell/cmdlets/class.ts

@@ -1023,6 +1023,11 @@ export class CmdletClass extends Class {
         yield `await ${$this.state.project.serviceNamespace.moduleClass.declaration}.Instance.Signal(${id.value}, ${token.value}, ${messageData.value}, (i,t,m) => ((${ClientRuntime.IEventListener})this).Signal(i,t,()=> ${ClientRuntime.EventDataConverter}.ConvertFrom( m() ) as ${ClientRuntime.EventData} ), ${$this.invocationInfo.value}, this.ParameterSetName, ${$this.correlationId.value}, ${$this.processRecordId.value}, null );`;
         yield If(`${token.value}.IsCancellationRequested`, Return());
       }
+      else {
+        // In Non-Azure Modes, emit the Signal method without coorelation and processrecordid
+        yield `await ${$this.state.project.serviceNamespace.moduleClass.declaration}.Instance.Signal(${id.value}, ${token.value}, ${messageData.value}, (i,t,m) => ((${ClientRuntime.IEventListener})this).Signal(i,t,()=> ${ClientRuntime.EventDataConverter}.ConvertFrom( m() ) as ${ClientRuntime.EventData} ), ${$this.invocationInfo.value}, this.ParameterSetName, null );`;
+        yield If(`${token.value}.IsCancellationRequested`, Return());
+      }
       yield `WriteDebug($"{id}: {(messageData().Message ?? ${System.String.Empty})}");`;
       // any handling of the signal on our side...
     });

powershell/module/module-class.ts

@@ -105,6 +105,7 @@ export class ModuleClass extends Class {
 
         yield $this.fPipeline.assignPrivate(ClientRuntime.HttpPipeline.new(ClientRuntime.HttpClientFactory.new(System.Net.Http.HttpClient.new())));
         yield $this.fPipelineWithProxy.assignPrivate(ClientRuntime.HttpPipeline.new(ClientRuntime.HttpClientFactory.new(System.Net.Http.HttpClient.new($this.fHandler))));
+        yield "Init();";
       }
     }));
 
@@ -113,6 +114,7 @@ export class ModuleClass extends Class {
     this.add(new PartialMethod('AfterCreatePipeline', dotnet.Void, { parameters: [this.pInvocationInfo, this.pPipeline] }));
     this.add(new PartialMethod('CustomInit', dotnet.Void));
 
+
     /* Setting the Proxy */
     this.add(new Method('SetProxyConfiguration', dotnet.Void, {
       parameters: [this.pProxy, this.pProxyCredential, this.pUseDefaultCredentials],
@@ -128,8 +130,23 @@ export class ModuleClass extends Class {
   }
 
   createInitAndPipeline(namespace: Namespace) {
+
     const $this = this;
-    // non-azure init method
+    // Custom Event Listener without Azure Spefic concepts. (ProcessId and CorelationId)
+    const customEventListenerFunc = System.Func(
+      dotnet.String,
+      System.Threading.CancellationToken,
+      System.Func(System.EventArgs),
+      this.incomingSignalFunc,
+      InvocationInfo,
+      dotnet.String,
+      System.Exception,
+    /* returns */ System.Threading.Tasks.Task());
+
+    const incomingSignalDelegate = namespace.add(new Alias('SignalDelegate', this.incomingSignalFunc));
+    const eventListenerDelegate = namespace.add(new Alias('EventListenerDelegate', customEventListenerFunc));
+    const EventListener = this.add(new Property('EventListener', eventListenerDelegate, { description: 'A delegate that gets called for each signalled event' }));
+
     this.initMethod.add(function* () {
       yield '// called at module init time...';
       yield 'CustomInit();';
@@ -152,6 +169,20 @@ export class ModuleClass extends Class {
     });
 
     this.add(new LambdaProperty('Name', dotnet.String, new StringExpression(this.state.project.moduleName), { description: 'The Name of this module ' }));
+
+    // Add Signal extensibility point
+    const pSignal = new Parameter('signal', incomingSignalDelegate, { description: 'The callback for the event dispatcher ' });
+    // Emit signal extensibility points that called EventListenerDelegate, allowing us to handle Signals emitted by the Pipeline in the Auth Module
+    const signalImpl = this.add(new Method('Signal', System.Threading.Tasks.Task(), {
+      parameters: [this.pId, this.pToken, this.pGetEventData, pSignal, this.pInvocationInfo, this.pParameterSetName, this.pException], async: Modifier.Async,
+      description: 'Called to dispatch events to the common module listener',
+      returnsDescription: `A <see cref="${System.Threading.Tasks.Task()}" /> that will be complete when handling of the event is completed.`
+    }));
+
+    signalImpl.push(Using('NoSynchronizationContext', ''));
+    signalImpl.add(function* () {
+      yield `await ${EventListener.value}?.Invoke(${$this.pId.value},${$this.pToken.value},${$this.pGetEventData.value}, ${pSignal.value}, ${$this.pInvocationInfo}, ${$this.pParameterSetName},${$this.pException});`;
+    });
   }
 
   createAzureInitAndPipeline(namespace: Namespace) {