summaryrefslogtreecommitdiff
path: root/py-fake-service/app/pixelated_user_agent.py
blob: 41407b1c73a65c2592df6ea6c01bca254c7bbf09 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
from flask import Flask, request, Response, redirect

import json
import datetime
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://example.wazokazi.is:8154\
                    /go/static/py-mediumtagged.tar.gz'
client = None
converter = None
account = None
loaded = 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)


@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 loaded
    if not loaded:
        load_mailset('mediumtagged')
        loaded = True

    return app.send_static_file('index.html')


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


if __name__ == '__main__':
    setup()