#   mutagengui.py: GTK based file tagger.
#   Copyright (C) 2009-2020 Stephen Fairchild (s-fairchild@users.sourceforge.net)
#
#   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.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program in the file entitled COPYING.
#   If not, see <http://www.gnu.org/licenses/>.

__all__ = ['MutagenGUI']

import os
import sys
import string
import re
import gettext

from gi.repository import Gtk
from gi.repository import Pango
from gi.repository import GLib
import mutagen
import mutagen.id3 as id3
from mutagen.mp3 import MP3
from mutagen.apev2 import APEv2, APETextValue
from mutagen.musepack import Musepack
from mutagen.monkeysaudio import MonkeysAudio
from mutagen.asf import ASF, ASFUnicodeAttribute

from idjc import FGlobs
from .tooltips import set_tip
from idjc.prelims import ProfileManager

t = gettext.translation(FGlobs.package_name, FGlobs.localedir, fallback=True)
_ = t.gettext

pm = ProfileManager()

def write_error_dialog(error, window):
    dialog = Gtk.Dialog(title=_('Tag Write Failed'))
    dialog.set_modal(True)
    dialog.set_transient_for(window)
    dialog.set_border_width(8)
    dialog.set_size_request(300, 60)
    content_vbox = dialog.get_content_area()
    content_vbox.set_spacing(10)
    label = Gtk.Label(str(error))
    label.set_max_width_chars(40)
    label.set_line_wrap(True)
    label.set_single_line_mode(False)
    label.set_selectable(True)
    content_vbox.add(label)
    label.show()
    dialog.add_button(_("OK"), Gtk.ResponseType.OK)
    dialog.run()
    dialog.destroy()


def mono_label(text):
    label = Gtk.Label.new(f"<span font-family='monospace'>{text}</span>")
    label.set_use_markup(True)
    label.set_margin_top(2)
    return label


def right_label(text):
    label = Gtk.Label.new(text)
    label.set_xalign(1.0)
    return label


class FreeTagFrame(Gtk.Frame):
    def __init__(self):
        Gtk.Frame.__init__(self)
        sw = Gtk.ScrolledWindow()
        sw.set_border_width(5)
        sw.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.ALWAYS)
        self.add(sw)
        sw.show()

        text_view = Gtk.TextView()
        text_view.set_wrap_mode(Gtk.WrapMode.CHAR)
        self.text_buffer = text_view.get_buffer()
        text_tag = self.text_buffer.create_tag("font", font="monospace 11")
        sw.add(text_view)
        text_view.show()
        self.text_buffer.connect("changed", self._on_buffer_changed)

    def _on_buffer_changed(self, text_buffer):
        """Keep all text in the same font."""

        text_buffer.apply_tag_by_name("font", *text_buffer.get_bounds())


class MutagenTagger(Gtk.VBox):
    """Base class for ID3Tagger and NativeTagger."""

    def __init__(self, pathname):
        Gtk.VBox.__init__(self)
        self.pathname = pathname


