]>
Commit | Line | Data |
---|---|---|
1 | import tornado.ioloop | |
2 | import tornado.web | |
3 | import os, os.path | |
4 | import tornado.httpserver | |
5 | import sqlite3 | |
6 | import arrow | |
7 | import datetime | |
8 | from rd import RD, Link | |
9 | db = None | |
10 | #insert into user (name,email) values('mikael','mikael@frykholm.com'); | |
11 | #insert into entry (userid,text) values (1,'My thoughts on ostatus'); | |
12 | import tornado.options | |
13 | ||
14 | settings = { | |
15 | "static_path": os.path.join(os.path.dirname(__file__), "static"), | |
16 | "cookie_secret": "__TODO:_GENERATE_YOUR_OWN_RANDOM_VALUE_HERE__", | |
17 | "login_url": "/login", | |
18 | "xsrf_cookies": False, | |
19 | "domain":"https://ronin.frykholm.com", | |
20 | ||
21 | } | |
22 | class PushHandler(tornado.web.RequestHandler): | |
23 | #curl -v -k "https://ronin.frykholm.com/hub" -d "hub.callback=a" -d "hub.mode=b" -d "hub.topic=c" -d "hub.verify=d" | |
24 | def post(self): | |
25 | """ Someone wants to subscribe to hub_topic feed""" | |
26 | hub_callback = self.get_argument('hub.callback') | |
27 | hub_mode = self.get_argument('hub.mode') | |
28 | hub_topic = self.get_argument('hub.topic') | |
29 | hub_verify = self.get_argument('hub.verify') | |
30 | hub_lease_seconds = self.get_argument('hub.lease_seconds','') | |
31 | hub_secret = self.get_argument('hub.sercret','') | |
32 | hub_verify_token = self.get_argument('hub.verify_token','') | |
33 | print(self.request.body) | |
34 | if hub_mode == 'unsubscribe': | |
35 | pass #FIXME | |
36 | path = hub_topic.split(self.settings['domain'])[1] | |
37 | user = path.split('user/')[1] | |
38 | row = db.execute("select id from user where name=?",(user,)).fetchone() | |
39 | if row: | |
40 | db.execute("INSERT into subscriptions (userid, expires, callback, verified) values (?,?,?,?)",(row['id'],datetime.datetime.now(),hub_callback,False)) | |
41 | db.commit() | |
42 | self.set_status(202) | |
43 | #TODO add GET callback with the same data we got | |
44 | #TODO store secret, add it to outgoing feeds with hmac | |
45 | class XrdHandler(tornado.web.RequestHandler): | |
46 | def get(self): | |
47 | self.render("templates/xrd.xml", hostname="ronin.frykholm.com", url=self.settings['domain']) | |
48 | ||
49 | class FingerHandler(tornado.web.RequestHandler): | |
50 | def get(self): | |
51 | user = self.get_argument('resource') | |
52 | user = user.split('acct:')[1] | |
53 | (user,domain) = user.split('@') | |
54 | rows = db.execute("select id from user where user.name=?",(user,)).fetchone() | |
55 | if not rows: | |
56 | self.set_status(404) | |
57 | self.write("Not found") | |
58 | self.finish() | |
59 | return | |
60 | lnk = Link(rel='http://spec.example.net/photo/1.0', | |
61 | type='image/jpeg', | |
62 | href='{}/static/{}.jpg'.format(self.settings['domain'],user)) | |
63 | lnk.titles.append(('User Photo', 'en')) | |
64 | lnk.titles.append(('Benutzerfoto', 'de')) | |
65 | lnk.properties.append(('http://spec.example.net/created/1.0', '1970-01-01')) | |
66 | lnk2 = Link(rel='http://schemas.google.com/g/2010#updates-from', | |
67 | type='application/atom+xml', | |
68 | href='{}/user/{}'.format(self.settings['domain'],user)) | |
69 | ||
70 | rd = RD(subject='{}/{}'.format(self.settings['domain'],user)) | |
71 | rd.properties.append('http://spec.example.net/type/person') | |
72 | rd.links.append(lnk) | |
73 | rd.links.append(lnk2) | |
74 | self.write(rd.to_json()) | |
75 | #self.render("templates/user.xml", user=user) | |
76 | ||
77 | class UserHandler(tornado.web.RequestHandler): | |
78 | def get(self, user): | |
79 | entries = db.execute("select entry.id,text,ts from user,entry where user.id=entry.userid and user.name=?",(user,)) | |
80 | #import pdb;pdb.set_trace() | |
81 | self.set_header("Content-Type", 'application/atom+xml') | |
82 | self.render("templates/feed.xml", | |
83 | user=user, | |
84 | feed_url="{}/user/{}".format(self.settings['domain'], user), | |
85 | hub_url="{}/hub".format(self.settings['domain']), | |
86 | entries=entries, | |
87 | arrow=arrow ) | |
88 | ||
89 | application = tornado.web.Application([ | |
90 | (r"/.well-known/host-meta", XrdHandler), | |
91 | (r"/.well-known/webfinger", FingerHandler), | |
92 | (r"/user/(.+)", UserHandler), | |
93 | (r"/hub", PushHandler), | |
94 | ],debug=True,**settings) | |
95 | srv = tornado.httpserver.HTTPServer(application, ssl_options={ | |
96 | "certfile": "ronin.frykholm.com.pem", | |
97 | "keyfile": "ronin.frykholm.com.key", | |
98 | }) | |
99 | def setup_db(path): | |
100 | print("No db found, creating in {}".format(path)) | |
101 | con = sqlite3.connect(path) | |
102 | con.execute(""" create table user (id integer primary key, | |
103 | name varchar, | |
104 | email varchar); | |
105 | create table entry (id integer primary key, | |
106 | userid INTEGER, | |
107 | text varchar, | |
108 | ts timestamp default current_timestamp, | |
109 | FOREIGN KEY(userid) REFERENCES user(id)); | |
110 | create table subscriptions (id integer primary key, | |
111 | userid integer, | |
112 | expires datetime, | |
113 | callback varchar, | |
114 | verified bool, | |
115 | FOREIGN KEY(userid) REFERENCES user(id));""") | |
116 | con.commit() | |
117 | ||
118 | if __name__ == "__main__": | |
119 | dbPath = 'friends.db' | |
120 | tornado.options.parse_command_line() | |
121 | if not os.path.exists(dbPath): | |
122 | setup_db(dbPath) | |
123 | db = sqlite3.connect(dbPath, detect_types=sqlite3.PARSE_DECLTYPES|sqlite3.PARSE_COLNAMES) | |
124 | db.row_factory = sqlite3.Row | |
125 | srv.listen(443) | |
126 | tornado.ioloop.IOLoop.instance().start() |