fase(5): findings module complete
Some checks failed
ABE Exploratory Testing / explore (push) Has been cancelled
Some checks failed
ABE Exploratory Testing / explore (push) Has been cancelled
This commit is contained in:
138
dist/modules/findings/infrastructure/repositories/KyselyFindingRepository.js
vendored
Normal file
138
dist/modules/findings/infrastructure/repositories/KyselyFindingRepository.js
vendored
Normal file
@@ -0,0 +1,138 @@
|
||||
"use strict";
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.KyselyFindingRepository = void 0;
|
||||
const Finding_1 = require("../../domain/entities/Finding");
|
||||
const UniqueId_1 = require("../../../../shared/domain/UniqueId");
|
||||
const Severity_1 = require("../../domain/value-objects/Severity");
|
||||
const FindingType_1 = require("../../domain/value-objects/FindingType");
|
||||
const FindingStatus_1 = require("../../domain/value-objects/FindingStatus");
|
||||
const Evidence_1 = require("../../domain/value-objects/Evidence");
|
||||
class KyselyFindingRepository {
|
||||
constructor(db) {
|
||||
this.db = db;
|
||||
}
|
||||
async save(finding) {
|
||||
const row = {
|
||||
id: finding.id.toString(),
|
||||
session_id: finding.sessionId,
|
||||
type: finding.type.value,
|
||||
severity: finding.severity.value,
|
||||
description: finding.description,
|
||||
status: finding.status.value,
|
||||
action_trace_json: JSON.stringify(finding.actionTrace),
|
||||
evidence_json: JSON.stringify(finding.evidence.toJSON()),
|
||||
screenshot_path: finding.evidence.screenshotPath ?? null,
|
||||
dom_snapshot_path: finding.evidence.domSnapshotPath ?? null,
|
||||
browser: finding.browser ?? null,
|
||||
browser_version: finding.browserVersion ?? null,
|
||||
ai_enrichment_json: finding.aiEnrichment ? JSON.stringify(finding.aiEnrichment) : null,
|
||||
created_at: finding.createdAt.getTime(),
|
||||
resolved_at: finding.resolvedAt ? finding.resolvedAt.getTime() : null,
|
||||
};
|
||||
await this.db.insertInto('findings').values(row).execute();
|
||||
}
|
||||
async findById(id) {
|
||||
const row = await this.db
|
||||
.selectFrom('findings')
|
||||
.selectAll()
|
||||
.where('id', '=', id)
|
||||
.executeTakeFirst();
|
||||
return row ? this.toDomain(row) : undefined;
|
||||
}
|
||||
async findAll(filters) {
|
||||
let query = this.db.selectFrom('findings').selectAll();
|
||||
if (filters?.sessionId) {
|
||||
query = query.where('session_id', '=', filters.sessionId);
|
||||
}
|
||||
if (filters?.severity) {
|
||||
query = query.where('severity', '=', filters.severity);
|
||||
}
|
||||
if (filters?.type) {
|
||||
query = query.where('type', '=', filters.type);
|
||||
}
|
||||
if (filters?.status) {
|
||||
query = query.where('status', '=', filters.status);
|
||||
}
|
||||
if (filters?.search) {
|
||||
query = query.where('description', 'like', `%${filters.search}%`);
|
||||
}
|
||||
const rows = await query.orderBy('created_at', 'desc').execute();
|
||||
return rows.map((row) => this.toDomain(row));
|
||||
}
|
||||
async update(finding) {
|
||||
await this.db
|
||||
.updateTable('findings')
|
||||
.set({
|
||||
status: finding.status.value,
|
||||
ai_enrichment_json: finding.aiEnrichment ? JSON.stringify(finding.aiEnrichment) : null,
|
||||
resolved_at: finding.resolvedAt ? finding.resolvedAt.getTime() : null,
|
||||
})
|
||||
.where('id', '=', finding.id.toString())
|
||||
.execute();
|
||||
}
|
||||
async count(filters) {
|
||||
let query = this.db.selectFrom('findings').select(eb => eb.fn.countAll().as('cnt'));
|
||||
if (filters?.sessionId) {
|
||||
query = query.where('session_id', '=', filters.sessionId);
|
||||
}
|
||||
if (filters?.severity) {
|
||||
query = query.where('severity', '=', filters.severity);
|
||||
}
|
||||
if (filters?.type) {
|
||||
query = query.where('type', '=', filters.type);
|
||||
}
|
||||
if (filters?.status) {
|
||||
query = query.where('status', '=', filters.status);
|
||||
}
|
||||
const result = await query.executeTakeFirst();
|
||||
return Number(result?.cnt ?? 0);
|
||||
}
|
||||
async countBySeverity() {
|
||||
const rows = await this.db
|
||||
.selectFrom('findings')
|
||||
.select(['severity', eb => eb.fn.countAll().as('cnt')])
|
||||
.groupBy('severity')
|
||||
.execute();
|
||||
const result = {};
|
||||
for (const row of rows) {
|
||||
result[row.severity] = Number(row.cnt);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
toDomain(row) {
|
||||
const actionTrace = this.parseJson(row.action_trace_json, []);
|
||||
const evidenceData = this.parseJson(row.evidence_json, {});
|
||||
const aiEnrichment = row.ai_enrichment_json
|
||||
? this.parseJson(row.ai_enrichment_json, undefined)
|
||||
: undefined;
|
||||
const props = {
|
||||
sessionId: row.session_id,
|
||||
severity: Severity_1.Severity.fromString(row.severity),
|
||||
type: FindingType_1.FindingType.fromString(row.type),
|
||||
description: row.description,
|
||||
evidence: Evidence_1.Evidence.create({
|
||||
screenshotPath: row.screenshot_path ?? undefined,
|
||||
domSnapshotPath: row.dom_snapshot_path ?? undefined,
|
||||
httpLog: evidenceData.httpLog ?? [],
|
||||
rawErrors: evidenceData.rawErrors ?? [],
|
||||
}),
|
||||
status: FindingStatus_1.FindingStatus.fromString(row.status),
|
||||
actionTrace,
|
||||
browser: row.browser ?? undefined,
|
||||
browserVersion: row.browser_version ?? undefined,
|
||||
aiEnrichment,
|
||||
createdAt: new Date(row.created_at),
|
||||
resolvedAt: row.resolved_at ? new Date(row.resolved_at) : undefined,
|
||||
};
|
||||
return Finding_1.Finding.reconstitute(props, UniqueId_1.UniqueId.from(row.id));
|
||||
}
|
||||
parseJson(json, fallback) {
|
||||
try {
|
||||
return JSON.parse(json);
|
||||
}
|
||||
catch {
|
||||
return fallback;
|
||||
}
|
||||
}
|
||||
}
|
||||
exports.KyselyFindingRepository = KyselyFindingRepository;
|
||||
Reference in New Issue
Block a user