mirror of https://gitlab.com/bashrc2/epicyon
312 lines
15 KiB
Python
312 lines
15 KiB
Python
__filename__ = "manualapprove.py"
|
|
__author__ = "Bob Mottram"
|
|
__license__ = "AGPL3+"
|
|
__version__ = "1.2.0"
|
|
__maintainer__ = "Bob Mottram"
|
|
__email__ = "bob@libreserver.org"
|
|
__status__ = "Production"
|
|
__module_group__ = "ActivityPub"
|
|
|
|
import os
|
|
from follow import followed_account_accepts
|
|
from follow import followed_account_rejects
|
|
from follow import remove_from_follow_requests
|
|
from utils import load_json
|
|
from utils import remove_domain_port
|
|
from utils import get_port_from_domain
|
|
from utils import get_user_paths
|
|
from utils import acct_dir
|
|
from threads import thread_with_trace
|
|
|
|
|
|
def manual_deny_follow_request(session, base_dir: str,
|
|
http_prefix: str,
|
|
nickname: str, domain: str, port: int,
|
|
deny_handle: str,
|
|
federation_list: [],
|
|
send_threads: [], post_log: [],
|
|
cached_webfingers: {}, person_cache: {},
|
|
debug: bool,
|
|
project_version: str,
|
|
signing_priv_key_pem: str) -> None:
|
|
"""Manually deny a follow request
|
|
"""
|
|
accounts_dir = acct_dir(base_dir, nickname, domain)
|
|
|
|
# has this handle already been rejected?
|
|
rejected_follows_filename = accounts_dir + '/followrejects.txt'
|
|
if os.path.isfile(rejected_follows_filename):
|
|
if deny_handle in open(rejected_follows_filename).read():
|
|
remove_from_follow_requests(base_dir, nickname, domain,
|
|
deny_handle, debug)
|
|
print(deny_handle +
|
|
' has already been rejected as a follower of ' + nickname)
|
|
return
|
|
|
|
remove_from_follow_requests(base_dir, nickname, domain, deny_handle, debug)
|
|
|
|
# Store rejected follows
|
|
try:
|
|
with open(rejected_follows_filename, 'a+') as rejects_file:
|
|
rejects_file.write(deny_handle + '\n')
|
|
except OSError:
|
|
print('EX: unable to append ' + rejected_follows_filename)
|
|
|
|
deny_nickname = deny_handle.split('@')[0]
|
|
deny_domain = \
|
|
deny_handle.split('@')[1].replace('\n', '').replace('\r', '')
|
|
deny_port = port
|
|
if ':' in deny_domain:
|
|
deny_port = get_port_from_domain(deny_domain)
|
|
deny_domain = remove_domain_port(deny_domain)
|
|
followed_account_rejects(session, base_dir, http_prefix,
|
|
nickname, domain, port,
|
|
deny_nickname, deny_domain, deny_port,
|
|
federation_list,
|
|
send_threads, post_log,
|
|
cached_webfingers, person_cache,
|
|
debug, project_version,
|
|
signing_priv_key_pem)
|
|
|
|
print('Follow request from ' + deny_handle + ' was denied.')
|
|
|
|
|
|
def manual_deny_follow_request_thread(session, base_dir: str,
|
|
http_prefix: str,
|
|
nickname: str, domain: str, port: int,
|
|
deny_handle: str,
|
|
federation_list: [],
|
|
send_threads: [], post_log: [],
|
|
cached_webfingers: {}, person_cache: {},
|
|
debug: bool,
|
|
project_version: str,
|
|
signing_priv_key_pem: str) -> None:
|
|
"""Manually deny a follow request, within a thread so that the
|
|
user interface doesn't lag
|
|
"""
|
|
thr = \
|
|
thread_with_trace(target=manual_deny_follow_request,
|
|
args=(session, base_dir,
|
|
http_prefix,
|
|
nickname, domain, port,
|
|
deny_handle,
|
|
federation_list,
|
|
send_threads, post_log,
|
|
cached_webfingers, person_cache,
|
|
debug,
|
|
project_version,
|
|
signing_priv_key_pem), daemon=True)
|
|
thr.start()
|
|
send_threads.append(thr)
|
|
|
|
|
|
def _approve_follower_handle(account_dir: str, approve_handle: str) -> None:
|
|
""" Record manually approved handles so that if they unfollow and then
|
|
re-follow later then they don't need to be manually approved again
|
|
"""
|
|
approved_filename = account_dir + '/approved.txt'
|
|
if os.path.isfile(approved_filename):
|
|
if approve_handle not in open(approved_filename).read():
|
|
try:
|
|
with open(approved_filename, 'a+') as approved_file:
|
|
approved_file.write(approve_handle + '\n')
|
|
except OSError:
|
|
print('EX: unable to append ' + approved_filename)
|
|
else:
|
|
try:
|
|
with open(approved_filename, 'w+') as approved_file:
|
|
approved_file.write(approve_handle + '\n')
|
|
except OSError:
|
|
print('EX: unable to write ' + approved_filename)
|
|
|
|
|
|
def manual_approve_follow_request(session, base_dir: str,
|
|
http_prefix: str,
|
|
nickname: str, domain: str, port: int,
|
|
approve_handle: str,
|
|
federation_list: [],
|
|
send_threads: [], post_log: [],
|
|
cached_webfingers: {}, person_cache: {},
|
|
debug: bool,
|
|
project_version: str,
|
|
signing_priv_key_pem: str) -> None:
|
|
"""Manually approve a follow request
|
|
"""
|
|
handle = nickname + '@' + domain
|
|
print('Manual follow accept: ' + handle +
|
|
' approving follow request from ' + approve_handle)
|
|
account_dir = base_dir + '/accounts/' + handle
|
|
approve_follows_filename = account_dir + '/followrequests.txt'
|
|
if not os.path.isfile(approve_follows_filename):
|
|
print('Manual follow accept: follow requests file ' +
|
|
approve_follows_filename + ' not found')
|
|
return
|
|
|
|
# is the handle in the requests file?
|
|
approve_follows_str = ''
|
|
with open(approve_follows_filename, 'r') as fp_foll:
|
|
approve_follows_str = fp_foll.read()
|
|
exists = False
|
|
approve_handle_full = approve_handle
|
|
if approve_handle in approve_follows_str:
|
|
exists = True
|
|
elif '@' in approve_handle:
|
|
group_account = False
|
|
if approve_handle.startswith('!'):
|
|
group_account = True
|
|
req_nick = approve_handle.split('@')[0].replace('!', '')
|
|
req_domain = approve_handle.split('@')[1].strip()
|
|
req_prefix = http_prefix + '://' + req_domain
|
|
paths = get_user_paths()
|
|
for user_path in paths:
|
|
if req_prefix + user_path + req_nick in approve_follows_str:
|
|
exists = True
|
|
approve_handle_full = req_prefix + user_path + req_nick
|
|
if group_account:
|
|
approve_handle_full = '!' + approve_handle_full
|
|
break
|
|
if not exists:
|
|
print('Manual follow accept: ' + approve_handle_full +
|
|
' not in requests file "' +
|
|
approve_follows_str.replace('\n', ' ') +
|
|
'" ' + approve_follows_filename)
|
|
return
|
|
|
|
with open(approve_follows_filename + '.new', 'w+') as approvefilenew:
|
|
update_approved_followers = False
|
|
follow_activity_filename = None
|
|
with open(approve_follows_filename, 'r') as approvefile:
|
|
for handle_of_follow_requester in approvefile:
|
|
# is this the approved follow?
|
|
if handle_of_follow_requester.startswith(approve_handle_full):
|
|
handle_of_follow_requester = \
|
|
handle_of_follow_requester.replace('\n', '')
|
|
handle_of_follow_requester = \
|
|
handle_of_follow_requester.replace('\r', '')
|
|
port2 = port
|
|
if ':' in handle_of_follow_requester:
|
|
port2 = \
|
|
get_port_from_domain(handle_of_follow_requester)
|
|
requests_dir = account_dir + '/requests'
|
|
follow_activity_filename = \
|
|
requests_dir + '/' + \
|
|
handle_of_follow_requester + '.follow'
|
|
if os.path.isfile(follow_activity_filename):
|
|
follow_json = load_json(follow_activity_filename)
|
|
if follow_json:
|
|
approve_nickname = approve_handle.split('@')[0]
|
|
approve_domain = approve_handle.split('@')[1]
|
|
approve_domain = \
|
|
approve_domain.replace('\n', '')
|
|
approve_domain = \
|
|
approve_domain.replace('\r', '')
|
|
approve_port = port2
|
|
if ':' in approve_domain:
|
|
approve_port = \
|
|
get_port_from_domain(approve_domain)
|
|
approve_domain = \
|
|
remove_domain_port(approve_domain)
|
|
print('Manual follow accept: Sending Accept for ' +
|
|
handle + ' follow request from ' +
|
|
approve_nickname + '@' + approve_domain)
|
|
followed_account_accepts(session, base_dir,
|
|
http_prefix,
|
|
nickname, domain, port,
|
|
approve_nickname,
|
|
approve_domain,
|
|
approve_port,
|
|
follow_json['actor'],
|
|
federation_list,
|
|
follow_json,
|
|
send_threads, post_log,
|
|
cached_webfingers,
|
|
person_cache,
|
|
debug,
|
|
project_version, False,
|
|
signing_priv_key_pem)
|
|
update_approved_followers = True
|
|
else:
|
|
# this isn't the approved follow so it will remain
|
|
# in the requests file
|
|
approvefilenew.write(handle_of_follow_requester)
|
|
|
|
followers_filename = account_dir + '/followers.txt'
|
|
if update_approved_followers:
|
|
# update the followers
|
|
print('Manual follow accept: updating ' + followers_filename)
|
|
if os.path.isfile(followers_filename):
|
|
if approve_handle_full not in open(followers_filename).read():
|
|
try:
|
|
with open(followers_filename, 'r+') as followers_file:
|
|
content = followers_file.read()
|
|
if approve_handle_full + '\n' not in content:
|
|
followers_file.seek(0, 0)
|
|
followers_file.write(approve_handle_full + '\n' +
|
|
content)
|
|
except Exception as ex:
|
|
print('WARN: Manual follow accept. ' +
|
|
'Failed to write entry to followers file ' + str(ex))
|
|
else:
|
|
print('WARN: Manual follow accept: ' + approve_handle_full +
|
|
' already exists in ' + followers_filename)
|
|
else:
|
|
print('Manual follow accept: first follower accepted for ' +
|
|
handle + ' is ' + approve_handle_full)
|
|
try:
|
|
with open(followers_filename, 'w+') as followers_file:
|
|
followers_file.write(approve_handle_full + '\n')
|
|
except OSError:
|
|
print('EX: unable to write ' + followers_filename)
|
|
|
|
# only update the follow requests file if the follow is confirmed to be
|
|
# in followers.txt
|
|
if approve_handle_full in open(followers_filename).read():
|
|
# mark this handle as approved for following
|
|
_approve_follower_handle(account_dir, approve_handle)
|
|
# update the follow requests with the handles not yet approved
|
|
os.rename(approve_follows_filename + '.new', approve_follows_filename)
|
|
# remove the .follow file
|
|
if follow_activity_filename:
|
|
if os.path.isfile(follow_activity_filename):
|
|
try:
|
|
os.remove(follow_activity_filename)
|
|
except OSError:
|
|
print('EX: manual_approve_follow_request ' +
|
|
'unable to delete ' + follow_activity_filename)
|
|
else:
|
|
try:
|
|
os.remove(approve_follows_filename + '.new')
|
|
except OSError:
|
|
print('EX: manual_approve_follow_request unable to delete ' +
|
|
approve_follows_filename + '.new')
|
|
|
|
|
|
def manual_approve_follow_request_thread(session, base_dir: str,
|
|
http_prefix: str,
|
|
nickname: str, domain: str, port: int,
|
|
approve_handle: str,
|
|
federation_list: [],
|
|
send_threads: [], post_log: [],
|
|
cached_webfingers: {},
|
|
person_cache: {},
|
|
debug: bool,
|
|
project_version: str,
|
|
signing_priv_key_pem: str) -> None:
|
|
"""Manually approve a follow request, in a thread so as not to cause
|
|
the UI to lag
|
|
"""
|
|
thr = \
|
|
thread_with_trace(target=manual_approve_follow_request,
|
|
args=(session, base_dir,
|
|
http_prefix,
|
|
nickname, domain, port,
|
|
approve_handle,
|
|
federation_list,
|
|
send_threads, post_log,
|
|
cached_webfingers, person_cache,
|
|
debug,
|
|
project_version,
|
|
signing_priv_key_pem), daemon=True)
|
|
thr.start()
|
|
send_threads.append(thr)
|