Merge pull request #824 from ianhi/fix-discard-all

check for changed files when discarding all changes

Author: fcollonval

src/model.ts

@@ -383,21 +383,25 @@ export class GitExtension implements IGitExtension {
     const data = await this._taskHandler.execute<Git.ICheckoutResult>(
       'git:checkout',
       async () => {
-        const d = await requestAPI<Git.ICheckoutResult>(
-          'checkout',
-          'POST',
-          body
-        );
-
+        let changes;
         if (body.checkout_branch) {
-          const changes = await this._changedFiles(
+          changes = await this._changedFiles(
             this._currentBranch.name,
             body.branchname
           );
-          changes.files?.forEach(file => this._revertFile(file));
+        } else if (body.filename) {
+          changes = { files: [body.filename] };
         } else {
-          this._revertFile(options.filename);
+          changes = await this._changedFiles('WORKING', 'HEAD');
         }
+
+        const d = await requestAPI<Git.ICheckoutResult>(
+          'checkout',
+          'POST',
+          body
+        );
+
+        changes.files?.forEach(file => this._revertFile(file));
         return d;
       }
     );