142 lines
3.3 KiB
Go
142 lines
3.3 KiB
Go
// Copyright 2019-present Facebook Inc. All rights reserved.
|
|
// This source code is licensed under the Apache 2.0 license found
|
|
// in the LICENSE file in the root directory of this source tree.
|
|
|
|
package printer
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"entgo.io/ent/entc/gen"
|
|
|
|
"github.com/olekukonko/tablewriter"
|
|
"github.com/olekukonko/tablewriter/tw"
|
|
)
|
|
|
|
// A Config controls the output of Fprint.
|
|
type Config struct {
|
|
io.Writer
|
|
}
|
|
|
|
// Print prints a table description of the graph to the given writer.
|
|
func (p Config) Print(g *gen.Graph) {
|
|
for _, n := range g.Nodes {
|
|
p.node(n)
|
|
}
|
|
}
|
|
|
|
// Fprint executes "pretty-printer" on the given writer.
|
|
func Fprint(w io.Writer, g *gen.Graph) {
|
|
Config{Writer: w}.Print(g)
|
|
}
|
|
|
|
// node returns description of a type. The format of the description is:
|
|
//
|
|
// Type:
|
|
// <Fields Table>
|
|
//
|
|
// <Edges Table>
|
|
func (p Config) node(t *gen.Type) {
|
|
var (
|
|
b strings.Builder
|
|
id []*gen.Field
|
|
table = tablewriter.NewWriter(&b)
|
|
header = []string{"Field", "Type", "Unique", "Optional", "Nillable", "Default", "UpdateDefault", "Immutable", "StructTag", "Validators", "Comment"}
|
|
)
|
|
b.WriteString(t.Name + ":\n")
|
|
table.Options(
|
|
tablewriter.WithHeaderConfig(tw.CellConfig{
|
|
Padding: tw.CellPadding{
|
|
Global: tw.Padding{
|
|
Left: tw.Space,
|
|
Right: tw.Space,
|
|
},
|
|
},
|
|
Formatting: tw.CellFormatting{
|
|
AutoFormat: tw.Off,
|
|
},
|
|
}),
|
|
tablewriter.WithRendition(tw.Rendition{
|
|
Symbols: tw.NewSymbols(tw.StyleASCII),
|
|
}),
|
|
)
|
|
table.Header(header)
|
|
var alignment = make([]tw.Align, 0)
|
|
if t.ID != nil {
|
|
id = append(id, t.ID)
|
|
}
|
|
for _, f := range append(id, t.Fields...) {
|
|
v := reflect.ValueOf(*f)
|
|
row := make([]string, len(header))
|
|
for i := 0; i < len(row)-1; i++ {
|
|
field := v.FieldByNameFunc(func(name string) bool {
|
|
// The first field is mapped from "Name" to "Field".
|
|
return name == "Name" && i == 0 || name == header[i]
|
|
})
|
|
row[i] = fmt.Sprint(field.Interface())
|
|
_, err := strconv.Atoi(row[i])
|
|
if err == nil {
|
|
alignment = append(alignment, tw.AlignRight)
|
|
} else {
|
|
alignment = append(alignment, tw.AlignLeft)
|
|
}
|
|
}
|
|
row[len(row)-1] = f.Comment()
|
|
_ = table.Append(row)
|
|
table.Options(
|
|
tablewriter.WithRowAlignmentConfig(
|
|
tw.CellAlignment{PerColumn: alignment},
|
|
),
|
|
)
|
|
}
|
|
err := table.Render()
|
|
if err != nil {
|
|
return
|
|
}
|
|
// Create new table for edges
|
|
table = tablewriter.NewWriter(&b)
|
|
table.Options(
|
|
tablewriter.WithHeaderConfig(tw.CellConfig{
|
|
Formatting: tw.CellFormatting{AutoFormat: tw.Off},
|
|
Padding: tw.CellPadding{
|
|
Global: tw.Padding{
|
|
Left: tw.Space,
|
|
Right: tw.Space,
|
|
},
|
|
},
|
|
}),
|
|
tablewriter.WithRendition(tw.Rendition{
|
|
Symbols: tw.NewSymbols(tw.StyleASCII),
|
|
}),
|
|
)
|
|
table.Header([]string{"Edge", "Type", "Inverse", "BackRef", "Relation", "Unique", "Optional", "Comment"})
|
|
hasEdges := false
|
|
for _, e := range t.Edges {
|
|
hasEdges = true
|
|
err := table.Append([]string{
|
|
e.Name,
|
|
e.Type.Name,
|
|
strconv.FormatBool(e.IsInverse()),
|
|
e.Inverse,
|
|
e.Rel.Type.String(),
|
|
strconv.FormatBool(e.Unique),
|
|
strconv.FormatBool(e.Optional),
|
|
e.Comment(),
|
|
})
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
if hasEdges {
|
|
err := table.Render()
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
io.WriteString(p, strings.ReplaceAll(b.String(), "\n", "\n\t")+"\n")
|
|
}
|