class WMATagger(MutagenTagger):
    """Handles tagging of WMA files"""

    primary_data = ("Title", "Author")
    secondaries = ("WM/AlbumTitle", "WM/AlbumArtist", "WM/Year", "WM/Genre")

    def save_tag(self, window):
        """Updates the tag with the GUI data."""

        tag = self.tag
        tb = self.tag_frame.text_buffer

        for key in self.text_set:
            try:
                del tag[key]
            except KeyError:
                pass

        for each in self.primary_line:
            val = each[1].get_text().strip()
            if val:
                tag[each[0]] = val
            else:
                try:
                    del tag[each[0]]
                except KeyError:
                    pass

        lines = tb.get_text(tb.get_start_iter(), tb.get_end_iter(), False).splitlines()
        for line in lines:
            try:
                key, val = line.split("=", 1)
            except ValueError:
                continue
            else:
                key = key.strip()
                val = val.strip()
                if val:
                    try:
                        tag[key] += [ASFUnicodeAttribute(val)]
                    except (KeyError, AttributeError):
                        try:
                            tag[key] = [
                                    ASFUnicodeAttribute(val)]
                        except KeyError:
                            print("Unacceptable key", key)
        try:
            tag.save()
        except mutagen.MutagenError as e:
            write_error_dialog(e, window)

    def load_tag(self):
        """(re)Writes the tag data to the GUI."""

        tag = self.tag

        for each in self.primary_line:
            try:
                data = tag[each[0]]
            except KeyError:
                pass
            else:
                each[1].set_text("/".join(y.value for y in data))

        additional = []

        for key in self.secondaries:
            values = tag.get(key, [ASFUnicodeAttribute("")])
            for val in values:
                additional.append(f"{key}={val}")

        for key in self.text_set:
            if key not in self.primary_data and key not in self.secondaries:
                values = tag[key]
                for val in values:
                    additional.append(f"{key}={val}")

        self.tag_frame.text_buffer.set_text("\n".join(additional))

    def __init__(self, pathname):
        MutagenTagger.__init__(self, pathname)
        try:
            self.tag = mutagen.asf.ASF(pathname)
            if not isinstance(self.tag, mutagen.asf.ASF):
                raise mutagen.asf.error
        except mutagen.asf.error:
            print("Not a real wma/asf file apparently.")
            self.tag = None
            return

        grid = Gtk.Grid()
        grid.set_border_width(5)
        grid.set_column_spacing(8)
        grid.set_row_homogeneous(True)
        self.add(grid)

        self.primary_line = []
        for row, (text, entry) in enumerate(
                            (x, Gtk.Entry()) for x in self.primary_data):
            self.primary_line.append((text, entry))
            entry.set_hexpand(True)
            grid.attach(right_label(text), 0, row, 1, 1)
            grid.attach(entry, 1, row, 1, 1)
        grid.show_all()

        self.tag_frame = FreeTagFrame()
        self.tag_frame.set_border_width(5)
        self.add(self.tag_frame)
        self.tag_frame.show()

        self.text_set = []

        for key, val in self.tag.items():
            if key not in self.primary_line and all(isinstance(v, (
                                ASFUnicodeAttribute, str)) for v in val):
                self.text_set.append(key)



