From 637693dda12c47113dc2f9b99f54d0146712bd52 Mon Sep 17 00:00:00 2001
From: Tanmai Gopal <tanmaig@hasura.io>
Date: Wed, 23 Oct 2024 03:59:02 -0700
Subject: [PATCH] The entire calendar app works end to end!

---
 src/duckduckapi.ts | 10 ++++++++++
 src/functions.ts   | 14 +++++++++++---
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/duckduckapi.ts b/src/duckduckapi.ts
index 76933ec..78aa269 100644
--- a/src/duckduckapi.ts
+++ b/src/duckduckapi.ts
@@ -12,6 +12,7 @@ import {
     Connector,
     Forbidden,
 } from "@hasura/ndc-sdk-typescript";
+import { JSONValue } from '@hasura/ndc-lambda-sdk';
 
 import path from "node:path"
 import { FunctionsSchema, getNdcSchema, printRelaxedTypesWarning } from "./lambda-sdk/schema";
@@ -276,4 +277,13 @@ export async function makeConnector(dda: duckduckapi): Promise<Connector<Configu
   };
 
   return Promise.resolve(connector);
+}
+
+export function getTokensFromHeader(headers: JSONValue, service: string): {access_token: string | null, refresh_token: string | null}  {
+  const oauthServices = headers.value as any;
+  const decodedServices = Buffer.from(oauthServices['x-hasura-oauth-services'] as string, 'base64').toString('utf-8');
+  const serviceTokens = JSON.parse(decodedServices);
+  const access_token = serviceTokens[service]?.access_token;
+  const refresh_token = serviceTokens[service]?.refresh_token;
+  return {access_token, refresh_token};
 }
\ No newline at end of file
diff --git a/src/functions.ts b/src/functions.ts
index 2bfc1fd..469c8c8 100644
--- a/src/functions.ts
+++ b/src/functions.ts
@@ -1,5 +1,6 @@
 import { JSONValue } from '@hasura/ndc-lambda-sdk';
 import { CalendarSyncManager } from './google-calendar-sync';
+import { getTokensFromHeader } from './duckduckapi';
 
 export let loaderStatus: string = 'stopped';
 
@@ -11,11 +12,18 @@ export let loaderStatus: string = 'stopped';
  */
 export async function __dda_loader_init(headers: JSONValue): Promise<string> {
 
-    console.log(JSON.stringify(headers.value));
+    const oauthServices = headers.value as any;
+    const {access_token} = getTokensFromHeader(headers, 'google-calendar');
+
+    if (!access_token) {
+      console.log(headers.value);
+      loaderStatus = 'google-calendar key not found in oauth services. Login to google-calendar?';
+      return loaderStatus;
+    }
 
     const syncManager = new CalendarSyncManager(
-      headers.value['google-calendar'].token,
-      1 // sync every 5 minutes
+      access_token,
+      1 // sync every minute
     );
     const result = await syncManager.test();