# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
#
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    help = 'Checks for dead links.'

    def handle(self, **options):
        from catalogue.models import Book
        from urllib.request import urlopen, HTTPError, URLError
        from django.urls import reverse
        from django.contrib.sites.models import Site

        domain = Site.objects.get_current().domain

        fields = [
            (
                Book,
                [
                    ('wiki_link', lambda b: b.wiki_link),
                    ('źródło', lambda b: b.get_extra_info_json().get('source_url')),
                ],
                'admin:catalogue_book_change'
            ),
        ]

        for model, model_fields, admin_name in fields:
            for obj in model.objects.all():
                clean = True
                for name, get in model_fields:
                    url = get(obj)
                    if url:
                        try:
                            urlopen(url)
                        except (HTTPError, URLError, ValueError) as e:
                            if clean:
                                clean = False
                                print(str(obj).encode('utf-8'))
                                print(('Na stronie: https://%s%s' % (domain, obj.get_absolute_url())).encode('utf-8'))
                                print(
                                    ('Administracja: https://%s%s' % (domain, reverse(admin_name, args=[obj.pk])))
                                    .encode('utf-8'))
                                if obj.get_extra_info_json().get('about'):
                                    print(('Redakcja: %s' % (obj.get_extra_info_json().get('about'),)).encode('utf-8'))
                            print(('    %s (%s): %s' % (name, getattr(e, 'code', 'błąd'), url)).encode('utf-8'))
                if not clean:
                    print()