chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,75 @@
|
||||
import { isComputeRegionAccessible, resolveComputeAccess } from "~/v3/regionAccess.server";
|
||||
import { BaseService, ServiceValidationError } from "./baseService.server";
|
||||
|
||||
export class SetDefaultRegionService extends BaseService {
|
||||
public async call({
|
||||
projectId,
|
||||
regionId,
|
||||
isAdmin = false,
|
||||
}: {
|
||||
projectId: string;
|
||||
regionId: string;
|
||||
isAdmin?: boolean;
|
||||
}) {
|
||||
const workerGroup = await this._prisma.workerInstanceGroup.findFirst({
|
||||
where: {
|
||||
id: regionId,
|
||||
},
|
||||
});
|
||||
|
||||
if (!workerGroup) {
|
||||
throw new ServiceValidationError("Region not found");
|
||||
}
|
||||
|
||||
const project = await this._prisma.project.findFirst({
|
||||
where: {
|
||||
id: projectId,
|
||||
},
|
||||
include: {
|
||||
organization: { select: { featureFlags: true } },
|
||||
},
|
||||
});
|
||||
|
||||
if (!project) {
|
||||
throw new ServiceValidationError("Project not found");
|
||||
}
|
||||
|
||||
// If their project is restricted, only allow them to set default regions that are allowed
|
||||
if (!isAdmin) {
|
||||
if (project.allowedWorkerQueues.length > 0) {
|
||||
if (!project.allowedWorkerQueues.includes(workerGroup.masterQueue)) {
|
||||
throw new ServiceValidationError("You're not allowed to set this region as default");
|
||||
}
|
||||
} else {
|
||||
if (workerGroup.hidden) {
|
||||
throw new ServiceValidationError("This region is not available to you");
|
||||
}
|
||||
|
||||
if (workerGroup.workloadType === "MICROVM") {
|
||||
const hasComputeAccess = await resolveComputeAccess(
|
||||
this._prisma,
|
||||
project.organization.featureFlags
|
||||
);
|
||||
|
||||
if (!isComputeRegionAccessible(workerGroup, hasComputeAccess)) {
|
||||
throw new ServiceValidationError("This region requires compute access");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
await this._prisma.project.update({
|
||||
where: {
|
||||
id: projectId,
|
||||
},
|
||||
data: {
|
||||
defaultWorkerGroupId: regionId,
|
||||
},
|
||||
});
|
||||
|
||||
return {
|
||||
id: workerGroup.id,
|
||||
name: workerGroup.name,
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user