# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
#
from os.path import abspath, dirname, join
import tempfile
from traceback import extract_stack
from django.conf import settings
from django.test import TestCase
from django.test.utils import override_settings
from slugify import slugify
from librarian import WLURI


@override_settings(
    MEDIA_ROOT=tempfile.mkdtemp(prefix='djangotest_'),
    CATALOGUE_DONT_BUILD={'pdf', 'mobi', 'epub', 'txt', 'fb2', 'cover'},
    NO_SEARCH_INDEX=True,
    CELERY_TASK_ALWAYS_EAGER=True,
    CACHES={
        'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'},
    },
)
class WLTestCase(TestCase):
    """
        Generic base class for tests. Adds settings freeze and clears MEDIA_ROOT.
    """
    longMessage = True


class PersonStub:

    def __init__(self, first_names, last_name):
        self.first_names = first_names
        self.last_name = last_name

    def readable(self):
        return " ".join(self.first_names + (self.last_name,))


class BookInfoStub:
    _empty_fields = ['cover_url', 'variant_of']
    # allow single definition for multiple-value fields
    _salias = {
        'authors': 'author',
        'genres': 'genre',
        'epochs': 'epoch',
        'kinds': 'kind',
    }

    def __init__(self, **kwargs):
        self.__dict = kwargs

    def __setattr__(self, key, value):
        if not key.startswith('_'):
            self.__dict[key] = value
        return object.__setattr__(self, key, value)

    def __getattr__(self, key):
        try:
            return self.__dict[key]
        except KeyError as e:
            if key in self._empty_fields:
                return None
            if key in self._salias:
                return [getattr(self, self._salias[key])]
            raise AttributeError(e)

    def to_dict(self):
        return dict((key, str(value)) for key, value in self.__dict.items())


def info_args(title, language=None):
    """ generate some keywords for comfortable BookInfoCreation  """
    slug = str(slugify(title))
    if language is None:
        language = 'pol'
    return {
        'title': str(title),
        'url': WLURI(slug),
        'about': "http://wolnelektury.pl/example/URI/%s" % slug,
        'language': language,
    }


def get_fixture(path, app=None):
    if app is not None:
        mod_path = app.__file__
        f_path = join(dirname(abspath(mod_path)), 'tests/files', path)
    else:
        mod_path = extract_stack(limit=2)[0][0]
        f_path = join(dirname(abspath(mod_path)), 'files', path)
    return f_path


def get_mp3_length(path):
    return 60