bookwyrm/bookwyrm/wellknown.py

''' responds to various requests to /.well-know '''

from dateutil.relativedelta import relativedelta
from django.http import HttpResponseNotFound
from django.http import JsonResponse
from django.utils import timezone

from bookwyrm import models
from bookwyrm.settings import DOMAIN, VERSION


def webfinger(request):
    ''' allow other servers to ask about a user '''
    if request.method != 'GET':
        return HttpResponseNotFound()

    resource = request.GET.get('resource')
    if not resource and not resource.startswith('acct:'):
        return HttpResponseNotFound()

    username = resource.replace('acct:', '')
    try:
        user = models.User.objects.get(username=username)
    except models.User.DoesNotExist:
        return HttpResponseNotFound('No account found')

    return JsonResponse({
        'subject': 'acct:%s' % (user.username),
        'links': [
            {
                'rel': 'self',
                'type': 'application/activity+json',
                'href': user.remote_id
            }
        ]
    })


def nodeinfo_pointer(request):
    ''' direct servers to nodeinfo '''
    if request.method != 'GET':
        return HttpResponseNotFound()

    return JsonResponse({
        'links': [
            {
                'rel': 'http://nodeinfo.diaspora.software/ns/schema/2.0',
                'href': 'https://%s/nodeinfo/2.0' % DOMAIN
            }
        ]
    })


def nodeinfo(request):
    ''' basic info about the server '''
    if request.method != 'GET':
        return HttpResponseNotFound()

    status_count = models.Status.objects.filter(user__local=True).count()
    user_count = models.User.objects.filter(local=True).count()

    month_ago = timezone.now() - relativedelta(months=1)
    last_month_count = models.User.objects.filter(
        local=True,
        last_active_date__gt=month_ago
    ).count()

    six_months_ago = timezone.now() - relativedelta(months=6)
    six_month_count = models.User.objects.filter(
        local=True,
        last_active_date__gt=six_months_ago
    ).count()

    site = models.SiteSettings.get()
    return JsonResponse({
        'version': '2.0',
        'software': {
            'name': 'bookwyrm',
            'version': VERSION
        },
        'protocols': [
            'activitypub'
        ],
        'usage': {
            'users': {
                'total': user_count,
                'activeMonth': last_month_count,
                'activeHalfyear': six_month_count,
            },
            'localPosts': status_count,
        },
        'openRegistrations': site.allow_registration,
    })


def instance_info(request):
    ''' let's talk about your cool unique instance '''
    if request.method != 'GET':
        return HttpResponseNotFound()

    user_count = models.User.objects.filter(local=True).count()
    status_count = models.Status.objects.filter(user__local=True).count()

    site = models.SiteSettings.get()
    return JsonResponse({
        'uri': DOMAIN,
        'title': site.name,
        'short_description': '',
        'description': site.instance_description,
        'version': '0.0.1',
        'stats': {
            'user_count': user_count,
            'status_count': status_count,
        },
        'thumbnail': 'https://%s/static/images/logo.png' % DOMAIN,
        'languages': [
            'en'
        ],
        'registrations': site.allow_registration,
        'approval_required': False,
    })


def peers(request):
    ''' list of federated servers this instance connects with '''
    if request.method != 'GET':
        return HttpResponseNotFound()

    names = models.FederatedServer.objects.values_list('server_name', flat=True)
    return JsonResponse(list(names), safe=False)