hk4e/gdconf/game_data_config_test.go

package gdconf

import (
	"encoding/json"
	"image"
	"image/color"
	"image/jpeg"
	"log"
	"os"
	"strings"
	"testing"
	"time"

	"hk4e/common/config"
	"hk4e/pkg/logger"

	"github.com/hjson/hjson-go/v4"
)

type TableField struct {
	FieldName  string `json:"field_name"`
	FieldType  string `json:"field_type"`
	OriginName string `json:"origin_name"`
}

type TableStructMapping struct {
	TableName string        `json:"table_name"`
	FieldList []*TableField `json:"field_list"`
}

// 生成最终服务器读取的配置表
func TestGenGdCsv(t *testing.T) {
	tableStructMappingList := make([]*TableStructMapping, 0)
	configFileData, err := os.ReadFile("./table_struct_mapping.json")
	if err != nil {
		log.Printf("open config file error: %v", err)
		return
	}
	err = json.Unmarshal(configFileData, &tableStructMappingList)
	if err != nil {
		log.Printf("parse config file error: %v", err)
		return
	}
	for _, tableStructMapping := range tableStructMappingList {
		txtFileData, err := os.ReadFile("./game_data_config/txt/" + tableStructMapping.TableName + ".txt")
		if err != nil {
			log.Printf("read txt file error: %v", err)
			continue
		}
		// 转换txt配置表格式为csv
		originCsv := string(txtFileData)
		originCsv = strings.ReplaceAll(originCsv, "\r\n", "\n")
		originCsv = strings.ReplaceAll(originCsv, "\r", "\n")
		originCsv = strings.ReplaceAll(originCsv, ",", "#")
		originCsv = strings.ReplaceAll(originCsv, ";", "#")
		originCsv = strings.ReplaceAll(originCsv, "\t", ",")
		originCsvLineList := strings.Split(originCsv, "\n")
		if len(originCsvLineList) == 0 {
			log.Printf("origin csv file is empty")
			continue
		}
		originCsvHeadList := strings.Split(originCsvLineList[0], ",")
		if len(originCsvHeadList) == 0 {
			log.Printf("origin csv file head is empty")
			continue
		}
		fieldNameHead := ""
		fieldTypeHead := ""
		for index, originCsvHead := range originCsvHeadList {
			for _, tableField := range tableStructMapping.FieldList {
				if originCsvHead == tableField.OriginName {
					// 字段名匹配成功
					fieldNameHead += tableField.FieldName
					fieldTypeHead += tableField.FieldType
				}
			}
			if index < len(originCsvHeadList)-1 {
				fieldNameHead += ","
				fieldTypeHead += ","
			}
		}
		fieldNameHead += "\n"
		fieldTypeHead += "\n"
		gdCsvFile := fieldNameHead + fieldTypeHead + originCsv
		err = os.WriteFile("./game_data_config/csv/"+tableStructMapping.TableName+".csv", []byte(gdCsvFile), 0644)
		if err != nil {
			log.Printf("write gd csv file error: %v", err)
			continue
		}
	}
}

// 测试初始化加载配置表
func TestInitGameDataConfig(t *testing.T) {
	config.InitConfig("./bin/application.toml")
	logger.InitLogger("InitGameDataConfig")
	logger.Info("start load conf")
	InitGameDataConfig()
	logger.Info("load conf finish, conf: %v", CONF)
	time.Sleep(time.Second)
}

func CheckJsonLoop(path string, errorJsonFileList *[]string, totalJsonFileCount *int) {
	fileList, err := os.ReadDir(path)
	if err != nil {
		logger.Error("open dir error: %v", err)
		return
	}
	for _, file := range fileList {
		fileName := file.Name()
		if file.IsDir() {
			CheckJsonLoop(path+"/"+fileName, errorJsonFileList, totalJsonFileCount)
		}
		if split := strings.Split(fileName, "."); split[len(split)-1] != "json" {
			continue
		}
		fileData, err := os.ReadFile(path + "/" + fileName)
		if err != nil {
			logger.Error("open file error: %v", err)
			continue
		}
		var obj any
		err = hjson.Unmarshal(fileData, &obj)
		if err != nil {
			*errorJsonFileList = append(*errorJsonFileList, path+"/"+fileName+", err: "+err.Error())
		}
		*totalJsonFileCount++
	}
}

// 测试加载json配置
func TestCheckJsonValid(t *testing.T) {
	config.InitConfig("./bin/application.toml")
	logger.InitLogger("CheckJsonValid")
	errorJsonFileList := make([]string, 0)
	totalJsonFileCount := 0
	CheckJsonLoop("./game_data_config/json", &errorJsonFileList, &totalJsonFileCount)
	for _, v := range errorJsonFileList {
		logger.Info("%v", v)
	}
	logger.Info("err json file count: %v, total count: %v", len(errorJsonFileList), totalJsonFileCount)
	time.Sleep(time.Second)
}

// 场景lua区块配置坐标范围可视化
func TestSceneBlock(t *testing.T) {
	config.InitConfig("./bin/application.toml")
	logger.InitLogger("SceneBlock")
	InitGameDataConfig()
	scene, exist := CONF.SceneDetailMap[3]
	if !exist {
		panic("scene 3 not exist")
	}
	logger.Info("scene info: %v", scene.SceneConfig)
	for _, block := range scene.BlockMap {
		block.BlockRange.Min.X *= -1.0
		block.BlockRange.Max.X *= -1.0
		block.BlockRange.Min.Z *= -1.0
		block.BlockRange.Max.Z *= -1.0
	}
	minX := 0.0
	maxX := 0.0
	minZ := 0.0
	maxZ := 0.0
	for _, block := range scene.BlockMap {
		if block.BlockRange.Min.X < minX {
			minX = block.BlockRange.Min.X
		}
		if block.BlockRange.Max.X > maxX {
			maxX = block.BlockRange.Max.X
		}
		if block.BlockRange.Min.Z < minZ {
			minZ = block.BlockRange.Min.Z
		}
		if block.BlockRange.Max.Z > maxZ {
			maxZ = block.BlockRange.Max.Z
		}
	}
	logger.Info("minX: %v, maxX: %v, minZ: %v, maxZ: %v", minX, maxX, minZ, maxZ)
	img := image.NewRGBA(image.Rect(0, 0, int(maxX-minX), int(maxZ-minZ)))
	rectColor := uint8(0)
	for _, block := range scene.BlockMap {
		maxW := int(block.BlockRange.Min.X - minX)
		maxH := int(block.BlockRange.Min.Z - minZ)
		minW := int(block.BlockRange.Max.X - minX)
		minH := int(block.BlockRange.Max.Z - minZ)
		for w := minW; w <= maxW; w++ {
			for h := minH; h <= maxH; h++ {
				img.SetRGBA(w, h, color.RGBA{R: rectColor, G: rectColor, B: rectColor, A: 255})
			}
		}
		rectColor += 5
		if rectColor > 255 {
			rectColor = 0
		}
	}
	file, err := os.Create("./bin/block.jpg")
	if err != nil {
		return
	}
	defer func() {
		_ = file.Close()
	}()
	err = jpeg.Encode(file, img, &jpeg.Options{
		Quality: 100,
	})
	if err != nil {
		return
	}
	logger.Info("test finish")
	time.Sleep(time.Second)
}