package main

import (
	"go/ast"
	"os"
	"strconv"
	"strings"
)

type TypesInfo struct {
	Types
	Direction string
	State     string
}

type PacketInfo struct {
	TName, Name, Direction, State, PacketId, ProtocolVersion, MinecraftVersion string
}

func GeneratePacketInfoFile(versions []string, packetInfos []PacketInfo) (err error) {
	f := NewFile("proto_generated")
	imports := []string{"github.com/admin-else/strom/proto_base"}
	for _, v := range versions {
		vUnderscore := strings.ReplaceAll(v, ".", "_")
		imports = append(imports, "github.com/admin-else/strom/proto_generated/v"+vUnderscore)
	}
	AppendDecl(f, Import(imports...))

	var packetInfoExprs []ast.Expr
	for _, p := range packetInfos {
		vUnderscore := "v" + strings.ReplaceAll(p.MinecraftVersion, ".", "_")
		typeName := p.TName
		var typeExpr ast.Expr
		if typeName == "struct{}" {
			typeExpr = &ast.StructType{Fields: &ast.FieldList{}}
		} else if strings.HasPrefix(typeName, "proto_base.") {
			typeExpr = Selector("proto_base", strings.TrimPrefix(typeName, "proto_base."))
		} else {
			typeExpr = Selector(vUnderscore, typeName)
		}

		packetId, _ := strconv.ParseInt(p.PacketId, 0, 32)
		protocolVersion, _ := strconv.Atoi(p.ProtocolVersion)

		packetInfoExprs = append(packetInfoExprs, CompLit(nil, []ast.Expr{
			KeyValueExpr(Ident("Type"), AddrOf(CompLit(typeExpr, nil))),
			KeyValueExpr(Ident("Name"), StrLit(p.Name)),
			KeyValueExpr(Ident("Direction"), Selector("proto_base", p.Direction)),
			KeyValueExpr(Ident("state"), Selector("proto_base", p.State)),
			KeyValueExpr(Ident("PacketId"), NumLit(int(packetId))),
			KeyValueExpr(Ident("ProtocolVersion"), NumLit(protocolVersion)),
		}))
	}
	AppendDecl(f, VarValues("Packets", CompLit(Slice(Selector("proto_base", "PacketInfo")), packetInfoExprs)))
	out, err := os.Create("proto_generated/packet_info.go")
	if err != nil {
		return
	}
	defer out.Close()

	return PrintToFile(f, out)
}