adding alias types

package.json

@@ -1,6 +1,6 @@
 {
     "name": "typescript-code-generator",
-    "version": "0.0.7",
+    "version": "0.0.8",
     "description": "",
     "scripts": {
         "build": "tsc",

src/AliasTypeBuilder.ts

@@ -0,0 +1,40 @@
+import { BuilderBase } from "./BuilderBase";
+import { StringBuilder } from "./StringBuilder";
+
+export class AliasTypeBuilder extends BuilderBase {
+    private readonly name: string;
+    private readonly type: string;
+    private export?: boolean;
+
+    public constructor(name: string, type: string) {
+        super();
+        if (!this.isValidObjectName(name)) {
+        throw new Error(`Invalid name for type: ${name}`);
+        }
+        this.name = name;
+        this.type = type;
+    }
+
+    public getName(): string {
+        return this.name;
+    }
+
+    public withExport(): this {
+        this.export = true;
+        return this;
+    }
+
+    public toArray(): Array<string> {
+        const builder = new StringBuilder();
+        const parts = [];
+        if (this.export) {
+            parts.push("export");
+        }
+        parts.push("type");
+        parts.push(this.name);
+        parts.push("=");
+        parts.push(this.type);
+        builder.append(parts.join(" "));
+        return builder.toArray();
+    }
+}

src/CodeDocument.ts

@@ -1,10 +1,14 @@
+import { AliasTypeBuilder } from "./AliasTypeBuilder";
 import { ClassBuilder } from "./ClassBuilder";
 import { InterfaceBuilder } from "./InterfaceBuilder";
 import { StringBuilder } from "./StringBuilder";
+import { TypeBuilder } from "./TypeBuilder";
 
 export class CodeDocument {
   private classes = new Array<ClassBuilder>();
   private interfaces = new Array<InterfaceBuilder>();
+  private types = new Array<TypeBuilder>();
+  private aliasTypes = new Array<AliasTypeBuilder>();
   private headComments: Array<string> = [];
   private imports = new Array<Import>();
 
@@ -48,6 +52,16 @@ export class CodeDocument {
     return this;
   }
 
+  public addType(type: TypeBuilder): this {
+    this.types.push(type);
+    return this;
+  }
+
+  public addAliasType(type: AliasTypeBuilder): this {
+    this.aliasTypes.push(type);
+    return this;
+  }
+
   public toString(): string {
     const builder = new StringBuilder();
 
@@ -69,6 +83,18 @@ export class CodeDocument {
       builder.addNewline();
     }
 
+    if (this.aliasTypes.length > 0) {
+      this.aliasTypes.forEach((type) => {
+        builder.appendLines(type.toArray()).addNewline();
+      });
+    }
+
+    if (this.types.length > 0) {
+      this.types.forEach((type) => {
+        builder.appendLines(type.toArray()).addNewline();
+      });
+    }
+
     this.interfaces.forEach((interfaceObj) => {
       builder.appendLines(interfaceObj.toArray()).addNewline();
     });