__filename__ = "daemon.py" __author__ = "Bob Mottram" __license__ = "AGPL3+" __version__ = "0.0.1" __maintainer__ = "Bob Mottram" __email__ = "bob@freedombone.net" __status__ = "Production" from http.server import BaseHTTPRequestHandler, HTTPServer #import socketserver import json import cgi from pprint import pprint from session import createSession from httpsig import testHttpsig from cache import testCache from webfinger import webfingerMeta from webfinger import webfingerLookup from person import personLookup from person import personKeyLookup from person import personOutboxJson from inbox import inboxPermittedMessage from follow import getFollowingFeed import os import sys # domain name of this server thisDomain='' # List of domains to federate with federationList=[] # Avoid giant messages maxMessageLength=5000 # maximum number of posts to list in outbox feed maxPostsInFeed=20 # number of follows/followers per page followsPerPage=12 # Whether to use https useHttps=True def readFollowList(filename: str): """Returns a list of ActivityPub addresses to follow """ followlist=[] if not os.path.isfile(filename): return followlist followUsers = open(filename, "r") for u in followUsers: if u not in followlist: username,domain = parseHandle(u) if username: followlist.append(username+'@'+domain) followUsers.close() return followlist class PubServer(BaseHTTPRequestHandler): def _set_headers(self,fileFormat): self.send_response(200) self.send_header('Content-type', fileFormat) self.end_headers() def _404(self): self.send_response(404) self.send_header('Content-Type', 'text/html; charset=utf-8') self.end_headers() self.wfile.write("

404 Not Found

".encode('utf-8')) def _webfinger(self) -> bool: if not self.path.startswith('/.well-known'): return False if self.path.startswith('/.well-known/host-meta'): wfResult=webfingerMeta() if wfResult: self._set_headers('application/xrd+xml') self.wfile.write(wfResult.encode('utf-8')) return wfResult=webfingerLookup(self.path) if wfResult: self._set_headers('application/json') self.wfile.write(json.dumps(wfResult).encode('utf-8')) else: self._404() return True def _permittedDir(self,path): if path.startswith('/wfendpoints') or \ path.startswith('/keys') or \ path.startswith('/accounts'): return False return True def do_GET(self): try: if self.GETbusy: self.send_response(429) self.end_headers() return except: pass self.GETbusy=True if not self._permittedDir(self.path): self._404() self.GETbusy=False return # get webfinger endpoint for a person if self._webfinger(): self.GETbusy=False return # get outbox feed for a person outboxFeed=personOutboxJson(thisDomain,self.path,useHttps,maxPostsInFeed) if outboxFeed: self._set_headers('application/json') self.wfile.write(json.dumps(outboxFeed).encode('utf-8')) self.GETbusy=False return following=getFollowingFeed(thisDomain,self.path,useHttps,followsPerPage) if following: self._set_headers('application/json') self.wfile.write(json.dumps(following).encode('utf-8')) self.GETbusy=False return followers=getFollowingFeed(thisDomain,self.path,useHttps,followsPerPage,'followers') if followers: self._set_headers('application/json') self.wfile.write(json.dumps(followers).encode('utf-8')) self.GETbusy=False return # look up a person getPerson = personLookup(thisDomain,self.path) if getPerson: self._set_headers('application/json') self.wfile.write(json.dumps(getPerson).encode('utf-8')) self.GETbusy=False return personKey = personKeyLookup(thisDomain,self.path) if personKey: self._set_headers('text/html; charset=utf-8') self.wfile.write(personKey.encode('utf-8')) self.GETbusy=False return # check that a json file was requested if not self.path.endswith('.json'): self._404() self.GETbusy=False return # check that the file exists baseDir=os.getcwd() filename=baseDir+self.path if os.path.isfile(filename): self._set_headers('application/json') with open(filename, 'r', encoding='utf8') as File: content = File.read() contentJson=json.loads(content) self.wfile.write(json.dumps(contentJson).encode('utf8')) else: self._404() self.GETbusy=False def do_HEAD(self): self._set_headers('application/json') def do_POST(self): try: if self.POSTbusy: self.send_response(429) self.end_headers() return except: pass self.POSTbusy=True ctype, pdict = cgi.parse_header(self.headers.getheader('content-type')) # refuse to receive non-json content if ctype != 'application/json': self.send_response(400) self.end_headers() self.POSTbusy=False return # read the message and convert it into a python dictionary length = int(self.headers.getheader('content-length')) if length>maxMessageLength: self.send_response(400) self.end_headers() self.POSTbusy=False return message = json.loads(self.rfile.read(length)) if not inboxPermittedMessage(message,federationList): self.send_response(403) self.end_headers() else: # add a property to the object, just to mess with data message['received'] = 'ok' # send the message back self._set_headers('application/json') self.wfile.write(json.dumps(message).encode('utf-8')) self.POSTbusy=False def runDaemon(domain: str,port=80,fedList=[],useTor=False) -> None: global thisDomain global federationList thisDomain=domain federationList=fedList.copy() if len(domain)==0: domain='127.0.0.1' if '.' not in domain: print('Invalid domain: ' + domain) return session = createSession(useTor) print('Running tests...') testHttpsig() testCache() print('Tests succeeded\n') serverAddress = ('', port) httpd = HTTPServer(serverAddress, PubServer) print('Running ActivityPub daemon on ' + thisDomain + ' port ' + str(port)) httpd.serve_forever()