# -*- coding: UTF-8 -*-
import sys
import os
import urllib2
import re

def GetHTML(URL):
	obj  = urllib2.urlopen(URL)
	html = obj.read()
	obj.close()
	return html
	
# Version 확인을 위한 클래스
class Version:

	major = None
	minor = None
	build = None
	
	# 로컬 버전 문자열 읽음.
	def FromLocalVersionString(self, localVersionString):
		vers = localVersionString.split('.')
		self.major = int(vers[0])
		self.minor = int('%s%s' % (vers[1], vers[2]))
		self.build = int(vers[3])
	
	# 최신 버전 (웹페이지에서 확인한) 문자열 읽음
	def FromCurrentVersionString(self, currentVersionString):
		vers = currentVersionString.split('.')
		self.major = int(vers[0])
		self.minor = int(vers[1])
		self.build = int(vers[2])
	
	def __str__(self):
		return '%d/%d/%d' % (self.major, self.minor, self.build)
	
	# 로컬 버전 <= 최신버전이므로 !=, == 만 정의.
	def __eq__(self, other):
		return isinstance(other, self.__class__) and self.__dict__ == other.__dict__

	def __ne__(self, other):
		return not self.__eq__(other)		

	
def CheckUpdate(config):
	# check CCleaner.exe exists!
	if os.path.exists(config['ccleaner_path']) == False:
		print >> sys.stderr, '%s does not exist!' % config['ccleaner_path']
		return 1
	
	# local ccleaner version	
	print "Checking installed CCleaner's version..."
	command = 'wmic DATAFILE WHERE NAME="%s" GET version > version.txt' % config['ccleaner_path']
	os.system(command)
	
	# get version
	with open('version.txt', 'r') as f:
		text   = unicode(f.read(), 'utf-16')
		iv_txt = text.split(u'\n')[1].strip()		

	# current release version
	print 'Checking current CCleaner version at Piriform...'
	release_html = GetHTML(config['release_url'])
	release_exp  = re.compile(config['release_re'] , re.DOTALL|re.MULTILINE)
	release_srch = release_exp.search(release_html).groups()
	
	cv_txt  = release_srch[0].strip()[1:] # exclude heading 'v'
	reldate = release_srch[1].strip()
	
	# To canonical versino expression.
	installed_ver = Version()
	current_ver  = Version()
	
	installed_ver.FromLocalVersionString(iv_txt)
	current_ver.FromCurrentVersionString(cv_txt)
	
	print 'Installed CCleaner version:', installed_ver
	print 'Current CCleaner version:', current_ver, reldate
	
	# compare two
	if installed_ver == current_ver:
		print 'You\'re using current version. Nothing to do.'
		return 0	
	else:
		print 'There\'s a new update available!'
	
	# download current release
	download_html = GetHTML(config['download_url'])
	download_exp  = re.compile(config['download_re'])
	download_srch = download_exp.search(download_html)
	
	url           = download_srch.groups()[0]
	filename      = url.split('/')[-1]		
	
	print 'Begin downloading...'
	obj = urllib2.urlopen(url)
	with open(filename, 'wb') as f:
		f.write(obj.read())
		obj.close()
		
	print 'Downloading complete!'
	
	cmd = filename + ' ' + config['install_arg']
	
	print 'Installing...'
	os.system(cmd)
	print 'Complete!'	
	return 0
	
def ParseConfig(configfile):
	config = {}
	with open(configfile, 'r') as f:
		for l in f:
			colon = l.find(':')
			prop  = l[0:colon].strip()
			val   = l[colon+1:].strip()
			
			config[prop] = val
			
	return  config

def main(argv):	

	# parse config file
	configfile = './CCleanerAutoUpdate.cfg'
	if len(argv) == 2:
		configfile = argv[1]
	config = ParseConfig(configfile)
	
	# begin the job
	return CheckUpdate(config)
		
if __name__ == '__main__':
	sys.exit(main(sys.argv))