30d365aeb3
* Parse more import syntax Signed-off-by: Nick Cameron <nrc@ncameron.org> * Remove unnecessary Vec from VariableDeclaration Signed-off-by: Nick Cameron <nrc@ncameron.org> * Parse export import Signed-off-by: Nick Cameron <nrc@ncameron.org> * Factor out an execution module Signed-off-by: Nick Cameron <nrc@ncameron.org> * imports: constants, globs, export import Signed-off-by: Nick Cameron <nrc@ncameron.org> * test fixups Signed-off-by: Nick Cameron <nrc@ncameron.org> --------- Signed-off-by: Nick Cameron <nrc@ncameron.org>
116 lines
3.3 KiB
TypeScript
116 lines
3.3 KiB
TypeScript
import { getNodePathFromSourceRange, getNodeFromPath } from './queryAst'
|
|
import { Identifier, assertParse, initPromise, Parameter } from './wasm'
|
|
import { err } from 'lib/trap'
|
|
|
|
beforeAll(async () => {
|
|
await initPromise
|
|
})
|
|
|
|
describe('testing getNodePathFromSourceRange', () => {
|
|
it('test it gets the right path for a `lineTo` CallExpression within a SketchExpression', () => {
|
|
const code = `
|
|
const myVar = 5
|
|
const sk3 = startSketchAt([0, 0])
|
|
|> lineTo([1, 2], %)
|
|
|> lineTo([3, 4], %, $yo)
|
|
|> close(%)
|
|
`
|
|
const subStr = 'lineTo([3, 4], %, $yo)'
|
|
const lineToSubstringIndex = code.indexOf(subStr)
|
|
const sourceRange: [number, number, boolean] = [
|
|
lineToSubstringIndex,
|
|
lineToSubstringIndex + subStr.length,
|
|
true,
|
|
]
|
|
|
|
const ast = assertParse(code)
|
|
const nodePath = getNodePathFromSourceRange(ast, sourceRange)
|
|
const _node = getNodeFromPath<any>(ast, nodePath)
|
|
if (err(_node)) throw _node
|
|
const { node } = _node
|
|
|
|
expect([node.start, node.end, true]).toEqual(sourceRange)
|
|
expect(node.type).toBe('CallExpression')
|
|
})
|
|
it('gets path right for function definition params', () => {
|
|
const code = `fn cube = (pos, scale) => {
|
|
const sg = startSketchAt(pos)
|
|
|> line([0, scale], %)
|
|
|> line([scale, 0], %)
|
|
|> line([0, -scale], %)
|
|
|
|
return sg
|
|
}
|
|
|
|
const b1 = cube([0,0], 10)`
|
|
const subStr = 'pos, scale'
|
|
const subStrIndex = code.indexOf(subStr)
|
|
const sourceRange: [number, number, boolean] = [
|
|
subStrIndex,
|
|
subStrIndex + 'pos'.length,
|
|
true,
|
|
]
|
|
|
|
const ast = assertParse(code)
|
|
const nodePath = getNodePathFromSourceRange(ast, sourceRange)
|
|
const _node = getNodeFromPath<Parameter>(ast, nodePath)
|
|
if (err(_node)) throw _node
|
|
const node = _node.node
|
|
|
|
expect(nodePath).toEqual([
|
|
['body', ''],
|
|
[0, 'index'],
|
|
['declaration', 'VariableDeclaration'],
|
|
['init', ''],
|
|
['params', 'FunctionExpression'],
|
|
[0, 'index'],
|
|
])
|
|
expect(node.type).toBe('Parameter')
|
|
expect(node.identifier.name).toBe('pos')
|
|
})
|
|
it('gets path right for deep within function definition body', () => {
|
|
const code = `fn cube = (pos, scale) => {
|
|
const sg = startSketchAt(pos)
|
|
|> line([0, scale], %)
|
|
|> line([scale, 0], %)
|
|
|> line([0, -scale], %)
|
|
|
|
return sg
|
|
}
|
|
|
|
const b1 = cube([0,0], 10)`
|
|
const subStr = 'scale, 0'
|
|
const subStrIndex = code.indexOf(subStr)
|
|
const sourceRange: [number, number, boolean] = [
|
|
subStrIndex,
|
|
subStrIndex + 'scale'.length,
|
|
true,
|
|
]
|
|
|
|
const ast = assertParse(code)
|
|
const nodePath = getNodePathFromSourceRange(ast, sourceRange)
|
|
const _node = getNodeFromPath<Identifier>(ast, nodePath)
|
|
if (err(_node)) throw _node
|
|
const node = _node.node
|
|
expect(nodePath).toEqual([
|
|
['body', ''],
|
|
[0, 'index'],
|
|
['declaration', 'VariableDeclaration'],
|
|
['init', ''],
|
|
['body', 'FunctionExpression'],
|
|
['body', 'FunctionExpression'],
|
|
[0, 'index'],
|
|
['declaration', 'VariableDeclaration'],
|
|
['init', ''],
|
|
['body', 'PipeExpression'],
|
|
[2, 'index'],
|
|
['arguments', 'CallExpression'],
|
|
[0, 'index'],
|
|
['elements', 'ArrayExpression'],
|
|
[0, 'index'],
|
|
])
|
|
expect(node.type).toBe('Identifier')
|
|
expect(node.name).toBe('scale')
|
|
})
|
|
})
|