import { useRouter } from "next/router";
import { useCallback } from "react";
import { useInitialData } from "hooks/useServerSideData";
import { useProfile } from "./profile";
import { useCurrentProject } from "./projects";
import {
useBBQuery,
useManagementApiMutation,
useManagementApiQuery,
useMutate,
} from "./api";
export function useDeployments(projectId?: number) {
const [initialData] = useInitialData();
const { data, isLoading } = useManagementApiQuery({
path: "/projects/{project_id}/list_deployments",
pathParams: {
project_id: projectId || 0,
},
queryParams: {
includeLocal: true,
},
swrOptions: {
revalidateOnMount: initialData === undefined,
refreshInterval: 30 * 1000,
keepPreviousData: false,
},
});
return { deployments: data, isLoading };
}
export function useDefaultDevDeployment(projectId: number | undefined) {
const member = useProfile();
const { deployments } = useDeployments(projectId);
const cloudDev = deployments?.find(
(d) =>
d.deploymentType === "dev" &&
d.kind === "cloud" &&
d.creator === member?.id &&
d.isDefault,
);
const localDev = deployments?.find(
(d) =>
d.deploymentType === "dev" &&
d.kind === "local" &&
d.creator === member?.id &&
d.isActive,
);
// Prefer local deployments if they exist.
return localDev ?? cloudDev;
}
export function useCurrentDeployment() {
const project = useCurrentProject();
const { deployments, isLoading } = useDeployments(project?.id);
const { push, query } = useRouter();
const deployment = deployments?.find((d) => d.name === query?.deploymentName);
// The deployment doesn't exist.
if (
!isLoading &&
project &&
deployments &&
deployments.length > 0 &&
!deployment &&
!!query.deploymentName
) {
void push("/404");
}
return deployment;
}
export function useProvisionDeployment(projectId: number) {
return useManagementApiMutation({
path: "/projects/{project_id}/create_deployment",
pathParams: {
project_id: projectId,
},
mutateKey: `/projects/{project_id}/list_deployments`,
mutatePathParams: {
project_id: projectId,
},
});
}
export function useDeploymentById(
teamId: number,
deploymentId?: number | string,
) {
const { data: deployment } = useBBQuery({
path: "/teams/{team_id}/deployments/{deployment_id}",
pathParams: {
team_id: teamId,
deployment_id: deploymentId?.toString() || "",
},
});
return deployment;
}
export function useDeleteDeployment(
projectId: number,
deploymentName: string,
settingsUrl: string,
) {
const deleteDeployment = useManagementApiMutation({
path: "/deployments/{deployment_name}/delete",
method: "post",
pathParams: { deployment_name: deploymentName || "" },
mutateKey: `/projects/{project_id}/list_deployments`,
mutatePathParams: { project_id: projectId },
successToast: "Deleted deployment.",
redirectTo: settingsUrl,
});
const mutateBBApi = useMutate();
return useCallback(async () => {
await deleteDeployment();
// Manually revalidate the BB API instances list
await mutateBBApi(
[
"/projects/{project_id}/instances",
{
params: {
path: { project_id: projectId.toString() },
},
},
],
undefined,
{ revalidate: true },
);
}, [deleteDeployment, mutateBBApi, projectId]);
}
