serverless-spy 0.0.0

package/cli/package.json

@@ -0,0 +1,14 @@
+{
+  "name": "serverless-spy-cli",
+  "version": "0.0.0",
+  "description": "",
+  "scripts": {},
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@aws-sdk/client-sts": "^3.142.0",
+    "@aws-sdk/credential-providers": "^3.142.0",
+    "esbuild-runner": "^2.2.1",
+    "ws": "^8.8.1"
+  }
+}

package/cli/serverlessSpy.js

@@ -0,0 +1,73 @@
+function run() {
+  const tableBody = document.getElementById("tableBody");
+  var modal = document.getElementById("myModal");
+  var span = document.getElementsByClassName("close")[0];
+  const modalContent = document.getElementById("modalContent");
+  span.onclick = function () {
+    modal.style.display = "none";
+  };
+  window.onclick = function (event) {
+    if (event.target == modal) {
+      modal.style.display = "none";
+    }
+  };
+
+  tableBody.addEventListener("click", function (e) {
+    const dataJson = JSON.stringify(
+      JSON.parse(e.target.parentElement.children[2].textContent),
+      null,
+      2
+    );
+    modalContent.innerText = dataJson;
+    modal.style.display = "block";
+    //alert();
+  });
+
+  const url = "SERVERLESS_SPY_WS_URL";
+  //"wss://m6g3w6ttdh.execute-api.eu-west-1.amazonaws.com/prod?X-Amz-Security-Token=IQoJb3JpZ2luX2VjELL%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FwEaCWV1LXdlc3QtMSJGMEQCIDk%2BzQd0RC2DmEgCJEPoRFAkmnhCZl%2BSIlEG4TummRkIAiBIgNG8Yw%2FOAew3XNAE3yrNCK0G0q%2BZI8MwjEa8xoEonirrAQgbEAQaDDA3NjU4NTk2NzA3NSIMxVUxKhwdqGjFOsRtKsgB2eotQMFgl4CrD8uy%2Fe1UE6K4HVc6uWo0gdjgHMPb5C9Mq0wFgHTiDFjFxIFkJpmCiTRXKqNFrx5lpo9C5Ml1bpcv%2BTopT6wvpbFksxChhhnaqUDHiBaGuys8zEy8Bjlu4jF0nKccmUYdieD70%2FXz99OU4wKPxXyWbr0w82y1Xd6ag8g4VNx3EXEdULq7vkM6H2shSAH8EUlZoTD%2BqvIQvyKS8nta3ebhprO%2BsLRQv2YC6k1w46jHuOjtI7hSnmfSKnkUqAxS6g4wgZrFlwY6mQFNSXz7jSF3IRtdcpMEIp%2Bnghf2LhvZKeW6GkZs4DMeuhPjoBehfmXLJIJWgcvLoheQvIt28ddgHU60%2BoRKhZTxQKhpvSOaAbI6M%2B%2FyHvkC2hTR4fEBtESAMnzo8WocfxdaQQ0YVYq2m5NKSXs3QrrsYIIirkFYhB%2BjAwG8bqBAuE9SQNeIthHjT27hB7KUeIIoXtwSVGutv70%3D&X-Amz-Date=20220808T175057Z&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=ASIARDVHBDXR4QCRLL4O%2F20220808%2Feu-west-1%2Fexecute-api%2Faws4_request&X-Amz-SignedHeaders=host&X-Amz-Signature=d58ab2b792a69f20f8e3bf9bf4fb19c8ec8f271bd1b594f8cd25c02d4c85521d";
+  const ws = new WebSocket(url);
+
+  ws.addEventListener("open", function (event) {
+    console.log("connected " + new Date().toISOString());
+  });
+
+  ws.addEventListener("message", function ({ data }) {
+    //debugger;
+    //console.log(`From server: ${data}`);
+
+    let parsed;
+    try {
+      parsed = JSON.parse(data);
+    } catch (err) {
+      console.error("Can not parse " + data);
+    }
+
+    addLog(parsed);
+
+    // if (parsed) {
+    //   console.log(
+    //     `\x1b[47m\x1b[34m${parsed.timestamp} \x1b[31m🍕 ${
+    //       parsed.serviceKey
+    //     }\x1b[0m\x1b[32m\n${JSON.stringify(parsed.data, null, 2)}\x1b[0m`
+    //   );
+    // }
+  });
+  ws.addEventListener("close", function () {
+    console.log("disconnected " + new Date().toISOString());
+  });
+
+  function addLog(data) {
+    const newRow = document.createElement("tr");
+    newRow.innerHTML = `
+    <td>${new Date(data.timestamp).toLocaleTimeString()}</td>
+    <td>${data.serviceKey}</td>
+    <td>${JSON.stringify(data.data)}</td>
+    `;
+    tableBody.append(newRow);
+  }
+}
+
+document.addEventListener("DOMContentLoaded", function () {
+  console.log("RUN");
+  run();
+});

