git-chopstick-core 0.1.2 → 0.1.3

package/src/git/status.ts

@@ -1,502 +0,0 @@
-import { getFilesWithConflictMarkers } from './diff-check.js'
-import {
-  WorkingDirectoryStatus,
-  WorkingDirectoryFileChange,
-  AppFileStatus,
-  FileEntry,
-  GitStatusEntry,
-  AppFileStatusKind,
-  UnmergedEntry,
-  ConflictedFileStatus,
-  UnmergedEntrySummary,
-} from '../models/status.js'
-import {
-  parsePorcelainStatus,
-  mapStatus,
-  IStatusEntry,
-  IStatusHeader,
-  isStatusHeader,
-  isStatusEntry,
-} from '../lib/status-parser.js'
-import { DiffSelectionType, DiffSelection } from '../models/diff/index.js'
-import { Repository } from '../models/repository.js'
-import { IAheadBehind } from '../models/branch.js'
-import { fatalError } from '../lib/fatal-error.js'
-import { isMergeHeadSet, isSquashMsgSet } from './merge.js'
-import { getBinaryPaths } from './diff.js'
-import { getRebaseInternalState } from './rebase.js'
-import { RebaseInternalState } from '../models/rebase.js'
-import { isCherryPickHeadFound } from './cherry-pick.js'
-import { git } from './index.js'
-
-/** The encapsulation of the result from 'git status' */
-export interface IStatusResult {
-  /** The name of the current branch */
-  readonly currentBranch?: string
-
-  /** The name of the current upstream branch */
-  readonly currentUpstreamBranch?: string
-
-  /** The SHA of the tip commit of the current branch */
-  readonly currentTip?: string
-
-  /** How many commits ahead and behind
-   *  the `currentBranch` is compared to the `currentUpstreamBranch`
-   */
-  readonly branchAheadBehind?: IAheadBehind
-
-  /** true if the repository exists at the given location */
-  readonly exists: boolean
-
-  /** true if repository is in a conflicted state */
-  readonly mergeHeadFound: boolean
-
-  /** true merge --squash operation started */
-  readonly squashMsgFound: boolean
-
-  /** details about the rebase operation, if found */
-  readonly rebaseInternalState: RebaseInternalState | null
-
-  /** true if repository is in cherry pick state */
-  readonly isCherryPickingHeadFound: boolean
-
-  /** the absolute path to the repository's working directory */
-  readonly workingDirectory: WorkingDirectoryStatus
-
-  /** whether conflicting files present on repository */
-  readonly doConflictedFilesExist: boolean
-}
-
-interface IStatusHeadersData {
-  currentBranch?: string
-  currentUpstreamBranch?: string
-  currentTip?: string
-  branchAheadBehind?: IAheadBehind
-  match: RegExpMatchArray | null
-}
-
-type ConflictFilesDetails = {
-  conflictCountsByPath: ReadonlyMap<string, number>
-  binaryFilePaths: ReadonlyArray<string>
-}
-
-function parseConflictedState(
-  entry: UnmergedEntry,
-  path: string,
-  conflictDetails: ConflictFilesDetails
-): ConflictedFileStatus {
-  switch (entry.action) {
-    case UnmergedEntrySummary.BothAdded: {
-      const isBinary = conflictDetails.binaryFilePaths.includes(path)
-      if (!isBinary) {
-        return {
-          kind: AppFileStatusKind.Conflicted,
-          entry,
-          conflictMarkerCount:
-            conflictDetails.conflictCountsByPath.get(path) || 0,
-        }
-      } else {
-        return {
-          kind: AppFileStatusKind.Conflicted,
-          entry,
-        }
-      }
-    }
-    case UnmergedEntrySummary.BothModified: {
-      const isBinary = conflictDetails.binaryFilePaths.includes(path)
-      if (!isBinary) {
-        return {
-          kind: AppFileStatusKind.Conflicted,
-          entry,
-          conflictMarkerCount:
-            conflictDetails.conflictCountsByPath.get(path) || 0,
-        }
-      } else {
-        return {
-          kind: AppFileStatusKind.Conflicted,
-          entry,
-        }
-      }
-    }
-    default:
-      return {
-        kind: AppFileStatusKind.Conflicted,
-        entry,
-      }
-  }
-}
-
-function convertToAppStatus(
-  path: string,
-  entry: FileEntry,
-  conflictDetails: ConflictFilesDetails,
-  oldPath?: string
-): AppFileStatus {
-  if (entry.kind === 'ordinary') {
-    switch (entry.type) {
-      case 'added':
-        return {
-          kind: AppFileStatusKind.New,
-          submoduleStatus: entry.submoduleStatus,
-        }
-      case 'modified':
-        return {
-          kind: AppFileStatusKind.Modified,
-          submoduleStatus: entry.submoduleStatus,
-        }
-      case 'deleted':
-        return {
-          kind: AppFileStatusKind.Deleted,
-          submoduleStatus: entry.submoduleStatus,
-        }
-    }
-  } else if (entry.kind === 'copied' && oldPath != null) {
-    return {
-      kind: AppFileStatusKind.Copied,
-      oldPath,
-      submoduleStatus: entry.submoduleStatus,
-      renameIncludesModifications: false,
-    }
-  } else if (entry.kind === 'renamed' && oldPath != null) {
-    return {
-      kind: AppFileStatusKind.Renamed,
-      oldPath,
-      submoduleStatus: entry.submoduleStatus,
-      renameIncludesModifications:
-        entry.workingTree === GitStatusEntry.Modified ||
-        (entry.renameOrCopyScore !== undefined &&
-          entry.renameOrCopyScore < 100),
-    }
-  } else if (entry.kind === 'untracked') {
-    return {
-      kind: AppFileStatusKind.Untracked,
-      submoduleStatus: entry.submoduleStatus,
-    }
-  } else if (entry.kind === 'conflicted') {
-    return parseConflictedState(entry, path, conflictDetails)
-  }
-
-  return fatalError(`Unknown file status ${status}`)
-}
-
-// List of known conflicted index entries for a file, extracted from mapStatus
-// inside `app/src/lib/status-parser.ts` for convenience
-const conflictStatusCodes = ['DD', 'AU', 'UD', 'UA', 'DU', 'AA', 'UU']
-
-/**
- *  Retrieve the status for a given repository,
- *  and fail gracefully if the location is not a Git repository
- */
-export async function getStatus(
-  repository: Repository
-): Promise<IStatusResult | null>
-export async function getStatus(
-  repository: Repository,
-  includeUntracked: boolean
-): Promise<IStatusResult | null>
-export async function getStatus(
-  repository: Repository,
-  includeUntracked: boolean,
-  rejectOnError: true
-): Promise<IStatusResult>
-export async function getStatus(
-  repository: Repository,
-  includeUntracked: boolean,
-  rejectOnError: false
-): Promise<IStatusResult | null>
-export async function getStatus(
-  repository: Repository,
-  includeUntracked = true,
-  rejectOnError = false
-): Promise<IStatusResult | null> {
-  const args = [
-    '--no-optional-locks',
-    'status',
-    ...(includeUntracked ? ['--untracked-files=all'] : []),
-    '--branch',
-    '--porcelain=2',
-    '-z',
-  ]
-
-  const { stdout, exitCode } = await git(args, repository.path, 'getStatus', {
-    successExitCodes: new Set(rejectOnError ? [0] : [0, 128]),
-    encoding: 'buffer',
-  })
-
-  if (exitCode === 128) {
-    console.debug(
-      `'git status' returned 128 for '${repository.path}' and is likely missing its .git directory`
-    )
-    return null
-  }
-
-  const parsed = parsePorcelainStatus(stdout)
-  const headers = parsed.filter(isStatusHeader)
-  const entries = parsed.filter(isStatusEntry)
-
-  const mergeHeadFound = await isMergeHeadSet(repository)
-  const conflictedFilesInIndex = entries.filter(e =>
-    conflictStatusCodes.includes(e.statusCode)
-  )
-  const rebaseInternalState = await getRebaseInternalState(repository)
-
-  const conflictDetails = await getConflictDetails(
-    repository,
-    mergeHeadFound,
-    conflictedFilesInIndex,
-    rebaseInternalState
-  )
-
-  // Map of files keyed on their paths.
-  const files = entries.reduce(
-    (files, entry) => buildStatusMap(files, entry, conflictDetails),
-    new Map<string, WorkingDirectoryFileChange>()
-  )
-
-  const {
-    currentBranch,
-    currentUpstreamBranch,
-    currentTip,
-    branchAheadBehind,
-  } = headers.reduce(parseStatusHeader, {
-    currentBranch: undefined,
-    currentUpstreamBranch: undefined,
-    currentTip: undefined,
-    branchAheadBehind: undefined,
-    match: null,
-  })
-
-  const workingDirectory = WorkingDirectoryStatus.fromFiles([...files.values()])
-
-  const isCherryPickingHeadFound = await isCherryPickHeadFound(repository)
-
-  const squashMsgFound = await isSquashMsgSet(repository)
-
-  return {
-    currentBranch,
-    currentTip,
-    currentUpstreamBranch,
-    branchAheadBehind,
-    exists: true,
-    mergeHeadFound,
-    rebaseInternalState,
-    workingDirectory,
-    isCherryPickingHeadFound,
-    squashMsgFound,
-    doConflictedFilesExist: conflictedFilesInIndex.length > 0,
-  }
-}
-
-/**
- *
- * Update map of working directory changes with a file status entry.
- * Reducer(ish).
- *
- * (Map is used here to maintain insertion order.)
- */
-function buildStatusMap(
-  files: Map<string, WorkingDirectoryFileChange>,
-  entry: IStatusEntry,
-  conflictDetails: ConflictFilesDetails
-): Map<string, WorkingDirectoryFileChange> {
-  const status = mapStatus(
-    entry.statusCode,
-    entry.submoduleStatusCode,
-    entry.renameOrCopyScore
-  )
-
-  if (status.kind === 'ordinary') {
-    // when a file is added in the index but then removed in the working
-    // directory, the file won't be part of the commit, so we can skip
-    // displaying this entry in the changes list
-    if (
-      status.index === GitStatusEntry.Added &&
-      status.workingTree === GitStatusEntry.Deleted
-    ) {
-      return files
-    }
-  }
-
-  if (status.kind === 'untracked') {
-    // when a delete has been staged, but an untracked file exists with the
-    // same path, we should ensure that we only draw one entry in the
-    // changes list - see if an entry already exists for this path and
-    // remove it if found
-    files.delete(entry.path)
-  }
-
-  // for now we just poke at the existing summary
-  const appStatus = convertToAppStatus(
-    entry.path,
-    status,
-    conflictDetails,
-    entry.oldPath
-  )
-
-  const initialSelectionType =
-    appStatus.kind === AppFileStatusKind.Modified &&
-    appStatus.submoduleStatus !== undefined &&
-    !appStatus.submoduleStatus.commitChanged
-      ? DiffSelectionType.None
-      : DiffSelectionType.All
-
-  const selection = DiffSelection.fromInitialSelection(initialSelectionType)
-
-  files.set(
-    entry.path,
-    new WorkingDirectoryFileChange(entry.path, appStatus, selection)
-  )
-  return files
-}
-
-/**
- * Update status header based on the current header entry.
- * Reducer.
- */
-function parseStatusHeader(results: IStatusHeadersData, header: IStatusHeader) {
-  let {
-    currentBranch,
-    currentUpstreamBranch,
-    currentTip,
-    branchAheadBehind,
-    match,
-  } = results
-  const value = header.value
-
-  // This intentionally does not match branch.oid initial
-  if ((match = value.match(/^branch\.oid ([a-f0-9]+)$/))) {
-    currentTip = match[1]
-  } else if ((match = value.match(/^branch.head (.*)/))) {
-    if (match[1] !== '(detached)') {
-      currentBranch = match[1]
-    }
-  } else if ((match = value.match(/^branch.upstream (.*)/))) {
-    currentUpstreamBranch = match[1]
-  } else if ((match = value.match(/^branch.ab \+(\d+) -(\d+)$/))) {
-    const ahead = parseInt(match[1], 10)
-    const behind = parseInt(match[2], 10)
-
-    if (!isNaN(ahead) && !isNaN(behind)) {
-      branchAheadBehind = { ahead, behind }
-    }
-  }
-  return {
-    currentBranch,
-    currentUpstreamBranch,
-    currentTip,
-    branchAheadBehind,
-    match,
-  }
-}
-
-async function getMergeConflictDetails(
-  repository: Repository,
-  conflictedFilesInIndex: ReadonlyArray<IStatusEntry>
-) {
-  const conflictCountsByPath = await getFilesWithConflictMarkers(
-    repository.path
-  )
-  const binaryFilePaths = await getBinaryPaths(
-    repository,
-    'MERGE_HEAD',
-    conflictedFilesInIndex
-  )
-  return {
-    conflictCountsByPath,
-    binaryFilePaths,
-  }
-}
-
-async function getRebaseConflictDetails(
-  repository: Repository,
-  conflictedFilesInIndex: ReadonlyArray<IStatusEntry>
-) {
-  const conflictCountsByPath = await getFilesWithConflictMarkers(
-    repository.path
-  )
-  const binaryFilePaths = await getBinaryPaths(
-    repository,
-    'REBASE_HEAD',
-    conflictedFilesInIndex
-  )
-  return {
-    conflictCountsByPath,
-    binaryFilePaths,
-  }
-}
-
-/**
- * We need to do these operations to detect conflicts that were the result
- * of popping a stash into the index
- */
-async function getWorkingDirectoryConflictDetails(
-  repository: Repository,
-  conflictedFilesInIndex: ReadonlyArray<IStatusEntry>
-) {
-  const conflictCountsByPath = await getFilesWithConflictMarkers(
-    repository.path
-  )
-  let binaryFilePaths: ReadonlyArray<string> = []
-  try {
-    // its totally fine if HEAD doesn't exist, which throws an error
-    binaryFilePaths = await getBinaryPaths(
-      repository,
-      'HEAD',
-      conflictedFilesInIndex
-    )
-  } catch (error) {}
-
-  return {
-    conflictCountsByPath,
-    binaryFilePaths,
-  }
-}
-
-/**
- * gets the conflicted files count and binary file paths in a given repository.
- * for computing an `IStatusResult`.
- *
- * @param repository to get details from
- * @param mergeHeadFound whether a merge conflict has been detected
- * @param conflictedFilesInIndex all files marked as being conflicted in the
- *                               index. Used to check for files using the binary
- *                               merge driver and whether it looks like a stash
- *                               has introduced conflicts
- * @param rebaseInternalState details about the current rebase operation (if
- * found)
- */
-async function getConflictDetails(
-  repository: Repository,
-  mergeHeadFound: boolean,
-  conflictedFilesInIndex: ReadonlyArray<IStatusEntry>,
-  rebaseInternalState: RebaseInternalState | null
-): Promise<ConflictFilesDetails> {
-  try {
-    if (mergeHeadFound) {
-      return await getMergeConflictDetails(repository, conflictedFilesInIndex)
-    }
-
-    if (rebaseInternalState !== null) {
-      return await getRebaseConflictDetails(repository, conflictedFilesInIndex)
-    }
-
-    // If there's conflicted files in the index but we don't have a merge head
-    // or a rebase internal state, then we're likely in a situation where a
-    // stash has introduced conflicts
-    if (conflictedFilesInIndex.length > 0) {
-      return await getWorkingDirectoryConflictDetails(
-        repository,
-        conflictedFilesInIndex
-      )
-    }
-  } catch (error) {
-    console.error(
-      'Unexpected error from git operations in getConflictDetails',
-      error
-    )
-  }
-  return {
-    conflictCountsByPath: new Map<string, number>(),
-    binaryFilePaths: new Array<string>(),
-  }
-}

