iterative/vscode-dvc

import { EventEmitter } from 'vscode'
import { Args } from './constants'
import { getOptions } from './options'
import { Cli, CliResult, CliStarted } from '..'
import { Config } from '../../config'

export class DvcCli extends Cli {
  public autoRegisteredCommands: string[] = []

  protected readonly extensionConfig: Config

  constructor(
    config: Config,
    emitters?: {
      processStarted: EventEmitter<CliStarted>
      processCompleted: EventEmitter<CliResult>
    }
  ) {
    super(emitters)

    this.extensionConfig = config
  }

  protected executeDvcProcess(cwd: string, ...args: Args): Promise<string> {
    const options = this.getOptions(cwd, ...args)
    return this.executeProcess(options)
  }

  protected createBackgroundDvcProcess(
    cwd: string,
    ...args: Args
  ): Promise<string> {
    const options = this.getOptions(cwd, ...args)
    return this.createBackgroundProcess(options)
  }

  protected getOptions(cwd: string, ...args: Args) {
    return getOptions({
      PYTHONPATH: this.extensionConfig.getPYTHONPATH(),
      args: [...args],
      cliPath: this.extensionConfig.getCliPath(),
      cwd,
      pythonBinPath: this.extensionConfig.getPythonBinPath()
    })
  }
}