class ID3Tagger(MutagenTagger):
    """ID3 tagging with Mutagen."""

    primary_data = (("TIT2", _('title')), ("TPE1", _('artist')),
                         ("TALB", _('album')), ("TRCK", _('track/total')),
                         ("TCON", _('genre')), ("TDRC", _('record date')))

    def save_tag(self, window):
        """Updates the tag with the GUI data."""

        tag = self.tag

        # Remove all text tags.
        for fid in [fid for fid in tag if fid[0] == 'T']:
            del tag[fid]

        # Add the primary tags.
        for fid, entry in self.primary_line:
            text = entry.get_text().strip()
            if text:
                frame = getattr(id3, fid)
                tag[fid] = frame(3, [text])

        # Add the freeform text tags.
        tb = self.tag_frame.text_buffer
        lines = tb.get_text(tb.get_start_iter(), tb.get_end_iter(), False).splitlines()

        for line in lines:
            try:
                fid, val = line.split(":", 1)

            except ValueError:
                continue

            fid = fid.strip()
            val = val.strip()

            try:
                frame = id3.Frames[fid]
            except NameError:
                continue

            if not issubclass(frame, id3.TextFrame):
                continue

            if frame is id3.TXXX:
                try:
                    key, val = val.split(u"=", 1)

                except ValueError:
                    continue

                f = frame(3, key.strip(), [val.strip()])
                tag[f.HashKey] = f

            else:
                try:
                    val_list = tag[fid].text
                except KeyError:
                    tag[fid] = frame(3, [val])
                else:
                    val_list.append(val)

        try:
            tag.save()
        except mutagen.MutagenError as e:
            write_error_dialog(e, window)


    def load_tag(self):
        """(re)Writes the tag data to the GUI."""

        additional = []
        done = []

        for fid, entry in self.primary_line:
            try:
                frame = self.tag[fid]
                if fid[0] == "T":
                    try:
                        entry.set_text(frame.text[0])
                    except TypeError:
                        # Handle occurrence of ID3Timestamp.
                        entry.set_text(str(frame.text[0]))
                    for each in frame.text[1:]:
                        additional.append(f"{fid}:{each}")
            except KeyError:
                entry.set_text("")

            done.append(fid)

        for fid, frame in self.tag.items():
            if fid[0] == "T" and fid not in done:
                sep = "=" if fid.startswith("TXXX:") else ":"
                for text in frame.text:
                    additional.append(f"{fid}{sep}"
                                      f"{text if type(text) is str else text.text}")

        self.tag_frame.text_buffer.set_text("\n".join(additional))

    def __init__(self, pathname, force=False):
        MutagenTagger.__init__(self, pathname)
        if force:
            try:
                self.tag = mutagen.File(pathname)
                if not isinstance(self.tag, MP3):
                    raise mutagen.mp3.error
            except mutagen.mp3.error:
                print("Not a real mp3 file apparently.")
                self.tag = None
                return
            try:
                self.tag.add_tags()
                print("Added ID3 tags to", pathname)
            except mutagen.id3.error:
                print("Existing ID3 tags found.")
        else:
            try:
                # Obtain ID3 tags from a non mp3 file.
                self.tag = mutagen.id3.ID3(pathname)
            except mutagen.id3.error:
                self.tag = None
                return

        grid = Gtk.Grid()
        grid.set_border_width(5)
        grid.set_column_spacing(8)
        self.pack_start(grid, False, False, 0)

        self.primary_line = []
        for i, (frame, text, entry) in enumerate(
                            (x, y, Gtk.Entry()) for x, y in self.primary_data):
            self.primary_line.append((frame, entry))
            grid.attach(mono_label(frame), 0, i, 1, 1)
            grid.attach(right_label(text), 1, i, 1, 1)
            grid.attach(entry, 2, i, 1, 1)
            entry.set_hexpand(True)
        grid.show_all()

        self.tag_frame = FreeTagFrame()
        self.tag_frame.set_vexpand(True)
        set_tip(self.tag_frame, _('Add any other ID3 text frames here.\ne.g. '
        'TIT2:Alternate Title\nThis will be appended onto the main TIT2 tag.'
        '\n\nEnter user defined text frames like this:\nTXXX:foo=bar\n\n'
        'For more information visit www.id3.org.'))
        self.tag_frame.set_border_width(5)
        # TC: Remaining textual ID3 data is show below this heading.
        self.tag_frame.set_label(_(' Additional Text Frames '))
        self.add(self.tag_frame)
        self.tag_frame.show()


