Fix to parse array with comments when no trailing comma (#7994)

* Fix to parse array with comments when no trailing comma

* Fix parsing array with block comment and no whitespace

Author: jtran

rust/kcl-lib/src/parsing/parser.rs

@@ -787,7 +787,7 @@ fn array_separator(i: &mut TokenSlice) -> ModalResult<()> {
         // Normally you need a comma.
         comma_sep,
         // But, if the array is ending, no need for a comma.
-        peek(preceded(opt(whitespace), close_bracket)).void(),
+        peek(preceded((opt(non_code_node), opt(whitespace)), close_bracket)).void(),
     ))
     .parse_next(i)
 }
@@ -800,7 +800,7 @@ pub(crate) fn array_elem_by_elem(i: &mut TokenSlice) -> ModalResult<Node<ArrayEx
         0..,
         alt((
             terminated(expression.map(NonCodeOr::Code), array_separator),
-            terminated(non_code_node.map(NonCodeOr::NonCode), whitespace),
+            terminated(non_code_node.map(NonCodeOr::NonCode), opt(whitespace)),
         )),
     )
     .context(expected("array contents, a list of elements (like [1, 2, 3])"))
@@ -4841,6 +4841,26 @@ export fn cos(num: number(rad)): number(_) {}"#;
         let _arr = array_elem_by_elem(&mut tokens.as_slice()).unwrap();
     }
 
+    #[test]
+    fn array_no_trailing_comma_with_comment() {
+        let program = r#"[
+            1, // one
+            2, // two
+            3  // three
+        ]"#;
+        let module_id = ModuleId::default();
+        let tokens = crate::parsing::token::lex(program, module_id).unwrap();
+        let _arr = array_elem_by_elem(&mut tokens.as_slice()).unwrap();
+    }
+
+    #[test]
+    fn array_block_comment_no_whitespace() {
+        let program = r#"[1/* comment*/]"#;
+        let module_id = ModuleId::default();
+        let tokens = crate::parsing::token::lex(program, module_id).unwrap();
+        let _arr = array_elem_by_elem(&mut tokens.as_slice()).unwrap();
+    }
+
     #[test]
     fn basic_if_else() {
         let some_program_string = "if true {

rust/kcl-lib/src/unparser.rs

@@ -2451,6 +2451,30 @@ fn ghi(part001) {
         );
     }
 
+    #[test]
+    fn test_recast_array_no_trailing_comma_with_comments() {
+        let some_program_string = r#"[
+  1, // one
+  2, // two
+  3  // three
+]"#;
+        let program = crate::parsing::top_level_parse(some_program_string).unwrap();
+
+        let recasted = program.recast_top(&Default::default(), 0);
+        assert_eq!(
+            recasted,
+            r#"[
+  1,
+  // one
+  2,
+  // two
+  3,
+  // three
+]
+"#
+        );
+    }
+
     #[test]
     fn test_recast_negative_var() {
         let some_program_string = r#"w = 20