]>
Commit | Line | Data |
---|---|---|
b3102128 MF |
1 | import tornado.ioloop |
2 | import tornado.web | |
ddf4123d | 3 | import os, os.path |
b3102128 | 4 | import tornado.httpserver |
ddf4123d MF |
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 | if hub_mode == 'unsubscribe': | |
34 | pass #FIXME | |
35 | path = hub_topic.split(self.settings['domain'])[1] | |
36 | user = path.split('user/')[1] | |
37 | row = db.execute("select id from user where name=?",(user,)).fetchone() | |
38 | if row: | |
39 | db.execute("INSERT into subscriptions (userid, expires, callback, verified) values (?,?,?,?)",(row['id'],datetime.datetime.now(),hub_callback,False)) | |
40 | db.commit() | |
41 | self.set_status(202) | |
b3102128 MF |
42 | class XrdHandler(tornado.web.RequestHandler): |
43 | def get(self): | |
ddf4123d MF |
44 | self.render("templates/xrd.xml", hostname="ronin.frykholm.com", url=self.settings['domain']) |
45 | ||
46 | class FingerHandler(tornado.web.RequestHandler): | |
47 | def get(self): | |
48 | user = self.get_argument('resource') | |
49 | user = user.split('acct:')[1] | |
50 | (user,domain) = user.split('@') | |
51 | rows = db.execute("select id from user where user.name=?",(user,)).fetchone() | |
52 | if not rows: | |
53 | self.set_status(404) | |
54 | self.write("Not found") | |
55 | self.finish() | |
56 | return | |
57 | lnk = Link(rel='http://spec.example.net/photo/1.0', | |
58 | type='image/jpeg', | |
59 | href='{}/static/{}.jpg'.format(self.settings['domain'],user)) | |
60 | lnk.titles.append(('User Photo', 'en')) | |
61 | lnk.titles.append(('Benutzerfoto', 'de')) | |
62 | lnk.properties.append(('http://spec.example.net/created/1.0', '1970-01-01')) | |
63 | lnk2 = Link(rel='http://schemas.google.com/g/2010#updates-from', | |
64 | type='application/atom+xml', | |
65 | href='{}/user/{}'.format(self.settings['domain'],user)) | |
66 | ||
67 | rd = RD(subject='{}/{}'.format(self.settings['domain'],user)) | |
68 | rd.properties.append('http://spec.example.net/type/person') | |
69 | rd.links.append(lnk) | |
70 | rd.links.append(lnk2) | |
71 | self.write(rd.to_json()) | |
72 | #self.render("templates/user.xml", user=user) | |
b3102128 MF |
73 | |
74 | class UserHandler(tornado.web.RequestHandler): | |
75 | def get(self, user): | |
ddf4123d MF |
76 | entries = db.execute("select entry.id,text,ts from user,entry where user.id=entry.userid and user.name=?",(user,)) |
77 | #import pdb;pdb.set_trace() | |
78 | self.set_header("Content-Type", 'application/atom+xml') | |
79 | self.render("templates/feed.xml", | |
80 | user=user, | |
81 | feed_url="{}/user/{}".format(self.settings['domain'], user), | |
82 | hub_url="{}/hub".format(self.settings['domain']), | |
83 | entries=entries, | |
84 | arrow=arrow ) | |
b3102128 MF |
85 | |
86 | application = tornado.web.Application([ | |
87 | (r"/.well-known/host-meta", XrdHandler), | |
ddf4123d | 88 | (r"/.well-known/webfinger", FingerHandler), |
b3102128 | 89 | (r"/user/(.+)", UserHandler), |
ddf4123d MF |
90 | (r"/hub", PushHandler), |
91 | ],debug=True,**settings) | |
b3102128 | 92 | srv = tornado.httpserver.HTTPServer(application, ssl_options={ |
ddf4123d MF |
93 | "certfile": "ronin.frykholm.com.pem", |
94 | "keyfile": "ronin.frykholm.com.key", | |
b3102128 | 95 | }) |
ddf4123d MF |
96 | def setup_db(path): |
97 | print("No db found, creating in {}".format(path)) | |
98 | con = sqlite3.connect(path) | |
99 | con.execute(""" create table user (id integer primary key, | |
100 | name varchar, | |
101 | email varchar); | |
102 | create table entry (id integer primary key, | |
103 | userid INTEGER, | |
104 | text varchar, | |
105 | ts timestamp default current_timestamp, | |
106 | FOREIGN KEY(userid) REFERENCES user(id)); | |
107 | create table subscriptions (id integer primary key, | |
108 | userid integer, | |
109 | expires datetime, | |
110 | callback varchar, | |
111 | verified bool, | |
112 | FOREIGN KEY(userid) REFERENCES user(id));""") | |
113 | con.commit() | |
b3102128 | 114 | |
b3102128 | 115 | if __name__ == "__main__": |
ddf4123d MF |
116 | dbPath = 'friends.db' |
117 | tornado.options.parse_command_line() | |
118 | if not os.path.exists(dbPath): | |
119 | setup_db(dbPath) | |
120 | db = sqlite3.connect(dbPath, detect_types=sqlite3.PARSE_DECLTYPES|sqlite3.PARSE_COLNAMES) | |
121 | db.row_factory = sqlite3.Row | |
b3102128 | 122 | srv.listen(443) |
3ffcd3f8 | 123 | tornado.ioloop.IOLoop.instance().start() |