Assign explicit values to TS enums (#24)

* Assign explicit values to TS enums TypeScript enums without explicitly assigned values are difficult to use if they are defined in declaration files. * Test TS enum variants as types
bytecodealliance · Jun 28, 2021 · a2f2d54 · a2f2d54
1 parent dd234ab
commit a2f2d54
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/crates/gen-js/src/lib.rs b/crates/gen-js/src/lib.rs
@@ -347,10 +347,10 @@ impl Generator for Js {
         } else if variant.is_enum() {
             self.src
                 .ts(&format!("export enum {} {{\n", name.to_camel_case()));
-            for case in variant.cases.iter() {
+            for (i, case) in variant.cases.iter().enumerate() {
                 self.docs(&case.docs);
-                self.src.ts(&case.name.to_camel_case());
-                self.src.ts(",\n");
+                let name = case.name.to_camel_case();
+                self.src.ts(&format!("{} = {},\n", name, i));
             }
             self.src.ts("}\n");
 

diff --git a/crates/gen-js/tests/run.ts b/crates/gen-js/tests/run.ts
@@ -370,6 +370,11 @@ function test_variants(wasm: exports.Wasm) {
   assert.deepStrictEqual(wasm.invert_bool(true), false);
   assert.deepStrictEqual(wasm.invert_bool(false), true);
 
+  {
+    const a: exports.E1.A = exports.E1.A;
+    const b: exports.E1.B = exports.E1.B;
+  }
+
   {
     const [a1, a2, a3, a4, a5, a6] = wasm.variant_casts([
       { tag: 'a', val: 1 },