class MP4Tagger(MutagenTagger):
    """MP4 tagging with Mutagen."""

    primary_data = (("\xa9nam", _('Title')), ("\xa9ART", _('Artist')),
                         ("\xa9alb", _('Album')), ("trkn", _('Track')),
                         ("\xa9gen", _('Genre')), ("\xa9day", _('Year')))

    def save_tag(self, window):
        """Updates the tag with the GUI data."""

        tag = self.tag
        for fid, entry in self.primary_line:
            text = entry.get_text().strip()
            if fid == "trkn":
                mo1 = re.search(r"\d+", text)
                try:
                    track = int(text[mo1.start():mo1.end()])
                except AttributeError:
                    new_val = None
                else:
                    text = text[mo1.end():]
                    mo2 = re.search(r"\d+", text)
                    try:
                        total = int(text[mo2.start():mo2.end()])
                    except AttributeError:
                        new_val = [(track, 0)]
                    else:
                        new_val = [(track, total)]
            else:
                new_val = [text] if text else None

            if new_val is not None:
                tag[fid] = new_val
            else:
                try:
                    del tag[fid]
                except KeyError:
                    pass

        try:
            tag.save()
        except mutagen.MutagenError as e:
            write_error_dialog(e, window)


    def load_tag(self):
        """(re)Writes the tag data to the GUI."""

        additional = []

        for fid, entry in self.primary_line:
            try:
                frame = self.tag[fid][0]
            except KeyError:
                entry.set_text("")
            else:
                if fid == "trkn":
                    if frame[1]:
                        entry.set_text("%d/%d" % frame)
                    else:
                        entry.set_text(str(frame[0]))
                else:
                    entry.set_text(frame)

    def __init__(self, pathname):
        MutagenTagger.__init__(self, pathname)
        try:
            self.tag = mutagen.mp4.MP4(pathname)
            if not isinstance(self.tag, mutagen.mp4.MP4):
                raise mutagen.mp4.error
        except mutagen.mp4.error:
            print("Not a real mp4 file apparently.")
            self.tag = None
            return

        grid = Gtk.Grid()
        grid.set_border_width(5)
        grid.set_column_spacing(8)
        grid.set_row_homogeneous(True)
        self.add(grid)

        self.primary_line = []
        for row, (frame, text, entry) in enumerate(
                            (x, y, Gtk.Entry()) for x, y in self.primary_data):
            self.primary_line.append((frame, entry))
            entry.set_hexpand(True)
            grid.attach(right_label(text), 0, row, 1, 1)
            grid.attach(entry, 1, row, 1, 1)
        grid.show_all()


class NativeTagger(MutagenTagger):
    """Native format tagging with Mutagen. Mostly FLAC and Ogg."""

    blacklist = "coverart", "metadata_block_picture"

    def save_tag(self, window):
        """Updates the tag with the GUI data."""

        tag = self.tag

        for key in [key for key in tag if key not in self.blacklist]:
            del tag[key]

        tb = self.tag_frame.text_buffer
        lines = tb.get_text(tb.get_start_iter(), tb.get_end_iter(), False).splitlines()

        for line in lines:
            try:
                key, val = line.split("=", 1)
            except ValueError:
                continue
            else:
                key = key.strip()
                val = val.strip()
                if key not in self.blacklist and val:
                    try:
                        tag[key] += [val]
                    except (KeyError, AttributeError):
                        try:
                            tag[key] = [val]
                        except KeyError:
                            print("Unacceptable key", key)

        try:
            tag.save()
        except mutagen.MutagenError as e:
            write_error_dialog(e, window)

    def load_tag(self):
        """(re)Writes the tag data to the GUI."""

        tag = self.tag
        lines = []
        primaries = "title", "artist", "author", "album",\
                                  "tracknumber", "tracktotal", "genre", "date"
        for key in primaries:
            try:
                values = tag[key]
            except KeyError:
                lines.append(f"{key}=")
            else:
                for val in values:
                    lines.append(f"{key}={val}")

        for key, values in tag.items():
            if key not in primaries and key not in self.blacklist:
                for val in values:
                    lines.append(f"{key}={val}")

        self.tag_frame.text_buffer.set_text("\n".join(lines))


    def __init__(self, pathname, ext):
        MutagenTagger.__init__(self, pathname)
        self.tag = mutagen.File(pathname)
        if isinstance(self.tag, (MP3, APEv2)):
            # MP3 and APEv2 have their own specialised tagger.
            self.tag = None
            return

        self.tag_frame = FreeTagFrame()
        self.tag_frame.set_vexpand(True)
        self.add(self.tag_frame)
        self.tag_frame.show()



