Add support for typescript enums.

Enums can be typechecked now and also minified.

Author: calebegg

src/sickle.ts

@@ -150,6 +150,9 @@ class Annotator {
         this.visitTypeAlias(<ts.TypeAliasDeclaration>node);
         this.writeNode(node);
         break;
+      case ts.SyntaxKind.EnumDeclaration:
+        this.visitEnum(<ts.EnumDeclaration>node);
+        break;
       default:
         this.writeNode(node);
         break;
@@ -263,6 +266,24 @@ class Annotator {
     this.emit(': void;\n');
   }
 
+  private visitEnum(node: ts.EnumDeclaration) {
+    this.emit('/** @typedef {number} */\n');
+    this.writeNode(node);
+    this.emit('\n');
+    let i = 0;
+    for (let member of node.members) {
+      this.emit(`/** @type {${node.name.getText()}} */\n`);
+      this.emit(`(<any>${node.name.getText()}).${member.name.getText()} = `);
+      if (member.initializer) {
+        this.visit(member.initializer);
+      } else {
+        this.emit(String(i));
+        i++;
+      }
+      this.emit(';\n');
+    }
+  }
+
   private writeNode(node: ts.Node, skipComments: boolean = false) {
     if (node.getChildCount() == 0) {
       // Directly write complete tokens.

test_files/enum.ts

@@ -0,0 +1 @@
+enum FooEnum {XYZ, PI = 3.14159}

test_files/es6/enum.js

@@ -0,0 +1,10 @@
+/** @typedef {number} */
+var FooEnum;
+(function (FooEnum) {
+    FooEnum[FooEnum["XYZ"] = 0] = "XYZ";
+    FooEnum[FooEnum["PI"] = 3.14159] = "PI";
+})(FooEnum || (FooEnum = {}));
+/** @type {FooEnum} */
+FooEnum.XYZ = 0;
+/** @type {FooEnum} */
+FooEnum.PI = 3.14159;

test_files/sickle/enum.ts

@@ -0,0 +1,6 @@
+/** @typedef {number} */
+enum FooEnum {XYZ, PI = 3.14159}
+/** @type {FooEnum} */
+(<any>FooEnum).XYZ = 0;
+/** @type {FooEnum} */
+(<any>FooEnum).PI =  3.14159;