added httpResponse and stock classes

Author: Bob Wall

sample-apps/robinhood-for-reddit/handlers/hotStock/hotStockHandler.ts

@@ -1,5 +1,5 @@
 import { Context, Handler } from 'aws-lambda';
-import { IHttpResponse, IStock } from './types';
+import { HttpResponse, Stock } from './types';
 import { HotStockService } from './hotStockService';
 
 /**
@@ -8,26 +8,53 @@ import { HotStockService } from './hotStockService';
  * @param context context object
  */
 const process: Handler = async (event: any, context: Context) => {
-  console.info('hotStockHandler.process', { event });
-
   try {
+    console.info('hotStockHandler.process', { event });
+
     // get the array of hot stocks from the hot stock service
-    const hotStocks: IStock[] = await HotStockService.getHotStocks();
+    const hotStocks: Stock[] = await HotStockService.getHotStocks();
 
-    // create a valid response object
-    const response: IHttpResponse = {
+    /*
+      create a valid response object
+        - sets cors and cookie settings
+        - sets to cache for 4 hours
+    */
+    const response = new HttpResponse({
       statusCode: 200,
       body: JSON.stringify(hotStocks),
       headers: {
-        "Access-Control-Allow-Origin" : "*", // Required for CORS support to work
-        "Access-Control-Allow-Credentials" : true // Required for cookies, authorization headers with HTTPS
+        "Access-Control-Allow-Origin" : "*",                        // Required for CORS support to work
+        "Access-Control-Allow-Credentials" : true,                  // Required for cookies, authorization headers with HTTPS
+        "Cache-Control": "public, max-age=14400, s-maxage=14400'"   // cache the response for 4 hours
+      }
+    });
+
+    // return the response back
+    context.succeed(response);
+  } catch (error) {
+    // report the error to the console
+    console.error('hotStockHandler.error', { error });
+
+    let body = '';
+
+    try {
+      body = JSON.stringify(error);
+    } catch (ex) {
+      body = 'Unable to stringify result';
+    }
+
+    const response = new HttpResponse({
+      statusCode: 500,
+      body,
+      headers: {
+        "Access-Control-Allow-Origin" : "*",                        // Required for CORS support to work
+        "Access-Control-Allow-Credentials" : true,                  // Required for cookies, authorization headers with HTTPS
+        "Cache-Control": "no-cache"                                 // No Cache
       }
-    };
+    });
 
+    // return back as success, but this is has a failed statusCode
     context.succeed(response);
-  } catch (exception) {
-    console.error('hotStockHandler.error', { exception });
-    context.fail(exception);
   }
 };
 

sample-apps/robinhood-for-reddit/handlers/hotStock/types/IHttpResponse.ts

@@ -1,5 +1,5 @@
 export interface IHttpResponse {
-  statusCode: number;
-  headers?: any;
   body?: string;
+  headers?: any;
+  statusCode: number;
 }

sample-apps/robinhood-for-reddit/handlers/hotStock/types/httpResponse.ts

@@ -0,0 +1,15 @@
+import { IHttpResponse } from './IHttpResponse';
+
+export class HttpResponse implements IHttpResponse {
+
+  public body?: string;
+  public headers?: any;
+  public statusCode: number;
+
+  constructor(response: IHttpResponse) {
+    this.body = response.body;   
+    this.headers = response.headers;
+    this.statusCode = response.statusCode;
+  }
+
+}

sample-apps/robinhood-for-reddit/handlers/hotStock/types/index.ts

@@ -1,3 +1,4 @@
+export * from './httpResponse';
 export * from './IHttpResponse';
 export * from './IStock';
 export * from './StockMarket';

sample-apps/robinhood-for-reddit/handlers/hotStock/types/stock.ts

@@ -0,0 +1,34 @@
+import { IStock } from './IStock';
+
+export class Stock implements IStock {
+
+  public bloombergTerminalUrl?: string;
+  public companyName: string;
+  public companyUrl: string;
+  public hasElonMuskTweetedAboutIt: boolean;
+  public id: string;
+  public isHot: boolean;
+  public marketSymbol: string;
+  public redditMentions: number;
+  public stockMarket: StockMarket;
+  public updatedAt: Date;
+  public value: number;
+
+  constructor(response: IHttpResponse) {
+    this.id = (response.id)
+      ? response.id
+      :uuid();
+
+    this.bloombergTerminalUrl = response.bloombergTerminalUrl;
+    this.companyName = response.companyName;
+    this.companyUrl = response.companyUrl;
+    this.hasElonMuskTweetedAboutIt = response.hasElonMuskTweetedAboutIt;
+    this.isHot = response.isHot;
+    this.marketSymbol = response.marketSymbol;
+    this.redditMentions = response.redditMentions;
+    this.stockMarket = response.stockMarket;
+    this.updatedAt = response.updatedAt;
+    this.value = response.value;
+  }
+
+}