feat: add entry-points for tools

grizzly/__main__.py

@@ -1,11 +1,10 @@
 # This Source Code Form is subject to the terms of the Mozilla Public
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
-from .args import GrizzlyArgs
 from .main import main
 
 __author__ = "Tyson Smith"
 __credits__ = ["Tyson Smith", "Jesse Schwartzentruber"]
 
 
-raise SystemExit(main(GrizzlyArgs().parse_args()))
+raise SystemExit(main())

grizzly/main.py

@@ -10,6 +10,7 @@
 from sapphire import CertificateBundle, Sapphire
 
 from .adapter import Adapter
+from .args import GrizzlyArgs
 from .common.plugins import load_plugin
 from .common.reporter import (
     FailedLaunchReporter,
@@ -37,8 +38,18 @@
 LOG = getLogger(__name__)
 
 
-def main(args: Namespace) -> int:
+def main(args: Namespace | None = None) -> int:
+    """CLI for `grizzly`.
+
+    Arguments:
+        args: Result from `GrizzlyArgs.parse_args`.
+
+    Returns:
+        Exit.SUCCESS (0) for success otherwise a different Exit code is returned.
+    """
+    args = args or GrizzlyArgs().parse_args()
     configure_logging(args.log_level)
+
     LOG.info("Starting Grizzly (%d)", getpid())
     LOG.debug("grizzly-framework version: %s", package_version("grizzly-framework"))
 

grizzly/reduce/__main__.py

@@ -4,6 +4,5 @@
 """Grizzly reducer CLI.
 """
 from . import ReduceManager
-from .args import ReduceArgs
 
-raise SystemExit(ReduceManager.main(ReduceArgs().parse_args()))
+raise SystemExit(ReduceManager.main())

grizzly/reduce/core.py

@@ -39,6 +39,7 @@
 )
 from ..replay import ReplayManager, ReplayResult
 from ..target import AssetManager, Target, TargetLaunchError, TargetLaunchTimeout
+from .args import ReduceArgs
 from .exceptions import GrizzlyReduceBaseException, NotReproducible
 from .strategies import STRATEGIES
 
