Merge pull request #124 from metagov/feat/snapshot

feat: support pagination to get proposal from snapshot.
metagov · Aug 23, 2023 · 0c0dbc1 · 0c0dbc1
2 parents 191647e + 946164f
commit 0c0dbc1
Showing 1 changed file with 83 additions and 77 deletions.
diff --git a/Implementations/API/backend/functions/snapshot/getProposals.ts b/Implementations/API/backend/functions/snapshot/getProposals.ts
@@ -1,68 +1,73 @@
-import { APIGatewayProxyHandlerV2 } from 'aws-lambda'
-import { snapshotApiConfig } from 'functions/config'
-import fetch from 'node-fetch'
+import { APIGatewayProxyHandlerV2 } from "aws-lambda";
+import { snapshotApiConfig } from "functions/config";
+import fetch from "node-fetch";
 
 function apiRequest(path: string, method: string, data: any) {
-    return fetch(path, {
-        headers: {
-            'Content-Type': 'application/json',
-        },
-        method,
-        redirect: 'follow',
-        body: JSON.stringify(data),
-    }).then((res) => res.json())
+  return fetch(path, {
+    headers: {
+      "Content-Type": "application/json",
+    },
+    method,
+    redirect: "follow",
+    body: JSON.stringify(data),
+  }).then((res) => res.json());
 }
 
 const calculateSatatus = (
-    processed: boolean,
-    cancelled: boolean,
-    passed: boolean,
-    votingStarts: number,
-    votingEnds: number,
-    graceEnds: number,
-    sponsored: boolean
+  processed: boolean,
+  cancelled: boolean,
+  passed: boolean,
+  votingStarts: number,
+  votingEnds: number,
+  graceEnds: number,
+  sponsored: boolean
 ) => {
-    const now = Date.now()
-    if (processed) {
-        return 'processed'
-    } else if (passed) {
-        return 'passed'
-    } else if (cancelled) {
-        return 'cancelled'
-    } else if (now >= votingEnds && graceEnds > now) {
-        return 'grace'
-    } else if (now >= votingStarts && votingEnds > now) {
-        return 'voting'
-    } else if (sponsored) {
-        return 'sponsored'
-    } else {
-        return 'submitted'
-    }
-}
+  const now = Date.now();
+  if (processed) {
+    return "processed";
+  } else if (passed) {
+    return "passed";
+  } else if (cancelled) {
+    return "cancelled";
+  } else if (now >= votingEnds && graceEnds > now) {
+    return "grace";
+  } else if (now >= votingStarts && votingEnds > now) {
+    return "voting";
+  } else if (sponsored) {
+    return "sponsored";
+  } else {
+    return "submitted";
+  }
+};
 
 export const handler: APIGatewayProxyHandlerV2 = async (event) => {
-    console.log({graphConfig: snapshotApiConfig})
-    const path = snapshotApiConfig['1']
-    if (!path) return { statusCode: 400, message: 'Missing config for network' }
+  console.log({ graphConfig: snapshotApiConfig });
+  const path = snapshotApiConfig["1"];
+  if (!path) return { statusCode: 400, message: "Missing config for network" };
+
+  const eventId = event?.pathParameters?.id;
+  if (!eventId) return { statusCode: 400, message: "Missing id" };
 
-    const eventId = event?.pathParameters?.id
-    if (!eventId) return { statusCode: 400, message: 'Missing id' }
+  const take = parseInt(<string>event?.queryStringParameters?.take) || 20; // Get the requested page number
+  const skip = parseInt(<string>event?.queryStringParameters?.skip) || 0; // Number of items per page
 
-    const template = {
-        '@context': {
-            '@vocab': 'http://daostar.org/',
-        },
-        type: 'DAO',
-        name: eventId,
-    }
+  const template = {
+    "@context": {
+      "@vocab": "http://daostar.org/",
+    },
+    type: "DAO",
+    name: eventId,
+  };
 
-    const query = `query GetProposals($dao: String!) {
+  const query = `query GetProposals($dao: String!, $first: Int!, $skip: Int!) {
         proposals (
             where: {
               space_in: [$dao],
             },
             orderBy: "created",
-            orderDirection: desc
+            orderDirection: desc,
+            first: $first,
+            skip: $skip
           ) {
             id
             title
@@ -78,37 +83,38 @@ export const handler: APIGatewayProxyHandlerV2 = async (event) => {
               name
             }
           }
-      }`
+      }`;
 
-    const data = {
-        query,
-        variables: { dao: eventId },
-    }
+  const data = {
+    query,
+    variables: { dao: eventId, first: take, skip },
+  };
 
-    const res = (await apiRequest(path, 'POST', data)) as any
+  const res = (await apiRequest(path, "POST", data)) as any;
 
-    if (!(res.data.proposals && res.data.proposals.length > 0)) return { statusCode: 404, message: 'DAO not found' }
-    console.log({ data: res.data.proposals })
-    const proposals = res.data.proposals
+  if (!(res.data.proposals && res.data.proposals.length > 0))
+    return { statusCode: 404, message: "DAO not found" };
+  console.log({ data: res.data.proposals });
+  const proposals = res.data.proposals;
 
-    const proposalsFormatted = proposals.map((p: any) => {
-        const status = p.state
-        return {
-            id: p.id,
-            type: 'proposal',
-            status: status,
-        }
-    })
+  const proposalsFormatted = proposals.map((p: any) => {
+    const status = p.state;
+    return {
+      id: p.id,
+      type: "proposal",
+      status: status,
+    };
+  });
 
-    const transformed = { proposals: proposalsFormatted, ...template }
+  const transformed = { proposals: proposalsFormatted, ...template };
 
-    return transformed
-        ? {
-              statusCode: 200,
-              body: JSON.stringify(transformed),
-          }
-        : {
-              statusCode: 404,
-              body: JSON.stringify({ error: true }),
-          }
-}
+  return transformed
+    ? {
+        statusCode: 200,
+        body: JSON.stringify(transformed),
+      }
+    : {
+        statusCode: 404,
+        body: JSON.stringify({ error: true }),
+      };
+};