Add severity ordering and filtering controls

Author: Poryaei

README.md

@@ -29,6 +29,7 @@ This project focuses on exactly that.
 - Validate multiple files or directories in one command
 - Emit SARIF output for CI pipelines, review bots, and code scanning tools
 - Audit large script corpora and generate JSON and Markdown reports
+- Control which severities are emitted with professional CLI toggles
 
 ## Best Fit Use Cases
 
@@ -78,6 +79,18 @@ Emit SARIF for CI or review tooling:
 pine-validator path\to\pine-scripts --sarif
 ```
 
+Show only errors and warnings:
+
+```powershell
+pine-validator path\to\script.pine --no-hints --no-information
+```
+
+Show only hints during cleanup:
+
+```powershell
+pine-validator path\to\script.pine --no-errors --no-warnings --hints
+```
+
 Validate from stdin:
 
 ```powershell
@@ -154,6 +167,12 @@ Output shape:
 - `--agent-json` for richer debugging output intended for agents and automation
 - directory and multi-file batch validation
 - `--sarif` for CI, code scanning, and external review integrations
+- `--errors/--no-errors`
+- `--warnings/--no-warnings`
+- `--information/--no-information`
+- `--hints/--no-hints`
+
+Diagnostics are ordered by severity first, then by source location. By default the CLI prints errors first, then warnings, then information messages, and finally hints.
 
 Exit codes:
 

src/pinescript_validator/cli.py

@@ -6,7 +6,7 @@
 import sys
 from pathlib import Path
 
-from .diagnostics import Severity
+from .diagnostics import Diagnostic, Severity
 from .sarif import build_sarif_run
 from .validator import PineScriptValidator
 
@@ -25,6 +25,30 @@ def build_parser() -> argparse.ArgumentParser:
         action="store_true",
         help="Print SARIF output for CI, code scanning, and machine-readable review tooling.",
     )
+    parser.add_argument(
+        "--errors",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+        help="Include or exclude error diagnostics.",
+    )
+    parser.add_argument(
+        "--warnings",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+        help="Include or exclude warning diagnostics.",
+    )
+    parser.add_argument(
+        "--information",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+        help="Include or exclude information diagnostics.",
+    )
+    parser.add_argument(
+        "--hints",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+        help="Include or exclude hint diagnostics.",
+    )
     return parser
 
 
@@ -64,6 +88,33 @@ def _validate_file_targets(validator: PineScriptValidator, paths: list[Path]) ->
     return results
 
 
+def _selected_severities(args: argparse.Namespace) -> set[Severity]:
+    selected: set[Severity] = set()
+    if args.errors:
+        selected.add(Severity.ERROR)
+    if args.warnings:
+        selected.add(Severity.WARNING)
+    if args.information:
+        selected.add(Severity.INFORMATION)
+    if args.hints:
+        selected.add(Severity.HINT)
+    return selected
+
+
+def _filter_results(results: list[dict[str, object]], selected: set[Severity]) -> list[dict[str, object]]:
+    filtered: list[dict[str, object]] = []
+    for item in results:
+        diagnostics = [diagnostic for diagnostic in item["diagnostics"] if diagnostic.severity in selected]
+        filtered.append(
+            {
+                "path": item["path"],
+                "text": item["text"],
+                "diagnostics": diagnostics,
+            }
+        )
+    return filtered
+
+
 def _summary(results: list[dict[str, object]]) -> dict[str, int]:
     diagnostics = [diagnostic for item in results for diagnostic in item["diagnostics"]]
     return {
@@ -81,6 +132,9 @@ def main(argv: list[str] | None = None) -> int:
     validator = PineScriptValidator()
     if sum(1 for flag in (args.json, args.agent_json, args.sarif) if flag) > 1:
         raise SystemExit("Choose only one of --json, --agent-json, or --sarif.")
+    selected_severities = _selected_severities(args)
+    if not selected_severities:
+        raise SystemExit("At least one diagnostic severity must remain enabled.")
 
     if "-" in args.paths:
         if len(args.paths) != 1:
@@ -94,9 +148,10 @@ def main(argv: list[str] | None = None) -> int:
         if not file_paths:
             raise SystemExit("No .pine files matched the provided targets.")
         results = _validate_file_targets(validator, file_paths)
-        file_path = results[0]["path"] if len(results) == 1 else None
-        text = results[0]["text"] if len(results) == 1 else ""
-        diagnostics = results[0]["diagnostics"] if len(results) == 1 else []
+    results = _filter_results(results, selected_severities)
+    file_path = results[0]["path"] if len(results) == 1 else None
+    text = results[0]["text"] if len(results) == 1 else ""
+    diagnostics = results[0]["diagnostics"] if len(results) == 1 else []
 
     if args.sarif:
         print(json.dumps(build_sarif_run(results), ensure_ascii=False, indent=2))

src/pinescript_validator/validator.py

@@ -78,5 +78,5 @@ def _dedupe_and_sort(diagnostics: list[Diagnostic]) -> list[Diagnostic]:
                 continue
             seen.add(key)
             output.append(diagnostic)
-        output.sort(key=lambda item: (item.line, item.column, item.severity, item.message))
+        output.sort(key=lambda item: (item.severity, item.line, item.column, item.message))
         return output

tests/test_validator.py

@@ -1,13 +1,14 @@
 from __future__ import annotations
 
 import unittest
+from argparse import Namespace
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
+from pinescript_validator.cli import _expand_paths, _filter_results, _selected_severities
 from pinescript_validator.diagnostics import Severity
 from pinescript_validator.sarif import build_sarif_run
 from pinescript_validator.validator import PineScriptValidator
-from pinescript_validator.cli import _expand_paths
 
 
 class ValidatorTests(unittest.TestCase):
@@ -479,6 +480,29 @@ def test_build_sarif_run_emits_results(self) -> None:
         self.assertEqual(len(sarif["runs"]), 1)
         self.assertGreaterEqual(len(sarif["runs"][0]["results"]), 1)
 
+    def test_diagnostics_are_sorted_by_severity_before_location(self) -> None:
+        diagnostics = self.validator.validate_text('indicator("X")\nvalue = close\nplot(close, invalid_param=true)')
+
+        severities = [diagnostic.severity for diagnostic in diagnostics]
+        self.assertEqual(severities, sorted(severities))
+
+    def test_selected_severities_respects_cli_toggles(self) -> None:
+        selected = _selected_severities(
+            Namespace(errors=True, warnings=False, information=False, hints=True)
+        )
+
+        self.assertEqual(selected, {Severity.ERROR, Severity.HINT})
+
+    def test_filter_results_removes_disabled_severities(self) -> None:
+        diagnostics = self.validator.validate_text('indicator("X")\nvalue = close\nplot(close, invalid_param=true)')
+        filtered = _filter_results(
+            [{"path": Path("sample.pine"), "text": "", "diagnostics": diagnostics}],
+            {Severity.ERROR},
+        )
+
+        self.assertTrue(filtered[0]["diagnostics"])
+        self.assertTrue(all(item.severity == Severity.ERROR for item in filtered[0]["diagnostics"]))
+
 
 if __name__ == "__main__":
     unittest.main()