@@ -752,16 +753,17 @@ def report(
             self._status.last_reports = new_reports
 
     @classmethod
-    def main(cls, args: Namespace) -> int:
+    def main(cls, args: Namespace | None = None) -> int:
         """CLI for `grizzly.reduce`.
 
         Arguments:
             args: Result from `ReduceArgs.parse_args`.
 
         Returns:
-            0 for success. non-0 indicates a problem.
+            Exit.SUCCESS (0) for success otherwise a different Exit code is returned.
         """
         # pylint: disable=too-many-return-statements
+        args = args or ReduceArgs().parse_args()
         configure_logging(args.log_level)
         setlocale(LC_ALL, "")
 

grizzly/reduce/crash.py

@@ -14,15 +14,16 @@
 LOG = getLogger(__name__)
 
 
-def main(args: Namespace) -> int:
+def main(args: Namespace | None = None) -> int:
     """CLI for `grizzly.reduce.crash`.
 
     Arguments:
         args: Result from `ReduceArgs.parse_args`.
 
     Returns:
-        0 for success. non-0 indicates a problem.
+        Exit.SUCCESS (0) for success otherwise a different Exit code is returned.
     """
+    args = args or ReduceFuzzManagerIDArgs().parse_args()
     configure_logging(args.log_level)
     with load_fm_data(args.input, load_bucket=not args.sig) as (crash, bucket):
         LOG.info(
@@ -61,4 +62,4 @@ def main(args: Namespace) -> int:
 
 
 if __name__ == "__main__":
-    raise SystemExit(main(ReduceFuzzManagerIDArgs().parse_args()))
+    raise SystemExit(main())

grizzly/replay/__main__.py

@@ -1,7 +1,6 @@
 # This Source Code Form is subject to the terms of the Mozilla Public
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
-from .args import ReplayArgs
 from .replay import ReplayManager
 
-raise SystemExit(ReplayManager.main(ReplayArgs().parse_args()))
+raise SystemExit(ReplayManager.main())

grizzly/replay/bugzilla.py

@@ -5,27 +5,28 @@
 from logging import getLogger
 
 from ..common.bugzilla import BugzillaBug
-from ..common.utils import configure_logging
+from ..common.utils import Exit, configure_logging
 from .args import ReplayFuzzBugzillaArgs
 from .replay import ReplayManager
 
 LOG = getLogger(__name__)
 
 
-def main(args: Namespace) -> int:
+def main(args: Namespace | None = None) -> int:
     """CLI for `grizzly.replay.bugzilla`.
 
     Arguments:
         args: Result from `ReplayArgs.parse_args`.
 
     Returns:
-        0 for success. non-0 indicates a problem.
+        Exit.SUCCESS (0) for success otherwise a different Exit code is returned.
     """
+    args = args or ReplayFuzzBugzillaArgs().parse_args()
     configure_logging(args.log_level)
     bug = BugzillaBug.load(args.input)
     if bug is None:
         LOG.info("Failed to load Bug %d from Bugzilla", args.input)
-        return 1
+        return Exit.ERROR
     LOG.info("Loaded Bug %d from Bugzilla", args.input)
     with bug:
         args.asset.extend(
@@ -35,10 +36,10 @@ def main(args: Namespace) -> int:
         testcases = bug.testcases()
         if not testcases:
             LOG.error("No test case data attached to bug %d", args.input)
-            return 1
+            return Exit.ERROR
         args.input = testcases
         return ReplayManager.main(args)
 
 
 if __name__ == "__main__":
-    raise SystemExit(main(ReplayFuzzBugzillaArgs().parse_args()))
+    raise SystemExit(main())

grizzly/replay/crash.py

@@ -18,15 +18,16 @@
 LOG = getLogger(__name__)
 
 
-def main(args: Namespace) -> int:
+def main(args: Namespace | None = None) -> int:
     """CLI for `grizzly.replay.crash`.
 
     Arguments:
         args: Result from `ReplayArgs.parse_args`.
 
     Returns:
-        0 for success. non-0 indicates a problem.
+        Exit.SUCCESS (0) for success otherwise a different Exit code is returned.
     """
+    args = args or ReplayFuzzManagerIDArgs().parse_args()
     configure_logging(args.log_level)
     with load_fm_data(args.input, load_bucket=not args.sig) as (crash, bucket):
         LOG.info("Loaded crash %d from FuzzManager", crash.crash_id)
@@ -77,4 +78,4 @@ def modify_args(args: Namespace, crash: CrashEntry, bucket: Bucket | None) -> Na
 
 
 if __name__ == "__main__":
-    raise SystemExit(main(ReplayFuzzManagerIDArgs().parse_args()))
+    raise SystemExit(main())

grizzly/replay/replay.py

@@ -40,6 +40,7 @@
     TargetLaunchError,
     TargetLaunchTimeout,
 )
+from .args import ReplayArgs
 
 if TYPE_CHECKING:
     from argparse import Namespace
@@ -600,8 +601,18 @@ def harness_fn(_: str) -> bytes:  # pragma: no cover
                 result.report.cleanup()
 
     @classmethod
-    def main(cls, args: Namespace) -> int:
+    def main(cls, args: Namespace | None = None) -> int:
+        """CLI for `grizzly.reduce`.
+
+        Arguments:
+            args: Result from `ReplayArgs.parse_args`.
+
+        Returns:
+            Exit.SUCCESS (0) for success otherwise a different Exit code is returned.
+        """
+        args = args or ReplayArgs().parse_args()
         configure_logging(args.log_level)
+
         LOG.info("Starting Grizzly Replay")
         LOG.debug("grizzly-framework version: %s", package_version("grizzly-framework"))
 

grizzly/test_main.py

@@ -2,7 +2,6 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 """test Grizzly main"""
-from pathlib import Path
 from platform import system
 
 from pytest import mark, skip
@@ -86,7 +85,7 @@ def test_main_02(mocker, session_setup, exit_code, to_raise):
         session_obj.run.side_effect = TargetLaunchError("test", mocker.Mock())
     else:
         session_obj.run.side_effect = to_raise()
-    args = mocker.MagicMock(adapter="fake", binary=Path("bin"), time_limit=1, timeout=1)
+    args = mocker.MagicMock(adapter="fake", time_limit=1, timeout=1)
     assert main(args) == exit_code
     assert target_cls.return_value.cleanup.call_count == 1
 
@@ -111,7 +110,6 @@ def test_main_03(mocker, session_setup, test_limit, timeout):
     # no_harness=False for code coverage
     args = mocker.MagicMock(
         adapter="fake",
-        binary=Path("bin"),
         no_harness=False,
         time_limit=test_limit,
         timeout=timeout,
@@ -142,7 +140,6 @@ def test_main_04(
     assert sum((pernosco, rr, valgrind)) < 2, "test broken!"
     args = mocker.MagicMock(
         adapter="fake",
-        binary=Path("bin"),
         pernosco=pernosco,
         rr=rr,
         time_limit=1,
@@ -162,7 +159,6 @@ def test_main_05(mocker, session_setup):
     target_cls.return_value.https.return_value = False
     args = mocker.MagicMock(
         adapter="fake",
-        binary=Path("bin"),
         use_http=False,
         time_limit=1,
         timeout=1,

setup.cfg

@@ -44,7 +44,13 @@ zip_safe = False
 
 [options.entry_points]
 console_scripts =
-    grizzly.status = grizzly.common.status_reporter:main
+    grizzly = grizzly:__main__
+    grizzly-reduce = grizzly.reduce:__main__
+    grizzly-reduce-crash = grizzly.reduce.crash:__main__
+    grizzly-replay = grizzly.replay:__main__
+    grizzly-replay-bugzilla = grizzly.replay.bugzilla:__main__
+    grizzly-replay-crash = grizzly.replay.crash:__main__
+    grizzly-status = grizzly.common.status_reporter:__main__
 grizzly_adapters =
     no-op = grizzly.adapter.no_op_adapter:NoOpAdapter
 grizzly_targets =