package/src/git/submodule.ts

@@ -1,212 +0,0 @@
-import { git, IGitStringExecutionOptions } from './core.js'
-import { Repository } from '../models/repository.js'
-import { SubmoduleEntry } from '../models/submodule.js'
-import { pathExists } from '../lib/path-exists.js'
-import { executionOptionsWithProgress, IGitOutput } from '../lib/progress/index.js'
-import {
-  envForRemoteOperation,
-  getFallbackUrlForProxyResolve,
-} from './environment.js'
-import { AuthenticationErrors } from './authentication.js'
-import { IRemote } from '../models/remote.js'
-import { Progress } from '../models/progress.js'
-import { join, resolve } from 'path'
-import { readFile } from 'fs/promises'
-
-/**
- * Update submodules after a git operation.
- *
- * @param repository - The repository in which to update submodules
- * @param remote - The remote for environment setup (can be null)
- * @param progressCallback - An optional function which will be invoked
- *                           with information about the current progress
- *                           of the submodule update operation.
- * @param progressKind - The kind of progress event ('checkout', 'pull', etc.)
- * @param title - The title to use for progress reporting
- * @param targetOrRemote - The target (for checkout) or remote name (for pull)
- * @param allowFileProtocol - Whether to allow file:// protocol for submodules
- */
-export async function updateSubmodulesAfterOperation<T extends Progress>(
-  repository: Repository,
-  remote: IRemote | null,
-  progressCallback: ((progress: T) => void) | undefined,
-  progressKind: T['kind'],
-  title: string,
-  targetOrRemote: string,
-  allowFileProtocol: boolean
-): Promise<void> {
-  const opts: IGitStringExecutionOptions = {
-    env: await envForRemoteOperation(
-      getFallbackUrlForProxyResolve(repository, remote)
-    ),
-    expectedErrors: AuthenticationErrors,
-  }
-
-  const args = [
-    ...(allowFileProtocol ? ['-c', 'protocol.file.allow=always'] : []),
-    'submodule',
-    'update',
-    '--init',
-    '--recursive',
-  ]
-
-  if (!progressCallback) {
-    await git(args, repository.path, 'updateSubmodules', opts)
-    return
-  }
-
-  // Initial progress
-  progressCallback({
-    kind: progressKind,
-    title,
-    description: 'Updating submodules',
-    value: 0,
-    // Add the target or remote field based on the progress kind
-    ...(progressKind === 'checkout'
-      ? { target: targetOrRemote }
-      : { remote: targetOrRemote }),
-  } as T)
-
-  let submoduleEventCount = 0
-
-  const progressOpts = await executionOptionsWithProgress(
-    { ...opts, trackLFSProgress: true },
-    {
-      parse(line: string): IGitOutput {
-        if (
-          line.match(/^Submodule path (.)+?: checked out /) ||
-          line.startsWith('Cloning into ')
-        ) {
-          submoduleEventCount += 1
-        }
-
-        return {
-          kind: 'context',
-          text: `Updating submodules: ${line}`,
-          // Math taken from https://math.stackexchange.com/a/2323106
-          // We do this to fake a progress that slows down as we process more
-          // events, as we don't know how many submodules there are upfront, or
-          // what does git have to do with them (cloning, just checking them
-          // out...)
-          percent: 1 - Math.exp(-submoduleEventCount * 0.25),
-        }
-      },
-    },
-    progress => {
-      const description =
-        progress.kind === 'progress' ? progress.details.text : progress.text
-
-      const value = progress.percent
-
-      progressCallback({
-        kind: progressKind,
-        title,
-        description,
-        value,
-        ...(progressKind === 'checkout'
-          ? { target: targetOrRemote }
-          : { remote: targetOrRemote }),
-      } as T)
-    }
-  )
-
-  await git(args, repository.path, 'updateSubmodules', progressOpts)
-
-  // Final progress
-  progressCallback({
-    kind: progressKind,
-    title,
-    description: 'Submodules updated',
-    value: 1,
-    ...(progressKind === 'checkout'
-      ? { target: targetOrRemote }
-      : { remote: targetOrRemote }),
-  } as T)
-}
-
-export async function listSubmodules(
-  repository: Repository
-): Promise<ReadonlyArray<SubmoduleEntry>> {
-  const [submodulesFile, submodulesDir] = await Promise.all([
-    pathExists(join(repository.path, '.gitmodules')),
-    pathExists(join(repository.path, '.git', 'modules')),
-  ])
-
-  if (!submodulesFile && !submodulesDir) {
-    // repo path + .gitmodules and + .git/modules covers the vast majority of
-    // "normal" repositories but if we're in a linked worktree the modules
-    // directory is actually in the git common dir so we'll also check for the
-    // existence of the modules directory there as well before giving up on the
-    // existence of submodules in this repo. We're reading the commondir file
-    // ourselves here instead of calling out to git to avoid the cost of
-    // spawning a process on Windows
-    const commonDirPath = join(repository.resolvedGitDir, 'commondir')
-    const commonDir = await readFile(commonDirPath, 'utf8')
-      .then(content => content.replace(/\r?\n$/, ''))
-      .then(p => (p ? resolve(repository.resolvedGitDir, p) : null))
-      .catch(() => null)
-
-    if (!commonDir || !(await pathExists(join(commonDir, 'modules')))) {
-      console.info('No submodules found. Skipping "git submodule status"')
-      return []
-    }
-  }
-
-  // We don't recurse when listing submodules here because we don't have a good
-  // story about managing these currently. So for now we're only listing
-  // changes to the top-level submodules to be consistent with `git status`
-  const { stdout, exitCode } = await git(
-    ['submodule', 'status', '--'],
-    repository.path,
-    'listSubmodules',
-    { successExitCodes: new Set([0, 128]) }
-  )
-
-  if (exitCode === 128) {
-    // unable to parse submodules in repository, giving up
-    return []
-  }
-
-  const submodules = new Array<SubmoduleEntry>()
-
-  // entries are of the format:
-  //  1eaabe34fc6f486367a176207420378f587d3b48 git (v2.16.0-rc0)
-  //
-  // first character:
-  //   - " " if no change
-  //   - "-" if the submodule is not initialized
-  //   - "+" if the currently checked out submodule commit does not match the SHA-1 found in the index of the containing repository
-  //   - "U" if the submodule has merge conflicts
-  //
-  // then the 40-character SHA represents the current commit
-  //
-  // then the path to the submodule
-  //
-  // then the output of `git describe` for the submodule in braces
-  // we're not leveraging this in the app, so go and read the docs
-  // about it if you want to learn more:
-  //
-  // https://git-scm.com/docs/git-describe
-  const statusRe = /^.([^ ]+) (.+) \((.+?)\)$/gm
-
-  for (const [, sha, path, describe] of stdout.matchAll(statusRe)) {
-    submodules.push({ path, url: '', describe, sha, status: { commitChanged: false, modifiedChanges: false, untrackedChanges: false } })
-  }
-
-  return submodules
-}
-
-export async function resetSubmodulePaths(
-  repository: Repository,
-  paths: ReadonlyArray<string>
-): Promise<void> {
-  if (paths.length === 0) {
-    return
-  }
-
-  await git(
-    ['submodule', 'update', '--recursive', '--force', '--', ...paths],
-    repository.path,
-    'updateSubmodule'
-  )
-}