refactor: add helper

Author: psteinroe

crates/pgt_workspace/src/workspace/server/change.rs

@@ -249,19 +249,7 @@ impl Document {
 
         // if within a statement, we can modify it if the change results in also a single statement
         if affected_indices.len() == 1 {
-            let start_byte = new_content
-                .char_indices()
-                .nth(usize::from(affected_range.start()))
-                .map(|(i, _)| i)
-                .unwrap_or(new_content.len());
-
-            let end_byte = new_content
-                .char_indices()
-                .nth(usize::from(affected_range.end()))
-                .map(|(i, _)| i)
-                .unwrap_or(new_content.len());
-
-            let changed_content = &new_content[start_byte..end_byte];
+            let changed_content = get_affected(&new_content, affected_range);
 
             let (new_ranges, diags) =
                 document::split_with_diagnostics(changed_content, Some(affected_range.start()));
@@ -314,19 +302,7 @@ impl Document {
         }
 
         // in any other case, parse the full affected range
-        let start_byte = new_content
-            .char_indices()
-            .nth(usize::from(full_affected_range.start()))
-            .map(|(i, _)| i)
-            .unwrap_or(new_content.len());
-
-        let end_byte = new_content
-            .char_indices()
-            .nth(usize::from(full_affected_range.end()))
-            .map(|(i, _)| i)
-            .unwrap_or(new_content.len());
-
-        let changed_content = &new_content[start_byte..end_byte];
+        let changed_content = get_affected(&new_content, full_affected_range);
 
         let (new_ranges, diags) =
             document::split_with_diagnostics(changed_content, Some(full_affected_range.start()));
@@ -430,6 +406,22 @@ impl ChangeParams {
     }
 }
 
+fn get_affected(content: &str, range: TextRange) -> &str {
+    let start_byte = content
+        .char_indices()
+        .nth(usize::from(range.start()))
+        .map(|(i, _)| i)
+        .unwrap_or(content.len());
+
+    let end_byte = content
+        .char_indices()
+        .nth(usize::from(range.end()))
+        .map(|(i, _)| i)
+        .unwrap_or(content.len());
+
+    &content[start_byte..end_byte]
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;