class ApeTagger(MutagenTagger):
    """APEv2 tagging with Mutagen."""

    opener = {"ape": MonkeysAudio, "mpc": Musepack }

    def save_tag(self, window):
        """Updates the tag with the GUI data."""

        tag = self.tag

        for k in [k for k, v in tag.items() if isinstance(v, APETextValue)]:
            del tag[k]

        tb = self.tag_frame.text_buffer
        lines = tb.get_text(tb.get_start_iter(), tb.get_end_iter(), False).splitlines()

        for line in lines:
            try:
                key, val = line.split("=", 1)
            except ValueError:
                continue
            else:
                key = key.strip()
                val = val.strip()
                if val:
                    try:
                        tag[key].value += "\0" + val
                    except (KeyError, AttributeError):
                        try:
                            tag[key] = APETextValue(val, 0)
                        except KeyError:
                            print("Unacceptable key", key)

        try:
            tag.save()
        except mutagen.MutagenError as e:
            write_error_dialog(e, window)

    def load_tag(self):
        """(re)Writes the tag data to the GUI."""

        tag = self.tag
        lines = []
        primaries = "TITLE", "ARTIST", "AUTHOR", "ALBUM",\
                                  "TRACKNUMBER", "TRACKTOTAL", "GENRE", "DATE"

        for key in primaries:
            try:
                values = tag[key]
            except KeyError:
                lines.append(f"{key}=")
            else:
                for val in values:
                    lines.append(f"{key}={val}")

        for key, values in tag.items():
            if key not in primaries and isinstance(values, APETextValue):
                for val in values:
                    lines.append(f"{key}={val}")

        self.tag_frame.text_buffer.set_text("\n".join(lines))

    def __init__(self, pathname, extension):
        MutagenTagger.__init__(self, pathname)

        try:
            self.tag = self.opener[extension](pathname)
        except KeyError:
            try:
                self.tag = APEv2(pathname)
            except:
                print("ape tag not found")
                self.tag = None
                return
            else:
                print("ape tag found on non-native format")
        except:
            print("failed to create tagger for native format")
            self.tag = None
            return
        else:
            try:
                self.tag.add_tags()
            except:
                print("ape tag found on native format")
            else:
                print("no existing ape tags found")

        self.tag_frame = FreeTagFrame()
        self.tag_frame.set_vexpand(True)
        self.add(self.tag_frame)
        self.tag_frame.show()



