goalrescha.go

//go:generate go get -u github.com/valyala/quicktemplate/qtc
//go:generate go run github.com/valyala/quicktemplate/qtc -dir=goqtpl

package goalrescha

import (
	"bytes"
	"fmt"
	"os/exec"
	"strconv"

	"github.com/unsafe-risk/go-alrescha/goqtpl"
	"github.com/unsafe-risk/go-alrescha/parser"
)

func MakeGoCode(data []byte, PackageName string) ([]byte, error) {
	var buffer *bytes.Buffer = bytes.NewBuffer(nil)
	info, err := parser.ParseGenerateInfo(data)
	if err != nil {
		return nil, err
	}

	fmt.Fprintf(buffer, "package %s\n", PackageName)
	fmt.Fprintln(buffer)
	fmt.Fprintln(buffer, "import (")
	fmt.Fprintln(buffer, "\t\"io\"")
	fmt.Fprintln(buffer, "\t\"math\"")
	fmt.Fprintln(buffer, ")")
	fmt.Fprintln(buffer)
	fmt.Fprintln(buffer, "var _ = math.Float64bits")
	fmt.Fprintln(buffer, "var _ = io.EOF")
	fmt.Fprintln(buffer, goqtpl.MaxLenConst(info.Max))

	for i := range info.IDL.Types {
		qtplStructs := make([]goqtpl.StructField, 0, len(info.IDL.Types))
		for j := range info.IDL.Types[i].Fields {
			if info.IDL.Types[i].Fields[j].FieldType == "array" {
				qtplStructs = append(qtplStructs, goqtpl.StructField{
					Name: info.IDL.IndexPathMap[info.IDL.Types[i].Fields[j].Index],
					Type: "[" + strconv.Itoa(info.IDL.Types[i].Fields[j].ArrayLength) + "]" + ConvertToGoType(info.IDL.Types[i].Fields[j].CodeType),
				})
			} else if info.IDL.Types[i].Fields[j].FieldType == "list" {
				qtplStructs = append(qtplStructs, goqtpl.StructField{
					Name: info.IDL.IndexPathMap[info.IDL.Types[i].Fields[j].Index],
					Type: "[]" + ConvertToGoType(info.IDL.Types[i].Fields[j].CodeType),
				})
			} else {
				qtplStructs = append(qtplStructs, goqtpl.StructField{
					Name: info.IDL.IndexPathMap[info.IDL.Types[i].Fields[j].Index],
					Type: ConvertToGoType(info.IDL.Types[i].Fields[j].CodeType),
				})
			}
		}
		fmt.Fprintln(buffer, CleanCode(goqtpl.GenerateGoStruct(info.IDL.IndexPathMap[info.IDL.Types[i].Index], qtplStructs)))
	}
	for i := range info.Structs {
		fmt.Fprintln(buffer, CleanCode(goqtpl.MakeMarshal(info.Structs[i].Name, info.Structs[i].Types)))
		fmt.Fprintln(buffer, CleanCode(goqtpl.MakeUnmarshal(info.Structs[i].Name, info.Structs[i].Types)))
	}
	return []byte(CleanCode(buffer.String())), nil
}

func ConvertToGoType(RawType string) string {
	switch RawType {
	case "i8":
		return "int8"
	case "i16":
		return "int16"
	case "i32":
		return "int32"
	case "i64":
		return "int64"
	case "u8":
		return "uint8"
	case "u16":
		return "uint16"
	case "u32":
		return "uint32"
	case "u64":
		return "uint64"
	case "f32":
		return "float32"
	case "f64":
		return "float64"
	case "str":
		return "string"
	case "bytes":
		return "[]byte"
	}
	return RawType
}

func CleanCode(in string) string {
	buf := bytes.NewBuffer(nil)
	buf.WriteString(in)
	outbuf := bytes.NewBuffer(nil)
	gfmt := exec.Command("gofmt")
	gfmt.Stdin = buf
	gfmt.Stdout = outbuf
	gfmt.Start()
	gfmt.Wait()
	return outbuf.String()
}