You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

250 lines
5.7 KiB
Go

package genstruct
import (
"fmt"
5 years ago
"sort"
"strings"
5 years ago
"github.com/xxjwxc/gormt/data/config"
"github.com/xxjwxc/gormt/data/view/cnf"
"github.com/xxjwxc/gormt/data/view/generate"
)
// SetName Setting element name.设置元素名字
func (e *GenElement) SetName(name string) {
e.Name = name
}
// SetType Setting element type.设置元素类型
func (e *GenElement) SetType(tp string) {
e.Type = tp
}
// SetNotes Setting element notes.设置注释
func (e *GenElement) SetNotes(notes string) {
e.Notes = strings.Replace(notes, "\n", ",", -1)
}
// AddTag Add a tag .添加一个tag标记
func (e *GenElement) AddTag(k string, v string) {
if e.Tags == nil {
e.Tags = make(map[string][]string)
}
e.Tags[k] = append(e.Tags[k], v)
}
// Generate Get the result data.获取结果数据
func (e *GenElement) Generate() string {
tag := ""
if e.Tags != nil {
5 years ago
var ks []string
for k := range e.Tags {
ks = append(ks, k)
}
sort.Strings(ks)
var tags []string
for _, v := range ks {
tags = append(tags, fmt.Sprintf(`%v:"%v"`, v, strings.Join(e.Tags[v], ";")))
}
5 years ago
tag = fmt.Sprintf("`%v`", strings.Join(tags, " "))
}
var p generate.PrintAtom
if len(e.Notes) > 0 {
p.Add(e.Name, e.Type, tag, "// "+e.Notes)
} else {
p.Add(e.Name, e.Type, tag)
}
return p.Generates()[0]
}
5 years ago
// GenerateColor Get the result data.获取结果数据
func (e *GenElement) GenerateColor() string {
tag := ""
if e.Tags != nil {
var ks []string
for k := range e.Tags {
ks = append(ks, k)
}
sort.Strings(ks)
var tags []string
for _, v := range ks {
tags = append(tags, fmt.Sprintf(`%v:"%v"`, v, strings.Join(e.Tags[v], ";")))
}
tag = fmt.Sprintf("`%v`", strings.Join(tags, " "))
}
var p generate.PrintAtom
if len(e.Notes) > 0 {
p.Add(e.Name, "\033[32;1m "+e.Type+" \033[0m", "\033[31;1m "+tag+" \033[0m", "\033[32;1m // "+e.Notes+" \033[0m")
} else {
p.Add(e.Name, "\033[32;1m "+e.Type+" \033[0m", "\033[31;1m "+tag+" \033[0m")
}
return p.Generates()[0]
}
//////////////////////////////////////////////////////////////////////////////
// struct
//////////////////////////////////////////////////////////////////////////////
// SetCreatTableStr Set up SQL create statement, backup use setup create statement, backup use.设置创建语句,备份使用
5 years ago
func (s *GenStruct) SetCreatTableStr(sql string) {
5 years ago
s.SQLBuildStr = sql
5 years ago
}
// SetStructName Setting the name of struct.设置struct名字
func (s *GenStruct) SetStructName(name string) {
s.Name = name
}
// SetNotes set the notes.设置注释
5 years ago
func (s *GenStruct) SetNotes(notes string) {
if len(notes) == 0 {
notes = "[...]" // default of struct notes(for export ).struct 默认注释(为了导出注释)
}
notes = s.Name + " " + notes
5 years ago
a := strings.Split(notes, "\n")
var text []string
for _, v := range a {
// if len(v) > 0 {
text = append(text, "// "+v)
// }
5 years ago
}
s.Notes = strings.Join(text, "\r\n")
}
// AddElement Add one or more elements.添加一个/或多个元素
5 years ago
func (s *GenStruct) AddElement(e ...GenElement) {
s.Em = append(s.Em, e...)
}
// Generates Get the result data.获取结果数据
func (s *GenStruct) Generates() []string {
var p generate.PrintAtom
if config.GetIsOutSQL() {
5 years ago
p.Add("/******sql******")
5 years ago
p.Add(s.SQLBuildStr)
5 years ago
p.Add("******sql******/")
}
5 years ago
p.Add(s.Notes)
p.Add("type", s.Name, "struct {")
mp := make(map[string]bool, len(s.Em))
for _, v := range s.Em {
if !mp[v.Name] {
mp[v.Name] = true
p.Add(v.Generate())
}
}
p.Add("}")
return p.Generates()
}
5 years ago
// \033[3%d;%dm -%d;%d-colors!\033[0m\n
// GeneratesColor Get the result data on color.获取结果数据 带颜色
func (s *GenStruct) GeneratesColor() []string {
var p generate.PrintAtom
if config.GetIsOutSQL() {
p.Add("\033[32;1m /******sql******\033[0m")
p.Add(s.SQLBuildStr)
p.Add("\033[32;1m ******sql******/ \033[0m")
}
p.Add("\033[32;1m " + s.Notes + " \033[0m")
p.Add("\033[34;1m type \033[0m", s.Name, "\033[34;1m struct \033[0m {")
mp := make(map[string]bool, len(s.Em))
5 years ago
for _, v := range s.Em {
if !mp[v.Name] {
mp[v.Name] = true
p.Add(" \t\t" + v.GenerateColor())
}
5 years ago
}
p.Add(" }")
return p.Generates()
}
//////////////////////////////////////////////////////////////////////////////
// package
//////////////////////////////////////////////////////////////////////////////
// SetPackage Defining package names.定义包名
func (p *GenPackage) SetPackage(pname string) {
p.Name = pname
}
// AddImport Add import by type.通过类型添加import
func (p *GenPackage) AddImport(imp string) {
if p.Imports == nil {
p.Imports = make(map[string]string)
}
p.Imports[imp] = imp
}
// AddStruct Add a structure.添加一个结构体
func (p *GenPackage) AddStruct(st GenStruct) {
p.Structs = append(p.Structs, st)
}
// Generate Get the result data.获取结果数据
func (p *GenPackage) Generate() string {
p.genimport() // auto add import .补充 import
var pa generate.PrintAtom
pa.Add("package", p.Name)
// add import
if p.Imports != nil {
pa.Add("import (")
for _, v := range p.Imports {
pa.Add(v)
}
pa.Add(")")
}
// -----------end
// add struct
for _, v := range p.Structs {
for _, v1 := range v.Generates() {
pa.Add(v1)
}
}
// -----------end
// add func
for _, v := range p.FuncStrList {
pa.Add(v)
}
// -----------end
// output.输出
strOut := ""
for _, v := range pa.Generates() {
strOut += v + "\n"
}
return strOut
}
5 years ago
// AddFuncStr add func coding string.添加函数串
func (p *GenPackage) AddFuncStr(src string) {
p.FuncStrList = append(p.FuncStrList, src)
}
// compensate and import .获取结果数据
func (p *GenPackage) genimport() {
for _, v := range p.Structs {
for _, v1 := range v.Em {
if v2, ok := cnf.EImportsHead[v1.Type]; ok {
if len(v2) > 0 {
p.AddImport(v2)
}
}
}
}
}