import os

ProtoCodeGO = './cmd/gameserver/action/protocode.go'
ProtoCodeGOTest = './cmd/test/action/protocode.go'
ProtoCodeFile = './protos/protocode.proto'
ProtoFileDir = "./protos"

ProtoCodeStr = "syntax = \"proto3\";\noption go_package = \"../pb;pb\";\n\npackage protocode;\n\nenum ProtoCode\n{{\n  " \
               "UNKNOWN = 0;\n {}\n}} "
ProtoCodeLineReq = "\t{}Rpc = {};\n"
ProtoCodeLineRsp = "\t{}Rsp = {};\n"
ProtoCodeLineNty = "\t{}Nty = {};\n"

GoProtoCodeStr = "package action\n\nimport (\n\t\"pro2d/common/logger\"\n\t\"pro2d/pb\"\n)\n\nfunc GetActionMap() " \
                 "map[interface{{}}]interface{{}} {{\n\tlogger.Debug(\"init protocode...\")\n\tam := make(map[interface{{" \
                 "}}]interface{{}})\n{}\n\treturn am\n}}"
GoProtoCodeLine = "\tam[uint32(pb.ProtoCode_{}Rpc)] = {}Rpc\n"

GoProtoCodeTestStr = "package action\n\nimport (\n\t\"pro2d/pb\"\n)\n\nfunc GetTestActionMap() " \
                 "map[interface{{}}]interface{{}} {{\n\tam := make(map[interface{{" \
                 "}}]interface{{}})\n{}\n\treturn am\n}}"
GoProtoCodeTestReqLine = "\tam[uint32(pb.ProtoCode_{}Rpc)] = {}Rsp\n"
GoProtoCodeTestRspLine = "\tam[uint32(pb.ProtoCode_{}Rsp)] = {}Rsp\n"
GoProtoCodeTestNtyLine = "\tam[uint32(pb.ProtoCode_{}Nty)] = {}Nty\n"

def generatorProto(path):
    RpcStartCode = 500
    NtyStartCode = 1000

    files = os.listdir(path)
    ProtoCodeData = ""
    GoCodeData = ""
    GoCodeTestData = ""
    for file in files:
        if file.find("account.proto") != -1:
            continue

        if os.path.isdir(file):
            continue

        with open(path + "/" + file, 'r', encoding='utf-8', errors='ignore') as f:
            lines = f.readlines()  # 读取所有行
            firstline = lines[0]
            if firstline.find("proto3") == -1:
                continue

            # req + rar
            for line in lines:
                if line.find("message") == -1:
                    continue
                sline = line.split(' ')
                if len(sline) < 2:
                    continue

                messageStr = sline[1].replace('\n', '').replace('{', "")
                n1 = messageStr.find('Req')
                n2 = messageStr.find('Rar')
                loginReq = messageStr.find('LoginReq')

                if n1 != -1:
                    RpcStartCode = RpcStartCode + 1
                    ProtoCodeData += ProtoCodeLineReq.format(messageStr[:n1], RpcStartCode)
                    if loginReq != -1:
                        continue
                    GoCodeData += GoProtoCodeLine.format(messageStr[:n1], messageStr[:n1])
                    GoCodeTestData += GoProtoCodeTestReqLine.format(messageStr[:n1], messageStr[:n1])
                elif n2 != -1:
                    RpcStartCode = RpcStartCode + 1
                    ProtoCodeData += ProtoCodeLineReq.format(messageStr[:n2], RpcStartCode)
                    if loginReq != -1:
                        continue
                    GoCodeData += GoProtoCodeLine.format(messageStr[:n2], messageStr[:n2])
                    GoCodeTestData += GoProtoCodeTestReqLine.format(messageStr[:n2], messageStr[:n2])
            ProtoCodeData += '\n'
            # nty
            for line in lines:
                if line.find("message") == -1:
                    continue
                sline = line.split(' ')
                if len(sline) < 2:
                    continue

                messageStr = sline[1].replace('\n', '').replace('{', "")
                n3 = messageStr.find('Nty')
                loginReq = messageStr.find('LoginReq')

                if n3 != -1:
                    NtyStartCode = NtyStartCode + 1
                    ProtoCodeData += ProtoCodeLineNty.format(messageStr[:n3], NtyStartCode)
                    if loginReq != -1:
                        continue
                    GoCodeTestData += GoProtoCodeTestNtyLine.format(messageStr[:n3], messageStr[:n3])




    # protocode.go
    gostr = GoProtoCodeStr.format(GoCodeData)
    fo = open(ProtoCodeGO, "w")
    fo.write(gostr)
    fo.close()

    # protocode.go
    gostr = GoProtoCodeTestStr.format(GoCodeTestData)
    fo = open(ProtoCodeGOTest, "w")
    fo.write(gostr)
    fo.close()

    #protocode.proto
    protostr = ProtoCodeStr.format(ProtoCodeData)
    fo = open(ProtoCodeFile, "w")
    fo.write(protostr)
    fo.close()


if __name__ == "__main__":
    generatorProto(ProtoFileDir)