#! /usr/bin/env python
# encoding: utf-8

import ansiterm
import os,re,logging,traceback,sys
from Constants import*
zones=''
verbose=0
colors_lst={'USE':True,'BOLD':'\x1b[01;1m','RED':'\x1b[01;31m','GREEN':'\x1b[32m','YELLOW':'\x1b[33m','PINK':'\x1b[35m','BLUE':'\x1b[01;34m','CYAN':'\x1b[36m','NORMAL':'\x1b[0m','cursor_on':'\x1b[?25h','cursor_off':'\x1b[?25l',}
got_tty=False
term=os.environ.get('TERM','dumb')
if not term in['dumb','emacs']:
	try:
		got_tty=sys.stderr.isatty()or(sys.platform=='win32'and term in['xterm','msys'])
	except AttributeError:
		pass
import Utils
if not got_tty or'NOCOLOR'in os.environ:
	colors_lst['USE']=False
def get_color(cl):
	if not colors_lst['USE']:return''
	return colors_lst.get(cl,'')
class foo(object):
	def __getattr__(self,a):
		return get_color(a)
	def __call__(self,a):
		return get_color(a)
colors=foo()
re_log=re.compile(r'(\w+): (.*)',re.M)
class log_filter(logging.Filter):
	def __init__(self,name=None):
		pass
	def filter(self,rec):
		rec.c1=colors.PINK
		rec.c2=colors.NORMAL
		rec.zone=rec.module
		if rec.levelno>=logging.INFO:
			if rec.levelno>=logging.ERROR:
				rec.c1=colors.RED
			elif rec.levelno>=logging.WARNING:
				rec.c1=colors.YELLOW
			else:
				rec.c1=colors.GREEN
			return True
		zone=''
		m=re_log.match(rec.msg)
		if m:
			zone=rec.zone=m.group(1)
			rec.msg=m.group(2)
		if zones:
			return getattr(rec,'zone','')in zones or'*'in zones
		elif not verbose>2:
			return False
		return True
class formatter(logging.Formatter):
	def __init__(self):
		logging.Formatter.__init__(self,LOG_FORMAT,HOUR_FORMAT)
	def format(self,rec):
		if rec.levelno>=logging.WARNING or rec.levelno==logging.INFO:
			try:
				return'%s%s%s'%(rec.c1,rec.msg.decode('utf-8'),rec.c2)
			except:
				return rec.c1+rec.msg+rec.c2
		return logging.Formatter.format(self,rec)
def debug(*k,**kw):
	if verbose:
		k=list(k)
		k[0]=k[0].replace('\n',' ')
		logging.debug(*k,**kw)
def error(*k,**kw):
	logging.error(*k,**kw)
	if verbose>1:
		if isinstance(k[0],Utils.WafError):
			st=k[0].stack
		else:
			st=traceback.extract_stack()
		if st:
			st=st[:-1]
			buf=[]
			for filename,lineno,name,line in st:
				buf.append('  File "%s", line %d, in %s'%(filename,lineno,name))
				if line:
					buf.append('	%s'%line.strip())
			if buf:logging.error("\n".join(buf))
warn=logging.warn
info=logging.info
def init_log():
	log=logging.getLogger()
	log.handlers=[]
	log.filters=[]
	hdlr=logging.StreamHandler()
	hdlr.setFormatter(formatter())
	log.addHandler(hdlr)
	log.addFilter(log_filter())
	log.setLevel(logging.DEBUG)
init_log()

