feat: add code to dianostic

Author: benfdking

crates/cli/tests/json/hql_file.stdout

@@ -1 +1 @@
-{"tests/lint/hql_file.hql":[{"range":{"start":{"line":1,"character":7},"end":{"line":1,"character":7}},"message":"Expected only single space before \"1\". Found \"   \".","severity":"Error","source":"sqruff"},{"range":{"start":{"line":1,"character":11},"end":{"line":1,"character":11}},"message":"Files must end with a single trailing newline.","severity":"Error","source":"sqruff"}]}
+{"tests/lint/hql_file.hql":[{"range":{"start":{"line":1,"character":7},"end":{"line":1,"character":7}},"message":"Expected only single space before \"1\". Found \"   \".","severity":"Error","source":"sqruff","code":"LT01"},{"range":{"start":{"line":1,"character":11},"end":{"line":1,"character":11}},"message":"Files must end with a single trailing newline.","severity":"Error","source":"sqruff","code":"LT12"}]}

crates/cli/tests/json/test_fail_whitespace_before_comma.stdout

@@ -1 +1 @@
-{"tests/lint/test_fail_whitespace_before_comma.sql":[{"range":{"start":{"line":1,"character":8},"end":{"line":1,"character":8}},"message":"Column expression without alias. Use explicit `AS` clause.","severity":"Error","source":"sqruff"},{"range":{"start":{"line":1,"character":11},"end":{"line":1,"character":11}},"message":"Column expression without alias. Use explicit `AS` clause.","severity":"Error","source":"sqruff"},{"range":{"start":{"line":1,"character":9},"end":{"line":1,"character":9}},"message":"Unexpected whitespace before comma.","severity":"Error","source":"sqruff"},{"range":{"start":{"line":1,"character":11},"end":{"line":1,"character":11}},"message":"Expected single whitespace between \",\" and \"4\".","severity":"Error","source":"sqruff"},{"range":{"start":{"line":1,"character":12},"end":{"line":1,"character":12}},"message":"Files must end with a single trailing newline.","severity":"Error","source":"sqruff"}]}
+{"tests/lint/test_fail_whitespace_before_comma.sql":[{"range":{"start":{"line":1,"character":8},"end":{"line":1,"character":8}},"message":"Column expression without alias. Use explicit `AS` clause.","severity":"Error","source":"sqruff","code":"AL03"},{"range":{"start":{"line":1,"character":11},"end":{"line":1,"character":11}},"message":"Column expression without alias. Use explicit `AS` clause.","severity":"Error","source":"sqruff","code":"AL03"},{"range":{"start":{"line":1,"character":9},"end":{"line":1,"character":9}},"message":"Unexpected whitespace before comma.","severity":"Error","source":"sqruff","code":"LT01"},{"range":{"start":{"line":1,"character":11},"end":{"line":1,"character":11}},"message":"Expected single whitespace between \",\" and \"4\".","severity":"Error","source":"sqruff","code":"LT01"},{"range":{"start":{"line":1,"character":12},"end":{"line":1,"character":12}},"message":"Files must end with a single trailing newline.","severity":"Error","source":"sqruff","code":"LT12"}]}

crates/lib/src/cli/json_types.rs

@@ -5,6 +5,7 @@ use sqruff_lib_core::errors::SQLBaseError;
 
 impl From<SQLBaseError> for Diagnostic {
     fn from(value: SQLBaseError) -> Self {
+        let code = value.rule.map(|rule| rule.code.to_string());
         Diagnostic {
             range: Range {
                 start: Position::new(value.line_no as u32, value.line_pos as u32),
@@ -17,6 +18,7 @@ impl From<SQLBaseError> for Diagnostic {
                 DiagnosticSeverity::Error
             },
             source: Some("sqruff".to_string()),
+            code,
             // code: todo!(),
             // source: Some(value.get_source().to_string()),
             // code: Some(DiagnosticCode {
@@ -65,15 +67,24 @@ pub struct Diagnostic {
     pub severity: DiagnosticSeverity,
     /// A human-readable string describing the source of this diagnostic, e.g. 'typescript' or 'super lint'.
     source: Option<String>,
-    // A code or identifier for this diagnostic. Should be used for later processing, e.g. when providing {@link CodeActionContext code actions}.
-    // code: Option<DiagnosticCode>,
+    // The diagnostic's code, which might appear in the user interface.
+    code: Option<String>,
+    // An optional property to describe the error code.
+    // code_description: Option<CodeDescription>,
     // TODO Maybe implement
     // An array of related diagnostic information, e.g. when symbol-names within a scope collide all definitions can be marked via this property.
     // related_information: Vec<DiagnosticRelatedInformation>,
     // Additional metadata about the diagnostic.
     // tags: Vec<DiagnosticTag>,
 }
 
+// Structure to capture a description for an error code.
+// #[derive(Serialize)]
+// pub struct CodeDescription {
+//     /// An URI to open with more information about the diagnostic error.
+//     href: String,
+// }
+
 // /// Represents a related message and source code location for a diagnostic. This should be used to point to code locations that cause or are related to a diagnostics, e.g when duplicating a symbol in a scope.
 // #[derive(Serialize)]
 // struct DiagnosticCode {