#!/usr/bin/python
# -*- coding: GBK -*-
#
# @todo: 
#
# @author: Alee
# @date 2018-1-3 ÏÂÎç05:47:38
# @version 1.0
#
# @note: 
#
#---------------------------------------------------------------------

#ϵͳ
import os
import sys
import time
import datetime
from binary.PyNetwork import *
from threading import Thread
import logging
from DataReader import MapEventPointReader, ChinMapReader

import ConfigurationReader.RobotsConfigReader
import ConfigurationReader.ConfigIniReader 
from Robot.RobotMgr import *

import traceback
import base64
import pymongo
import datetime


ServerDBConfigPath1 = r"\Config\config.ini"
ServerDBConfigPath2 = r"\PyMongoDataServer.ini"

g_starttime = time.time()

BASE64_ENCODE_CNT = 3
XOR_KEY = 151
def GetEncodePsw(psw):
    ret = ""
    try:
        for i in range(BASE64_ENCODE_CNT):
            psw = base64.decodestring(psw)
    except:

        return False,ret
    
    for i in psw:
        ret += chr(ord(i)^XOR_KEY)
    return True, ret

TYPE_Time_Format = "%Y-%m-%d %H:%M:%S"
def GetDateTimeByStr(timeStr):
    timeStr = timeStr.strip().split(".")[0]
    try:
        return  datetime.datetime.strptime(timeStr, TYPE_Time_Format)
    except BaseException , e :
        return

    return

def ChangeTimeStrToNum(timeStr, timeFormat=TYPE_Time_Format):
    timeStr = datetime.datetime.strptime(timeStr, timeFormat).timetuple() 
    return int(time.mktime(timeStr)) 


# Êý¾Ý¿â¹ÜÀí
class MongoDBRobot():
    def __init__(self):
        self.user = ""
        self.pwd = ""
        self.dbIP = ""
        self.dbName = ""
        
        self.logdb = None
        self.interfaceDatadb = None
        self.userdb = None
        self.connection = None
    
        config =  ConfigurationReader.ConfigIniReader.GetConfig()
        
        self.PlayerOffTime = config.GetPlayerOffTime() #Íæ¼ÒÏÂÏßʱ³¤
        self.ServerDBConfigPath = config.GetServerDBConfigPath()    #Êý¾Ý¿â·¾¶
        self.InitDBCon()
        
        self.tokenList = []   # Êý¾Ý¿â»ñÈ¡£¬µÇ¼ȥ³ý
        self.maxPlayerCnt = config.GetMaxPlayerCnt()
        self.loginCnt = 0
        self.AccountSource = config.GetAccountSource()
        
    def InitDBCon(self):
        dbConfig = ConfigurationReader.ConfigIniReader.AppConfig(self.ServerDBConfigPath + ServerDBConfigPath1)
        dbConfig.SetSection( "auth" )
        self.user = dbConfig.GetValue("userdb_user")
        self.pwd = GetEncodePsw(dbConfig.GetValue("userdb_pwd"))[1]
        
        dbConfig2 = ConfigurationReader.ConfigIniReader.AppConfig(self.ServerDBConfigPath + ServerDBConfigPath2)
        dbConfig2.SetSection( "connect" )
        self.dbIP = dbConfig2.GetValue("USER_DB_IP")
        self.connection = pymongo.Connection(self.dbIP, 27017, auto_start_request=False)
        db = self.connection.admin
        if not db.authenticate(self.user, self.pwd):
            logging.error( "!!!InitDBCon error" )
            return
        
        self.userdb = self.connection[dbConfig2.GetValue("USER_DB_NAME")]
        
        logging.info("conn ok-----------")
        
    # 1.»ñÈ¡ÍÑ»ú¹ÒÍæ¼ÒÊý¾Ý--GameLog
    def GetTJGRobot(self):
        if self.AccountSource == 0:
            index = 0
            if len(sys.argv) == 2:
                index = sys.argv[1]
            clientMac = "abc"
            for i in xrange(self.maxPlayerCnt):
                accID = "ice%s_%s@yun@s1"%(index, i)
    
                account = ("123456789012345678901234567890ab", accID, clientMac, 1)
                if account not in self.tokenList:
                    self.tokenList.append(account)
        else:
            col = self.userdb["tagDSAccount"]
            account = None
            for rec in col.find(limit=self.maxPlayerCnt):
                account = ("123456789012345678901234567890ab", str(rec['ACCID']), 'anc', 1)
                if account not in self.tokenList:
                    self.tokenList.append(account)
            
            print account
        if self.tokenList:
            logging.info( "´ýµÇ¼µÄÍÑ»ú¹ÒÍæ¼Ò----:%s"%(len(self.tokenList)))
            #logging.debug( "´ýµÇ¼µÄÍÑ»ú¹ÒÍæ¼Ò----tokenList:%s"%(self.tokenList))
        
        return
    
    def GetTokenList(self):
        return self.tokenList
    
    def AddCrossRobotInfo(self, accID, IPList):
        account = ("123456789012345678901234567890ab", accID, "abc", 1, IPList)
        if account not in self.tokenList:
            self.tokenList.append(account)
    
    def RemoveToken(self, value):
        self.tokenList.remove(value)
    
