Merge pull request #16 from dyka3773/develop

feat: Add essential CLI enhancements (--version and --quiet flags)

Author: dyka3773

README.md

@@ -92,6 +92,15 @@ mvn-tree-visualizer --filename "maven_dependency_file" --output "dependencies.js
 mvn-tree-visualizer --filename "maven_dependency_file" --output "diagram.html" --watch
 ```
 
+#### Quiet Mode (For Automation/Scripts)
+```bash
+# Only show errors, suppress success messages
+mvn-tree-visualizer --filename "maven_dependency_file" --output "diagram.html" --quiet
+
+# Short form also available
+mvn-tree-visualizer --filename "maven_dependency_file" --output "diagram.html" -q
+```
+
 > **💡 Tip:** In watch mode, the tool will monitor for changes to your Maven dependency files and automatically regenerate the diagram. Perfect for development workflows! Press `Ctrl+C` to stop watching.
 
 ### Step 3: View the output
@@ -123,6 +132,8 @@ Each example includes:
 | `--watch` | Watch for file changes and auto-regenerate diagram | `False` |
 | `--directory` | The directory to scan for the Maven dependency file(s) | current directory |
 | `--keep-tree` | Keep the intermediate `dependency_tree.txt` file | `False` |
+| `--quiet`, `-q` | Suppress all console output except errors | `False` |
+| `--version`, `-v` | Show the current version and exit | - |
 | `--help` | Show the help message and exit | - |
 
 ### Theme Options

pyproject.toml

@@ -48,6 +48,7 @@ dev = [
     "build>=1.0.0",
     "git-changelog>=2.5.3",
     "tox>=4.28.4",
+    "pytest>=8.4.1",
 ]
 
 [build-system]

src/mvn_tree_visualizer/cli.py

@@ -1,5 +1,8 @@
 import argparse
+import sys
 import time
+import traceback
+from importlib import metadata
 from pathlib import Path
 from typing import NoReturn
 
@@ -12,6 +15,14 @@
 from .validation import find_dependency_files, validate_dependency_files, validate_output_directory
 
 
+def get_version() -> str:
+    """Get the current version of the package."""
+    try:
+        return metadata.version("mvn-tree-visualizer")
+    except metadata.PackageNotFoundError:
+        return "unknown"
+
+
 def generate_diagram(
     directory: str,
     output_file: str,
@@ -20,6 +31,7 @@ def generate_diagram(
     output_format: str,
     show_versions: bool,
     theme: str = "minimal",
+    quiet: bool = False,
 ) -> None:
     """Generate the dependency diagram with comprehensive error handling."""
     timestamp = time.strftime("%H:%M:%S")
@@ -31,7 +43,7 @@ def generate_diagram(
 
         # Show what files we found
         dependency_files = find_dependency_files(directory, filename)
-        if len(dependency_files) > 1:
+        if len(dependency_files) > 1 and not quiet:
             print(f"[{timestamp}] Found {len(dependency_files)} dependency files")
 
         # Setup paths
@@ -103,31 +115,42 @@ def generate_diagram(
         except Exception as e:
             raise OutputGenerationError(f"Error generating {output_format.upper()} output: {e}")
 
-        print(f"[{timestamp}] ✅ Diagram generated and saved to {output_file}")
+        if not quiet:
+            print(f"[{timestamp}] SUCCESS: Diagram generated and saved to {output_file}")
 
     except MvnTreeVisualizerError as e:
         # Our custom errors already have helpful messages
-        print(f"[{timestamp}] ❌ Error: {e}")
+        print(f"[{timestamp}] ERROR: {e}", file=sys.stderr)
+        raise  # Re-raise the exception for the caller to handle
     except KeyboardInterrupt:
-        print(f"\n[{timestamp}] ⏹️  Operation cancelled by user")
+        print(f"\n[{timestamp}] Operation cancelled by user", file=sys.stderr)
+        raise  # Re-raise for the caller to handle
     except Exception as e:
         # Unexpected errors
-        print(f"[{timestamp}] ❌ Unexpected error: {e}")
-        print("This is an internal error. Please report this issue with the following details:")
-        print(f"  - Directory: {directory}")
-        print(f"  - Filename: {filename}")
-        print(f"  - Output: {output_file}")
-        print(f"  - Format: {output_format}")
-        import traceback
+        print(f"[{timestamp}] UNEXPECTED ERROR: {e}", file=sys.stderr)
+        print("This is an internal error. Please report this issue with the following details:", file=sys.stderr)
+        print(f"  - Directory: {directory}", file=sys.stderr)
+        print(f"  - Filename: {filename}", file=sys.stderr)
+        print(f"  - Output: {output_file}", file=sys.stderr)
+        print(f"  - Format: {output_format}", file=sys.stderr)
 
         traceback.print_exc()
+        raise  # Re-raise for the caller to handle
 
 
 def cli() -> NoReturn:
     parser = argparse.ArgumentParser(
         prog="mvn-tree-visualizer",
         description="Generate a dependency diagram from a file.",
     )
+
+    parser.add_argument(
+        "-v",
+        "--version",
+        action="version",
+        version=f"mvn-tree-visualizer {get_version()}",
+    )
+
     parser.add_argument(
         "directory",
         type=str,
@@ -186,6 +209,13 @@ def cli() -> NoReturn:
         help="Theme for the diagram visualization. Default is 'minimal'.",
     )
 
+    parser.add_argument(
+        "-q",
+        "--quiet",
+        action="store_true",
+        help="Suppress all console output except errors. Perfect for CI/CD pipelines and scripted usage.",
+    )
+
     args = parser.parse_args()
     directory: str = args.directory
     output_file: str = args.output
@@ -195,28 +225,46 @@ def cli() -> NoReturn:
     show_versions: bool = args.show_versions
     watch_mode: bool = args.watch
     theme: str = args.theme
+    quiet: bool = args.quiet
 
     # Generate initial diagram
-    print("Generating initial diagram...")
-    generate_diagram(directory, output_file, filename, keep_tree, output_format, show_versions, theme)
+    if not quiet:
+        timestamp = time.strftime("%Y-%m-%d %H:%M:%S")
+        print(f"[{timestamp}] Generating initial diagram...")
+
+    try:
+        generate_diagram(directory, output_file, filename, keep_tree, output_format, show_versions, theme, quiet)
+    except MvnTreeVisualizerError:
+        sys.exit(1)
+    except KeyboardInterrupt:
+        sys.exit(130)  # Standard exit code for SIGINT
+    except Exception:
+        sys.exit(1)
 
     if not watch_mode:
-        print("You can open it in your browser to view the dependency tree.")
-        print("Thank you for using mvn-tree-visualizer!")
+        if not quiet:
+            print("You can open it in your browser to view the dependency tree.")
+            print("Thank you for using mvn-tree-visualizer!")
         return
 
     # Watch mode
     def regenerate_callback():
         """Callback function for file watcher."""
-        generate_diagram(directory, output_file, filename, keep_tree, output_format, show_versions, theme)
+        try:
+            generate_diagram(directory, output_file, filename, keep_tree, output_format, show_versions, theme, quiet)
+        except Exception:
+            # In watch mode, we don't want to exit on errors, just log them
+            print("Error during diagram regeneration:", file=sys.stderr)
+            traceback.print_exc()
 
     watcher = FileWatcher(directory, filename, regenerate_callback)
     watcher.start()
 
     try:
         watcher.wait()
     finally:
-        print("Thank you for using mvn-tree-visualizer!")
+        if not quiet:
+            print("Thank you for using mvn-tree-visualizer!")
 
 
 if __name__ == "__main__":