Added Take Theirs and Keep Yours commands to UI

User Story #884949

package.json

@@ -77,7 +77,7 @@
         },
         {
           "command": "tfvc.Undo",
-          "when": "scmProvider == tfvc",
+          "when": "scmProvider == tfvc && scmResourceGroup != conflicts",
           "group": "1_modification@1"
         },
         {
@@ -90,9 +90,19 @@
           "when": "scmProvider == tfvc && scmResourceGroup == included",
           "group": "1_modification@2"
         },
+        {
+          "command": "tfvc.ResolveTakeTheirs",
+          "when": "scmProvider == tfvc && scmResourceGroup == conflicts",
+          "group": "1_modification@1"
+        },
+        {
+          "command": "tfvc.ResolveKeepYours",
+          "when": "scmProvider == tfvc && scmResourceGroup == conflicts",
+          "group": "1_modification@2"
+        },
         {
           "command": "tfvc.Undo",
-          "when": "scmProvider == tfvc",
+          "when": "scmProvider == tfvc && scmResourceGroup != conflicts",
           "group": "inline@1"
         },
         {
@@ -104,6 +114,16 @@
           "command": "tfvc.Exclude",
           "when": "scmProvider == tfvc && scmResourceGroup == included",
           "group": "inline@2"
+        },
+        {
+          "command": "tfvc.ResolveTakeTheirs",
+          "when": "scmProvider == tfvc && scmResourceGroup == conflicts",
+          "group": "inline@1"
+        },
+        {
+          "command": "tfvc.ResolveKeepYours",
+          "when": "scmProvider == tfvc && scmResourceGroup == conflicts",
+          "group": "inline@2"
         }
       ]
     },
@@ -275,6 +295,24 @@
           "dark": "resources/icons/dark/refresh.svg"
         }
       },
+      {
+        "command": "tfvc.ResolveKeepYours",
+        "title": "Resove: Keep Yours",
+        "category": "TFVC",
+        "icon": {
+          "light": "resources/icons/light/resolve-keepyours.svg",
+          "dark": "resources/icons/dark/resolve-keepyours.svg"
+        }
+      },
+      {
+        "command": "tfvc.ResolveTakeTheirs",
+        "title": "Resolve: Take Theirs",
+        "category": "TFVC",
+        "icon": {
+          "light": "resources/icons/light/resolve-taketheirs.svg",
+          "dark": "resources/icons/dark/resolve-taketheirs.svg"
+        }
+      },
       {
         "command": "tfvc.Sync",
         "title": "Sync",

src/extension.ts

@@ -8,6 +8,7 @@ import { commands, Disposable, ExtensionContext } from "vscode";
 import { CommandNames, TfvcCommandNames } from "./helpers/constants";
 import { ExtensionManager } from "./extensionmanager";
 import { TfvcSCMProvider } from "./tfvc/tfvcscmprovider";
+import { AutoResolveType } from "./tfvc/interfaces";
 
 let _extensionManager: ExtensionManager;
 let _scmProvider: TfvcSCMProvider;
@@ -62,6 +63,12 @@ export async function activate(context: ExtensionContext) {
         }));
     context.subscriptions.push(commands.registerCommand(TfvcCommandNames.Refresh, () => _extensionManager.Tfvc.TfvcRefresh()));
     context.subscriptions.push(commands.registerCommand(TfvcCommandNames.ShowOutput, () => _extensionManager.Tfvc.TfvcShowOutput()));
+    context.subscriptions.push(commands.registerCommand(TfvcCommandNames.ResolveKeepYours, (...args) => {
+            _extensionManager.Tfvc.TfvcResolve(args ? args[0] : undefined, AutoResolveType.KeepYours);
+        }));
+    context.subscriptions.push(commands.registerCommand(TfvcCommandNames.ResolveTakeTheirs, (...args) => {
+            _extensionManager.Tfvc.TfvcResolve(args ? args[0] : undefined, AutoResolveType.TakeTheirs);
+        }));
     context.subscriptions.push(commands.registerCommand(TfvcCommandNames.Checkin, () => _extensionManager.Tfvc.TfvcCheckin()));
     context.subscriptions.push(commands.registerCommand(TfvcCommandNames.Sync, () => _extensionManager.Tfvc.TfvcSync()));
 }

src/helpers/constants.ts

