[PRISM] Support US-ASCII symbols

peterzhu2118 · peterzhu2118 · commit e0fd1b022f12 · 2024-01-26T16:13:04.000-05:00
diff --git a/prism_compile.c b/prism_compile.c
@@ -216,6 +216,9 @@ parse_string_symbol(const pm_symbol_node_t *symbol, const pm_parser_t *parser)
     else if (symbol->base.flags & PM_SYMBOL_FLAGS_FORCED_BINARY_ENCODING) {
         encoding = "ASCII-8BIT";
     }
+    else if (symbol->base.flags & PM_SYMBOL_FLAGS_FORCED_US_ASCII_ENCODING) {
+        encoding = "US-ASCII";
+    }
     const uint8_t *start = pm_string_source(&symbol->unescaped);
     return parse_symbol(start, start + pm_string_length(&symbol->unescaped), encoding);
 }
diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb
@@ -791,6 +791,12 @@ def test_SymbolNode
         # -*- coding: us-ascii -*-
         :"\xff"
       RUBY
+
+      # Test US-ASCII symbol in a ASCII-8BIT file
+      assert_prism_eval(<<~'RUBY', raw: true)
+        # -*- coding: us-ascii -*-
+        :a
+      RUBY
     end
 
     def test_XStringNode

Original file line number	Diff line number	Diff line change
`@@ -216,6 +216,9 @@ parse_string_symbol(const pm_symbol_node_t symbol, const pm_parser_t parser)`
`216`	`216`	`else if (symbol->base.flags & PM_SYMBOL_FLAGS_FORCED_BINARY_ENCODING) {`
`217`	`217`	`encoding = "ASCII-8BIT";`
`218`	`218`	`}`
	`219`	`+ else if (symbol->base.flags & PM_SYMBOL_FLAGS_FORCED_US_ASCII_ENCODING) {`
	`220`	`+ encoding = "US-ASCII";`
	`221`	`+ }`
`219`	`222`	`const uint8_t *start = pm_string_source(&symbol->unescaped);`
`220`	`223`	`return parse_symbol(start, start + pm_string_length(&symbol->unescaped), encoding);`
`221`	`224`	`}`