# # Copyright (c) 2014 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
from flask import Flask, request, Response, redirect
import os
import csv
import json
import datetime
import mailbox
import StringIO
import requests
from adapter import MailService
from search import SearchQuery

app = Flask(__name__, static_url_path='', static_folder='../../web-ui/app')
MEDIUM_TAGGED_URL = 'https://static.wazokazi.is/py-mediumtagged.tar.gz'
client = None
converter = None
account = None
autoload = os.environ.get('AUTOLOAD', False)
mail_service = MailService()


def respond_json(entity):
    response = json.dumps(entity)
    return Response(response=response, mimetype="application/json")


@app.route('/disabled_features')
def disabled_features():
    return respond_json([])


@app.route('/mails', methods=['POST'])
def save_draft_or_send():
    mail = request.json
    if mail['ident']:
        ident = mail_service.send(mail)
    else:
        ident = mail_service.save_draft(mail)

    return respond_json({'ident': ident})


@app.route('/mails', methods=['PUT'])
def update_draft():
    mail = request.json
    ident = mail_service.update_draft(mail)
    return respond_json({'ident': ident})


@app.route('/mails')
def mails():
    query = SearchQuery.compile(request.args.get('q', ''))
    page = request.args.get('p', '')
    window_size = request.args.get('w', '')
    fetched_mails = mail_service.mails(query, page, window_size)

    mails = [mail.__dict__ for mail in fetched_mails]
    response = {
        "stats": {
            "total": len(mails),
            "read": 0,
            "starred": 0,
            "replied": 0
        },
        "mails": mails
    }

    return respond_json(response)


@app.route('/mail/<int:mail_id>', methods=['DELETE'])
def delete_mails(mail_id):
    mail_service.delete_mail(mail_id)
    return respond_json(None)


@app.route('/tags')
def tags():
    tags = mail_service.tagsset.all_tags()
    return respond_json([tag.__dict__ for tag in tags])


@app.route('/mail/<int:mail_id>')
def mail(mail_id):
    return respond_json(mail_service.mail(mail_id).__dict__)


@app.route('/mail/<int:mail_id>/tags', methods=['POST'])
def mail_tags(mail_id):
    new_tags = request.json['newtags']
    mail_service.update_tags_for(mail_id, new_tags)
    return respond_json(request.json['newtags'])


@app.route('/mail/<int:mail_id>/read', methods=['POST'])
def mark_mail_as_read(mail_id):
    mail_service.mark_as_read(mail_id)
    return ""


@app.route('/contacts')
def contacts():
    contacts_query = request.args.get('q')
    return respond_json(
        {'contacts': mail_service.search_contacts(contacts_query)})


@app.route('/draft_reply_for/<int:mail_id>')
def draft_reply_for(mail_id):
    mail = mail_service.draft_reply_for(mail_id)
    if mail:
        return respond_json(mail.__dict__)
    else:
        return respond_json(None)


def utf_8_encoder(unicode_csv_data):
    for line in unicode_csv_data:
        yield line.encode('utf-8')


@app.route('/control/mailset/csv/load', methods=['POST'])
def load_mailset_from_csv():
    csv_data = request.form.keys()[0]
    csv_string = StringIO.StringIO(csv_data)
    csv_reader = csv.reader(utf_8_encoder(csv_string))
    headers = csv_reader.next()
    messages = []
    for row in csv_reader:
        mail = ""
        row[3] = ', '.join(filter(len, row[3].split(' ')))
        for header, value in zip(headers, row):
            if header == 'Body':
                mail += "\n"
            else:
                mail += header + ": "
            mail += value + "\n"
        messages.append(mailbox.mboxMessage(mail))
    mail_service.index_messages(messages)
    return respond_json(None)


@app.route('/control/mailset/<mailset>/load', methods=['POST'])
def load_mailset(mailset):
    import os
    from tarfile import TarFile
    from gzip import GzipFile
    mbox_root = os.path.join(os.environ['HOME'], 'mailsets')
    if not os.path.isdir(os.path.join(mbox_root)):
        os.mkdir(mbox_root)

    if len(os.listdir(mbox_root)) == 0:
        response = requests.get(MEDIUM_TAGGED_URL, verify=False)
        mbox_archive_path = os.path.join(mbox_root, 'py-mediumtagged.tar.gz')
        mbox_archive = open(mbox_archive_path, 'w')
        mbox_archive.write(response.content)
        mbox_archive.close()
        gzippedfile = GzipFile(filename=mbox_archive_path)
        tarfile = TarFile(fileobj=gzippedfile)
        tarfile.extractall(path=mbox_root)

    mail_service.reset()
    mail_service.load_mailset()

    return respond_json(None)


@app.route('/')
def index():
    global autoload
    if autoload:
        load_mailset('mediumtagged')
        autoload = False

    return app.send_static_file('index.html')


def setup():
    app.run(host="0.0.0.0", debug=True, port=4567)


if __name__ == '__main__':
    setup()