cosmos-explorer/src/hooks/useDatabases.tsx

import { DatabaseModel } from "Contracts/DataModels";
import useSWR from "swr";
import { getDatabaseEndpoint } from "../Common/DatabaseAccountUtility";
import { configContext } from "../ConfigContext";
import { ApiType } from "../UserContext";
import { getCopyJobAuthorizationHeader } from "../Utils/CopyJobAuthUtils";

const apiVersion = "2023-09-15";
export interface FetchDatabasesListParams {
  subscriptionId: string;
  resourceGroupName: string;
  accountName: string;
  apiType?: ApiType;
}

const buildReadDatabasesListUrl = (params: FetchDatabasesListParams): string => {
  const { subscriptionId, resourceGroupName, accountName, apiType } = params;
  const databaseEndpoint = getDatabaseEndpoint(apiType);

  let armEndpoint = configContext.ARM_ENDPOINT;
  if (armEndpoint.endsWith("/")) {
    armEndpoint = armEndpoint.slice(0, -1);
  }
  return `${armEndpoint}/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/${databaseEndpoint}?api-version=${apiVersion}`;
};

const fetchDatabasesList = async (
  subscriptionId: string,
  resourceGroupName: string,
  accountName: string,
  apiType: ApiType,
): Promise<DatabaseModel[]> => {
  const uri = buildReadDatabasesListUrl({ subscriptionId, resourceGroupName, accountName, apiType });
  const headers = getCopyJobAuthorizationHeader();

  const response = await fetch(uri, {
    method: "GET",
    headers: headers,
  });

  if (!response.ok) {
    throw new Error("Failed to fetch databases");
  }

  const data = await response.json();
  return data.value;
};

export function useDatabases(
  subscriptionId: string,
  resourceGroupName: string,
  accountName: string,
  apiType: ApiType,
): DatabaseModel[] | undefined {
  const { data } = useSWR(
    () =>
      subscriptionId && resourceGroupName && accountName && apiType
        ? ["fetchDatabasesLinkedToResource", subscriptionId, resourceGroupName, accountName, apiType]
        : undefined,
    (_, subscriptionId, resourceGroupName, accountName, apiType) =>
      fetchDatabasesList(subscriptionId, resourceGroupName, accountName, apiType),
  );

  return data;
}