cosmos-explorer/src/Common/dataAccess/deleteUserDefinedFunction.ts

import { AuthType } from "../../AuthType";
import { client } from "../CosmosClient";
import { deleteSqlUserDefinedFunction } from "../../Utils/arm/generatedClients/2020-04-01/sqlResources";
import { handleError } from "../ErrorHandlingUtils";
import { logConsoleProgress } from "../../Utils/NotificationConsoleUtils";
import { userContext } from "../../UserContext";

export async function deleteUserDefinedFunction(databaseId: string, collectionId: string, id: string): Promise<void> {
  const clearMessage = logConsoleProgress(`Deleting user defined function ${id}`);
  try {
    if (window.authType === AuthType.AAD && !userContext.useSDKOperations) {
      await deleteSqlUserDefinedFunction(
        userContext.subscriptionId,
        userContext.resourceGroup,
        userContext.databaseAccount.name,
        databaseId,
        collectionId,
        id
      );
    } else {
      await client()
        .database(databaseId)
        .container(collectionId)
        .scripts.userDefinedFunction(id)
        .delete();
    }
  } catch (error) {
    handleError(error, `Error while deleting user defined function ${id}`, "DeleteUserDefinedFunction");
    throw error;
  } finally {
    clearMessage();
  }
}