4585671a5d
* Refactor getNodePathFromSourceRange * A snapshot a day keeps the bugs away! 📷🐛 (OS: namespace-profile-ubuntu-8-cores) * Trigger CI * A snapshot a day keeps the bugs away! 📷🐛 (OS: namespace-profile-ubuntu-8-cores) * A snapshot a day keeps the bugs away! 📷🐛 (OS: namespace-profile-ubuntu-8-cores) * Trigger CI * A snapshot a day keeps the bugs away! 📷🐛 (OS: namespace-profile-ubuntu-8-cores) * Trigger CI * A snapshot a day keeps the bugs away! 📷🐛 (OS: namespace-profile-ubuntu-8-cores) * A snapshot a day keeps the bugs away! 📷🐛 (OS: namespace-profile-ubuntu-8-cores) * Trigger CI --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
115 lines
3.2 KiB
TypeScript
115 lines
3.2 KiB
TypeScript
import { getNodeFromPath } from './queryAst'
|
|
import { getNodePathFromSourceRange } from 'lang/queryAstNodePathUtils'
|
|
import {
|
|
Identifier,
|
|
assertParse,
|
|
initPromise,
|
|
Parameter,
|
|
SourceRange,
|
|
topLevelRange,
|
|
} 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 = topLevelRange(
|
|
lineToSubstringIndex,
|
|
lineToSubstringIndex + subStr.length
|
|
)
|
|
|
|
const ast = assertParse(code)
|
|
const nodePath = getNodePathFromSourceRange(ast, sourceRange)
|
|
const _node = getNodeFromPath<any>(ast, nodePath)
|
|
if (err(_node)) throw _node
|
|
const { node } = _node
|
|
|
|
expect(topLevelRange(node.start, node.end)).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 = topLevelRange(subStrIndex, subStrIndex + 'pos'.length)
|
|
|
|
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 = topLevelRange(subStrIndex, subStrIndex + 'scale'.length)
|
|
|
|
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')
|
|
})
|
|
})
|