#!/usr/bin/python
import string
import sys
import os
from os import listdir
from os.path import isdir
import time
from optparse import OptionParser

#config
LEAVESCAN = 'D:\leavescan.exe '

os_type = 'unix'
if sys.platform.lower().startswith('win'):
    os_type = 'windows'

class TestResult:
	def __init__(self):
		self.allOutput=""
		self.lines={}

	def setAllOutput(self,allOuptput):
		self.allOuptput = allOuptput

	def parseLines(self):
		tmp_lines=self.allOuptput.splitlines()
		for tmp_line in tmp_lines:
			line_no = getLeaveScanMsgLine(tmp_line)
			self.lines[line_no]=tmp_line

	def checkResult(self,warnItems):
		result = ''
		ok=0
		for warnItem in warnItems:
			warnLine = warnItem.line
			if not warnItem.semi:
				if self.lines.has_key(warnLine):
					outputLine = self.lines[warnLine]
					for keyword in warnItem.checkExist:
						if outputLine.find(keyword)<0:
							result= result+ 'CHECK:"'+keyword+'" is expected in the output:"'+outputLine+'"\n'
							ok=1
					for keyword in warnItem.checkNoExist:
						if outputLine.find(keyword)>=0:
							result= result+ 'CHECK:"'+keyword+'" is expected not in the output:"'+outputLine+'"\n'
							ok=1
				else:
					for keyword in warnItem.checkExist:
						result= result+ 'CHECK:"'+keyword+'" is expected in the output of line'+str(warnLine)+'\n'
						ok = 1
			else:
				if self.lines.has_key(warnLine):
					outputLine = self.lines[warnLine]
					for keyword in warnItem.checkExist:
						if outputLine.find(keyword)>=0:
							result= result+ 'SEMI:"'+keyword+'" is expected in the output:"'+outputLine+'"\n'
							ok=1
					for keyword in warnItem.checkNoExist:
						if outputLine.find(keyword)<0:
							result= result+ 'SEMI:"'+keyword+'" is expected not in the output:"'+outputLine+'"\n'
							ok=1
				else:
					for keyword in warnItem.checkNoExist:
						result= result+ 'SEMI:"'+keyword+'" is expected in the output of line'+str(warnLine)+'\n'
						ok = 1


		if ok==0:
			return 'OK'
		else:
			return result;

class CheckPoint:
	def __init__(self):
		self.line = -1
		self.checkExist=[]
		self.checkNoExist=[]
		self.semi = False
	def setCheckPoint(self,warnString,semi = False):
		warnString=warnString.strip()
		if len(warnString)<=0:
			return 
		keywordList = warnString.split(',')
		for keyword in keywordList:
			keyword = keyword.strip()
			if len(keyword)>0:
				if keyword[0]=='+':
					self.checkExist.append(keyword[1:])
				elif keyword[0]=='-':
					self.checkNoExist.append(keyword[1:])
				else:
					self.checkExist.append(keyword)
		self.semi = semi

	def printSelf(self):
		printType = 0;
		print 'line:'+ str(self.line)
		if len(self.checkExist)>0:
			print '		checkExist'
			printType +=1;
			for checkWord in self.checkExist:
				print '				'+checkWord
		if len(self.checkNoExist)>0:
			print '		checkNoExist'
			printType +=1;
			for checkWord in self.checkNoExist:
				print '				'+checkWord
		if printType == 0:
			print  '	nothing to check'



class TestcaseConfig:
	def __init__(self):
		self.desc =''
		self.option =''
		self.expect =''
		self.warnItems =[]
		self.author = ''
		self.type = ''
		self.date =''
	def printSelf(self):
		print 'desc:'+self.desc
		print 'option:'+self.option
		print 'author:'+self.author
		print 'type:'+self.type
		print 'date:'+self.date
		print 'expect:'+self.expect
		print 'check:'
		for warnItem in self.warnItems:
			print '	line:'+str(warnItem.line)
			printType = 0;
			if len(warnItem.checkExist)>0:
				print '		checkExist'
				printType +=1;
				for checkWord in warnItem.checkExist:
					print '				'+checkWord
			if len(warnItem.checkNoExist)>0:
				print '		checkNoExist'
				printType +=1;
				for checkWord in warnItem.checkNoExist:
					print '				'+checkWord
			if printType == 0:

				print  '	nothing to check'



def getTestcaseConfig(fileName):
	file = open(fileName,"r")
	fileLine = 0
	testcaseConfig = TestcaseConfig()
	for line in file.readlines():
		if not line.endswith('\\'):
			fileLine += 1
		line = line.strip()
		scommnet = line.find('//')
		if scommnet == 0:
			if (line.find('desc:'))>0:
				desc = line[line.find('desc:')+5:].strip()
				testcaseConfig.desc=desc
			elif line.find('option:')>0:
				option = line[line.find('option:')+7:].strip()
				testcaseConfig.option=option
			elif line.find('expect file:')>0:
				expect = line[line.find('expect file:')+12:].strip()
				#get the content of the file
				data = open(expect,"r")
				for dataline in data.readlines():
					testcaseConfig.expect += dataline
			elif line.find('expect:')>0:
				expect = line[line.find('expect:')+7:].strip()
				testcaseConfig.expect=expect
			elif line.find('author:')>0:
				expect = line[line.find('author:')+7:].strip()
				testcaseConfig.author=expect
			elif line.find('type:')>0:
				expect = line[line.find('type:')+5:].strip()
				testcaseConfig.type=expect
			elif line.find('date:')>0:
				expect = line[line.find('date:')+5:].strip()
				testcaseConfig.date=expect

		if scommnet >=0:
			line =line[scommnet+2:]
			if line.find('check:')>=0:
				line = line[line.find('check:')+6:].strip()
				warnItem = CheckPoint()
				warnItem.setCheckPoint(line)
				warnItem.line = fileLine
				testcaseConfig.warnItems.append(warnItem)
			elif line.find('semi:')>=0:
				line = line[line.find('semi:')+5:].strip()
				warnItem = CheckPoint()
				warnItem.setCheckPoint(line,True)
				warnItem.line = fileLine
				testcaseConfig.warnItems.append(warnItem)

				
	return testcaseConfig

