# Copyright (C) 2005  Michael Urman
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.

import warnings

from calibre_plugins.AudioM3U.mutagen._util import DictMixin, loadfile


class FileType(DictMixin):
    """FileType(filething, **kwargs)

    Args:
        filething (filething): A filename or a file-like object

    Subclasses might take further options via keyword arguments.

    An abstract object wrapping tags and audio stream information.

    Each file format has different potential tags and stream
    information.

    FileTypes implement an interface very similar to Metadata; the
    dict interface, save, load, and delete calls on a FileType call
    the appropriate methods on its tag data.

    Attributes:
        info (`StreamInfo`): contains length, bitrate, sample rate
        tags (`Tags`): metadata tags, if any, otherwise `None`
    """

    __module__ = "mutagen"

    info = None
    tags = None
    filename = None
    _mimes = ["application/octet-stream"]

    def __init__(self, *args, **kwargs):
        if not args and not kwargs:
            warnings.warn("FileType constructor requires a filename",
                          DeprecationWarning)
        else:
            self.load(*args, **kwargs)

    @loadfile()
    def load(self, filething, *args, **kwargs):
        raise NotImplementedError

    def __getitem__(self, key):
        """Look up a metadata tag key.

        If the file has no tags at all, a KeyError is raised.
        """

        if self.tags is None:
            raise KeyError(key)
        else:
            return self.tags[key]

    def __setitem__(self, key, value):
        """Set a metadata tag.

        If the file has no tags, an appropriate format is added (but
        not written until save is called).
        """

        if self.tags is None:
            self.add_tags()
        self.tags[key] = value

    def __delitem__(self, key):
        """Delete a metadata tag key.

        If the file has no tags at all, a KeyError is raised.
        """

        if self.tags is None:
            raise KeyError(key)
        else:
            del self.tags[key]

    def keys(self):
        """Return a list of keys in the metadata tag.

        If the file has no tags at all, an empty list is returned.
        """

        if self.tags is None:
            return []
        else:
            return self.tags.keys()

    @loadfile(writable=True)
    def delete(self, filething=None):
        """delete(filething=None)

        Remove tags from a file.

        In cases where the tagging format is independent of the file type
        (for example `mutagen.id3.ID3`) all traces of the tagging format will
        be removed.
        In cases where the tag is part of the file type, all tags and
        padding will be removed.

        The tags attribute will be cleared as well if there is one.

        Does nothing if the file has no tags.

        Raises:
            mutagen.MutagenError: if deleting wasn't possible
        """

        if self.tags is not None:
            return self.tags.delete(filething)

    @loadfile(writable=True)
    def save(self, filething=None, **kwargs):
        """save(filething=None, **kwargs)

        Save metadata tags.

        Raises:
            MutagenError: if saving wasn't possible
        """

        if self.tags is not None:
            return self.tags.save(filething, **kwargs)

    def pprint(self):
        """
        Returns:
            text: stream information and comment key=value pairs.
        """

        stream = "%s (%s)" % (self.info.pprint(), self.mime[0])
        try:
            tags = self.tags.pprint()
        except AttributeError:
            return stream
        else:
            return stream + ((tags and "\n" + tags) or "")

    def add_tags(self):
        """Adds new tags to the file.

        Raises:
            mutagen.MutagenError:
                if tags already exist or adding is not possible.
        """

        raise NotImplementedError

    @property
    def mime(self):
        """A list of mime types (:class:`mutagen.text`)"""

        mimes = []
        for Kind in type(self).__mro__:
            for mime in getattr(Kind, '_mimes', []):
                if mime not in mimes:
                    mimes.append(mime)
        return mimes

    @staticmethod
    def score(filename, fileobj, header):
        """Returns a score for how likely the file can be parsed by this type.

        Args:
            filename (fspath): a file path
            fileobj (fileobj): a file object open in rb mode. Position is
                undefined
            header (bytes): data of undefined length, starts with the start of
                the file.

        Returns:
            int: negative if definitely not a matching type, otherwise a score,
                the bigger the more certain that the file can be loaded.
        """

        raise NotImplementedError


class StreamInfo(object):
    """Abstract stream information object.

    Provides attributes for length, bitrate, sample rate etc.

    See the implementations for details.
    """

    __module__ = "mutagen"

    def pprint(self):
        """
        Returns:
            text: Print stream information
        """

        raise NotImplementedError


