feat: add context to linter rules

Author: tobymao

sqlmesh/core/context.py

@@ -614,7 +614,7 @@ def load(self, update_schemas: bool = True) -> GenericContext[C]:
 
             config = loader.config
             self._linters[config.project] = Linter.from_rules(
-                BUILTIN_RULES.union(project.user_rules), config.linter
+                self, BUILTIN_RULES.union(project.user_rules), config.linter
             )
 
         uncached = set()

sqlmesh/core/linter/definition.py

@@ -11,6 +11,9 @@
 from sqlmesh.core.linter.rule import Rule, RuleViolation
 from sqlmesh.core.console import LinterConsole, get_console
 
+if t.TYPE_CHECKING:
+    from sqlmesh.core.context import GenericContext
+
 
 def select_rules(all_rules: RuleSet, rule_names: t.Set[str]) -> RuleSet:
     if "all" in rule_names:
@@ -28,15 +31,23 @@ def select_rules(all_rules: RuleSet, rule_names: t.Set[str]) -> RuleSet:
 
 class Linter:
     def __init__(
-        self, enabled: bool, all_rules: RuleSet, rules: RuleSet, warn_rules: RuleSet
+        self,
+        context: GenericContext,
+        enabled: bool,
+        all_rules: RuleSet,
+        rules: RuleSet,
+        warn_rules: RuleSet,
     ) -> None:
+        self.context = context
         self.enabled = enabled
         self.all_rules = all_rules
         self.rules = rules
         self.warn_rules = warn_rules
 
     @classmethod
-    def from_rules(cls, all_rules: RuleSet, config: LinterConfig) -> Linter:
+    def from_rules(
+        cls, context: GenericContext, all_rules: RuleSet, config: LinterConfig
+    ) -> Linter:
         ignored_rules = select_rules(all_rules, config.ignored_rules)
         included_rules = all_rules.difference(ignored_rules)
 
@@ -49,7 +60,7 @@ def from_rules(cls, all_rules: RuleSet, config: LinterConfig) -> Linter:
                 f"Rules cannot simultaneously warn and raise an error: [{overlapping_rules}]"
             )
 
-        return Linter(config.enabled, all_rules, rules, warn_rules)
+        return Linter(context, config.enabled, all_rules, rules, warn_rules)
 
     def lint_model(
         self, model: Model, console: LinterConsole = get_console()