microblog.pub/config.py

import subprocess
import os
import yaml
from pymongo import MongoClient
import requests
from itsdangerous import JSONWebSignatureSerializer
from datetime import datetime

from little_boxes.utils import strtobool
from utils.key import KEY_DIR, get_key, 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('utf-8')  # type: ignore


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(),
}