Skip to content

Commit c5d8fc9

Browse files
committed
feat: improve the plugin state variables
1 parent 287edfe commit c5d8fc9

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

src/index.ts

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,12 @@ import type { Serverless, ServerlessSecretHooks, ServerlessSecretOptions } from
55
class ServerlessAWSSecret {
66
hooks: ServerlessSecretHooks;
77
options: ServerlessSecretOptions;
8-
providerCopy: Serverless['service']['provider'];
9-
region: string;
8+
serverless: Serverless;
109

1110
constructor(serverless: Serverless) {
1211
this.setOptions(serverless);
1312

14-
const { provider } = serverless.service;
15-
this.region = provider.region;
16-
this.providerCopy = provider;
13+
this.serverless = serverless;
1714

1815
this.hooks = {
1916
'before:package:initialize': this.loadSecrets.bind(this),
@@ -22,7 +19,7 @@ class ServerlessAWSSecret {
2219
}
2320

2421
async loadSecrets() {
25-
const client = new SecretsManagerClient({ region: this.region });
22+
const client = new SecretsManagerClient({ region: this.serverless.service.provider.region });
2623
const command = new GetSecretValueCommand({ SecretId: this.options.secretId });
2724

2825
const { SecretString } = await client.send(command);
@@ -34,7 +31,7 @@ class ServerlessAWSSecret {
3431
const secrets = JSON.parse(SecretString);
3532

3633
let replaceCount = 0;
37-
for (const [key, value] of Object.entries(this.providerCopy.environment)) {
34+
for (const [key, value] of Object.entries(this.serverless.service.provider.environment)) {
3835
if (value?.startsWith(this.options.secretPrefix!)) {
3936
const secretKey = value.replace(this.options.secretPrefix!, '');
4037

@@ -46,7 +43,7 @@ class ServerlessAWSSecret {
4643
console.log(`[serverless-aws-secrets]: Replacing ${key} with secret of ${secretKey}`);
4744
}
4845

49-
this.providerCopy.environment[key] = secrets[secretKey];
46+
this.serverless.service.provider.environment[key] = secrets[secretKey];
5047

5148
++replaceCount;
5249
}

0 commit comments

Comments
 (0)