Adding option to give an initial pattern

Makefile

@@ -1,13 +1,13 @@
 BUMP_PART ?=
-
+PYTEST_COV ?= xml
 setup:
 	python3 -m pip install poetry
 
 install:
 	poetry install
 
 test:
-	poetry run pytest -ssv --cov --cov-report=xml
+	poetry run pytest -ssv --cov --cov-report=$(PYTEST_COV)
 
 build:
 	poetry build

rexi/cli.py

@@ -15,10 +15,19 @@ def rexi_cli(
     input_file: Annotated[
         Optional[typer.FileText],
         typer.Option(
-            "--input", "-i",
+            "--input",
+            "-i",
             help="Input file to pass to rexi; if not provided, stdin will be used.",
         ),
     ] = None,
+    initial_pattern: Annotated[
+        Optional[str],
+        typer.Option(
+            "--pattern",
+            "-p",
+            help="Initial regex pattern to use",
+        ),
+    ] = None,
 ) -> None:
     if input_file:
         input_text = input_file.read()
@@ -29,5 +38,5 @@ def rexi_cli(
         except OSError:
             pass
         sys.stdin = open("/dev/tty", "rb")  # type: ignore[assignment]
-    app: RexiApp[int] = RexiApp(input_text)
+    app: RexiApp[int] = RexiApp(input_text, initial_pattern=initial_pattern)
     app.run()

rexi/rexi.py

@@ -33,15 +33,21 @@ def __repr__(self) -> str:
 class RexiApp(App[ReturnType]):
     CSS_PATH = "rexi.tcss"
 
-    def __init__(self, input_content: str, start_mode: str = "finditer"):
+    def __init__(
+        self,
+        input_content: str,
+        start_mode: str = "finditer",
+        initial_pattern: Optional[str] = None,
+    ):
         super().__init__()
         self.input_content: str = input_content
         self.regex_modes: list[str] = ["finditer", "match"]
         self.regex_current_mode: str = start_mode
+        self.initial_pattern = initial_pattern
 
     def compose(self) -> ComposeResult:
         with Horizontal(id="inputs"):
-            yield Input(placeholder="Enter regex pattern")
+            yield Input(value=self.initial_pattern, placeholder="Enter regex pattern")
             yield Select(
                 zip(self.regex_modes, self.regex_modes),
                 id="select",

tests/test_cli.py

@@ -7,7 +7,7 @@
 from rexi.cli import app
 
 
-def test_stdin_cli(monkeypatch: MonkeyPatch) -> None:
+def test_on_args(monkeypatch: MonkeyPatch) -> None:
     """
     Couldn't find a better way to test the CLI without patching everything :(
     """
@@ -23,11 +23,11 @@ def test_stdin_cli(monkeypatch: MonkeyPatch) -> None:
         monkeypatch.setattr("builtins.open", open_mock)
         runner.invoke(app, input=a)
     open_mock.assert_called_once_with("/dev/tty", "rb")
-    class_mock.assert_called_once_with(text.decode())
+    class_mock.assert_called_once_with(text.decode(), initial_pattern=None)
     instance_mock.run.assert_called_once()
 
 
-def test_file_input_cli(monkeypatch: MonkeyPatch, tmp_path: Path) -> None:
+def test_file_input(monkeypatch: MonkeyPatch, tmp_path: Path) -> None:
     runner = CliRunner()
     text = "This iS! aTe xt2 F0r T3sT!ng"
     (tmp_path / "text_file").write_text(text)
@@ -37,4 +37,18 @@ def test_file_input_cli(monkeypatch: MonkeyPatch, tmp_path: Path) -> None:
         class_mock.return_value = instance_mock
         monkeypatch.setattr("rexi.cli.RexiApp", class_mock)
         runner.invoke(app, args=["-i", str(tmp_path / "text_file")])
-    class_mock.assert_called_once_with(text)
+    class_mock.assert_called_once_with(text, initial_pattern=None)
+
+
+def test_initial_pattern(monkeypatch: MonkeyPatch, tmp_path: Path) -> None:
+    runner = CliRunner()
+    text = "This iS! aTe xt2 F0r T3sT!ng"
+    (tmp_path / "text_file").write_text(text)
+    class_mock = Mock()
+    instance_mock = Mock()
+    with monkeypatch.context():
+        class_mock.return_value = instance_mock
+        monkeypatch.setattr("rexi.cli.RexiApp", class_mock)
+        runner.invoke(app, args=["-i", str(tmp_path / "text_file"), "--pattern", "wtf"])
+    class_mock.assert_called_once_with(text, initial_pattern="wtf")
+

tests/test_ui.py

@@ -36,6 +36,18 @@ async def test_input_box(start_mode: str, pattern: str, expected_output: str) ->
         assert result == expected_output
 
 
+async def test_input_box_with_initial_pattern() -> None:
+    app: RexiApp[int] = RexiApp(
+        "This iS! aTe xt2 F0r T3sT!ng", initial_pattern="(This.*iS!)"
+    )
+    async with app.run_test():
+        result = str(cast(Static, app.query_one("#output")).renderable)
+        assert (
+            result
+            == f"{UNDERLINE}{Fore.RED}This iS!{Fore.RESET}{RESET_UNDERLINE} aTe xt2 F0r T3sT!ng"
+        )
+
+
 async def test_switch_modes() -> None:
     app: RexiApp[int] = RexiApp("This iS! aTe xt2 F0r T3sT!ng")
     async with app.run_test() as pilot: