from flask import Flask

from flask import request

from flask import render_template, url_for

from flask import make_response

from flask import jsonify

import logging

from logging import FileHandler

import pymongo

import os

import lxml.html

import urllib2

import StringIO

import json

import urllib

import conf

import sweetmaker

app = Flask(__name__)

SWEET_STORE_URL = conf.SWEET_STORE_URL

@app.route('/', methods=['GET'])

def index():

    if request.args.has_key('url'):

        return render_template('index.html', url=request.args['url'])

    else:

        return render_template('index.html')

@app.route('/fetch', methods=['GET'])

def fetch():

    connection = pymongo.MongoClient('localhost',27017)

    db = connection['mural']

    collection = db['data']

    ret = {}

    x = 0

    resource = "default"

    if request.args.has_key('uri'):

        resource = request.args['uri']

    for i in collection.find({'uri':resource}):

        del(i['_id'])

        ret[x] = i

        x = x + 1

    else:

        for i in collection.find():

            del(i['_id'])

            ret[x] = i

            x = x + 1

    if len(ret) == 0:

        ret['error'] = "Sorry! No re-treats for you."

    return jsonify(ret)

@app.route('/sweets', methods=['GET'])

def displaySweet():

    return render_template('sweets.html')

@app.route('/search', methods=['GET'])

def search():

    connection = pymongo.MongoClient('localhost',27017)

    db = connection['mural']

    collection = db['data']

    y = 0

    ret = {}

    keywords_dict = json.loads(request.args['data'])

    #keywords = json.loads(keywords_dict)['data']

    if 'nodes' in keywords_dict:

        for i in collection.find():

            try:

                if 'how' in i:

                    i['nodes'] = i['how']

            except:

                pass

            for node in keywords_dict['nodes']:

                try:

                    if node in i['nodes']:

                        del(i['_id'])

                        ret[y] = i

                        y = y + 1

                except:

                    pass

    elif 'where' in keywords_dict:

        for i in collection.find({'uri': keywords_dict['where']}):

            del(i['_id'])

            ret[y] = i

            y = y + 1

        for i in collection.find({'where': {'$regex':\

                                            keywords_dict['where']}}):

            del(i['_id'])

            ret[y] = i

            y = y + 1

    return render_template('blank.html', content = ret, flag =\

                           request.args['flag'])

@app.route('/submit', methods=['POST'])

def submit():

    c = pymongo.MongoClient('localhost',27017)

    db = c['mural']

    coll = db['data']

    requestData = json.loads(request.form['data'])

    try:

        for i in requestData:

            coll.insert(i)

        print 'inserted'

        print requestData

        response = make_response()

        response.headers['Access-Control-Allow-Origin'] = '*'

        response.status_code = 200

        for i in requestData:

            del(i['_id'])

            i['how'] = '{concepts: ' + ', '.join(i['how']) + '}'

            #i['how'] = attribs

        print 'payload for sweet'

        print requestData

        sweetmaker.sweet(SWEET_STORE_URL, requestData)

    except:

        response = make_response()

        response.status_code = 500

        response.data = "Your post could not be saved. Try posting again."

    return response

@app.route('/web/', methods=['GET'])

def web():

  return render_template('web.html')

@app.route('/SWeeText', methods=['GET'])

def SWeeText():

    if request.args.has_key('url'):

        # Log -- comment them

        print "Got URL " + request.args['url'] + " .. Fetching and Parsing.."

        myhandler1 = urllib2.Request(request.args['url'], headers={'User-Agent': "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:20.0) Gecko/20100101 Firefox/20.0"})

        a = urllib2.urlopen(myhandler1)

        page = a.read()

        a.close()

        try:

            page = unicode(page, 'utf-8')

        except UnicodeDecodeError:

            pass

        root = lxml.html.parse(StringIO.StringIO(page)).getroot()

        root.make_links_absolute(request.args['url'], resolve_base_href = True)

        # Log -- comment them

        #print "Page parsed.. Preparing to send.."

        # inject the JS toolbar to annotate text

        jq = root.makeelement('script')

        jq.set('src', url_for('static', filename='jquery-1.9.1.min.js'))

        script = root.makeelement('script')

        script.set('src', url_for('static', filename='text-annotation.js'))

        config_script = root.makeelement('script')

        config_script.set('src', url_for('static', filename='config.js'))

        tree = root.makeelement('script')

        tree.set('src', url_for('static', filename='tree.js'))

        bs_js = root.makeelement('script')

        bs_js.set('src', url_for('static', filename='bootstrap.js'))

        jit = root.makeelement('script')

        jit.set('src', url_for('static', filename='jit.js'))

        us = root.makeelement('script')

        us.set('src', url_for('static', filename='underscore-min-1.4.4.js'))

        link = root.makeelement('link')

        link.set('href', url_for('static', filename='text-annotation.css'))

        link.set('type', 'text/css')

        link.set('rel', 'stylesheet')

        bs = root.makeelement('link')

        bs.set('href', url_for('static', filename='bootstrap.css'))

        bs.set('type', 'text/css')

        bs.set('rel', 'stylesheet')

        tree_css = root.makeelement('link')

        tree_css.set('href', url_for('static', filename='tree.css'))

        tree_css.set('type', 'text/css')

        tree_css.set('rel', 'stylesheet')

        root.head.append(bs)

        root.head.append(link)

        root.head.append(tree_css)

        root.head.append(jq)

        root.head.append(bs_js)

        root.head.append(jit)

        root.head.append(us)

        root.head.append(tree)

        root.head.append(script)

        root.head.append(config_script)

        return lxml.html.tostring(root)

#Log the errors, don't depend on apache to log it for you.

    fil = FileHandler(os.path.join(os.path.dirname(__file__), 'logme'),mode='a')

    fil.setLevel(logging.ERROR)

    app.logger.addHandler(fil)

if __name__ == "__main__":

    app.run(debug=True, host='0.0.0.0')