Create projects.ts controller

src/controller/projects.ts

@@ -0,0 +1,94 @@
+import { IProject, IProjectPlan, Project } from "@models/Project";
+import { Types } from "mongoose";
+import crypto from "crypto";
+import { AccessKey, IKeyRole } from "@models/AccessKey";
+import { Member } from "@models/Member";
+import { planLimits } from "../limits/plans";
+import { License } from "@models/License";
+import { Permission } from "@models/Permission";
+import { Group } from "@models/Group";
+import { MetaData } from "@models/MetaData";
+
+export const checkProjectAccess = (requiredPermission: IKeyRole) => async (userId: string, projectId: string) => {
+    if (!Types.ObjectId.isValid(projectId))
+        return { code: 3, message: "Invalid object id provided" };
+
+    let project = await Project.findOne({ _id: projectId, creatorId: userId });
+    if (project !== null) return project;
+
+    const projectMember = await Member.findOne({ memberId: userId || "", accepted: true });
+    if (projectMember === null) return { code: 5009, message: "The provided project id does not exist" };
+
+    project = await Project.findById(projectMember.projectId);
+    if (project === null) return { code: 5009, message: "The provided project id does not exist" };
+
+    if (projectMember.role === IKeyRole.ADMIN) return project;
+    if (requiredPermission === IKeyRole.MANAGE && projectMember.role === IKeyRole.MANAGE) return project;
+    if (requiredPermission === IKeyRole.VIEW && (projectMember.role === IKeyRole.VIEW || projectMember.role === IKeyRole.MANAGE)) return project;
+
+    return { code: 5009, message: "The provided project id does not exist" };
+};
+
+const projectMapper = (project: IProject) => ({
+    id: project._id, name: project.name, validationKey: project.validationKey,
+    defaults: project.defaults,
+});
+
+export const listProjects = async (userId?: string) => {
+    const projects = await Project.find({ creatorId: userId || "" });
+
+    const memberProjects = await Member.find({ memberId: userId || "", accepted: true });
+    for (const project of memberProjects) {
+        const foundProject = await Project.findById(project.projectId);
+        if (foundProject !== null) projects.push(foundProject);
+    }
+
+    return projects.map(project => projectMapper(project));
+};
+
+export const getProject = async (projectId: string, userId: string) => {
+    const project = await checkProjectAccess(IKeyRole.VIEW)(userId, projectId);
+    if ("code" in project) return project;
+
+    return projectMapper(project);
+};
+
+export const createProject = async (name: string, userId: string) => {
+    const count = await Project.countDocuments({ creatorId: userId,
+        plan: IProjectPlan.PERSONAL });
+
+    if (count > planLimits["account"].FREE_PROJECTS) return { code: 95, message: "You have exceeded the free project limit" };
+    if (count > 100) return { code: 95, message: "You have exceeded the project limit" };
+
+    await Project.create({ name, creatorId: userId });
+};
+
+export const deleteProject = async (id: string, userId: string) => {
+    const project = await checkProjectAccess(IKeyRole.ADMIN)(userId, id);
+    if ("code" in project) return project;
+
+    License.deleteMany({ projectId: project.id });
+    Permission.deleteMany({ projectId: project.id });
+    Group.deleteMany({ projectId: project.id });
+    MetaData.deleteMany({ projectId: project.id });
+    AccessKey.deleteMany({ projectId: project.id });
+    Member.deleteMany({ projectId: project.id });
+
+    await project?.deleteOne();
+};
+
+export const patchProject = async (id: string, userId: string, config: {
+    name: string, defaults: { licenseKey: string, groups: [], permissions: [], expirationDate: Date }
+}) => {
+    const project = await checkProjectAccess(IKeyRole.MANAGE)(userId, id);
+    if ("code" in project) return project;
+
+    await project.updateOne({ name: config.name, defaults: Object.assign(project.defaults, config.defaults) });
+};
+
+export const regenerateKey = async (id: string, userId: string) => {
+    const project = await checkProjectAccess(IKeyRole.MANAGE)(userId, id);
+    if ("code" in project) return project;
+
+    await project.updateOne({ validationKey: crypto.randomBytes(24).toString("hex") });
+};