github.com/easysoft/zendata@v0.0.0-20240513203326-705bd5a7fd67/internal/server/service/common.go

package serverService

import (
	"fmt"
	"strings"

	consts "github.com/easysoft/zendata/internal/pkg/const"
	"github.com/easysoft/zendata/internal/pkg/domain"
	"github.com/easysoft/zendata/internal/pkg/model"
	logUtils "github.com/easysoft/zendata/pkg/utils/log"
	"github.com/easysoft/zendata/pkg/utils/vari"
)

func genFieldFromZdField(treeNode model.ZdField, refer model.ZdRefer, field *domain.DefField) {
	field.Field = treeNode.Field
	field.Note = treeNode.Note

	field.Range = treeNode.Range
	field.Value = treeNode.Exp
	field.Prefix = treeNode.Prefix
	field.Postfix = treeNode.Postfix
	field.Divider = treeNode.Divider
	field.Loop = treeNode.Loop
	field.Loopfix = treeNode.Loopfix
	field.Format = treeNode.Format

	field.Type = treeNode.Type
	field.Mode = treeNode.Mode
	if field.Type == consts.FieldTypeList {
		field.Type = ""
	}
	if field.Mode == consts.ModeParallel {
		field.Mode = ""
	}

	field.Length = treeNode.Length
	field.LeftPad = treeNode.LeftPad
	field.RightPad = treeNode.RightPad
	field.Rand = treeNode.Rand

	field.Config = treeNode.Config
	field.Use = treeNode.Use
	field.From = treeNode.From
	field.Select = treeNode.Select
	field.Where = treeNode.Where
	field.Limit = treeNode.Limit

	if field.Range != "" {
		return
	}

	// deal with refer
	if refer.Type != "" {
		logUtils.PrintTo(refer.Type)

		if refer.Type == "excel" {
			field.From = refer.File
			field.Select = refer.ColName
			field.Where = refer.Condition

		} else if refer.Type == consts.ResTypeRanges || refer.Type == consts.ResTypeInstances { // medium{2}
			arr := strings.Split(refer.ColName, ",")
			arrNew := make([]string, 0)

			for _, item := range arr {
				if refer.Count > 0 {
					item = fmt.Sprintf("%s{%d}", item, refer.Count)
				}

				arrNew = append(arrNew, item)
			}

			field.From = refer.File
			field.Use = strings.Join(arrNew, ",")

		} else if refer.Type == consts.ResTypeYaml { // dir/content.yaml{3}
			arr := strings.Split(refer.File, ",")
			arrNew := make([]string, 0)
			for _, item := range arr {
				if refer.Count > 0 {
					item = fmt.Sprintf("%s{%d}", item, refer.Count)
				}

				arrNew = append(arrNew, item)
			}
			field.Range = strings.Join(arrNew, ",")

		} else if refer.Type == consts.ResTypeText { // dir/users.txt:2,dir/file.txt:3
			arr := strings.Split(refer.File, ",")
			arrNew := make([]string, 0)
			for _, item := range arr {
				if refer.Rand {
					item = fmt.Sprintf("%s:R", item)
				} else if refer.Step > 1 {
					item = fmt.Sprintf("%s:%d", item, refer.Step)
				}

				arrNew = append(arrNew, item)
			}
			field.Range = strings.Join(arrNew, ",")

		} else if refer.Type == consts.ResTypeConfig { // dir/users.txt:2
			field.Config = fmt.Sprintf("%s", refer.File)

		} else if refer.Type == consts.ResTypeValue { //
			field.Value = fmt.Sprintf("%s", refer.Value)

		}
	}
}

func GetRelatedPathWithResDir(p string) (ret string) {
	rpl := vari.WorkDir + consts.ResDirYaml + consts.PthSep
	ret = strings.Replace(p, rpl, "", 1)

	rpl = vari.WorkDir + consts.ResDirUsers + consts.PthSep
	ret = strings.Replace(ret, rpl, "", 1)

	return
}

//func instancesItemToResInstForExport(item model.ZdInstancesItem) (inst model.ResInstancesItem) {
//	inst.Note = item.Note
//
//	for _, child := range item.Fields {
//		childField := model.DefField{}
//		instancesItemToResFieldForExport(*child, &childField)
//
//		inst.Fields = append(inst.Fields, childField)
//	}
//
//	if len(inst.Fields) == 0 {
//		inst.Fields = nil
//	}
//	if len(inst.Froms) == 0 {
//		inst.Froms = nil
//	}
//
//	return
//}
//func instancesItemToResFieldForExport(item model.ZdInstancesItem, field *model.DefField) {
//	for _, item := range item.Fields {
//		childField := model.DefField{}
//		instancesItemToResFieldForExport(*item, &childField)
//
//		field.Fields = append(field.Fields, childField)
//	}
//
//	for _, from := range item.Froms { // only one level
//		childField := model.DefField{}
//		genFieldFromZdInstancesItem(*from, &childField)
//
//		field.Froms = append(field.Froms, childField)
//	}
//
//	if len(field.Fields) == 0 {
//		field.Fields = nil
//	}
//	if len(field.Froms) == 0 {
//		field.Froms = nil
//	}
//}
//}
//func genFieldFromZdInstancesItem(item model.ZdInstancesItem, field *model.DefField) () {
//	field.Field = item.Field
//	field.Note = item.Note
//
//	field.Range = item.Range
//	field.Value = item.Exp
//	field.Prefix = item.Prefix
//	field.Postfix = item.Postfix
//	field.Loop = item.Loop
//	field.Loopfix = item.Loopfix
//	field.Format = item.Format
//	field.DefType = item.DefType
//	field.Mode = item.Mode
//	field.Length = item.Length
//	field.LeftPad = item.LeftPad
//	field.RightPad = item.RightPad
//	field.Rand = item.Rand
//
//	field.Config = item.Config
//	field.Use = item.Use
//	field.From = item.From
//	field.Select = item.Select
//	field.Where = item.Where
//	field.Limit = item.Limit
//}