/
table.go
60 lines (49 loc) · 1.21 KB
/
table.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package php81
import (
"bytes"
"strings"
"text/template"
"github.com/go-bridget/mig/cmd/mig/internal"
)
func columnDefaultValue(column *internal.Column) string {
return typeAlias(column.DataType).Default
}
func columnToNativeType(column *internal.Column) string {
return typeAlias(column.DataType).Type
}
func RenderTable(data *internal.Table) (string, error) {
funcMap := template.FuncMap{
"NL": func() string { return "\n" },
"Camel": internal.Camel,
"DefaultValue": columnDefaultValue,
"ToNativeType": columnToNativeType,
}
tpl, err := template.New("render-php81-table").Funcs(funcMap).Parse(tableTemplate)
if err != nil {
return "", err
}
out := new(bytes.Buffer)
err = tpl.Execute(out, data)
if err != nil {
return "", err
}
sOut := out.String()
sOut = strings.ReplaceAll(sOut, "\n\n\n", "\n\n")
sOut = strings.ReplaceAll(sOut, "-\n", "")
sOut = strings.ReplaceAll(sOut, "\n-", "")
sOut = strings.TrimSpace(sOut) + "\n"
return sOut, nil
}
const tableTemplate = `
<?php
/** {{ .Comment }} */
class {{ .Name | Camel }}
{
public function __construct(
-{{range .Columns}}
/** {{ .Comment }} */
public ${{.Name}} ?{{. | ToNativeType}} = {{. | DefaultValue}};
{{end}}
- ) {}
}
`