#!/usr/bin/python -OO

import sys
import config
import cgi
import gentoo
import os
import re
from MySQLdb import escape_string

form = cgi.FieldStorage()
type = form.getvalue("type","")
sstring = form.getvalue("sstring","")
offset = form.getvalue("offset","0")

if not sstring:
	sys.exit(0)

if type == 'desc':
	col = "description"
else:
	col = "name"

def query_to_dict(q):
	pkginfo = {}
	keys = ('category','name','homepage','description','license')
	for i in range(len(keys)):
		try:
			pkginfo[keys[i]] = q[i]
		except IndexError:
			continue
	return pkginfo

def write_to_cache(s):
    try:
        open(cachefile,'w').write(s)
    except IOError:
        pass

def sort_by_weight(a, b):
    """Right now we just sort based on whether the sstring is in the name"""
    a_name = a['name']
    b_name = b['name']
    
    a_match = re.search(sstring, a_name, re.IGNORECASE)
    b_match = re.search(sstring, b_name, re.IGNORECASE)
    
    if a_match and b_match:
        return 0
    if a_match:
        return -1
    return 1

# if it's in the cache, just write that out
qs = sys.argv[1]
cachefile = os.path.join(config.LOCALHOME,'search/cache',qs)
if os.path.exists(cachefile):
	#sys.stdout.write('<div class="centerpage">using cached results</div>\n')
	sys.stdout.write(open(cachefile,'r').read())
	sys.exit(0)

escaped = escape_string(sstring)
query = ('SELECT category,name,homepage,description,license '
	'FROM package WHERE name REGEXP "%s" or description REGEXP "%s" '
	'ORDER BY category LIMIT %s,%s' 
	% (escaped,escaped,offset,config.MAXPERPAGE))

import ebuilddb
db = ebuilddb.db_connect()
c = db.cursor()
try:
	c.execute(query)
	results = c.fetchall()
except:
	results = None

if not results:
	s = ( '<div class="centerpage">\n'
	'<table width="100%%" border="0"  align="center"'
	' cellspacing="0"><tr><td colspan="3" class="fields">'
	'Nothing found.<br></td></tr>\n'
	'<tr><td class="item" colspan="3">'
	'<p>I could not find any ebuild that match'
	' your query.  Try a different query or check the'
	' <a HREF="%s">'
	'packages.gentoo.org main page</a>.</p></td></tr></table>\n'
	'</div>\n' % config.FEHOME)
	sys.stdout.write(s)
	write_to_cache(s)
	sys.exit(0)

pkgs = [ query_to_dict(i) for i in results ]
pkgs.sort(sort_by_weight)

s = '\n'.join([gentoo.package_to_html(pkg,db) for pkg in pkgs])
#for pkg in pkgs:
#	#print pkg
#	html = gentoo.package_to_html(pkg,db)
#	s = '%s\n%s' % (s,gentoo.package_to_html(pkg,db))

if offset != "0":
	s = '%s<a href="?sstring=%s;offset=%s">[Previous]</a> ' % (s,sstring,int(offset) - config.MAXPERPAGE)
if len(results) == config.MAXPERPAGE:
	s = '%s<a href="?sstring=%s;offset=%s">[Next]</a> ' % (s,sstring,int(offset) + config.MAXPERPAGE)

sys.stdout.write(s)
write_to_cache(s)