110 lines
2.7 KiB
Python
110 lines
2.7 KiB
Python
import os
|
|
import subprocess
|
|
from datetime import datetime
|
|
|
|
import requests
|
|
import yaml
|
|
from itsdangerous import JSONWebSignatureSerializer
|
|
from pymongo import MongoClient
|
|
|
|
from little_boxes import strtobool
|
|
from utils.key import KEY_DIR
|
|
from utils.key import get_key
|
|
from utils.key import get_secret_key
|
|
|
|
|
|
def noop():
|
|
pass
|
|
|
|
|
|
CUSTOM_CACHE_HOOKS = False
|
|
try:
|
|
from cache_hooks import purge as custom_cache_purge_hook
|
|
except ModuleNotFoundError:
|
|
custom_cache_purge_hook = noop
|
|
|
|
VERSION = (
|
|
subprocess.check_output(["git", "describe", "--always"]).split()[0].decode("utf-8")
|
|
)
|
|
|
|
DEBUG_MODE = strtobool(os.getenv("MICROBLOGPUB_DEBUG", "false"))
|
|
|
|
|
|
CTX_AS = "https://www.w3.org/ns/activitystreams"
|
|
CTX_SECURITY = "https://w3id.org/security/v1"
|
|
AS_PUBLIC = "https://www.w3.org/ns/activitystreams#Public"
|
|
HEADERS = [
|
|
"application/activity+json",
|
|
"application/ld+json;profile=https://www.w3.org/ns/activitystreams",
|
|
'application/ld+json; profile="https://www.w3.org/ns/activitystreams"',
|
|
"application/ld+json",
|
|
]
|
|
|
|
|
|
with open(os.path.join(KEY_DIR, "me.yml")) as f:
|
|
conf = yaml.load(f)
|
|
|
|
USERNAME = conf["username"]
|
|
NAME = conf["name"]
|
|
DOMAIN = conf["domain"]
|
|
SCHEME = "https" if conf.get("https", True) else "http"
|
|
BASE_URL = SCHEME + "://" + DOMAIN
|
|
ID = BASE_URL
|
|
SUMMARY = conf["summary"]
|
|
ICON_URL = conf["icon_url"]
|
|
PASS = conf["pass"]
|
|
PUBLIC_INSTANCES = conf.get("public_instances", [])
|
|
# TODO(tsileo): choose dark/light style
|
|
THEME_COLOR = conf.get("theme_color")
|
|
|
|
USER_AGENT = (
|
|
f"{requests.utils.default_user_agent()} " f"(microblog.pub/{VERSION}; +{BASE_URL})"
|
|
)
|
|
|
|
mongo_client = MongoClient(
|
|
host=[os.getenv("MICROBLOGPUB_MONGODB_HOST", "localhost:27017")]
|
|
)
|
|
|
|
DB_NAME = "{}_{}".format(USERNAME, DOMAIN.replace(".", "_"))
|
|
DB = mongo_client[DB_NAME]
|
|
|
|
|
|
def _drop_db():
|
|
if not DEBUG_MODE:
|
|
return
|
|
|
|
mongo_client.drop_database(DB_NAME)
|
|
|
|
|
|
KEY = get_key(ID, USERNAME, DOMAIN)
|
|
|
|
|
|
JWT_SECRET = get_secret_key("jwt")
|
|
JWT = JSONWebSignatureSerializer(JWT_SECRET)
|
|
|
|
|
|
def _admin_jwt_token() -> str:
|
|
return JWT.dumps({"me": "ADMIN", "ts": datetime.now().timestamp()}).decode( # type: ignore
|
|
"utf-8"
|
|
)
|
|
|
|
|
|
ADMIN_API_KEY = get_secret_key("admin_api_key", _admin_jwt_token)
|
|
|
|
ME = {
|
|
"@context": [CTX_AS, CTX_SECURITY],
|
|
"type": "Person",
|
|
"id": ID,
|
|
"following": ID + "/following",
|
|
"followers": ID + "/followers",
|
|
"liked": ID + "/liked",
|
|
"inbox": ID + "/inbox",
|
|
"outbox": ID + "/outbox",
|
|
"preferredUsername": USERNAME,
|
|
"name": NAME,
|
|
"summary": SUMMARY,
|
|
"endpoints": {},
|
|
"url": ID,
|
|
"icon": {"mediaType": "image/png", "type": "Image", "url": ICON_URL},
|
|
"publicKey": KEY.to_dict(),
|
|
}
|