2021-04-20 12:32:02 +02:00
|
|
|
import { Knex } from 'knex';
|
|
|
|
import { Logger, LogProvider } from '../logger';
|
|
|
|
|
2021-07-07 10:46:50 +02:00
|
|
|
import NotFoundError from '../error/notfound-error';
|
2022-02-21 12:46:28 +01:00
|
|
|
import { IProject, IProjectWithCount } from '../types/model';
|
2021-08-19 13:25:36 +02:00
|
|
|
import {
|
2021-08-12 15:04:37 +02:00
|
|
|
IProjectHealthUpdate,
|
|
|
|
IProjectInsert,
|
2021-10-01 10:59:43 +02:00
|
|
|
IProjectQuery,
|
2021-08-12 15:04:37 +02:00
|
|
|
IProjectStore,
|
|
|
|
} from '../types/stores/project-store';
|
2021-09-24 13:55:00 +02:00
|
|
|
import { DEFAULT_ENV } from '../util/constants';
|
2022-02-21 12:46:28 +01:00
|
|
|
import metricsHelper from '../util/metrics-helper';
|
|
|
|
import { DB_TIME } from '../metric-events';
|
|
|
|
import EventEmitter from 'events';
|
2020-09-28 21:54:44 +02:00
|
|
|
|
2021-11-30 14:05:44 +01:00
|
|
|
const COLUMNS = [
|
|
|
|
'id',
|
|
|
|
'name',
|
|
|
|
'description',
|
|
|
|
'created_at',
|
|
|
|
'health',
|
2021-11-30 15:25:52 +01:00
|
|
|
'updated_at',
|
2021-11-30 14:05:44 +01:00
|
|
|
];
|
2020-09-28 21:54:44 +02:00
|
|
|
const TABLE = 'projects';
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
class ProjectStore implements IProjectStore {
|
2021-04-20 12:32:02 +02:00
|
|
|
private db: Knex;
|
|
|
|
|
|
|
|
private logger: Logger;
|
|
|
|
|
2022-02-21 12:46:28 +01:00
|
|
|
private timer: Function;
|
|
|
|
|
|
|
|
constructor(db: Knex, eventBus: EventEmitter, getLogger: LogProvider) {
|
2020-09-28 21:54:44 +02:00
|
|
|
this.db = db;
|
2021-08-12 15:04:37 +02:00
|
|
|
this.logger = getLogger('project-store.ts');
|
2022-02-21 12:46:28 +01:00
|
|
|
this.timer = (action) =>
|
|
|
|
metricsHelper.wrapTimer(eventBus, DB_TIME, {
|
|
|
|
store: 'project',
|
|
|
|
action,
|
|
|
|
});
|
2020-09-28 21:54:44 +02:00
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
2021-04-20 12:32:02 +02:00
|
|
|
fieldToRow(data): IProjectInsert {
|
2020-09-28 21:54:44 +02:00
|
|
|
return {
|
|
|
|
id: data.id,
|
|
|
|
name: data.name,
|
|
|
|
description: data.description,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
destroy(): void {}
|
|
|
|
|
|
|
|
async exists(id: string): Promise<boolean> {
|
|
|
|
const result = await this.db.raw(
|
2022-02-21 12:46:28 +01:00
|
|
|
`SELECT EXISTS(SELECT 1 FROM ${TABLE} WHERE id = ?) AS present`,
|
2021-08-12 15:04:37 +02:00
|
|
|
[id],
|
|
|
|
);
|
|
|
|
const { present } = result.rows[0];
|
|
|
|
return present;
|
|
|
|
}
|
|
|
|
|
2022-02-21 12:46:28 +01:00
|
|
|
async getProjectsWithCounts(
|
|
|
|
query?: IProjectQuery,
|
|
|
|
): Promise<IProjectWithCount[]> {
|
|
|
|
const projectTimer = this.timer('getProjectsWithCount');
|
|
|
|
let projects = this.db(TABLE)
|
|
|
|
.select(
|
|
|
|
this.db.raw(
|
|
|
|
'projects.id, projects.name, projects.description, projects.health, projects.updated_at, count(features.name) AS number_of_features',
|
|
|
|
),
|
|
|
|
)
|
|
|
|
.leftJoin('features', 'features.project', 'projects.id')
|
|
|
|
.groupBy('projects.id');
|
|
|
|
if (query) {
|
|
|
|
projects = projects.where(query);
|
|
|
|
}
|
|
|
|
const projectAndFeatureCount = await projects;
|
|
|
|
|
|
|
|
// @ts-ignore
|
|
|
|
const projectsWithFeatureCount = projectAndFeatureCount.map(
|
|
|
|
this.mapProjectWithCountRow,
|
|
|
|
);
|
|
|
|
projectTimer();
|
|
|
|
const memberTimer = this.timer('getMemberCount');
|
|
|
|
const memberCount = await this.db.raw(
|
|
|
|
`SELECT count(role_id) as member_count, project FROM role_user GROUP BY project`,
|
|
|
|
);
|
|
|
|
memberTimer();
|
|
|
|
const memberMap = new Map<string, number>(
|
|
|
|
memberCount.rows.map((c) => [c.project, Number(c.member_count)]),
|
|
|
|
);
|
|
|
|
return projectsWithFeatureCount.map((r) => {
|
|
|
|
return { ...r, memberCount: memberMap.get(r.id) };
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
|
|
|
mapProjectWithCountRow(row): IProjectWithCount {
|
|
|
|
return {
|
|
|
|
name: row.name,
|
|
|
|
id: row.id,
|
|
|
|
description: row.description,
|
|
|
|
health: row.health,
|
|
|
|
featureCount: row.number_of_features,
|
|
|
|
memberCount: row.number_of_users || 0,
|
|
|
|
updatedAt: row.updated_at,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2021-10-01 10:59:43 +02:00
|
|
|
async getAll(query: IProjectQuery = {}): Promise<IProject[]> {
|
2020-09-28 21:54:44 +02:00
|
|
|
const rows = await this.db
|
|
|
|
.select(COLUMNS)
|
|
|
|
.from(TABLE)
|
2021-10-01 10:59:43 +02:00
|
|
|
.where(query)
|
2020-09-28 21:54:44 +02:00
|
|
|
.orderBy('name', 'asc');
|
|
|
|
|
|
|
|
return rows.map(this.mapRow);
|
|
|
|
}
|
|
|
|
|
2021-04-30 12:51:46 +02:00
|
|
|
async get(id: string): Promise<IProject> {
|
2020-09-28 21:54:44 +02:00
|
|
|
return this.db
|
|
|
|
.first(COLUMNS)
|
|
|
|
.from(TABLE)
|
|
|
|
.where({ id })
|
|
|
|
.then(this.mapRow);
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
async hasProject(id: string): Promise<boolean> {
|
|
|
|
const result = await this.db.raw(
|
2022-02-21 12:46:28 +01:00
|
|
|
`SELECT EXISTS(SELECT 1 FROM ${TABLE} WHERE id = ?) AS present`,
|
2021-08-12 15:04:37 +02:00
|
|
|
[id],
|
|
|
|
);
|
|
|
|
const { present } = result.rows[0];
|
|
|
|
return present;
|
2020-09-28 21:54:44 +02:00
|
|
|
}
|
|
|
|
|
2021-07-07 10:46:50 +02:00
|
|
|
async updateHealth(healthUpdate: IProjectHealthUpdate): Promise<void> {
|
|
|
|
await this.db(TABLE)
|
|
|
|
.where({ id: healthUpdate.id })
|
2021-11-30 15:25:52 +01:00
|
|
|
.update({ health: healthUpdate.health, updated_at: new Date() });
|
2021-07-07 10:46:50 +02:00
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
async create(project: IProjectInsert): Promise<IProject> {
|
|
|
|
const row = await this.db(TABLE)
|
2020-09-28 21:54:44 +02:00
|
|
|
.insert(this.fieldToRow(project))
|
2021-08-12 15:04:37 +02:00
|
|
|
.returning('*');
|
2022-01-13 11:14:17 +01:00
|
|
|
return this.mapRow(row[0]);
|
2020-09-28 21:54:44 +02:00
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
2021-04-20 12:32:02 +02:00
|
|
|
async update(data): Promise<void> {
|
2020-09-28 21:54:44 +02:00
|
|
|
try {
|
|
|
|
await this.db(TABLE)
|
|
|
|
.where({ id: data.id })
|
|
|
|
.update(this.fieldToRow(data));
|
|
|
|
} catch (err) {
|
|
|
|
this.logger.error('Could not update project, error: ', err);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
async importProjects(projects: IProjectInsert[]): Promise<IProject[]> {
|
2021-03-12 11:08:10 +01:00
|
|
|
const rows = await this.db(TABLE)
|
|
|
|
.insert(projects.map(this.fieldToRow))
|
|
|
|
.returning(COLUMNS)
|
|
|
|
.onConflict('id')
|
|
|
|
.ignore();
|
|
|
|
if (rows.length > 0) {
|
2021-09-24 13:55:00 +02:00
|
|
|
await this.addDefaultEnvironment(rows);
|
2021-03-12 11:08:10 +01:00
|
|
|
return rows.map(this.mapRow);
|
|
|
|
}
|
|
|
|
return [];
|
|
|
|
}
|
|
|
|
|
2021-09-24 13:55:00 +02:00
|
|
|
async addDefaultEnvironment(projects: any[]): Promise<void> {
|
2021-08-12 15:04:37 +02:00
|
|
|
const environments = projects.map((p) => ({
|
2021-07-30 13:38:28 +02:00
|
|
|
project_id: p.id,
|
2021-09-24 13:55:00 +02:00
|
|
|
environment_name: DEFAULT_ENV,
|
2021-07-30 13:38:28 +02:00
|
|
|
}));
|
|
|
|
await this.db('project_environments')
|
|
|
|
.insert(environments)
|
|
|
|
.onConflict(['project_id', 'environment_name'])
|
|
|
|
.ignore();
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
async deleteAll(): Promise<void> {
|
2021-03-12 11:08:10 +01:00
|
|
|
await this.db(TABLE).del();
|
|
|
|
}
|
|
|
|
|
2021-07-07 10:46:50 +02:00
|
|
|
async delete(id: string): Promise<void> {
|
2020-09-28 21:54:44 +02:00
|
|
|
try {
|
2021-08-12 15:04:37 +02:00
|
|
|
await this.db(TABLE).where({ id }).del();
|
2020-09-28 21:54:44 +02:00
|
|
|
} catch (err) {
|
|
|
|
this.logger.error('Could not delete project, error: ', err);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
async deleteEnvironmentForProject(
|
|
|
|
id: string,
|
|
|
|
environment: string,
|
|
|
|
): Promise<void> {
|
2021-07-07 10:46:50 +02:00
|
|
|
await this.db('project_environments')
|
|
|
|
.where({
|
|
|
|
project_id: id,
|
|
|
|
environment_name: environment,
|
|
|
|
})
|
|
|
|
.del();
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
async addEnvironmentToProject(
|
|
|
|
id: string,
|
|
|
|
environment: string,
|
|
|
|
): Promise<void> {
|
|
|
|
await this.db('project_environments')
|
|
|
|
.insert({ project_id: id, environment_name: environment })
|
|
|
|
.onConflict(['project_id', 'environment_name'])
|
|
|
|
.ignore();
|
|
|
|
}
|
|
|
|
|
|
|
|
async getEnvironmentsForProject(id: string): Promise<string[]> {
|
|
|
|
return this.db('project_environments')
|
|
|
|
.where({
|
|
|
|
project_id: id,
|
|
|
|
})
|
2021-09-13 10:23:57 +02:00
|
|
|
.pluck('environment_name');
|
2021-08-12 15:04:37 +02:00
|
|
|
}
|
|
|
|
|
2021-07-07 10:46:50 +02:00
|
|
|
async getMembers(projectId: string): Promise<number> {
|
|
|
|
const rolesFromProject = this.db('role_permission')
|
|
|
|
.select('role_id')
|
|
|
|
.distinct()
|
|
|
|
.where({ project: projectId });
|
|
|
|
|
|
|
|
const numbers = await this.db('role_user')
|
|
|
|
.countDistinct('user_id as members')
|
|
|
|
.whereIn('role_id', rolesFromProject)
|
|
|
|
.first();
|
|
|
|
const { members } = numbers;
|
|
|
|
if (typeof members === 'string') {
|
|
|
|
return parseInt(members, 10);
|
|
|
|
}
|
|
|
|
return members;
|
|
|
|
}
|
|
|
|
|
2021-08-27 10:10:14 +02:00
|
|
|
async count(): Promise<number> {
|
|
|
|
return this.db
|
|
|
|
.count('*')
|
|
|
|
.from(TABLE)
|
|
|
|
.then((res) => Number(res[0].count));
|
|
|
|
}
|
|
|
|
|
2021-08-12 15:04:37 +02:00
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
2021-04-20 12:32:02 +02:00
|
|
|
mapRow(row): IProject {
|
2020-09-28 21:54:44 +02:00
|
|
|
if (!row) {
|
|
|
|
throw new NotFoundError('No project found');
|
|
|
|
}
|
|
|
|
|
|
|
|
return {
|
|
|
|
id: row.id,
|
|
|
|
name: row.name,
|
|
|
|
description: row.description,
|
|
|
|
createdAt: row.created_at,
|
2021-07-07 10:46:50 +02:00
|
|
|
health: row.health || 100,
|
2021-11-30 15:25:52 +01:00
|
|
|
updatedAt: row.updated_at || new Date(),
|
2020-09-28 21:54:44 +02:00
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
2022-02-21 12:46:28 +01:00
|
|
|
|
2021-04-20 12:32:02 +02:00
|
|
|
export default ProjectStore;
|