aboutsummaryrefslogtreecommitdiff
path: root/serve.py
blob: d6e074bfa31f981c566d2e658dd7e2907f399c79 (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
#!/usr/bin/env python3
import gevent.monkey

gevent.monkey.patch_all()
import bottle
import gevent.lock

import argparse
import os
import pathlib
import typing

import tagrss

parser = argparse.ArgumentParser()
parser.add_argument("--host", default="localhost")
parser.add_argument("--port", default=8000, type=int)
parser.add_argument("--storage-path", required=True)
args = parser.parse_args()

storage_path: pathlib.Path = pathlib.Path(args.storage_path)

tagrss_lock = gevent.lock.RLock()
core = tagrss.TagRss(storage_path=storage_path)


def parse_space_separated_tags(inp: str) -> list[str]:
    tags = set()
    tag = ""
    escaped = False
    for c in inp:
        match c:
            case "\\":
                if not escaped:
                    escaped = True
                    continue
            case " ":
                if not escaped:
                    tags.add(tag)
                    tag = ""
                    continue
        escaped = False
        tag += c
    if tag:
        tags.add(tag)
    return list(sorted(tags))


def serialise_tags(tags: list[str]) -> str:
    result = ""
    for i, tag in enumerate(tags):
        if i > 0:
            result += " "
        result += (tag.replace("\\", "\\\\")).replace(" ", "\\ ")
    return result


@bottle.route("/")
def index():
    with tagrss_lock:
        entries = core.get_entries(limit=100)
        return bottle.template("index", entries=entries, core=core)


@bottle.get("/add_feed")
def add_feed_view():
    return bottle.template("add_feed")


@bottle.post("/add_feed")
def add_feed_effect():
    feed_source: str = bottle.request.forms.get("feed_source")  # type: ignore
    tags = parse_space_separated_tags(bottle.request.forms.get("tags"))  # type: ignore

    already_present: bool = False
    with tagrss_lock:
        try:
            core.add_feed(feed_source=feed_source, tags=tags)
        except tagrss.FeedAlreadyAddedError:
            already_present = True
        # TODO: handle FeedFetchError too
    return bottle.template(
        "add_feed",
        after_add=True,
        feed_source=feed_source,
        already_present=already_present,
    )


@bottle.get("/manage_feed")
def manage_feed_view():
    try:
        feed_id_raw: str = bottle.request.query["feed"]  # type: ignore
        feed_id: int = int(feed_id_raw)
    except KeyError:
        raise bottle.HTTPError(400, "Feed ID not given.")
    feed: dict[str, typing.Any] = {}
    feed["id"] = feed_id
    feed["source"] = core.get_feed_source(feed_id)
    feed["title"] = core.get_feed_title(feed_id)
    feed["tags"] = core.get_feed_tags(feed_id)
    feed["serialised_tags"] = serialise_tags(feed["tags"])
    return bottle.template("manage_feed", feed=feed)

@bottle.post("/manage_feed")
def manage_feed_effect_update():
    feed_id: int = int(bottle.request.forms["id"]) # type: ignore
    feed_source: str = bottle.request.forms["source"] # type: ignore
    feed_title: str = bottle.request.forms["title"] # type: ignore
    feed_tags: list[str] = parse_space_separated_tags(bottle.request.forms["tags"]) # type: ignore
    core.set_feed_source(feed_id, feed_source)
    core.set_feed_title(feed_id, feed_title)
    core.set_feed_tags(feed_id, feed_tags)
    return bottle.redirect(f"/manage_feed?feed={feed_id}")

@bottle.get("/delete")
def delete_view():
    return bottle.static_file("delete.html", root="views")


@bottle.post("/delete")
def delete_effect():
    feed_id: int = int(bottle.request.forms["id"])  # type: ignore
    core.delete_feed(feed_id)
    return bottle.redirect("/delete")


@bottle.get("/static/<path:path>")
def serve_static(path):
    return bottle.static_file(path, pathlib.Path(os.getcwd(), "static"))


bottle.run(host=args.host, port=args.port, server="gevent")
core.close()