#!/usr/bin/env python
# -*- coding: utf-8  -*-
'''
webchecklinks.py
 A web wrapper around the checklinks library
 (c) Dispenser, 2007-2008
'''

import cgi, re
import checklink, wikipedia, parser
import cgitb; cgitb.enable(logdir='./logs/', context=10)

def printu(ustr):
	try:
		print ustr
	except:
		print ustr.encode('utf-8')
	
def printFile(name, arg1="$1"):
	f = open('./text/%s.html' % name )
	print f.read().replace('$1', arg1.encode('utf-8'))
	f.close()

def addEntry(self, page, url, refId, context, status, reason, redirect, rank, comment):
	#printu(u'%s %s - %s' % (status, reason, wikilink))
	if not rank < threshold:
		classes = 'dead-%s'%rank
		if url.encode('utf-8')+'\n' in open('/home/dispenser/webcite_requests.txt','r'):
			classes += ' webcite'
		if redirect:
			classes += ' redirect'
		printu('<tr class="%s"><td>%s</td><td>%s</td><td><abbr title="%s">%s</abbr></td><td>%s</td></tr>' % (classes, refId or '', context.replace('[[', '[&#91;'), reason, status, rank and (comment or reason) or ''))
	
def textbox(name, value, label = None, attrib=''):
	if label is None:
		label = '%s: ' % name.capitalize()
	print '<label for="%s">%s</label><input type="text" name="%s" value="%s" id="%s" %s/>' % (name, label, name, value, name, attrib)

def checkbox(name, checked, label = None, attr=''):
	if checked:
		attr += ' checked="checked"'
	print '<input type="checkbox" name="%s" id="%s" %s /><label for="%s">%s</label>' % (name, name, attr, name, label or name.capitalize())

def main():
	global threshold
	#global hostname

	form = cgi.FieldStorage()
	checklink.DEBUG	= bool(form.getfirst('debug', False))
	checklink.SOURCE = bool(form.getfirst('source', False))
	checklink.config.max_external_links	= int(form.getfirst('threads',   30))
	checklink.config.defaulttimeout 	= int(form.getfirst('timeout',   30))
	checklink.config.httpDebug      	= int(form.getfirst('httpDebug',  0))
	alwaysUseGet = bool(form.getfirst('alwaysUseGet',	False))
	htmlmode     = bool(form.getfirst('html',			False))
	nothread     = bool(form.getfirst('nothread',		False))
	threshold    = int( form.getfirst('threshold',		0))
	if form.getfirst('hostname'):
		host = form.getfirst('hostname').split('.')
		site = wikipedia.Site(host[0], host[1])
	else:
		site = wikipedia.getSite()
	page = wikipedia.Page(site, form.getfirst('page', ''))
	#hostname = page.hostname()
	
	wikipedia.startContent(u'Checklinks: %s' % page.title(), form=False, notice='')
	print '<form action="webchecklinks.py" style="text-align:center;" onsubmit="hideViewer()">'
	print '<input name="page" id="page" size="40" value="%s" onchange="fixTitle(this)" />' % page.aslink()[2:-2].encode('utf-8')
	if checklink.DEBUG:
		print '<br/>'
		textbox('threshold', threshold, attrib=' size="1"')
		textbox('threads', checklink.config.max_external_links, attrib=' size="2"')
		textbox('timeout', checklink.config.defaulttimeout, attrib=' size="3"')
		textbox('httpDebug', checklink.config.httpDebug, 'HTTP debug level: ', ' size="3"')
		print '<br/>'
		checkbox('alwaysUseGet', alwaysUseGet, 'Force downloading')
		checkbox('debug', checklink.DEBUG, 'Debug')
		checkbox('html', htmlmode, 'HTML input')
		checkbox('nothread', nothread, 'Disable threading')
		checkbox('source', checklink.SOURCE, 'Print source')
	print '<input type="submit" value="Check page" /></form>'
	try:
		page.get()
	except wikipedia.NoPage, e:
		printu('NoPage error encountered (%s)' % e)
		return
	except wikipedia.IsRedirectPage:
		link = wikipedia.Page(page.site(), page._redirarg).aslink()[2:-2]
		printu('<img src="http://%s/skins-1.5/common/images/redirectltr.png" alt="#REDIRECT " /><a href="webchecklinks.py?page=%s" class="redirectText">%s</a>' % (page.hostname(), link, link))
		return
	try:
		printFile('ChecklinksHeader')
		print '<table id="linktable" class="">'
		printu('<tr class="page"><th colspan="6"><a id="%s" href="http://%s/wiki/%s" title="%s" class="extiw">%s</a></th></tr>' % (parser.escapeId(page.title().encode('utf-8')), page.hostname(), page.urlname(), page.aslink()[2:-2], page.title()))
		printFile('tableHeader')
		if htmlmode:
			checklink.checkMWhtml(page, addEntry, alwaysUseGet, not nothread)
		else:
			checklink.checkMediaWikiPage(page, addEntry, alwaysUseGet, not nothread)
	finally:
		print '</table>'

if __name__ == "__main__" and wikipedia.handleUrlAndHeader():
	try:
		main()
	finally:
		wikipedia.endContent()