@@ -44,6 +44,8 @@ export class TfvcCommandNames {
     static OpenDiff: string = TfvcCommandNames.CommandPrefix + "OpenDiff";
     static OpenFile: string = TfvcCommandNames.CommandPrefix + "OpenFile";
     static Refresh: string = TfvcCommandNames.CommandPrefix + "Refresh";
+    static ResolveKeepYours: string = TfvcCommandNames.CommandPrefix + "ResolveKeepYours";
+    static ResolveTakeTheirs: string = TfvcCommandNames.CommandPrefix + "ResolveTakeTheirs";
     static ShowOutput: string = TfvcCommandNames.CommandPrefix + "ShowOutput";
     static Status: string = TfvcCommandNames.CommandPrefix + "Status";
     static Sync: string = TfvcCommandNames.CommandPrefix + "Sync";

src/helpers/strings.ts

@@ -67,6 +67,7 @@ export class Strings {
     static UndoChanges: string = "Undo Changes";
     static NoChangesToCheckin: string = "There are no changes to checkin. Changes must be added to the 'Included' section to be checked in.";
     static AllFilesUpToDate: string = "All files are up to date.";
+    static CommandRequiresFileContext: string = "This command requires a file context and can only be executed from the TFVC viewlet window.";
 
     // TFVC viewlet Strings
     static ExcludedGroupName: string = "Excluded changes";
@@ -85,5 +86,14 @@ export class Strings {
     static ConflictAlreadyDeleted: string = "ALREADY DELETED";
     static ConflictAlreadyExists: string = "ALREADY EXISTS";
     static ConflictDeletedLocally: string = "DELETED LOCALLY";
+
+    // TFVC AutoResolveType Strings
+    static AutoResolveTypeAutoMerge: string = "Auto Merge";
+    static AutoResolveTypeDeleteConflict: string = "Delete Conflict";
+    static AutoResolveTypeKeepYours: string = "Keep Yours";
+    static AutoResolveTypeKeepYoursRenameTheirs: string = "Keep Yours Rename Theirs";
+    static AutoResolveTypeOverwriteLocal: string = "Overwrite Local";
+    static AutoResolveTypeTakeTheirs: string = "Take Theirs";
+
 }
 /* tslint:enable:variable-name */

src/tfvc/repository.ts

@@ -8,14 +8,15 @@ import { TeamServerContext} from "../contexts/servercontext";
 import { Logger } from "../helpers/logger";
 import { ITfvcCommand, IExecutionResult } from "./interfaces";
 import { Tfvc } from "./tfvc";
-import { IArgumentProvider, IConflict, IItemInfo, IPendingChange, ISyncResults, IWorkspace } from "./interfaces";
+import { AutoResolveType, IArgumentProvider, IConflict, IItemInfo, IPendingChange, ISyncResults, IWorkspace } from "./interfaces";
 import { Add } from "./commands/add";
 import { Checkin } from "./commands/checkin";
 import { FindConflicts } from "./commands/findconflicts";
 import { FindWorkspace } from "./commands/findworkspace";
 import { GetInfo } from "./commands/getinfo";
 import { GetFileContent } from "./commands/getfilecontent";
 import { GetVersion } from "./commands/getversion";
+import { ResolveConflicts } from "./commands/resolveconflicts";
 import { Status } from "./commands/status";
 import { Sync } from "./commands/sync";
 import { Undo } from "./commands/undo";
@@ -96,6 +97,12 @@ export class Repository {
             new Status(this._serverContext, ignoreFiles === undefined ? true : ignoreFiles));
     }
 
+    public async ResolveConflicts(itemPaths: string[], autoResolveType: AutoResolveType): Promise<IConflict[]> {
+        Logger.LogDebug(`TFVC Repository.ResolveConflicts`);
+        return this.RunCommand<IConflict[]>(
+            new ResolveConflicts(this._serverContext, itemPaths, autoResolveType));
+    }
+
     public async Sync(itemPaths: string[], recursive: boolean): Promise<ISyncResults> {
         Logger.LogDebug(`TFVC Repository.Sync`);
         return this.RunCommand<ISyncResults>(

src/tfvc/tfvc-extension.ts

@@ -19,7 +19,7 @@ import { TfvcSCMProvider } from "./tfvcscmprovider";
 import { TfvcErrorCodes } from "./tfvcerror";
 import { Repository } from "./repository";
 import { UIHelper } from "./uihelper";
-import { ICheckinInfo, IItemInfo, IPendingChange, ISyncResults } from "./interfaces";
+import { AutoResolveType, ICheckinInfo, IItemInfo, IPendingChange, ISyncResults } from "./interfaces";
 import { TfvcOutput } from "./tfvcoutput";
 
 export class TfvcExtension  {
@@ -134,6 +134,23 @@ export class TfvcExtension  {
         }
     }
 
+    public async TfvcResolve(uri: Uri, autoResolveType: AutoResolveType): Promise<void> {
+        this.displayErrors(async () => {
+            if (uri) {
+                let localPath: string = TfvcSCMProvider.GetPathFromUri(uri);
+                const resolveTypeString: string = UIHelper.GetDisplayTextForAutoResolveType(autoResolveType);
+                const basename: string = path.basename(localPath);
+                const message: string = `Are you sure you want to resolve changes in ${basename} as ${resolveTypeString}?`;
+                if (await UIHelper.PromptForConfirmation(message, resolveTypeString)) {
+                    await this._repo.ResolveConflicts([localPath], autoResolveType);
+                    TfvcSCMProvider.Refresh();
+                }
+            } else {
+                this._manager.DisplayWarningMessage(Strings.CommandRequiresFileContext);
+            }
+        });
+    }
+
     public async TfvcShowOutput(): Promise<void> {
         TfvcOutput.Show();
     }
@@ -202,14 +219,10 @@ export class TfvcExtension  {
             if (pathToUndo) {
                 const basename: string = path.basename(pathToUndo);
                 const message: string = `Are you sure you want to undo changes in ${basename}?`;
-                //TODO: use Modal api once vscode.d.ts exposes it (currently proposed)
-                const pick: string = await window.showWarningMessage(message, /*{ modal: true },*/ Strings.UndoChanges);
-                if (pick !== Strings.UndoChanges) {
-                    return;
+                if (await UIHelper.PromptForConfirmation(message, Strings.UndoChanges)) {
+                    this._manager.Telemetry.SendEvent(TfvcTelemetryEvents.Undo);
+                    await this._repo.Undo([pathToUndo]);
                 }
-
-                this._manager.Telemetry.SendEvent(TfvcTelemetryEvents.Undo);
-                await this._repo.Undo([pathToUndo]);
             }
         } catch (err) {
             this._manager.DisplayErrorMessage(err.message);
@@ -261,6 +274,19 @@ export class TfvcExtension  {
         }
     }
 
+    private async displayErrors(funcToTry: () => Promise<void>): Promise<void> {
+        if (!this._manager.EnsureInitialized(RepositoryType.TFVC)) {
+            this._manager.DisplayErrorMessage();
+            return;
+        }
+
+        try {
+            await funcToTry();
+        } catch (err) {
+            this._manager.DisplayErrorMessage(err.message);
+        }
+    }
+
     public async InitializeClients(repoType: RepositoryType): Promise<void> {
         //We only need to initialize for Tfvc repositories
         if (repoType !== RepositoryType.TFVC) {

src/tfvc/uihelper.ts

@@ -6,7 +6,7 @@
 
 import { QuickPickItem, window, workspace } from "vscode";
 import { Strings } from "../helpers/strings";
-import { IPendingChange, ISyncResults, ISyncItemResult, SyncType } from "./interfaces";
+import { AutoResolveType, IPendingChange, ISyncResults, ISyncItemResult, SyncType } from "./interfaces";
 import { TfvcOutput } from "./tfvcoutput";
 
 var path = require("path");
@@ -106,6 +106,18 @@ export class UIHelper {
         }
     }
 
+    public static GetDisplayTextForAutoResolveType(type: AutoResolveType): string {
+        switch (type) {
+            case AutoResolveType.AutoMerge: return Strings.AutoResolveTypeAutoMerge;
+            case AutoResolveType.DeleteConflict: return Strings.AutoResolveTypeDeleteConflict;
+            case AutoResolveType.KeepYours: return Strings.AutoResolveTypeKeepYours;
+            case AutoResolveType.KeepYoursRenameTheirs: return Strings.AutoResolveTypeKeepYoursRenameTheirs;
+            case AutoResolveType.OverwriteLocal: return Strings.AutoResolveTypeOverwriteLocal;
+            case AutoResolveType.TakeTheirs: return Strings.AutoResolveTypeTakeTheirs;
+            default: return Strings.AutoResolveTypeAutoMerge;
+        }
+    }
+
     public static GetFileName(change: IPendingChange) {
         if (change && change.localItem) {
             var filename = path.parse(change.localItem).base;
@@ -122,4 +134,11 @@ export class UIHelper {
 
         return change.localItem;
     }
+
+    public static async PromptForConfirmation(message: string, okButtonText?: string): Promise<boolean> {
+        okButtonText = okButtonText ? okButtonText : "OK";
+        //TODO: use Modal api once vscode.d.ts exposes it (currently proposed)
+        const pick: string = await window.showWarningMessage(message, /*{ modal: true },*/ okButtonText);
+        return pick === okButtonText;
+    }
 }