@xube/kit-aws 0.0.59 → 0.0.61

package/dist/index.d.ts

@@ -1,4 +1,7 @@
+export * from "./resources/cognito/authenticate";
+export * from "./resources/cognito/register";
 export * from "./resources/dynamodb/get";
 export * from "./resources/dynamodb/put";
 export * from "./resources/dynamodb/query";
 export * from "./resources/dynamodb/transform";
+export * from "./resources/parameter-store/get";

package/dist/index.js

@@ -14,7 +14,10 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     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("./resources/cognito/authenticate"), exports);
+__exportStar(require("./resources/cognito/register"), exports);
 __exportStar(require("./resources/dynamodb/get"), exports);
 __exportStar(require("./resources/dynamodb/put"), exports);
 __exportStar(require("./resources/dynamodb/query"), exports);
 __exportStar(require("./resources/dynamodb/transform"), exports);
+__exportStar(require("./resources/parameter-store/get"), exports);

package/dist/resources/cognito/authenticate.d.ts

@@ -0,0 +1,6 @@
+import { AuthFlowType } from "@aws-sdk/client-cognito-identity-provider";
+import { AuthenticationRequest, AuthenticationResponse } from "@xube/kit-aws-schema";
+import { XubeLog } from "@xube/kit-log";
+import { XubeResponse } from "@xube/kit-request";
+export { AuthFlowType };
+export declare const authenticateUser: (request: AuthenticationRequest, authFlow: typeof AuthFlowType.USER_PASSWORD_AUTH, clientId: string, log?: XubeLog) => Promise<XubeResponse<AuthenticationResponse>>;

package/dist/resources/cognito/authenticate.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.authenticateUser = exports.AuthFlowType = void 0;
+const client_cognito_identity_provider_1 = require("@aws-sdk/client-cognito-identity-provider");
+Object.defineProperty(exports, "AuthFlowType", { enumerable: true, get: function () { return client_cognito_identity_provider_1.AuthFlowType; } });
+const kit_constants_1 = require("@xube/kit-constants");
+const kit_log_1 = require("@xube/kit-log");
+const kit_request_1 = require("@xube/kit-request");
+const client = new client_cognito_identity_provider_1.CognitoIdentityProviderClient({
+    region: process.env.AWS_REGION,
+});
+const authenticateUser = async (request, authFlow, clientId, log = kit_log_1.XubeLog.getInstance()) => {
+    log.info("Authenticating user");
+    try {
+        const initiateAuthCommand = new client_cognito_identity_provider_1.InitiateAuthCommand({
+            AuthFlow: authFlow,
+            AuthParameters: {
+                USERNAME: request.email,
+                PASSWORD: request.password,
+            },
+            ClientId: clientId,
+        });
+        log.info("Sending authentication request");
+        const authResponse = await client.send(initiateAuthCommand);
+        const authenticationResponse = {
+            accessToken: authResponse.AuthenticationResult?.AccessToken || "",
+            refreshToken: authResponse.AuthenticationResult?.RefreshToken || "",
+            expiry: authResponse.AuthenticationResult?.ExpiresIn || 0,
+            token: authResponse.AuthenticationResult?.IdToken || "",
+            deviceKey: authResponse.AuthenticationResult?.NewDeviceMetadata?.DeviceKey,
+        };
+        log.info("Authentication successful");
+        return new kit_request_1.XubeResponse({
+            statusCode: kit_constants_1.StatusCode.OK,
+            data: authenticationResponse,
+        });
+    }
+    catch (e) {
+        log.info("Authentication failed" + JSON.stringify(e));
+        throw e;
+    }
+};
+exports.authenticateUser = authenticateUser;

package/dist/resources/cognito/register.d.ts

@@ -0,0 +1,4 @@
+import { AuthenticationRequest, AuthenticationResponse } from "@xube/kit-aws-schema";
+import { XubeLog } from "@xube/kit-log";
+import { XubeResponse } from "@xube/kit-request";
+export declare const register: (credentials: AuthenticationRequest, clientId: string, log?: XubeLog) => Promise<XubeResponse<AuthenticationResponse>>;

