1003 lines
34 KiB
Python
1003 lines
34 KiB
Python
import functools
|
||
import random
|
||
import re
|
||
import signal
|
||
from functools import partial
|
||
from os import path, listdir
|
||
from typing_extensions import deprecated
|
||
from urllib.parse import parse_qs, urlparse, unquote, ParseResult, urlencode, urlunparse
|
||
import time
|
||
from files.helpers.marseyfx.parser import parse_emoji
|
||
from files.helpers.marseyfx.tokenizer import Tokenizer
|
||
|
||
from sqlalchemy.sql import func
|
||
|
||
import bleach
|
||
from bleach.css_sanitizer import CSSSanitizer
|
||
from bleach.linkifier import LinkifyFilter
|
||
from bs4 import BeautifulSoup
|
||
from mistletoe import markdown
|
||
|
||
from files.classes.domains import BannedDomain
|
||
from files.classes.mod_logs import ModAction
|
||
from files.classes.notifications import Notification
|
||
from files.classes.group import Group
|
||
from files.classes.follows import Follow
|
||
|
||
from files.helpers.config.const import *
|
||
from files.helpers.const_stateful import *
|
||
from files.helpers.regex import *
|
||
from files.helpers.get import *
|
||
|
||
TLDS = ( # Original gTLDs and ccTLDs
|
||
'ac','ad','ae','aero','af','ag','ai','al','am','an','ao','aq','ar','arpa','as','asia','at',
|
||
'au','aw','ax','az','ba','bb','bd','be','bf','bg','bh','bi','biz','bj','bm','bn','bo','br',
|
||
'bs','bt','bv','bw','by','bz','ca','cafe','cat','cc','cd','cf','cg','ch','ci','ck','cl',
|
||
'cm','cn','co','com','coop','cr','cu','cv','cx','cy','cz','de','dj','dk','dm','do','dz','ec',
|
||
'edu','ee','eg','er','es','et','eu','fi','fj','fk','fm','fo','fr','ga','gb','gd','ge','gf',
|
||
'gg','gh','gi','gl','gm','gn','gov','gp','gq','gr','gs','gt','gu','gw','gy','hk','hm','hn',
|
||
'hr','ht','hu','id','ie','il','im','in','info','int','io','iq','ir','is','it','je','jm','jo',
|
||
'jobs','jp','ke','kg','kh','ki','km','kn','kp','kr','kw','ky','kz','la','lb','lc','li','lk',
|
||
'lr','ls','lt','lu','lv','ly','ma','mc','md','me','mg','mh','mil','mk','ml','mm','mn','mo',
|
||
'mobi','mp','mq','mr','ms','mt','mu','museum','mv','mw','mx','my','mz','na','name',
|
||
'nc','ne','net','nf','ng','ni','nl','no','np','nr','nu','nz','om','org','pa','pe','pf','pg',
|
||
'ph','pk','pl','pm','pn','post','pr','pro','ps','pt','pw','py','qa','re','ro','rs','ru','rw',
|
||
'sa','sb','sc','sd','se','sg','sh','si','sj','sk','sl','sm','sn','so','social','sr','ss','st',
|
||
'su','sv','sx','sy','sz','tc','td','tel','tf','tg','th','tj','tk','tl','tm','tn','to','tp',
|
||
'tr','travel','tt','tv','tw','tz','ua','ug','uk','us','uy','uz','va','vc','ve','vg','vi','vn',
|
||
'vu','wf','ws','xn','xxx','ye','yt','yu','za','zm','zw',
|
||
# New gTLDs
|
||
'app','cleaning','club','dev','farm','florist','fun','gay','lgbt','life','lol',
|
||
'moe','mom','monster','new','news','online','pics','press','pub','site','blog',
|
||
'vip','win','world','wtf','xyz','video','host','art','media','wiki','tech',
|
||
'cooking','network','party','goog','markets','today','beauty','camp','top',
|
||
'red','city','quest','works','soy',
|
||
)
|
||
|
||
allowed_tags = ('a','audio','b','big','blockquote','br','center','code','del','details','em','g','h1','h2','h3','h4','h5','h6','hr','i','img','li','lite-youtube','marquee','ol','p','pre','rp','rt','ruby','small','span','spoiler','strike','strong','sub','summary','sup','table','tbody','td','th','thead','tr','u','ul','video')
|
||
|
||
allowed_global_styles = ['background-color', 'color', 'filter', 'font-weight', 'text-align']
|
||
|
||
additional_img_styles = ['transform']
|
||
|
||
allowed_styles = allowed_global_styles + additional_img_styles
|
||
|
||
def allowed_attributes(tag, name, value):
|
||
|
||
if name == 'style': return True
|
||
|
||
if tag == 'marquee':
|
||
if name in {'direction', 'behavior', 'scrollamount'}: return True
|
||
if name in {'height', 'width'}:
|
||
try: value = int(value.replace('px', ''))
|
||
except: return False
|
||
if 0 < value <= 250: return True
|
||
|
||
if tag == 'a':
|
||
if name == 'href' and '\\' not in value and 'xn--' not in value:
|
||
return True
|
||
if name == 'rel' and value == 'nofollow noopener': return True
|
||
if name == 'target' and value == '_blank': return True
|
||
|
||
if tag == 'img':
|
||
if name in {'src','data-src'}: return is_safe_url(value)
|
||
if name == 'loading' and value == 'lazy': return True
|
||
if name == 'data-bs-toggle' and value == 'tooltip': return True
|
||
if name in {'g','b','glow','party'} and not value: return True
|
||
if name in {'alt','title'}: return True
|
||
if name == 'class' and value == 'img': return True
|
||
if name == 'data-user-submitted' and not value: return True
|
||
if name == 'data-kind' and value in EMOJI_KINDS: return True
|
||
|
||
if tag == 'lite-youtube':
|
||
if name == 'params' and value.startswith('autoplay=1&modestbranding=1'): return True
|
||
if name == 'videoid': return True
|
||
|
||
if tag == 'video':
|
||
if name == 'controls' and value == '': return True
|
||
if name == 'preload' and value == 'none': return True
|
||
if name == 'src': return is_safe_url(value)
|
||
|
||
if tag == 'audio':
|
||
if name == 'src': return is_safe_url(value)
|
||
if name == 'controls' and value == '': return True
|
||
if name == 'preload' and value == 'none': return True
|
||
|
||
if tag == 'p':
|
||
if name == 'class' and value in {'mb-0','resizable','yt','text-center'}: return True
|
||
|
||
if tag == 'span':
|
||
if name == 'data-bs-toggle' and value == 'tooltip': return True
|
||
if name == 'title': return True
|
||
if name == 'alt': return True
|
||
if name == 'cide' and not value: return True
|
||
if name == 'bounce' and not value: return True
|
||
|
||
if tag == 'table':
|
||
if name == 'class' and value == 'table': return True
|
||
|
||
return False
|
||
|
||
def build_url_re(tlds, protocols):
|
||
"""Builds the url regex used by linkifier
|
||
|
||
If you want a different set of tlds or allowed protocols, pass those in
|
||
and stomp on the existing ``url_re``::
|
||
|
||
from bleach import linkifier
|
||
|
||
my_url_re = linkifier.build_url_re(my_tlds_list, my_protocols)
|
||
|
||
linker = LinkifyFilter(url_re=my_url_re)
|
||
|
||
"""
|
||
return re.compile(
|
||
r"""\(*# Match any opening parentheses.
|
||
\b(?<![@.#:])(?:(?:{0}):/{{0,3}}(?:(?:\w+:)?\w+@)?)?# http://
|
||
([\w-]+\.)+(?:{1})(?:\:[0-9]+)?(?!\.\w)\b# xx.yy.tld(:##)?
|
||
(?:[/?][^#\s\{{\}}\|\\\^\[\]`<>"]*)?
|
||
# /path/zz (excluding "unsafe" chars from RFC 1738,
|
||
# except for ~, which happens in practice)
|
||
(?:\#[^#\s\|\\\^\[\]`<>"]*)?
|
||
# #hash (excluding "unsafe" chars from RFC 1738,
|
||
# except for ~, which happens in practice)
|
||
""".format(
|
||
"|".join(sorted(protocols)), "|".join(sorted(tlds))
|
||
),
|
||
re.VERBOSE | re.UNICODE,
|
||
)
|
||
|
||
url_re = build_url_re(tlds=TLDS, protocols=['http', 'https'])
|
||
|
||
def create_comment_duplicated(text_html):
|
||
new_comment = Comment(author_id=AUTOJANNY_ID,
|
||
parent_post=None,
|
||
body_html=text_html,
|
||
distinguish_level=6,
|
||
is_bot=True)
|
||
g.db.add(new_comment)
|
||
g.db.flush()
|
||
|
||
new_comment.top_comment_id = new_comment.id
|
||
|
||
return new_comment.id
|
||
|
||
def send_repeatable_notification_duplicated(uid, text):
|
||
|
||
if uid in BOT_IDs: return
|
||
|
||
text_html = sanitize(text)
|
||
|
||
existing_comments = g.db.query(Comment.id).filter_by(author_id=AUTOJANNY_ID, parent_post=None, body_html=text_html, is_bot=True).order_by(Comment.id).all()
|
||
|
||
for c in existing_comments:
|
||
existing_notif = g.db.query(Notification.user_id).filter_by(user_id=uid, comment_id=c.id).one_or_none()
|
||
if not existing_notif:
|
||
notif = Notification(comment_id=c.id, user_id=uid)
|
||
g.db.add(notif)
|
||
return
|
||
|
||
cid = create_comment_duplicated(text_html)
|
||
notif = Notification(comment_id=cid, user_id=uid)
|
||
g.db.add(notif)
|
||
|
||
|
||
def execute_blackjack(v, target, body, kind):
|
||
if not blackjack or not body: return False
|
||
|
||
execute = False
|
||
for x in blackjack.split(','):
|
||
if all(i in body.lower() for i in x.split()):
|
||
execute = True
|
||
|
||
if not execute: return False
|
||
|
||
v.shadowbanned = AUTOJANNY_ID
|
||
|
||
ma = ModAction(
|
||
kind="shadowban",
|
||
user_id=AUTOJANNY_ID,
|
||
target_user_id=v.id,
|
||
_note='reason: "Blackjack"'
|
||
)
|
||
g.db.add(ma)
|
||
|
||
v.ban_reason = "Blackjack"
|
||
g.db.add(v)
|
||
|
||
notified_ids = [x[0] for x in g.db.query(User.id).filter(User.admin_level >= PERMS['BLACKJACK_NOTIFICATIONS'])]
|
||
extra_info = kind
|
||
|
||
if target:
|
||
if kind == 'post':
|
||
extra_info = target.permalink
|
||
elif kind == 'report':
|
||
extra_info = f"reports on {target.permalink}"
|
||
elif kind in {'comment', 'message'}:
|
||
for id in notified_ids:
|
||
n = Notification(comment_id=target.id, user_id=id)
|
||
g.db.add(n)
|
||
|
||
extra_info = None
|
||
|
||
if extra_info:
|
||
for id in notified_ids:
|
||
send_repeatable_notification_duplicated(id, f"Blackjack by @{v.username}: {extra_info}")
|
||
return True
|
||
|
||
def find_all_emote_endings(word):
|
||
endings = []
|
||
|
||
if path.isfile(f'files/assets/images/emojis/{word}.webp'):
|
||
return endings, word
|
||
|
||
is_non_ending_found = False
|
||
while not is_non_ending_found:
|
||
if word.endswith('pat'):
|
||
if 'pat' in endings:
|
||
is_non_ending_found = True
|
||
continue
|
||
endings.append('pat')
|
||
word = word[:-3]
|
||
continue
|
||
|
||
if word.endswith('talking'):
|
||
if 'talking' in endings:
|
||
is_non_ending_found = True
|
||
continue
|
||
endings.append('talking')
|
||
word = word[:-7]
|
||
continue
|
||
|
||
if word.endswith('genocide'):
|
||
if 'genocide' in endings:
|
||
is_non_ending_found = True
|
||
continue
|
||
endings.append('genocide')
|
||
word = word[:-8]
|
||
continue
|
||
|
||
if word.endswith('love'):
|
||
if 'love' in endings:
|
||
is_non_ending_found = True
|
||
continue
|
||
endings.append('love')
|
||
word = word[:-4]
|
||
continue
|
||
|
||
is_non_ending_found = True
|
||
|
||
return endings, word
|
||
|
||
class RenderEmojisResult:
|
||
emojis_used: set[str]
|
||
heavy_count = 0
|
||
tags: list[str]
|
||
|
||
def __init__(self):
|
||
self.emojis_used = set()
|
||
self.tags = []
|
||
|
||
def update(self, other):
|
||
self.emojis_used |= other.emojis_used
|
||
self.heavy_count += other.heavy_count
|
||
self.tags.extend(other.tags)
|
||
|
||
def render_emojis(markup: str, **kwargs):
|
||
result = RenderEmojisResult()
|
||
last_match_end = 0
|
||
|
||
golden = kwargs.get('golden', True)
|
||
permit_big = kwargs.get('permit_big', True)
|
||
|
||
for emoji_match in marseyfx_emoji_regex.finditer(markup):
|
||
previous_text = markup[last_match_end:emoji_match.start()]
|
||
if previous_text != '':
|
||
result.tags.append(previous_text)
|
||
last_match_end = emoji_match.end()
|
||
|
||
emoji_str = emoji_match.group()[1:-1] # Cut off colons
|
||
|
||
tokenizer = Tokenizer(emoji_str)
|
||
success, emoji, _ = parse_emoji(tokenizer)
|
||
if success:
|
||
result.emojis_used.add(emoji.name)
|
||
|
||
if not permit_big:
|
||
emoji.is_big = False
|
||
|
||
emoji_html = emoji.create_el(tokenizer)
|
||
result.tags.append(emoji_html)
|
||
|
||
if len(tokenizer.errors) > 0:
|
||
soup = BeautifulSoup()
|
||
err_tag = soup.new_tag('pre', attrs={'class': 'marseyfx-error'})
|
||
nl = "\n "
|
||
err_tag.string = 'MarseyFX error:' + nl + nl.join(map(str,tokenizer.errors))
|
||
result.tags.append(err_tag)
|
||
#result.tags.append(f':{emoji_str}:')
|
||
|
||
result.tags.append(markup[last_match_end:])
|
||
return result
|
||
|
||
@deprecated("Use the new one")
|
||
def old_render_emoji(html, regexp, golden, emojis_used, b=False, is_title=False):
|
||
emojis = list(regexp.finditer(html))
|
||
captured = set()
|
||
|
||
for i in emojis:
|
||
if i.group(0) in captured: continue
|
||
captured.add(i.group(0))
|
||
|
||
emoji = i.group(1).lower()
|
||
attrs = ''
|
||
if b: attrs += ' b'
|
||
if is_title: emoji = emoji.replace('#','')
|
||
if golden and len(emojis) <= 20 and ('marsey' in emoji or emoji in marseys_const2):
|
||
if random.random() < 0.005:
|
||
attrs += ' ' + random.choice(('g', 'glow', 'party'))
|
||
|
||
old = emoji
|
||
emoji = emoji.replace('!','').replace('#','')
|
||
|
||
if emoji.endswith('random'):
|
||
kind = emoji.split('random')[0].title()
|
||
if kind == 'Donkeykong': kind = 'Donkey Kong'
|
||
elif kind == 'Marseyflag': kind = 'Marsey Flags'
|
||
elif kind == 'Marseyalphabet': kind = 'Marsey Alphabet'
|
||
|
||
if kind in EMOJI_KINDS:
|
||
emoji = g.db.query(Emoji.name).filter_by(kind=kind).order_by(func.random()).first()[0]
|
||
|
||
|
||
emoji_partial_pat = '<img alt=":{0}:" loading="lazy" src="{1}"{2}>'
|
||
emoji_partial = '<img alt=":{0}:" data-bs-toggle="tooltip" loading="lazy" src="{1}" title=":{0}:"{2}>'
|
||
emoji_html = None
|
||
|
||
ending_modifiers, emoji = find_all_emote_endings(emoji)
|
||
|
||
is_talking = 'talking' in ending_modifiers
|
||
is_patted = 'pat' in ending_modifiers
|
||
is_talking_first = ending_modifiers.index('pat') > ending_modifiers.index('talking') if is_talking and is_patted else False
|
||
is_loved = 'love' in ending_modifiers
|
||
is_genocided = 'genocide' in ending_modifiers
|
||
is_user = emoji.startswith('@')
|
||
|
||
end_modifier_length = 3 if is_patted else 0
|
||
end_modifier_length = end_modifier_length + 7 if is_talking else end_modifier_length
|
||
|
||
hand_html = f'<img loading="lazy" src="{SITE_FULL_IMAGES}/i/hand.webp">' if is_patted and emoji != 'marseyunpettable' else ''
|
||
talking_html = f'<img loading="lazy" src="{SITE_FULL_IMAGES}/i/talking.webp">' if is_talking else ''
|
||
loved_html = f'<img loading="lazy" src="{SITE_FULL_IMAGES}/i/love-foreground.webp" alt=":{old}:" {attrs}><img loading="lazy" alt=":{old}:" src="{SITE_FULL_IMAGES}/i/love-background.webp" {attrs}>'
|
||
genocide_attr = ' cide' if is_genocided else ''
|
||
|
||
modifier_html = ''
|
||
if (is_talking and is_patted):
|
||
modifier_html = f'{talking_html}{hand_html}' if is_talking_first else f'{hand_html}{talking_html}'
|
||
elif (is_patted):
|
||
modifier_html = hand_html
|
||
elif (is_talking):
|
||
modifier_html = talking_html
|
||
|
||
if(is_loved):
|
||
modifier_html = f'{modifier_html}{loved_html}'
|
||
|
||
kind = g.db.query(Emoji.kind).filter(Emoji.name==emoji).one_or_none()
|
||
if kind:
|
||
attrs += ' data-kind="' + kind[0].replace('"', '') + '"'
|
||
|
||
if (is_patted and emoji != 'marseyunpettable') or is_talking or is_genocided or is_loved:
|
||
if path.isfile(f"files/assets/images/emojis/{emoji}.webp"):
|
||
emoji_html = f'<span alt=":{old}:" data-bs-toggle="tooltip" title=":{old}:"{genocide_attr}>{modifier_html}{emoji_partial_pat.format(old, f"{SITE_FULL_IMAGES}/e/{emoji}.webp", attrs)}</span>'
|
||
elif is_user:
|
||
if u := get_user(emoji[1:], graceful=True):
|
||
emoji_html = f'<span alt=":{old}:" data-bs-toggle="tooltip" title=":{old}:"{genocide_attr}>{modifier_html}{emoji_partial_pat.format(old, f"/pp/{u.id}", attrs)}</span>'
|
||
elif path.isfile(f'files/assets/images/emojis/{emoji}.webp'):
|
||
emoji_html = emoji_partial.format(old, f'{SITE_FULL_IMAGES}/e/{emoji}.webp', attrs)
|
||
|
||
|
||
if emoji_html:
|
||
emojis_used.add(emoji)
|
||
html = re.sub(f'(?<!"){i.group(0)}(?![^<]*<\/(code|pre)>)', emoji_html, html)
|
||
return html
|
||
|
||
|
||
def with_sigalrm_timeout(timeout):
|
||
'Use SIGALRM to raise an exception if the function executes for longer than timeout seconds'
|
||
|
||
# while trying to test this using time.sleep I discovered that gunicorn does in fact do some
|
||
# async so if we timeout on that (or on a db op) then the process is crashed without returning
|
||
# a proper 500 error. Oh well.
|
||
def sig_handler(signum, frame):
|
||
print("Timeout!", flush=True)
|
||
raise Exception("Timeout")
|
||
|
||
def inner(func):
|
||
@functools.wraps(func)
|
||
def wrapped(*args, **kwargs):
|
||
signal.signal(signal.SIGALRM, sig_handler)
|
||
signal.alarm(timeout)
|
||
try:
|
||
return func(*args, **kwargs)
|
||
finally:
|
||
signal.alarm(0)
|
||
return wrapped
|
||
return inner
|
||
|
||
def remove_cuniform(sanitized):
|
||
if not sanitized: return ""
|
||
sanitized = sanitized.replace('\u200e','').replace('\u200b','').replace('\u202e','').replace("\ufeff", "").replace("\u033f","").replace("\u0589", ":")
|
||
sanitized = sanitized.replace("𒐪","").replace("𒐫","").replace("﷽","")
|
||
sanitized = sanitized.replace("\r\n", "\n")
|
||
sanitized = sanitized.replace("’", "'")
|
||
return sanitized.strip()
|
||
|
||
def get_youtube_id_and_t(url):
|
||
params = parse_qs(urlparse(url).query, keep_blank_values=True)
|
||
|
||
id = params.get('v')
|
||
|
||
if not id: return (None, None)
|
||
|
||
id = id[0]
|
||
|
||
t = None
|
||
split = id.split('?t=')
|
||
if len(split) == 2:
|
||
id = split[0]
|
||
t = split[1]
|
||
|
||
id = id.split('?')[0]
|
||
|
||
return (id, t)
|
||
|
||
def handle_youtube_links(url):
|
||
url = url.replace('&','&')
|
||
params = parse_qs(urlparse(url).query, keep_blank_values=True)
|
||
html = None
|
||
id, t = get_youtube_id_and_t(url)
|
||
if not id: return None
|
||
if yt_id_regex.fullmatch(id):
|
||
if not t:
|
||
t = params.get('t', params.get('start', [0]))[0]
|
||
if isinstance(t, str):
|
||
t = t.replace('s','').replace('S','')
|
||
split = t.split('m')
|
||
if len(split) == 2:
|
||
minutes = int(split[0])
|
||
seconds = int(split[1])
|
||
t = minutes*60 + seconds
|
||
html = f'<lite-youtube videoid="{id}" params="autoplay=1&modestbranding=1'
|
||
if t:
|
||
html += f'&start={int(t)}'
|
||
html += '"></lite-youtube>'
|
||
return html
|
||
|
||
@with_sigalrm_timeout(10)
|
||
def sanitize(sanitized, golden=True, limit_pings=0, showmore=False, count_emojis=False, snappy=False, chat=False, blackjack=None, post_mention_notif=False, commenters_ping_post_id=None):
|
||
def error(error):
|
||
if chat:
|
||
return error, 403
|
||
else:
|
||
abort(403, error)
|
||
|
||
sanitized = html_comment_regex.sub('', sanitized)
|
||
sanitized = remove_cuniform(sanitized)
|
||
|
||
if not sanitized: return ''
|
||
|
||
if blackjack and execute_blackjack(g.v, None, sanitized, blackjack):
|
||
sanitized = 'g'
|
||
|
||
if '```' not in sanitized and '<pre>' not in sanitized:
|
||
sanitized = linefeeds_regex.sub(r'\1\n\n\2', sanitized)
|
||
|
||
sanitized = greentext_regex.sub(r'\1<g>\>\2</g>', sanitized)
|
||
sanitized = image_sub_regex.sub(r'', sanitized)
|
||
sanitized = image_check_regex.sub(r'\1', sanitized)
|
||
sanitized = link_fix_regex.sub(r'\1https://\2', sanitized)
|
||
|
||
if FEATURES['MARKUP_COMMANDS']:
|
||
sanitized = command_regex.sub(command_regex_matcher, sanitized)
|
||
|
||
sanitized = numbered_list_regex.sub(r'\1\. ', sanitized)
|
||
|
||
sanitized = strikethrough_regex.sub(r'\1<del>\2</del>', sanitized)
|
||
|
||
sanitized = sanitized.replace('_', '▔')
|
||
sanitized = markdown(sanitized)
|
||
sanitized = sanitized.replace('▔', '_').replace('%E2%96%94', '_')
|
||
|
||
sanitized = sanitized.replace('<a href="/%21', '<a href="/!')
|
||
|
||
sanitized = reddit_mention_regex.sub(r'<a href="https://old.reddit.com/\1" rel="nofollow noopener" target="_blank">/\1</a>', sanitized)
|
||
sanitized = sub_regex.sub(r'<a href="/\1">/\1</a>', sanitized)
|
||
|
||
v = getattr(g, 'v', None)
|
||
|
||
if FEATURES['PING_GROUPS']:
|
||
def group_replacer(m):
|
||
name = m.group(1).lower()
|
||
|
||
if name == 'everyone':
|
||
return f'<a href="/users">!{name}</a>'
|
||
elif name == 'jannies':
|
||
return f'<a href="/admins">!{name}</a>'
|
||
elif name == 'commenters' and commenters_ping_post_id:
|
||
return f'<a href="/!commenters/{commenters_ping_post_id}/{int(time.time())}">!{name}</a>'
|
||
elif name == 'followers':
|
||
return f'<a href="/id/{g.v.id}/followers">!{name}</a>'
|
||
elif g.db.get(Group, name):
|
||
return f'<a href="/!{name}">!{name}</a>'
|
||
else:
|
||
return m.group(0)
|
||
|
||
sanitized = group_mention_regex.sub(group_replacer, sanitized)
|
||
|
||
|
||
soup = BeautifulSoup(sanitized, 'lxml')
|
||
|
||
for tag in soup.find_all("img"):
|
||
if tag.get("src") and not tag["src"].startswith('/pp/') and not (snappy and tag["src"].startswith(f'{SITE_FULL_IMAGES}/e/')):
|
||
if not is_safe_url(tag["src"]):
|
||
a = soup.new_tag("a", href=tag["src"], rel="nofollow noopener", target="_blank")
|
||
a.string = tag["src"]
|
||
tag.replace_with(a)
|
||
continue
|
||
|
||
del tag["g"]
|
||
del tag["glow"]
|
||
del tag["party"]
|
||
|
||
tag["loading"] = "lazy"
|
||
tag["data-src"] = tag["src"]
|
||
tag["src"] = f"{SITE_FULL_IMAGES}/i/l.webp"
|
||
tag['alt'] = tag["data-src"]
|
||
tag['class'] = "img"
|
||
|
||
if tag.parent.name != 'a':
|
||
a = soup.new_tag("a", href=tag["data-src"])
|
||
if not is_site_url(a["href"]):
|
||
a["rel"] = "nofollow noopener"
|
||
a["target"] = "_blank"
|
||
tag = tag.replace_with(a)
|
||
a.append(tag)
|
||
|
||
tag["data-src"] = tag["data-src"]
|
||
tag["data-user-submitted"] = ""
|
||
|
||
sanitized = str(soup).replace('<html><body>','').replace('</body></html>','').replace('/>','>')
|
||
|
||
sanitized = spoiler_regex.sub(r'<spoiler>\1</spoiler>', sanitized)
|
||
|
||
sanitized = sanitized.replace('&','&')
|
||
|
||
sanitized = video_sub_regex.sub(r'<p class="resizable"><video controls preload="none" src="\1"></video></p>', sanitized)
|
||
sanitized = audio_sub_regex.sub(r'<audio controls preload="none" src="\1"></audio>', sanitized)
|
||
|
||
sanitized = sanitized.replace('<p></p>', '')
|
||
|
||
allowed_css_properties = allowed_styles.copy()
|
||
if g.v and g.v.chud:
|
||
allowed_css_properties.remove('filter')
|
||
|
||
css_sanitizer = CSSSanitizer(allowed_css_properties=allowed_css_properties)
|
||
sanitized = bleach.Cleaner(tags=allowed_tags,
|
||
attributes=allowed_attributes,
|
||
protocols=['http', 'https'],
|
||
css_sanitizer=css_sanitizer,
|
||
filters=[partial(LinkifyFilter, skip_tags=["pre"],
|
||
parse_email=False, url_re=url_re)]
|
||
).clean(sanitized)
|
||
|
||
|
||
#doing this here cuz of the linkifyfilter right above it (therefore unifying all link processing logic) <-- i have no clue what this means lol
|
||
soup = BeautifulSoup(sanitized, 'lxml')
|
||
|
||
# -- EMOJI RENDERING --
|
||
emojis_used = set()
|
||
|
||
for text_el in soup.find_all(text=True):
|
||
if not text_el.parent or text_el.parent.name in {'code', 'pre'}:
|
||
continue
|
||
res = render_emojis(text_el.text)
|
||
text_el.replace_with(*res.tags)
|
||
emojis_used.update(res.emojis_used)
|
||
|
||
if count_emojis:
|
||
for emoji in g.db.query(Emoji).filter(Emoji.submitter_id==None, Emoji.name.in_(emojis_used)):
|
||
emoji.count += 1
|
||
g.db.add(emoji)
|
||
|
||
# -- @ MENTIONS --
|
||
ping_count = 0
|
||
names = set(m.group(1) for m in mention_regex.finditer(sanitized))
|
||
|
||
users_list = get_users(names, graceful=True)
|
||
users_dict = {}
|
||
for u in users_list:
|
||
users_dict[u.username.lower()] = u
|
||
if u.original_username:
|
||
users_dict[u.original_username.lower()] = u
|
||
if u.prelock_username:
|
||
users_dict[u.prelock_username.lower()] = u
|
||
|
||
for text_el in soup.find_all(text=True):
|
||
start = 0
|
||
tags = []
|
||
for match in mention_regex.finditer(text_el):
|
||
tags.append(text_el[start:match.start()])
|
||
start = match.end()
|
||
name = match.group(1).lower()
|
||
u = users_dict.get(name)
|
||
if not u or (v and u.id in v.all_twoway_blocks) or (v and u.has_muted(v)):
|
||
tags.append(match.group(0))
|
||
continue
|
||
else:
|
||
link = soup.new_tag("a", href=f"/id/{u.id}")
|
||
img = soup.new_tag("img", loading="lazy", src=f"/pp/{u.id}")
|
||
link.extend([img, f'@{u.username}'])
|
||
tags.append(link)
|
||
ping_count += 1
|
||
if limit_pings and ping_count >= limit_pings:
|
||
if limit_pings and len(names) > limit_pings and v.admin_level < PERMS['POST_COMMENT_INFINITE_PINGS']:
|
||
error("Max ping limit is 5 for comments and 50 for posts!")
|
||
|
||
def replacer(m):
|
||
u = users_dict.get(m.group(1).lower())
|
||
if not u or (v and u.id in v.all_twoway_blocks) or (v and u.has_muted(v)):
|
||
return m.group(0)
|
||
#return f'<a href="/id/{u.id}"><img loading="lazy" src="/pp/{u.id}">@{u.username}</a>'
|
||
|
||
if not text_el.parent or text_el.parent.name in {'code', 'pre', 'a'}:
|
||
continue
|
||
text_el.replace_with(*tags, text_el[start:])
|
||
|
||
sanitized = mention_regex.sub(replacer, sanitized)
|
||
|
||
# -- LINKS --
|
||
|
||
links = soup.find_all("a")
|
||
|
||
if g.v and g.v.admin_level >= PERMS["IGNORE_DOMAIN_BAN"]:
|
||
banned_domains = []
|
||
else:
|
||
banned_domains = [x.domain for x in g.db.query(BannedDomain.domain)]
|
||
|
||
for link in links:
|
||
#remove empty links
|
||
if not link.contents or not str(link.contents[0]).strip():
|
||
link.extract()
|
||
continue
|
||
|
||
href = link.get("href")
|
||
if not href: continue
|
||
|
||
link["href"] = normalize_url(href)
|
||
if link.string == href:
|
||
link.string = link["href"]
|
||
href = link["href"]
|
||
|
||
def unlinkfy():
|
||
link.string = href
|
||
del link["href"]
|
||
|
||
#\ in href right after / makes most browsers ditch site hostname and allows for a host injection bypassing the check, see <a href="/\google.com">cool</a>
|
||
if "\\" in href:
|
||
unlinkfy()
|
||
continue
|
||
|
||
#don't allow something like this https://rdrama.net/post/78376/reminder-of-the-fact-that-our/2150032#context
|
||
domain = tldextract.extract(href).registered_domain
|
||
if domain and not allowed_domain_regex.fullmatch(domain):
|
||
unlinkfy()
|
||
continue
|
||
|
||
#check for banned domain
|
||
combined = (domain + urlparse(href).path).lower()
|
||
if any((combined.startswith(x) for x in banned_domains)):
|
||
unlinkfy()
|
||
continue
|
||
|
||
#don't allow something like this [@Aevann2](https://iplogger.org/1fRKk7)
|
||
if str(link.string).startswith('@') and not href.startswith('/'):
|
||
unlinkfy()
|
||
continue
|
||
|
||
#don't allow something like this [!jannies](https://iplogger.org/1fRKk7)
|
||
if str(link.string).startswith('!') and not href.startswith('/'):
|
||
unlinkfy()
|
||
continue
|
||
|
||
#don't allow something like this [https://rԁrama.net/leaderboard](https://iplogger.org/1fRKk7)
|
||
if not snappy and not post_mention_notif:
|
||
string_domain = tldextract.extract(str(link.string)).registered_domain
|
||
if string_domain and string_domain != tldextract.extract(href).registered_domain:
|
||
link.string = href
|
||
|
||
#insert target="_blank" and ref="nofollower noopener" for external link
|
||
if not href.startswith('/') and not href.startswith(f'{SITE_FULL}/'):
|
||
link["target"] = "_blank"
|
||
link["rel"] = "nofollow noopener"
|
||
|
||
sanitized = str(soup).replace('<html><body>','').replace('</body></html>','').replace('/>','>')
|
||
|
||
captured = []
|
||
for i in youtube_regex.finditer(sanitized):
|
||
if i.group(0) in captured: continue
|
||
captured.append(i.group(0))
|
||
|
||
html = handle_youtube_links(i.group(1))
|
||
if html:
|
||
if not chat:
|
||
html = f'<p class="resizable yt">{html}</p>'
|
||
sanitized = sanitized.replace(i.group(0), html)
|
||
|
||
if showmore:
|
||
# Insert a show more button if the text is too long or has too many paragraphs
|
||
CHARLIMIT = 3000
|
||
pos = 0
|
||
for _ in range(20):
|
||
pos = sanitized.find('</p>', pos + 4)
|
||
if pos < 0:
|
||
break
|
||
if (pos < 0 and len(sanitized) > CHARLIMIT) or pos > CHARLIMIT:
|
||
pos = CHARLIMIT - 500
|
||
if pos >= 0:
|
||
sanitized = (sanitized[:pos] + showmore_regex.sub(r'\1<p><button class="showmore">SHOW MORE</button></p><d class="d-none">\2</d>', sanitized[pos:], count=1))
|
||
|
||
if "style" in sanitized and "filter" in sanitized:
|
||
if sanitized.count("blur(") + sanitized.count("drop-shadow(") > 5:
|
||
error("Too many filters!")
|
||
|
||
return sanitized.strip()
|
||
|
||
def allowed_attributes_emojis(tag, name, value):
|
||
if tag == 'img':
|
||
if name == 'src':
|
||
if '\\' in value: return False
|
||
if value.startswith('/') : return True
|
||
if value.startswith(f'{SITE_FULL_IMAGES}/') : return True
|
||
if name == 'loading' and value == 'lazy': return True
|
||
if name == 'data-bs-toggle' and value == 'tooltip': return True
|
||
if name in {'g','glow','party'} and not value: return True
|
||
if name in {'alt','title'}: return True
|
||
|
||
if tag == 'span':
|
||
if name == 'data-bs-toggle' and value == 'tooltip': return True
|
||
if name == 'title': return True
|
||
if name == 'alt': return True
|
||
if name == 'cide' and not value: return True
|
||
return False
|
||
|
||
|
||
@with_sigalrm_timeout(2)
|
||
def filter_emojis_only(title, golden=True, count_emojis=False):
|
||
|
||
title = title.replace("\n", "").replace("\r", "").replace("\t", "").replace('<','<').replace('>','>')
|
||
|
||
title = remove_cuniform(title)
|
||
|
||
res = render_emojis(title) #old_render_emoji(title, emoji_regex2, golden, emojis_used, is_title=True)
|
||
|
||
title = ''.join(map(str, res.tags))
|
||
|
||
if count_emojis:
|
||
for emoji in g.db.query(Emoji).filter(Emoji.submitter_id==None, Emoji.name.in_(emojis_used)):
|
||
emoji.count += 1
|
||
g.db.add(emoji)
|
||
|
||
title = strikethrough_regex.sub(r'\1<del>\2</del>', title)
|
||
|
||
title = bleach.clean(title, tags=['img','del','span'], attributes=allowed_attributes_emojis, protocols=['http','https']).replace('\n','')
|
||
|
||
if len(title) > POST_TITLE_HTML_LENGTH_LIMIT:
|
||
abort(400, "Rendered title is too big!")
|
||
|
||
title = title.strip()
|
||
|
||
return title
|
||
|
||
def is_whitelisted(domain, k):
|
||
if domain.endswith('pullpush.io'):
|
||
return True
|
||
if 'sort' in k.lower() or 'query' in k.lower():
|
||
return True
|
||
if k in {'_x_tr_hl','_x_tr_pto','_x_tr_sl','_x_tr_tl','abstract_id','after','article','bill_id','c','clip','comments','context','count','f','fbid','format','forum_id','i','ID','id','lb','list','oldid','p','page','post_id','postid','q','run','scrollToComments','search','sl','sp','story_fbid','tab','term','text','thread_id','threadid','ticket_form_id','time_continue','title','title_no','tl','token','topic','type','tz1','tz2','u','udca','url','v','vid','viewkey'}:
|
||
return True
|
||
if k == 't' and domain != 'twitter.com':
|
||
return True
|
||
if k == 'oid' and domain != 'quora.com':
|
||
return True
|
||
return False
|
||
|
||
|
||
def normalize_url(url):
|
||
url = unquote(url)
|
||
|
||
url = reddit_domain_regex.sub(r'\1https://old.reddit.com/\3', url)
|
||
|
||
url = url.replace("https://music.youtube.com/watch?v=", "https://youtube.com/watch?v=") \
|
||
.replace("https://www.youtube.com", "https://youtube.com") \
|
||
.replace("https://m.youtube.com", "https://youtube.com") \
|
||
.replace("https://youtube.com/shorts/", "https://youtube.com/watch?v=") \
|
||
.replace("https://youtube.com/v/", "https://youtube.com/watch?v=") \
|
||
.replace("https://mobile.twitter.com", "https://twitter.com") \
|
||
.replace("https://x.com", "https://twitter.com") \
|
||
.replace("https://www.twitter.com", "https://twitter.com") \
|
||
.replace("https://nitter.net/", "https://twitter.com/") \
|
||
.replace("https://nitter.42l.fr/", "https://twitter.com/") \
|
||
.replace("https://nitter.net/", "https://twitter.com/") \
|
||
.replace("https://m.facebook.com", "https://facebook.com") \
|
||
.replace("https://en.m.wikipedia.org", "https://en.wikipedia.org") \
|
||
.replace("https://www.instagram.com", "https://instagram.com") \
|
||
.replace("https://www.tiktok.com", "https://tiktok.com") \
|
||
.replace("https://imgur.com/", "https://i.imgur.com/") \
|
||
.replace("/giphy.gif", "/giphy.webp") \
|
||
.replace('https://www.google.com/amp/s/', 'https://') \
|
||
.replace('https://amp.', 'https://') \
|
||
.replace('https://cnn.com/cnn/', 'https://edition.cnn.com/') \
|
||
.replace('/amp/', '/') \
|
||
.replace('https://letmegooglethat.com/?q=', 'https://google.com/search?q=') \
|
||
.replace('https://lmgtfy.app/?q=', 'https://google.com/search?q=') \
|
||
|
||
if url.endswith('.amp'):
|
||
url = url.split('.amp')[0]
|
||
|
||
url = giphy_regex.sub(r'\1.webp', url)
|
||
|
||
if not url.startswith('/') and not url.startswith('https://rdrama.net') and not url.startswith('https://watchpeopledie.tv'):
|
||
try: parsed_url = urlparse(url)
|
||
except:
|
||
print(url, flush=True)
|
||
abort(500)
|
||
|
||
netloc = parsed_url.netloc
|
||
path = parsed_url.path.rstrip('/')
|
||
qd = parse_qs(parsed_url.query, keep_blank_values=True)
|
||
|
||
filtered = {}
|
||
|
||
if netloc == 'youtu.be':
|
||
filtered['v'] = path.lstrip('/')
|
||
netloc = 'youtube.com'
|
||
path = '/watch'
|
||
|
||
filtered |= {k: val for k, val in qd.items() if is_whitelisted(netloc, k)}
|
||
|
||
if netloc == 'old.reddit.com' and reddit_comment_link_regex.fullmatch(url):
|
||
filtered['context'] = 8
|
||
|
||
new_url = ParseResult(scheme="https",
|
||
netloc=netloc,
|
||
path=path,
|
||
params=parsed_url.params,
|
||
query=urlencode(filtered, doseq=True),
|
||
fragment=parsed_url.fragment)
|
||
url = urlunparse(new_url)
|
||
|
||
url = url.rstrip('/')
|
||
|
||
url = imgur_regex.sub(r'\1_d.webp?maxwidth=9999&fidelity=grand', url)
|
||
|
||
return url
|
||
|
||
def validate_css(css):
|
||
if '@import' in css:
|
||
return False, "CSS @import statements are not allowed!"
|
||
|
||
if '/*' in css:
|
||
return False, "CSS comments are not allowed!"
|
||
|
||
for i in css_url_regex.finditer(css):
|
||
url = i.group(1)
|
||
if not is_safe_url(url):
|
||
domain = tldextract.extract(url).registered_domain
|
||
return False, f"The domain '{domain}' is not allowed here!"
|
||
|
||
return True, ""
|
||
|
||
def torture_chud(string, username):
|
||
if not string: return string
|
||
for k, l in CHUD_REPLACEMENTS.items():
|
||
string = string.replace(k, l)
|
||
string = torture_regex.sub(rf'\1@{username}\3', string)
|
||
string = torture_regex2.sub(rf'\1@{username} is\3', string)
|
||
string = torture_regex3.sub(rf"\1@{username}'s\3", string)
|
||
return string
|
||
|
||
def torture_queen(string, key):
|
||
if not string: return string
|
||
result = initial_part_regex.search(string)
|
||
initial = result.group(1) if result else ""
|
||
string = string.lower()
|
||
string = initial_part_regex.sub("", string)
|
||
string = sentence_ending_regex.sub(", and", string)
|
||
string = superlative_regex.sub(r"literally \g<1>", string)
|
||
string = totally_regex.sub(r"totally \g<1>", string)
|
||
string = single_repeatable_punctuation.sub(r"\g<1>\g<1>\g<1>", string)
|
||
string = greeting_regex.sub(r"hiiiiiiiiii", string)
|
||
string = like_after_regex.sub(r"\g<1> like", string)
|
||
string = like_before_regex.sub(r"like \g<1>", string)
|
||
string = redpilled_regex.sub(r"goodpill\g<2>", string)
|
||
string = based_and_x_pilled_regex.sub(r"comfy \g<2> vibes", string)
|
||
string = based_regex.sub(r"comfy", string)
|
||
string = x_pilled_regex.sub(r"\g<2> vibes", string)
|
||
string = xmax_regex.sub(r"normalize good \g<2>s", string)
|
||
string = xmaxing_regex.sub(r"normalizing good \g<2>s", string)
|
||
string = xmaxed_regex.sub(r"normalized good \g<2>s", string)
|
||
|
||
string = normal_punctuation_regex.sub("", string)
|
||
string = more_than_one_comma_regex.sub(",", string)
|
||
if string[-5:] == ', and':
|
||
string = string[:-5]
|
||
|
||
random.seed(key)
|
||
if random.random() < PHRASE_CHANCE:
|
||
girl_phrase = random.choice(GIRL_PHRASES)
|
||
string = girl_phrase.replace("$", string)
|
||
string = initial + string
|
||
return string
|
||
|
||
def torture_object(obj, torture_method):
|
||
#torture body_html
|
||
if obj.body_html and '<p>&&' not in obj.body_html and '<p>$$' not in obj.body_html and '<p>##' not in obj.body_html:
|
||
soup = BeautifulSoup(obj.body_html, 'lxml')
|
||
tags = soup.html.body.find_all(lambda tag: tag.name not in {'blockquote','codeblock','pre'} and tag.string, recursive=False)
|
||
i = 0
|
||
for tag in tags:
|
||
i+=1
|
||
key = obj.id+i
|
||
tag.string.replace_with(torture_method(tag.string, key))
|
||
obj.body_html = str(soup).replace('<html><body>','').replace('</body></html>','')
|
||
|
||
#torture title_html and check for chud_phrase in plain title and leave if it's there
|
||
if isinstance(obj, Post):
|
||
obj.title_html = obj.title_html
|
||
|
||
def complies_with_chud(obj):
|
||
#check for cases where u should leave
|
||
if not (obj.chudded or obj.queened): return True
|
||
if obj.author.marseyawarded: return True
|
||
|
||
if isinstance(obj, Post):
|
||
if obj.id in ADMIGGER_THREADS: return True
|
||
if obj.sub == "chudrama": return True
|
||
elif obj.parent_post:
|
||
if obj.parent_post in ADMIGGER_THREADS: return True
|
||
if obj.post.sub == "chudrama": return True
|
||
|
||
if obj.chudded:
|
||
#perserve old body_html to be used in checking for chud phrase
|
||
old_body_html = obj.body_html
|
||
|
||
# TODO: Replace this code to make it more generic
|
||
#torture body_html
|
||
if obj.body_html and '<p>&&' not in obj.body_html and '<p>$$' not in obj.body_html and '<p>##' not in obj.body_html:
|
||
soup = BeautifulSoup(obj.body_html, 'lxml')
|
||
tags = soup.html.body.find_all(lambda tag: tag.name not in {'blockquote','codeblock','pre'} and tag.string, recursive=False)
|
||
for tag in tags:
|
||
tag.string.replace_with(torture_chud(tag.string, obj.author.username))
|
||
obj.body_html = str(soup).replace('<html><body>','').replace('</body></html>','')
|
||
|
||
#torture title_html and check for chud_phrase in plain title and leave if it's there
|
||
if isinstance(obj, Post):
|
||
obj.title_html = torture_chud(obj.title_html, obj.author.username)
|
||
if not obj.author.chud or obj.author.chud_phrase in obj.title.lower():
|
||
return True
|
||
|
||
#check for chud_phrase in body_html
|
||
if old_body_html:
|
||
excluded_tags = {'del','sub','sup','marquee','spoiler','lite-youtube','video','audio'}
|
||
soup = BeautifulSoup(old_body_html, 'lxml')
|
||
tags = soup.html.body.find_all(lambda tag: tag.name not in excluded_tags and not tag.attrs, recursive=False)
|
||
for tag in tags:
|
||
for text in tag.find_all(text=True, recursive=False):
|
||
if not obj.author.chud or obj.author.chud_phrase in text.lower():
|
||
return True
|
||
|
||
return False
|
||
elif obj.queened:
|
||
torture_object(obj, torture_queen)
|
||
return True
|