import json from sys import stdout import gevent from flask import g from pywebpush import webpush from files.classes import Comment, Notification, PushSubscription from .config.const import * from .regex import * from .sanitize import * def create_comment(text_html): new_comment = Comment(author_id=AUTOJANNY_ID, parent_submission=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(uid, text): if uid in bots: return text_html = sanitize(text) existing_comments = g.db.query(Comment.id).filter_by(author_id=AUTOJANNY_ID, parent_submission=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(text_html) notif = Notification(comment_id=cid, user_id=uid) g.db.add(notif) def send_notification(uid, text): if uid in bots: return cid = notif_comment(text) add_notif(cid, uid) def notif_comment(text): text_html = sanitize(text) existing = g.db.query(Comment.id).filter_by(author_id=AUTOJANNY_ID, parent_submission=None, body_html=text_html, is_bot=True).one_or_none() if existing: return existing[0] else: return create_comment(text_html) def notif_comment2(p): search_html = f'% has mentioned you: %' existing = g.db.query(Comment.id).filter(Comment.author_id == AUTOJANNY_ID, Comment.parent_submission == None, Comment.body_html.like(search_html)).first() if existing: return existing[0] else: text = f"@{p.author.username} has mentioned you: [{p.title}](/post/{p.id})" if p.sub: text += f" in /h/{p.sub}" text_html = sanitize(text) return create_comment(text_html) def add_notif(cid, uid): if uid in bots: return existing = g.db.query(Notification.user_id).filter_by(comment_id=cid, user_id=uid).one_or_none() if not existing: notif = Notification(comment_id=cid, user_id=uid) g.db.add(notif) def NOTIFY_USERS(text, v): # Restrict young accounts from generating notifications if v.age < NOTIFICATION_SPAM_AGE_THRESHOLD: return set() notify_users = set() for word, id in NOTIFIED_USERS.items(): if id == 0 or v.id == id: continue if word in text.lower() and id not in notify_users: notify_users.add(id) names = set(m.group(2) for m in mention_regex.finditer(text)) for user in get_users(names, graceful=True): if v.id != user.id and not v.any_block_exists(user): notify_users.add(user.id) if SITE_NAME == "WPD" and 'daisy' in text.lower(): admin_ids = [x[0] for x in g.db.query(User.id).filter(User.admin_level >= PERMS['NOTIFICATIONS_SPECIFIC_WPD_COMMENTS']).all()] notify_users.update(admin_ids) return notify_users - bots def push_notif(uid, title, body, url): if VAPID_PUBLIC_KEY == DEFAULT_CONFIG_VALUE: return if len(body) > PUSH_NOTIF_LIMIT: body = body[:PUSH_NOTIF_LIMIT] + "..." subscriptions = g.db.query(PushSubscription.subscription_json).filter_by(user_id=uid).all() subscriptions = [x[0] for x in subscriptions] gevent.spawn(_push_notif_thread, subscriptions, title, body, url) def _push_notif_thread(subscriptions, title, body, url): for subscription in subscriptions: try: response = webpush( subscription_info=json.loads(subscription), data=json.dumps({ "title": title, "body": body, 'url': url, 'icon': f'{SITE_FULL}/icon.webp?v=1', }), vapid_private_key=VAPID_PRIVATE_KEY, vapid_claims={"sub": f"mailto:{EMAIL}"} ) except: continue