package/dist/resources/cognito/register.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.register = void 0;
+const client_cognito_identity_provider_1 = require("@aws-sdk/client-cognito-identity-provider");
+const kit_constants_1 = require("@xube/kit-constants");
+const kit_log_1 = require("@xube/kit-log");
+const kit_request_1 = require("@xube/kit-request");
+const authenticate_1 = require("./authenticate");
+const cognitoIDPClient = new client_cognito_identity_provider_1.CognitoIdentityProviderClient({
+    region: process.env.AWS_REGION,
+});
+const register = async (credentials, clientId, log = kit_log_1.XubeLog.getInstance()) => {
+    try {
+        const response = await cognitoIDPClient.send(new client_cognito_identity_provider_1.SignUpCommand({
+            ClientId: clientId,
+            Username: credentials.email,
+            Password: credentials.password,
+        }));
+        if (!response.UserSub) {
+            log.info(`User was not created. Response from Cognito: ${JSON.stringify(response)}`);
+            return new kit_request_1.XubeResponse({
+                statusCode: kit_constants_1.StatusCode.InternalError,
+                error: `User was not created`,
+            });
+        }
+        return await (0, authenticate_1.authenticateUser)(credentials, client_cognito_identity_provider_1.AuthFlowType.USER_PASSWORD_AUTH, clientId, log);
+    }
+    catch (e) {
+        if (e instanceof client_cognito_identity_provider_1.UsernameExistsException) {
+            return new kit_request_1.XubeResponse({
+                statusCode: kit_constants_1.StatusCode.Conflict,
+                error: `Username already exists`,
+            });
+        }
+        return new kit_request_1.XubeResponse({
+            statusCode: kit_constants_1.StatusCode.InternalError,
+            error: `Error signing up user: ${e}`,
+        });
+    }
+};
+exports.register = register;

package/dist/resources/parameter-store/get.d.ts

@@ -0,0 +1,3 @@
+import { XubeLog } from "@xube/kit-log";
+import { XubeResponse } from "@xube/kit-request";
+export declare const getParameter: (parameterName: string, log?: XubeLog) => Promise<XubeResponse<string>>;

package/dist/resources/parameter-store/get.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getParameter = void 0;
+const client_ssm_1 = require("@aws-sdk/client-ssm");
+const kit_constants_1 = require("@xube/kit-constants");
+const kit_log_1 = require("@xube/kit-log");
+const kit_request_1 = require("@xube/kit-request");
+const ssmClient = new client_ssm_1.SSMClient({ region: process.env.AWS_REGION });
+const getParameter = async (parameterName, log = kit_log_1.XubeLog.getInstance()) => {
+    const getParamCommand = new client_ssm_1.GetParameterCommand({
+        Name: parameterName,
+    });
+    let parameterValue;
+    try {
+        const paramResponse = await ssmClient.send(getParamCommand);
+        if (!paramResponse?.Parameter?.Value) {
+            throw "Parameter value does not exist";
+        }
+        parameterValue = paramResponse.Parameter.Value;
+        if (!parameterValue) {
+            log.info(`Parameter ${parameterName} does not exist`);
+            return new kit_request_1.XubeResponse({
+                statusCode: kit_constants_1.StatusCode.NotFound,
+                error: `Parameter ${parameterName} does not exist`,
+            });
+        }
+    }
+    catch (error) {
+        log.error(`Error getting parameter ${parameterName}: ${error}`);
+        return new kit_request_1.XubeResponse({
+            statusCode: kit_constants_1.StatusCode.InternalError,
+            error: `Error getting parameter ${parameterName}: ${error}`,
+        });
+    }
+    return new kit_request_1.XubeResponse({
+        statusCode: kit_constants_1.StatusCode.OK,
+        data: parameterValue,
+    });
+};
+exports.getParameter = getParameter;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xube/kit-aws",
-  "version": "0.0.59",
+  "version": "0.0.61",
   "description": "",
   "main": "dist/index.js",
   "scripts": {
@@ -18,16 +18,18 @@
   "homepage": "https://github.com/XubeLtd/dev-kit#readme",
   "devDependencies": {
     "@types/aws-lambda": "^8.10.119",
-    "@xube/kit-build": "^0.0.59"
+    "@xube/kit-build": "^0.0.61"
   },
   "dependencies": {
-    "@aws-sdk/client-dynamodb": "^3.427.0",
-    "@aws-sdk/lib-dynamodb": "^3.427.0",
-    "@aws-sdk/util-dynamodb": "^3.427.0",
-    "@xube/kit-aws-schema": "^0.0.59",
-    "@xube/kit-log": "^0.0.59",
-    "@xube/kit-request": "^0.0.59",
-    "@xube/kit-schema": "^0.0.59",
+    "@aws-sdk/client-cognito-identity-provider": "^3.433.0",
+    "@aws-sdk/client-dynamodb": "^3.433.0",
+    "@aws-sdk/client-ssm": "^3.433.0",
+    "@aws-sdk/lib-dynamodb": "^3.433.0",
+    "@aws-sdk/util-dynamodb": "^3.433.0",
+    "@xube/kit-aws-schema": "^0.0.61",
+    "@xube/kit-log": "^0.0.61",
+    "@xube/kit-request": "^0.0.61",
+    "@xube/kit-schema": "^0.0.61",
     "zod": "^3.22.4"
   }
 }

