Fix TypeScript Plugin Codegen

polywrap · Aug 20, 2021 · 90d18c2 · 90d18c2
1 parent 76eadb4
commit 90d18c2
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 6 deletions.
diff --git a/packages/schema/bind/src/bindings/plugin-ts/templates/types-ts.mustache b/packages/schema/bind/src/bindings/plugin-ts/templates/types-ts.mustache
@@ -28,13 +28,13 @@ export interface {{type}} {
 
 {{/objectTypes}}
 {{#enumTypes}}
-enum {{type}}Enum {
+export enum {{type}}Enum {
   {{#constants}}
   {{.}},
   {{/constants}}
 }
 
-type {{type}}String =
+export type {{type}}String =
   {{#constants}}
   | "{{.}}"
   {{/constants}}
@@ -55,12 +55,19 @@ export interface {{type}} {
 {{/importedObjectTypes}}
 {{#importedEnumTypes}}
 /* URI: "{{uri}}" */
-export enum {{type}} {
+export enum {{type}}Enum {
   {{#constants}}
   {{.}},
   {{/constants}}
 }
 
+export type {{type}}String =
+  {{#constants}}
+  | "{{.}}"
+  {{/constants}}
+
+export type {{type}} = {{type}}Enum | {{type}}String;
+
 {{/importedEnumTypes}}
 /// Imported Objects END ///
 

diff --git a/packages/test-cases/cases/bind/sanity/output/plugin-ts/types.ts b/packages/test-cases/cases/bind/sanity/output/plugin-ts/types.ts
@@ -60,12 +60,12 @@ export interface AnotherType {
   circular?: Types.CustomType | null;
 }
 
-enum CustomEnumEnum {
+export enum CustomEnumEnum {
   STRING,
   BYTES,
 }
 
-type CustomEnumString =
+export type CustomEnumString =
   | "STRING"
   | "BYTES"
 
@@ -91,11 +91,17 @@ export interface TestImport_AnotherObject {
 }
 
 /* URI: "testimport.uri.eth" */
-export enum TestImport_Enum {
+export enum TestImport_EnumEnum {
   STRING,
   BYTES,
 }
 
+export type TestImport_EnumString =
+  | "STRING"
+  | "BYTES"
+
+export type TestImport_Enum = TestImport_EnumEnum | TestImport_EnumString;
+
 /// Imported Objects END ///
 
 /// Imported Queries START ///