forked from rDrama/rDrama
1103 lines
33 KiB
Python
1103 lines
33 KiB
Python
import os
|
|
import time
|
|
import html
|
|
from io import BytesIO
|
|
from os import path
|
|
from shutil import copyfile
|
|
from sys import stdout
|
|
from urllib.parse import ParseResult, urlparse, urlunparse, unquote
|
|
|
|
import gevent
|
|
import requests
|
|
from PIL import Image
|
|
|
|
from files.__main__ import app, cache, limiter
|
|
from files.classes import *
|
|
from files.helpers.actions import *
|
|
from files.helpers.alerts import *
|
|
from files.helpers.config.const import *
|
|
from files.helpers.get import *
|
|
from files.helpers.regex import *
|
|
from files.helpers.sanitize import *
|
|
from files.helpers.settings import get_setting
|
|
from files.helpers.slots import *
|
|
from files.helpers.sorting_and_time import *
|
|
from files.routes.routehelpers import execute_shadowban_viewers_and_voters
|
|
from files.routes.wrappers import *
|
|
|
|
from .front import frontlist
|
|
from .users import userpagelisting
|
|
|
|
from files.__main__ import app, limiter, redis_instance
|
|
|
|
def _add_post_view(pid):
|
|
db = db_session()
|
|
|
|
p = db.get(Post, pid)
|
|
p.views += 1
|
|
db.add(p)
|
|
|
|
db.commit()
|
|
db.close()
|
|
stdout.flush()
|
|
|
|
@app.post("/publish/<int:pid>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@is_not_banned
|
|
def publish(pid, v):
|
|
p = get_post(pid)
|
|
if not p.private: return {"message": "Post published!"}
|
|
|
|
if p.author_id != v.id: abort(403)
|
|
p.private = False
|
|
p.created_utc = int(time.time())
|
|
g.db.add(p)
|
|
|
|
notify_users = NOTIFY_USERS(f'{p.title} {p.body}', v, ghost=p.ghost, log_cost=p)
|
|
|
|
if notify_users:
|
|
cid, text = notif_comment2(p)
|
|
if notify_users == 'everyone':
|
|
alert_everyone(cid)
|
|
else:
|
|
for x in notify_users:
|
|
add_notif(cid, x, text, pushnotif_url=p.permalink)
|
|
|
|
|
|
cache.delete_memoized(frontlist)
|
|
cache.delete_memoized(userpagelisting)
|
|
|
|
execute_snappy(p, v)
|
|
|
|
return {"message": "Post has been published successfully!"}
|
|
|
|
@app.get("/submit")
|
|
@app.get("/h/<sub>/submit")
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def submit_get(v:User, sub=None):
|
|
sub = get_sub_by_name(sub, graceful=True)
|
|
if request.path.startswith('/h/') and not sub: abort(404)
|
|
|
|
SUBS = [x[0] for x in g.db.query(Sub.name).order_by(Sub.name).all()]
|
|
|
|
return render_template("submit.html", SUBS=SUBS, v=v, sub=sub)
|
|
|
|
@app.get("/post/<int:pid>")
|
|
@app.get("/post/<int:pid>/<anything>")
|
|
@app.get("/h/<sub>/post/<int:pid>")
|
|
@app.get("/h/<sub>/post/<int:pid>/<anything>")
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@auth_desired_with_logingate
|
|
def post_id(pid, v, anything=None, sub=None):
|
|
p = get_post(pid, v=v)
|
|
if not User.can_see(v, p): abort(403)
|
|
|
|
if p.over_18 and not (v and v.over_18) and session.get('over_18_cookies', 0) < int(time.time()):
|
|
if g.is_api_or_xhr: abort(451, "Must be 18+ to view")
|
|
return render_template("errors/nsfw.html", v=v)
|
|
|
|
gevent.spawn(_add_post_view, pid)
|
|
|
|
if p.new: defaultsortingcomments = 'new'
|
|
elif v: defaultsortingcomments = v.defaultsortingcomments
|
|
else: defaultsortingcomments = "hot"
|
|
sort = request.values.get("sort", defaultsortingcomments)
|
|
|
|
if sort == 'saves':
|
|
sort = defaultsortingcomments
|
|
|
|
if not v:
|
|
result = cache.get(f'post_{p.id}_{sort}')
|
|
if result:
|
|
calc_users()
|
|
return result
|
|
|
|
if v:
|
|
execute_shadowban_viewers_and_voters(v, p)
|
|
# shadowban check is done in sort_objects
|
|
# output is needed: see comments.py
|
|
comments, output = get_comments_v_properties(v, None, Comment.parent_post == p.id, Comment.level < 10)
|
|
|
|
if sort == "hot":
|
|
pinned = [c[0] for c in comments.filter(Comment.stickied != None).order_by(Comment.created_utc.desc()).all()]
|
|
comments = comments.filter(Comment.stickied == None)
|
|
|
|
comments = comments.filter(Comment.level == 1)
|
|
comments = sort_objects(sort, comments, Comment)
|
|
comments = [c[0] for c in comments.all()]
|
|
else:
|
|
comments = g.db.query(Comment).filter(Comment.parent_post == p.id)
|
|
|
|
if sort == "hot":
|
|
pinned = comments.filter(Comment.stickied != None).order_by(Comment.created_utc.desc()).all()
|
|
comments = comments.filter(Comment.stickied == None)
|
|
|
|
comments = comments.filter(Comment.level == 1)
|
|
comments = sort_objects(sort, comments, Comment)
|
|
comments = comments.all()
|
|
|
|
offset = 0
|
|
ids = set()
|
|
|
|
threshold = 100
|
|
|
|
if p.comment_count > threshold+25 and not (v and v.client):
|
|
comments2 = []
|
|
count = 0
|
|
if p.created_utc > 1638672040: # TODO: migrate old comments to use top_comment_id
|
|
for comment in comments:
|
|
comments2.append(comment)
|
|
ids.add(comment.id)
|
|
count += g.db.query(Comment).filter_by(parent_post=p.id, top_comment_id=comment.id).count() + 1
|
|
if count > threshold: break
|
|
else:
|
|
for comment in comments:
|
|
comments2.append(comment)
|
|
ids.add(comment.id)
|
|
count += g.db.query(Comment).filter_by(parent_post=p.id, parent_comment_id=comment.id).count() + 1
|
|
if count > 20: break
|
|
|
|
if len(comments) == len(comments2): offset = 0
|
|
else: offset = 1
|
|
comments = comments2
|
|
|
|
p.replies = comments
|
|
|
|
if sort == "hot":
|
|
pinned2 = {}
|
|
for pin in pinned:
|
|
if pin.level > 1:
|
|
pinned2[pin.top_comment] = ''
|
|
if pin.top_comment in comments:
|
|
comments.remove(pin.top_comment)
|
|
else:
|
|
pinned2[pin] = ''
|
|
|
|
p.replies = list(pinned2.keys()) + p.replies
|
|
|
|
if v and v.client:
|
|
return p.json
|
|
|
|
template = "post.html"
|
|
if (p.is_banned or p.author.shadowbanned) \
|
|
and not (v and (v.admin_level >= PERMS['POST_COMMENT_MODERATION'] or p.author_id == v.id)):
|
|
template = "post_banned.html"
|
|
|
|
result = render_template(template, v=v, p=p, ids=list(ids),
|
|
sort=sort, render_replies=True, offset=offset, sub=p.subr,
|
|
fart=get_setting('fart_mode'))
|
|
|
|
if not v:
|
|
cache.set(f'post_{p.id}_{sort}', result, timeout=3600)
|
|
|
|
return result
|
|
|
|
@app.get("/view_more/<int:pid>/<sort>/<offset>")
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@auth_desired_with_logingate
|
|
def view_more(v, pid, sort, offset):
|
|
p = get_post(pid, v=v)
|
|
try:
|
|
offset = int(offset)
|
|
except: abort(400)
|
|
try: ids = set(int(x) for x in request.values.get("ids").split(','))
|
|
except: abort(400)
|
|
|
|
if v:
|
|
# shadowban check is done in sort_objects
|
|
# output is needed: see comments.py
|
|
comments, output = get_comments_v_properties(v, None, Comment.parent_post == pid, Comment.stickied == None, Comment.id.notin_(ids), Comment.level < 10)
|
|
comments = comments.filter(Comment.level == 1)
|
|
comments = sort_objects(sort, comments, Comment)
|
|
|
|
comments = [c[0] for c in comments.all()]
|
|
else:
|
|
comments = g.db.query(Comment).filter(
|
|
Comment.parent_post == pid,
|
|
Comment.level == 1,
|
|
Comment.stickied == None,
|
|
Comment.id.notin_(ids)
|
|
)
|
|
|
|
comments = sort_objects(sort, comments, Comment)
|
|
|
|
comments = comments.offset(offset).all()
|
|
|
|
comments2 = []
|
|
count = 0
|
|
if p.created_utc > 1638672040: # TODO: migrate old comments to use top_comment_id
|
|
for comment in comments:
|
|
comments2.append(comment)
|
|
ids.add(comment.id)
|
|
count += g.db.query(Comment).filter_by(parent_post=p.id, top_comment_id=comment.id).count() + 1
|
|
if count > 100: break
|
|
else:
|
|
for comment in comments:
|
|
comments2.append(comment)
|
|
ids.add(comment.id)
|
|
count += g.db.query(Comment).filter_by(parent_post=p.id, parent_comment_id=comment.id).count() + 1
|
|
if count > 20: break
|
|
|
|
if len(comments) == len(comments2): offset = 0
|
|
else: offset += 1
|
|
comments = comments2
|
|
|
|
return render_template("comments.html", v=v, comments=comments, p=p, ids=list(ids), render_replies=True, pid=pid, sort=sort, offset=offset)
|
|
|
|
|
|
@app.get("/more_comments/<int:cid>")
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@auth_desired_with_logingate
|
|
def more_comments(v, cid):
|
|
try: cid = int(cid)
|
|
except: abort(404)
|
|
|
|
tcid = g.db.query(Comment.top_comment_id).filter_by(id=cid).one_or_none()[0]
|
|
|
|
if v:
|
|
# shadowban check is done in sort_objects i think
|
|
# output is needed: see comments.py
|
|
comments, output = get_comments_v_properties(v, None, Comment.top_comment_id == tcid, Comment.level > 9)
|
|
comments = comments.filter(Comment.parent_comment_id == cid)
|
|
comments = [c[0] for c in comments.all()]
|
|
else:
|
|
c = get_comment(cid)
|
|
comments = c.replies(sort=request.values.get('sort'))
|
|
|
|
if comments: p = comments[0].post
|
|
else: p = None
|
|
|
|
return render_template("comments.html", v=v, comments=comments, p=p, render_replies=True)
|
|
|
|
def thumbnail_thread(pid:int, vid:int):
|
|
db = db_session()
|
|
def expand_url(post_url, fragment_url):
|
|
if fragment_url.startswith("https://"):
|
|
return fragment_url
|
|
elif fragment_url.startswith("https://"):
|
|
return f"https://{fragment_url.split('https://')[1]}"
|
|
elif fragment_url.startswith('//'):
|
|
return f"https:{fragment_url}"
|
|
elif fragment_url.startswith('/') and '\\' not in fragment_url:
|
|
parsed_url = urlparse(post_url)
|
|
return f"https://{parsed_url.netloc}{fragment_url}"
|
|
else:
|
|
return f"{post_url}/{fragment_url}"
|
|
|
|
p = db.get(Post, pid)
|
|
|
|
if not p or not p.url:
|
|
time.sleep(5)
|
|
p = db.get(Post, pid)
|
|
|
|
if not p or not p.url: return
|
|
|
|
fetch_url = p.url
|
|
|
|
if fetch_url.startswith('/') and '\\' not in fetch_url:
|
|
fetch_url = f"{SITE_FULL}{fetch_url}"
|
|
|
|
try:
|
|
x=requests.get(fetch_url, headers=HEADERS, timeout=5, proxies=proxies)
|
|
except:
|
|
db.close()
|
|
return
|
|
|
|
if x.status_code != 200:
|
|
db.close()
|
|
return
|
|
|
|
if x.headers.get("Content-Type","").startswith("text/html"):
|
|
soup=BeautifulSoup(x.content, 'lxml')
|
|
|
|
thumb_candidate_urls=[]
|
|
|
|
meta_tags = [
|
|
"drama:thumbnail",
|
|
"twitter:image",
|
|
"og:image",
|
|
"thumbnail"
|
|
]
|
|
|
|
for tag_name in meta_tags:
|
|
|
|
|
|
tag = soup.find(
|
|
'meta',
|
|
attrs={
|
|
"name": tag_name,
|
|
"content": True
|
|
}
|
|
)
|
|
if not tag:
|
|
tag = soup.find(
|
|
'meta',
|
|
attrs={
|
|
'property': tag_name,
|
|
'content': True
|
|
}
|
|
)
|
|
if tag:
|
|
thumb_candidate_urls.append(expand_url(p.url, tag['content']))
|
|
|
|
for tag in soup.find_all("img", attrs={'src':True}):
|
|
thumb_candidate_urls.append(expand_url(p.url, tag['src']))
|
|
|
|
|
|
for url in thumb_candidate_urls:
|
|
try:
|
|
image_req=requests.get(url, headers=HEADERS, timeout=5, proxies=proxies)
|
|
except:
|
|
continue
|
|
|
|
if image_req.status_code >= 400:
|
|
continue
|
|
|
|
if not image_req.headers.get("Content-Type","").startswith("image/"):
|
|
continue
|
|
|
|
if image_req.headers.get("Content-Type","").startswith("image/svg"):
|
|
continue
|
|
|
|
with Image.open(BytesIO(image_req.content)) as i:
|
|
if i.width < 30 or i.height < 30:
|
|
continue
|
|
break
|
|
else:
|
|
db.close()
|
|
return
|
|
|
|
elif x.headers.get("Content-Type","").startswith("image/"):
|
|
image_req=x
|
|
with Image.open(BytesIO(x.content)) as i:
|
|
size = len(i.fp.read())
|
|
if size > 8 * 1024 * 1024:
|
|
db.close()
|
|
return
|
|
|
|
else:
|
|
db.close()
|
|
return
|
|
|
|
name = f'/images/{time.time()}'.replace('.','') + '.webp'
|
|
|
|
with open(name, "wb") as file:
|
|
for chunk in image_req.iter_content(1024):
|
|
file.write(chunk)
|
|
|
|
v = db.get(User, vid)
|
|
url = process_image(name, v, resize=99, uploader_id=p.author_id, db=db)
|
|
if url:
|
|
p.thumburl = url
|
|
db.add(p)
|
|
db.commit()
|
|
db.close()
|
|
stdout.flush()
|
|
|
|
|
|
@app.post("/is_repost")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def is_repost(v):
|
|
not_a_repost = {'permalink': ''}
|
|
if not FEATURES['REPOST_DETECTION']:
|
|
return not_a_repost
|
|
|
|
url = request.values.get('url')
|
|
if not url or len(url) < MIN_REPOST_CHECK_URL_LENGTH:
|
|
abort(400)
|
|
|
|
url = normalize_url(url)
|
|
url = unquote(url)
|
|
parsed_url = urlparse(url)
|
|
|
|
domain = parsed_url.netloc
|
|
if domain in {'old.reddit.com','twitter.com','instagram.com','tiktok.com'} and '/search' not in url:
|
|
new_url = ParseResult(scheme="https",
|
|
netloc=parsed_url.netloc,
|
|
path=parsed_url.path,
|
|
params=parsed_url.params,
|
|
query=None,
|
|
fragment=parsed_url.fragment)
|
|
else:
|
|
qd = parse_qs(parsed_url.query, keep_blank_values=True)
|
|
filtered = {k: val for k, val in qd.items() if not k.startswith('utm_') and not k.startswith('ref_')}
|
|
|
|
new_url = ParseResult(scheme="https",
|
|
netloc=parsed_url.netloc,
|
|
path=parsed_url.path,
|
|
params=parsed_url.params,
|
|
query=urlencode(filtered, doseq=True),
|
|
fragment=parsed_url.fragment)
|
|
|
|
url = urlunparse(new_url)
|
|
url = url.rstrip('/')
|
|
|
|
search_url = url.replace('%', '').replace('\\', '').replace('_', '\_').strip()
|
|
repost = g.db.query(Post).filter(
|
|
Post.url.ilike(search_url),
|
|
Post.deleted_utc == 0,
|
|
Post.is_banned == False
|
|
).first()
|
|
if repost: return {'permalink': repost.permalink}
|
|
else: return not_a_repost
|
|
|
|
@app.post("/submit")
|
|
@app.post("/h/<sub>/submit")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(POST_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(POST_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@is_not_banned
|
|
def submit_post(v:User, sub=None):
|
|
url = request.values.get("url", "").strip()
|
|
|
|
if '\\' in url: abort(400)
|
|
|
|
title = sanitize_raw_title(request.values.get("title", ""))
|
|
body = sanitize_raw_body(request.values.get("body", ""), True)
|
|
|
|
if not title:
|
|
abort(400, "Please enter a better title!")
|
|
|
|
sub = request.values.get("sub", "").lower().replace('/h/','').strip()
|
|
|
|
if SITE == 'rdrama.net' and (v.chud == 1 or v.id == 253):
|
|
sub = 'chudrama'
|
|
|
|
title_html = filter_emojis_only(title, graceful=True, count_emojis=True)
|
|
|
|
if v.marseyawarded and not marseyaward_title_regex.fullmatch(title_html):
|
|
abort(400, "You can only type marseys!")
|
|
if len(title_html) > POST_TITLE_HTML_LENGTH_LIMIT:
|
|
abort(400, "Rendered title is too big!")
|
|
|
|
if sub == 'changelog':
|
|
abort(400, "/h/changelog is archived")
|
|
|
|
if sub in {'furry','vampire','racist','femboy','edgy'} and not v.client and not v.house.lower().startswith(sub):
|
|
abort(400, f"You need to be a member of House {sub.capitalize()} to post in /h/{sub}")
|
|
|
|
if sub and sub != 'none':
|
|
sname = sub.strip().lower()
|
|
sub = g.db.query(Sub.name).filter_by(name=sname).one_or_none()
|
|
if not sub: abort(400, f"/h/{sname} not found!")
|
|
sub = sub[0]
|
|
if v.exiled_from(sub): abort(400, f"You're exiled from /h/{sub}")
|
|
else: sub = None
|
|
|
|
if not sub and HOLE_REQUIRED:
|
|
abort(400, f"You must choose a {HOLE_NAME} for your post!")
|
|
|
|
if v.longpost and (len(body) < 280 or ' [](' in body or body.startswith('[](')):
|
|
abort(400, "You have to type more than 280 characters!")
|
|
elif v.bird and len(body) > 140:
|
|
abort(400, "You have to type less than 140 characters!")
|
|
|
|
|
|
embed = None
|
|
|
|
if url:
|
|
url = normalize_url(url)
|
|
url = unquote(url)
|
|
parsed_url = urlparse(url)
|
|
|
|
domain = parsed_url.netloc
|
|
if domain in {'old.reddit.com','twitter.com','instagram.com','tiktok.com'} and '/search' not in url:
|
|
new_url = ParseResult(scheme="https",
|
|
netloc=parsed_url.netloc,
|
|
path=parsed_url.path,
|
|
params=parsed_url.params,
|
|
query=None,
|
|
fragment=parsed_url.fragment)
|
|
else:
|
|
qd = parse_qs(parsed_url.query, keep_blank_values=True)
|
|
filtered = {k: val for k, val in qd.items() if not k.startswith('utm_') and not k.startswith('ref_')}
|
|
|
|
new_url = ParseResult(scheme="https",
|
|
netloc=parsed_url.netloc,
|
|
path=parsed_url.path,
|
|
params=parsed_url.params,
|
|
query=urlencode(filtered, doseq=True),
|
|
fragment=parsed_url.fragment)
|
|
|
|
url = urlunparse(new_url)
|
|
|
|
url = url.rstrip('/')
|
|
|
|
if v.admin_level < PERMS["IGNORE_DOMAIN_BAN"]:
|
|
y = tldextract.extract(url).registered_domain + parsed_url.path
|
|
y = y.lower()
|
|
banned_domains = g.db.query(BannedDomain).all()
|
|
for x in banned_domains:
|
|
if y.startswith(x.domain):
|
|
abort(400, f'Remove the banned link "{x.domain}" and try again!\nReason for link ban: "{x.reason}"')
|
|
|
|
if "twitter.com" == domain:
|
|
try:
|
|
embed = requests.get("https://publish.twitter.com/oembed", params={"url":url, "omit_script":"t"}, timeout=5).json()["html"]
|
|
embed = embed.replace('<a href', '<a rel="nofollow noopener" href')
|
|
except: pass
|
|
elif url.startswith('https://youtube.com/watch?'):
|
|
embed = handle_youtube_links(url)
|
|
elif SITE in domain and "/post/" in url and "context" not in url and url.count('/') < 6:
|
|
id = url.split("/post/")[1]
|
|
if "/" in id: id = id.split("/")[0]
|
|
embed = str(int(id))
|
|
|
|
|
|
if not url and not body and not request.files.get("file") and not request.files.get("file-url"):
|
|
abort(400, "Please enter a url or some text!")
|
|
|
|
if not IS_LOCALHOST:
|
|
dup = g.db.query(Post).filter(
|
|
Post.author_id == v.id,
|
|
Post.deleted_utc == 0,
|
|
Post.title == title,
|
|
Post.url == url,
|
|
Post.body == body
|
|
).one_or_none()
|
|
if dup:
|
|
return {"post_id": dup.id, "success": False}
|
|
|
|
if not execute_antispam_post_check(title, v, url):
|
|
return redirect("/notifications")
|
|
|
|
if len(url) > 2048:
|
|
abort(400, "There's a 2048 character limit for URLs!")
|
|
|
|
body = process_files(request.files, v, body)
|
|
body = body.strip()[:POST_BODY_LENGTH_LIMIT(v)] # process_files() adds content to the body, so we need to re-strip
|
|
|
|
body_html = sanitize(body, count_emojis=True, limit_pings=100)
|
|
|
|
if v.marseyawarded and marseyaward_body_regex.search(body_html):
|
|
abort(400, "You can only type marseys!")
|
|
|
|
if len(body_html) > POST_BODY_HTML_LENGTH_LIMIT:
|
|
abort(400, "Post body_html too long!")
|
|
|
|
flag_notify = (request.values.get("notify", "on") == "on")
|
|
flag_new = request.values.get("new", False, bool) or 'megathread' in title.lower()
|
|
flag_over_18 = FEATURES['NSFW_MARKING'] and request.values.get("over_18", False, bool)
|
|
flag_private = request.values.get("private", False, bool)
|
|
flag_ghost = request.values.get("ghost", False, bool) and v.can_post_in_ghost_threads
|
|
|
|
if flag_ghost: sub = None
|
|
|
|
if embed and len(embed) > 1500: embed = None
|
|
if embed: embed = embed.strip()
|
|
|
|
if url and url.startswith(f'{SITE_FULL}/'):
|
|
url = url.split(SITE_FULL)[1]
|
|
|
|
if url == '': url = None
|
|
|
|
flag_chudded = v.chud and sub != 'chudrama'
|
|
|
|
p = Post(
|
|
private=flag_private,
|
|
notify=flag_notify,
|
|
author_id=v.id,
|
|
over_18=flag_over_18,
|
|
new=flag_new,
|
|
app_id=v.client.application.id if v.client else None,
|
|
is_bot=(v.client is not None),
|
|
url=url,
|
|
body=body,
|
|
body_html=body_html,
|
|
embed=embed,
|
|
title=title,
|
|
title_html=title_html,
|
|
sub=sub,
|
|
ghost=flag_ghost,
|
|
chudded=flag_chudded,
|
|
)
|
|
|
|
g.db.add(p)
|
|
g.db.flush()
|
|
|
|
execute_under_siege(v, p, p.body, 'post')
|
|
|
|
process_poll_options(v, p)
|
|
|
|
for text in {p.body, p.title, p.url}:
|
|
if execute_blackjack(v, p, text, 'post'): break
|
|
|
|
vote = Vote(user_id=v.id,
|
|
vote_type=1,
|
|
post_id=p.id,
|
|
coins=0
|
|
)
|
|
g.db.add(vote)
|
|
|
|
if request.files.get('file-url') and not g.is_tor:
|
|
file = request.files['file-url']
|
|
|
|
if file.content_type.startswith('image/'):
|
|
name = f'/images/{time.time()}'.replace('.','') + '.webp'
|
|
file.save(name)
|
|
p.url = process_image(name, v)
|
|
|
|
name2 = name.replace('.webp', 'r.webp')
|
|
copyfile(name, name2)
|
|
p.thumburl = process_image(name2, v, resize=99)
|
|
elif file.content_type.startswith('video/'):
|
|
p.url = process_video(file, v)
|
|
name = f'/images/{time.time()}'.replace('.','') + '.webp'
|
|
try:
|
|
subprocess.run(['ffmpeg', '-y', '-loglevel', 'warning',
|
|
'-i', p.url, '-vf', "scale='iw':-2",
|
|
'-q:v', '3', '-frames:v', '1', name], check=True, timeout=SUBPROCESS_TIMEOUT_DURATION)
|
|
except:
|
|
if os.path.isfile(name):
|
|
os.remove(name)
|
|
else:
|
|
p.posterurl = name
|
|
name2 = name.replace('.webp', 'r.webp')
|
|
copyfile(name, name2)
|
|
p.thumburl = process_image(name2, v, resize=99)
|
|
elif file.content_type.startswith('audio/'):
|
|
p.url = process_audio(file, v)
|
|
else:
|
|
abort(415)
|
|
|
|
if not p.thumburl and p.url and p.domain != SITE:
|
|
gevent.spawn(thumbnail_thread, p.id, v.id)
|
|
|
|
if not p.private:
|
|
notify_users = NOTIFY_USERS(f'{title} {body}', v, ghost=p.ghost, log_cost=p)
|
|
|
|
if notify_users:
|
|
cid, text = notif_comment2(p)
|
|
if notify_users == 'everyone':
|
|
alert_everyone(cid)
|
|
else:
|
|
for x in notify_users:
|
|
add_notif(cid, x, text, pushnotif_url=p.permalink)
|
|
|
|
if not complies_with_chud(p):
|
|
p.is_banned = True
|
|
p.ban_reason = "AutoJanny"
|
|
|
|
body = CHUD_MSG.format(username=v.username, type='post', CHUD_PHRASE=v.chud_phrase)
|
|
body_jannied_html = sanitize(body)
|
|
|
|
|
|
c_jannied = Comment(author_id=AUTOJANNY_ID,
|
|
parent_post=p.id,
|
|
level=1,
|
|
over_18=False,
|
|
is_bot=True,
|
|
app_id=None,
|
|
distinguish_level=6,
|
|
body=body,
|
|
body_html=body_jannied_html,
|
|
ghost=p.ghost
|
|
)
|
|
|
|
g.db.add(c_jannied)
|
|
g.db.flush()
|
|
|
|
p.comment_count += 1
|
|
g.db.add(p)
|
|
|
|
c_jannied.top_comment_id = c_jannied.id
|
|
|
|
n = Notification(comment_id=c_jannied.id, user_id=v.id)
|
|
g.db.add(n)
|
|
|
|
autojanny = g.db.get(User, AUTOJANNY_ID)
|
|
autojanny.comment_count += 1
|
|
g.db.add(autojanny)
|
|
|
|
v.post_count = g.db.query(Post).filter_by(author_id=v.id, deleted_utc=0).count()
|
|
g.db.add(v)
|
|
|
|
execute_lawlz_actions(v, p)
|
|
|
|
if (SITE == 'rdrama.net'
|
|
and v.id in {TGTW_ID, SNALLY_ID}
|
|
and not (p.sub and p.subr.stealth)) and p.sub != 'slavshit' and not p.ghost:
|
|
p.stickied_utc = int(time.time()) + 28800
|
|
p.stickied = "AutoJanny"
|
|
|
|
if SITE == 'rdrama.net' and v.id == 7465 and "women's world cup" in p.title.lower():
|
|
p.stickied_utc = int(time.time()) + 28800
|
|
p.stickied = "AutoJanny"
|
|
|
|
cache.delete_memoized(frontlist)
|
|
cache.delete_memoized(userpagelisting)
|
|
|
|
key_pattern = app.config["CACHE_KEY_PREFIX"] + 'frontpage_*'
|
|
for key in redis_instance.scan_iter(key_pattern):
|
|
redis_instance.delete(key)
|
|
|
|
if not p.private:
|
|
execute_snappy(p, v)
|
|
|
|
if v.client: return p.json
|
|
else:
|
|
p.voted = 1
|
|
return {"post_id": p.id, "success": True}
|
|
|
|
@app.post("/delete_post/<int:pid>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def delete_post_pid(pid, v):
|
|
p = get_post(pid)
|
|
if p.author_id != v.id: abort(403)
|
|
|
|
# Temporary special logic by Carp request for events of 2022-10-10
|
|
if SITE_NAME == 'rDrama' and p.author_id == 3161: abort(403)
|
|
|
|
if not p.deleted_utc:
|
|
p.deleted_utc = int(time.time())
|
|
p.is_pinned = False
|
|
p.stickied = None
|
|
|
|
g.db.add(p)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
cache.delete_memoized(userpagelisting)
|
|
|
|
v.post_count = g.db.query(Post).filter_by(author_id=v.id, deleted_utc=0).count()
|
|
g.db.add(v)
|
|
|
|
return {"message": "Post deleted!"}
|
|
|
|
@app.post("/undelete_post/<int:pid>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def undelete_post_pid(pid, v):
|
|
p = get_post(pid)
|
|
if p.author_id != v.id: abort(403)
|
|
|
|
if p.deleted_utc:
|
|
p.deleted_utc = 0
|
|
g.db.add(p)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
cache.delete_memoized(userpagelisting)
|
|
|
|
v.post_count = g.db.query(Post).filter_by(author_id=v.id, deleted_utc=0).count()
|
|
g.db.add(v)
|
|
|
|
return {"message": "Post undeleted!"}
|
|
|
|
|
|
@app.post("/mark_post_nsfw/<int:pid>")
|
|
@feature_required('NSFW_MARKING')
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def mark_post_nsfw(pid, v):
|
|
p = get_post(pid)
|
|
|
|
if p.author_id != v.id and not v.admin_level >= PERMS['POST_COMMENT_MODERATION'] and not (p.sub and v.mods(p.sub)):
|
|
abort(403)
|
|
|
|
if p.over_18 and v.is_permabanned:
|
|
abort(403)
|
|
|
|
p.over_18 = True
|
|
g.db.add(p)
|
|
|
|
if p.author_id != v.id:
|
|
if v.admin_level >= PERMS['POST_COMMENT_MODERATION']:
|
|
ma = ModAction(
|
|
kind = "set_nsfw",
|
|
user_id = v.id,
|
|
target_post_id = p.id,
|
|
)
|
|
g.db.add(ma)
|
|
else:
|
|
ma = SubAction(
|
|
sub = p.sub,
|
|
kind = "set_nsfw",
|
|
user_id = v.id,
|
|
target_post_id = p.id,
|
|
)
|
|
g.db.add(ma)
|
|
send_repeatable_notification(p.author_id, f"@{v.username} (a site admin) has marked [{p.title}](/post/{p.id}) as +18")
|
|
|
|
return {"message": "Post has been marked as +18!"}
|
|
|
|
@app.post("/unmark_post_nsfw/<int:pid>")
|
|
@feature_required('NSFW_MARKING')
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def unmark_post_nsfw(pid, v):
|
|
p = get_post(pid)
|
|
|
|
if p.author_id != v.id and not v.admin_level >= PERMS['POST_COMMENT_MODERATION'] and not (p.sub and v.mods(p.sub)):
|
|
abort(403)
|
|
|
|
if p.over_18 and v.is_permabanned:
|
|
abort(403)
|
|
|
|
p.over_18 = False
|
|
g.db.add(p)
|
|
|
|
if p.author_id != v.id:
|
|
if v.admin_level >= PERMS['POST_COMMENT_MODERATION']:
|
|
ma = ModAction(
|
|
kind = "unset_nsfw",
|
|
user_id = v.id,
|
|
target_post_id = p.id,
|
|
)
|
|
g.db.add(ma)
|
|
else:
|
|
ma = SubAction(
|
|
sub = p.sub,
|
|
kind = "unset_nsfw",
|
|
user_id = v.id,
|
|
target_post_id = p.id,
|
|
)
|
|
g.db.add(ma)
|
|
send_repeatable_notification(p.author_id, f"@{v.username} (a site admin) has unmarked [{p.title}](/post/{p.id}) as +18")
|
|
|
|
return {"message": "Post has been unmarked as +18!"}
|
|
|
|
@app.post("/save_post/<int:pid>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def save_post(pid, v):
|
|
|
|
p = get_post(pid)
|
|
|
|
save = g.db.query(SaveRelationship).filter_by(user_id=v.id, post_id=p.id).one_or_none()
|
|
|
|
if not save:
|
|
new_save=SaveRelationship(user_id=v.id, post_id=p.id)
|
|
g.db.add(new_save)
|
|
cache.delete_memoized(userpagelisting)
|
|
|
|
return {"message": "Post saved!"}
|
|
|
|
@app.post("/unsave_post/<int:pid>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def unsave_post(pid, v):
|
|
|
|
p = get_post(pid)
|
|
|
|
save = g.db.query(SaveRelationship).filter_by(user_id=v.id, post_id=p.id).one_or_none()
|
|
|
|
if save:
|
|
g.db.delete(save)
|
|
cache.delete_memoized(userpagelisting)
|
|
|
|
return {"message": "Post unsaved!"}
|
|
|
|
@app.post("/pin/<int:post_id>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def pin_post(post_id, v):
|
|
p = get_post(post_id)
|
|
if p:
|
|
if v.id != p.author_id: abort(403, "Only the post author can do that!")
|
|
p.is_pinned = not p.is_pinned
|
|
g.db.add(p)
|
|
cache.delete_memoized(userpagelisting)
|
|
if p.is_pinned: return {"message": "Post pinned!"}
|
|
else: return {"message": "Post unpinned!"}
|
|
return abort(404, "Post not found!")
|
|
|
|
@app.put("/post/<int:post_id>/new")
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def set_new_sort(post_id:int, v:User):
|
|
p = get_post(post_id)
|
|
if not v.can_edit(p): abort(403, "Only the post author can do that!")
|
|
p.new = True
|
|
g.db.add(p)
|
|
|
|
if v.id != p.author_id:
|
|
ma = ModAction(
|
|
kind = "set_new",
|
|
user_id = v.id,
|
|
target_post_id = p.id,
|
|
)
|
|
g.db.add(ma)
|
|
send_repeatable_notification(p.author_id, f"@{v.username} (a site admin) has changed the the default sorting of comments on [{p.title}](/post/{p.id}) to `new`")
|
|
|
|
return {"message": "Changed the the default sorting of comments on this post to 'new'"}
|
|
|
|
|
|
@app.delete("/post/<int:post_id>/new")
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def unset_new_sort(post_id:int, v:User):
|
|
p = get_post(post_id)
|
|
if not v.can_edit(p): abort(403, "Only the post author can do that!")
|
|
p.new = None
|
|
g.db.add(p)
|
|
|
|
if v.id != p.author_id:
|
|
ma = ModAction(
|
|
kind = "set_hot",
|
|
user_id = v.id,
|
|
target_post_id = p.id,
|
|
)
|
|
g.db.add(ma)
|
|
send_repeatable_notification(p.author_id, f"@{v.username} (a site admin) has changed the the default sorting of comments on [{p.title}](/post/{p.id}) to `hot`")
|
|
|
|
return {"message": "Changed the the default sorting of comments on this post to 'hot'"}
|
|
|
|
|
|
extensions = IMAGE_FORMATS + VIDEO_FORMATS + AUDIO_FORMATS
|
|
|
|
@app.get("/submit/title")
|
|
@limiter.limit("3/minute", deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit("3/minute", deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@auth_required
|
|
def get_post_title(v):
|
|
POST_TITLE_TIMEOUT = 5
|
|
url = request.values.get("url")
|
|
if not url or '\\' in url: abort(400)
|
|
url = url.strip()
|
|
if not url.startswith('http'): abort(400)
|
|
|
|
checking_url = url.lower().split('?')[0].split('%3F')[0]
|
|
if any((checking_url.endswith(f'.{x}') for x in extensions)):
|
|
abort(400)
|
|
|
|
try:
|
|
x = gevent.with_timeout(POST_TITLE_TIMEOUT, requests.get, url, headers=HEADERS, timeout=POST_TITLE_TIMEOUT, proxies=proxies)
|
|
except: abort(400)
|
|
|
|
content_type = x.headers.get("Content-Type")
|
|
if not content_type or "text/html" not in content_type: abort(400)
|
|
|
|
# no you can't just parse html with reeeeeeeegex
|
|
match = html_title_regex.search(x.text)
|
|
if match and match.lastindex >= 1:
|
|
title = match.group(1)
|
|
else: abort(400)
|
|
|
|
title = html.unescape(title)
|
|
|
|
return {"url": url, "title": title}
|
|
|
|
@app.post("/edit_post/<int:pid>")
|
|
@limiter.limit('1/second', scope=rpath)
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
|
@limiter.limit("10/minute;100/hour;200/day", deduct_when=lambda response: response.status_code < 400)
|
|
@limiter.limit("10/minute;100/hour;200/day", deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
@is_not_permabanned
|
|
def edit_post(pid, v):
|
|
p = get_post(pid)
|
|
if not v.can_edit(p): abort(403)
|
|
|
|
# Disable edits on things older than 1wk unless it's a draft or editor is a jannie
|
|
if time.time() - p.created_utc > 7*24*60*60 and not p.private \
|
|
and v.admin_level < PERMS["IGNORE_1WEEk_EDITING_LIMIT"] and v.id not in EXEMPT_FROM_1WEEK_EDITING_LIMIT:
|
|
abort(403, "You can't edit posts older than 1 week!")
|
|
|
|
title = sanitize_raw_title(request.values.get("title", ""))
|
|
body = sanitize_raw_body(request.values.get("body", ""), True)
|
|
|
|
if v.id == p.author_id:
|
|
if v.longpost and (len(body) < 280 or ' [](' in body or body.startswith('[](')):
|
|
abort(403, "You have to type more than 280 characters!")
|
|
elif v.bird and len(body) > 140:
|
|
abort(403, "You have to type less than 140 characters!")
|
|
|
|
if not title:
|
|
abort(400, "Please enter a better title!")
|
|
|
|
|
|
if not p.private:
|
|
notify_users = NOTIFY_USERS(f'{title} {body}', v, oldtext=f'{p.title} {p.body}', ghost=p.ghost, log_cost=p)
|
|
if notify_users:
|
|
cid, text = notif_comment2(p)
|
|
if notify_users == 'everyone':
|
|
alert_everyone(cid)
|
|
else:
|
|
for x in notify_users:
|
|
add_notif(cid, x, text, pushnotif_url=p.permalink)
|
|
|
|
|
|
if title != p.title:
|
|
title_html = filter_emojis_only(title, golden=False)
|
|
|
|
if v.id == p.author_id and v.marseyawarded and not marseyaward_title_regex.fullmatch(title_html):
|
|
abort(403, "You can only type marseys!")
|
|
|
|
if 'megathread' in title.lower() and 'megathread' not in p.title.lower():
|
|
p.new = True
|
|
|
|
p.title = title
|
|
p.title_html = title_html
|
|
|
|
body = process_files(request.files, v, body)
|
|
body = body.strip()[:POST_BODY_LENGTH_LIMIT(v)] # process_files() may be adding stuff to the body
|
|
|
|
if body != p.body:
|
|
body_html = sanitize(body, golden=False, limit_pings=100)
|
|
|
|
if v.id == p.author_id and v.marseyawarded and marseyaward_body_regex.search(body_html):
|
|
abort(403, "You can only type marseys!")
|
|
|
|
|
|
p.body = body
|
|
|
|
process_poll_options(v, p)
|
|
|
|
for text in [p.body, p.title, p.url]:
|
|
if execute_blackjack(v, p, text, 'post'): break
|
|
|
|
if len(body_html) > POST_BODY_HTML_LENGTH_LIMIT:
|
|
abort(400, "Post body_html too long!")
|
|
|
|
p.body_html = body_html
|
|
|
|
|
|
if not complies_with_chud(p):
|
|
abort(403, f'You have to include "{v.chud_phrase}" in your post!')
|
|
|
|
|
|
if v.id == p.author_id:
|
|
if int(time.time()) - p.created_utc > 60 * 3: p.edited_utc = int(time.time())
|
|
g.db.add(p)
|
|
else:
|
|
ma=ModAction(
|
|
kind="edit_post",
|
|
user_id=v.id,
|
|
target_post_id=p.id
|
|
)
|
|
g.db.add(ma)
|
|
|
|
return {"message": "Post edited successfully!"}
|