import User from '../types/user'; import { AccessService } from './access-service'; import NameExistsError from '../error/name-exists-error'; import InvalidOperationError from '../error/invalid-operation-error'; import { nameType } from '../routes/util'; import { projectSchema } from './project-schema'; import NotFoundError from '../error/notfound-error'; import { PROJECT_CREATED, PROJECT_DELETED, PROJECT_UPDATED, } from '../types/events'; import { IUnleashStores } from '../types/stores'; import { IUnleashConfig } from '../types/option'; import { IProject, IProjectOverview, IProjectWithCount, IUserWithRole, FeatureToggle, RoleName, } from '../types/model'; import { IEnvironmentStore } from '../types/stores/environment-store'; import { IFeatureTypeStore } from '../types/stores/feature-type-store'; import { IFeatureToggleStore } from '../types/stores/feature-toggle-store'; import { IFeatureEnvironmentStore } from '../types/stores/feature-environment-store'; import { IProjectQuery, IProjectStore } from '../types/stores/project-store'; import { IRole } from '../types/stores/access-store'; import { IEventStore } from '../types/stores/event-store'; import FeatureToggleService from './feature-toggle-service'; import { CREATE_FEATURE, UPDATE_FEATURE } from '../types/permissions'; import NoAccessError from '../error/no-access-error'; import IncompatibleProjectError from '../error/incompatible-project-error'; const getCreatedBy = (user: User) => user.email || user.username; const DEFAULT_PROJECT = 'default'; export interface UsersWithRoles { users: IUserWithRole[]; roles: IRole[]; } export default class ProjectService { private store: IProjectStore; private accessService: AccessService; private eventStore: IEventStore; private featureToggleStore: IFeatureToggleStore; private featureTypeStore: IFeatureTypeStore; private featureEnvironmentStore: IFeatureEnvironmentStore; private environmentStore: IEnvironmentStore; private logger: any; private featureToggleService: FeatureToggleService; constructor( { projectStore, eventStore, featureToggleStore, featureTypeStore, environmentStore, featureEnvironmentStore, }: Pick< IUnleashStores, | 'projectStore' | 'eventStore' | 'featureToggleStore' | 'featureTypeStore' | 'environmentStore' | 'featureEnvironmentStore' >, config: IUnleashConfig, accessService: AccessService, featureToggleService: FeatureToggleService, ) { this.store = projectStore; this.environmentStore = environmentStore; this.featureEnvironmentStore = featureEnvironmentStore; this.accessService = accessService; this.eventStore = eventStore; this.featureToggleStore = featureToggleStore; this.featureTypeStore = featureTypeStore; this.featureToggleService = featureToggleService; this.logger = config.getLogger('services/project-service.js'); } async getProjects(query?: IProjectQuery): Promise { const projects = await this.store.getAll(query); const projectsWithCount = await Promise.all( projects.map(async (p) => { let featureCount = 0; let memberCount = 0; try { featureCount = await this.featureToggleService.getFeatureCountForProject( p.id, ); memberCount = await this.getMembers(p.id); } catch (e) { this.logger.warn('Error fetching project counts', e); } return { ...p, featureCount, memberCount }; }), ); return projectsWithCount; } async getProject(id: string): Promise { return this.store.get(id); } async createProject(newProject: IProject, user: User): Promise { const data = await projectSchema.validateAsync(newProject); await this.validateUniqueId(data.id); await this.store.create(data); const enabledEnvironments = await this.environmentStore.getAll({ enabled: true, }); // TODO: Only if enabled! await Promise.all( enabledEnvironments.map(async (e) => { await this.featureEnvironmentStore.connectProject( e.name, data.id, ); }), ); await this.accessService.createDefaultProjectRoles(user, data.id); await this.eventStore.store({ type: PROJECT_CREATED, createdBy: getCreatedBy(user), data, project: newProject.id, }); return data; } async updateProject(updatedProject: IProject, user: User): Promise { const preData = await this.store.get(updatedProject.id); const project = await projectSchema.validateAsync(updatedProject); await this.store.update(project); await this.eventStore.store({ type: PROJECT_UPDATED, project: project.id, createdBy: getCreatedBy(user), data: project, preData, }); } async checkProjectsCompatibility( feature: FeatureToggle, newProjectId: string, ): Promise { const featureEnvs = await this.featureEnvironmentStore.getAll({ feature_name: feature.name, }); const newEnvs = await this.store.getEnvironmentsForProject( newProjectId, ); return featureEnvs.every( (e) => !e.enabled || newEnvs.includes(e.environment), ); } async changeProject( newProjectId: string, featureName: string, user: User, currentProjectId: string, ): Promise { const feature = await this.featureToggleStore.get(featureName); if (feature.project !== currentProjectId) { throw new NoAccessError(UPDATE_FEATURE); } const project = await this.getProject(newProjectId); if (!project) { throw new NotFoundError(`Project ${newProjectId} not found`); } const authorized = await this.accessService.hasPermission( user, CREATE_FEATURE, newProjectId, ); if (!authorized) { throw new NoAccessError(CREATE_FEATURE); } const isCompatibleWithTargetProject = await this.checkProjectsCompatibility(feature, newProjectId); if (!isCompatibleWithTargetProject) { throw new IncompatibleProjectError(newProjectId); } const updatedFeature = await this.featureToggleService.changeProject( featureName, newProjectId, user.username, ); await this.featureToggleService.updateFeatureStrategyProject( featureName, newProjectId, ); return updatedFeature; } async deleteProject(id: string, user: User): Promise { if (id === DEFAULT_PROJECT) { throw new InvalidOperationError( 'You can not delete the default project!', ); } const toggles = await this.featureToggleStore.getAll({ project: id, archived: false, }); if (toggles.length > 0) { throw new InvalidOperationError( 'You can not delete a project with active feature toggles', ); } await this.store.delete(id); await this.eventStore.store({ type: PROJECT_DELETED, createdBy: getCreatedBy(user), project: id, }); await this.accessService.removeDefaultProjectRoles(user, id); } async validateId(id: string): Promise { await nameType.validateAsync(id); await this.validateUniqueId(id); return true; } async validateUniqueId(id: string): Promise { const exists = await this.store.hasProject(id); if (exists) { throw new NameExistsError('A project with this id already exists.'); } } // RBAC methods async getUsersWithAccess(projectId: string): Promise { const [roles, users] = await this.accessService.getProjectRoleUsers( projectId, ); return { roles, users, }; } // TODO: should be an event too async addUser( projectId: string, roleId: number, userId: number, ): Promise { const [roles, users] = await this.accessService.getProjectRoleUsers( projectId, ); const role = roles.find((r) => r.id === roleId); if (!role) { throw new NotFoundError( `Could not find roleId=${roleId} on project=${projectId}`, ); } const alreadyHasAccess = users.some((u) => u.id === userId); if (alreadyHasAccess) { throw new Error(`User already have access to project=${projectId}`); } await this.accessService.addUserToRole(userId, role.id); } // TODO: should be an event too async removeUser( projectId: string, roleId: number, userId: number, ): Promise { const roles = await this.accessService.getRolesForProject(projectId); const role = roles.find((r) => r.id === roleId); if (!role) { throw new NotFoundError( `Couldn't find roleId=${roleId} on project=${projectId}`, ); } if (role.name === RoleName.OWNER) { const users = await this.accessService.getUsersForRole(role.id); if (users.length < 2) { throw new Error('A project must have at least one owner'); } } await this.accessService.removeUserFromRole(userId, role.id); } async getMembers(projectId: string): Promise { return this.store.getMembers(projectId); } async getProjectOverview( projectId: string, archived: boolean = false, ): Promise { const project = await this.store.get(projectId); const environments = await this.store.getEnvironmentsForProject( projectId, ); const features = await this.featureToggleService.getFeatureOverview( projectId, archived, ); const members = await this.store.getMembers(projectId); return { name: project.name, environments, description: project.description, health: project.health, features, members, version: 1, }; } } module.exports = ProjectService;