package/src/index.ts

@@ -1,4 +1,7 @@
+export * from "./resources/cognito/authenticate";
+export * from "./resources/cognito/register";
 export * from "./resources/dynamodb/get";
 export * from "./resources/dynamodb/put";
 export * from "./resources/dynamodb/query";
 export * from "./resources/dynamodb/transform";
+export * from "./resources/parameter-store/get";

package/src/resources/cognito/authenticate.ts

@@ -0,0 +1,62 @@
+import {
+  AuthFlowType,
+  CognitoIdentityProviderClient,
+  InitiateAuthCommand,
+  InitiateAuthCommandOutput,
+} from "@aws-sdk/client-cognito-identity-provider";
+import {
+  AuthenticationRequest,
+  AuthenticationResponse,
+} from "@xube/kit-aws-schema";
+import { StatusCode } from "@xube/kit-constants";
+import { XubeLog } from "@xube/kit-log";
+import { XubeResponse } from "@xube/kit-request";
+
+export { AuthFlowType };
+
+const client = new CognitoIdentityProviderClient({
+  region: process.env.AWS_REGION,
+});
+export const authenticateUser = async (
+  request: AuthenticationRequest,
+  authFlow: typeof AuthFlowType.USER_PASSWORD_AUTH,
+  clientId: string,
+  log: XubeLog = XubeLog.getInstance()
+): Promise<XubeResponse<AuthenticationResponse>> => {
+  log.info("Authenticating user");
+  try {
+    const initiateAuthCommand = new InitiateAuthCommand({
+      AuthFlow: authFlow,
+      AuthParameters: {
+        USERNAME: request.email,
+        PASSWORD: request.password,
+      },
+      ClientId: clientId,
+    });
+
+    log.info("Sending authentication request");
+
+    const authResponse: InitiateAuthCommandOutput = await client.send(
+      initiateAuthCommand
+    );
+
+    const authenticationResponse: AuthenticationResponse = {
+      accessToken: authResponse.AuthenticationResult?.AccessToken || "",
+      refreshToken: authResponse.AuthenticationResult?.RefreshToken || "",
+      expiry: authResponse.AuthenticationResult?.ExpiresIn || 0,
+      token: authResponse.AuthenticationResult?.IdToken || "",
+      deviceKey:
+        authResponse.AuthenticationResult?.NewDeviceMetadata?.DeviceKey,
+    };
+
+    log.info("Authentication successful");
+
+    return new XubeResponse({
+      statusCode: StatusCode.OK,
+      data: authenticationResponse,
+    });
+  } catch (e) {
+    log.info("Authentication failed" + JSON.stringify(e));
+    throw e;
+  }
+};

