All files / languages/turtle/providers turtle-prefix-completion-provider.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
 
 
 
 
 
 
 
 
 
22x
 
 
7x
 
 
22x
22x
 
 
 
7x
 
7x
1x
 
 
6x
 
 
6x
1x
 
 
5x
5x
 
5x
1x
 
 
4x
 
4x
1x
 
 
 
3x
 
3x
1x
 
 
2x
2x
2x
 
2x
1x
 
 
 
 
1x
 
 
 
import * as vscode from 'vscode';
import { RdfToken } from '@faubulous/mentor-rdf-parsers';
import { container } from 'tsyringe';
import { ServiceToken } from '@src/services/tokens';
import { IDocumentContextService } from '@src/services/document';
import { IPrefixLookupService } from '@src/services/document';
import { TurtleDocument } from '@src/languages/turtle/turtle-document';
import { TurtleFeatureProvider } from '@src/languages/turtle/turtle-feature-provider';
 
export class TurtlePrefixCompletionProvider extends TurtleFeatureProvider implements vscode.InlineCompletionItemProvider {
	protected readonly prefixTokenTypes = new Set([RdfToken.PREFIX.name, RdfToken.TTL_PREFIX.name]);
 
	private get contextService() {
		return container.resolve<IDocumentContextService>(ServiceToken.DocumentContextService);
	}
 
	constructor(readonly onComplete: (uri: string) => string) {
		super();
	}
 
	provideInlineCompletionItems(document: vscode.TextDocument, position: vscode.Position, completion: vscode.InlineCompletionContext): vscode.ProviderResult<vscode.InlineCompletionItem[] | vscode.InlineCompletionList> {
		const context = this.contextService.getDocumentContext(document, TurtleDocument);
 
		if (!context) {
			return null;
		}
 
		const n = context.getTokenIndexAtPosition(position);
 
		// We also need the previous token to determine if this is a prefix definition.
		if (n < 1) {
			return null;
		}
 
		const currentToken = context.tokens[n];
		const currentType = currentToken.tokenType.name;
 
		if (!currentType || currentType !== RdfToken.PNAME_NS.name) {
			return;
		}
 
		const previousToken = context.tokens[n - 1];
 
		if (!previousToken) {
			return null;
		}
 
		// Only do inline completion for prefix defitions.
		const previousType = previousToken.tokenType.name;
 
		if (!previousType || !this.prefixTokenTypes.has(previousType)) {
			return null;
		}
 
		const prefix = currentToken.image.split(":")[0];
		const prefixLookup = container.resolve<IPrefixLookupService>(ServiceToken.PrefixLookupService);
		const uri = prefixLookup.getUriForPrefix(document.uri.toString(), prefix);
 
		if (uri) {
			return [{
				insertText: this.onComplete(uri),
				range: new vscode.Range(position, position)
			}];
		} else {
			return [];
		}
	}
}