powershell/module/module-class.ts (329 lines of code) (raw):
/*---------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
import { Access, Alias, Class, ClassType, Constructor, dotnet, Field, ImportDirective, If, LambdaMethod, LambdaProperty, LazyProperty, LiteralExpression, LocalVariable, MemberVariable, Method, Modifier, Namespace, Parameter, ParameterModifier, PartialMethod, Property, Return, Statements, StringExpression, System, TypeDeclaration, Using, valueOf, Variable, Else } from '@azure-tools/codegen-csharp';
import { InvocationInfo, PSCredential, IArgumentCompleter, CompletionResult, CommandAst, CompletionResultType, PSCmdlet, } from '../internal/powershell-declarations';
import { State } from '../internal/state';
import { ClientRuntime } from '../llcsharp/exports';
import { DeepPartial } from '@azure-tools/codegen';
export class NewModuleClass extends Class {
// get the name of the client API class
TaskOfHttpResponseMessage = System.Threading.Tasks.Task(System.Net.Http.HttpResponseMessage);
// lets the common code call the signal again (recursive! careful!)
incomingSignalFunc = System.Func(
dotnet.String,
System.Threading.CancellationToken,
System.Func(System.EventArgs),
/* returns */ System.Threading.Tasks.Task());
eventListenerFunc = System.Func(
dotnet.String,
System.Threading.CancellationToken,
System.Func(System.EventArgs),
this.incomingSignalFunc,
InvocationInfo,
dotnet.String,
dotnet.String,
dotnet.String,
System.Exception,
/* returns */ System.Threading.Tasks.Task());
IEventListenerExpanded = [
System.Threading.CancellationToken, /* token */
System.Action(), /* Cancel() */
this.incomingSignalFunc,
];
nextStep = System.Func(
System.Net.Http.HttpRequestMessage,
...this.IEventListenerExpanded,
/* returns */ this.TaskOfHttpResponseMessage);
initMethod = this.add(new Method('Init', dotnet.Void, { description: 'Initialization steps performed after the module is loaded.' }));
createPipelineMethod!: Method;
pInvocationInfo = new Parameter('invocationInfo', InvocationInfo, { description: 'The <see cref="System.Management.Automation.InvocationInfo" /> from the cmdlet' });
pPipeline = new Parameter('pipeline', ClientRuntime.HttpPipeline, { modifier: ParameterModifier.Ref, description: 'The HttpPipeline for the request' });
pProxy = new Parameter('proxy', System.Uri, { description: 'The HTTP Proxy to use.' });
pProxyCredential = new Parameter('proxyCredential', PSCredential, { description: 'The HTTP Proxy Credentials' });
pUseDefaultCredentials = new Parameter('proxyUseDefaultCredentials', dotnet.Bool, { description: 'True if the proxy should use default credentials' });
pCorrelationId = new Parameter('correlationId', dotnet.String, { description: 'the cmdlet\'s correlation id.' });
pParameterName = new Parameter('parameterName', dotnet.String, { description: 'The name of the parameter to get the value for.' });
pId = new Parameter('id', dotnet.String, { description: 'The ID of the event ' });
pToken = new Parameter('token', System.Threading.CancellationToken, { description: 'The cancellation token for the event ' });
pGetEventData = new Parameter('getEventData', System.Func(System.EventArgs), { description: 'A delegate to get the detailed event data' });
pParameterSetName = new Parameter('parameterSetName', dotnet.String, { description: 'the cmdlet\'s parameterset name.' });
pParameterSetNameWithDefault = new Parameter('parameterSetName', dotnet.String, { description: 'the cmdlet\'s parameterset name.', defaultInitializer: dotnet.Null });
pExtensibleParameters = new Parameter('extensibleParameters', System.Collections.Generic.IDictionary(dotnet.String, dotnet.Object), { description: 'a dict for extensible parameters', defaultInitializer: dotnet.Null });
pProcessRecordId = new Parameter('processRecordId', dotnet.String, { description: 'the cmdlet\'s process record correlation id.' });
pException = new Parameter('exception', System.Exception, { description: 'the exception that is being thrown (if available)' });
fPipeline = this.add(new Field('_pipeline', ClientRuntime.HttpPipeline, { access: Access.Private, description: 'the ISendAsync pipeline instance' }));
fPipelineWithProxy = this.add(new Field('_pipelineWithProxy', ClientRuntime.HttpPipeline, { access: Access.Private, description: 'the ISendAsync pipeline instance (when proxy is enabled)' }));
fHandler = this.add(new Field('_handler', System.Net.Http.HttpClientHandler, { initialValue: System.Net.Http.HttpClientHandler.new() }));
fWebProxy = this.add(new Field('_webProxy', System.Net.WebProxy, { initialValue: System.Net.WebProxy.new() }));
fUseProxy = this.add(new Field('_useProxy', dotnet.Bool, { initialValue: dotnet.False }));
constructor(namespace: Namespace, private readonly state: State, objectInitializer?: DeepPartial<NewModuleClass>) {
super(namespace, 'Module');
this.apply(objectInitializer);
this.partial = true;
this.description = 'A class that contains the module-common code and data.';
const $this = this;
// Lock for the singleton
const singletonLock = new Field('_singletonLock', System.Object, { readonly: Modifier.ReadOnly, access: Access.Private, static: Modifier.Static, initialValue: System.Object.new() });
this.add(singletonLock);
const initLock = new Field('_initLock', System.Object, { readonly: Modifier.ReadOnly, access: Access.Private, static: Modifier.Static, initialValue: System.Object.new() });
this.add(initLock);
const fInstance = new Field('_instance', this, { access: Access.Private, static: Modifier.Static });
const getInstanceFunc = `if (${fInstance.name} == null) { lock (${singletonLock.name}) { if (${fInstance.name} == null) { ${fInstance.name} = new Module(); }}} return ${fInstance.name};`;
this.add(fInstance);
this.add(new Field('_init', dotnet.Bool, { access: Access.Private, static: Modifier.Static, initialValue: dotnet.False }));
this.add(new Property('Instance', this, {
getAccess: Access.Public,
static: Modifier.Static,
get: getInstanceFunc,
description: 'the singleton of this module class'
}));
const clientAPI = new ClassType(this.state.model.language.csharp?.namespace, this.state.model.language.csharp?.name || '');
const clientProperty = this.add(new Property('ClientAPI', clientAPI, { description: 'The instance of the Client API' }));
if (this.state.project.azure) {
this.createAzureInitAndPipeline(namespace);
} else {
this.createInitAndPipeline(namespace);
}
this.add(new Constructor(this, {
access: Access.Private,
description: 'Creates the module instance.',
body: function* () {
yield '// constructor';
yield clientProperty.assignPrivate(clientAPI.new());
yield `${$this.fHandler}.Proxy = ${$this.fWebProxy};`;
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))));
}
}));
/* extensibility points */
this.add(new PartialMethod('BeforeCreatePipeline', dotnet.Void, { parameters: [this.pInvocationInfo, this.pPipeline] }));
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],
*body() {
yield `${$this.fUseProxy} = proxy != null;`;
yield If(`${$this.pProxy} == null`, function* () {
yield 'return;';
});
yield '// set the proxy configuration';
yield `${$this.fWebProxy}.Address = proxy;`;
yield `${$this.fWebProxy}.BypassProxyOnLocal = false;`;
yield If(`${$this.pUseDefaultCredentials}`, function* () {
yield `${$this.fWebProxy}.Credentials = null;`;
yield `${$this.fWebProxy}.UseDefaultCredentials = true;`;
});
yield Else(function* () {
yield `${$this.fWebProxy}.UseDefaultCredentials = false;`;
yield `${$this.fWebProxy}.Credentials = proxyCredential ?.GetNetworkCredential();`;
});
}
}));
}
createInitAndPipeline(namespace: Namespace) {
const $this = this;
// 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' }));
// non-azure init method
this.initMethod.add(function* () {
yield '// called at module init time...';
yield If('_init == false', function* () {
yield `lock (_initLock) {
if (_init == false) {
CustomInit();
_init = true;
}
}`;
});
});
this.createPipelineMethod = this.add(new Method('CreatePipeline', ClientRuntime.HttpPipeline, {
parameters: [this.pInvocationInfo, this.pParameterSetNameWithDefault, this.pExtensibleParameters],
description: 'Creates an instance of the HttpPipeline for each call.',
returnsDescription: `An instance of ${ClientRuntime.HttpPipeline} for the remote call.`
}));
// non-azure createPipeline method
this.createPipelineMethod.add(function* () {
const pip = new LocalVariable('pipeline', ClientRuntime.HttpPipeline, { initializer: 'null' });
yield pip.declarationStatement;
yield `BeforeCreatePipeline(${$this.pInvocationInfo.use}, ref ${pip});`;
yield pip.assign(`(${pip} ?? (${$this.fUseProxy} ? ${$this.fPipelineWithProxy} : ${$this.fPipeline})).Clone()`);
yield `AfterCreatePipeline(${$this.pInvocationInfo.use}, ref ${pip});`;
yield Return(pip);
});
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* () {
// Emit call to EventListener after explicit null check.
// Not using Null-Conditional operator causes Null Reference exception when Func<Task> is null, due to awaiting null Task.
yield If(`${EventListener.value} != null`, `await ${EventListener.value}.Invoke(${$this.pId.value},${$this.pToken.value},${$this.pGetEventData.value}, ${pSignal.value}, ${$this.pInvocationInfo}, ${$this.pParameterSetName},${$this.pException});`);
});
}
createAzureInitAndPipeline(namespace: Namespace) {
const $this = this;
const sendAsyncStep = namespace.add(new Alias('SendAsyncStepDelegate',
System.Func(
System.Net.Http.HttpRequestMessage,
...this.IEventListenerExpanded,
this.nextStep, /* Next( ...) */
/* returns */ this.TaskOfHttpResponseMessage)));
const isDataPlane = !!this.state.project.endpointResourceIdKeyName;
const pipelineChangeDelegate = namespace.add(new Alias('PipelineChangeDelegate', System.Action(sendAsyncStep.fullDefinition)));
const getParameterDelegate = namespace.add(new Alias('GetParameterDelegate', System.Func(
dotnet.String, /* resourceId */
dotnet.String, /* moduleName */
InvocationInfo, /* invocationInfo */
dotnet.String, /* correlationId */
dotnet.String, /* parameterName */
/* returns */ dotnet.Object)));
const moduleLoadPipelineDelegate = namespace.add(new Alias('ModuleLoadPipelineDelegate', System.Action(
dotnet.String, /* resourceId */
dotnet.String, /* moduleName */
pipelineChangeDelegate.fullDefinition, /* prependStep */
pipelineChangeDelegate.fullDefinition))); /* appendStep */
const newRequestPipelineDelegate = new Alias('NewRequestPipelineDelegate', System.Action(
InvocationInfo, /* invocationInfo */
dotnet.String, /* correlationId */
dotnet.String, /* processRecordId */
pipelineChangeDelegate.fullDefinition, /* prependStep */
pipelineChangeDelegate.fullDefinition)); /* appendStep */
const argumentCompleterDelegate = namespace.add(new Alias('ArgumentCompleterDelegate', System.Func(
dotnet.String, /* completerName */
InvocationInfo, /* invocationInfo */
dotnet.String, /* correlationId */
dotnet.StringArray, /* resourceTypes */
dotnet.StringArray, /* parentResourceParameterNames */
/* returns */ dotnet.StringArray,
)));
const getTelemetryIdDelegate = namespace.add(new Alias('GetTelemetryIdDelegate', System.Func(
/* returns */ dotnet.String,
)));
const telemetryDelegate = namespace.add(new Alias('TelemetryDelegate', System.Action(
dotnet.String, /* operation name */
InvocationInfo, /* invocationInfo */
dotnet.String, /* parameter set name */
PSCmdlet, /* pscmdlet */
)));
const sanitizerDelegate = new Alias('SanitizerDelegate', System.Action(
dotnet.Object, /* sendToPipeline */
dotnet.String /* telemetryId */
));
const getTelemetryInfoDelegate = new Alias('GetTelemetryInfoDelegate', System.Func(
dotnet.String /* telemetryId */,
/* returns */ System.Collections.Generic.Dictionary(System.String, System.String)
));
const tokenAudienceConverterDelegate = new Alias('TokenAudienceConverterDelegate',
System.Func(
dotnet.String,
dotnet.String,
dotnet.String,
dotnet.String,
System.Uri,
/* returns */ dotnet.String));
const authorizeRequestDelegate = new Alias('AuthorizeRequestDelegate',
System.Action(
InvocationInfo,
dotnet.String,
dotnet.String,
System.Action(sendAsyncStep.fullDefinition),
System.Action(sendAsyncStep.fullDefinition),
tokenAudienceConverterDelegate.fullDefinition,
System.Collections.Generic.IDictionary(dotnet.String, dotnet.Object)
));
if (isDataPlane) {
namespace.add(tokenAudienceConverterDelegate);
namespace.add(authorizeRequestDelegate);
namespace.add(new ImportDirective('System.Collections.Generic'));
}
namespace.add(newRequestPipelineDelegate);
const incomingSignalDelegate = namespace.add(new Alias('SignalDelegate', this.incomingSignalFunc));
const eventListenerDelegate = namespace.add(new Alias('EventListenerDelegate', this.eventListenerFunc));
namespace.add(new Alias('NextDelegate', this.nextStep));
/* AzAccounts VTable properties */
const OnModuleLoad = this.add(new Property('OnModuleLoad', moduleLoadPipelineDelegate, { description: 'The delegate to call when this module is loaded (supporting a commmon module).' }));
const OnNewRequest = new Property('OnNewRequest', newRequestPipelineDelegate, { description: 'The delegate to call before each new request (supporting a commmon module).' });
const AddRequestUserAgentHandler = new Property('AddRequestUserAgentHandler', newRequestPipelineDelegate, { description: 'The delegate to call before each new request to add request user agent.' });
const AddPatchRequestUriHandler = new Property('AddPatchRequestUriHandler', newRequestPipelineDelegate, { description: 'The delegate to call before each new request to patch request uri.' });
const AddAuthorizeRequestHandler = new Property('AddAuthorizeRequestHandler', authorizeRequestDelegate, { description: 'The delegate to call before each new request to add authorization.' });
this.add(new Property('GetTelemetryId', getTelemetryIdDelegate, { description: 'The delegate to get the telemetry Id.' }));
this.add(new Property('Telemetry', telemetryDelegate, { description: 'The delegate for creating a telemetry.' }));
const SanitizeOutput = this.add(new Property('SanitizeOutput', sanitizerDelegate, { description: 'The delegate to call in WriteObject to sanitize the output object.' }));
namespace.add(sanitizerDelegate);
const GetTelemetryInfo = this.add(new Property('GetTelemetryInfo', getTelemetryInfoDelegate, { description: 'The delegate to get the telemetry info.' }));
namespace.add(getTelemetryInfoDelegate);
if (isDataPlane) {
this.add(AddRequestUserAgentHandler);
this.add(AddPatchRequestUriHandler);
this.add(AddAuthorizeRequestHandler);
} else {
this.add(OnNewRequest);
}
const GetParameterValue = this.add(new Property('GetParameterValue', getParameterDelegate, { description: 'The delegate to call to get parameter data from a common module.' }));
const EventListener = this.add(new Property('EventListener', eventListenerDelegate, { description: 'A delegate that gets called for each signalled event' }));
const ArgumentCompleter = this.add(new Property('ArgumentCompleter', argumentCompleterDelegate, { description: 'Gets completion data for azure specific fields' }));
const ProfileName = this.add(new Property('ProfileName', System.String, { description: 'The name of the currently selected Azure profile' }));
const moduleIdentity = this.add(new LambdaProperty('Name', dotnet.String, new StringExpression(this.state.project.moduleName), { description: 'The Name of this module ' }));
const currentProfile = this.add(new Field('Profile', dotnet.String, { initialValue: System.String.Empty, description: 'The currently selected profile.' }));
const moduleResourceId = this.add(new LambdaProperty('ResourceId', dotnet.String, new StringExpression(this.state.project.moduleName), { description: 'The ResourceID for this module (azure arm).' }));
/* get parameter method (calls azAccounts) */
this.add(new LambdaMethod('GetParameter', dotnet.Object, new LiteralExpression(`${GetParameterValue.value}?.Invoke( ${moduleResourceId.value}, ${moduleIdentity.value}, ${$this.pInvocationInfo.value}, ${$this.pCorrelationId.value},${$this.pParameterName.value} )`), {
parameters: [this.pInvocationInfo, this.pCorrelationId, this.pParameterName],
description: 'Gets parameters from a common module.',
returnsDescription: 'The parameter value from the common module. (Note: this should be type converted on the way back)'
}));
/* signal method (calls azAccounts) */
const pSignal = new Parameter('signal', incomingSignalDelegate, { description: 'The callback for the event dispatcher ' });
const signalImpl = this.add(new Method('Signal', System.Threading.Tasks.Task(), {
parameters: [this.pId, this.pToken, this.pGetEventData, pSignal, this.pInvocationInfo, this.pParameterSetName, this.pCorrelationId, this.pProcessRecordId, 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.pCorrelationId},${$this.pProcessRecordId},${$this.pException});`;
});
/* init method */
this.initMethod.add(function* () {
yield If('_init == false', function* () {
yield `lock (_initLock) {
if (_init == false) {
${OnModuleLoad.value}?.Invoke( ${moduleResourceId.value}, ${moduleIdentity.value} ,(step)=> { ${$this.fPipeline.value}.Prepend(step); } , (step)=> { ${$this.fPipeline.value}.Append(step); } );
${OnModuleLoad.value}?.Invoke( ${moduleResourceId.value}, ${moduleIdentity.value} ,(step)=> { ${$this.fPipelineWithProxy.value}.Prepend(step); } , (step)=> { ${$this.fPipelineWithProxy.value}.Append(step); } );
CustomInit();
_init = true;
}
}`;
});
});
this.createPipelineMethod = this.add(new Method('CreatePipeline', ClientRuntime.HttpPipeline, {
parameters: [this.pInvocationInfo, this.pCorrelationId, this.pProcessRecordId, this.pParameterSetNameWithDefault, this.pExtensibleParameters],
description: 'Creates an instance of the HttpPipeline for each call.',
returnsDescription: `An instance of ${ClientRuntime.HttpPipeline} for the remote call.`
}));
/* Add following three fields for data plane */
const fendpointResourceIdKeyName = new Field('_endpointResourceIdKeyName', dotnet.String, { access: Access.Private, initialValue: new StringExpression(this.state.project.endpointResourceIdKeyName) });
const fEndpointSuffixKeyName = new Field('_endpointSuffixKeyName', dotnet.String, { access: Access.Private, initialValue: new StringExpression(this.state.project.endpointSuffixKeyName) });
const fTokenAudienceConverter = new Field('_tokenAudienceConverter', tokenAudienceConverterDelegate, { access: Access.Private, initialValue: 'null' });
if (isDataPlane) {
this.add(fendpointResourceIdKeyName);
this.add(fEndpointSuffixKeyName);
this.add(fTokenAudienceConverter);
}
/* pipeline create method */
this.createPipelineMethod.add(function* () {
const pip = new LocalVariable('pipeline', ClientRuntime.HttpPipeline, { initializer: 'null' });
yield pip.declarationStatement;
yield `BeforeCreatePipeline(${$this.pInvocationInfo.use}, ref ${pip});`;
yield pip.assign(`(${pip} ?? (${$this.fUseProxy} ? ${$this.fPipelineWithProxy} : ${$this.fPipeline})).Clone()`);
yield `AfterCreatePipeline(${$this.pInvocationInfo.use}, ref ${pip});`;
yield `pipeline.Append(new Runtime.CmdInfoHandler(${$this.pProcessRecordId}, ${$this.pInvocationInfo.use}, ${$this.pParameterSetName}).SendAsync);`;
if (isDataPlane) {
yield `${AddRequestUserAgentHandler.value}?.Invoke( ${$this.pInvocationInfo.use}, ${$this.pCorrelationId},${$this.pProcessRecordId}, (step)=> { ${pip}.Prepend(step); } , (step)=> { ${pip}.Append(step); } );`;
yield `${AddPatchRequestUriHandler.value}?.Invoke( ${$this.pInvocationInfo.use}, ${$this.pCorrelationId},${$this.pProcessRecordId}, (step)=> { ${pip}.Prepend(step); } , (step)=> { ${pip}.Append(step); } );`;
yield `${AddAuthorizeRequestHandler.value}?.Invoke( ${$this.pInvocationInfo.use}, ${fendpointResourceIdKeyName},${fEndpointSuffixKeyName}, (step)=> { ${pip}.Prepend(step); } , (step)=> { ${pip}.Append(step); }, ${fTokenAudienceConverter}, ${$this.pExtensibleParameters} );`;
} else {
yield `${OnNewRequest.value}?.Invoke( ${$this.pInvocationInfo.use}, ${$this.pCorrelationId},${$this.pProcessRecordId}, (step)=> { ${pip}.Prepend(step); } , (step)=> { ${pip}.Append(step); } );`;
}
yield Return(pip);
});
}
}