Merge pull request #3 from anc95/develop

👏 MVP VERSION DONE

Author: anc95

README.md

@@ -1,2 +1,46 @@
 # golang-enum-to-ts
 Transform Golang `enum` type to Typescript enum
+
+## Function
+
+### Before (Golang)
+```golang
+package some
+
+type Status int
+
+const (
+	Todo Status = iota
+	Done
+	Pending
+	InProgress
+)
+
+type Sex string
+
+const (
+	Female Sex = "female"
+	Male   Sex = "male"
+)
+
+func Abctext() {
+	//dadsad
+}
+```
+
+### After (Typescript)
+
+```ts
+namespace some {
+  export enum Sex {
+    Female = 'female',
+    Male = 'male',
+  }
+  export enum Status {
+    Pending = 2,
+    InProgress = 3,
+    Todo = 0,
+    Done = 1,
+  }
+}
+```

src/ast/__snapshots__/ast_test.snap

@@ -3,15 +3,16 @@
 ast.File{
     Name: "some",
     Body: {
-        ast.TypeDeclaration{Id:"Status", Kind:"string"},
+        ast.TypeDeclaration{Id:"Status", Kind:"int"},
         ast.ConstDeclaration{
             Declarators: {
                 {Kind:"Status", Id:"Todo", Value:"iota"},
                 {Kind:"", Id:"Done", Value:""},
+                {Kind:"", Id:"Pending", Value:""},
                 {Kind:"", Id:"InProgress", Value:""},
             },
         },
-        ast.TypeDeclaration{Id:"Sex", Kind:"int"},
+        ast.TypeDeclaration{Id:"Sex", Kind:"string"},
         ast.ConstDeclaration{
             Declarators: {
                 {Kind:"Sex", Id:"Female", Value:"female"},

src/ast/ast.go

@@ -83,7 +83,7 @@ func (a *AstGenerator) readTypeDeclaration() TypeDeclaration {
 
 	next, _ = a.nextToken(true)
 
-	if next.Type != token.IntType {
+	if next.Type == token.IntType {
 		d.Kind = Int
 	} else {
 		d.Kind = String
@@ -134,6 +134,8 @@ func (a *AstGenerator) readConstDeclaration() ConstDeclaration {
 			if a.currentToken.Type == token.RightParentheses {
 				break
 			}
+
+			a.backToken()
 		}
 
 		a.matchNextLine()
@@ -188,3 +190,7 @@ func (a *AstGenerator) Gen() File {
 
 	return file
 }
+
+func NewAstGenerator(tokens []token.Token) AstGenerator {
+	return AstGenerator{Tokens: tokens, index: -1}
+}

src/generator/__snapshots__/generator_test.snap

@@ -0,0 +1,16 @@
+
+[TestNormal - 1]
+namespace some {
+  export enum Status {
+    Todo = 0,
+    Done = 1,
+    Pending = 2,
+    InProgress = 3,
+  }
+  export enum Sex {
+    Female = 'female',
+    Male = 'male',
+  }
+}
+
+---

src/generator/generator.go

@@ -1 +1,104 @@
 package generator
+
+import (
+	"fmt"
+	"strconv"
+
+	"github.com/anc95/golang-enum-to-ts/src/ast"
+)
+
+func getEnum(f ast.File) map[string]map[string]interface{} {
+	body := f.Body
+	enumType := make(map[string]string)
+	enumValue := make(map[string]map[string]interface{})
+
+	for i := 0; i < len(body); i++ {
+		decl := body[i]
+
+		switch decl.(type) {
+		case ast.TypeDeclaration:
+			decl := decl.(ast.TypeDeclaration)
+			enumType[decl.Id] = string(decl.Kind)
+		case ast.ConstDeclaration:
+			decl := decl.(ast.ConstDeclaration)
+			kind := ""
+			iotaValue := 0
+			useIota := false
+			iotaFlag := false
+			var prevValue interface{}
+
+			for _, x := range decl.Declarators {
+				if x.Kind != "" {
+					kind = x.Kind
+
+					if enumValue[kind] == nil {
+						enumValue[kind] = map[string]interface{}{}
+					}
+				}
+
+				if iotaFlag {
+					iotaValue += 1
+				}
+
+				if x.Value != "" {
+					if x.Value == "iota" {
+						useIota = true
+						iotaFlag = true
+						enumValue[kind][x.Id] = iotaValue
+					} else {
+						useIota = false
+
+						if kind == "int" || enumType[kind] == "int" {
+							val, _ := strconv.Atoi(x.Value)
+							enumValue[kind][x.Id] = val
+							prevValue = val
+						}
+
+						enumValue[kind][x.Id] = x.Value
+						prevValue = x.Value
+					}
+				} else {
+					if useIota {
+						enumValue[kind][x.Id] = iotaValue
+					} else {
+						enumValue[kind][x.Id] = prevValue
+					}
+				}
+			}
+		}
+	}
+
+	return enumValue
+}
+
+func GenerateTS(f ast.File) string {
+	enumValue := getEnum(f)
+	result := ""
+
+	for key, val := range enumValue {
+		ret := ""
+
+		for k, v := range val {
+			if ret != "" {
+				ret += "\n"
+			}
+
+			switch v.(type) {
+			case int:
+				ret += fmt.Sprintf("    %s = %d,", k, v)
+			case string:
+				ret += fmt.Sprintf("    %s = '%s',", k, v)
+			}
+		}
+
+		ret = fmt.Sprintf("  export enum %s {\n", key) + ret
+		ret += "\n  }\n"
+
+		result += ret
+	}
+
+	result = fmt.Sprintf("namespace %s {\n", f.Name) + result
+	result += "}\n"
+
+	return result
+}

src/generator/generator_test.go

@@ -0,0 +1,24 @@
+package generator
+
+import (
+	"os"
+	"path"
+	"testing"
+
+	"github.com/anc95/golang-enum-to-ts/src/ast"
+	"github.com/anc95/golang-enum-to-ts/src/token"
+	"github.com/gkampitakis/go-snaps/snaps"
+)
+
+func TestNormal(t *testing.T) {
+	wd, _ := os.Getwd()
+
+	a := path.Join(wd, "../test-cases/normal.go")
+	parser := token.NewParser(a)
+	tokens := parser.Parse()
+	ast := ast.NewAstGenerator(tokens)
+
+	result := GenerateTS(ast.Gen())
+
+	snaps.MatchSnapshot(t, result)
+}