epicyon/availability.py

161 lines
5.1 KiB
Python
Raw Normal View History

2020-04-01 19:39:27 +00:00
__filename__ = "availability.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
2021-01-26 10:07:42 +00:00
__version__ = "1.2.0"
2020-04-01 19:39:27 +00:00
__maintainer__ = "Bob Mottram"
2021-09-10 16:14:50 +00:00
__email__ = "bob@libreserver.org"
2020-04-01 19:39:27 +00:00
__status__ = "Production"
2021-06-26 11:16:41 +00:00
__module_group__ = "Profile Metadata"
2020-04-01 19:39:27 +00:00
2019-07-19 11:38:37 +00:00
import os
from webfinger import webfingerHandle
from auth import createBasicAuthHeader
from posts import getPersonBox
from session import postJson
2021-10-13 11:15:06 +00:00
from utils import hasObjectString
2020-12-16 10:30:54 +00:00
from utils import getFullDomain
2019-07-19 11:38:37 +00:00
from utils import getNicknameFromActor
from utils import getDomainFromActor
2019-10-22 11:55:06 +00:00
from utils import loadJson
from utils import saveJson
2021-07-13 21:59:53 +00:00
from utils import acctDir
2021-08-14 11:13:39 +00:00
from utils import localActorUrl
from utils import hasActor
2019-07-19 11:38:37 +00:00
2020-04-01 19:39:27 +00:00
def setAvailability(baseDir: str, nickname: str, domain: str,
2019-07-19 11:38:37 +00:00
status: str) -> bool:
"""Set an availability status
"""
# avoid giant strings
2020-04-01 19:39:27 +00:00
if len(status) > 128:
2019-07-19 11:38:37 +00:00
return False
2021-07-13 21:59:53 +00:00
actorFilename = acctDir(baseDir, nickname, domain) + '.json'
2019-07-19 11:38:37 +00:00
if not os.path.isfile(actorFilename):
return False
2020-04-01 19:39:27 +00:00
actorJson = loadJson(actorFilename)
2019-09-30 22:39:02 +00:00
if actorJson:
2020-04-01 19:39:27 +00:00
actorJson['availability'] = status
saveJson(actorJson, actorFilename)
2019-07-19 11:38:37 +00:00
return True
2020-04-01 19:39:27 +00:00
def getAvailability(baseDir: str, nickname: str, domain: str) -> str:
2019-07-19 11:38:37 +00:00
"""Returns the availability for a given person
"""
2021-07-13 21:59:53 +00:00
actorFilename = acctDir(baseDir, nickname, domain) + '.json'
2019-07-19 11:38:37 +00:00
if not os.path.isfile(actorFilename):
return False
2020-04-01 19:39:27 +00:00
actorJson = loadJson(actorFilename)
2019-09-30 22:39:02 +00:00
if actorJson:
2019-07-19 11:38:37 +00:00
if not actorJson.get('availability'):
return None
return actorJson['availability']
return None
2020-04-01 19:39:27 +00:00
def outboxAvailability(baseDir: str, nickname: str, messageJson: {},
2019-11-03 15:27:29 +00:00
debug: bool) -> bool:
2019-07-19 11:38:37 +00:00
"""Handles receiving an availability update
"""
if not messageJson.get('type'):
return False
2020-04-01 19:39:27 +00:00
if not messageJson['type'] == 'Availability':
2019-07-19 11:38:37 +00:00
return False
if not hasActor(messageJson, debug):
2019-07-19 11:38:37 +00:00
return False
2021-10-13 11:15:06 +00:00
if not hasObjectString(messageJson, debug):
2019-07-19 11:38:37 +00:00
return False
2020-04-01 19:39:27 +00:00
actorNickname = getNicknameFromActor(messageJson['actor'])
if actorNickname != nickname:
2019-07-19 11:38:37 +00:00
return False
2020-04-01 19:39:27 +00:00
domain, port = getDomainFromActor(messageJson['actor'])
status = messageJson['object'].replace('"', '')
return setAvailability(baseDir, nickname, domain, status)
def sendAvailabilityViaServer(baseDir: str, session,
nickname: str, password: str,
domain: str, port: int,
httpPrefix: str,
status: str,
cachedWebfingers: {}, personCache: {},
debug: bool, projectVersion: str,
signingPrivateKeyPem: str) -> {}:
2019-07-19 11:38:37 +00:00
"""Sets the availability for a person via c2s
"""
if not session:
print('WARN: No session for sendAvailabilityViaServer')
return 6
2020-12-16 10:30:54 +00:00
domainFull = getFullDomain(domain, port)
2020-03-22 21:16:02 +00:00
2021-08-14 11:13:39 +00:00
toUrl = localActorUrl(httpPrefix, nickname, domainFull)
ccUrl = toUrl + '/followers'
2019-07-19 11:38:37 +00:00
2020-04-01 19:39:27 +00:00
newAvailabilityJson = {
2019-07-19 11:38:37 +00:00
'type': 'Availability',
2021-08-14 11:13:39 +00:00
'actor': toUrl,
2021-06-22 12:42:52 +00:00
'object': '"' + status + '"',
2019-07-19 11:38:37 +00:00
'to': [toUrl],
'cc': [ccUrl]
}
2020-04-01 19:39:27 +00:00
handle = httpPrefix + '://' + domainFull + '/@' + nickname
2019-07-19 11:38:37 +00:00
# lookup the inbox for the To handle
2020-04-01 19:39:27 +00:00
wfRequest = webfingerHandle(session, handle, httpPrefix,
cachedWebfingers,
domain, projectVersion, debug, False,
signingPrivateKeyPem)
2019-07-19 11:38:37 +00:00
if not wfRequest:
if debug:
2021-03-18 10:01:01 +00:00
print('DEBUG: availability webfinger failed for ' + handle)
2019-07-19 11:38:37 +00:00
return 1
2020-06-23 10:41:12 +00:00
if not isinstance(wfRequest, dict):
2021-03-18 10:01:01 +00:00
print('WARN: availability webfinger for ' + handle +
' did not return a dict. ' + str(wfRequest))
2020-06-23 10:41:12 +00:00
return 1
2019-07-19 11:38:37 +00:00
2020-04-01 19:39:27 +00:00
postToBox = 'outbox'
2019-07-19 11:38:37 +00:00
# get the actor inbox for the To handle
2021-09-15 14:05:08 +00:00
originDomain = domain
(inboxUrl, pubKeyId, pubKey, fromPersonId, sharedInbox, avatarUrl,
displayName, _) = getPersonBox(signingPrivateKeyPem,
originDomain,
baseDir, session, wfRequest,
personCache, projectVersion,
httpPrefix, nickname,
domain, postToBox, 57262)
2020-03-22 21:16:02 +00:00
2019-07-19 11:38:37 +00:00
if not inboxUrl:
if debug:
2021-03-18 10:01:01 +00:00
print('DEBUG: availability no ' + postToBox +
' was found for ' + handle)
2019-07-19 11:38:37 +00:00
return 3
if not fromPersonId:
if debug:
2021-03-18 10:01:01 +00:00
print('DEBUG: availability no actor was found for ' + handle)
2019-07-19 11:38:37 +00:00
return 4
2020-03-22 21:16:02 +00:00
2020-04-01 19:39:27 +00:00
authHeader = createBasicAuthHeader(nickname, password)
2020-03-22 21:16:02 +00:00
2020-04-01 19:39:27 +00:00
headers = {
'host': domain,
'Content-type': 'application/json',
2020-03-22 20:36:19 +00:00
'Authorization': authHeader
}
2021-06-20 13:39:53 +00:00
postResult = postJson(httpPrefix, domainFull,
session, newAvailabilityJson, [],
2021-03-10 19:24:52 +00:00
inboxUrl, headers, 30, True)
2020-04-01 19:39:27 +00:00
if not postResult:
2021-03-18 10:01:01 +00:00
print('WARN: availability failed to post')
2019-07-19 11:38:37 +00:00
if debug:
print('DEBUG: c2s POST availability success')
return newAvailabilityJson