refactor: store contexture info into a AsyncLocalStorage (#57)

refactor: store contexture info into a AsyncLocalStorage


Refs: #38

---------

Signed-off-by: seven <[email protected]>

Author: Blankll

src/commands/deploy.ts

@@ -1,5 +1,5 @@
 import { deployStack } from '../stack';
-import { constructActionContext, getIacLocation, logger } from '../common';
+import { getIacLocation, logger, setContext } from '../common';
 import { parseYaml } from '../parser';
 
 export const deploy = async (
@@ -19,10 +19,10 @@ export const deploy = async (
   const iac = parseYaml(getIacLocation(options.location));
   logger.info('Yaml is valid! 🎉');
 
-  const context = constructActionContext({ ...options, stackName, iacProvider: iac.provider });
+  setContext({ ...options, stackName, iacProvider: iac.provider });
 
   logger.info('Deploying stack...');
-  await deployStack(stackName, iac, context);
+  await deployStack(stackName, iac);
 
   logger.info('Stack deployed! 🎉');
 };

src/commands/destroy.ts

@@ -1,4 +1,4 @@
-import { constructActionContext, getIacLocation, logger, rosStackDelete } from '../common';
+import { getContext, getIacLocation, logger, rosStackDelete, setContext } from '../common';
 import { parseYaml } from '../parser';
 
 export const destroyStack = async (
@@ -13,7 +13,8 @@ export const destroyStack = async (
   },
 ) => {
   const iac = parseYaml(getIacLocation(options.location));
-  const context = constructActionContext({ stackName, ...options, iacProvider: iac.provider });
+  setContext({ stackName, ...options, iacProvider: iac.provider });
+  const context = getContext();
   logger.info(
     `Destroying stack: ${stackName}, provider: ${context.provider}, region: ${context.region}...`,
   );

src/commands/index.ts

@@ -1,7 +1,7 @@
 #! /usr/bin/env node
 
 import { Command } from 'commander';
-import { constructActionContext, getVersion, logger } from '../common';
+import { getContext, getVersion, logger, setContext } from '../common';
 import { validate } from './validate';
 import { deploy } from './deploy';
 import { template } from './template';
@@ -16,7 +16,8 @@ program
   .command('show')
   .description('show string')
   .action(async (options) => {
-    const context = constructActionContext({ ...options });
+    setContext({ ...options });
+    const context = getContext();
     const result = await getIamInfo(context);
     console.log('result:', JSON.stringify(result));
   });

src/commands/template.ts

@@ -1,7 +1,7 @@
 import { TemplateFormat } from '../types';
 import yaml from 'yaml';
 import { generateStackTemplate } from '../stack/deploy';
-import { constructActionContext, getIacLocation, logger } from '../common';
+import { getIacLocation, logger, setContext } from '../common';
 import { parseYaml } from '../parser';
 
 export const template = (
@@ -10,9 +10,9 @@ export const template = (
 ) => {
   const iac = parseYaml(getIacLocation(options.location));
 
-  const context = constructActionContext({ ...options, stackName, provider: iac.provider.name });
+  setContext({ ...options, stackName, provider: iac.provider.name });
 
-  const { template } = generateStackTemplate(stackName, iac, context);
+  const { template } = generateStackTemplate(stackName, iac);
   if (typeof template === 'string') {
     logger.info(`\n${template}`);
   } else {

src/commands/validate.ts

@@ -1,11 +1,12 @@
-import { constructActionContext, logger } from '../common';
+import { getContext, logger, setContext } from '../common';
 import { parseYaml } from '../parser';
 
 export const validate = (
   stackName: string,
   options: { location: string | undefined; stage: string | undefined },
 ) => {
-  const context = constructActionContext({ stackName, ...options });
+  setContext({ stackName, ...options });
+  const context = getContext();
   parseYaml(context.iacLocation);
   logger.info('Yaml is valid! 🎉');
 };

src/common/actionContext.ts renamed to src/common/context.ts

@@ -1,6 +1,9 @@
-import { ActionContext, ServerlessIac } from '../types';
+import { Context, ServerlessIac } from '../types';
 import path from 'node:path';
 import { ProviderEnum } from './providerEnum';
+import { AsyncLocalStorage } from 'node:async_hooks';
+
+const asyncLocalStorage = new AsyncLocalStorage<Context>();
 
 export const getIacLocation = (location?: string): string => {
   const projectRoot = path.resolve(process.cwd());
@@ -12,7 +15,7 @@ export const getIacLocation = (location?: string): string => {
         path.resolve(projectRoot, 'serverless-insight.yml');
 };
 
-export const constructActionContext = (config: {
+export const setContext = (config: {
   stage?: string;
   stackName?: string;
   region?: string;
@@ -23,8 +26,8 @@ export const constructActionContext = (config: {
   location?: string;
   parameters?: { [key: string]: string };
   iacProvider?: ServerlessIac['provider'];
-}): ActionContext => {
-  return {
+}): void => {
+  const context = {
     stage: config.stage ?? 'default',
     stackName: config.stackName ?? '',
     provider: (config.provider ?? config.iacProvider?.name ?? ProviderEnum.ALIYUN) as ProviderEnum,
@@ -40,4 +43,14 @@ export const constructActionContext = (config: {
     iacLocation: getIacLocation(config.location),
     parameters: Object.entries(config.parameters ?? {}).map(([key, value]) => ({ key, value })),
   };
+
+  asyncLocalStorage.enterWith(context);
+};
+
+export const getContext = (): Context => {
+  const context = asyncLocalStorage.getStore();
+  if (!context) {
+    throw new Error('No context found');
+  }
+  return context;
 };

src/common/iacHelper.ts

@@ -1,7 +1,7 @@
 import path from 'node:path';
 import fs from 'node:fs';
 import * as ros from '@alicloud/ros-cdk-core';
-import { ActionContext } from '../types';
+import { Context } from '../types';
 import * as ossDeployment from '@alicloud/ros-cdk-ossdeployment';
 import crypto from 'node:crypto';
 
@@ -34,13 +34,13 @@ export const getFileSource = (
   return { source, objectKey };
 };
 
-const evalCtx = (value: string, ctx: ActionContext): string => {
+const evalCtx = (value: string, ctx: Context): string => {
   const containsStage = value.match(/\$\{ctx.\w+}/);
 
   return containsStage ? value.replace(/\$\{ctx.stage}/g, ctx.stage) : value;
 };
 
-export const replaceReference = <T>(value: T, ctx: ActionContext): T => {
+export const replaceReference = <T>(value: T, ctx: Context): T => {
   if (typeof value === 'string') {
     const matchVar = value.match(/^\$\{vars\.(\w+)}$/);
     const containsVar = value.match(/\$\{vars\.(\w+)}/);

src/common/imsClient.ts

@@ -1,8 +1,8 @@
 import Ims20190815, * as ims20190815 from '@alicloud/ims20190815';
 import * as openApi from '@alicloud/openapi-client';
-import { ActionContext } from '../types';
+import { Context } from '../types';
 
-export const getIamInfo = async (context: ActionContext) => {
+export const getIamInfo = async (context: Context) => {
   const imsClient = new Ims20190815(
     new openApi.Config({
       accessKeyId: context.accessKeyId,

src/common/index.ts

@@ -2,7 +2,7 @@ export * from './providerEnum';
 export * from './logger';
 export * from './getVersion';
 export * from './rosClient';
-export * from './actionContext';
+export * from './context';
 export * from './iacHelper';
 export * from './constants';
 export * from './imsClient';

src/common/rosAssets.ts

@@ -4,9 +4,10 @@ import * as ossDeployment from '@alicloud/ros-cdk-ossdeployment';
 import path from 'node:path';
 import JSZip from 'jszip';
 import { logger } from './logger';
-import { ActionContext, CdkAssets } from '../types';
+import { CdkAssets } from '../types';
 import { get, isEmpty } from 'lodash';
 import OSS from 'ali-oss';
+import { getContext } from './context';
 
 const buildAssets = (rootPath: string, relativePath: string): Array<ISource> => {
   const location = path.resolve(rootPath, relativePath);
@@ -71,10 +72,11 @@ type ConstructedAsset = {
   objectKey: string;
 };
 
-export const constructAssets = async (
-  { files, rootPath }: CdkAssets,
-  region: string,
-): Promise<Array<ConstructedAsset> | undefined> => {
+export const constructAssets = async ({
+  files,
+  rootPath,
+}: CdkAssets): Promise<Array<ConstructedAsset> | undefined> => {
+  const { region } = getContext();
   const assets = await Promise.all(
     Object.entries(files)
       .filter(([, fileItem]) => !fileItem.source.path.endsWith('.template.json'))
@@ -112,15 +114,13 @@ const ensureBucketExits = async (bucketName: string, ossClient: OSS) =>
     }
   });
 
-export const publishAssets = async (
-  assets: Array<ConstructedAsset> | undefined,
-  context: ActionContext,
-) => {
+export const publishAssets = async (assets: Array<ConstructedAsset> | undefined) => {
   if (!assets?.length) {
     logger.info('No assets to publish, skipped!');
     return;
   }
 
+  const context = getContext();
   const bucketName = assets[0].bucketName;
 
   const client = new OSS({
@@ -148,15 +148,12 @@ export const publishAssets = async (
   return bucketName;
 };
 
-export const cleanupAssets = async (
-  assets: Array<ConstructedAsset> | undefined,
-  context: ActionContext,
-) => {
+export const cleanupAssets = async (assets: Array<ConstructedAsset> | undefined) => {
   if (!assets?.length) {
     logger.info('No assets to cleanup, skipped!');
     return;
   }
-
+  const context = getContext();
   const bucketName = assets[0].bucketName;
 
   const client = new OSS({