package/cli/style.css

@@ -0,0 +1,43 @@
+.log {
+  white-space: nowrap
+}
+
+/* The Modal (background) 
+https://www.w3schools.com/howto/howto_css_modals.asp
+*/
+.modal {
+  display: none; /* Hidden by default */
+  position: fixed; /* Stay in place */
+  z-index: 1; /* Sit on top */
+  left: 0;
+  top: 0;
+  width: 100%; /* Full width */
+  height: 100%; /* Full height */
+  overflow: auto; /* Enable scroll if needed */
+  background-color: rgb(0,0,0); /* Fallback color */
+  background-color: rgba(0,0,0,0.4); /* Black w/ opacity */
+}
+
+/* Modal Content/Box */
+.modal-content {
+  background-color: #fefefe;
+  margin: 15% auto; /* 15% from the top and centered */
+  padding: 20px;
+  border: 1px solid #888;
+  width: 80%; /* Could be more or less, depending on screen size */
+}
+
+/* The Close Button */
+.close {
+  color: #aaa;
+  float: right;
+  font-size: 28px;
+  font-weight: bold;
+}
+
+.close:hover,
+.close:focus {
+  color: black;
+  text-decoration: none;
+  cursor: pointer;
+}

package/cli/ws.ts

@@ -0,0 +1,79 @@
+import * as fs from 'fs';
+import * as readline from 'readline';
+import WebSocket from 'ws';
+import { getWebSocketUrl } from '../common/getWebSocketUrl';
+
+async function run() {
+  // var credentialsProvider = awsCp.fromTemporaryCredentials();
+  // const credentials = await credentialsProvider();
+
+  // const client = new STSClient({});
+  // client.send()
+
+  // const command = new GetSessionTokenCommand({});
+  // const { Credentials: credentials } = await client.send(command);
+  // //const credentials = s.Credentials;
+
+  // console.log("credentials", credentials);
+
+  // const url = process.argv[2];
+  const url = 'wss://preh1xo1xh.execute-api.eu-west-1.amazonaws.com/prod';
+
+  // https://medium.com/@o.bredenberg/iam-sign-your-api-gateway-websocket-connection-request-no-custom-auth-from-your-frontend-65451166757d
+  const url2 = await getWebSocketUrl(url);
+
+  const ws = new WebSocket(url2);
+
+  const fileDescriptor = fs.openSync('spy_log.json', 'w');
+  // 📢📁⚡️💾 💽 💾 💿 📀🛢🪣📑🔊
+
+  ws.on('open', () => console.log(`connected ${new Date().toISOString()}`));
+  ws.on('message', (data: any) => {
+    // console.log(`From server: ${data}`);
+
+    let parsed;
+    try {
+      parsed = JSON.parse(data);
+    } catch (err) {
+      console.error(`Can not parse ${data}`);
+    }
+
+    if (parsed) {
+      console.log(
+        `\x1b[47m\x1b[34m${parsed.timestamp} \x1b[31m🍕 ${
+          parsed.serviceKey
+        }\x1b[0m\x1b[32m\n${JSON.stringify(parsed.data, null, 2)}\x1b[0m`
+      );
+    }
+
+    fs.write(fileDescriptor, data, (error) => {
+      if (!error) {
+        // fs.close(fileDescriptor);
+      } else {
+        console.error(error);
+      }
+    });
+    fs.write(fileDescriptor, '\n', (error) => {
+      if (!error) {
+        // fs.close(fileDescriptor);
+      } else {
+        console.error(error);
+      }
+    });
+  });
+  ws.on('close', () => {
+    console.log(`disconnected ${new Date().toISOString()}`);
+    fs.close(fileDescriptor);
+    process.exit();
+  });
+
+  readline
+    .createInterface({
+      input: process.stdin,
+      output: process.stdout,
+    })
+    .on('line', (data) => {
+      ws.send(data);
+    });
+}
+run().catch(console.error);

