iterative/vscode-dvc

import { collectDvcRootPids } from './collect'
import { processExists, stopProcesses } from '../../process/execution'

export const stopWorkspaceExperiment = async (dvcRoot: string, id: string) => {
  const pids = new Set<number>()

  await collectDvcRootPids(pids, dvcRoot)

  const notFound = `process executing ${id} was not found.`

  if (pids.size === 0) {
    return notFound
  }

  const [pid] = [...pids]

  if (!pid || !(await processExists(pid))) {
    return notFound
  }

  const failedToStop = `failed to kill ${id}.`

  try {
    const stopped = await stopProcesses([pid])
    return stopped ? `${id} has been killed.` : failedToStop
  } catch {
    return failedToStop
  }
}