/**
* @fileoverview Data import/export operation models.
*/
import { z } from 'zod';
/**
* Import operation input schema with validation.
*/
export const ImportDataInputSchema = z
.object({
format: z.enum(['graphson', 'csv'], {
errorMap: () => ({ message: 'Format must be either "graphson" or "csv"' }),
}),
data: z
.string()
.min(1, 'Data cannot be empty')
.max(50 * 1024 * 1024, 'Data size cannot exceed 50MB'), // 50MB limit
options: z
.object({
clear_graph: z.boolean().optional(),
batch_size: z
.number()
.positive('Batch size must be positive')
.max(10000, 'Batch size cannot exceed 10,000')
.optional(),
validate_schema: z.boolean().optional(),
})
.optional(),
})
.refine(
data => {
// Additional validation for GraphSON format
if (data.format === 'graphson') {
try {
JSON.parse(data.data);
return true;
} catch {
return false;
}
}
return true;
},
{
message: 'GraphSON data must be valid JSON',
path: ['data'],
}
);
export type ImportDataInput = z.infer<typeof ImportDataInputSchema>;
/**
* Export operation input schema with validation.
*/
export const ExportSubgraphInputSchema = z
.object({
traversal_query: z
.string()
.min(1, 'Traversal query cannot be empty')
.max(10000, 'Traversal query cannot exceed 10,000 characters')
.refine(
query => {
// Basic Gremlin syntax validation
const invalidPatterns = [';', '--', '/*', '*/', 'DROP', 'DELETE'];
return !invalidPatterns.some(pattern =>
query.toUpperCase().includes(pattern.toUpperCase())
);
},
{
message: 'Query contains potentially unsafe operations',
}
),
format: z.enum(['graphson', 'json', 'csv'], {
errorMap: () => ({ message: 'Format must be "graphson", "json", or "csv"' }),
}),
include_properties: z
.array(z.string().min(1, 'Property name cannot be empty'))
.max(100, 'Cannot include more than 100 properties')
.optional(),
exclude_properties: z
.array(z.string().min(1, 'Property name cannot be empty'))
.max(100, 'Cannot exclude more than 100 properties')
.optional(),
max_depth: z
.number()
.positive('Max depth must be positive')
.max(10, 'Max depth cannot exceed 10 levels')
.optional(),
})
.refine(
data => {
// Cannot have both include and exclude properties
return !(data.include_properties && data.exclude_properties);
},
{
message: 'Cannot specify both include_properties and exclude_properties',
path: ['include_properties'],
}
);
export type ExportSubgraphInput = z.infer<typeof ExportSubgraphInputSchema>;
