✨ feat(vscode): 代码阅读加 Celery 库规则,定时任务表里的任务名能跳到任务函数

This commit is contained in:
cruldra committed 2026-09-26 19:41:07 +08:00
1 parent 905d7845b5
commit 68104e2cba
22 files changed
+2228 -139

No files matched your search

+20 -14
View File
@@ -1,38 +1,44 @@
import type { Location, LocationLink, Position, Range, Uri } from 'vscode'
import type { Location, LocationLink, Position } from 'vscode'
import type { RuleLocation } from '../contract'
import type { DefinitionTarget } from './definitionAction'
import { commands, window, workspace } from 'vscode'
import { commands, Range, Uri, window, workspace } from 'vscode'
import { logger } from '../../logging/logger'
/** 查到的一处定义:文件和定义处那个名字的范围。 */
export interface FoundLocation { uri: Uri, range: Range }
/** VS Code 查到的一处定义,加上打开目标文件才知道的信息。 */
export interface Definition extends DefinitionTarget {
location: { uri: Uri, range: Range }
location: FoundLocation
/** 定义处在目标文件里的下标,预览靠它找到是哪块、哪一行属性 */
offset: number
/** 定义所在那一行的源码,列在挑选框里 */
lineText: string
}
/** 交给 VS Code 注册的定义提供者(Pylance、TypeScript 语言服务……),和普通编辑器里 ⌘ + 点击查的是同一套。 */
async function provided(uri: Uri, position: Position): Promise<(Location | LocationLink)[]> {
/** uri 里 position 处名字的定义:交给 VS Code 注册的定义提供者(Pylance、TypeScript 语言服务……),和普通编辑器里 ⌘ + 点击查的是同一套。 */
export async function languageDefinitions(uri: Uri, position: Position): Promise<FoundLocation[]> {
let provided: (Location | LocationLink)[]
try {
return await commands.executeCommand<(Location | LocationLink)[] | undefined>('vscode.executeDefinitionProvider', uri, position) ?? []
provided = await commands.executeCommand<(Location | LocationLink)[] | undefined>('vscode.executeDefinitionProvider', uri, position) ?? []
}
catch (err) {
logger.add({ level: 'warn', source: 'codeReader', message: `查定义失败 ${uri.toString()}:${position.line + 1}:${err instanceof Error ? err.message : String(err)}` })
return []
}
return provided.map(item => 'targetUri' in item
? { uri: item.targetUri, range: item.targetSelectionRange ?? item.targetRange }
: { uri: item.uri, range: item.range })
}
export async function hasDefinition(uri: Uri, position: Position): Promise<boolean> {
return (await provided(uri, position)).length > 0
/** 库规则查到的定义(RuleLocation)换成 VS Code 的位置。 */
export function ruleDefinitions(locations: readonly RuleLocation[]): FoundLocation[] {
return locations.map(({ uri, start, end }) => ({ uri: Uri.parse(uri), range: new Range(start.line, start.character, end.line, end.character) }))
}
/** uri 里 position 处名字的所有定义;目标打不开成文本的(例如 Python 命名空间包指向的目录)略过。 */
export async function findDefinitions(uri: Uri, position: Position, supports: (languageId: string) => boolean): Promise<Definition[]> {
const found = await Promise.all((await provided(uri, position)).map(async (item): Promise<Definition[]> => {
const location = 'targetUri' in item
? { uri: item.targetUri, range: item.targetSelectionRange ?? item.targetRange }
: { uri: item.uri, range: item.range }
/** 查到的各处定义打开目标文件补上行文字、下标;目标打不开成文本的(例如 Python 命名空间包指向的目录)略过。 */
export async function findDefinitions(locations: readonly FoundLocation[], supports: (languageId: string) => boolean): Promise<Definition[]> {
const found = await Promise.all(locations.map(async (location): Promise<Definition[]> => {
const { start } = location.range
try {
const document = await workspace.openTextDocument(location.uri)