1818
1919const fs = require ( 'fs' ) ;
2020const env = require ( 'env-var' ) ;
21- import * as cdk from '@ aws-cdk/core ' ;
21+ import * as cdk from 'aws-cdk-lib ' ;
2222
23+ import { AppConfig } from './app-config' ;
2324import { StackCommonProps } from './stack/base/base-stack' ;
2425
26+
27+ export class AppContextError extends Error {
28+ constructor ( message : string ) {
29+ super ( message ) ;
30+ this . name = "AppConfigFileFailError" ;
31+ }
32+ }
33+
34+ export enum ProjectPrefixType {
35+ NameStage ,
36+ NameHyphenStage ,
37+ Name
38+ }
39+
2540export interface AppContextProps {
26- appConfigEnvName : string ;
27- projectPrefix ?: string ;
41+ appConfigFileKey : string ;
2842 contextArgs ?: string [ ] ;
43+ projectPrefixType ?: ProjectPrefixType ;
2944}
3045
3146export class AppContext {
3247 public readonly cdkApp : cdk . App ;
33- public readonly appConfig : any | undefined ;
34- public readonly stackCommonProps : StackCommonProps | undefined ;
48+ public readonly appConfig : AppConfig ;
49+ public readonly stackCommonProps : StackCommonProps ;
3550
36- private projectPrefix : string | undefined ;
37- private infraConfigPath : string | undefined ;
51+ private readonly appContextProps : AppContextProps ;
3852
3953 constructor ( props : AppContextProps ) {
4054 this . cdkApp = new cdk . App ( ) ;
55+ this . appContextProps = props ;
56+
57+ try {
58+ const appConfigFile = this . findAppConfigFile ( props . appConfigFileKey ) ;
4159
42- this . projectPrefix = props . projectPrefix ;
60+ this . appConfig = this . loadAppConfigFile ( appConfigFile , props . contextArgs ) ;
4361
44- this . appConfig = this . loadConfig ( this . cdkApp , props . appConfigEnvName , props . contextArgs ) ;
62+ if ( this . appConfig != undefined ) {
63+ this . stackCommonProps = this . createStackCommonProps ( appConfigFile ) ;
64+ }
4565
46- if ( this . appConfig != undefined ) {
47- this . stackCommonProps = this . createStackProps ( ) ;
66+ } catch ( e ) {
67+ console . error ( `==> CDK App-Config File is empty,
68+ set up your environment variable(Usage: export ${ props . appConfigFileKey }
69+ or append inline-argurment(Usage: cdk list --context ${ props . appConfigFileKey } , e ) ;
70+ throw new AppContextError ( 'Fail to find App-Config json file' ) ;
4871 }
4972 }
5073
51- private createStackProps ( ) : StackCommonProps {
74+ public ready ( ) : boolean {
75+ return this . stackCommonProps ? true : false ;
76+ }
77+
78+ private createStackCommonProps ( appConfigFile : string ) : StackCommonProps {
5279 const stackProps : StackCommonProps = {
53- projectPrefix : this . projectPrefix ! ,
80+ projectPrefix : this . getProjectPrefix ( this . appConfig . Project . Name , this . appConfig . Project . Stage ) ,
5481 appConfig : this . appConfig ,
55- appConfigPath : this . infraConfigPath ! ,
82+ appConfigPath : appConfigFile ,
5683 env : {
5784 account : this . appConfig . Project . Account ,
5885 region : this . appConfig . Project . Region
5986 } ,
60- variable : { }
87+ variables : { }
6188 }
6289
6390 return stackProps ;
6491 }
6592
66- // must contain infra/config/xxxx.json
67- private loadConfig ( cdkApp : cdk . App , key : string , contextArgs ?: string [ ] ) : any {
68- var fromType = 'Input-Parameter' ;
69- var configFilePath = cdkApp . node . tryGetContext ( key ) ; //'APP_CONFIG'
93+ private findAppConfigFile ( appConfigKey : string ) : string {
94+ let fromType = 'InLine-Argument' ;
95+ let configFilePath = this . cdkApp . node . tryGetContext ( appConfigKey ) ;
7096
7197 if ( configFilePath == undefined ) {
72- configFilePath = env . get ( key ) . asString ( ) ;
98+ configFilePath = env . get ( appConfigKey ) . asString ( ) ;
7399
74100 if ( configFilePath != undefined && configFilePath . length > 0 ) {
75101 fromType = 'Environment-Variable' ;
@@ -79,62 +105,68 @@ export class AppContext {
79105 }
80106
81107 if ( configFilePath == undefined ) {
82- console . error ( `==> CDK App-Config File is empty,
83- please check your environment variable(Usage: export ${ key }
84- or input parameter(--context=${ key } ) ;
85- return false ;
108+ throw new Error ( 'Fail to find App-Config json file' )
86109 } else {
87110 console . info ( `==> CDK App-Config File is ${ configFilePath } ${ fromType } ) ;
88- return this . loadConfigFromFile ( configFilePath , cdkApp , contextArgs ) ;
89111 }
112+
113+ return configFilePath ;
90114 }
91115
92- private loadConfigFromFile ( filePath : string , app ?: cdk . App , contextArgs ?: string [ ] ) : any {
93- this . infraConfigPath = filePath ;
94- let config : any = JSON . parse ( fs . readFileSync ( filePath ) . toString ( ) ) ;
95-
116+ private getProjectPrefix ( projectName : string , projectStage : string ) : string {
117+ let prefix = `${ projectName } ${ projectStage } ;
118+
119+ if ( this . appContextProps . projectPrefixType === ProjectPrefixType . NameHyphenStage ) {
120+ prefix = `${ projectName } ${ projectStage } ;
121+ } else if ( this . appContextProps . projectPrefixType === ProjectPrefixType . Name ) {
122+ prefix = projectName ;
123+ }
124+
125+ return prefix ;
126+ }
127+
128+ private loadAppConfigFile ( filePath : string , contextArgs ?: string [ ] ) : any {
129+ let appConfig = JSON . parse ( fs . readFileSync ( filePath ) . toString ( ) ) ;
130+ let projectPrefix = this . getProjectPrefix ( appConfig . Project . Name , appConfig . Project . Stage ) ;
131+
96132 if ( contextArgs != undefined ) {
97- this . updateContextArgs ( config , app ! , contextArgs ) ;
133+ this . updateContextArgs ( appConfig , contextArgs ) ;
98134 }
99-
100- this . addPrefixIntoStackName ( config ) ;
101135
102- return config ;
136+ this . addPrefixIntoStackName ( appConfig , projectPrefix ) ;
137+
138+ return appConfig ;
103139 }
104-
105- private updateContextArgs ( config : any , app : cdk . App , contextArgs : string [ ] ) {
106- for ( var key of contextArgs ) {
140+
141+ private updateContextArgs ( appConfig : any , contextArgs : string [ ] ) {
142+ for ( let key of contextArgs ) {
107143 const jsonKeys = key . split ( '.' ) ;
108- let oldValue = '' ;
109- const newValue : string = app ?. node . tryGetContext ( key ) ;
110-
111- if ( jsonKeys . length == 1 ) {
112- oldValue = config [ jsonKeys [ 0 ] ] ;
113- config [ jsonKeys [ 0 ] ] = newValue ;
114- } else if ( jsonKeys . length == 2 ) {
115- oldValue = config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ]
116- config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] = newValue ;
117- } else if ( jsonKeys . length == 3 ) {
118- oldValue = config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] [ jsonKeys [ 2 ] ] ;
119- config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] [ jsonKeys [ 2 ] ] = newValue ;
120- } else if ( jsonKeys . length == 4 ) {
121- oldValue = config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] [ jsonKeys [ 2 ] ] [ jsonKeys [ 3 ] ] ;
122- config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] [ jsonKeys [ 2 ] ] [ jsonKeys [ 3 ] ] = newValue ;
123- } else if ( jsonKeys . length == 5 ) {
124- oldValue = config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] [ jsonKeys [ 2 ] ] [ jsonKeys [ 3 ] ] [ jsonKeys [ 4 ] ] ;
125- config [ jsonKeys [ 0 ] ] [ jsonKeys [ 1 ] ] [ jsonKeys [ 2 ] ] [ jsonKeys [ 3 ] ] [ jsonKeys [ 4 ] ] = newValue ;
144+ let oldValue = undefined ;
145+ const newValue : string = this . cdkApp . node . tryGetContext ( key ) ;
146+
147+ if ( newValue != undefined && jsonKeys . length > 0 ) {
148+ try {
149+ oldValue = jsonKeys . reduce ( ( reducer : any , pointer : string ) => reducer . hasOwnProperty ( pointer ) ? reducer [ pointer ] : undefined , appConfig ) ;
150+ } catch ( e ) {
151+ console . error ( `[ERROR] updateContextArgs: This key[${ key } , e ) ;
152+ throw e ;
153+ }
154+
155+ jsonKeys . reduce ( ( reducer : any , pointer : string , count : number ) => {
156+ if ( count == jsonKeys . length - 1 ) reducer [ pointer ] = newValue ;
157+ return reducer [ pointer ] ;
158+ } , appConfig ) ;
159+
160+ console . info ( `[INFO] updateContextArgs: Updated ${ key } ${ oldValue } ${ newValue } ) ;
126161 }
127-
128- console . error ( `updateContextArgs: ${ key } ${ oldValue } ${ newValue } ) ;
129162 }
130163 }
131164
132- private addPrefixIntoStackName ( config : any ) {
133- const projectPrefix = this . projectPrefix == undefined ? `${ config . Project . Name } ${ config . Project . Stage } : this . projectPrefix ;
134- this . projectPrefix = projectPrefix ;
135-
136- for ( const key in config . Stack ) {
137- config . Stack [ key ] . Name = `${ projectPrefix } ${ config . Stack [ key ] . Name } ;
165+ private addPrefixIntoStackName ( appConfig : any , projectPrefix : string ) {
166+ for ( const key in appConfig . Stack ) {
167+ const stackOriginalName = appConfig . Stack [ key ] . Name ;
168+ appConfig . Stack [ key ] . ShortStackName = stackOriginalName ;
169+ appConfig . Stack [ key ] . Name = `${ projectPrefix } ${ stackOriginalName } ;
138170 }
139171 }
140172}
0 commit comments