forked from indymedia/epicyon
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
__filename__ = "inbox.py"
|
|
__author__ = "Bob Mottram"
|
|
__license__ = "AGPL3+"
|
|
__version__ = "0.0.1"
|
|
__maintainer__ = "Bob Mottram"
|
|
__email__ = "bob@freedombone.net"
|
|
__status__ = "Production"
|
|
|
|
import json
|
|
import os
|
|
import datetime
|
|
from utils import urlPermitted
|
|
|
|
def inboxMessageHasParams(messageJson: {}) -> bool:
|
|
"""Checks whether an incoming message contains expected parameters
|
|
"""
|
|
expectedParams=['type','to','actor','object']
|
|
for param in expectedParams:
|
|
if not messageJson.get(param):
|
|
return False
|
|
return True
|
|
|
|
def inboxPermittedMessage(domain: str,messageJson: {},federationList: []) -> bool:
|
|
""" check that we are receiving from a permitted domain
|
|
"""
|
|
testParam='actor'
|
|
if not messageJson.get(testParam):
|
|
return False
|
|
actor=messageJson[testParam]
|
|
# always allow the local domain
|
|
if domain in actor:
|
|
return True
|
|
|
|
if not urlPermitted(actor,federationList):
|
|
return False
|
|
|
|
if messageJson.get('object'):
|
|
if messageJson['object'].get('inReplyTo'):
|
|
inReplyTo=messageJson['object']['inReplyTo']
|
|
if not urlPermitted(inReplyTo, federationList):
|
|
return False
|
|
|
|
return True
|
|
|
|
def receivePublicMessage(message: {}) -> bool:
|
|
print("TODO")
|
|
|
|
def validPublishedDate(published):
|
|
currTime=datetime.datetime.utcnow()
|
|
pubDate=datetime.datetime.strptime(published,"%Y-%m-%dT%H:%M:%SZ")
|
|
daysSincePublished = (currTime - pubTime).days
|
|
if daysSincePublished>30:
|
|
return False
|
|
return True
|
|
|
|
def receiveMessage(message: {},baseDir: str):
|
|
if not message.get('type'):
|
|
return
|
|
if message['type']!='Create':
|
|
return
|
|
if not message.get('published'):
|
|
return
|
|
# is the message too old?
|
|
if not validPublishedDate(message['published']):
|
|
return
|
|
if not message.get('to'):
|
|
return
|
|
if not message.get('id'):
|
|
return
|
|
for recipient in message['to']:
|
|
if recipient.endswith('/activitystreams#Public'):
|
|
receivePublicMessage(message)
|
|
continue
|
|
|
|
username=''
|
|
domain=''
|
|
messageId=message['id'].replace('/','_')
|
|
handle=username.lower()+'@'+domain.lower()
|
|
if not os.path.isdir(baseDir+'/accounts/'+handle):
|
|
os.mkdir(baseDir+'/accounts/'+handle)
|
|
if not os.path.isdir(baseDir+'/accounts/'+handle+'/inbox'):
|
|
os.mkdir(baseDir+'/accounts/'+handle+'/inbox')
|
|
filename=baseDir+'/accounts/'+handle+'/inbox/'+messageId+'.json'
|
|
with open(filename, 'w') as fp:
|
|
commentjson.dump(personJson, fp, indent=4, sort_keys=False)
|