checkpoint

src/hooks/useDatabaseAccounts.tsx

@@ -0,0 +1,52 @@
+import { useContext, useEffect, useState } from "react";
+import { AuthContext } from "../contexts/authContext";
+import { DatabaseAccount } from "../Contracts/DataModels";
+
+interface AccountListResult {
+  nextLink: string;
+  value: DatabaseAccount[];
+}
+
+export async function fetchDatabaseAccounts(
+  subscriptionIds: string[],
+  accessToken: string
+): Promise<DatabaseAccount[]> {
+  const headers = new Headers();
+  const bearer = `Bearer ${accessToken}`;
+
+  headers.append("Authorization", bearer);
+
+  if (!subscriptionIds || !subscriptionIds.length) {
+    return Promise.reject("No subscription passed in");
+  }
+
+  let accounts: Array<DatabaseAccount> = [];
+
+  const subscriptionFilter = "subscriptionId eq '" + subscriptionIds.join("' or subscriptionId eq '") + "'";
+  const urlFilter = `$filter=(${subscriptionFilter}) and (resourceType eq 'microsoft.documentdb/databaseaccounts')`;
+  let nextLink = `https://management.azure.com/resources?api-version=2020-01-01&${urlFilter}`;
+
+  while (nextLink) {
+    const response: Response = await fetch(nextLink, { headers });
+    const result: AccountListResult =
+      response.status === 204 || response.status === 304 ? undefined : await response.json();
+    if (!response.ok) {
+      throw result;
+    }
+    nextLink = result.nextLink;
+    accounts = [...accounts, ...result.value];
+  }
+  return accounts;
+}
+
+export function useDatabaseAccounts(subscriptionId: string): DatabaseAccount[] {
+  const { armToken } = useContext(AuthContext);
+  const [state, setState] = useState<DatabaseAccount[]>();
+
+  useEffect(() => {
+    if (subscriptionId && armToken) {
+      fetchDatabaseAccounts([subscriptionId], armToken).then(response => setState(response));
+    }
+  }, [subscriptionId, armToken]);
+  return state || [];
+}