Support parsing for const generic literals

CHANGELOG.md

@@ -1,5 +1,7 @@
 # Unreleased
 
+* Support parsing for const generic literals
+
 # v0.4.1
 
 * Minor fix to prevent panic on formatting files containing unbalanced parentheses in strings/chars/...

src/verus.pest

@@ -485,7 +485,13 @@ lifetime_arg = {
 }
 
 const_arg = {
-    expr
+    // Currently stable Rust only support literals as const args inside generic
+    // args. In general, it could be arbitrary expressions, but that causes
+    // headaches for parsing (in particular, we need to parse an `expr` that
+    // does _not_ have a `>` sign in it, but our `expr` parsing greedily eats
+    // any `>` that exists within it. Thus, for now, we simply restrict to
+    // literals.
+    literal
 }
 
 generic_args_binding = {

tests/verus-consistency.rs

@@ -2473,3 +2473,27 @@ fn fff() {
     } // verus!
     "###);
 }
+
+#[test]
+fn verus_const_generics() {
+    // Regression test for https://github.com/verus-lang/verusfmt/issues/90
+    let file = r#"
+verus! {
+       pub fn f<const N: u64>() -> u64 { N }
+ fn main() { f::<123>(); }
+}
+"#;
+    assert_snapshot!(parse_and_format(file).unwrap(), @r###"
+    verus! {
+
+    pub fn f<const N: u64>() -> u64 {
+        N
+    }
+
+    fn main() {
+        f::<123>();
+    }
+
+    } // verus!
+    "###);
+}