diff --git a/README.md b/README.md index b177151..06e1dad 100644 --- a/README.md +++ b/README.md @@ -64,7 +64,7 @@ This will output the contents of every file, with each file preceded by its rela files-to-prompt path/to/directory --cxml ``` -- `--markdown`: Output as Markdown with fenced code blocks. +- `-m/--markdown`: Output as Markdown with fenced code blocks. ```bash files-to-prompt path/to/directory --markdown diff --git a/files_to_prompt/cli.py b/files_to_prompt/cli.py index 37c408e..7eee04f 100644 --- a/files_to_prompt/cli.py +++ b/files_to_prompt/cli.py @@ -225,6 +225,8 @@ def read_paths_from_stdin(use_null_separator): help="Output in XML-ish format suitable for Claude's long context window.", ) @click.option( + "markdown", + "-m", "--markdown", is_flag=True, help="Output Markdown with fenced code blocks", diff --git a/tests/test_files_to_prompt.py b/tests/test_files_to_prompt.py index cd0581a..eecfc5d 100644 --- a/tests/test_files_to_prompt.py +++ b/tests/test_files_to_prompt.py @@ -377,7 +377,8 @@ def test_paths_from_arguments_and_stdin(tmpdir): assert "Contents of file2" in result.output -def test_markdown(tmpdir): +@pytest.mark.parametrize("option", ("-m", "--markdown")) +def test_markdown(tmpdir, option): runner = CliRunner() with tmpdir.as_cwd(): os.makedirs("test_dir") @@ -389,7 +390,7 @@ def test_markdown(tmpdir): f.write("This is javascript") with open("test_dir/code.unknown", "w") as f: f.write("This is an unknown file type") - result = runner.invoke(cli, ["test_dir", "--markdown"]) + result = runner.invoke(cli, ["test_dir", option]) assert result.exit_code == 0 actual = result.output expected = (