# -*- coding: utf-8 -*-
# This file is part of beets.
# Copyright 2017, Dorian Soergel.
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.

"""Gets work title, disambiguation, parent work and its disambiguation, composer, composer sort name and performers
"""

from __future__ import division, absolute_import, print_function

import subprocess

from beets import ui
from beets import util
from beets.plugins import BeetsPlugin

import musicbrainzngs

class ParentWorkPlugin(BeetsPlugin):

    def __init__(self):
        super(ParentWorkPlugin, self).__init__()
        self.import_stages = [self.imported]
    # possible options
    #        self.config.add({
    #            u'bin': u'KeyFinder',
    #            u'auto': True,
    #            u'overwrite': False,
    #        })
    #        self.add_media_field('work', 'work title')
    #        self.add_media_field('work_disamb', 'work title disambiguation')
    #        self.add_media_field('parent_work', 'parent work title')
    #        self.add_media_field('parent_work_disamb', 'parent work title disambiguation')
    #        self.add_media_field('parent_composer', 'parent composer name')
    #        self.add_media_field('parent_composer_sort', 'parent composer sort name')
    #        self.add_media_field('performer', 'performer name')
    #        self.add_media_field('performer_sort', 'performer sort name')
    
    def commands(self):
        cmd = ui.Subcommand('parentwork',
                            help=u'fetches parent works, composers and performers')
        cmd.func = self.command
        return [cmd]
        
    def command(self, lib, opts, args):
        self.find_key(lib.items(ui.decargs(args)))
             
    def imported(self, session, task):
        self.find_work(task.items)        
    
    def find_work(self, items):
        
        for item in items:
            performer            = []
            performer_sort       = []
            work                 = []
            work_disambig        = []
            parent_work          = []
            parent_work_disambig = []
            parent_composer      = []
            parent_composer_sort = []
            item.read()
            recording_id=item['mb_trackid']
            i=0
            while i<5:
                try: 
                    performer_types=['performer','instrument','vocals','conductor','orchestra','chorus master','concertmaster']
                    rec_rels=musicbrainzngs.get_recording_by_id(recording_id, includes=['work-rels', 'artist-rels'])
                    if 'artist-relation-list' in rec_rels['recording']:
                        for dudes in rec_rels['recording']['artist-relation-list']:
                            if dudes['type'] in performer_types:
                                performer.append(dudes['artist']['name'])
                                performer_sort.append(dudes['artist']['sort-name'])
                    if 'work-relation-list' in rec_rels['recording']:
                        for work_relation in rec_rels['recording']['work-relation-list']:
                            work_id=work_relation['work']['id']
                            work_info=musicbrainzngs.get_work_by_id(work_id, includes=["work-rels", "artist-rels"])
                            work.append(work_info['work']['title'])
                            if 'disambiguation' in work_info['work']:
                                work_disambig.append(work_info['work']['disambiguation'])
                            partof=True
                            while partof:
                                partof=False
                                if 'work-relation-list' in work_info['work']:
                                    for work_father in work_info['work']['work-relation-list']:
                                        if work_father['type'] == 'parts': 
                                            if 'direction' in work_father:
                                                if work_father['direction'] == 'backward':
                                                    father_id=work_father['work']['id']
                                                    partof=True
                                                    work_info=musicbrainzngs.get_work_by_id(father_id, includes=["work-rels", "artist-rels"])
                            if 'artist-relation-list' in work_info['work']:
                                for artist in work_info['work']['artist-relation-list']:
                                    if artist['type']=='composer':
                                        if artist['artist']['name'] in parent_composer:
                                            pass
                                        else: 
                                            parent_composer.append(artist['artist']['name'])
                                            parent_composer_sort.append(artist['artist']['sort-name'])
                            else:
                                print('no composer')
                                print('add one at')
                                print('https://musicbrainz.org/work/'+ work_info['work']['id'])
                            if work_info['work']['title'] in parent_work:
                                pass
                            else: 
                                parent_work.append(work_info['work']['title'])
                                if 'disambiguation' in work_info['work']:
                                    parent_work_disambig.append(work_info['work']['disambiguation'])
                except musicbrainzngs.musicbrainz.NetworkError:
                    continue
                    i=i+1
                break    
            
            item['parent_work']=parent_work
            item['parent_work_disambig']=parent_work_disambig
            item['work']=work
            item['work_disambig']=work_disambig
            item['performer']=performer
            item['performer_sort']=performer_sort
            item['parent_composer']=parent_composer
            item['parent_composer_sort']=parent_composer_sort
            
            item.try_write()
            item.store()