package/common/getWebSocketUrl.ts

@@ -0,0 +1,68 @@
+import { fromNodeProviderChain } from '@aws-sdk/credential-providers';
+import { Credentials } from '@aws-sdk/types';
+import * as aws4 from 'aws4';
+
+// ""wss://m6g3w6ttdh.execute-api.eu-west-1.amazonaws.com/prod";"
+
+export async function getWebSocketUrl(url: string, credentials?: Credentials) {
+  const urlParsed = parseUrl(url);
+  const hostParts = urlParsed!.host.split('.');
+
+  console.log('urlParsed', urlParsed);
+
+  if (!credentials) {
+    const credentialsProvider = fromNodeProviderChain();
+    credentials = await credentialsProvider();
+  }
+
+  const AWS_REGION = hostParts[2]; // The region of your API-gateway
+  const SOCKET_HOST = hostParts[0]; // Your API-gateway ID
+  const ENV = urlParsed!.pathname!.replace(/^\//, ''); // The stage of your target deployment
+  const WEBSOCKET_URL = `${SOCKET_HOST}.execute-api.${AWS_REGION}.amazonaws.com`; // Don't prepend with wss!
+  // const AWS_REGION = "eu-west-1"; // The region of your API-gateway
+  // const SOCKET_HOST = "m6g3w6ttdh"; // Your API-gateway ID
+  // const ENV = "prod"; // The stage of your target deployment
+  // const WEBSOCKET_URL = `${SOCKET_HOST}.execute-api.${AWS_REGION}.amazonaws.com`; // Don't prepend with wss!
+
+  // Get a signed path
+  const { path } = aws4.sign(
+    {
+      method: 'GET',
+      host: WEBSOCKET_URL,
+      // path: `/${ENV}?X-Amz-Security-Token=${encodeURIComponent(
+      //   credentials!.SessionToken!
+      // )}`,
+      path: `/${ENV}`,
+      service: 'execute-api',
+      region: AWS_REGION,
+      signQuery: true,
+    },
+    {
+      accessKeyId: credentials.accessKeyId,
+      secretAccessKey: credentials.secretAccessKey,
+      sessionToken: credentials.sessionToken,
+    }
+  );
+
+  const url2 = `wss://${WEBSOCKET_URL}${path}`;
+  console.log('path', url2);
+  return url2;
+}
+
+function parseUrl(href: string) {
+  const match = href.match(
+    /^(wss?:)\/\/(([^:/?#]*)(?::([0-9]+))?)([/]{0,1}[^?#]*)(\?[^#]*|)(#.*|)$/
+  );
+  return (
+    match && {
+      href,
+      protocol: match[1],
+      host: match[2],
+      hostname: match[3],
+      port: match[4],
+      pathname: match[5],
+      search: match[6],
+      hash: match[7],
+    }
+  );
+}

package/common/package.json

@@ -0,0 +1,22 @@
+{
+  "name": "serverless-spy-common",
+  "version": "1.0.0",
+  "description": "",
+  "main": "index.js",
+  "private": true,
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "author": "",
+  "license": "ISC",
+  "devDependencies": {
+    "@types/aws4": "^1.11.2"
+  },
+  "dependencies": {
+    "@aws-sdk/credential-providers": "^3.145.0",
+    "@aws-sdk/types": "^3.127.0",
+    "@aws-sdk/util-dynamodb": "^3.145.0",
+    "aws-lambda": "^1.0.7",
+    "aws4": "^1.11.0"
+  }
+}

package/common/spyEvents/DynamoDBSpyEvent.ts

@@ -0,0 +1,12 @@
+import { NativeAttributeValue } from '@aws-sdk/util-dynamodb';
+import { DynamoDBRecord } from 'aws-lambda';
+import { SpyEvent } from './SpyEvent';
+
+// asdasd
+export interface DynamoDBSpyEvent<TData = any> extends SpyEvent {
+  spyEventType: 'DynamoDB';
+  eventName: DynamoDBRecord['eventName'];
+  newImage: TData;
+  keys: Record<string, NativeAttributeValue>;
+  oldImage?: TData;
+}

package/common/spyEvents/EventBridgeRuleSpyEvent.ts

@@ -0,0 +1,13 @@
+import { SpyEvent } from './SpyEvent';
+
+export interface EventBridgeRuleSpyEvent<
+  MessageType = any,
+  EventBridgeDetailType extends string = string
+> extends SpyEvent {
+  spyEventType: 'EventBridgeRule';
+  detail: MessageType;
+  detailType: EventBridgeDetailType;
+  source: string;
+  time: string;
+  account: string;
+}

package/common/spyEvents/EventBridgeSpyEvent.ts

@@ -0,0 +1,13 @@
+import { SpyEvent } from './SpyEvent';
+
+export interface EventBridgeSpyEvent<
+  MessageType = any,
+  EventBridgeDetailType extends string = string
+> extends SpyEvent {
+  spyEventType: 'EventBridge';
+  detail: MessageType;
+  detailType: EventBridgeDetailType;
+  source: string;
+  time: string;
+  account: string;
+}

package/common/spyEvents/FunctionConsoleSpyEvent.ts

@@ -0,0 +1,11 @@
+import { SpyEvent } from './SpyEvent';
+
+export interface FunctionConsoleSpyEvent<TRequest = any> extends SpyEvent {
+  spyEventType: 'FunctionConsole';
+  request: TRequest;
+  console: {
+    type: 'log' | 'debug' | 'info' | 'error' | 'warn';
+    message?: any;
+    optionalParams: any[];
+  };
+}

package/common/spyEvents/FunctionContext.ts

@@ -0,0 +1,8 @@
+import { ClientContext, CognitoIdentity } from 'aws-lambda';
+
+export interface FunctionContext {
+  functionName: string;
+  awsRequestId: string;
+  identity?: CognitoIdentity | undefined;
+  clientContext?: ClientContext | undefined;
+}

package/common/spyEvents/FunctionErrorSpyEvent.ts

@@ -0,0 +1,9 @@
+import { FunctionContext } from './FunctionContext';
+import { SpyEvent } from './SpyEvent';
+
+export interface FunctionErrorSpyEvent<TRequest = any> extends SpyEvent {
+  spyEventType: 'FunctionError';
+  request: TRequest;
+  context: FunctionContext;
+  error: any;
+}

package/common/spyEvents/FunctionRequestSpyEvent.ts

@@ -0,0 +1,8 @@
+import { FunctionContext } from './FunctionContext';
+import { SpyEvent } from './SpyEvent';
+
+export interface FunctionRequestSpyEvent<TData = any> extends SpyEvent {
+  spyEventType: 'FunctionRequest';
+  request: TData;
+  context: FunctionContext;
+}

package/common/spyEvents/FunctionResponseSpyEvent.ts

@@ -0,0 +1,10 @@
+import { FunctionContext } from './FunctionContext';
+import { SpyEvent } from './SpyEvent';
+
+export interface FunctionResponseSpyEvent<TRequest = any, TResponse = any>
+  extends SpyEvent {
+  spyEventType: 'FunctionResponse';
+  request: TRequest;
+  context: FunctionContext;
+  response: TResponse;
+}

package/common/spyEvents/S3SpyEvent.ts

@@ -0,0 +1,9 @@
+import { SpyEvent } from './SpyEvent';
+
+export interface S3SpyEvent extends SpyEvent {
+  spyEventType: 'S3';
+  eventName: string;
+  eventTime: string;
+  bucket: string;
+  key: string;
+}

package/common/spyEvents/SnsSubscriptionSpyEvent.ts

@@ -0,0 +1,12 @@
+import { SNSMessageAttributes } from 'aws-lambda';
+import { SpyEvent } from './SpyEvent';
+
+export interface SnsSubscriptionSpyEvent<MessageType = any> extends SpyEvent {
+  spyEventType: 'FunctionSnsSubscription';
+  message: MessageType;
+  subject: string;
+  timestamp: string;
+  topicArn: string;
+  messageId: string;
+  messageAttributes: SNSMessageAttributes;
+}

package/common/spyEvents/SnsTopicSpyEvent.ts

@@ -0,0 +1,12 @@
+import { SNSMessageAttributes } from 'aws-lambda';
+import { SpyEvent } from './SpyEvent';
+
+export interface SnsTopicSpyEvent<MessageType = any> extends SpyEvent {
+  spyEventType: 'FunctionSnsTopic';
+  message: MessageType;
+  subject: string;
+  timestamp: string;
+  topicArn: string;
+  messageId: string;
+  messageAttributes: SNSMessageAttributes;
+}

package/common/spyEvents/SpyEvent.ts

@@ -0,0 +1,3 @@
+export interface SpyEvent {
+  spyEventType: string;
+}

package/common/spyEvents/SpyMessage.ts

@@ -0,0 +1,7 @@
+import { SpyEvent } from './SpyEvent';
+
+export type SpyMessage<T extends SpyEvent = SpyEvent> = {
+  timestamp: string;
+  serviceKey: string;
+  data: T;
+};

package/common/spyEvents/SqsSpyEvent.ts

@@ -0,0 +1,8 @@
+import { SQSMessageAttributes } from 'aws-lambda';
+import { SpyEvent } from './SpyEvent';
+
+export interface SqsSpyEvent<MessageType = any> extends SpyEvent {
+  spyEventType: 'Sqs';
+  body: MessageType;
+  messageAttributes: SQSMessageAttributes;
+}

package/functions/onConnect.ts

@@ -0,0 +1,29 @@
+import { DynamoDBClient, PutItemCommand } from '@aws-sdk/client-dynamodb';
+import { APIGatewayEvent } from 'aws-lambda';
+
+const ddb = new DynamoDBClient({
+  region: process.env.AWS_REGION,
+});
+
+exports.handler = async (event: APIGatewayEvent) => {
+  console.log('EVENT', JSON.stringify(event));
+
+  const putParams = new PutItemCommand({
+    TableName: process.env.TABLE_NAME as string,
+    Item: {
+      connectionId: { S: event.requestContext.connectionId! },
+    },
+  });
+
+  try {
+    await ddb.send(putParams);
+  } catch (err) {
+    console.error(err);
+    return {
+      statusCode: 500,
+      body: 'Failed to connect: ' + JSON.stringify(err),
+    };
+  }
+
+  return { statusCode: 200, body: 'Connected.' };
+};

package/functions/onDisconnect.ts

@@ -0,0 +1,28 @@
+import { DeleteItemCommand, DynamoDBClient } from '@aws-sdk/client-dynamodb';
+import { APIGatewayEvent } from 'aws-lambda';
+
+const ddb = new DynamoDBClient({
+  region: process.env.AWS_REGION,
+});
+export const handler = async (event: APIGatewayEvent) => {
+  console.log('EVENT', JSON.stringify(event));
+
+  const deleteParams = new DeleteItemCommand({
+    TableName: process.env.TABLE_NAME,
+    Key: {
+      connectionId: { S: event.requestContext.connectionId! },
+    },
+  });
+
+  try {
+    await ddb.send(deleteParams);
+  } catch (err) {
+    console.error(err);
+    return {
+      statusCode: 500,
+      body: 'Failed to disconnect: ' + JSON.stringify(err),
+    };
+  }
+
+  return { statusCode: 200, body: 'Disconnected.' };
+};