support maxspace

Author: CppCXY

crates/xmake_formatter/src/format/formatter_context.rs

@@ -22,4 +22,23 @@ impl FormatterContext {
             self.text.pop();
         }
     }
+
+    pub fn get_last_whitespace_count(&self) -> usize {
+        let mut count = 0;
+        for ch in self.text.chars().rev() {
+            if ch == ' ' {
+                count += 1;
+            } else {
+                break;
+            }
+        }
+        count
+    }
+
+    pub fn reset_whitespace_to(&mut self, n: usize) {
+        self.reset_whitespace();
+        if n > 0 {
+            self.text.push_str(&" ".repeat(n));
+        }
+    }
 }

crates/xmake_formatter/src/format/mod.rs

@@ -61,6 +61,15 @@ impl XmakeFormatter {
                             continue;
                         }
                     }
+                    (Some(TokenExpected::MaxSpace(n)), LuaTokenKind::TkWhitespace) => {
+                        if !context.is_line_first_token {
+                            let white_space_len = token.text().chars().count();
+                            if white_space_len > n {
+                                context.reset_whitespace_to(n);
+                                continue;
+                            }
+                        }
+                    }
                     (_, LuaTokenKind::TkEndOfLine) => {
                         // No space expected
                         context.reset_whitespace();
@@ -85,6 +94,14 @@ impl XmakeFormatter {
                                 context.text.push_str(&" ".repeat(*n));
                             }
                         }
+                        TokenExpected::MaxSpace(n) => {
+                            if !context.is_line_first_token {
+                                let current_spaces = context.get_last_whitespace_count();
+                                if current_spaces > *n {
+                                    context.reset_whitespace_to(*n);
+                                }
+                            }
+                        }
                     }
                 }
 

crates/xmake_formatter/src/format/syntax_node_change.rs

@@ -11,4 +11,5 @@ pub enum TokenNodeChange {
 #[derive(Debug, Clone, Copy)]
 pub enum TokenExpected {
     Space(usize),
+    MaxSpace(usize),
 }

crates/xmake_formatter/src/style_ruler/basic_space.rs

@@ -80,7 +80,16 @@ impl StyleRuler for BasicSpaceRuler {
                         f.add_token_left_expected(syntax_id, TokenExpected::Space(1));
                         f.add_token_right_expected(syntax_id, TokenExpected::Space(1));
                     }
-                    LuaTokenKind::TkColon | LuaTokenKind::TkDot => {
+                    LuaTokenKind::TkColon => {
+                        if is_parent_syntax(&token, LuaSyntaxKind::IndexExpr) {
+                            f.add_token_left_expected(syntax_id, TokenExpected::Space(0));
+                            f.add_token_right_expected(syntax_id, TokenExpected::Space(0));
+                            continue;
+                        }
+                        f.add_token_left_expected(syntax_id, TokenExpected::MaxSpace(1));
+                        f.add_token_right_expected(syntax_id, TokenExpected::MaxSpace(1));
+                    }
+                    LuaTokenKind::TkDot => {
                         f.add_token_left_expected(syntax_id, TokenExpected::Space(0));
                         f.add_token_right_expected(syntax_id, TokenExpected::Space(0));
                     }