| b69ab31 | | | 1 | /** |
| b69ab31 | | | 2 | * Copyright (c) Meta Platforms, Inc. and affiliates. |
| b69ab31 | | | 3 | * |
| b69ab31 | | | 4 | * This source code is licensed under the MIT license found in the |
| b69ab31 | | | 5 | * LICENSE file in the root directory of this source tree. |
| b69ab31 | | | 6 | */ |
| b69ab31 | | | 7 | |
| b69ab31 | | | 8 | import type { |
| b69ab31 | | | 9 | ApplyUncommittedChangesPreviewsFuncType, |
| b69ab31 | | | 10 | Dag, |
| b69ab31 | | | 11 | UncommittedChangesPreviewContext, |
| b69ab31 | | | 12 | } from '../previews'; |
| b69ab31 | | | 13 | import type {ChangedFile, CommitInfo, UncommittedChanges} from '../types'; |
| b69ab31 | | | 14 | |
| b69ab31 | | | 15 | import {Operation} from './Operation'; |
| b69ab31 | | | 16 | |
| b69ab31 | | | 17 | export class UncommitOperation extends Operation { |
| b69ab31 | | | 18 | /** |
| b69ab31 | | | 19 | * @param originalDotCommit the current dot commit, needed to track when optimistic state is resolved |
| b69ab31 | | | 20 | * @param changedFiles the files that are in the commit to be uncommitted. Must be fetched before running, as the CommitInfo object itself does not have file statuses. |
| b69ab31 | | | 21 | */ |
| b69ab31 | | | 22 | constructor( |
| b69ab31 | | | 23 | private originalDotCommit: CommitInfo, |
| b69ab31 | | | 24 | private changedFiles: Array<ChangedFile>, |
| b69ab31 | | | 25 | ) { |
| b69ab31 | | | 26 | super('UncommitOperation'); |
| b69ab31 | | | 27 | } |
| b69ab31 | | | 28 | |
| b69ab31 | | | 29 | static opName = 'Uncommit'; |
| b69ab31 | | | 30 | |
| b69ab31 | | | 31 | getArgs() { |
| b69ab31 | | | 32 | const args = ['uncommit']; |
| b69ab31 | | | 33 | return args; |
| b69ab31 | | | 34 | } |
| b69ab31 | | | 35 | |
| b69ab31 | | | 36 | optimisticDag(dag: Dag): Dag { |
| b69ab31 | | | 37 | const {hash, parents} = this.originalDotCommit; |
| b69ab31 | | | 38 | const p1 = parents.at(0); |
| b69ab31 | | | 39 | const commitHasChildren = (dag.children(hash)?.size ?? 0) > 0; |
| b69ab31 | | | 40 | if ( |
| b69ab31 | | | 41 | p1 == null || commitHasChildren |
| b69ab31 | | | 42 | ? // If the commit has children, then we know the uncommit is done when it's no longer the dot commit |
| b69ab31 | | | 43 | dag.get(hash)?.isDot !== true |
| b69ab31 | | | 44 | : // If the commit does not have children, if `hash` disappears and `p1` still exists, then uncommit is completed. |
| b69ab31 | | | 45 | dag.get(hash) == null || dag.get(p1) == null |
| b69ab31 | | | 46 | ) { |
| b69ab31 | | | 47 | return dag; |
| b69ab31 | | | 48 | } |
| b69ab31 | | | 49 | return commitHasChildren |
| b69ab31 | | | 50 | ? // Set `isDot` on `p1` and not `hash` |
| b69ab31 | | | 51 | dag.replaceWith([p1 as string, hash], (h, c) => { |
| b69ab31 | | | 52 | return c?.set('isDot', h === p1); |
| b69ab31 | | | 53 | }) |
| b69ab31 | | | 54 | : // Hide `hash` and set `isDot` on `p1`. |
| b69ab31 | | | 55 | dag.replaceWith([p1 as string, hash], (h, c) => { |
| b69ab31 | | | 56 | if (h === hash) { |
| b69ab31 | | | 57 | return undefined; |
| b69ab31 | | | 58 | } else { |
| b69ab31 | | | 59 | return c?.set('isDot', true); |
| b69ab31 | | | 60 | } |
| b69ab31 | | | 61 | }); |
| b69ab31 | | | 62 | } |
| b69ab31 | | | 63 | |
| b69ab31 | | | 64 | makeOptimisticUncommittedChangesApplier?( |
| b69ab31 | | | 65 | context: UncommittedChangesPreviewContext, |
| b69ab31 | | | 66 | ): ApplyUncommittedChangesPreviewsFuncType | undefined { |
| b69ab31 | | | 67 | const preexistingChanges = new Set(context.uncommittedChanges.map(change => change.path)); |
| b69ab31 | | | 68 | |
| b69ab31 | | | 69 | if (this.changedFiles.every(file => preexistingChanges.has(file.path))) { |
| b69ab31 | | | 70 | // once every file to uncommit appears in the output, the uncommit has reflected in the latest fetch. |
| b69ab31 | | | 71 | // TODO: we'll eventually limit how many uncommitted changes we pull in. When this happens, it's |
| b69ab31 | | | 72 | // possible the list of files won't include any of the changes being uncommitted (though this would be rare). |
| b69ab31 | | | 73 | // We should probably return undefined if the number of uncommitted changes >= max fetched. |
| b69ab31 | | | 74 | return undefined; |
| b69ab31 | | | 75 | } |
| b69ab31 | | | 76 | |
| b69ab31 | | | 77 | const func: ApplyUncommittedChangesPreviewsFuncType = (changes: UncommittedChanges) => { |
| b69ab31 | | | 78 | // You could have uncommitted changes before uncommitting, so we need to include |
| b69ab31 | | | 79 | // files from the commit AND the existing uncommitted changes. |
| b69ab31 | | | 80 | // But it's also possible to have changed a file changed by the commit, so we need to de-dupe. |
| b69ab31 | | | 81 | const newChanges = this.changedFiles.filter(file => !preexistingChanges.has(file.path)); |
| b69ab31 | | | 82 | return [...changes, ...newChanges]; |
| b69ab31 | | | 83 | }; |
| b69ab31 | | | 84 | return func; |
| b69ab31 | | | 85 | } |
| b69ab31 | | | 86 | } |