'use strict';

import * as vscode from 'vscode';

import * as path from 'path';

import {BaseFormatter} from './../formatters/baseFormatter';

import {YapfFormatter} from './../formatters/yapfFormatter';

import {AutoPep8Formatter} from './../formatters/autoPep8Formatter';

import * as settings from './../common/configSettings';

import * as telemetryHelper from '../common/telemetry';

import * as telemetryContracts from '../common/telemetryContracts';

export class PythonFormattingEditProvider implements vscode.DocumentFormattingEditProvider, vscode.DocumentRangeFormattingEditProvider {

private formatters = new Map<string, BaseFormatter>();

public constructor(context: vscode.ExtensionContext, outputChannel: vscode.OutputChannel, private settings: settings.IPythonSettings, workspaceRootPath: string) {

let yapfFormatter = new YapfFormatter(outputChannel, settings, workspaceRootPath);

let autoPep8 = new AutoPep8Formatter(outputChannel, settings, workspaceRootPath);

this.formatters.set(yapfFormatter.Id, yapfFormatter);

this.formatters.set(autoPep8.Id, autoPep8);

}

public provideDocumentFormattingEdits(document: vscode.TextDocument, options: vscode.FormattingOptions, token: vscode.CancellationToken): Thenable<vscode.TextEdit[]> {

return this.provideDocumentRangeFormattingEdits(document, null, options, token);

}

public provideDocumentRangeFormattingEdits(document: vscode.TextDocument, range: vscode.Range, options: vscode.FormattingOptions, token: vscode.CancellationToken): Thenable<vscode.TextEdit[]> {

let formatter = this.formatters.get(this.settings.formatting.provider);

let delays = new telemetryHelper.Delays();

return formatter.formatDocument(document, options, token, range).then(edits => {

delays.stop();

telemetryHelper.sendTelemetryEvent(telemetryContracts.IDE.Format, { Format_Provider: formatter.Id }, delays.toMeasures());

return edits;

});

}

}