Automatically capitalize TS class names (Also remove spaces)

Fixes #8

Author: OneOfOne

cmd/struct2ts/main.go

@@ -44,6 +44,7 @@ func init() {
 	KP.Flag("no-date", "Don't automatically handle time.Unix () <-> JS Date().").Short('D').BoolVar(&opts.NoDate)
 	KP.Flag("no-helpers", "Don't output the helpers.").Short('H').BoolVar(&opts.NoHelpers)
 	KP.Flag("no-default-values", "Don't assign default/zero values in the ctor.").Short('N').BoolVar(&opts.NoAssignDefaults)
+	KP.Flag("no-capitalize", "Don't capitalize TS class names.").Short('C').BoolVar(&opts.NoCapitalize)
 	KP.Flag("interface", "Only generate an interface (disables all the other options).").Short('i').BoolVar(&opts.InterfaceOnly)
 
 	KP.Flag("src-only", "Only output the Go code (helpful if you want to edit it yourself).").Short('s').BoolVar(&srcOnly)
@@ -206,13 +207,14 @@ func runStruct2TS(w io.Writer) error {
 		InterfaceOnly:    {{ .opts.InterfaceOnly    }},
 
 		NoConstructor: {{ .opts.NoConstructor }},
+		NoCapitalize:  {{ .opts.NoCapitalize }},
 		MarkOptional:  {{ .opts.MarkOptional  }},
 		NoToObject:    {{ .opts.NoToObject    }},
 		NoExports:     {{ .opts.NoExports        }},
 		NoHelpers:     {{ .opts.NoHelpers        }},
 		NoDate:        {{ .opts.NoDate        }},
 
-		ES6:        {{ .opts.ES6        }},
+		ES6:           {{ .opts.ES6 }},
 	})
 
 	{{ range $_, $t := .types }}

s2ts.go

@@ -9,6 +9,7 @@ import (
 	"log"
 	"reflect"
 	"strings"
+	"unicode"
 )
 
 type Options struct {
@@ -18,6 +19,7 @@ type Options struct {
 	InterfaceOnly    bool
 
 	MarkOptional  bool
+	NoCapitalize  bool
 	NoConstructor bool
 	NoToObject    bool
 	NoExports     bool
@@ -78,6 +80,9 @@ func (s *StructToTS) addType(t reflect.Type, name, prefix string) (out *Struct)
 
 	if name == "" {
 		name = t.Name()
+		if !s.opts.NoCapitalize {
+			name = capitalize(name)
+		}
 	}
 
 	out = &Struct{
@@ -247,3 +252,20 @@ func stripType(t reflect.Type) string {
 	}
 	return n
 }
+
+func capitalize(s string) string {
+	var last rune
+	return strings.Map(func(r rune) rune {
+		l := last
+		last = r
+		if unicode.IsSpace(r) {
+			return 0
+		}
+
+		if !unicode.IsLetter(l) && unicode.IsLetter(r) {
+			r = unicode.ToUpper(r)
+		}
+
+		return r
+	}, s)
+}