import os
import headphones
import shutil
import time

from subprocess import call
from headphones import logger
from lib.beets.mediafile import MediaFile

try:
    import argparse
except ImportError:
    import lib.argparse as argparse

def encode(albumPath):
	tempDirEncode=os.path.join(albumPath,"temp")
	musicFiles=[]
	musicFinalFiles=[]
	musicTempFiles=[]
	encoder =""
	startAlbumTime=time.clock()
	ifencoded=0
	
	if not os.path.exists(tempDirEncode):		
		os.mkdir(tempDirEncode)
	else:
		shutil.rmtree(tempDirEncode)
		time.sleep(1)
		os.mkdir(tempDirEncode)
		
	for r,d,f in os.walk(albumPath):
		for music in f:
			if any(music.lower().endswith('.' + x.lower()) for x in headphones.MEDIA_FORMATS):
				if (headphones.ENCODERLOSSLESS):
					if (music.lower().endswith('.flac')):
						musicFiles.append(os.path.join(r, music))
						musicTemp = os.path.normpath(os.path.splitext(music)[0]+'.'+headphones.ENCODEROUTPUTFORMAT).encode(headphones.SYS_ENCODING)
						musicTempFiles.append(os.path.join(tempDirEncode, musicTemp))
					else:
						logger.warn('Music "%s" is already encoded' % (music))
				else:
					musicFiles.append(os.path.join(r, music))
					musicTemp = os.path.normpath(os.path.splitext(music)[0]+'.'+headphones.ENCODEROUTPUTFORMAT).encode(headphones.SYS_ENCODING)
					musicTempFiles.append(os.path.join(tempDirEncode, musicTemp))
							
	if headphones.ENCODER=='lame':
		encoder=os.path.join(headphones.ENCODERFOLDER,'lame')
	elif headphones.ENCODER=='ffmpeg':
		encoder=os.path.join(headphones.ENCODERFOLDER,'ffmpeg')
	i=0
	for music in musicFiles:		
		infoMusic=MediaFile(music)
		if headphones.ENCODER == 'lame':
			if not any(music.lower().endswith('.' + x) for x in ["mp3", "wav"]):
				logger.warn('Lame cant encode "%s" format for "%s", use ffmpeg' % (os.path.splitext(music)[1],music))
			else:
				if (music.lower().endswith('.mp3') and (infoMusic.bitrate/1000<=headphones.BITRATE)): 
					logger.warn('Music "%s" has bitrate<="%skbit" will not be reencoded' % (music,headphones.BITRATE))
				else:
					command(encoder,music,musicTempFiles[i],albumPath)
					ifencoded=1
		else:
			if headphones.ENCODEROUTPUTFORMAT=='ogg':
				if music.lower().endswith('.ogg'):
					logger.warn('Can not reencode .ogg music "%s"' % (music))
				else:
					command(encoder,music,musicTempFiles[i],albumPath)
					ifencoded=1
			elif (headphones.ENCODEROUTPUTFORMAT=='mp3' or headphones.ENCODEROUTPUTFORMAT=='m4a'):
				if (music.lower().endswith('.'+headphones.ENCODEROUTPUTFORMAT) and (infoMusic.bitrate/1000<=headphones.BITRATE)):
					logger.warn('Music "%s" has bitrate<="%skbit" will not be reencoded' % (music,headphones.BITRATE))		
				else:
					command(encoder,music,musicTempFiles[i],albumPath)
					ifencoded=1
		i=i+1
		
	shutil.rmtree(tempDirEncode)
	time.sleep(1)	
	for r,d,f in os.walk(albumPath):
		for music in f:
			if any(music.lower().endswith('.' + x.lower()) for x in headphones.MEDIA_FORMATS):
				musicFinalFiles.append(os.path.join(r, music))
	
	if ifencoded==0:
		logger.info('Encoding for folder "%s" is not needed' % (albumPath))
	
	return musicFinalFiles
	
def command(encoder,musicSource,musicDest,albumPath):
	return_code=1
	cmd=''
	startMusicTime=time.clock()
	if headphones.ENCODER == 'lame':
		if headphones.ADVANCEDENCODER =='':
			cmd=encoder + ' -h'		
			if headphones.ENCODERVBRCBR=='cbr':
				cmd=cmd+ ' --resample ' + str(headphones.SAMPLINGFREQUENCY) + ' -b ' + str(headphones.BITRATE)
			elif headphones.ENCODERVBRCBR=='vbr':
				cmd=cmd+' -V'+str(headphones.ENCODERQUALITY)
			cmd=cmd+ ' ' + headphones.ADVANCEDENCODER
		else:
			cmd=cmd+' '+ headphones.ADVANCEDENCODER
		cmd=cmd+ ' "' + musicSource + '"'
		cmd=cmd+ ' "' + musicDest +'"'
		
	elif headphones.ENCODER == 'ffmpeg':
		cmd=encoder+ ' -i'
		cmd=cmd+ ' "' + musicSource + '"'
		if headphones.ADVANCEDENCODER =='':
			if headphones.ENCODEROUTPUTFORMAT=='ogg':
				cmd=cmd+ ' -acodec libvorbis'
			if headphones.ENCODEROUTPUTFORMAT=='m4a':
				cmd=cmd+ ' -strict experimental'
			if headphones.ENCODERVBRCBR=='cbr':
				cmd=cmd+ ' -ar ' + str(headphones.SAMPLINGFREQUENCY) + ' -ab ' + str(headphones.BITRATE) + 'k'
			elif headphones.ENCODERVBRCBR=='vbr':
				cmd=cmd+' -aq ' + str(headphones.ENCODERQUALITY)
			cmd=cmd+ ' -y -ac 2 -map_meta_data 0:0,s0 -vn'
		else:
			cmd=cmd+' '+ headphones.ADVANCEDENCODER
		cmd=cmd+ ' "' + musicDest + '"'
	print cmd
	time.sleep(10)
	return_code = call(cmd, shell=True)
	if (return_code==0) and (os.path.exists(musicDest)):
		os.remove(musicSource)
		shutil.move(musicDest,albumPath)
		logger.info('Music "%s" encoded in %s' % (musicSource,getTimeEncode(startMusicTime)))
		
def getTimeEncode(start):
	seconds =int(time.clock()-start)
	hours = seconds / 3600
	seconds -= 3600*hours
	minutes = seconds / 60
	seconds -= 60*minutes
	return "%02d:%02d:%02d" % (hours, minutes, seconds)