forked from MarseyWorld/MarseyWorld
318 lines
8.9 KiB
Python
318 lines
8.9 KiB
Python
import gevent.monkey
|
|
gevent.monkey.patch_all()
|
|
|
|
from os import environ, path
|
|
import secrets
|
|
from flask import *
|
|
from flask_caching import Cache
|
|
from flask_limiter import Limiter
|
|
from flask_compress import Compress
|
|
|
|
|
|
from flaskext.markdown import Markdown
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.exc import OperationalError
|
|
from sqlalchemy.orm import sessionmaker, scoped_session, Query as _Query
|
|
from sqlalchemy import *
|
|
from sqlalchemy.pool import QueuePool
|
|
import requests
|
|
import redis
|
|
import gevent
|
|
|
|
from redis import ConnectionPool
|
|
|
|
from werkzeug.middleware.proxy_fix import ProxyFix
|
|
|
|
app = Flask(__name__,
|
|
template_folder='./templates',
|
|
static_folder='./static'
|
|
)
|
|
app.wsgi_app = ProxyFix(app.wsgi_app, x_for=3)
|
|
app.url_map.strict_slashes = False
|
|
|
|
app.config["SITE_NAME"]=environ.get("SITE_NAME", "Drama").strip()
|
|
|
|
app.config["SITE_COLOR"]=environ.get("SITE_COLOR", "805ad5").strip()
|
|
|
|
app.config["DRAMAPATH"]=environ.get("DRAMAPATH", path.dirname(path.realpath(__file__)))
|
|
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
app.config['DATABASE_URL'] = environ.get(
|
|
"DATABASE_CONNECTION_POOL_URL",
|
|
environ.get("DATABASE_URL"))
|
|
|
|
app.config['SQLALCHEMY_READ_URIS'] = [
|
|
environ.get("DATABASE_CONNECTION_READ_01_URL"),
|
|
environ.get("DATABASE_CONNECTION_READ_02_URL"),
|
|
environ.get("DATABASE_CONNECTION_READ_03_URL")
|
|
]
|
|
|
|
app.config['SECRET_KEY'] = environ.get('MASTER_KEY')
|
|
app.config["SERVER_NAME"] = environ.get("domain").strip()
|
|
|
|
app.config["SHORT_DOMAIN"]=environ.get("SHORT_DOMAIN","").strip()
|
|
app.config["SESSION_COOKIE_NAME"] = "session_drama"
|
|
app.config["VERSION"] = "1.0.0"
|
|
app.config['MAX_CONTENT_LENGTH'] = 64 * 1024 * 1024
|
|
app.config["SESSION_COOKIE_SECURE"] = bool(int(environ.get("FORCE_HTTPS", 1)))
|
|
app.config["SESSION_COOKIE_SAMESITE"] = "Lax"
|
|
|
|
app.config["PERMANENT_SESSION_LIFETIME"] = 60 * 60 * 24 * 365
|
|
app.config["SESSION_REFRESH_EACH_REQUEST"] = True
|
|
|
|
app.config["FORCE_HTTPS"] = int(environ.get("FORCE_HTTPS", 1)) if ("localhost" not in app.config["SERVER_NAME"] and "127.0.0.1" not in app.config["SERVER_NAME"]) else 0
|
|
app.config["DISABLE_SIGNUPS"]=int(environ.get("DISABLE_SIGNUPS",0))
|
|
|
|
app.jinja_env.cache = {}
|
|
|
|
app.config["UserAgent"] = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36"
|
|
|
|
if "localhost" in app.config["SERVER_NAME"]:
|
|
app.config["CACHE_TYPE"] = "null"
|
|
else:
|
|
app.config["CACHE_TYPE"] = environ.get("CACHE_TYPE", 'filesystem').strip()
|
|
|
|
app.config["CACHE_DIR"] = environ.get("CACHE_DIR", "dramacache")
|
|
|
|
# captcha configs
|
|
app.config["HCAPTCHA_SITEKEY"] = environ.get("HCAPTCHA_SITEKEY","").strip()
|
|
app.config["HCAPTCHA_SECRET"] = environ.get(
|
|
"HCAPTCHA_SECRET","").strip()
|
|
app.config["SIGNUP_HOURLY_LIMIT"]=int(environ.get("SIGNUP_HOURLY_LIMIT",0))
|
|
|
|
# antispam configs
|
|
app.config["SPAM_SIMILARITY_THRESHOLD"] = float(
|
|
environ.get("SPAM_SIMILARITY_THRESHOLD", 0.5))
|
|
app.config["SPAM_SIMILAR_COUNT_THRESHOLD"] = int(
|
|
environ.get("SPAM_SIMILAR_COUNT_THRESHOLD", 5))
|
|
app.config["SPAM_URL_SIMILARITY_THRESHOLD"] = float(
|
|
environ.get("SPAM_URL_SIMILARITY_THRESHOLD", 0.1))
|
|
app.config["COMMENT_SPAM_SIMILAR_THRESHOLD"] = float(
|
|
environ.get("COMMENT_SPAM_SIMILAR_THRESHOLD", 0.5))
|
|
app.config["COMMENT_SPAM_COUNT_THRESHOLD"] = int(
|
|
environ.get("COMMENT_SPAM_COUNT_THRESHOLD", 5))
|
|
|
|
app.config["CACHE_REDIS_URL"] = environ.get(
|
|
"REDIS_URL").strip().lstrip() if environ.get("REDIS_URL") else None
|
|
app.config["CACHE_DEFAULT_TIMEOUT"] = 60
|
|
app.config["CACHE_KEY_PREFIX"] = "flask_caching_"
|
|
|
|
app.config["REDIS_POOL_SIZE"]=int(environ.get("REDIS_POOL_SIZE", 10))
|
|
|
|
redispool=ConnectionPool(
|
|
max_connections=app.config["REDIS_POOL_SIZE"],
|
|
host=app.config["CACHE_REDIS_URL"][8:]
|
|
) if app.config["CACHE_TYPE"]=="redis" else None
|
|
app.config["CACHE_OPTIONS"]={'connection_pool':redispool} if app.config["CACHE_TYPE"]=="redis" else {}
|
|
|
|
app.config["READ_ONLY"]=bool(int(environ.get("READ_ONLY", False)))
|
|
app.config["BOT_DISABLE"]=bool(int(environ.get("BOT_DISABLE", False)))
|
|
|
|
app.config["TENOR_KEY"]=environ.get("TENOR_KEY",'').strip()
|
|
|
|
|
|
Markdown(app)
|
|
cache = Cache(app)
|
|
Compress(app)
|
|
|
|
app.config["RATELIMIT_STORAGE_URL"] = environ.get("REDIS_URL").strip() if environ.get("REDIS_URL") else 'memory://'
|
|
app.config["RATELIMIT_KEY_PREFIX"] = "flask_limiting_"
|
|
app.config["RATELIMIT_ENABLED"] = True
|
|
app.config["RATELIMIT_DEFAULTS_DEDUCT_WHEN"]=lambda:True
|
|
app.config["RATELIMIT_DEFAULTS_EXEMPT_WHEN"]=lambda:False
|
|
app.config["RATELIMIT_HEADERS_ENABLED"]=True
|
|
|
|
#app.config["DISABLESIGNUPS"] = bool(int(environ.get("DISABLESIGNUPS", "0")))
|
|
|
|
|
|
def limiter_key_func():
|
|
return request.remote_addr
|
|
|
|
|
|
limiter = Limiter(
|
|
app,
|
|
key_func=limiter_key_func,
|
|
default_limits=["100/minute"],
|
|
headers_enabled=True,
|
|
strategy="fixed-window"
|
|
)
|
|
|
|
_engine=create_engine(
|
|
app.config['DATABASE_URL'],
|
|
poolclass=QueuePool,
|
|
pool_size=int(environ.get("PG_POOL_SIZE",10)),
|
|
pool_use_lifo=True
|
|
)
|
|
|
|
def retry(f):
|
|
|
|
def wrapper(self, *args, **kwargs):
|
|
try:
|
|
return f(self, *args, **kwargs)
|
|
except OperationalError as e:
|
|
#self.session.rollback()
|
|
#raise(DatabaseOverload)
|
|
print("sex")
|
|
except:
|
|
self.session.rollback()
|
|
return f(self, *args, **kwargs)
|
|
|
|
wrapper.__name__=f.__name__
|
|
return wrapper
|
|
|
|
|
|
class RetryingQuery(_Query):
|
|
|
|
@retry
|
|
def all(self):
|
|
return super().all()
|
|
|
|
@retry
|
|
def count(self):
|
|
return super().count()
|
|
|
|
@retry
|
|
def first(self):
|
|
return super().first()
|
|
|
|
db_session=scoped_session(sessionmaker(bind=_engine, query_cls=RetryingQuery))
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
#set the shared redis cache for misc stuff
|
|
|
|
r=redis.Redis(
|
|
host=app.config["CACHE_REDIS_URL"][8:],
|
|
decode_responses=True,
|
|
ssl_cert_reqs=None,
|
|
connection_pool=redispool
|
|
) if app.config["CACHE_REDIS_URL"] else None
|
|
|
|
local_ban_cache={}
|
|
|
|
UA_BAN_CACHE_TTL = int(environ.get("UA_BAN_CACHE_TTL", 3600))
|
|
|
|
|
|
|
|
# import and bind all routing functions
|
|
import drama.classes
|
|
from drama.routes import *
|
|
import drama.helpers.jinja2
|
|
|
|
@cache.memoize(UA_BAN_CACHE_TTL)
|
|
def get_useragent_ban_response(user_agent_str):
|
|
"""
|
|
Given a user agent string, returns a tuple in the form of:
|
|
(is_user_agent_banned, (insult, status_code))
|
|
"""
|
|
#if request.path.startswith("/socket.io/"):
|
|
# return False, (None, None)
|
|
|
|
result = g.db.query(
|
|
drama.classes.Agent).filter(
|
|
drama.classes.Agent.kwd.in_(
|
|
user_agent_str.split())).first()
|
|
if result:
|
|
return True, (result.mock or "Follow the robots.txt, dumbass",
|
|
result.status_code or 418)
|
|
return False, (None, None)
|
|
|
|
def drop_connection():
|
|
|
|
g.db.close()
|
|
gevent.getcurrent().kill()
|
|
|
|
|
|
# enforce https
|
|
@app.before_request
|
|
def before_request():
|
|
|
|
if request.method.lower() != "get" and app.config["READ_ONLY"]:
|
|
return {"error":f"{app.config['SITE_NAME']} is currently in read-only mode."}, 500
|
|
|
|
if app.config["BOT_DISABLE"] and request.headers.get("X-User-Type")=="Bot":
|
|
abort(503)
|
|
|
|
g.db = db_session()
|
|
|
|
g.timestamp = int(time.time())
|
|
|
|
session.permanent = True
|
|
|
|
ua_banned, response_tuple = get_useragent_ban_response(
|
|
request.headers.get("User-Agent", "NoAgent"))
|
|
if ua_banned and request.path != "/robots.txt":
|
|
return response_tuple
|
|
|
|
if app.config["FORCE_HTTPS"] and request.url.startswith(
|
|
"http://") and "localhost" not in app.config["SERVER_NAME"]:
|
|
url = request.url.replace("http://", "https://", 1)
|
|
return redirect(url, code=301)
|
|
|
|
if not session.get("session_id"):
|
|
session["session_id"] = secrets.token_hex(16)
|
|
|
|
ua=request.headers.get("User-Agent","")
|
|
if "CriOS/" in ua:
|
|
g.system="ios/chrome"
|
|
elif "Version/" in ua:
|
|
g.system="android/webview"
|
|
elif "Mobile Safari/" in ua:
|
|
g.system="android/chrome"
|
|
elif "Safari/" in ua:
|
|
g.system="ios/safari"
|
|
elif "Mobile/" in ua:
|
|
g.system="ios/webview"
|
|
else:
|
|
g.system="other/other"
|
|
|
|
|
|
def log_event(name, link):
|
|
|
|
x = requests.get(link)
|
|
|
|
if x.status_code != 200:
|
|
return
|
|
|
|
text = f'> **{name}**\r> {link}'
|
|
|
|
url = environ.get("DISCORD_WEBHOOK")
|
|
headers = {"Content-Type": "application/json"}
|
|
data = {"username": "drama",
|
|
"content": text
|
|
}
|
|
|
|
x = requests.post(url, headers=headers, json=data)
|
|
print(x.status_code)
|
|
|
|
|
|
@app.after_request
|
|
def after_request(response):
|
|
|
|
try:
|
|
g.db.commit()
|
|
g.db.close()
|
|
except Exception as e:
|
|
g.db.rollback()
|
|
g.db.close()
|
|
print(e)
|
|
abort(500)
|
|
|
|
response.headers.add('Access-Control-Allow-Headers', "Origin, X-Requested-With, Content-Type, Accept, x-auth")
|
|
response.headers.remove("Cache-Control")
|
|
response.headers.add("Cache-Control", "public, max-age=31536000")
|
|
response.headers.add("Access-Control-Allow-Origin", app.config["SERVER_NAME"])
|
|
|
|
response.headers.add("Strict-Transport-Security", "max-age=31536000")
|
|
response.headers.add("Referrer-Policy", "same-origin")
|
|
response.headers.add("Feature-Policy", "geolocation 'none'; midi 'none'; notifications 'none'; push 'none'; sync-xhr 'none'; microphone 'none'; camera 'none'; magnetometer 'none'; gyroscope 'none'; vibrate 'none'; fullscreen 'none'; payment 'none';")
|
|
if not request.path.startswith("/embed/"): response.headers.add("X-Frame-Options", "deny")
|
|
|
|
return response
|
|
|
|
|
|
@app.route("/<path:path>", subdomain="www")
|
|
def www_redirect(path):
|
|
|
|
return redirect(f"https://{app.config['SERVER_NAME']}/{path}") |