'use strict';

import * as vscode from 'vscode';

import { JediFactory } from '../languageServices/jediProxyFactory';

import { captureTelemetry } from '../telemetry';

import { EventName } from '../telemetry/constants';

import * as proxy from './jediProxy';

export class PythonReferenceProvider implements vscode.ReferenceProvider {

public constructor(private jediFactory: JediFactory) { }

private static parseData(data: proxy.IReferenceResult): vscode.Location[] {

if (data && data.references.length > 0) {

// tslint:disable-next-line:no-unnecessary-local-variable

const references = data.references.filter(ref => {

if (!ref || typeof ref.columnIndex !== 'number' || typeof ref.lineIndex !== 'number'

|| typeof ref.fileName !== 'string' || ref.columnIndex === -1 || ref.lineIndex === -1 || ref.fileName.length === 0) {

return false;

}

return true;

}).map(ref => {

const definitionResource = vscode.Uri.file(ref.fileName);

const range = new vscode.Range(ref.lineIndex, ref.columnIndex, ref.lineIndex, ref.columnIndex);

return new vscode.Location(definitionResource, range);

});

return references;

}

return [];

}

@captureTelemetry(EventName.REFERENCE)

public async provideReferences(document: vscode.TextDocument, position: vscode.Position, _context: vscode.ReferenceContext, token: vscode.CancellationToken): Promise<vscode.Location[] | undefined> {

const filename = document.fileName;

if (document.lineAt(position.line).text.match(/^\s*\/\//)) {

return;

}

if (position.character <= 0) {

return;

}

const range = document.getWordRangeAtPosition(position);

if (!range) {

return;

}

const columnIndex = range.isEmpty ? position.character : range.end.character;

const cmd: proxy.ICommand = {

command: proxy.CommandType.Usages,

fileName: filename,

columnIndex: columnIndex,

lineIndex: position.line

};

if (document.isDirty) {

cmd.source = document.getText();

}

const data = await this.jediFactory.getJediProxyHandler<proxy.IReferenceResult>(document.uri).sendCommand(cmd, token);

return data ? PythonReferenceProvider.parseData(data) : undefined;

}

}