#12782

Committed 12 Jul 2024 05:13PM UTC coverage: 85.671%. Remained the same

Build Type

push

Committed by

Commit Message

Merge 34ae893f2 into 0b04e82c5

Run Details

10889 of 13587 branches covered (80.14%)

Branch coverage included in aggregate %.

60 of 66 new or added lines in 7 files covered. (90.91%)

64 existing lines in 4 files now uncovered.

12315 of 13498 relevant lines covered (91.24%)

26499.85 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

92.5

/src/bscPlugin/codeActions/CodeActionsProcessor.ts

import type { Diagnostic } from 'vscode-languageserver';
import { CodeActionKind } from 'vscode-languageserver';
import { codeActionUtil } from '../../CodeActionUtil';
import type { DiagnosticMessageType } from '../../DiagnosticMessages';
import { DiagnosticCodeMap } from '../../DiagnosticMessages';
import type { BrsFile } from '../../files/BrsFile';
import type { BscFile } from '../../files/BscFile';
import type { XmlFile } from '../../files/XmlFile';
import type { OnGetCodeActionsEvent } from '../../interfaces';
import { ParseMode } from '../../parser/Parser';
import { util } from '../../util';
import { isBrsFile } from '../../astUtils/reflection';

export class CodeActionsProcessor {
    public constructor(
        public event: OnGetCodeActionsEvent
    ) {

    }

    public process() {
        for (const diagnostic of this.event.diagnostics) {
            if (diagnostic.code === DiagnosticCodeMap.cannotFindName || diagnostic.code === DiagnosticCodeMap.cannotFindFunction) {
                this.suggestCannotFindName(diagnostic as any);
            } else if (diagnostic.code === DiagnosticCodeMap.xmlComponentMissingExtendsAttribute) {
                this.addMissingExtends(diagnostic as any);
            }
        }
    }

    private suggestedImports = new Set<string>();

    /**
     * Generic import suggestion function. Shouldn't be called directly from the main loop, but instead called by more specific diagnostic handlers
     */
    private suggestImports(diagnostic: Diagnostic, key: string, files: BscFile[]) {
        //skip if we already have this suggestion
        if (this.suggestedImports.has(key) || !isBrsFile(this.event.file)) {
            return;
        }

        this.suggestedImports.add(key);
        // eslint-disable-next-line @typescript-eslint/dot-notation
        const importStatements = this.event.file['_cachedLookups'].importStatements;
        //find the position of the first import statement, or the top of the file if there is none
        const insertPosition = importStatements[importStatements.length - 1]?.tokens.import?.location?.range?.start ?? util.createPosition(0, 0);

        //find all files that reference this function
        for (const file of files) {
            const destPath = util.sanitizePkgPath(file.destPath);
            this.event.codeActions.push(
                codeActionUtil.createCodeAction({
                    title: `import "${destPath}"`,
                    diagnostics: [diagnostic],
                    isPreferred: false,
                    kind: CodeActionKind.QuickFix,
                    changes: [{
                        type: 'insert',
                        filePath: this.event.file.srcPath,
                        position: insertPosition,
                        newText: `import "${destPath}"\n`
                    }]
                })
            );
        }
    }

    private suggestCannotFindName(diagnostic: DiagnosticMessageType<'cannotFindName'>) {
        //skip if not a BrighterScript file
        if ((diagnostic.file as BrsFile).parseMode !== ParseMode.BrighterScript) {
            return;
        }
        const lowerName = (diagnostic.data.fullName ?? diagnostic.data.name).toLowerCase();

        this.suggestImports(
            diagnostic,
            lowerName,
            [
                ...this.event.program.findFilesForFunction(lowerName),
                ...this.event.program.findFilesForClass(lowerName),
                ...this.event.program.findFilesForNamespace(lowerName),
                ...this.event.program.findFilesForEnum(lowerName)
            ]
        );
    }

    private addMissingExtends(diagnostic: DiagnosticMessageType<'xmlComponentMissingExtendsAttribute'>) {
        const srcPath = this.event.file.srcPath;
        const { componentElement } = (this.event.file as XmlFile).parser.ast;
        //inject new attribute after the final attribute, or after the `<component` if there are no attributes
        const pos = (componentElement.attributes[componentElement.attributes.length - 1] ?? componentElement.tokens.startTagName)?.location?.range.end;
        this.event.codeActions.push(
            codeActionUtil.createCodeAction({
                title: `Extend "Group"`,
                diagnostics: [diagnostic],
                isPreferred: true,
                kind: CodeActionKind.QuickFix,
                changes: [{
                    type: 'insert',
                    filePath: srcPath,
                    position: pos,
                    newText: ' extends="Group"'
                }]
            })
        );
        this.event.codeActions.push(
            codeActionUtil.createCodeAction({
                title: `Extend "Task"`,
                diagnostics: [diagnostic],
                kind: CodeActionKind.QuickFix,
                changes: [{
                    type: 'insert',
                    filePath: srcPath,
                    position: pos,
                    newText: ' extends="Task"'
                }]
            })
        );
        this.event.codeActions.push(
            codeActionUtil.createCodeAction({
                title: `Extend "ContentNode"`,
                diagnostics: [diagnostic],
                kind: CodeActionKind.QuickFix,
                changes: [{
                    type: 'insert',
                    filePath: srcPath,
                    position: pos,
                    newText: ' extends="ContentNode"'
                }]
            })
        );
    }
}

