✨ feat(vscode): shell 被当成一串步骤执行时(Dockerfile 的 RUN)只用 && 连着的一串拆成一条一步;tree-sitter-bash 把 a && b > f 读成整串带重定向,重定向改算在最后那条命令上

This commit is contained in:
cruldra committed 2026-09-28 11:53:54 +08:00
1 parent 3353cf870c
commit 149cb51cc0
2 files changed
+90 -2

No files matched your search

@@ -138,6 +138,33 @@ describe('shell 属性表信息(facts)', () => {
})
})
describe('shell 被别的语言嵌着当步骤执行(Dockerfile 的 RUN)', () => {
const steps = async (source: string): Promise<{ blocks: Outline[], chained: boolean, headers: string[] } | undefined> => {
const parsed = await runtime.parse(adapter, 'shellscript', source)
try {
const found = adapter.steps?.(parsed.rootNode, source, 'shellscript')
return found && { blocks: outline(found.blocks), chained: found.chained, headers: found.blocks.map(block => source.slice(block.header.startIndex, block.header.endIndex)) }
}
finally {
parsed.delete()
}
}
it('只用 && 连着的一串拆成一条一步(一步失败就停);整串带着的重定向只算在最后一条上,它的头部延到重定向结尾', async () => {
expect(await steps('apt-get update && echo hi > /f && rm -rf /x')).toEqual({
blocks: [{ at: 1, kind: 'command' }, { at: 1, kind: 'command' }, { at: 1, kind: 'command' }],
chained: true,
headers: ['apt-get update', 'echo hi > /f', 'rm -rf /x'],
})
})
it('条件串、有 || 的串、几句分开写的照 toBlocks 一句一步,不算一步失败就停', async () => {
expect(await steps('[ -f x ] && rm x')).toMatchObject({ blocks: [{ at: 1, kind: 'if' }], chained: false })
expect(await steps('a && b || c')).toMatchObject({ blocks: [{ at: 1, kind: 'command' }], chained: false })
expect(await steps('a\nb')).toMatchObject({ blocks: [{ at: 1, kind: 'command' }, { at: 2, kind: 'command' }], chained: false })
})
})
describe('shell 命令、名字', () => {
const commands = (source: string): Promise<unknown[]> => nodeFacts(adapter, 'shellscript', source, node => adapter.command?.(node))
+63 -2
View File
@@ -1,4 +1,4 @@
import type { BlockShape, CodeBlock, CommandFacts, CommandWord, DeclaratorFacts, FunctionFacts, InterpreterFacts, LanguageAdapter, LocalSymbol, Node, OptionsFacts, PositionalParameter, Redirect, SourceRange, SwitchFacts, TextKind, TokenRole } from '../contract'
import type { BlockShape, CodeBlock, CommandFacts, CommandWord, DeclaratorFacts, FunctionFacts, InterpreterFacts, LanguageAdapter, LocalSymbol, Node, OptionsFacts, PositionalParameter, Redirect, SourceRange, StepBlocks, SwitchFacts, TextKind, TokenRole } from '../contract'
import type { CommentRules, ListBlocks } from './shared'
import * as path from 'node:path'
import { bashNameReader } from './bashNames'
@@ -461,7 +461,27 @@ function bashCommand(node: Node): CommandFacts | undefined {
if (!command || !name)
return undefined
const args = command.childrenForFieldName('argument').filter(arg => arg !== null)
return { name: commandWord(name), arguments: args.map(commandWord), redirects: redirectsOf(node) }
const inherited = node.type === 'command' ? redirectedChain(node) : undefined
return { name: commandWord(name), arguments: args.map(commandWord), redirects: [...redirectsOf(node), ...(inherited ? redirectsOf(inherited) : [])] }
}
/** 一串 `a && b` 里的最后一句。 */
function lastOperand(list: Node): Node | undefined {
return named(list).filter(child => child.type !== 'comment').at(-1)
}
/**
* tree-sitter-bash 把 `a && b > f` 读成整串带着重定向(redirected_statement 的 body 是 `a && b`),其实重定向只属于最后那条命令 b:
* command 是这样一串的最后一条时给出那个 redirected_statement,它的重定向算在 command 上。
*/
function redirectedChain(command: Node): Node | undefined {
let child = command
let parent = command.parent
while (parent?.type === 'list' && lastOperand(parent)?.id === child.id) {
child = parent
parent = parent.parent
}
return child.id !== command.id && parent?.type === 'redirected_statement' && parent.childForFieldName('body')?.id === child.id ? parent : undefined
}
// ─── 名字 ────────────────────────────────────────────────────────────────────
@@ -569,6 +589,46 @@ function bashTokenRole(leaf: Node): TokenRole {
return DECLARATION_KEYWORDS.has(leaf.type) ? 'storage' : 'plain'
}
/** 一串 `a && b` 本身(整串带着重定向的,见 redirectedChain,是它 body 里的那串)。 */
function chainList(node: Node): Node | undefined {
const body = node.type === 'redirected_statement' ? node.childForFieldName('body') : node
return body?.type === 'list' ? body : undefined
}
/** 一串里的各条命令,照源码顺序;整串带着重定向的往里拆(重定向算在最后一条上)。 */
function chainOperands(node: Node): Node[] {
const list = chainList(node)
return list ? named(list).filter(child => child.type !== 'comment').flatMap(chainOperands) : [node]
}
/** `a && b && c`:只用 && 连着的一串(都不是条件)。 */
function andChain(node: Node): boolean {
const operators = (list: Node): string[] => children(list).flatMap((child) => {
const inner = chainList(child)
return inner ? operators(inner) : child.isNamed ? [] : [child.type]
})
const list = chainList(node)
return !!list && operators(list).every(operator => operator === '&&') && !chainOperands(node).some(isTest)
}
/** 一步的块:整串带着的重定向算在最后一条上,它的头部延到重定向结尾。 */
function stepBlocks(operand: Node): CodeBlock[] {
const redirected = operand.type === 'command' ? redirectedChain(operand) : undefined
return statementBlocks(operand, true).map(block => redirected ? { ...block, header: { ...block.header, endIndex: redirected.endIndex, endRow: redirected.endPosition.row } } : block)
}
/**
* 被别的语言当一串步骤执行的 shell(Dockerfile 的 RUN):只有一句、是只用 && 连着的一串命令时拆开,一条命令一步(一步失败就停);
* 别的照 toBlocks 一句一步。
*/
function bashSteps(root: Node): StepBlocks {
const items = named(root)
const [only, ...more] = items.filter(item => item.type !== 'comment')
if (only && more.length === 0 && andChain(only))
return { blocks: chainOperands(only).flatMap(stepBlocks), chained: true }
return { blocks: statements(items, -1, true), chained: false }
}
/** bash / sh / zsh 脚本(VS Code 的 languageId 是 shellscript)的语言插件。@param wasmDir 语法包 .wasm 所在目录 */
export function createBashAdapter(wasmDir: string): LanguageAdapter {
return {
@@ -576,6 +636,7 @@ export function createBashAdapter(wasmDir: string): LanguageAdapter {
languageIds: ['shellscript'],
grammarPath: () => path.join(wasmDir, 'tree-sitter-bash.wasm'),
toBlocks: root => programBlocks(root),
steps: root => bashSteps(root),
tokenRole: bashTokenRole,
command: bashCommand,
symbols: root => bashSymbols(root),