# HG changeset patch
# User Martin von Zweigbergk <martinvonz@google.com>
# Date 1576197121 28800
# Node ID 9f5e94bbc6064d73cc80edab8576eb1effc71ec8
# Parent  245aec57d76a6b731c008aaf8a54e0188239a001
fix: move handling of --all into getrevstofix() for consistency

Differential Revision: https://phab.mercurial-scm.org/D8286

diff -r 245aec57d76a -r 9f5e94bbc606 hgext/fix.py
--- a/hgext/fix.py	Wed Mar 18 14:26:47 2020 +0100
+++ b/hgext/fix.py	Thu Dec 12 16:32:01 2019 -0800
@@ -251,9 +251,7 @@
     opts = pycompat.byteskwargs(opts)
     cmdutil.check_at_most_one_arg(opts, b'all', b'rev')
     cmdutil.check_incompatible_arguments(opts, b'working_dir', [b'all'])
-    if opts[b'all']:
-        opts[b'rev'] = [b'not public() and not obsolete()']
-        opts[b'working_dir'] = True
+
     with repo.wlock(), repo.lock(), repo.transaction(b'fix'):
         revstofix = getrevstofix(ui, repo, opts)
         basectxs = getbasectxs(repo, opts, revstofix)
@@ -399,9 +397,12 @@
 
 def getrevstofix(ui, repo, opts):
     """Returns the set of revision numbers that should be fixed"""
-    revs = set(scmutil.revrange(repo, opts[b'rev']))
-    if opts.get(b'working_dir'):
-        revs.add(wdirrev)
+    if opts[b'all']:
+        revs = repo.revs(b'(not public() and not obsolete()) or wdir()')
+    else:
+        revs = set(scmutil.revrange(repo, opts[b'rev']))
+        if opts.get(b'working_dir'):
+            revs.add(wdirrev)
     for rev in revs:
         checkfixablectx(ui, repo, repo[rev])
     # Allow fixing only wdir() even if there's an unfinished operation