#!/usr/bin/env python
 
 # Copyright (c) 2015 Edward Langley
 # All rights reserved.
 # 
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
 # are met:
 # 
 # Redistributions of source code must retain the above copyright notice,
 # this list of conditions and the following disclaimer.
 # 
 # Redistributions in binary form must reproduce the above copyright
 # notice, this list of conditions and the following disclaimer in the
 # documentation and/or other materials provided with the distribution.
 # 
 # Neither the name of the project's author nor the names of its
 # contributors may be used to endorse or promote products derived from
 # this software without specific prior written permission.
 # 
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 # FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 # HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 # TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 # LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 # NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 '''Sorts music into my preferred directory layout.
 Usage: find -name '*.mp3' -o '*.m4a' . . . | python sorter.py <destination_directory>
 
 Requires mutagen for tag handling (and, therefore, taglib)'''
 

 import os

 import shutil

 import string
 import mutagen
 import argparse
 
 parser=argparse.ArgumentParser()
 parser.add_argument('destination')
 args = parser.parse_args()
 
 destination = args.destination

 print('copying music to, %s' % destination)

 
 import sys
 
 class PathFinder(object):
     def __init__(self):
         pass
 

     def construct_tag_map(self, data):
         tag_map = {}
         for (tag, _) in data.tags:
             tag_map[tag.lower()] = tag
         return tag_map
 

     def get_path(self, data, orig_name):

         tag_map = self.construct_tag_map(data)
         print(tag_map, tag_map['title'], data.get(tag_map['title']))

         orig_name = orig_name.rpartition('.')[0]

         title=self.clean(data.get(tag_map['title'], ['Unknown Title'])[0])

         if title.strip() == '': title = orig_name

         album=self.clean(data.get(tag_map['album'], ['Unknown Album'])[0])

         if album.strip() == '': album = 'Unknown Album'
 

         if 'albumartist' in data: artist = data[tag_map['albumartist']][0]
         elif 'performer' in data: artist = data[tag_map['performer']][0]
         elif 'conductor' in data: artist = data[tag_map['conductor']][0]
         else: artist = data.get(tag_map['artist'], ['Unknown Artist'])[0]

         artist = self.clean(artist)
         if artist.strip() == '': artist = 'Unknown Artist'
 

         tracknumber=self.clean(data.get(tag_map['tracknumber'], ['--'])[0])

         if tracknumber != '--':
            try:
                tracknumber = tracknumber.partition('_')[0]
                tracknumber = '%02d' % int(tracknumber)
            except ValueError:
                pass

         discnumber=self.clean(data.get(tag_map['discnumber'], ['--'])[0].partition('/')[0])
         print(f'discnumber is: {discnumber}')
         if discnumber.strip() not in {'', '--'}: discnumber = '%02d' % int(discnumber)

         result =[ os.path.join(artist[0].upper(),artist,album), '%s_%s' % (tracknumber, title) ]

         if discnumber.strip() not in {'', '--'}:

             result[1] = '%s_%s' % (discnumber, result[1])
         return result
 
     validchars = frozenset('_.- %s%s' % (string.ascii_letters, string.digits))
     def clean(self, strin):
         strin = ''.join((c if c in self.validchars else " ") for c in strin)
         strin = '_'.join(strin.split())
         if len(strin) > 40:
             pa = strin[:28]
             pb = strin[-31:]
             strin = '...'.join([pa,pb])
         return strin
 
 
 
 
 
 pf = PathFinder()
 for line in sys.stdin:
     line = line.strip()
     try:
         file = mutagen.File(line, easy=True)
         x = 0
         suf = ''
         path, name = pf.get_path(file, os.path.basename(line))
         ext = line.rpartition('.')[-1]
 
         path = os.path.join(destination, path)
 

         checkname = '%s.%s' % ('__'.join([_f for _f in [name, suf] if _f]), ext)

         while os.path.exists(os.path.join(path, '%s.%s' % ('__'.join([_f for _f in [name, suf] if _f]), ext))):

             x += 1
             suf = '%02d' % x
 

         name = '%s.%s' % ('__'.join([_f for _f in [name, suf] if _f]), ext)

         try:
             os.makedirs(path)

         except OSError as e:

             if e.errno == 17:
                 pass
             else:
                 raise
 

         print(os.path.join(path, name))

         shutil.copy(line, os.path.join(path, name))

     except Exception as e:
         print('Error Processing: %s' % line,e, file=sys.stderr)
         print('Error Processing: %s' % line,e)