def execute(filename , option):
	commandLine = LEAVESCAN + option + ' ' +filename
	output=''
	for outputLine in os.popen(commandLine).readlines():
		output += outputLine
	return output
	
def analysis(result,testcaseConfig):
	"""
	no use in the version
	for expect keyword
	"""
	if not testcaseConfig.expect=='':
		if testcaseConfig.expect == result:
			return 1;
		else:
			print testcaseConfig.expect
			if DEBUG:
				print '------'
				print result 
				print '========'
			return 0;

	
def excuteCmd(cmdLine):
	i, o = os.popen4(cmdLine)
   	i.close()
    	out = o.read()
    	o.close()
	return out


def getLeaveScanMsgLine(line):
	p1=line.find('(')
	if(p1<0):
		return -2
	p2=line.find(')')
	if(p2<2):
		return -2
	return int(line[p1+1:p2])

def getFormatTime():
	curTime = time.localtime()
	return str(curTime[0])+"_"+str(curTime[1])+"_"+str(curTime[2])+"_"+str(curTime[3])+"_"+str(curTime[4])+"_"+str(curTime[5])

def getCurTime():
	curTime = time.localtime()
	return "'"+str(curTime[0])+"-"+str(curTime[1])+"-"+str(curTime[2])+" "+str(curTime[3])+":"+str(curTime[4])+":"+str(curTime[5])+".000000'"

def getTestcaseName(fileName):
	fileNameList = os.path.splitext(fileName) 
	return fileNameList[len(fileNameList)-2]

def runTest(testFile,logFile,reportFile):
	testcaseConfig = getTestcaseConfig(testFile)
	cmdString = LEAVESCAN+testcaseConfig.option+' '+testFile
	logFile.write(cmdString+'\n')
	startTime = getCurTime()
	result = excuteCmd(cmdString)
	endtime = getCurTime()
	logFile.write(result+'\n')

	lr = TestResult()
	lr.setAllOutput(result)
	lr.parseLines()
	
	result=lr.checkResult(testcaseConfig.warnItems)
	testcaseResult = 'PASS' # for csv file
	if not result=='OK':
		logFile.write('ERROR:'+testFile+'\n')
		logFile.write('DETAIL:'+'\n')
		logFile.write(result)
		testcaseResult = 'FAILTURE'
		print testcaseResult+':'+testFile

	else:
		logFile.write('PASS\n')
	
	reportFile.write(getTestcaseName(testFile)+','+startTime+','+endtime+','+testcaseResult+'\n')
	if result=='OK':
		return 1
	else:
		return 0


if __name__ == '__main__':
	#pass argu
	testFileList=[]
	optParser = OptionParser('let -f file_list -o output_dir')
	optParser.add_option("-f","--file",action = "store",type = "string",dest = "fileListName")
	optParser.add_option("-o","--outputdir",action = "store",type = "string",dest = "outputName")
	optParser.add_option("","--std-in",action = "store_true",dest = "listFromStdin",default =False)
	optParser.add_option("-e","--execute",action = "store",type="string", dest = "exePath", default="")
	options, args = optParser.parse_args(sys.argv[1:])
	#check exe exsitence
	if not options.exePath == "":
		if os.path.exists(options.exePath):
			LEAVESCAN = options.exePath+" "
		else:
			print "WARN-"+options.exePath+" dosen't exist"
			print "WARN-will use default config:"+LEAVESCAN
	if not os.path.exists(LEAVESCAN.strip()):
		print "WARN-please set Leavescan's address"
	else:
		fileListName = options.fileListName
		outputDirName = options.outputName
		if not os.path.exists(outputDirName):
			os.makedirs(outputDirName)
		#get each testcase in the list 
		pwd=os.getcwd()+'\\'
		if options.listFromStdin :
			listFile = sys.stdin
		else:
			listFile = open(fileListName,'r')

		for curFile in listFile.readlines():
			curFile=curFile.strip()
			if(curFile.find('#')==0):
				1+1
			elif os.path.exists(curFile):
				testFileList.append(curFile)
			else:
				print 'WARN-file '+curFile+ 'doesn\'t exist.ignore'
		#other config
		DEBUG = 0;
		totalRun=0;
		totalPass=0;
		logFileName = outputDirName+'\\'+getFormatTime()+'.log'
		reportFileName = outputDirName+'\\'+getFormatTime()+'.csv'
		logFile = open(logFileName,'w')
		reportFile = open(reportFileName,'w')
		reportFile.write('TestCaseId,StartTime,EndTime,Result\n')

		for testFileName in testFileList:
		#test it
			totalRun+=1;
			totalPass+=runTest(testFileName,logFile,reportFile)
		#summary
		print 'RSLT-Total:'+str(totalRun)+';Pass:'+str(totalPass)
		#environment
		logFile.close()
		reportFile.close()