#!/usr/bin/env python3

import os
#import sys
#sys.path.append(os.path.dirname(os.path.abspath(__file__)))

import asyncio
import uvloop
from sanic import Sanic
import sanic.response
from sanic.log import log as logging
from whoosh import index
from whoosh import qparser
import jinja2
import shared

def SearchHandler(query, tmpl):
    response = sanic.response.text(
        "You seem to have forgot to enter what you want to search for. Please try again.",
        status=400
    )

    if not query:
        return response

    query = query.replace('+', ' AND ').replace(' -', ' NOT ')
    ix = index.open_dir(os.path.abspath(os.path.join(
            shared.config.get('target', 'builddir'),
            shared.config.get('var', 'searchdb')
    )))

    qp = qparser.MultifieldParser(
        ["title", "content"],
        schema = shared.schema
    )

    q = qp.parse(query)
    r = ix.searcher().search(q, sortedby="weight", limit=100)
    logging.info("results for '%s': %i", query, len(r))
    results = {}
    for result in r:
        if result['url'] in results.keys():
            continue

        res = {
            'title': result['title'],
            #'url': result['url'],
            'highlight': result.highlights("content"),
        }
        if 'img' in result:
            res['img'] = result['img']
        results[result['url']] = res

    tvars = {
        'term': query,
        'posts': results,
    }

    logging.info("collected %i results to render", len(results))
    response = sanic.response.html(tmpl.render(tvars), status=200)
    return response

if __name__ == '__main__':
    asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
    app = Sanic()


    jldr = jinja2.FileSystemLoader(
        searchpath=shared.config.get('source', 'templatesdir')
    )
    jenv = jinja2.Environment(loader=jldr)
    tmpl = jenv.get_template('searchresults.html')

    @app.route("/search", methods=["GET"])
    async def search(request):
        query = request.args.get('s')
        r = SearchHandler(query, tmpl)
        return r

    app.run(host="127.0.0.1", port=8001, debug=True)