#coding:utf8
import os
import json
import re
import sys
import shutil
import tempfile

print sys.path
sys.path.append(".")
#sys.path.append("protoc")
#sys.path.append("c#protoGen")

'''
{
	moduleName: {
		"id":name,

		"methodName":1,
		"methodName2": 2,
	}
}
'''


pat = re.compile('message (\w+)')

#msgID = 1
moduleID = 1

msgMap = {}
newMsgMap = {}
if os.path.exists('nameMap.json'):
	con = open('nameMap.json').read()
	msgMap = json.loads(con)

'''
分配一个模块的 消息ID
'''
def getMsgID(moduleName):
	#global msgID
	#global moduleID
	global msgMap
	msgID = 1
	for k in msgMap[moduleName]:
		if k != 'id':
			if msgMap[moduleName][k] >= msgID and k.find('Push') == -1:
				msgID = msgMap[moduleName][k]+1
	return msgID

'''
Push Msg ID
'''
def getPushMsgID(moduleName):
	global msgMap
	msgID = 1000
	for k in msgMap[moduleName]:
		if k != 'id':
			if msgMap[moduleName][k] >= msgID:
				msgID = msgMap[moduleName][k]+1
	return msgID				


'''
分配一个模块ID
'''
def getModuleID(modName):
	global moduleID
	global msgMap
	for k in msgMap:
		if msgMap[k]["id"] >= moduleID:
			moduleID = msgMap[k]["id"]+1
	if moduleID >= 256:
		raise Exception("Module Id too Big "+modName+" "+moduleID)
	return moduleID



f1 = os.listdir('protos')
#f1 = ["CommonProtoc.proto", "MCopyProtoc.proto"]
allProtos = []
for i in f1:
	if i.find('.proto') != -1  and i.find("dump") == -1 and i.find('.swp') == -1:
		modName = i.replace('.proto', '')
		if msgMap.get(modName) == None and modName[0] == 'M':
			msgMap[modName] = {"id":getModuleID(modName)}
		#newMsgMap[modName] = {}	
			
		allProtos.append(i)

		profile = open("protos/"+i).read()
		#all Messsage
		mat = pat.findall(profile)
		for n in mat:
			if msgMap.get(modName) != None and msgMap[modName].get(n) == None:
				if n[:2] == 'CG' or n[:2] == 'GC':
					print "Add Name is", n
					if n.find("Push") != -1:
						msgMap[modName][n] = getPushMsgID(modName)
					else:
						msgMap[modName][n] = getMsgID(modName)
			newMsgMap[n] = True

#检查是否有MsgName 在nameMap.json 里面但是不在 proto里 表示这个Message被删除了 可以从MsgName 中去掉
cleanMsgMap = {}
for mod in msgMap:
	newMod = {}
	for msg in msgMap[mod]:
		if msg != 'id':
			if newMsgMap.get(msg) != None:
				newMod[msg] = msgMap[mod][msg]
		else:
			newMod["id"] = msgMap[mod]["id"]
	cleanMsgMap[mod] = newMod

msgMap = cleanMsgMap

						
comStr = ''
for a in allProtos:
	comStr += 'protos/%s ' % (a)

print "all protos"
print comStr

os.system('mono ProtoGen.exe %s protos/google/protobuf/csharp_options.proto protos/google/protobuf/descriptor.proto --proto_path=protos' % (comStr))

if os.path.exists("temp"):
	#os.system("rm -r temp")
	shutil.rmtree("temp")
#os.system('copy -r protos temp')

shutil.copytree("protos", "temp")

allF = os.listdir('temp')
csharpRep = '''
import "google/protobuf/csharp_options.proto";
option (google.protobuf.csharp_file_options).namespace = "ChuMeng";
option (google.protobuf.csharp_file_options).umbrella_classname = "CommonProtoc";
'''
newoption = '''
option optimize_for = LITE_RUNTIME;
'''

comStr2 = ''
for p in allF:
    if p.find('.dump') == -1 and p.find('.swp') == -1 and p.endswith('.proto'):
        name = os.path.join('temp', p)
        if not os.path.isdir(name):
            con = open(name).readlines()
            ncon = ''
            for l in con:
                #if l.find('csharp') == -1 and l.find('optimize_for') == -1:
                    ncon += l
                    
            #con = con.replace(csharpRep, '')
            f = open(name, 'w')
            f.write(ncon)
            f.close()
            comStr2 += name+" "

print 'cmdstr'
print comStr2
os.system("protoc --python_out=py --java_out=java %s --proto_path=temp" % (comStr2))
shutil.rmtree("temp")
	
	
	
		
f = os.listdir('.')
delFile = [
'Util2.cs',
'Util3.cs',
'CSharpOptions.cs',
'DescriptorProtoFile.cs',
]
for i in f:
	if i.find('.cs') != -1 :
		if i in delFile:
			os.remove(i)
		else:
			#os.system('mv %s cs/%s' % (i, i))
			shutil.move(i, "cs/%s" % (i))

f = open('nameMap.json', 'w')
f.write(json.dumps(msgMap, indent=4, sort_value=True, separators=(', ', ': ')))
f.close()


import GenUtilFile
GenUtilFile.GenUtil(msgMap.values())

os.chdir('cs')
os.system('python UpdateSln.py')
os.system('xbuild new.csproj')
os.chdir('..')

U3D = '/Users/liyong/Desktop/allUnity/Assets/Plugins'
os.system('cp cs/obj/Debug/protoDll.dll %s' % (U3D))

U3D = '/Users/liyong/Desktop/allUnity/Assets/scripts/util'
os.system('cp Util2.cs %s' % (U3D))


U3D = '/Users/liyong/Desktop/allUnity/Assets/Resources'
os.system('cp nameMap.json %s' % (U3D))

Server = '/Users/liyong/Projects/SocketServer/packages'
os.system('cp cs/obj/Debug/protoDll.dll %s' % (Server))

Server2 = '/Users/liyong/Projects/SocketServer/SocketServer/bin/Debug'
os.system('cp nameMap.json %s' % (Server2))


f = open('Util2.cs', 'r')
con = f.read();
con = con.replace('using UnityEngine;', '')
nf = open('Util3.cs', 'w')
nf.write(con)
nf.close()

Server3 = '/Users/liyong/Projects/SocketServer/SocketServer/Common/Util2.cs'
os.system('cp Util3.cs %s' % (Server3))