feat: basic support for range for

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "cxx2flow"
-version = "0.5.9"
+version = "0.5.10"
 edition = "2018"
 authors = ["mgt <[email protected]>"] 
 description = "Convert your C/C++ code to control flow chart"

README-en.md

@@ -97,3 +97,4 @@ https://github.com/Enter-tainer/cxx2flow
 
 - The support of preprocessor is based on `cpp`, and is disabled by default. `--cpp` flag is needed to enable it. It will fail if `cpp` does not exist in `PATH`.
 - Supported control flow keyword: while，for，if，break，continue，break，return，switch, goto, do-while。
+- Very basic support for range based loop in C++ 11.

README.md

@@ -98,3 +98,4 @@ https://github.com/Enter-tainer/cxx2flow
 
 - 对于预处理器的支持基于 `cpp` ，默认关闭，需要使用 `--cpp` 参数手动启用。如果 `PATH` 中不存在 `cpp` 则会失败。
 - 支持的控制流语句有：while，for，if，break，continue，break，return，switch, goto, do-while。
+- 对 range for 有基本支持。部分情况下，受到 tree-sitter-cpp 能力限制，会出现一些问题

assets/rangefor.cpp

@@ -0,0 +1,29 @@
+void basic() {
+    for (auto i : vec) {
+        i = i + 1;
+    }
+}
+
+void auto_ref() {
+    for (auto & i : map) {
+        a;
+    }
+}
+
+void const_auto_rvref() {
+    for (const auto &&i : vec) {
+        i - 1;
+    }
+}
+
+void auto_destruct() {
+    for (auto [x, y] : vec) {
+        x + y;
+    }
+}
+
+void int_init() {
+    for (int i : vec) {
+        i;
+    }
+}

src/parser.rs

@@ -193,6 +193,7 @@ fn parse_single_stat(stat: Node, content: &[u8]) -> Result<Rc<RefCell<Ast>>> {
         "while_statement" => parse_while_stat(stat, content),
         "do_statement" => parse_do_while_stat(stat, content),
         "for_statement" => parse_for_stat(stat, content),
+        "for_range_loop" => parse_range_for_stat(stat, content),
         "switch_statement" => parse_switch_stat(stat, content),
         "goto_statement" => parse_goto_stat(stat, content),
         "expression_statement" | "declaration" => {
@@ -422,3 +423,31 @@ fn parse_for_stat(for_stat: Node, content: &[u8]) -> Result<Rc<RefCell<Ast>>> {
     )));
     Ok(res)
 }
+
+fn parse_range_for_stat(range_for_stat: Node, content: &[u8]) -> Result<Rc<RefCell<Ast>>> {
+    let ty = range_for_stat.child_by_field_name("type").ok_or(Error::ChildNotFound)?;
+    let declarator = range_for_stat.child_by_field_name("declarator").ok_or(Error::ChildNotFound)?;
+    let range = range_for_stat.child_by_field_name("right").ok_or(Error::ChildNotFound)?;
+    let body = range_for_stat.child_by_field_name("body").ok_or(Error::ChildNotFound)?;
+    let body = parse_stat(body, content)?;
+    let type_text = ty.utf8_text(content)?;
+    let init_text = declarator.utf8_text(content)?;
+    let range_text = range.utf8_text(content)?;
+    let real_init_text = format!("{init_text}_iter = {range_text}.begin()");
+    let real_cond_text = format!("{init_text}_iter != {range_text}.end()");
+    let real_update_text = format!("++{init_text}_iter");
+    let res = Rc::new(RefCell::new(Ast::new(
+        AstNode::For {
+            init: real_init_text,
+            cond: real_cond_text,
+            upd: real_update_text,
+            body: Rc::new(RefCell::new(Ast::new(AstNode::Compound(vec![
+                Rc::new(RefCell::new(Ast::new(AstNode::Stat(format!("{type_text} {init_text} = *{init_text}_iter")), range_for_stat.byte_range(), None))),
+                body
+            ]), range_for_stat.byte_range(), None))),
+        },
+        range_for_stat.byte_range(),
+        None,
+    )));
+    Ok(res)
+}