class WorkerThread(Thread):
    def __init__(self, mgr, index ):
        Thread.__init__(self, None, None, "WorkerThreadPy%s"%index)
        self.mgr = mgr
        self.clsTick = 30
        self.lastClsTick = 0

    def run(self):
        logging.debug( "thread %s start" % self.getName() )
        while 1:
            try:
                #Ïß³Ì1 »ñÈ¡ÍÑ»ú¹ÒÍæ¼ÒÊý¾Ý£¬¼ÓÈëµ½robotmgr
                self.mgr.ThreadRun()
                time.sleep( 0.01 )   # Ãë       
                if time.time() - self.lastClsTick > self.clsTick:
                    os.system("cls")
                    self.lastClsTick = time.time()
                    
            except Exception, e:
                print str(e)
                print traceback.print_exc()
                logging.error( "WorkerThread: %s"%str(e) )
                logging.error( "WorkerThread: %s"%traceback.print_exc() )
            
    # 1.»ñÈ¡ÍÑ»ú¹ÒÍæ¼ÒÊý¾Ý--GameLog
    # 2.³õʼ»¯ÍÑ»ú¹ÒÍæ¼ÒµÄrobot¼ÓÈëmgr
    # 3.ÇëÇóÐÂtokenµÇ¼--InterfaceData
    # 4.µÇ¼ºóÔÚ0107loadmapok°üÇ°·¢ËÍÍÑ»úµÇ¼ÏûÏ¢£¬ÓÃÓÚÇø·ÖÍæ¼ÒÕæʵµÇ¼
    # 5.ÍÑ»ú¹ÒÍæ¼Ò·À³ÁÃÔÂß¼­ÐÞÕý£¬¸ù¾Ý4Ìõ¼þ
            
def StartApp( PyBaseRoot ):
    logging.info( "PyBaseRoot:%s" % PyBaseRoot )
    titleMark = 0
    if len(sys.argv) == 2:
        titleMark = sys.argv[1]
    
    os.system("title %s-Robot-%s"%(titleMark, datetime.datetime.today()))
    
    #¶ÁÈ¡¸÷ÖÖÅäÖã¨IP£¬¶Ë¿Ú£¬VERSION_NOµÈ£©
    ConfigurationReader.ConfigIniReader.ReadConfig( PyBaseRoot + "Configuration\\Config.ini" )
    config =  ConfigurationReader.ConfigIniReader.GetConfig()
    #logging.info( "ConfigData:%s" % str(config) )
    
    dbRobot = MongoDBRobot()
    
    #¶ÁÈ¡¸÷ÖÖÊý¾Ý£¨µØͼ£©, ÍÑ»ú¹ÒÖ±½Ó·ÉÐÐ
    #MapDataReader.ReadMapData( PyBaseRoot )
    #TransportDataReader.ReadTransportData( PyBaseRoot )
    MapEventPointReader.ReadMapEventPoint(PyBaseRoot)
    ChinMapReader.ReadChinMapData(PyBaseRoot)
    
    #³õʼ»¯ÍøÂ磨Asio£©
    asioMgr = CAsioMgr()
    ret = asioMgr.InitModule( config.GetVersionNo(), config.GetPackStartCount(),\
                              config.GetKeyString(), PyBaseRoot + config.GetSendKeyDictFilePath() )
    if not ret:
        logging.fatal( "Init PyNetwork failed." )
        return
    else:
        logging.debug( "Init PyNetwork success." )
     
    #´´½¨ÍøÂçÄ£¿é¹¤×÷Ị̈߳¬ÕâЩÏß³ÌÊýÁ¿²»Ò˹ý¶à£¬ÒÔÃâ³öÏÖ´óÁ¿¾ºÕù  
    #½¨ÒéÈ¡Öµ·¶Î§ÔÚ nThreadsµ½2*nThreadsÖ®¼ä
    nThreads = asioMgr.GetProcessorNumber()
    logging.info( "Host has %d logic processor." % nThreads )
        
    IO_WORKER_THREAD_NUM = config.GetIOThreadNum()
    threadList = []
    logging.info( "Create %d IO Threads....." % IO_WORKER_THREAD_NUM )
    for i in range( IO_WORKER_THREAD_NUM ):
        th = WorkerThread( asioMgr, i )
        threadList.append( th )
        th.start()

    dbRobot.GetTJGRobot()
    #´´½¨RobotMgr£¬Ìṩ»úÆ÷ÈË×ÜÊý£¨´ËMgrÄÚ²¿»á´´½¨AIÏß³ÌÓÃÓÚ½øÐÐAI´¦Àí£©
    CreateRobotMgr( asioMgr, dbRobot)