1
0
mirror of https://github.com/ManyakRus/crud_generator.git synced 2024-12-16 21:52:42 +02:00
crud_generator/internal/create_files/create_files.go
2024-02-21 13:29:56 +03:00

1307 lines
35 KiB
Go

package create_files
import (
"errors"
"fmt"
"github.com/ManyakRus/crud_generator/internal/config"
"github.com/ManyakRus/crud_generator/internal/constants"
"github.com/ManyakRus/crud_generator/internal/mini_func"
"github.com/ManyakRus/crud_generator/internal/types"
"github.com/ManyakRus/starter/log"
"github.com/ManyakRus/starter/micro"
"github.com/iancoleman/strcase"
"github.com/jinzhu/inflection"
"strconv"
"strings"
)
// FindSingularName - возвращает наименование в единственном числе
func FindSingularName(s string) string {
var Otvet string
if s == "" {
return Otvet
}
Otvet = inflection.Singular(s)
return Otvet
}
// FormatName - возвращает наименование в формате CamelCase
func FormatName(Name string) string {
Otvet := Name
Otvet1, ok := types.MapReplaceName[Name]
if ok == true {
return Otvet1
}
switch strings.ToLower(Name) {
case "id":
Otvet = "ID"
return Otvet
}
Otvet = strcase.ToCamel(Otvet)
//_id в конце заменяем на ID
lenName := len(Name)
lenOtvet := len(Otvet)
if lenName >= 3 {
last3 := strings.ToLower(Name[lenName-3:])
if last3 == "_id" {
Otvet = Otvet[:lenOtvet-2] + "ID"
}
}
return Otvet
}
// DeleteFuncFromFuncName - удаляет функцию из текста начиная с объявления функции
func DeleteFuncFromFuncName(Text, FuncName string) string {
Otvet := Text
TextFind := "\nfunc " + FuncName + "("
pos1 := strings.Index(Otvet, TextFind)
if pos1 < 0 {
return Otvet
}
s2 := Otvet[pos1+1:]
posEnd := strings.Index(s2, "\n}")
if posEnd < 0 {
return Otvet
}
Otvet = Otvet[:pos1-1] + Otvet[pos1+posEnd+3:]
return Otvet
}
// DeleteFuncFromComment - удаляет функцию из текста начиная с комментария
func DeleteFuncFromComment(Text, Comment string) string {
Otvet := Text
TextFind := Comment
pos1 := strings.Index(Otvet, TextFind)
if pos1 < 0 {
return Otvet
}
s2 := Otvet[pos1+1:]
posEnd := strings.Index(s2, "\n}")
if posEnd < 0 {
return Otvet
}
Otvet = Otvet[:pos1-1] + Otvet[pos1+posEnd+3:]
return Otvet
}
// DeleteCommentFromFuncName - удаляет комментарий с названием функции
func DeleteCommentFromFuncName(Text, FuncName string) string {
Otvet := Text
TextFind := "//" + FuncName + "()"
Otvet = strings.ReplaceAll(Otvet, TextFind, "")
return Otvet
}
// DeleteLineWithComment - удаляет текст от комментария до конца строки
func DeleteLineWithComment(Text, FuncName string) string {
Otvet := Text
TextFind := "//" + FuncName + "()"
pos1 := strings.Index(Otvet, TextFind)
if pos1 < 0 {
return Otvet
}
s2 := Otvet[pos1:]
posEnd := strings.Index(s2, "\n")
if posEnd < 0 {
return Otvet
}
Otvet = Otvet[:pos1-1] + Otvet[pos1+posEnd+1:]
return Otvet
}
// FindPrimaryKeyNameTypeGo - возвращает наименование и тип golang колонки PrimaryKey
func FindPrimaryKeyNameTypeGo(Table1 *types.Table) (string, string) {
Otvet := ""
Type := ""
for _, Column1 := range Table1.MapColumns {
if Column1.IsIdentity == true {
return Column1.NameGo, Column1.TypeGo
}
}
return Otvet, Type
}
// FindPrimaryKeyNameType - возвращает наименование и тип БД колонки PrimaryKey
func FindPrimaryKeyNameType(Table1 *types.Table) (string, string) {
Otvet := ""
Type := ""
for _, Column1 := range Table1.MapColumns {
if Column1.IsIdentity == true {
return Column1.Name, Column1.Type
}
}
return Otvet, Type
}
//// FindPrimaryKeyNameTypeGo - возвращает наименование и тип golang колонки PrimaryKey
//func FindPrimaryKeyNameTypeGo(Table1 *types.Table) (string, string) {
// Otvet := ""
// Type := ""
//
// for _, Column1 := range Table1.MapColumns {
// if Column1.IsIdentity == true {
// return Column1.NameGo, Column1.TypeGo
// }
// }
//
// return Otvet, Type
//}
// ReplacePrimaryKeyID - заменяет "ID" на название колонки PrimaryKey
func ReplacePrimaryKeyID(Text string, Table1 *types.Table) string {
Otvet := Text
ColumnName, ColumnTypeGo := FindPrimaryKeyNameTypeGo(Table1)
if mini_func.IsNumberType(ColumnTypeGo) == true {
Otvet = strings.ReplaceAll(Otvet, "Otvet.ID", "Otvet."+ColumnName)
} else if ColumnTypeGo == "string" {
Otvet = strings.ReplaceAll(Otvet, "Otvet.ID == 0", "Otvet."+ColumnName+" == \"\"")
Otvet = strings.ReplaceAll(Otvet, "Otvet.ID", "Otvet."+ColumnName)
}
return Otvet
}
// AddSkipNow - добавляет строку t.SkipNow()
func AddSkipNow(Text string, Table1 *types.Table) string {
Otvet := Text
if Table1.IDMinimum == "" || Table1.IDMinimum == "0" {
TextFind := "(t *testing.T) {"
Otvet = strings.ReplaceAll(Otvet, TextFind, TextFind+"\n\tt.SkipNow() //now rows in DB\n")
}
return Otvet
}
// IsGoodTable - возвращает ошибку если таблица неправильная
func IsGoodTable(Table1 *types.Table) error {
var err error
TableName := Table1.Name
ColumnName, _ := FindPrimaryKeyNameTypeGo(Table1)
if ColumnName == "" {
TextError := fmt.Sprint("Wrong table ", Table1.Name, " error: not found Primary key")
err = errors.New(TextError)
}
if strings.HasPrefix(TableName, "DELETED_") == true {
TextError := fmt.Sprint("Wrong table ", Table1.Name, " error: name = DELETED_")
err = errors.New(TextError)
}
return err
}
// PrintableString - возвращает строку без запрещённых символов
func PrintableString(s string) string {
Otvet := s
Otvet = strconv.Quote(Otvet) //экранирование символов
len1 := len(Otvet)
if len1 > 0 {
Otvet = Otvet[1 : len1-1]
}
return Otvet
}
// Find_Template_DB_Foldername - возвращает путь к папке
func Find_Template_DB_Foldername() string {
Otvet := ""
DirBin := micro.ProgramDir_bin()
DirTemplates := DirBin + config.Settings.TEMPLATE_FOLDERNAME + micro.SeparatorFile()
DirTemplatesDB := DirTemplates + config.Settings.TEMPLATE_FOLDERNAME_CRUD + micro.SeparatorFile()
Otvet = DirTemplatesDB
return Otvet
}
// DeleteImportModel - удаляет лишний импорт модели
func DeleteImportModel(s string) string {
Otvet := s
ModelURL := FindModelURL()
ImportName := micro.LastWord(ModelURL)
pos1 := strings.Index(Otvet, ImportName+".")
if pos1 < 0 {
Otvet = strings.ReplaceAll(Otvet, `"`+ModelURL+`"\n`, "")
Otvet = strings.ReplaceAll(Otvet, `"`+ModelURL+`"`, "")
}
return Otvet
}
// FindModelURL - возвращает URL репозитория с пакетом "model"
func FindModelURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_MODEL
return Otvet
}
// FindGRPCServerURL - возвращает URL репозитория с пакетом "server_grpc"
func FindGRPCServerURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_GRPC_SERVER
return Otvet
}
// FindGRPCClientURL - возвращает URL репозитория с пакетом "client_grpc"
func FindGRPClientURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_GRPC_CLIENT
return Otvet
}
// FindNRPClientURL - возвращает URL репозитория с пакетом "client_nrpc"
func FindNRPClientURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_NRPC_CLIENT
return Otvet
}
// FindGRPCConstantsURL - возвращает URL репозитория с пакетом "client_grpc"
func FindGRPCConstantsURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_GRPC + "/" + "grpc_constants"
return Otvet
}
// FindNRPCServerlURL - возвращает URL репозитория с пакетом "server_nrpc"
func FindNRPCServerlURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_NRPC_SERVER
return Otvet
}
// FindProtobufURL - возвращает URL репозитория с файлом .proto
func FindProtobufURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_GRPC_PROTO
return Otvet
}
// FindTablesURL - возвращает URL репозитория с пакетом "tables"
func FindTablesURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_TABLES
return Otvet
}
// FindTableNameURL - возвращает URL репозитория с пакетом "tables" + TableName
func FindTableNameURL(TableName string) string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_TABLES + "/" + config.Settings.PREFIX_TABLE + TableName
return Otvet
}
// FindProtoURL - возвращает URL репозитория с пакетом "grpc_proto"
func FindProtoURL() string {
Otvet := ""
TEMPLATE_FOLDERNAME_GRPC_PROTO := strings.ToLower(config.Settings.TEMPLATE_FOLDERNAME_GRPC_PROTO)
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + TEMPLATE_FOLDERNAME_GRPC_PROTO + "/" + "grpc_proto"
return Otvet
}
// FindModelTableURL - возвращает URL репозитория model для таблицы TableName
func FindModelTableURL(TableName string) string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_MODEL + "/" + TableName
return Otvet
}
// FindCrudStarterTableURL - возвращает URL репозитория crud_starter для таблицы TableName
func FindCrudStarterTableURL(TableName string) string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_CRUD_STARTER + "/" + constants.STARTER_TABLES_PREFIX + TableName
return Otvet
}
// FindCrudTableURL - возвращает URL репозитория crud для таблицы TableName
func FindCrudTableURL(TableName string) string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_CRUD + "/" + constants.CRUD_TABLES_FREFIX + TableName
return Otvet
}
// FindNRPC_Client_URL - возвращает URL репозитория с пакетом "nrpc_client"
func FindNRPC_Client_URL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_NRPC_CLIENT
return Otvet
}
// FindDBConstantsURL - возвращает URL репозитория с пакетом db "constants"
func FindDBConstantsURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_DB + "/" + "constants"
return Otvet
}
// FindNRPCConstantsURL - возвращает URL репозитория с пакетом db "constants"
func FindNRPCConstantsURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_NRPC + "/" + "nrpc_constants"
return Otvet
}
// FindCrudStarterURL - возвращает URL репозитория с пакетом "crud_starter"
func FindCrudStarterURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_CRUD_STARTER
return Otvet
}
// Find_GRPC_NRPC_URL - возвращает URL репозитория с пакетом "crud_starter"
func Find_GRPC_NRPC_URL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_GRPC_NRPC
return Otvet
}
// FindCalcStructVersionURL - возвращает URL репозитория с пакетом "calc_struct_version"
func FindCalcStructVersionURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDERNAME_DB + "/" + "calc_struct_version"
return Otvet
}
// FindCrudFunctionsURL - возвращает URL репозитория с пакетом crud_functions
func FindCrudFunctionsURL() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/" + config.Settings.TEMPLATE_FOLDER_CRUD_FUNCTIONS
return Otvet
}
// FindTextDefaultGORMValue - возвращает значение по умолчанию для столбца Column1 для тегов в GORM
func FindTextDefaultGORMValue(Column1 *types.Column) string {
var Otvet string
ColumnName := Column1.Name
//ищем в файле настроек nullable.json
is_nullable_config, has_is_nullable_config := types.MapNullableFileds[ColumnName]
//
sValue := ""
Type_go := Column1.TypeGo
if Column1.TableKey != "" && Column1.IsNullable == true && (is_nullable_config == true || has_is_nullable_config == false) {
sValue = "null"
} else {
switch Type_go {
case "string":
sValue = "\\\"\\\""
case "int", "int32", "int64", "float32", "float64", "uint", "uint32", "uint64":
sValue = "0"
case "time.Time":
sValue = "null"
}
}
if sValue != "" {
Otvet = ";default:" + sValue
}
return Otvet
}
// FindTextDefaultValue - возвращает значение по умолчанию для типа
func FindTextDefaultValue(Type_go string) string {
var Otvet string
switch Type_go {
case "string":
Otvet = `""`
case "int", "int32", "int64", "float32", "float64", "uint", "uint32", "uint64":
Otvet = "0"
case "time.Time":
Otvet = "time.Time{}"
case "bool":
Otvet = "false"
}
return Otvet
}
// FindTextDefaultValueSQL - возвращает значение по умолчанию для типа
func FindTextDefaultValueSQL(Type_go string) string {
var Otvet string
switch Type_go {
case "string":
Otvet = `''`
case "int", "int32", "int64", "float32", "float64", "uint", "uint32", "uint64":
Otvet = "0"
case "time.Time":
Otvet = "null"
case "bool":
Otvet = "false"
}
return Otvet
}
// FindURL_Alias - возвращает URL репозитория с пакетом "alias"
func FindURL_Alias() string {
Otvet := ""
if config.Settings.TEMPLATE_FOLDERNAME_ALIAS == "" {
return Otvet
}
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/"
Otvet = Otvet + config.Settings.TEMPLATE_FOLDERNAME_ALIAS
return Otvet
}
// FindURL_Tables - возвращает URL репозитория с пакетом "tables"
func FindURL_Tables() string {
Otvet := ""
Otvet = config.Settings.SERVICE_REPOSITORY_URL + "/"
Otvet = Otvet + config.Settings.TEMPLATE_FOLDERNAME_TABLES
return Otvet
}
// AddImport - добавляет URL в секцию Import, если его там нет
func AddImport(Text, URL string) string {
Otvet := Text
//если уже есть импорт
pos1 := strings.Index(Otvet, `"`+URL+`"`)
if pos1 >= 0 {
return Otvet
}
//
TextFind := "import ("
LenFind := len(TextFind)
pos1 = strings.Index(Otvet, TextFind)
if pos1 < 0 {
log.Error("not found word: import (")
return Otvet
}
Otvet = Otvet[:pos1+LenFind] + "\n\t" + `"` + URL + `"` + Otvet[pos1+LenFind:]
return Otvet
}
// CheckAndAddImport - добавляет URL в секцию Import, если его там нет, если он нужен
func CheckAndAddImport(Text, URL string) string {
Otvet := Text
//проверим используется или нет
ModuleName := micro.LastWord(URL)
pos1 := strings.Index(Otvet, ModuleName+".")
if pos1 < 0 {
return Otvet
}
Otvet = AddImport(Text, URL)
return Otvet
}
// AddImportTime - добавляет покет в секцию Import, если его там нет
func AddImportTime(TextModel string) string {
Otvet := TextModel
//если уже есть импорт
pos1 := strings.Index(Otvet, `"time"`)
if pos1 >= 0 {
return Otvet
}
//
pos1 = strings.Index(Otvet, "import (")
if pos1 < 0 {
log.Error("not found word: import (")
return TextModel
}
Otvet = Otvet[:pos1+8] + "\n\t" + `"time"` + Otvet[pos1+8:]
return Otvet
}
// AddImportTimestamp - добавляет покет в секцию Import, если его там нет
func AddImportTimestamp(TextModel string) string {
Otvet := TextModel
//если уже есть импорт
pos1 := strings.Index(Otvet, `"google.golang.org/protobuf/types/known/timestamppb"`)
if pos1 >= 0 {
return Otvet
}
//
pos1 = strings.Index(Otvet, "import (")
if pos1 < 0 {
log.Error("not found word: import (")
return TextModel
}
Otvet = Otvet[:pos1+8] + "\n\t" + `"google.golang.org/protobuf/types/known/timestamppb"` + Otvet[pos1+8:]
return Otvet
}
// CheckAndAddImportAlias - добавляет покет в секцию Alias, если его там нет
func CheckAndAddImportAlias(TextModel string) string {
Otvet := TextModel
//если уже есть импорт
pos1 := strings.Index(Otvet, `/alias`)
if pos1 >= 0 {
return Otvet
}
//если нету alias
pos1 = strings.Index(Otvet, `alias.`)
if pos1 < 0 {
return Otvet
}
//
pos1 = strings.Index(Otvet, "import (")
if pos1 < 0 {
log.Error("not found word: import (")
return TextModel
}
AliasURL := FindURL_Alias()
Otvet = Otvet[:pos1+8] + "\n\t" + `"` + AliasURL + `"` + Otvet[pos1+8:]
return Otvet
}
// CheckAndAddImportTime_FromTable - добавляет пакет "time" в секцию Import, если его там нет
func CheckAndAddImportTime_FromTable(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
HasTimeColumn := Has_ColumnType_Time(Table1)
if HasTimeColumn == false {
return Otvet
}
Otvet = AddImportTime(Otvet)
return Otvet
}
// CheckAndAddImportTime_FromText - добавляет пакет "time" в секцию Import, если его там нет
func CheckAndAddImportTime_FromText(Text string) string {
Otvet := Text
pos1 := strings.Index(Text, " time.")
if pos1 < 0 {
return Otvet
}
Otvet = AddImportTime(Otvet)
return Otvet
}
// CheckAndAddImportTimestamp_FromText - добавляет пакет "time" в секцию Import, если его там нет
func CheckAndAddImportTimestamp_FromText(Text string) string {
Otvet := Text
pos1 := strings.Index(Text, " timestamppb.")
if pos1 < 0 {
return Otvet
}
Otvet = AddImportTimestamp(Otvet)
return Otvet
}
// DeleteTemplateRepositoryImports - удаляет импорты репозитория шаблона
func DeleteTemplateRepositoryImports(Text string) string {
Otvet := Text
if config.Settings.TEMPLATE_REPOSITORY_URL == "" {
return Otvet
}
//
TextFind := "import ("
pos1 := strings.Index(Otvet, TextFind)
if pos1 < 0 {
log.Error("not found word: import (")
return Otvet
}
TextFind = `"` + config.Settings.TEMPLATE_REPOSITORY_URL
pos1 = strings.Index(Otvet, TextFind)
if pos1 < 0 {
return Otvet
}
s2 := Otvet[pos1:]
posEnd := strings.Index(s2, "\n")
if posEnd < 0 {
return Otvet
}
Otvet = Otvet[:pos1-1] + Otvet[pos1+posEnd+1:]
Otvet = DeleteTemplateRepositoryImports(Otvet)
return Otvet
}
// ReplaceServiceURLImports - заменяет URL репозитория шаблона на URL репозитория сервиса
func ReplaceServiceURLImports(Text string) string {
Otvet := Text
if config.Settings.SERVICE_REPOSITORY_URL == "" {
return Otvet
}
if config.Settings.TEMPLATE_REPOSITORY_URL == "" {
return Otvet
}
if config.Settings.USE_DEFAULT_TEMPLATE == false {
return Otvet
}
//
TextFind := "import ("
pos1 := strings.Index(Otvet, TextFind)
if pos1 < 0 {
return Otvet
}
TEMPLATE_REPOSITORY_URL := config.Settings.TEMPLATE_REPOSITORY_URL
SERVICE_REPOSITORY_URL := config.Settings.SERVICE_REPOSITORY_URL
Otvet = strings.ReplaceAll(Otvet, TEMPLATE_REPOSITORY_URL, SERVICE_REPOSITORY_URL)
return Otvet
}
// ReplaceModelAndTableName - заменяет имя модели и имя таблицы в шаблоне на новые
func ReplaceModelAndTableName(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
Otvet = strings.ReplaceAll(Otvet, config.Settings.TEXT_TEMPLATE_MODEL, Table1.NameGo)
Otvet = strings.ReplaceAll(Otvet, config.Settings.TEXT_TEMPLATE_TABLENAME, Table1.Name)
return Otvet
}
// FindModelComment - возвращает комментарий для модели
func FindModelComment(Table1 *types.Table) string {
Otvet := ""
TableName := Table1.Name
ModelName := Table1.NameGo
COMMENT_MODEL_STRUCT := config.Settings.COMMENT_MODEL_STRUCT
Otvet = `// ` + ModelName + ` - ` + COMMENT_MODEL_STRUCT + TableName + `: ` + Table1.Comment
return Otvet
}
// FindModelNameComment - возвращает комментарий для названия модели
func FindModelNameComment(ModelName string, Table1 *types.Table) string {
Otvet := ""
TableName := Table1.Name
//ModelName := Table1.NameGo
COMMENT_MODEL_STRUCT := config.Settings.COMMENT_MODEL_STRUCT
Otvet = `// ` + ModelName + ` - ` + COMMENT_MODEL_STRUCT + TableName + `: ` + Table1.Comment
return Otvet
}
// ReplacePackageName - заменяет имя пакета в шаблоне на новое
func ReplacePackageName(Text, PackageName string) string {
Otvet := Text
//найдём имя каталога, это будет имя пакета
PackageName = micro.DeleteEndSlash(PackageName)
PackageName = micro.LastWord(PackageName)
//
TextFind := "package "
pos1 := strings.Index(Otvet, TextFind)
if pos1 < 0 {
log.Error("not found word: package ")
return Otvet
}
s2 := Otvet[pos1:]
posEnd := strings.Index(s2, "\n")
if posEnd < 0 {
log.Error("not found word: \n")
return Otvet
}
Otvet = Otvet[:pos1+len(TextFind)] + PackageName + Otvet[pos1+posEnd:]
return Otvet
}
// DeleteEmptyImport - удаляет пустой импорт
func DeleteEmptyImport(Text string) string {
Otvet := Text
sFind := `import (
)
`
Otvet = strings.ReplaceAll(Otvet, sFind, "")
return Otvet
}
// DeleteFuncDelete - удаляет функцию Delete()
func DeleteFuncDelete(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
//проверим есть ли колонка IsDeleted
if Has_Column_IsDeleted(Table1) == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// Delete ")
Otvet = DeleteFuncFromFuncName(Otvet, "Delete")
return Otvet
}
// DeleteFuncRestore - удаляет функцию Restore()
func DeleteFuncRestore(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
//проверим есть ли колонка IsDeleted
if Has_Column_IsDeleted(Table1) == true && config.Settings.HAS_IS_DELETED == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// Restore ")
Otvet = DeleteFuncFromFuncName(Otvet, "Restore")
return Otvet
}
// DeleteFuncFind_byExtID - удаляет функцию Find_ByExtID()
func DeleteFuncFind_byExtID(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
//проверка есть ли колонки ExtID и ConnectionID
if Has_Column_ExtID_ConnectionID(Table1) == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// Find_ByExtID ")
Otvet = DeleteFuncFromFuncName(Otvet, "Find_ByExtID")
return Otvet
}
// DeleteFuncDeleteCtx - удаляет функцию Delete_ctx()
func DeleteFuncDeleteCtx(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
//проверим есть ли колонка IsDeleted
if Has_Column_IsDeleted(Table1) == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// Delete_ctx ")
Otvet = DeleteFuncFromFuncName(Otvet, "Delete_ctx")
return Otvet
}
// DeleteFuncRestoreCtx - удаляет функцию Restore_ctx()
func DeleteFuncRestoreCtx(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
//проверим есть ли колонка IsDeleted
if Has_Column_IsDeleted(Table1) == true && config.Settings.HAS_IS_DELETED == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// Restore_ctx ")
Otvet = DeleteFuncFromFuncName(Otvet, "Restore_ctx")
return Otvet
}
// DeleteFuncFind_byExtIDCtx - удаляет функцию Find_ByExtID_ctx()
func DeleteFuncFind_byExtIDCtx(TextModel string, Table1 *types.Table) string {
Otvet := TextModel
//проверка есть ли колонки ExtID и ConnectionID
if Has_Column_ExtID_ConnectionID(Table1) == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// Find_ByExtID_ctx ")
Otvet = DeleteFuncFromFuncName(Otvet, "Find_ByExtID_ctx")
return Otvet
}
// DeleteFuncTestDelete - удаляет функцию Delete()
func DeleteFuncTestDelete(Text string, Table1 *types.Table) string {
Otvet := Text
//проверим есть ли колонка IsDeleted
if Has_Column_IsDeleted(Table1) == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// TestDelete ")
Otvet = DeleteFuncFromFuncName(Otvet, "TestDelete")
return Otvet
}
// DeleteFuncTestRestore - удаляет функцию Restore()
func DeleteFuncTestRestore(Text string, Table1 *types.Table) string {
Otvet := Text
//проверим есть ли колонка IsDeleted
if Has_Column_IsDeleted(Table1) == true && config.Settings.HAS_IS_DELETED == true {
return Otvet
}
Otvet = DeleteFuncFromComment(Otvet, "\n// TestRestore ")
Otvet = DeleteFuncFromFuncName(Otvet, "TestRestore")
return Otvet
}
// DeleteFuncFind_byExtID - удаляет функцию Find_ByExtID()
func DeleteFuncTestFind_byExtID(Text string, Table1 *types.Table) string {
Otvet := Text
//проверка есть ли колонки ExtID и ConnectionID
if Has_Column_ExtID_ConnectionID(Table1) == true {
return Otvet
}
//
Otvet = DeleteFuncFromComment(Otvet, "\n// TestFind_ByExtID ")
Otvet = DeleteFuncFromFuncName(Otvet, "TestFind_ByExtID")
return Otvet
}
// DeleteEmptyLines - удаляет пустые строки
func DeleteEmptyLines(Text string) string {
Otvet := Text
Otvet = strings.ReplaceAll(Otvet, "\n\n\n", "\n\n")
Otvet = strings.ReplaceAll(Otvet, "\n//\n\n", "\n\n")
Otvet = strings.ReplaceAll(Otvet, "\n\t//\n\n", "\n\n")
//Otvet = strings.ReplaceAll(Otvet, "\r\r", "\r")
//Otvet = strings.ReplaceAll(Otvet, "\r\n", "\n")
//Otvet = strings.ReplaceAll(Otvet, "}\n\n", "}\n")
pos1 := strings.Index(Otvet, "\n\n\n")
if pos1 >= 0 {
Otvet = DeleteEmptyLines(Otvet)
}
//удалим последние 2 пустые строки
HasSuffix := strings.HasSuffix(Otvet, "\n\n")
if HasSuffix == true {
Otvet = Otvet[:len(Otvet)-1]
}
return Otvet
}
// DeleteLastUnderline - удаляет последний символ подчёркивания
func DeleteLastUnderline(s string) string {
Otvet := s
if s == "" {
return Otvet
}
Otvet = strings.TrimSuffix(Otvet, "_")
return Otvet
}
// FindLastGoodPos - возвращает позицию последнего нахождения, с новой строки
func FindLastGoodPos(s, TextFind string) int {
Otvet := -1
pos1 := strings.LastIndex(s, TextFind)
if pos1 < 0 {
return Otvet
}
pos2 := strings.Index(s[pos1:], "\n")
if pos2 < 0 {
return Otvet
}
Otvet = pos1 + pos2 + 1
return Otvet
}
// AddInterfaceFunction - добавляет функцию в интерфейс
func AddInterfaceFunction(s, TextAdd string) string {
Otvet := s
//Проверим такая функция уже есть
pos1 := strings.Index(Otvet, TextAdd)
if pos1 >= 0 {
return Otvet
}
//найдём начало интефейса
sFind := " interface {"
pos1 = FindLastGoodPos(Otvet, sFind)
if pos1 < 0 {
log.Error("FindLastGoodPos() error: not found: ", sFind)
return Otvet
}
s2 := Otvet[pos1:]
pos2 := strings.Index(s2, "\n}")
if pos2 < 0 {
log.Error("FindLastGoodPos() error: not found: \\n")
return Otvet
}
PosStart := pos1 + pos2
Otvet = Otvet[:PosStart] + TextAdd + Otvet[PosStart:]
return Otvet
}
// FindTextProtobufRequest - возвращает имя message из .proto, в зависимости от типа, а также название поля
func FindTextProtobufRequest(TypeGo string) (string, string) {
Otvet := "RequestID"
TextRequestFieldName := "ID"
switch TypeGo {
case "int", "int64":
{
Otvet = "RequestId"
TextRequestFieldName = "ID"
}
case "int32":
{
Otvet = "RequestInt32"
TextRequestFieldName = "Int32"
}
case "string":
{
Otvet = "RequestString"
TextRequestFieldName = "StringFind"
}
case "time.Time":
{
Otvet = "RequestDate"
TextRequestFieldName = "Date"
}
case "float32":
{
Otvet = "RequestFloat32"
TextRequestFieldName = "Float32"
}
case "float64":
{
Otvet = "RequestFloat64"
TextRequestFieldName = "Float64"
}
case "bool":
{
Otvet = "RequestBool"
TextRequestFieldName = "Bool"
}
}
return Otvet, TextRequestFieldName
}
// FindTextProtobufRequest_ID_Type - возвращает имя message из .proto для двух параметров ID + Type,в зависимости от типа, а также название поля
// возвращает:
// Otvet - имя message из .proto
// TextRequestFieldName - название поля в Request
// TextRequestFieldGolang - название поля в Request с преобразованием в тип гоу
func FindTextProtobufRequest_ID_Type(Table1 *types.Table, Column1 *types.Column, VariableName string) (string, string, string) {
Otvet := "RequestID"
TextRequestFieldName := "ID"
TextRequestFieldGolang := "ID"
TypeGo := Column1.TypeGo
TableName := Table1.Name
ColumnName := Column1.Name
switch TypeGo {
case "int", "int64":
{
Otvet = "Request_ID_Int64"
TextRequestFieldName = "Int64"
TextRequestFieldGolang = VariableName + "Int64"
}
case "int32":
{
Otvet = "Request_ID_Int32"
TextRequestFieldName = "Int32"
TextRequestFieldGolang = VariableName + "Int32"
}
case "string":
{
Otvet = "Request_ID_String"
TextRequestFieldName = "StringFind"
TextRequestFieldGolang = VariableName + "StringFind"
}
case "time.Time":
{
Otvet = "Request_ID_Date"
TextRequestFieldName = "Date"
TextRequestFieldGolang = VariableName + "Date.AsTime()"
}
case "float32":
{
Otvet = "Request_ID_Float32"
TextRequestFieldName = "Float32"
TextRequestFieldGolang = VariableName + "Float32"
}
case "float64":
{
Otvet = "Request_ID_Float64"
TextRequestFieldName = "Float64"
TextRequestFieldGolang = VariableName + "Float64"
}
case "bool":
{
Otvet = "Request_ID_Bool"
TextRequestFieldName = "Bool"
TextRequestFieldGolang = VariableName + "Bool"
}
}
TextConvert, ok := types.MapConvertID[TableName+"."+ColumnName]
if ok == true {
TextRequestFieldGolang = TextConvert + "(" + VariableName + TextRequestFieldName + ")"
}
return Otvet, TextRequestFieldName, TextRequestFieldGolang
}
// ConvertIdToAlias - заменяет ID на Alias
func ConvertIdToAlias(Text string, Table1 *types.Table) string {
Otvet := Text
TableName := Table1.Name
IDName, _ := FindPrimaryKeyNameType(Table1)
TextConvert, ok := types.MapConvertID[TableName+"."+IDName]
if ok == false {
return Otvet
}
Otvet = strings.ReplaceAll(Otvet, "Request.ID", TextConvert+"(Request.ID)")
if TextConvert[:6] != "alias." {
return Otvet
}
URL := FindURL_Alias()
if URL == "" {
return Otvet
}
Otvet = AddImport(Otvet, URL)
return Otvet
}
// FindTextConvertGolangTypeToProtobufType - возвращает имя переменной + имя колонки, преобразованное в тип protobuf
func FindTextConvertGolangTypeToProtobufType(Table1 *types.Table, Column1 *types.Column, VariableName string) string {
Otvet := VariableName + Column1.NameGo
//TableName := Table1.Name
//IDName := Column1.Name
////alias в Int64
//TextConvert, ok := types.MapConvertID[TableName+"."+IDName]
//if ok == true {
// Otvet = TextConvert + "(" + VariableName + Column1.NameGo + ")"
//}
//time.Time в timestamppb
//if Column1.TypeGo == "time.Time" {
// Otvet = "timestamppb.New(" + VariableName + Column1.NameGo + ")"
//}
//преобразуем alias в обычный тип, и дату в timestamp
switch Column1.TypeGo {
case "time.Time":
Otvet = "timestamppb.New(" + VariableName + Column1.NameGo + ")"
case "string":
Otvet = "string(" + VariableName + Column1.NameGo + ")"
case "int64":
Otvet = "int64(" + VariableName + Column1.NameGo + ")"
case "int32":
Otvet = "int32(" + VariableName + Column1.NameGo + ")"
case "bool":
Otvet = "bool(" + VariableName + Column1.NameGo + ")"
case "float32":
Otvet = "float32(" + VariableName + Column1.NameGo + ")"
case "float64":
Otvet = "float64(" + VariableName + Column1.NameGo + ")"
}
return Otvet
}
// FindTextConvertProtobufTypeToGolangType - возвращает имя переменной + имя колонки, преобразованное в тип golang из protobuf
func FindTextConvertProtobufTypeToGolangType(Table1 *types.Table, Column1 *types.Column, VariableName string) string {
Otvet := VariableName + Column1.NameGo
TableName := Table1.Name
IDName := Column1.Name
//alias в Int64
TextConvert, ok := types.MapConvertID[TableName+"."+IDName]
if ok == true {
Otvet = TextConvert + "(" + VariableName + Column1.NameGo + ")"
return Otvet
}
//time.Time в timestamppb
if Column1.TypeGo == "time.Time" {
Otvet = VariableName + Column1.NameGo + ".AsTime()"
return Otvet
}
////преобразуем alias в обычный тип, и дату в timestamp
//switch Column1.TypeGo {
//case "time.Time":
// Otvet = "timestamppb.New(" + VariableName + Column1.NameGo + ")"
//case "string":
// Otvet = "string(" + VariableName + Column1.NameGo + ")"
//case "int64":
// Otvet = "int64(" + VariableName + Column1.NameGo + ")"
//case "int32":
// Otvet = "int32(" + VariableName + Column1.NameGo + ")"
//case "bool":
// Otvet = "bool(" + VariableName + Column1.NameGo + ")"
//case "float32":
// Otvet = "float32(" + VariableName + Column1.NameGo + ")"
//case "float64":
// Otvet = "float64(" + VariableName + Column1.NameGo + ")"
//}
return Otvet
}
// FindTextEqualEmpty - находит текст сравнение с пустым значением
func FindTextEqualEmpty(Column1 *types.Column, VariableName string) string {
Otvet := ""
DefaultValue := FindTextDefaultValue(Column1.TypeGo)
Otvet = VariableName + " == " + DefaultValue
if DefaultValue == "time.Time{}" {
Otvet = VariableName + ".IsZero() == true"
}
return Otvet
}
//// AddSkipNowEveryFunc - добавляет функцию SkipNow() для каждой тестовой функции
//func AddSkipNowEveryFunc(Text string) string {
// Otvet := ""
//
// sFind := "\nfunc "
// Mass := make([]string, 0)
// Mass = strings.Split(Text, sFind)
// for _, v := range Mass {
// pos1 := strings.Index(v, sFind)
// if pos1 < 0 {
// continue
// }
//
// s2 := Text[pos1:]
// pos2 := strings.Index(s2, "\n")
// if pos2 < 0 {
// continue
// }
// v = v[:pos1+pos2] + "\n\tt.SkipNow() //нет строк в БД \n" + v[pos1+pos2:]
// Otvet = Otvet + v
// }
//
// return Otvet
//}