Fix mrcrowl#100: Add purge support

package.json

@@ -244,6 +244,23 @@
         "title": "%command.pull%",
         "category": "Hg"
       },
+      {
+        "command": "hg.purge",
+        "title": "%command.purge%",
+        "category": "Hg",
+        "icon": "$(discard)"
+      },
+      {
+        "command": "hg.purgeFiles",
+        "title": "%command.purgeFiles%",
+        "category": "Hg",
+        "icon": "$(discard)"
+      },
+      {
+        "command": "hg.purgeAll",
+        "title": "%command.purgeAll%",
+        "category": "Hg"
+      },
       {
         "command": "hg.push",
         "title": "%command.push%",
@@ -492,6 +509,18 @@
           "command": "hg.pull",
           "when": "config.hg.enabled && !hg.missing && hgOpenRepositoryCount != 0"
         },
+        {
+          "command": "hg.purge",
+          "when": "false"
+        },
+        {
+          "command": "hg.purgeFiles",
+          "when": "config.hg.enabled && !hg.missing && hgOpenRepositoryCount != 0"
+        },
+        {
+          "command": "hg.purgeAll",
+          "when": "config.hg.enabled && !hg.missing && hgOpenRepositoryCount != 0"
+        },
         {
           "command": "hg.push",
           "when": "config.hg.enabled && !hg.missing && hgOpenRepositoryCount != 0"
@@ -604,6 +633,11 @@
           "group": "4_stage",
           "when": "config.hg.enabled && !hg.missing && scmProvider == hg"
         },
+        {
+          "command": "hg.purgeAll",
+          "group": "4_stage",
+          "when": "config.hg.enabled && !hg.missing && scmProvider == hg"
+        },
         {
           "command": "hg.mergeHeads",
           "group": "5_merge",
@@ -663,6 +697,16 @@
           "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
           "group": "1_modification"
         },
+        {
+          "command": "hg.purgeFiles",
+          "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
+          "group": "1_modification"
+        },
+        {
+          "command": "hg.purgeFiles",
+          "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
+          "group": "inline1"
+        },
         {
           "command": "hg.addAll",
           "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
@@ -745,11 +789,21 @@
           "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == working",
           "group": "inline"
         },
+        {
+          "command": "hg.purge",
+          "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
+          "group": "inline1"
+        },
         {
           "command": "hg.add",
           "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
           "group": "inline"
         },
+        {
+          "command": "hg.purge",
+          "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",
+          "group": "1_modification"
+        },
         {
           "command": "hg.add",
           "when": "config.hg.enabled && !hg.missing && scmProvider == hg && scmResourceGroup == untracked",

package.nls.json

@@ -23,6 +23,9 @@
 	"command.update": "Update to...",
 	"command.branch": "Create Branch...",
 	"command.pull": "Pull",
+	"command.purge": "Purge",
+	"command.purgeFiles": "Purge All Untracked Files",
+	"command.purgeAll": "Purge All Untracked and Ignored Files",
 	"command.push": "Push",
 	"command.pushTo": "Push to...",
 	"command.add": "Add Files",

src/commands.ts

@@ -559,6 +559,39 @@ export class CommandCenter {
 		}
 	}
 
+	@command('hg.purgeFiles', { repository: true })
+	async purgeFiles(repository: Repository): Promise<void> {
+		return await repository.purge();
+	}
+
+	@command('hg.purgeAll', { repository: true })
+	async purgeAll(repository: Repository): Promise<void> {
+		return await repository.purgeAll();
+	}
+
+	@command('hg.purge')
+	async purge(...resourceStates: SourceControlResourceState[]): Promise<void> {
+		if (resourceStates.length === 0) {
+			const resource = this.getSCMResource();
+
+			if (!resource) {
+				return;
+			}
+
+			resourceStates = [resource];
+		}
+
+		const scmResources = resourceStates
+			.filter(s => s instanceof Resource && s.resourceGroup instanceof UntrackedGroup) as Resource[];
+
+		if (!scmResources.length) {
+			return;
+		}
+
+		const resources = scmResources.map(r => r.resourceUri);
+		await this.runByRepository(resources, async (repository, uris) => repository.purge(...uris));
+	}
+
 	private async smartCommit(repository: Repository, getCommitMessage: () => Promise<string | undefined>, opts: CommitOptions = {}): Promise<boolean> {
 		// validate no conflicts
 		const numConflictResources = repository.conflictGroup.resources.length;

src/hg.ts

@@ -58,6 +58,11 @@ export interface UnshelveOptions {
 	keep?: boolean;
 }
 
+export interface PurgeOptions {
+	paths?: string[];
+	all?: boolean;
+}
+
 export interface IMergeResult {
 	unresolvedCount: number;
 }
@@ -352,6 +357,7 @@ export const HgErrorCodes = {
 	CantAccessRemote: 'CantAccessRemote',
 	RepositoryNotFound: 'RepositoryNotFound',
 	NoSuchFile: 'NoSuchFile',
+	ExtensionMissing: 'ExtensionMissing',
 	BranchAlreadyExists: 'BranchAlreadyExists',
 	NoRollbackInformationAvailable: 'NoRollbackInformationAvailable',
 	UntrackedFilesDiffer: 'UntrackedFilesDiffer',
@@ -487,6 +493,10 @@ export class Hg {
 			else if (/no such file/.test(result.stderr)) {
 				hgErrorCode = HgErrorCodes.NoSuchFile;
 			}
+			else if (/unknown command/.test(result.stderr)) {
+				hgErrorCode = HgErrorCodes.ExtensionMissing;
+				result.stderr = result.stderr.replace(/.*(unknown command '.*?').*/s, '$1');
+			}
 
 			if (options.logErrors !== false && result.stderr) {
 				this.log(`${result.stderr}\n`);
@@ -883,6 +893,19 @@ export class Repository {
 		}
 	}
 
+	async purge(opts: PurgeOptions): Promise<void> {
+		const args = ['purge'];
+
+		if (opts.paths && opts.paths.length) {
+			args.push.apply(args, opts.paths);
+		}
+		if (opts.all) {
+			args.push('--all');
+		}
+
+		await this.run(args);
+	}
+
 	async unshelveAbort(): Promise<void> {
 		await this.run(['unshelve', '--abort']);
 	}

src/interaction.ts

@@ -216,6 +216,10 @@ export namespace interaction {
                 message = localize('unshelve in progress', "There is already an unshelve operation in progress.");
                 options.modal = false;
                 break;
+            case HgErrorCodes.ExtensionMissing:
+                message = localize('extension missing', "Extension missing: {0}", err.stderr);
+                options.modal = false;
+                break;
 
             default:
                 const hint = (err.stderr || err.message || String(err))

src/repository.ts

@@ -687,6 +687,23 @@ export class Repository implements IDisposable {
         });
     }
 
+    @throttle
+    async purge(...uris: Uri[]): Promise<void> {
+        let resources: Resource[];
+        if (uris.length === 0) {
+            resources = this._groups.untracked.resources;
+        } else {
+            resources = this.mapResources(uris);
+        }
+        const relativePaths: string[] = resources.map(r => this.mapResourceToRepoRelativePath(r));
+        await this.run(Operation.Clean, () => this.repository.purge({paths: relativePaths}));
+    }
+
+    @throttle
+    async purgeAll(): Promise<void> {
+        await this.run(Operation.Clean, () => this.repository.purge({all: true}));
+    }
+
     @throttle
     async branch(name: string, opts?: { allowBranchReuse: boolean }): Promise<void> {
         const hgOpts = opts && {