class MutagenGUI:
    ext2name = {"aac": "AAC", "mp3": "ID3", "mp2": "ID3", "mp4": "MP4", "m4a": "MP4", "spx": "Speex",
                "flac": "FLAC", "ogg": "Ogg Vorbis", "oga": "XIPH Ogg audio", "opus": "Ogg Opus",
                "m4b": "MP4", "m4p": "MP4", "wma": "Windows Media Audio"}

    def destroy_and_quit(self, widget, data = None):
        Gtk.main_quit()
        sys.exit(0)

    def update_playlists(self, _, pathname, idjcroot):
        idjcroot.player_left.update_playlist(pathname)
        idjcroot.player_right.update_playlist(pathname)
        idjcroot.background.player.update_playlist(pathname)

    @staticmethod
    def is_supported(pathname):
        supported = [ "mp2", "mp3", "ogg", "oga" ]
        if FGlobs.avenabled:
            supported += ["aac", "mp4", "m4a", "m4b", "m4p", "ape", "mpc", "wma"]
        if FGlobs.flacenabled:
            supported.append("flac")
        if FGlobs.speexenabled:
            supported.append("spx")
        if FGlobs.opusenabled:
            supported.append("opus")
        extension = os.path.splitext(pathname)[1][1:].lower()
        if supported.count(extension) != 1:
            if extension:
                print("File type", extension, "is not supported for tagging")
            return False
        else:
            return extension

    def __init__(self, pathname, encoding, idjcroot = None):
        if not os.path.isfile(pathname or ""):
            print(f"tagger: bad pathname: {pathname}")
            return

        extension = self.is_supported(pathname)
        if extension == False:
            print("tagger: file extension", extension, "not supported.")
            return

        self.window = Gtk.Window()
        self.window.set_border_width(4)
        if idjcroot is not None:
            idjcroot.window_group.add_window(self.window)
        self.window.set_size_request(550, 450)
        # TC: Window title.
        self.window.set_title(_('IDJC Tagger') + pm.title_extra)
        self.window.set_destroy_with_parent(True)
        self.window.set_resizable(True)
        if idjcroot == None:
            self.window.connect("destroy", self.destroy_and_quit)
        vbox = Gtk.VBox()
        vbox.set_spacing(3)
        self.window.add(vbox)
        vbox.show()
        label = Gtk.Label()
        label.set_markup("<b>" + _('Filename:') + " " + \
                         GLib.markup_escape_text(os.path.split(
                             pathname)[1]) + "</b>")
        vbox.pack_start(label, False, False, 6)
        label.show()

        hbox = Gtk.HBox()
        bb = Gtk.ButtonBox()
        bb.set_spacing(3)
        bb.set_layout(Gtk.ButtonBoxStyle.END)
        hbox.pack_end(bb, True)
        bb.show()

        close_button = Gtk.Button.new_with_label(_("Close"))
        close_button.connect_object("clicked", Gtk.Window.destroy, self.window)
        bb.add(close_button)
        close_button.show()

        apply_button = Gtk.Button.new_with_label(_("Apply"))
        if idjcroot is not None:
            apply_button.connect_after("clicked", self.update_playlists,
                                                            pathname, idjcroot)
        bb.add(apply_button)
        apply_button.show()

        reload_button = Gtk.Button.new_from_icon_name("edit-undo-symbolic",
                                                      Gtk.IconSize.BUTTON)
        # reload_button.set_label(_("Undo"))
        hbox.pack_start(reload_button, False, False, 0)
        reload_button.show()
        vbox.pack_end(hbox, False, False, 0)
        hbox.show()
        hbox = Gtk.HBox()
        vbox.pack_end(hbox, False, False, 2)
        hbox.show()

        notebook = Gtk.Notebook()
        vbox.pack_start(notebook, True, True, 0)
        notebook.show()

        try:
            self.ape = ApeTagger(pathname, extension)

            if extension in ("mp3", "aac"):
                self.id3 = ID3Tagger(pathname, True)
                self.native = None
            else:
                self.id3 = ID3Tagger(pathname, False)
                if extension in ("mp4", "m4a", "m4b", "m4p"):
                    self.native = MP4Tagger(pathname)
                elif extension == "wma":
                    self.native = WMATagger(pathname)
                elif extension in ("ape", "mpc"):
                    # APE tags are native to this format.
                    self.native = None
                else:
                    self.native = NativeTagger(pathname, ext=extension)

            if self.id3 is not None and self.id3.tag is not None:
                reload_button.connect("clicked", lambda x: self.id3.load_tag())
                apply_button.connect("clicked", lambda x: self.id3.save_tag(self.window))
                label = Gtk.Label("ID3")
                notebook.append_page(self.id3, label)
                self.id3.show()

            if self.ape is not None and self.ape.tag is not None:
                reload_button.connect("clicked", lambda x: self.ape.load_tag())
                apply_button.connect("clicked", lambda x: self.ape.save_tag(self.window))
                label = Gtk.Label("APE v2")
                notebook.append_page(self.ape, label)
                self.ape.show()

            if self.native is not None and self.native.tag is not None:
                reload_button.connect("clicked",
                                            lambda x: self.native.load_tag())
                apply_button.connect("clicked",
                                            lambda x: self.native.save_tag(self.window))
                label = Gtk.Label.new(_('Native') + " (" + self.ext2name[
                                                            extension] + ")")
                notebook.append_page(self.native, label)
                self.native.show()

            reload_button.clicked()

            apply_button.connect_object_after("clicked",
                                            Gtk.Window.destroy, self.window)
            self.window.show()
        except IOError as e:
            print(e)
            self.window.destroy()