/*---------------------------------------------------------------------------------------------

* Copyright (c) Microsoft Corporation. All rights reserved.

* Licensed under the MIT License. See License.txt in the project root for license information.

*--------------------------------------------------------------------------------------------*/

import * as vscode from 'vscode';

import { API, Repository } from '../typings/git';

import { GitExtension } from '../typings/git';

export function getAPI() {

const gitExtension = vscode.extensions.getExtension<GitExtension>('vscode.git')!.exports;

const git = gitExtension.getAPI(1);

return git;

}

export class Model implements vscode.Disposable {

private _onDidOpenRepository = new vscode.EventEmitter<Repository>();

readonly onDidOpenRepository: vscode.Event<Repository> = this._onDidOpenRepository.event;

private _onDidCloseRepository = new vscode.EventEmitter<Repository>();

readonly onDidCloseRepository: vscode.Event<Repository> = this._onDidCloseRepository.event;

private _gitApi: API;

private _openRepositories: Repository[] = [];

get repositories(): Repository[] {

return this._openRepositories;

}

private _disposables: vscode.Disposable[];

constructor() {

this._disposables = [];

this._gitApi = getAPI();

this._openRepositories = this._gitApi.repositories;

this._disposables.push(this._gitApi.onDidCloseRepository(this._onDidCloseGitRepository.bind(this)));

this._disposables.push(this._gitApi.onDidOpenRepository(this._onDidOpenGitRepository.bind(this)));

}

private _onDidCloseGitRepository(repository: Repository) {

this._openRepositories = this._gitApi.repositories;

this._openRepositories = this.repositories.filter(e => e !== repository);

this._onDidCloseRepository.fire(repository);

}

private _onDidOpenGitRepository(repository: Repository) {

this._openRepositories = this._gitApi.repositories;

this._onDidOpenRepository.fire(repository);

}

public openRepository(repository: Repository) {

this._openRepositories.push(repository);

this._onDidOpenRepository.fire(repository);

}

public closeRepository(repository: Repository) {

this._openRepositories = this._openRepositories.filter(e => e !== repository);

this._onDidCloseRepository.fire(repository);

}

public getRepository(folder: vscode.WorkspaceFolder): Repository {

return this._openRepositories.filter(repository => (repository as any).workspaceFolder === folder)[0];

}

public dispose() {

this._disposables.forEach(d => d.dispose());

this._disposables = [];

this._openRepositories = [];

}

}