Skip to content

add support for single function deployments #21

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

add support for single function deployments #21

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
104 changes: 65 additions & 39 deletions dist/serverless-ssm-fetch.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,15 @@ var SsmFetch = function () {
this.serverless = serverless;
this.options = options;

serverless.configSchemaHandler.defineProvider('aws', {
provider: {
type: "object",
properties: {
ssmToEnvironment: { type: 'array' }
}
}
});

serverless.configSchemaHandler.defineFunctionProperties('aws', {
properties: {
ssmToEnvironment: { type: 'array' }
Expand All @@ -47,6 +56,10 @@ var SsmFetch = function () {
};

this.hooks = {
'package:function:package': function packageFunctionPackage() {
_this._triggeredFromHook = true;
return _this.serverless.pluginManager.run(['serverless-ssm-fetch', 'parameter']);
},
'after:package:cleanup': function afterPackageCleanup() {
_this._triggeredFromHook = true;
return _this.serverless.pluginManager.run(['serverless-ssm-fetch', 'parameter']);
Expand All @@ -58,7 +71,7 @@ var SsmFetch = function () {
return BbPromise.bind(_this).then(function () {
return _this.getParameter(log);
}).then(function () {
return _this.assignParameter(log);
return _this.assignParameters(log);
});
}
};
Expand Down Expand Up @@ -131,61 +144,74 @@ var SsmFetch = function () {
});
}
}, {
key: 'assignParameter',
value: function assignParameter(log) {
key: 'assignParameters',
value: function assignParameters(log) {
var _this3 = this;

// forEach function to deploy
Object.keys(this.serverless.service.functions).forEach(function (functionName) {
// Aliases of the current function path and the got ssm parameters path
var currentFunction = _this3.serverless.service.functions[functionName];
var fetchedSsmParameters = _this3.serverless.serverlessSsmFetch;
if (this.options.function) {
// Function to deploy
var currentFunction = this.serverless.service.functions[this.options.function];
this.assignParameter(log, currentFunction);
} else {
// forEach function to deploy
Object.keys(this.serverless.service.functions).forEach(function (functionName) {
// Aliases of the current function path and the got ssm parameters path
var currentFunction = _this3.serverless.service.functions[functionName];
_this3.assignParameter(log, currentFunction);
});
}
}
}, {
key: 'assignParameter',
value: function assignParameter(log, currentFunction) {
var _this4 = this;

if (_this3.isSet(currentFunction.ssmToEnvironment)) {
// If the property `ssmToEnvironment` has been set at the function level
var fetchedSsmParameters = this.serverless.serverlessSsmFetch;

// Creates the function `environment` property if it doesn't already exist
if (!_this3.isSet(currentFunction.environment)) {
currentFunction.environment = {};
}
if (this.isSet(currentFunction.ssmToEnvironment)) {
// If the property `ssmToEnvironment` has been set at the function level

// forEach ssmParameter assigned over `ssmToEnvironment` function property...
currentFunction.ssmToEnvironment.forEach(function (ssmParameterToAssign) {
if (_this3.isSet(fetchedSsmParameters[ssmParameterToAssign])) {
// merges it into the function `environment` property
currentFunction.environment[ssmParameterToAssign] = fetchedSsmParameters[ssmParameterToAssign];
}
});
} else {
// Else, the property `ssmToEnvironment` has NOT been set at the function level

// Creates the function `environment` property if it doesn't already exist
if (!_this3.isSet(currentFunction.environment)) {
currentFunction.environment = {};
}
// Creates the function `environment` property if it doesn't already exist
if (!this.isSet(currentFunction.environment)) {
currentFunction.environment = {};
}

// Merges ALL the fetched ssmParameters
Object.keys(fetchedSsmParameters).forEach(function (ssmParameterToAssign) {
// forEach ssmParameter assigned over `ssmToEnvironment` function property...
currentFunction.ssmToEnvironment.forEach(function (ssmParameterToAssign) {
if (_this4.isSet(fetchedSsmParameters[ssmParameterToAssign])) {
// merges it into the function `environment` property
currentFunction.environment[ssmParameterToAssign] = fetchedSsmParameters[ssmParameterToAssign];
});
}
});
} else {
// Else, the property `ssmToEnvironment` has NOT been set at the function level

// Creates the function `environment` property if it doesn't already exist
if (!this.isSet(currentFunction.environment)) {
currentFunction.environment = {};
}

log.info('> serverless-ssm-fetch: Function "' + functionName + '" set environment variables:');
log.info(JSON.stringify(Object.keys(currentFunction.environment), null, 2));
});
// Merges ALL the fetched ssmParameters
Object.keys(fetchedSsmParameters).forEach(function (ssmParameterToAssign) {
currentFunction.environment[ssmParameterToAssign] = fetchedSsmParameters[ssmParameterToAssign];
});
}

log.info('> serverless-ssm-fetch: Function "' + currentFunction.name + '" set environment variables:');
log.info(JSON.stringify(Object.keys(currentFunction.environment), null, 2));
}
}, {
key: 'validate',
value: function validate() {
var _this4 = this;
var _this5 = this;

setTimeout(function () {
if (_this4.serverless.service.provider.name !== 'aws') {
throw new _this4.serverless.classes.Error('> serverless-ssm-fetch: The plugin "serverless-ssm-fetch" is only available for `aws` provider.');
if (_this5.serverless.service.provider.name !== 'aws') {
throw new _this5.serverless.classes.Error('> serverless-ssm-fetch: The plugin "serverless-ssm-fetch" is only available for `aws` provider.');
}

if (!_this4.isSet(_this4.serverless.service.custom) || !_this4.isSet(_this4.serverless.service.custom['serverlessSsmFetch'])) {
throw new _this4.serverless.classes.Error('> serverless-ssm-fetch: You are using the plugin "serverless-ssm-fetch". You must set a `custom.serverlessSsmFetch` element in your serverless conf file.');
if (!_this5.isSet(_this5.serverless.service.custom) || !_this5.isSet(_this5.serverless.service.custom['serverlessSsmFetch'])) {
throw new _this5.serverless.classes.Error('> serverless-ssm-fetch: You are using the plugin "serverless-ssm-fetch". You must set a `custom.serverlessSsmFetch` element in your serverless conf file.');
}
}, 5000);
}
Expand Down
Loading