import debug from 'debug'; import assert from 'assert'; import { Database } from './database'; import { deploymentToGqlType, projectMemberToGqlType, projectToGqlType, environmentVariableToGqlType, isUserOwner } from './utils'; import { Environment } from './entity/Deployment'; const log = debug('snowball:database'); export const createResolvers = async (db: Database): Promise => { return { Query: { // TODO: add custom type for context user: (_: any, __: any, context: any) => { return db.getUser(context.userId); }, organizations: async (_:any, __: any, context: any) => { const organizations = await db.getOrganizationsByUserId(context.userId); const orgsWithProjectsPromises = organizations.map(async (org) => { const dbProjects = await db.getProjectsByOrganizationId(org.id); const projectsPromises = dbProjects.map(async (dbProject) => { const dbProjectMembers = await db.getProjectMembersByProjectId(dbProject.id); const dbEnvironmentVariables = await db.getEnvironmentVariablesByProjectId(dbProject.id); const projectMembers = dbProjectMembers.map(dbProjectMember => { return projectMemberToGqlType(dbProjectMember); }); const environmentVariables = dbEnvironmentVariables.map(dbEnvironmentVariable => { return environmentVariableToGqlType(dbEnvironmentVariable); }); return projectToGqlType(dbProject, projectMembers, environmentVariables); }); const projects = await Promise.all(projectsPromises); return { ...org, projects }; }); // TODO: Add organizationMembers field when / if required const orgsWithProjects = await Promise.all(orgsWithProjectsPromises); return orgsWithProjects; }, project: async (_: any, { projectId }: { projectId: string }) => { const dbProject = await db.getProjectByProjectId(projectId); return dbProject ? projectToGqlType(dbProject, [], []) : null; }, deployments: async (_: any, { projectId }: { projectId: string }) => { const dbDeployments = await db.getDeploymentsByProjectId(projectId); const deployments = dbDeployments.map(dbDeployment => { return deploymentToGqlType(dbDeployment); }); return deployments; }, environmentVariables: async (_: any, { projectId }: { projectId: string }) => { const dbEnvironmentVariables = await db.getEnvironmentVariablesByProjectId(projectId); const environmentVariables = dbEnvironmentVariables.map(dbEnvironmentVariable => { return environmentVariableToGqlType(dbEnvironmentVariable); }); return environmentVariables; }, projectMembers: async (_: any, { projectId }: { projectId: string }) => { const dbProjectMembers = await db.getProjectMembersByProjectId(projectId); const projectMembers = dbProjectMembers.map(dbProjectMember => { return projectMemberToGqlType(dbProjectMember); }); return projectMembers; }, searchProjects: async (_: any, { searchText }: { searchText: string }, context: any) => { const dbProjects = await db.getProjectsBySearchText(context.userId, searchText); const projects = dbProjects.map((project) => { return projectToGqlType(project, [], []); }); return projects; } }, Mutation: { removeMember: async (_: any, { memberId }: { memberId: string }, context: any) => { try { const member = await db.getProjectMemberByMemberId(memberId); if (member.member.id === context.userId) { throw new Error('Invalid operation: cannot remove self'); } const memberProject = member.project; assert(memberProject); if (isUserOwner(String(context.userId), String(memberProject.owner.id))) { return db.removeProjectMemberById(memberId); } else { throw new Error('Invalid operation: not authorized'); } } catch (err) { log(err); return false; } }, addEnvironmentVariables: async (_: any, { projectId, environmentVariables }: { projectId: string, environmentVariables: { environments: string[], key: string, value: string}[] }) => { try { return db.addEnvironmentVariablesByProjectId(projectId, environmentVariables); } catch (err) { log(err); return false; } }, updateDeploymentToProd: async (_: any, { deploymentId }: {deploymentId: string }) => { try { return db.updateDeploymentById(deploymentId, { environment: Environment.Production }); } catch (err) { log(err); return false; } }, updateProject: async (_: any, { projectId, updateProject }: { projectId: string, updateProject: { name: string, description: string } }) => { try { return db.updateProjectById(projectId, updateProject); } catch (err) { log(err); return false; } }, redeployToProd: async (_: any, { deploymentId }: {deploymentId: string }) => { try { return db.redeployToProdById(deploymentId); } catch (err) { log(err); return false; } } } }; };