Generate UnmarshalJSON to handle emptyStringInt types

This commit changes the code generator to generate a `UnmarshalJSON` for each
struct, so that if unmarshalled it properly handles UniFis varying integer values
via the `emptyStringInt` type.

Structs not including a field of `int` type will still have the function generated,
but it will effectively do nothing.

Fixes #18

fields/api.go.tmpl

@@ -2,6 +2,18 @@
 
 {{ define "field" }}
 	{{ .FieldName }} {{ if .IsArray }}[]{{end}}{{ .FieldType }} `json:"{{ .JSONName }}{{ if .OmitEmpty }},omitempty{{ end }}"` {{ if .FieldValidation }}// {{ .FieldValidation }}{{ end }} {{- end }}
+{{ define "field-emptyStringInt" }}
+	{{- if ne .FieldType "int" }}{{else}}
+	    {{ .FieldName }} {{ if .IsArray }}[]{{end}}emptyStringInt `json:"{{ .JSONName }}{{ if .OmitEmpty }}{{ end }}"`{{ end }} {{- end }}
+{{ define "typecast" }}
+    {{- if eq .FieldType "int" }}{{- if .IsArray }}
+    dst.{{ .FieldName }}= make([]int, len(aux.{{ .FieldName }}))
+    for i, v := range aux.{{ .FieldName }} {
+        dst.{{ .FieldName }}[i] = int(v)
+    }
+    {{- else }}
+    dst.{{ .FieldName }} = int(aux.{{ .FieldName }})
+    {{- end }}{{- end }}{{- end }}
 {{ define "field-embed" }}
 	{{ .FieldName }} {{ if .IsArray }}[]{{end}}{{ if not .Fields }}{{ .FieldType }}{{ else }}struct {
 	{{ range $fk, $fv := .Fields }}{{ if not $fv }}
@@ -14,13 +26,15 @@ package unifi
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 )
 
 // just to fix compile issues with the import
 var (
-	_ fmt.Formatter
 	_ context.Context
+	_ fmt.Formatter
+	_ json.Marshaler
 )
 
 {{ if embedTypes -}}
@@ -36,6 +50,28 @@ type {{ $k }} struct {
 	{{ range $fk, $fv := $v.Fields }}{{ if not $fv }}
 	{{ else }}{{- template "field" $fv }}{{ end }}{{ end }}
 }
+
+func (dst *{{ $k }}) UnmarshalJSON(b []byte) error {
+	type Alias {{ $k }}
+	aux := &struct {
+	    {{- range $fk, $fv := $v.Fields }}{{ if not $fv }}
+	    {{- else }}{{- template "field-emptyStringInt" $fv }}{{ end }}{{- end }}
+
+		*Alias
+	}{
+		Alias: (*Alias)(dst),
+	}
+
+	err := json.Unmarshal(b, &aux)
+	if err != nil {
+		return fmt.Errorf("unable to unmarshal alias: %w", err)
+	}
+
+    {{- range $fk, $fv := $v.Fields }}{{ if not $fv }}
+    {{- else }}{{- template "typecast" $fv }}{{ end }}{{ end }}
+
+	return nil
+}
 {{ end }}
 {{- end -}}
 

unifi/account.go

@@ -3,33 +3,8 @@ package unifi
 import (
 	"context"
 	"encoding/json"
-	"fmt"
 )
 
-func (dst *Account) UnmarshalJSON(b []byte) error {
-	type Alias Account
-	aux := &struct {
-		TunnelType       emptyStringInt `json:"tunnel_type"`
-		TunnelMediumType emptyStringInt `json:"tunnel_medium_type"`
-		VLAN             emptyStringInt `json:"vlan"`
-
-		*Alias
-	}{
-		Alias: (*Alias)(dst),
-	}
-
-	err := json.Unmarshal(b, &aux)
-	if err != nil {
-		return fmt.Errorf("unable to unmarshal alias: %w", err)
-	}
-
-	dst.TunnelType = int(aux.TunnelType)
-	dst.TunnelMediumType = int(aux.TunnelMediumType)
-	dst.VLAN = int(aux.VLAN)
-
-	return nil
-}
-
 func (dst *Account) MarshalJSON() ([]byte, error) {
 	type Alias Account
 	aux := &struct {