1	import type { Diagnostic } from 'vscode-languageserver';
2	import { CodeActionKind } from 'vscode-languageserver';	1✔
3	import { codeActionUtil } from '../../CodeActionUtil';	1✔
4	import type { DiagnosticMessageType } from '../../DiagnosticMessages';
5	import { DiagnosticCodeMap } from '../../DiagnosticMessages';	1✔
6	import type { BrsFile } from '../../files/BrsFile';
7	import type { BscFile } from '../../files/BscFile';
8	import type { XmlFile } from '../../files/XmlFile';
9	import type { OnGetCodeActionsEvent } from '../../interfaces';
10	import { ParseMode } from '../../parser/Parser';	1✔
11	import { util } from '../../util';	1✔
12	import { isBrsFile } from '../../astUtils/reflection';	1✔
13
14	export class CodeActionsProcessor {	1✔
15	public constructor(
16	public event: OnGetCodeActionsEvent	10✔
17	) {
18
19	}
20
21	public process() {
22	for (const diagnostic of this.event.diagnostics) {	10✔
23	if (diagnostic.code === DiagnosticCodeMap.cannotFindName \|\| diagnostic.code === DiagnosticCodeMap.cannotFindFunction) {	12✔
24	this.suggestCannotFindName(diagnostic as any);	8✔
25	} else if (diagnostic.code === DiagnosticCodeMap.xmlComponentMissingExtendsAttribute) {	4✔
26	this.addMissingExtends(diagnostic as any);	2✔
27	}
28	}
29	}
30
31	private suggestedImports = new Set<string>();	10✔
32
33	/**
34	* Generic import suggestion function. Shouldn't be called directly from the main loop, but instead called by more specific diagnostic handlers
35	*/
36	private suggestImports(diagnostic: Diagnostic, key: string, files: BscFile[]) {
37	//skip if we already have this suggestion
38	if (this.suggestedImports.has(key) \|\| !isBrsFile(this.event.file)) {	7!
UNCOV 39	return;	×
40	}
41
42	this.suggestedImports.add(key);	7✔
43	// eslint-disable-next-line @typescript-eslint/dot-notation
44	const importStatements = this.event.file['_cachedLookups'].importStatements;	7✔
45	//find the position of the first import statement, or the top of the file if there is none
46	const insertPosition = importStatements[importStatements.length - 1]?.tokens.import?.location?.range?.start ?? util.createPosition(0, 0);	7✔
47
48	//find all files that reference this function
49	for (const file of files) {	7✔
50	const destPath = util.sanitizePkgPath(file.destPath);	9✔
51	this.event.codeActions.push(	9✔
52	codeActionUtil.createCodeAction({
53	title: `import "${destPath}"`,
54	diagnostics: [diagnostic],
55	isPreferred: false,
56	kind: CodeActionKind.QuickFix,
57	changes: [{
58	type: 'insert',
59	filePath: this.event.file.srcPath,
60	position: insertPosition,
61	newText: `import "${destPath}"\n`
62	}]
63	})
64	);
65	}
66	}
67
68	private suggestCannotFindName(diagnostic: DiagnosticMessageType<'cannotFindName'>) {
69	//skip if not a BrighterScript file
70	if ((diagnostic.file as BrsFile).parseMode !== ParseMode.BrighterScript) {	8✔
71	return;	1✔
72	}
73	const lowerName = (diagnostic.data.fullName ?? diagnostic.data.name).toLowerCase();	7!
74
75	this.suggestImports(	7✔
76	diagnostic,
77	lowerName,
78	[
79	...this.event.program.findFilesForFunction(lowerName),
80	...this.event.program.findFilesForClass(lowerName),
81	...this.event.program.findFilesForNamespace(lowerName),
82	...this.event.program.findFilesForEnum(lowerName)
83	]
84	);
85	}
86
87	private addMissingExtends(diagnostic: DiagnosticMessageType<'xmlComponentMissingExtendsAttribute'>) {
88	const srcPath = this.event.file.srcPath;	2✔
89	const { componentElement } = (this.event.file as XmlFile).parser.ast;	2✔
90	//inject new attribute after the final attribute, or after the `<component` if there are no attributes
91	const pos = (componentElement.attributes[componentElement.attributes.length - 1] ?? componentElement.tokens.startTagName)?.location?.range.end;	2!
92	this.event.codeActions.push(	2✔
93	codeActionUtil.createCodeAction({
94	title: `Extend "Group"`,
95	diagnostics: [diagnostic],
96	isPreferred: true,
97	kind: CodeActionKind.QuickFix,
98	changes: [{
99	type: 'insert',
100	filePath: srcPath,
101	position: pos,
102	newText: ' extends="Group"'
103	}]
104	})
105	);
106	this.event.codeActions.push(	2✔
107	codeActionUtil.createCodeAction({
108	title: `Extend "Task"`,
109	diagnostics: [diagnostic],
110	kind: CodeActionKind.QuickFix,
111	changes: [{
112	type: 'insert',
113	filePath: srcPath,
114	position: pos,
115	newText: ' extends="Task"'
116	}]
117	})
118	);
119	this.event.codeActions.push(	2✔
120	codeActionUtil.createCodeAction({
121	title: `Extend "ContentNode"`,
122	diagnostics: [diagnostic],
123	kind: CodeActionKind.QuickFix,
124	changes: [{
125	type: 'insert',
126	filePath: srcPath,
127	position: pos,
128	newText: ' extends="ContentNode"'
129	}]
130	})
131	);
132	}
133	}

rokucommunity / brighterscript / #12782

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous