#MUDPyE - (M)ulti-(U)ser (D)imension (Py)thon (E)ngine
#Copyright (C) 2005  Corey Staten

#This program is free software; you can redistribute it and/or
#modify it under the terms of the GNU General Public License
#as published by the Free Software Foundation; either version 2
#of the License, or (at your option) any later version.

#This program is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#GNU General Public License for more details.

#You should have received a copy of the GNU General Public License
#along with this program; if not, write to the Free Software
#Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

#Send feedback/questions to MUDPyE@gmail.com

import gzip
import logging
import logging.config
import os.path
import time

from mpmudobject import Ex

loggers = ["engine", "engine.db", "engine.server", "engine.fileloader", "mud", "mud.scripterror"]
mudWorld = None

class ZipLogHandler(logging.Handler):
  def __init__(self, fileName, maxSize, zipDir):
    logging.Handler.__init__(self)
    self.fileName = os.path.normpath(fileName)
    self.maxSize = maxSize
    self.zipDir = zipDir
    self.fileObj = file(fileName, "a")
    print "Successfully opened logfile [%s]." % fileName

  def emit(self, record):
    self.fileObj.write("".join((self.format(record), "\n")))
    if self.fileObj.tell() >= self.maxSize:
      self.fileObj.close()
      zipFileName = ("%s %s.gz" % (time.asctime(), os.path.split(self.fileName)[1]))
      zipFilePath = os.path.join(self.zipDir, zipFileName).replace(":", "#")
      try:
        zipFileObj = file(zipFilePath, "wb")
        zipFileObj = gzip.GzipFile(zipFileName, "wb", 9, zipFileObj)
        logFileObj = file(self.fileName, "r")
        buffer = logFileObj.read(8192)
        while len(buffer):
          zipFileObj.write(buffer)
          buffer = logFileObj.read(8192)
        zipFileObj.close()
        logFileObj.close()
      except IOError:
        print "Error saving zipped log file."
      self.fileObj = file(self.fileName, "w")

class MudHandler(logging.Handler):
  def __init__(self):
    logging.Handler.__init__(self)
    self.ready = True

  def emit(self, record):
    print self.format(record)
    #Temporarily disable self from logging, since some of these functions may emit logs.
    if self.ready:
      self.ready = False
      if mudWorld.__dict__.has_key("objDB") and mudWorld.objDB.HasObj(mudWorld.loggingObjID):
        logObj = mudWorld.objDB.GetObjRef(mudWorld.loggingObjID)
        Ex(logObj, "_Sys_LogString", self.format(record))
      self.ready = True

def InitLogging(mudWorld, fileName):
  exec "import mplog" in logging.__dict__
  globals()["mudWorld"] = mudWorld
  logging.config.fileConfig(fileName)
  loggerDict = {}
  map(loggerDict.setdefault, loggers, [logging.getLogger(logger) for logger in loggers])
  return loggerDict