Update

action.yml

@@ -36,6 +36,17 @@ inputs:
     description: 'Item Value, Only required for batch-put operation'
     required: false
 
+  #Update Operation
+  updateExpression:
+    description: 'Attribute variables, Only required for update operation'
+    required: false
+  expressionAttributeValues:
+    description: 'Attribute values to be updated, Only required for update operation'
+    required: false
+  expressionAttributeFiles:
+    description: 'Attribute files to be updated, Only required for update operation'
+    required: false
+
 outputs:
   item:
     description: 'JSON-serialized Item'

dist/index.js

@@ -11,7 +11,7 @@ const processor = new processor_1.Processor();
         operation: (_a = core.getInput("operation")) === null || _a === void 0 ? void 0 : _a.toLowerCase(),
         region: core.getInput("region"),
         table: core.getInput("table"),
-        // Get / Delete Operation
+        // Get / Delete / Update Operation
         key: helpers_1.forgivingJSONParse(core.getInput("key")),
         consistent: helpers_1.forgivingJSONParse(core.getInput("consistent")),
         // Put Operation
@@ -20,6 +20,10 @@ const processor = new processor_1.Processor();
         // BatchPut Operation
         items: helpers_1.forgivingJSONParse(core.getInput("items")),
         files: core.getInput("files"),
+        // Update Operation
+        updateExpression: core.getInput("updateExpression"),
+        expressionAttributeValues: core.getInput("expressionAttributeValues"),
+        expressionAttributeFiles: core.getInput("expressionAttributeFiles")
     };
     const output = await processor.process(input);
     if (output) {

dist/operations/batch-put.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BatchPutOperation = void 0;
 const glob = require("@actions/glob");
-const Joi = require("@hapi/joi");
+const Joi = require("joi");
 const fs_1 = require("fs");
 const helpers_1 = require("../helpers");
 const BaseInputSchema = Joi.object({

dist/operations/delete.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DeleteOperation = void 0;
-const Joi = require("@hapi/joi");
+const Joi = require("joi");
 const helpers_1 = require("../helpers");
 const InputSchema = Joi.object({
     operation: Joi.string().lowercase().valid("delete").required(),

dist/operations/get.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GetOperation = void 0;
-const Joi = require("@hapi/joi");
+const Joi = require("joi");
 const helpers_1 = require("../helpers");
 const InputSchema = Joi.object({
     operation: Joi.string().lowercase().valid("get").required(),

dist/operations/index.js

@@ -7,11 +7,12 @@ var __createBinding = (this && this.__createBinding) || (Object.create ? (functi
     o[k2] = m[k];
 }));
 var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !exports.hasOwnProperty(p)) __createBinding(exports, m, p);
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./base"), exports);
 __exportStar(require("./batch-put"), exports);
 __exportStar(require("./delete"), exports);
 __exportStar(require("./get"), exports);
 __exportStar(require("./put"), exports);
+__exportStar(require("./update"), exports);

dist/operations/put.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PutOperation = void 0;
-const Joi = require("@hapi/joi");
+const Joi = require("joi");
 const fs_1 = require("fs");
 const helpers_1 = require("../helpers");
 const BaseInputSchema = Joi.object({

dist/operations/update.js

@@ -0,0 +1,96 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateOperation = void 0;
+const Joi = require("joi");
+const fs_1 = require("fs");
+const helpers_1 = require("../helpers");
+const BaseInputSchema = Joi.object({
+    operation: Joi.string().lowercase().valid("update").required(),
+    region: Joi.string().lowercase().required(),
+    table: Joi.string().required(),
+    updateExpression: Joi.string().required()
+});
+const InputSchema = Joi.alternatives([
+    BaseInputSchema.append({
+        expressionAttributeValues: Joi.string().required(),
+        key: Joi.object().required(),
+    }),
+    BaseInputSchema.append({
+        expressionAttributeFiles: Joi.string().required(),
+        key: Joi.object().required(),
+    }),
+]).required();
+class UpdateOperation {
+    constructor() {
+        this.name = "update";
+    }
+    async validate(input) {
+        const validationResult = InputSchema.validate(input, {
+            stripUnknown: true,
+        });
+        if (validationResult.error) {
+            throw validationResult.error;
+        }
+        return validationResult.value;
+    }
+    async execute(input) {
+        const ddb = helpers_1.createClient(input.region);
+        let updateExp = ``;
+        let attValues = {};
+        const expressions = await this.buildExpression(updateExp, input);
+        const attributes = await this.buildAttributes(expressions, attValues, input);
+        console.log(expressions);
+        console.log(attributes);
+        await ddb.update({
+            TableName: input.table,
+            Key: input.key,
+            UpdateExpression: expressions,
+            ExpressionAttributeValues: attributes
+        }).promise();
+    }
+    async read(path) {
+        const content = await fs_1.promises.readFile(path, { encoding: "utf8" });
+        return JSON.parse(content);
+    }
+    async buildExpression(updateExp, input) {
+        const updateExpressions = input.updateExpression.split(',');
+        for (let i = 0; i < updateExpressions.length; i++) {
+            if (i === 0 && updateExpressions.length > 1) {
+                updateExp = `set ${updateExpressions[i]} = :${updateExpressions[i]},`;
+            }
+            else if (i === 0 && updateExpressions.length === 1) {
+                updateExp = `set ${updateExpressions[i]} = :${updateExpressions[i]}`;
+            }
+            else if (i === updateExpressions.length - 1) {
+                updateExp += ` ${updateExpressions[i]} = :${updateExpressions[i]}`;
+            }
+            else {
+                updateExp += ` ${updateExpressions[i]} = :${updateExpressions[i]},`;
+            }
+        }
+        return updateExp;
+    }
+    async buildAttributes(updateExp, attValues, input) {
+        const updateExpressions = updateExp.split(',');
+        if (input.expressionAttributeValues) {
+            const expAttValues = input.expressionAttributeValues.split(', ');
+            const keyValues = [];
+            for (let i = 0; i < updateExpressions.length; i++) {
+                updateExpressions[i] = updateExpressions[i].substring(updateExpressions[i].indexOf(":"));
+                keyValues[i] = [updateExpressions[i], expAttValues[i]];
+            }
+            attValues = Object.fromEntries(keyValues);
+        }
+        else if (input.expressionAttributeFiles) {
+            const expAttValues = input.expressionAttributeFiles.split(',');
+            const keyValues = [];
+            for (let i = 0; i < updateExpressions.length; i++) {
+                updateExpressions[i] = updateExpressions[i].substring(updateExpressions[i].indexOf(":"));
+                keyValues[i] = [updateExpressions[i], await this.read(expAttValues[i])];
+            }
+            attValues = Object.fromEntries(keyValues);
+        }
+        return attValues;
+    }
+}
+exports.UpdateOperation = UpdateOperation;

dist/processor.js

@@ -9,6 +9,7 @@ class Processor {
             new operations_1.DeleteOperation(),
             new operations_1.GetOperation(),
             new operations_1.PutOperation(),
+            new operations_1.UpdateOperation()
         ];
     }
     async process(input) {