a392163630
- Add braceDepth tracking to Map/Set type parsing for inline records - Fix multiline state block joining to preserve type expressions - Fix multiline object literal parsing in assignments - Add smart set-difference for Maps vs Sets - Fix bare notIn for Sets - Add non-null assertion (!) to Map.get() calls - Add implicit event interface generation from emit statements - Protect object literals from this. prefix injection
982 lines
No EOL
38 KiB
JavaScript
982 lines
No EOL
38 KiB
JavaScript
import fs from 'fs';
|
|
/**
|
|
* Parse a .speck file and return the AST
|
|
*/
|
|
export function parseSpeckFile(filePath) {
|
|
const content = fs.readFileSync(filePath, 'utf-8');
|
|
return parseSpeckContent(content, filePath);
|
|
}
|
|
/**
|
|
* Parse SpeckDL content and return the AST
|
|
*/
|
|
export function parseSpeckContent(content, filePath) {
|
|
const specks = [];
|
|
const lines = content.split('\n');
|
|
let i = 0;
|
|
while (i < lines.length) {
|
|
const line = lines[i].trim();
|
|
// Skip empty lines and comments
|
|
if (!line || line.startsWith('//') || line.startsWith('/*')) {
|
|
i++;
|
|
continue;
|
|
}
|
|
// Handle import statements
|
|
if (line.startsWith('import ')) {
|
|
const importNode = parseImportStatement(line);
|
|
// Import is a special case - it's not part of a speck
|
|
// We'll handle it separately in a future update
|
|
i++;
|
|
continue;
|
|
}
|
|
// Handle speck definitions
|
|
if (line.startsWith('speck ')) {
|
|
const speck = parseSpeck(lines, i);
|
|
if (speck) {
|
|
specks.push(speck);
|
|
// Move index past the speck we just parsed
|
|
i++; // Increment by 1, the loop will continue from next line
|
|
}
|
|
else {
|
|
i++;
|
|
}
|
|
continue;
|
|
}
|
|
i++;
|
|
}
|
|
return { specks };
|
|
}
|
|
function parseSpeck(lines, startIndex) {
|
|
const headerMatch = lines[startIndex].match(/^speck\s+([\w-]+)\s*{/);
|
|
if (!headerMatch)
|
|
return null;
|
|
const name = headerMatch[1];
|
|
const members = [];
|
|
let i = startIndex + 1;
|
|
let braceCount = 1;
|
|
while (i < lines.length && braceCount > 0) {
|
|
const rawLine = lines[i];
|
|
const line = rawLine.trim();
|
|
if (!line || line.startsWith('//') || line.startsWith('/*')) {
|
|
i++;
|
|
continue;
|
|
}
|
|
// Count braces on this line first
|
|
const openCount = (rawLine.match(/{/g) || []).length;
|
|
const closeCount = (rawLine.match(/}/g) || []).length;
|
|
// Check for block-opening members (those that open braces on the same line)
|
|
const blockStarters = ['state:', 'init:', 'action ', 'event ', 'provenance ', 'bom ', 'interface ',
|
|
'state {', 'init {', 'verify ', 'constraint '];
|
|
const isBlockStarter = blockStarters.some(s => line.startsWith(s));
|
|
if (isBlockStarter && openCount > 0) {
|
|
// Multi-line block: parse it and skip to its end
|
|
const member = parseMemberBlock(lines, i);
|
|
if (member) {
|
|
members.push(member);
|
|
}
|
|
// Skip past this block if it spans multiple lines
|
|
if (closeCount < openCount) {
|
|
const endIdx = findBlockEnd(lines, i + 1, braceCount + openCount);
|
|
i = endIdx + 1;
|
|
}
|
|
else {
|
|
i++;
|
|
}
|
|
continue;
|
|
}
|
|
// Handle constraint/verify blocks that may span multiple lines without braces
|
|
// (e.g. constraint Name: forall r in [0..4]:)
|
|
if (line.startsWith('constraint ') || line.startsWith('verify ')) {
|
|
const member = parseMultiLineConstraintVerify(lines, i);
|
|
if (member) {
|
|
members.push(member);
|
|
// Skip past the consumed continuation lines
|
|
while (i + 1 < lines.length) {
|
|
const nextLine = lines[i + 1].trim();
|
|
if (!nextLine || nextLine.startsWith('//') || nextLine.startsWith('/*') ||
|
|
nextLine.startsWith('constraint ') || nextLine.startsWith('verify ') ||
|
|
nextLine.startsWith('constraint:') || nextLine.startsWith('verify:') ||
|
|
nextLine.startsWith('state') || nextLine.startsWith('init') ||
|
|
nextLine.startsWith('action ') || nextLine.startsWith('event ') ||
|
|
nextLine.startsWith('interface ') || nextLine.startsWith('provenance ') ||
|
|
nextLine.startsWith('bom ') || nextLine.startsWith('}') ||
|
|
nextLine.startsWith('next:') || nextLine.startsWith('invariant ')) {
|
|
break;
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
// Single-line: update brace count, then parse if still inside
|
|
braceCount += openCount - closeCount;
|
|
if (braceCount <= 0) {
|
|
i++;
|
|
break;
|
|
}
|
|
const member = parseMember(line);
|
|
if (member) {
|
|
members.push(member);
|
|
}
|
|
i++;
|
|
}
|
|
return { type: 'speck', name, members };
|
|
}
|
|
function findBlockEnd(lines, startIndex, startBraceCount) {
|
|
let braceCount = startBraceCount;
|
|
let i = startIndex;
|
|
while (i < lines.length) {
|
|
const line = lines[i];
|
|
for (const ch of line) {
|
|
if (ch === '{')
|
|
braceCount++;
|
|
if (ch === '}') {
|
|
braceCount--;
|
|
if (braceCount < startBraceCount) {
|
|
return i;
|
|
}
|
|
}
|
|
}
|
|
i++;
|
|
}
|
|
return i;
|
|
}
|
|
// Check if an expression has balanced braces, parens, and brackets
|
|
function isExpressionComplete(expr) {
|
|
let braces = 0, parens = 0, brackets = 0;
|
|
for (const char of expr) {
|
|
if (char === '{')
|
|
braces++;
|
|
else if (char === '}')
|
|
braces--;
|
|
else if (char === '(')
|
|
parens++;
|
|
else if (char === ')')
|
|
parens--;
|
|
else if (char === '[')
|
|
brackets++;
|
|
else if (char === ']')
|
|
brackets--;
|
|
}
|
|
return braces === 0 && parens === 0 && brackets === 0;
|
|
}
|
|
function parseMemberBlock(lines, startIndex) {
|
|
const firstLine = lines[startIndex].trim();
|
|
const startBraceCount = (firstLine.match(/{/g) || []).length;
|
|
if (firstLine.startsWith('state:')) {
|
|
return parseStateBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('state {') || firstLine === 'state') {
|
|
return parseStateBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('init:')) {
|
|
return parseInitBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('init {') || firstLine === 'init') {
|
|
return parseInitBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('action ')) {
|
|
return parseActionBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('event ')) {
|
|
return parseEventBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('interface ')) {
|
|
return parseInterfaceBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('provenance ')) {
|
|
return { type: 'provenance', clauses: [] };
|
|
}
|
|
if (firstLine.startsWith('bom ')) {
|
|
return { type: 'bom' };
|
|
}
|
|
if (firstLine.startsWith('verify ')) {
|
|
return parseVerifyBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
if (firstLine.startsWith('constraint ')) {
|
|
return parseConstraintBlockMultiline(lines, startIndex, startBraceCount);
|
|
}
|
|
return null;
|
|
}
|
|
function parseMember(line) {
|
|
// input:
|
|
if (line.startsWith('input:')) {
|
|
const typeExpr = parseTypeExpr(line.substring(6).trim());
|
|
return { type: 'input', typeExpr };
|
|
}
|
|
// output:
|
|
if (line.startsWith('output:')) {
|
|
const typeExpr = parseTypeExpr(line.substring(7).trim());
|
|
return { type: 'output', typeExpr };
|
|
}
|
|
// constraint Name: expr (single-line named constraint)
|
|
if (line.startsWith('constraint ')) {
|
|
const match = line.match(/^constraint\s+(\w+):\s*(.+)$/);
|
|
if (match) {
|
|
return { type: 'constraint', name: match[1], expr: match[2].trim() };
|
|
}
|
|
return null; // multi-line constraint block (handled by parseMemberBlock)
|
|
}
|
|
// constraint: expr (unnamed, single-line)
|
|
if (line.startsWith('constraint:')) {
|
|
return { type: 'constraint', expr: line.substring(11).trim() };
|
|
}
|
|
// verify Always(Name) { depth N } or verify Always(Name) (single-line forms)
|
|
if (line.startsWith('verify ')) {
|
|
return parseVerifySingleLine(line);
|
|
}
|
|
// verify: expr (old unnamed single-line)
|
|
if (line.startsWith('verify:')) {
|
|
return { type: 'verify', name: 'unnamed', temporalExpr: line.substring(7).trim() };
|
|
}
|
|
// interface:
|
|
if (line.startsWith('interface ')) {
|
|
return parseInterface(line);
|
|
}
|
|
// event:
|
|
if (line.startsWith('event ')) {
|
|
return parseEvent(line);
|
|
}
|
|
// provenance {
|
|
if (line.startsWith('provenance ')) {
|
|
return { type: 'provenance', clauses: [] };
|
|
}
|
|
// review:
|
|
if (line.startsWith('review:')) {
|
|
const kind = line.substring(7).trim();
|
|
return { type: 'review', kind };
|
|
}
|
|
// derives from:
|
|
if (line.startsWith('derives from ')) {
|
|
return parseDerives(line);
|
|
}
|
|
// satisfies:
|
|
if (line.startsWith('satisfies ')) {
|
|
return parseSatisfies(line);
|
|
}
|
|
// author:
|
|
if (line.startsWith('author:')) {
|
|
return parseAuthor(line);
|
|
}
|
|
// source:
|
|
if (line.startsWith('source:')) {
|
|
return parseSource(line);
|
|
}
|
|
// state: or state {
|
|
if (line.startsWith('state:')) {
|
|
const varsStr = line.substring(6).trim();
|
|
// state: { ... } - single-line form
|
|
if (varsStr.startsWith('{') && varsStr.endsWith('}')) {
|
|
const inner = varsStr.slice(1, -1).trim();
|
|
return parseStateBlock(inner);
|
|
}
|
|
// state: { - multi-line opening (handled by parseStateMultiline)
|
|
return { type: 'state', variables: [] };
|
|
}
|
|
if (line.startsWith('state {') || line === 'state') {
|
|
// state { ... } inline or state { on its own line (handled by parseMemberBlock)
|
|
if (line.startsWith('state {')) {
|
|
const inner = line.substring(6).trim();
|
|
const content = inner.startsWith('{') ? inner.slice(1).trim() : inner;
|
|
if (content.endsWith('}')) {
|
|
return parseStateBlock(content.slice(0, -1).trim());
|
|
}
|
|
}
|
|
return { type: 'state', variables: [] };
|
|
}
|
|
// init: or init {
|
|
if (line.startsWith('init:')) {
|
|
const initStr = line.substring(5).trim();
|
|
if (initStr.startsWith('{') && initStr.endsWith('}')) {
|
|
const inner = initStr.slice(1, -1).trim();
|
|
return parseInitBlock(inner);
|
|
}
|
|
return { type: 'init', assignments: [] };
|
|
}
|
|
if (line.startsWith('init {') || line === 'init') {
|
|
// init { ... } inline
|
|
if (line.startsWith('init {')) {
|
|
const inner = line.substring(5).trim();
|
|
const content = inner.startsWith('{') ? inner.slice(1).trim() : inner;
|
|
if (content.endsWith('}')) {
|
|
return parseInitBlock(content.slice(0, -1).trim());
|
|
}
|
|
}
|
|
return { type: 'init', assignments: [] };
|
|
}
|
|
// action keyword
|
|
if (line.startsWith('action ')) {
|
|
return parseActionHeader(line);
|
|
}
|
|
// bom {
|
|
if (line.startsWith('bom ')) {
|
|
// Simplified - full BOM block parsing would need multi-line context
|
|
return {
|
|
type: 'bom',
|
|
};
|
|
}
|
|
return null;
|
|
}
|
|
function parseStateBlock(inner) {
|
|
const variables = [];
|
|
if (!inner.trim())
|
|
return { type: 'state', variables };
|
|
const varStrs = inner.split(/;|\n/).map(s => s.trim()).filter(Boolean);
|
|
for (const vStr of varStrs) {
|
|
const withDefault = vStr.match(/^(\w+)\s*:\s*(.+?)\s*:=\s*(.+)$/);
|
|
if (withDefault) {
|
|
// Strip trailing comments from type expression
|
|
let typeStr = withDefault[2].trim();
|
|
const commentIdx = typeStr.indexOf('//');
|
|
if (commentIdx >= 0)
|
|
typeStr = typeStr.substring(0, commentIdx).trim();
|
|
variables.push({
|
|
name: withDefault[1],
|
|
typeExpr: parseTypeExpr(typeStr),
|
|
defaultInit: withDefault[3].trim(),
|
|
});
|
|
continue;
|
|
}
|
|
const withoutDefault = vStr.match(/^(\w+)\s*:\s*(.+)$/);
|
|
if (withoutDefault) {
|
|
// Strip trailing comments from type expression
|
|
let typeStr = withoutDefault[2].trim();
|
|
const commentIdx = typeStr.indexOf('//');
|
|
if (commentIdx >= 0)
|
|
typeStr = typeStr.substring(0, commentIdx).trim();
|
|
variables.push({
|
|
name: withoutDefault[1],
|
|
typeExpr: parseTypeExpr(typeStr),
|
|
});
|
|
}
|
|
}
|
|
return { type: 'state', variables };
|
|
}
|
|
function parseInitBlock(inner) {
|
|
const assignments = [];
|
|
if (!inner.trim())
|
|
return { type: 'init', assignments };
|
|
const assignStrs = inner.split(/;|\n/).map(s => s.trim()).filter(Boolean);
|
|
for (const aStr of assignStrs) {
|
|
const match = aStr.match(/^(\w+)\s*:=\s*(.+)$/);
|
|
if (match) {
|
|
assignments.push({ name: match[1], expr: match[2].trim() });
|
|
}
|
|
}
|
|
return { type: 'init', assignments };
|
|
}
|
|
function parseActionHeader(line) {
|
|
const match = line.match(/^action\s+(\w+)\s*(?:\(([^)]*)\))?\s*\{/);
|
|
if (!match)
|
|
return null;
|
|
const name = match[1];
|
|
const paramsStr = match[2] || '';
|
|
const params = [];
|
|
if (paramsStr.trim()) {
|
|
const paramPairs = paramsStr.split(',').map(s => s.trim()).filter(Boolean);
|
|
for (const p of paramPairs) {
|
|
const pMatch = p.match(/^(\w+)\s*:\s*(.+)$/);
|
|
if (pMatch) {
|
|
params.push({ name: pMatch[1], type: parseTypeExpr(pMatch[2].trim()) });
|
|
}
|
|
}
|
|
}
|
|
return { type: 'action', name, params, statements: [] };
|
|
}
|
|
function parseImportStatement(line) {
|
|
const match = line.match(/^import\s+(['"])(.+?)\1(?:\s+as\s+(\w+))?(?:\s+version\s+(['"])(.+?)\4)?(?:\s+hash\s+(['"])(.+?)\6)?/);
|
|
if (!match) {
|
|
throw new Error(`Failed to parse import statement: ${line}`);
|
|
}
|
|
const [, , path, , alias, , version, , hash] = match;
|
|
return {
|
|
type: 'import',
|
|
path,
|
|
alias,
|
|
version: version || undefined,
|
|
hash: hash || undefined,
|
|
};
|
|
}
|
|
function parseTypeExpr(expr) {
|
|
expr = expr.trim();
|
|
// Primitive types
|
|
const primitives = ['Nat', 'Int', 'Real', 'Bool', 'String', 'Bytes'];
|
|
if (primitives.includes(expr)) {
|
|
return { type: 'primitive', name: expr };
|
|
}
|
|
// List<T>
|
|
if (expr.startsWith('List<') && expr.endsWith('>')) {
|
|
const inner = expr.substring(5, expr.length - 1).trim();
|
|
return { type: 'list', elementType: parseTypeExpr(inner) };
|
|
}
|
|
// Set<T>
|
|
if (expr.startsWith('Set<') && expr.endsWith('>')) {
|
|
const inner = expr.substring(4, expr.length - 1).trim();
|
|
return { type: 'set', elementType: parseTypeExpr(inner) };
|
|
}
|
|
// Map<K, V>
|
|
if (expr.startsWith('Map<') && expr.endsWith('>')) {
|
|
const inner = expr.substring(4, expr.length - 1);
|
|
// Find the first-level comma (not inside nested generics or record literals)
|
|
let depth = 0;
|
|
let braceDepth = 0;
|
|
let commaIdx = -1;
|
|
for (let i = 0; i < inner.length; i++) {
|
|
if (inner[i] === '<' || inner[i] === '(')
|
|
depth++;
|
|
else if (inner[i] === '>' || inner[i] === ')')
|
|
depth--;
|
|
else if (inner[i] === '{')
|
|
braceDepth++;
|
|
else if (inner[i] === '}')
|
|
braceDepth--;
|
|
else if (inner[i] === ',' && depth === 0 && braceDepth === 0) {
|
|
commaIdx = i;
|
|
break;
|
|
}
|
|
}
|
|
if (commaIdx !== -1) {
|
|
const keyStr = inner.substring(0, commaIdx).trim();
|
|
const valStr = inner.substring(commaIdx + 1).trim();
|
|
return { type: 'map', keyType: parseTypeExpr(keyStr), valueType: parseTypeExpr(valStr) };
|
|
}
|
|
// Fallback: treat as ident if malformed
|
|
return { type: 'ident', name: expr };
|
|
}
|
|
// List(T) — parenthetical syntax (SpeckDL canonical form)
|
|
if (expr.startsWith('List(') && expr.endsWith(')')) {
|
|
const inner = expr.substring(5, expr.length - 1).trim();
|
|
return { type: 'list', elementType: parseTypeExpr(inner) };
|
|
}
|
|
// Set(T) — parenthetical syntax (SpeckDL canonical form)
|
|
if (expr.startsWith('Set(') && expr.endsWith(')')) {
|
|
const inner = expr.substring(4, expr.length - 1).trim();
|
|
return { type: 'set', elementType: parseTypeExpr(inner) };
|
|
}
|
|
// Map(K, V) — parenthetical syntax (SpeckDL canonical form)
|
|
if (expr.startsWith('Map(') && expr.endsWith(')')) {
|
|
const inner = expr.substring(4, expr.length - 1);
|
|
let depth = 0;
|
|
let braceDepth = 0;
|
|
let commaIdx = -1;
|
|
for (let i = 0; i < inner.length; i++) {
|
|
if (inner[i] === '(' || inner[i] === '<')
|
|
depth++;
|
|
else if (inner[i] === ')' || inner[i] === '>')
|
|
depth--;
|
|
else if (inner[i] === '{')
|
|
braceDepth++;
|
|
else if (inner[i] === '}')
|
|
braceDepth--;
|
|
else if (inner[i] === ',' && depth === 0 && braceDepth === 0) {
|
|
commaIdx = i;
|
|
break;
|
|
}
|
|
}
|
|
if (commaIdx !== -1) {
|
|
const keyStr = inner.substring(0, commaIdx).trim();
|
|
const valStr = inner.substring(commaIdx + 1).trim();
|
|
return { type: 'map', keyType: parseTypeExpr(keyStr), valueType: parseTypeExpr(valStr) };
|
|
}
|
|
// Fallback: treat as ident if malformed
|
|
return { type: 'ident', name: expr };
|
|
}
|
|
// List type: [T]
|
|
if (expr.startsWith('[') && expr.endsWith(']')) {
|
|
const elementType = parseTypeExpr(expr.substring(1, expr.length - 1));
|
|
return { type: 'list', elementType };
|
|
}
|
|
// Record type: { ... }
|
|
if (expr.startsWith('{') && expr.endsWith('}')) {
|
|
const content = expr.substring(1, expr.length - 1).trim();
|
|
const fields = [];
|
|
if (content) {
|
|
// Split by comma and parse each field
|
|
const fieldStrs = content.split(',').map(s => s.trim());
|
|
for (const fieldStr of fieldStrs) {
|
|
const fieldMatch = fieldStr.match(/^(\w+)\s*:\s*(.+)$/);
|
|
if (fieldMatch) {
|
|
const [, name, typeStr] = fieldMatch;
|
|
fields.push({ name, type: parseTypeExpr(typeStr.trim()) });
|
|
}
|
|
}
|
|
}
|
|
return { type: 'record', fields };
|
|
}
|
|
// Identifier (for interface references, etc.)
|
|
return { type: 'ident', name: expr };
|
|
}
|
|
function parseInterface(line) {
|
|
// interface Name {
|
|
const match = line.match(/^interface\s+(\w+)\s*{/);
|
|
if (!match) {
|
|
throw new Error(`Failed to parse interface: ${line}`);
|
|
}
|
|
const name = match[1];
|
|
// This is a simplified parser - in a real implementation, we'd need to handle multi-line interfaces
|
|
return {
|
|
type: 'interface',
|
|
name,
|
|
methods: [],
|
|
};
|
|
}
|
|
function parseEvent(line) {
|
|
// event Name {
|
|
const match = line.match(/^event\s+(\w+)\s*{/);
|
|
if (!match) {
|
|
throw new Error(`Failed to parse event: ${line}`);
|
|
}
|
|
const name = match[1];
|
|
return {
|
|
type: 'event',
|
|
name,
|
|
fields: [],
|
|
};
|
|
}
|
|
function parseProvenanceBlock(lines, line) {
|
|
// This is a simplified parser - in a real implementation, we'd parse the full block
|
|
return {
|
|
type: 'provenance',
|
|
clauses: [],
|
|
};
|
|
}
|
|
function parseDerives(line) {
|
|
// derives from Name via "rationale"
|
|
const match = line.match(/^derives\s+from\s+(\w+(?:-\w+)*)(?:\s+via\s+(['"])(.+?)\2)?/);
|
|
if (!match) {
|
|
console.warn(`Failed to parse derives statement: ${line}`);
|
|
return { type: 'derives', from: 'unknown' };
|
|
}
|
|
const [, from, , via] = match;
|
|
return {
|
|
type: 'derives',
|
|
from,
|
|
via: via || undefined,
|
|
};
|
|
}
|
|
function parseSatisfies(line) {
|
|
// satisfies Name [clause "X.Y.Z"]
|
|
// e.g., satisfies RESILIENCE-REQ-01 clause "3.1.2"
|
|
// e.g., satisfies RESILIENCE-REQ-02
|
|
const match = line.match(/^satisfies\s+([\w-]+(?:-[\w]+)*)(?:\s+clause\s+(['"])(.+?)\2)?/);
|
|
if (!match) {
|
|
console.warn(`Failed to parse satisfies statement: ${line}`);
|
|
return { type: 'satisfies', requirement: 'unknown' };
|
|
}
|
|
const [, requirement, , clause] = match;
|
|
return {
|
|
type: 'satisfies',
|
|
requirement,
|
|
clause: clause || undefined,
|
|
};
|
|
}
|
|
function parseAuthor(line) {
|
|
// author: "Name" <"email">
|
|
// e.g., author: "Scott Scoble" <"scott@scoble.me">
|
|
const match = line.match(/^author:\s+(['"])(.+?)\1(?:\s+<(['"])(.+?)\3>)?/);
|
|
if (!match) {
|
|
// Try without angle brackets or with different format
|
|
console.warn(`Failed to parse author statement: ${line}`);
|
|
return { type: 'author', name: 'unknown' };
|
|
}
|
|
const [, , name, , email] = match;
|
|
return {
|
|
type: 'author',
|
|
name,
|
|
email: email || undefined,
|
|
};
|
|
}
|
|
function parseSource(line) {
|
|
// source: kind ref "identifier"
|
|
// e.g., source: architecture_review ref "AR-2024-0215"
|
|
const match = line.match(/^source:\s+(\w+)(?:\s+ref\s+(['"])(.+?)\2)?/);
|
|
if (!match) {
|
|
console.warn(`Failed to parse source statement: ${line}`);
|
|
return { type: 'source', kind: 'document' };
|
|
}
|
|
const [, kind, , ref] = match;
|
|
const validKinds = [
|
|
'conversation', 'meeting', 'document', 'regulation',
|
|
'architecture_review', 'threat_model', 'compliance_audit'
|
|
];
|
|
if (!validKinds.includes(kind)) {
|
|
console.warn(`Invalid source kind: ${kind}, defaulting to document`);
|
|
return { type: 'source', kind: 'document', ref: ref || undefined };
|
|
}
|
|
return {
|
|
type: 'source',
|
|
kind: kind,
|
|
ref: ref || undefined,
|
|
};
|
|
}
|
|
function parseStateBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const endIndex = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIndex).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
// Join lines: use '; ' between complete variable declarations
|
|
const inner = innerLines.reduce((acc, line, i) => {
|
|
if (i === 0)
|
|
return line;
|
|
const prevLine = innerLines[i - 1];
|
|
// Check if this line starts a new variable declaration (name: type)
|
|
const isNewVar = /^\w+\s*:/.test(line);
|
|
// Check if prev line ends with a continuation character
|
|
const prevIsContinuation = /[{(,]\s*$/.test(prevLine);
|
|
if (isNewVar && !prevIsContinuation) {
|
|
return acc + '; ' + line;
|
|
}
|
|
return acc + ' ' + line;
|
|
}, '');
|
|
return parseStateBlock(inner);
|
|
}
|
|
function parseInitBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const endIndex = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIndex).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const inner = innerLines.reduce((acc, line, i) => {
|
|
if (i === 0)
|
|
return line;
|
|
const prevLine = innerLines[i - 1];
|
|
const isNewVar = /^\w+\s*:/.test(line);
|
|
const prevIsContinuation = /[{(,]\s*$/.test(prevLine);
|
|
if (isNewVar && !prevIsContinuation) {
|
|
return acc + '; ' + line;
|
|
}
|
|
return acc + ' ' + line;
|
|
}, '');
|
|
return parseInitBlock(inner);
|
|
}
|
|
function parseActionBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const firstLine = lines[startIndex].trim();
|
|
const header = parseActionHeader(firstLine);
|
|
if (!header)
|
|
return null;
|
|
const endIndex = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const bodyLines = lines.slice(startIndex + 1, endIndex).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const statements = [];
|
|
for (let i = 0; i < bodyLines.length; i++) {
|
|
const stmt = bodyLines[i];
|
|
// precondition: expr
|
|
if (stmt.startsWith('precondition:')) {
|
|
let expr = stmt.substring(13).trim();
|
|
// support 'or' continuation lines
|
|
while (i + 1 < bodyLines.length && bodyLines[i + 1].startsWith('or ')) {
|
|
i++;
|
|
expr += ' || ' + bodyLines[i].substring(3).trim();
|
|
}
|
|
statements.push({ type: 'precondition', expr });
|
|
continue;
|
|
}
|
|
// postcondition: expr
|
|
if (stmt.startsWith('postcondition:')) {
|
|
let expr = stmt.substring(14).trim();
|
|
// support 'or' continuation lines
|
|
while (i + 1 < bodyLines.length && bodyLines[i + 1].startsWith('or ')) {
|
|
i++;
|
|
expr += ' || ' + bodyLines[i].substring(3).trim();
|
|
}
|
|
statements.push({ type: 'postcondition', expr });
|
|
continue;
|
|
}
|
|
// require expr
|
|
if (stmt.startsWith('require ')) {
|
|
let expr = stmt.substring(8).trim();
|
|
// support multi-line continuation (forall, or, etc.)
|
|
while (i + 1 < bodyLines.length) {
|
|
const nextLine = bodyLines[i + 1];
|
|
// Stop if next line starts a new statement keyword
|
|
if (nextLine.startsWith('precondition:') || nextLine.startsWith('postcondition:') ||
|
|
nextLine.startsWith('require ') || nextLine.startsWith('return ') ||
|
|
nextLine.startsWith('emit ') || nextLine.startsWith('let ') ||
|
|
nextLine.includes(':=')) {
|
|
break;
|
|
}
|
|
i++;
|
|
expr += ' ' + nextLine;
|
|
}
|
|
statements.push({ type: 'require', expr });
|
|
continue;
|
|
}
|
|
// return expr
|
|
if (stmt.startsWith('return ')) {
|
|
statements.push({ type: 'return', expr: stmt.substring(7).trim() });
|
|
continue;
|
|
}
|
|
// emit Event { fields }
|
|
const emitMatch = stmt.match(/^emit\s+(\w+)\s*\{\s*(.*)\s*\}$/);
|
|
if (emitMatch) {
|
|
const eventName = emitMatch[1];
|
|
const fieldsStr = emitMatch[2];
|
|
const fields = [];
|
|
if (fieldsStr) {
|
|
const fieldPairs = fieldsStr.split(',').map(s => s.trim()).filter(Boolean);
|
|
for (const fp of fieldPairs) {
|
|
const colonIdx = fp.indexOf(':');
|
|
if (colonIdx > 0) {
|
|
fields.push({
|
|
name: fp.substring(0, colonIdx).trim(),
|
|
value: fp.substring(colonIdx + 1).trim(),
|
|
});
|
|
}
|
|
}
|
|
}
|
|
statements.push({ type: 'emit', event: eventName, fields });
|
|
continue;
|
|
}
|
|
// let name := expr
|
|
const letMatch = stmt.match(/^let\s+(\w+)\s*:=\s*(.+)$/);
|
|
if (letMatch) {
|
|
statements.push({ type: 'let', name: letMatch[1], expr: letMatch[2].trim() });
|
|
continue;
|
|
}
|
|
// assignment: target := expr (may span multiple lines)
|
|
const assignMatch = stmt.match(/^(\w+(?:\[.*?\])?)\s*:=\s*(.+)$/);
|
|
if (assignMatch) {
|
|
let target = assignMatch[1];
|
|
let expr = assignMatch[2].trim();
|
|
// Handle multiline expressions: join subsequent lines until braces/parens are balanced
|
|
while (i + 1 < bodyLines.length && !isExpressionComplete(expr)) {
|
|
i++;
|
|
expr += ' ' + bodyLines[i];
|
|
}
|
|
statements.push({ type: 'assign', target, expr });
|
|
}
|
|
}
|
|
return { ...header, statements };
|
|
}
|
|
function parseEventBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const firstLine = lines[startIndex].trim();
|
|
const header = parseEvent(firstLine);
|
|
const endIndex = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIndex).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const fields = [];
|
|
for (const line of innerLines) {
|
|
const match = line.match(/^(\w+)\s*:\s*(.+?),?$/);
|
|
if (match) {
|
|
fields.push({ name: match[1], type: parseTypeExpr(match[2].replace(/,$/, '').trim()) });
|
|
}
|
|
}
|
|
return { ...header, fields };
|
|
}
|
|
function parseInterfaceBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const firstLine = lines[startIndex].trim();
|
|
const header = parseInterface(firstLine);
|
|
const endIndex = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIndex).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const methods = [];
|
|
for (const line of innerLines) {
|
|
// name(params) -> returnType
|
|
const match = line.match(/^(\w+)\s*\(([^)]*)\)\s*-\u003e\s*(.+?),?$/);
|
|
if (match) {
|
|
const name = match[1];
|
|
const paramsStr = match[2];
|
|
const returnTypeStr = match[3].replace(/,$/, '').trim();
|
|
const params = [];
|
|
if (paramsStr.trim()) {
|
|
const paramPairs = paramsStr.split(',').map(s => s.trim()).filter(Boolean);
|
|
for (const p of paramPairs) {
|
|
const pMatch = p.match(/^(\w+)\s*:\s*(.+)$/);
|
|
if (pMatch) {
|
|
params.push({ name: pMatch[1], type: parseTypeExpr(pMatch[2].trim()) });
|
|
}
|
|
}
|
|
}
|
|
methods.push({ name, params, returnType: parseTypeExpr(returnTypeStr) });
|
|
}
|
|
}
|
|
return { ...header, methods };
|
|
}
|
|
// --- Multi-line constraint/verify parsing (no braces needed) ---
|
|
/**
|
|
* Parse a constraint or verify that may span multiple lines without braces.
|
|
* Used for forall-style constraints like:
|
|
* constraint Name: forall r in [0..4]:
|
|
* (expr)
|
|
*
|
|
* and verify blocks like:
|
|
* verify Always(Name)
|
|
* verify Always(Name) { depth N }
|
|
*/
|
|
function parseMultiLineConstraintVerify(lines, startIndex) {
|
|
const line = lines[startIndex].trim();
|
|
if (line.startsWith('verify ')) {
|
|
return parseVerifyAnyForm(lines, startIndex);
|
|
}
|
|
if (line.startsWith('constraint ')) {
|
|
return parseConstraintAnyForm(lines, startIndex);
|
|
}
|
|
return null;
|
|
}
|
|
function parseConstraintAnyForm(lines, startIndex) {
|
|
const line = lines[startIndex].trim();
|
|
const namedMatch = line.match(/^constraint\s+(\w+):(.*)$/);
|
|
if (!namedMatch)
|
|
return null;
|
|
const name = namedMatch[1];
|
|
let expr = namedMatch[2].trim();
|
|
// If opens a brace, treat as block
|
|
if (expr.startsWith('{')) {
|
|
const afterBrace = expr.substring(1).trim();
|
|
const openCount = (line.match(/{/g) || []).length;
|
|
const endIdx = findBlockEnd(lines, startIndex + 1, openCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIdx).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const inner = innerLines.join(' ');
|
|
expr = (afterBrace + ' ' + inner).trim();
|
|
}
|
|
else {
|
|
// No braces — but might span multiple lines (forall continuation)
|
|
const continuationLines = [];
|
|
for (let j = startIndex + 1; j < lines.length; j++) {
|
|
const nl = lines[j].trim();
|
|
if (!nl || nl.startsWith('//') || nl.startsWith('/*') ||
|
|
nl.startsWith('constraint ') || nl.startsWith('verify ') ||
|
|
nl.startsWith('constraint:') || nl.startsWith('verify:') ||
|
|
nl.startsWith('state') || nl.startsWith('init') ||
|
|
nl.startsWith('action ') || nl.startsWith('event ') ||
|
|
nl.startsWith('interface ') || nl.startsWith('provenance ') ||
|
|
nl.startsWith('bom ') || nl.startsWith('}') ||
|
|
nl.startsWith('next:') || nl.startsWith('invariant ')) {
|
|
break;
|
|
}
|
|
continuationLines.push(nl);
|
|
}
|
|
if (continuationLines.length > 0) {
|
|
expr = expr + ' ' + continuationLines.join(' ');
|
|
}
|
|
}
|
|
return { type: 'constraint', name, expr: expr.trim() };
|
|
}
|
|
function parseVerifyAnyForm(lines, startIndex) {
|
|
const line = lines[startIndex].trim();
|
|
// Single-line: verify Always(Name) { depth N }
|
|
const fullMatch = line.match(/^verify\s+((?:Always|always|Eventually|eventually)\s*\(\s*(\w+)\s*\))\s*\{\s*depth\s+(\d+)\s*\}/);
|
|
if (fullMatch) {
|
|
return {
|
|
type: 'verify',
|
|
name: fullMatch[2],
|
|
temporalExpr: fullMatch[1].trim(),
|
|
depth: parseInt(fullMatch[3], 10),
|
|
};
|
|
}
|
|
// Multi-line: verify Always(Name) { ... }
|
|
if (line.includes('{')) {
|
|
const headerMatch = line.match(/^verify\s+((?:Always|always|Eventually|eventually)\s*\(\s*(\w+)\s*\))\s*\{/);
|
|
if (headerMatch) {
|
|
const temporalExpr = headerMatch[1].trim();
|
|
const name = headerMatch[2];
|
|
const openCount = (line.match(/{/g) || []).length;
|
|
const endIdx = findBlockEnd(lines, startIndex + 1, openCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIdx).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const inner = innerLines.join(' ');
|
|
const afterBrace = line.substring(line.indexOf('{') + 1).trim();
|
|
const combined = (afterBrace + ' ' + inner).trim();
|
|
const depthMatch = combined.match(/\bdepth\s+(\d+)\b/);
|
|
const depth = depthMatch ? parseInt(depthMatch[1], 10) : undefined;
|
|
return { type: 'verify', name, temporalExpr, depth };
|
|
}
|
|
}
|
|
// Simple: verify Always(Name)
|
|
const simpleMatch = line.match(/^verify\s+((?:Always|always|Eventually|eventually)\s*\(\s*(\w+)\s*\))\s*$/);
|
|
if (simpleMatch) {
|
|
return {
|
|
type: 'verify',
|
|
name: simpleMatch[2],
|
|
temporalExpr: simpleMatch[1].trim(),
|
|
};
|
|
}
|
|
return null;
|
|
}
|
|
// --- Verify/Constraint block parsing ---
|
|
/**
|
|
* Parse a verify block opened with `verify Always(Name) { ... }` or similar.
|
|
* The first line already contains the header; we need to parse the body inside { }.
|
|
*/
|
|
function parseVerifyBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const firstLine = lines[startIndex].trim();
|
|
// Match: verify Always(Name) { depth N } or verify Always(Name) {
|
|
const headerMatch = firstLine.match(/^verify\s+((?:Always|always|Eventually|eventually)\s*\(\s*(\w+)\s*\))\s*\{/);
|
|
if (!headerMatch)
|
|
return null;
|
|
const temporalExpr = headerMatch[1].trim();
|
|
const name = headerMatch[2];
|
|
// Check if the brace closes on the same line (single-line form)
|
|
const openBraceCount = (firstLine.match(/{/g) || []).length;
|
|
const closeBraceCount = (firstLine.match(/}/g) || []).length;
|
|
if (closeBraceCount >= openBraceCount) {
|
|
// Single-line: parse depth from same line
|
|
const afterBrace = firstLine.substring(firstLine.indexOf('{') + 1);
|
|
const beforeClose = afterBrace.substring(0, afterBrace.lastIndexOf('}')).trim();
|
|
const depthMatch = beforeClose.match(/\bdepth\s+(\d+)\b/);
|
|
const depth = depthMatch ? parseInt(depthMatch[1], 10) : undefined;
|
|
return { type: 'verify', name, temporalExpr, depth };
|
|
}
|
|
// Multi-line block
|
|
let depth;
|
|
const endIdx = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIdx).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const inner = innerLines.join(' ');
|
|
const afterBrace = firstLine.substring(firstLine.indexOf('{') + 1).trim();
|
|
const depthMatch = (afterBrace + ' ' + inner).match(/\bdepth\s+(\d+)\b/);
|
|
if (depthMatch) {
|
|
depth = parseInt(depthMatch[1], 10);
|
|
}
|
|
return { type: 'verify', name, temporalExpr, depth };
|
|
}
|
|
/**
|
|
* Parse a single-line verify statement: verify Always(Name) { depth N }
|
|
*/
|
|
function parseVerifySingleLine(line) {
|
|
// verify Always(Name) { depth N }
|
|
const fullMatch = line.match(/^verify\s+((?:Always|always|Eventually|eventually)\s*\(\s*(\w+)\s*\))\s*\{\s*depth\s+(\d+)\s*\}/);
|
|
if (fullMatch) {
|
|
return {
|
|
type: 'verify',
|
|
name: fullMatch[2],
|
|
temporalExpr: fullMatch[1].trim(),
|
|
depth: parseInt(fullMatch[3], 10),
|
|
};
|
|
}
|
|
// verify Always(Name) (no depth, no braces — just a statement)
|
|
const simpleMatch = line.match(/^verify\s+((?:Always|always|Eventually|eventually)\s*\(\s*(\w+)\s*\))\s*$/);
|
|
if (simpleMatch) {
|
|
return {
|
|
type: 'verify',
|
|
name: simpleMatch[2],
|
|
temporalExpr: simpleMatch[1].trim(),
|
|
};
|
|
}
|
|
return null;
|
|
}
|
|
/**
|
|
* Parse a constraint block opened with `constraint Name: ...` which may
|
|
* be multi-line (e.g. forall spanning multiple lines).
|
|
*/
|
|
function parseConstraintBlockMultiline(lines, startIndex, startBraceCount) {
|
|
const firstLine = lines[startIndex].trim();
|
|
// Match: constraint Name: expr (possibly opening a {} block, or just a forall on next line)
|
|
// For the SQLiteWAL style: constraint Name: expr (single line, no braces needed)
|
|
const namedMatch = firstLine.match(/^constraint\s+(\w+):(.*)$/);
|
|
if (namedMatch) {
|
|
const name = namedMatch[1];
|
|
let expr = namedMatch[2].trim();
|
|
// If the expression opens a brace, collect the block
|
|
if (expr.startsWith('{')) {
|
|
const endIdx = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIdx).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
const inner = innerLines.join(' ');
|
|
// Combine: content after { on first line + content from inner lines
|
|
const afterBrace = expr.substring(1).trim();
|
|
expr = (afterBrace + ' ' + inner).trim();
|
|
}
|
|
else {
|
|
// No braces, but might still be multiline (forall spanning multiple lines)
|
|
// Check if next line(s) continue the expression
|
|
const endIdx = findBlockEnd(lines, startIndex + 1, startBraceCount);
|
|
const innerLines = lines.slice(startIndex + 1, endIdx).map(l => l.trim()).filter(l => l && !l.startsWith('//') && !l.startsWith('/*'));
|
|
if (innerLines.length > 0) {
|
|
expr = expr + ' ' + innerLines.join(' ');
|
|
}
|
|
}
|
|
return { type: 'constraint', name, expr: expr.trim() };
|
|
}
|
|
return null;
|
|
}
|
|
//# sourceMappingURL=parser.js.map
|