number: hex literal and decimal

Author: aspeddro

grammar.js

@@ -1371,9 +1371,10 @@ module.exports = grammar({
     extension_identifier: $ => /[a-zA-Z0-9_\.]+/,
 
     number: $ => {
+      // OCaml: https://github.com/tree-sitter/tree-sitter-ocaml/blob/f1106bf834703f1f2f795da1a3b5f8f40174ffcc/ocaml/grammar.js#L26
       const hex_literal = seq(
-        choice('0x', '0X'),
-        /[\da-fA-F](_?[\da-fA-F])*/
+        optional(choice('-', '+')),
+        /0[xX][0-9A-Fa-f][0-9A-Fa-f_]*(\.[0-9A-Fa-f_]*)?([pP][+\-]?[0-9][0-9_]*)?[g-zG-Z]?/
       )
 
       const decimal_digits = /\d(_?\d)*/
@@ -1387,8 +1388,8 @@ module.exports = grammar({
       const bigint_literal = seq(choice(hex_literal, binary_literal, octal_literal, decimal_digits), 'n')
 
       const decimal_integer_literal = choice(
-        '0',
-        seq(optional('0'), /[1-9]/, optional(seq(optional('_'), decimal_digits)))
+        repeat('0'),
+        seq(repeat('0'), /[1-9]/, optional(seq(optional('_'), decimal_digits)))
       )
 
       const decimal_literal = seq(

test/corpus/literals.txt

@@ -42,6 +42,24 @@ Numbers
 0o1234L
 10l
 0xffffffffl
+-0x3.p+1
+0x1p2047
+-0x1p2047
+0x1.2p2047
+0x1p-1022
+0x0.0000_0000_0000_1p-1022
+0x1p-52
+0x1.0000_0000_0000_1
+0x1p-1023
+0xffffp0
+00007.0
+02e3
+0000.2
+00.
+000.
+01.
+001.
+002e3
 
 ---
 
@@ -72,6 +90,24 @@ Numbers
   (expression_statement (number))
   (expression_statement (number))
   (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
+  (expression_statement (number))
   (expression_statement (number)))
 
 ==========================================