Worked out the fiddly

Author: blopker

codebook/src/lib.rs

@@ -249,6 +249,7 @@ impl CodeDictionary {
                 let current_line = node_start.row as u32;
                 let current_column = node_start.column as u32;
                 let words = self.get_words_from_text(node_text);
+                println!("Found Capture:: {node_text:?}");
                 println!("Words:: {words:?}");
                 println!("Column: {current_column}");
                 println!("Line: {current_line}");
@@ -257,7 +258,8 @@ impl CodeDictionary {
                     println!("Checking: {:?}", split);
                     for split_word in split {
                         if !self.check(&split_word.word) {
-                            let base_start_char = text_start_char + current_column;
+                            let offset = if text_line == 0 { current_column } else { 0 };
+                            let base_start_char = text_start_char + offset;
                             let location = TextRange {
                                 start_char: base_start_char + split_word.start_char,
                                 end_char: base_start_char

codebook/src/queries.rs

@@ -11,6 +11,7 @@ pub static LANGUAGE_SETTINGS: [LanguageSetting; 7] = [
                     pattern: (identifier) @identifier)
                 (let_declaration
                     pattern: (identifier) @identifier)
+                (line_comment) @comment
                 (string_content) @string
                 (char_literal) @string
                 "#,

codebook/tests/test_go.rs

@@ -5,16 +5,16 @@ mod utils;
 #[test]
 fn test_go_location() {
     let sample_text = r#"
-        func mispeledFuncion() string {
-            return ""
-        }"#;
+    func mispeledFuncion() string {
+        return ""
+    }"#;
     let expected = vec![
         SpellCheckResult::new(
             "mispeled".to_string(),
             vec!["misspelled", "dispelled", "misspell", "misled"],
             vec![TextRange {
-                start_char: 13,
-                end_char: 21,
+                start_char: 9,
+                end_char: 17,
                 start_line: 1,
                 end_line: 1,
             }],
@@ -23,8 +23,8 @@ fn test_go_location() {
             "Funcion".to_string(),
             vec!["Function"],
             vec![TextRange {
-                start_char: 21,
-                end_char: 28,
+                start_char: 17,
+                end_char: 24,
                 start_line: 1,
                 end_line: 1,
             }],

codebook/tests/test_python.rs

@@ -6,13 +6,13 @@ mod utils;
 fn test_python_simple() {
     let processor = utils::get_processor();
     let sample_text = r#"
-        def calculat_user_age(bithDate: str) -> int:
+        def calculat_user_age(bithDate) -> int:
             # This is an examle_function that calculates age
             usrAge = get_curent_date() - bithDate
             userAge
     "#;
-    let expected = vec!["bith", "calculat", "examle", "usr"];
-    let binding = processor.spell_check(sample_text, "rust").to_vec();
+    let expected = vec!["bith", "calculat", "curent", "examle", "usr"];
+    let binding = processor.spell_check(sample_text, "python").to_vec();
     let mut misspelled = binding
         .iter()
         .map(|r| r.word.as_str())
@@ -45,10 +45,10 @@ multi_line_comment = '''
             "linet".to_string(),
             vec![],
             vec![TextRange {
-                start_char: 57,
-                end_char: 62,
-                start_line: 2,
-                end_line: 2,
+                start_char: 12,
+                end_char: 17,
+                start_line: 3,
+                end_line: 3,
             }],
         ),
     ];

codebook/tests/test_rust.rs

@@ -1,7 +1,7 @@
 use codebook::{SpellCheckResult, TextRange};
 
 mod utils;
-
+// im a bd speler
 #[test]
 fn test_rust_simple() {
     let processor = utils::get_processor();