@dawans/promptshield
Version:
Secure your LLM stack with enterprise-grade RulePacks for AI safety scanning
117 lines (116 loc) • 4.31 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.DefaultValidationEngine = void 0;
const Result_1 = require("../../../../shared/types/Result");
const path = __importStar(require("path"));
const fs = __importStar(require("fs"));
/**
* Default validation engine implementation
*/
class DefaultValidationEngine {
constructor() {
this.validators = new Map();
// Register default validators will be done by dependency injection
}
async validate(target, options) {
try {
// Determine validation type based on target
const validationType = this.determineValidationType(target);
// Get appropriate validator
const validator = this.validators.get(validationType);
if (!validator) {
return (0, Result_1.err)(new Error(`No validator found for type: ${validationType}`));
}
// Check if validator supports this target
if (!validator.supports(target, options)) {
return (0, Result_1.err)(new Error(`Validator does not support target: ${target}`));
}
// Perform validation
return await validator.validate(target, options);
}
catch (error) {
return (0, Result_1.err)(new Error(`Validation failed: ${error}`));
}
}
async validateBatch(targets, options) {
try {
const results = [];
for (const target of targets) {
const result = await this.validate(target, options);
if (result.isErr()) {
return (0, Result_1.err)(result.error);
}
results.push(result.value);
}
return (0, Result_1.ok)(results);
}
catch (error) {
return (0, Result_1.err)(new Error(`Batch validation failed: ${error}`));
}
}
registerValidator(type, validator) {
this.validators.set(type, validator);
}
getSupportedTypes() {
return Array.from(this.validators.keys());
}
determineValidationType(target) {
// Check if target exists
if (!fs.existsSync(target)) {
return 'unknown';
}
const ext = path.extname(target).toLowerCase();
const basename = path.basename(target).toLowerCase();
// RulePack files
if (ext === '.yaml' || ext === '.yml') {
return 'rulepack';
}
// Input files
if (ext === '.json' ||
ext === '.ndjson' ||
ext === '.jsonl' ||
ext === '.txt') {
return 'input-file';
}
// Config files
if (basename.includes('config') || basename.includes('settings')) {
return 'config';
}
// Default to input file
return 'input-file';
}
}
exports.DefaultValidationEngine = DefaultValidationEngine;