extension.js

const vscode = require('vscode');
const openers = require('./openers');
const tokenInput = require('./token_input');
const tokenService = require('./token_service');
const pipelineActionsPicker = require('./pipeline_actions_picker');
const searchInput = require('./search_input');
const snippetInput = require('./snippet_input');
const sidebar = require('./sidebar');
const ciConfigValidator = require('./ci_config_validator');
const IssuableDataProvider = require('./data_providers/issuable').DataProvider;
const CurrentBranchDataProvider = require('./data_providers/current_branch').DataProvider;

let context = null;
vscode.gitLabWorkflow = {
  sidebarDataProviders: [],
};

const registerSidebarTreeDataProviders = () => {
  const assignedIssuesDataProvider = new IssuableDataProvider({
    fetcher: 'fetchIssuesAssignedToMe',
    noItemText: 'There is no issue assigned to you.',
  });

  const createdIssuesDataProvider = new IssuableDataProvider({
    fetcher: 'fetchIssuesCreatedByMe',
    noItemText: 'There is no issue created by you.',
  });

  const assignedMrsDataProvider = new IssuableDataProvider({
    fetcher: 'fetchMergeRequestsAssignedToMe',
    issuableType: 'merge request',
    noItemText: 'There is no MR assigned to you.',
  });

  const createdMrsDataProvider = new IssuableDataProvider({
    fetcher: 'fetchMergeRequestsCreatedByMe',
    issuableType: 'merge request',
    noItemText: 'There is no MR created by you.',
  });

  const allProjectMrsDataProvider = new IssuableDataProvider({
    fetcher: 'fetchAllProjectMergeRequests',
    issuableType: 'merge request',
    noItemText: 'The project has no merge requests',
  });

  const currentBranchDataProvider = new CurrentBranchDataProvider();

  const register = (name, provider) => {
    vscode.window.registerTreeDataProvider(name, provider);
    vscode.gitLabWorkflow.sidebarDataProviders.push(provider);
  };

  register('issuesAssignedToMe', assignedIssuesDataProvider);
  register('issuesCreatedByMe', createdIssuesDataProvider);
  register('mrsAssignedToMe', assignedMrsDataProvider);
  register('mrsCreatedByMe', createdMrsDataProvider);
  register('allProjectMrs', allProjectMrsDataProvider);
  register('currentBranchInfo', currentBranchDataProvider);
};

const registerCommands = () => {
  const commands = {
    'gl.showIssuesAssignedToMe': openers.showIssues,
    'gl.showMergeRequestsAssignedToMe': openers.showMergeRequests,
    'gl.setToken': tokenInput.showInput,
    'gl.removeToken': tokenInput.removeTokenPicker,
    'gl.openActiveFile': openers.openActiveFile,
    'gl.openCurrentMergeRequest': openers.openCurrentMergeRequest,
    'gl.openCreateNewIssue': openers.openCreateNewIssue,
    'gl.openCreateNewMR': openers.openCreateNewMr,
    'gl.openProjectPage': openers.openProjectPage,
    'gl.openCurrentPipeline': openers.openCurrentPipeline,
    'gl.pipelineActions': pipelineActionsPicker.showPicker,
    'gl.issueSearch': searchInput.showIssueSearchInput,
    'gl.mergeRequestSearch': searchInput.showMergeRequestSearchInput,
    'gl.compareCurrentBranch': openers.compareCurrentBranch,
    'gl.createSnippet': snippetInput.show,
    'gl.validateCIConfig': ciConfigValidator.validate,
    'gl.refreshSidebar': sidebar.refresh,
  };

  Object.keys(commands).forEach(cmd => {
    context.subscriptions.push(vscode.commands.registerCommand(cmd, commands[cmd]));
  });

  registerSidebarTreeDataProviders();
};

const init = () => {
  tokenService.init(context);
};

const activate = ctx => {
  context = ctx;
  registerCommands();
  init();
};

exports.init = init;
exports.activate = activate;