staticoso/Source/Modules/ActivityPub.py

""" ================================= |
| This file is part of                |
|   staticoso                         |
| Just a simple Static Site Generator |
|                                     |
| Licensed under the AGPLv3 license   |
|   Copyright (C) 2022, OctoSpacc     |
| ================================= """

import time
from Libs.dateutil.parser import parse as date_parse
from Libs.mastodon import Mastodon
from Modules.HTML import *
from Modules.Utils import *

def MastodonGetSession(InstanceURL, Token):
	return Mastodon(
		api_base_url=InstanceURL,
		access_token=Token)

def MastodonGetMyID(Session):
	return Session.me()['id']

def MastodonGetPostsFromUserID(Session, UserID):
	return Session.account_statuses(
		UserID,
		exclude_replies=True)

def MastodonDoPost(Session, Text, Lang=None, Visibility='public'):
	if Text:
		return Session.status_post(
			Text,
			language=Lang,
			visibility=Visibility)

def MastodonGetLinkFromPost(Post, Domain=None):
	Parse = MkSoup(Post['content'])
	if Parse.a:
		Link = Parse.find_all('a')[-1]['href']
		if not Domain or (Domain and Link.startswith(Domain)):
			return {
				'Post': Post['uri'],
				'Link': Link}
	return None

def MastodonGetAllLinkPosts(Session, Domain=None):
	Posts = []
	for p in MastodonGetPostsFromUserID(Session, MastodonGetMyID(Session)):
		Post = MastodonGetLinkFromPost(p, Domain)
		if Post:
			Posts += [Post]
	return Posts

def MastodonShare(Flags, Pages, Locale):
	SaidPosting = False
	SiteDomain, SiteLang = Flags['SiteDomain'], Flags['SiteLang']
	InstanceURL, Token = Flags['MastodonURL'], Flags['MastodonToken']
	TypeFilter, HoursLimit, CategoryFilter = Flags['ActivityPubTypeFilter'], Flags['ActivityPubHoursLimit'], Flags['FeedCategoryFilter']
	Session = MastodonGetSession(InstanceURL, Token)
	Posts = MastodonGetAllLinkPosts(Session, SiteDomain)
	Pages.sort()
	for File, Content, Titles, Meta, ContentHTML, SlimHTML, Description, Image in Pages:
		if (not TypeFilter or (TypeFilter and (Meta['Type'] == TypeFilter or TypeFilter == '*'))) and (not CategoryFilter or (CategoryFilter and (CategoryFilter in Meta['Categories'] or CategoryFilter == '*'))):
			URL = f"{SiteDomain}/{StripExt(File)}.html"
			DoPost = True
			for p in Posts:
				if p['Link'] in [URL]+Meta['URLs']:
					DoPost = False
					break

			if DoPost and Meta['Feed'] == 'True' and (not HoursLimit or (Meta['CreatedOn'] and time.time() - time.mktime(date_parse(Meta['CreatedOn']).timetuple()) < 60*60*HoursLimit)):
				Desc = ''
				Paragraphs = MkSoup(ContentHTML).p.get_text().split('\n')
				Read = '...' + Locale['ReadFullPost'] + ':\n'
				for p in Paragraphs:
					if p and len(Read+Desc+p)+25 < 500:
						Desc += p + '\n\n'
					else:
						if Desc:
							break
						else:
							Desc = p[:500-25-5-len(Read)] + '...'

				if not SaidPosting:
					print("[I] Posting to Mastodon")
					SaidPosting = True

				time.sleep(5) # Prevent flooding
				Post = MastodonGetLinkFromPost(
					Post=MastodonDoPost(
						Session,
						Text=Desc+Read+URL,
						Lang=SiteLang),
					Domain=SiteDomain)
				if Post:
					Posts += [Post]

	return Posts