From 4fbc86e4b68ea1c4d771fd050daf24acbe33cbd3 Mon Sep 17 00:00:00 2001
From: PhatPhuckDave <david.majdandzic@hotmail.com>
Date: Thu, 12 Sep 2024 01:03:24 +0200
Subject: [PATCH] Code polish

---
 class.tmpl | 10 +++++-----
 main.go    |  4 ++++
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/class.tmpl b/class.tmpl
index 593963a..26dd9a4 100644
--- a/class.tmpl
+++ b/class.tmpl
@@ -1,10 +1,7 @@
----@diagnostic disable: missing-return
+---@diagnostic disable: missing-return, lowercase-global
 {{range .Fields -}}
 ---@field {{.Name}} {{.Type}} {{.Comment}}
 {{end -}}
-{{range .Constructors -}}
----@overload fun({{range $index, $param := .Params}}{{if $index}}, {{end}}{{$param.Name}}: {{$param.Type}}{{end}}): {{$.ClassName}}
-{{end -}}
 ---@class {{.ClassName}}
 {{.ClassName}} = {
 	{{$n := len .Methods -}}
@@ -23,6 +20,9 @@
 	{{end}}{{end}}
 }
 
----@type {{$.ClassName}}
 -- Define class globally so it's "available" to other files
+---@type {{$.ClassName}}
+{{range .Constructors -}}
+---@overload fun({{range $index, $param := .Params}}{{if $index}}, {{end}}{{$param.Name}}: {{$param.Type}}{{end}}): {{$.ClassName}}
+{{end -}}
 {{$.ClassName}} = nil
\ No newline at end of file
diff --git a/main.go b/main.go
index 3f49ce3..5e90c11 100644
--- a/main.go
+++ b/main.go
@@ -120,6 +120,8 @@ func ParseFile(filename string) (*Class, error) {
 	})
 	res.Constructors = append(res.Constructors, resConstructor)
 
+	// constructorDetails := codeblocks.Eq(1)
+
 	log.Printf("%#v", res)
 
 	return &res, nil
@@ -131,6 +133,8 @@ func MapType(t string) string {
 		return "any"
 	case "int":
 		return "number"
+	case "float":
+		return "number"
 	case "string":
 		return "string"
 	case "boolean":