@loadfile(method=False)
def File(filething, options=None, easy=False):
    """File(filething, options=None, easy=False)

    Guess the type of the file and try to open it.

    The file type is decided by several things, such as the first 128
    bytes (which usually contains a file type identifier), the
    filename extension, and the presence of existing tags.

    If no appropriate type could be found, None is returned.

    Args:
        filething (filething)
        options: Sequence of :class:`FileType` implementations,
            defaults to all included ones.
        easy (bool):  If the easy wrappers should be returned if available.
            For example :class:`EasyMP3 <mp3.EasyMP3>` instead of
            :class:`MP3 <mp3.MP3>`.

    Returns:
        FileType: A FileType instance for the detected type or `None` in case
            the type couldn't be determined.

    Raises:
        MutagenError: in case the detected type fails to load the file.
    """

    if options is None:
        from calibre_plugins.AudioM3U.mutagen.asf import ASF
        from calibre_plugins.AudioM3U.mutagen.apev2 import APEv2File
        from calibre_plugins.AudioM3U.mutagen.flac import FLAC
        if easy:
            from calibre_plugins.AudioM3U.mutagen.easyid3 import EasyID3FileType as ID3FileType
        else:
            from calibre_plugins.AudioM3U.mutagen.id3 import ID3FileType
        if easy:
            from calibre_plugins.AudioM3U.mutagen.mp3 import EasyMP3 as MP3
        else:
            from calibre_plugins.AudioM3U.mutagen.mp3 import MP3
        from calibre_plugins.AudioM3U.mutagen.oggflac import OggFLAC
        from calibre_plugins.AudioM3U.mutagen.oggspeex import OggSpeex
        from calibre_plugins.AudioM3U.mutagen.oggtheora import OggTheora
        from calibre_plugins.AudioM3U.mutagen.oggvorbis import OggVorbis
        from calibre_plugins.AudioM3U.mutagen.oggopus import OggOpus
        if easy:
            from calibre_plugins.AudioM3U.mutagen.trueaudio import EasyTrueAudio as TrueAudio
        else:
            from calibre_plugins.AudioM3U.mutagen.trueaudio import TrueAudio
        from calibre_plugins.AudioM3U.mutagen.wavpack import WavPack
        if easy:
            from calibre_plugins.AudioM3U.mutagen.easymp4 import EasyMP4 as MP4
        else:
            from calibre_plugins.AudioM3U.mutagen.mp4 import MP4
        from calibre_plugins.AudioM3U.mutagen.musepack import Musepack
        from calibre_plugins.AudioM3U.mutagen.monkeysaudio import MonkeysAudio
        from calibre_plugins.AudioM3U.mutagen.optimfrog import OptimFROG
        from calibre_plugins.AudioM3U.mutagen.aiff import AIFF
        from calibre_plugins.AudioM3U.mutagen.aac import AAC
        from calibre_plugins.AudioM3U.mutagen.ac3 import AC3
        from calibre_plugins.AudioM3U.mutagen.smf import SMF
        from calibre_plugins.AudioM3U.mutagen.tak import TAK
        from calibre_plugins.AudioM3U.mutagen.dsf import DSF
        from calibre_plugins.AudioM3U.mutagen.dsdiff import DSDIFF
        from calibre_plugins.AudioM3U.mutagen.wave import WAVE
        options = [MP3, TrueAudio, OggTheora, OggSpeex, OggVorbis, OggFLAC,
                   FLAC, AIFF, APEv2File, MP4, ID3FileType, WavPack,
                   Musepack, MonkeysAudio, OptimFROG, ASF, OggOpus, AAC, AC3,
                   SMF, TAK, DSF, DSDIFF, WAVE]

    if not options:
        return None

    fileobj = filething.fileobj

    try:
        header = fileobj.read(128)
    except IOError:
        header = b""

    # Sort by name after score. Otherwise import order affects
    # Kind sort order, which affects treatment of things with
    # equals scores.
    results = [(Kind.score(filething.name, fileobj, header), Kind.__name__)
               for Kind in options]

    results = list(zip(results, options))
    results.sort()
    (score, name), Kind = results[-1]
    if score > 0:
        try:
            fileobj.seek(0, 0)
        except IOError:
            pass
        return Kind(fileobj, filename=filething.filename)
    else:
        return None