package/src/resources/cognito/register.ts

@@ -0,0 +1,65 @@
+import {
+  AuthFlowType,
+  CognitoIdentityProviderClient,
+  SignUpCommand,
+  UsernameExistsException,
+} from "@aws-sdk/client-cognito-identity-provider";
+import {
+  AuthenticationRequest,
+  AuthenticationResponse,
+} from "@xube/kit-aws-schema";
+import { StatusCode } from "@xube/kit-constants";
+import { XubeLog } from "@xube/kit-log";
+import { XubeResponse } from "@xube/kit-request";
+import { authenticateUser } from "./authenticate";
+
+const cognitoIDPClient: CognitoIdentityProviderClient =
+  new CognitoIdentityProviderClient({
+    region: process.env.AWS_REGION,
+  });
+
+export const register = async (
+  credentials: AuthenticationRequest,
+  clientId: string,
+  log: XubeLog = XubeLog.getInstance()
+): Promise<XubeResponse<AuthenticationResponse>> => {
+  try {
+    const response = await cognitoIDPClient.send(
+      new SignUpCommand({
+        ClientId: clientId,
+        Username: credentials.email,
+        Password: credentials.password,
+      })
+    );
+
+    if (!response.UserSub) {
+      log.info(
+        `User was not created. Response from Cognito: ${JSON.stringify(
+          response
+        )}`
+      );
+      return new XubeResponse({
+        statusCode: StatusCode.InternalError,
+        error: `User was not created`,
+      });
+    }
+
+    return await authenticateUser(
+      credentials,
+      AuthFlowType.USER_PASSWORD_AUTH,
+      clientId,
+      log
+    );
+  } catch (e) {
+    if (e instanceof UsernameExistsException) {
+      return new XubeResponse({
+        statusCode: StatusCode.Conflict,
+        error: `Username already exists`,
+      });
+    }
+    return new XubeResponse({
+      statusCode: StatusCode.InternalError,
+      error: `Error signing up user: ${e}`,
+    });
+  }
+};

package/src/resources/parameter-store/get.ts

@@ -0,0 +1,46 @@
+import { GetParameterCommand, SSMClient } from "@aws-sdk/client-ssm";
+import { StatusCode } from "@xube/kit-constants";
+import { XubeLog } from "@xube/kit-log";
+import { XubeResponse } from "@xube/kit-request";
+
+const ssmClient: SSMClient = new SSMClient({ region: process.env.AWS_REGION });
+
+export const getParameter = async (
+  parameterName: string,
+  log: XubeLog = XubeLog.getInstance()
+): Promise<XubeResponse<string>> => {
+  const getParamCommand = new GetParameterCommand({
+    Name: parameterName,
+  });
+
+  let parameterValue: string;
+
+  try {
+    const paramResponse = await ssmClient.send(getParamCommand);
+
+    if (!paramResponse?.Parameter?.Value) {
+      throw "Parameter value does not exist";
+    }
+
+    parameterValue = paramResponse.Parameter.Value;
+
+    if (!parameterValue) {
+      log.info(`Parameter ${parameterName} does not exist`);
+      return new XubeResponse({
+        statusCode: StatusCode.NotFound,
+        error: `Parameter ${parameterName} does not exist`,
+      });
+    }
+  } catch (error) {
+    log.error(`Error getting parameter ${parameterName}: ${error}`);
+    return new XubeResponse({
+      statusCode: StatusCode.InternalError,
+      error: `Error getting parameter ${parameterName}: ${error}`,
+    });
+  }
+
+  return new XubeResponse({
+    statusCode: StatusCode.OK,
+    data: parameterValue,
+  });
+};