mirror of https://gitlab.com/octtspacc/staticoso
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
""" ================================= |
|
|
| 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
|