#!/usr/bin/python3.4

## Generate pdf form lilypond file
## by using standard command
##
## Most of programming was done by Sven Axelsson, http://svenax.net/

import codecs, os
from argparse import ArgumentParser

class MakeDrum:
	LILYPOND = 'lilypond'
	VERSION  = '0.9.5'
	MASTER_DIR = os.path.dirname(os.path.abspath(__file__))
	RUN_DIR = os.path.abspath(os.curdir)
	TMP_DIR  = './tmp'
	TMP_PREFIX = 'tmp_'

	def __init__(self):
		# Gather options and create the template file
		usage = __file__
		parser = ArgumentParser(usage)

		parser.add_argument('--version',
			action='store_true', dest='show_version', default=False,
			help='show makeDrum version and exit')
		parser.add_argument('--lilyversion',
			action='store_true', dest='show_lilyversion', default=False,
			help='show Lilypond version and exit')

		parser.add_argument('-x', '--drumfile',
			dest='lilydrum', default=os.path.join(self.MASTER_DIR,'lilydrum.ly'),
			help='Use the specified file for drums')
		parser.add_argument('-c', '--pipefile',
			dest='lilypipe', default='bagpipe.ly',
			help='Use the specified file for pipes')
		parser.add_argument('-i', '--include',
			dest='includes', nargs='*', default=[],action='append',
			help='Include the specified file for compiling')

		parser.add_argument('-p', '--paper-size',
			dest='papersize', default='a4',
			help='Paper size. Default: A4')
		parser.add_argument('-o', '--orientation',
			dest='orientation', default='landscape',
			help='Paper orientation. Default: landscape')
		parser.add_argument('-s', '--staff-size',
			dest='staffsize', default='20',
			help='Staff size. Default: 16pt.')
		parser.add_argument('-w', '--view-spacing',
			action='store_true', dest='view_spacing', default=False,
			help='Turn on "Paper.annotatespacing".')

		parser.add_argument('-r','--suffix',
			dest='suffix', default='',
			help='String added at end of pdf\'s filename')
		parser.add_argument('-g','--generated',
			dest='gen_out', default=self.TMP_DIR,
			help='Put generated lilyfiles in $gen_out')
		parser.add_argument('--no-compile', default=True,
			action='store_false', dest='compile',
			help='Do not compile generated Lilypond files')
		parser.add_argument('--no-log',
			action='store_false', dest='log', default=True,
			help='Do not generate log files.')
		parser.add_argument('--no-cleanup',
			action='store_false', dest='clean', default=True,
			help='Leave all temporary files in place')
		parser.add_argument('-d', '--out-dir',
			dest='out_dir', default='pdf',
			help='Output dir, for lilypond. If it doesn\'t exist, try to create it')
		parser.add_argument('music_file',
			default='', nargs='*',
			help='file to process')
		parser.add_argument('-@', '--list_file',
			dest='list_file', default='',
			help='list of files to process')

		self.args = parser.parse_args()

		if self.args.show_version:
			print(__name__, ' ', self.VERSION)
			return

		if self.args.show_lilyversion:
			print(os.system(self.LILYPOND+' --version'))
			return

		if self.args.view_spacing:
			self.args.view_spacing = "##t"
		else:
			self.args.view_spacing = "##f"

		# Input files
		if self.args.list_file != '':
			self.args.music_file.append(open(self.args.list_file, 'r').readlines())
			close(self.args.list_file)

		# Check for files
		if not self.args.music_file:
			parser.print_usage()
			return

		# Check for files to include
		self.args.includes = [el for elements in self.args.includes for el in elements]

		# Clean up of files
		self.remove_tmp_dir = self.args.clean

		if not os.path.exists(os.path.join(os.path.curdir, self.TMP_DIR)):
			try: os.makedirs(os.path.join(os.path.curdir, self.TMP_DIR))
			except:
				print('Seems like no temporary directory can be created')
				return
		if not os.path.exists(os.path.join(os.path.curdir, self.args.out_dir)):
			try: os.makedirs(os.path.join(os.path.curdir, self.args.out_dir))
			except:
				print('Seems like no output directory can be created')
				return

		for file_path in self.args.music_file:
			self.processit(self.TMP_DIR, os.path.join(self.RUN_DIR, file_path), self.args.gen_out, self.args.compile)

		#if not
		#os.rmdir(self.TMP_DIR)


	def processit(self, tmp_dir, file, gen_out, compile):
		tmp_file = self.maketemplate(tmp_dir, file)

		if gen_out is not None and gen_out != tmp_dir:
			new_tmp_file = os.path.basename(tmp_file).replace(self.TMP_PREFIX, '');
			print ('Moving ', tmp_file, ' to ', new_tmp_file, end=' ', flush=True)
			gen_dir = os.path.join(self.RUN_DIR, gen_out);
			# if not dir $gen_out, make it
			if not os.path.exists(gen_dir):
				try: os.makedirs(gen_dir)
				except:
					print('[Error]')
					print(' ! Seems like the {} directory cannot be created'.format(gen_dir))
					return
			# mv file to dir, remove self.TMP_PREFIX
			os.rename(tmp_file, os.path.join(gen_dir, new_tmp_file))
			tmp_file = new_tmp_file
			print('[OK]')

		if compile:
			if self.args.log:
				logfile = os.path.join(self.TMP_DIR, os.path.relpath(file).replace(".ly", '').replace('/', '-')+'.log')
				log = ' > '+logfile+' 2>&1'
			else:
				log = ''

			print ('Compiling ', file, end=' ', flush=True)
			if not self.args.log:
				print()
			lilyout = os.path.join(self.RUN_DIR, self.args.out_dir, os.path.basename(tmp_file).replace(self.TMP_PREFIX, '').replace(".ly", ''))
			lilycmd = self.LILYPOND+' --pdf --output='+lilyout+' '+tmp_file+log

			if os.system(lilycmd) != 0:
				self.remove_tmp_dir = False
				print ('[Error]')
				if self.args.log:
					print (' ! Did not compile, please see the log at ', logfile)
			else :
				print ('[OK]')
	
				if self.args.clean:
					#remove files
					if self.args.log:
						os.remove(logfile)
					os.remove(tmp_file)

	def maketemplate(self, tmp_dir, file):
		lily_includes = ''
		include_drum_file = False
		include_pipe_file = False
		# find out whether drum, pipes, or full score
		for ext in ['full', 'side', 'tenor', 'bass', 'drum', 'snare']:
			if ext in file:
				include_drum_file = True
				break

		for ext in ['full', 'pipes']:
			if ext in file:
				include_pipe_file = True
				break

		if include_drum_file:
			self.args.includes.insert(0,self.args.lilydrum)

		if include_pipe_file:
			self.args.includes.insert(0, self.args.lilypipe)

		for f in self.args.includes:
			lily_includes = lily_includes + "\n\\include \"{}\"".format(f)

		# set up a tmp file with template and file combined
		tmp_file = os.path.join(tmp_dir, self.TMP_PREFIX + os.path.relpath(file).replace('../','').replace('music/','',1).replace('/', '-')[:-3] + self.args.suffix + '.ly')
		print(tmp_file)
		out_file = codecs.open(tmp_file, 'w+', 'utf8')
		out_file.write(u'\ufeff')

		out_file.write(u"""
% Generated from """+file+""" by """+__file__+""" version """+self.VERSION+"""

\\version "2.18.0"

#(ly:set-option 'point-and-click #f)
"""+ lily_includes +"""


#(set-global-staff-size """+self.args.staffsize+""")
#(set-default-paper-size \""""+self.args.papersize+"""\" '"""+self.args.orientation+""")

%\layout {
% \context {
%   \Score {
%     \override NonMusicalPaperColumn #'line-break-permission = ##f
%   }
% }
%}
% The tune to generate.
		""")

		# Read lily file into tmp file
		music = codecs.open(file, 'r', 'utf8').read()
		if music.startswith(u'\ufeff'): music = music[1:]
		music = music.split(u'\n')
		printit = 1
		for line in music:
			if line.startswith(u'\\include "lilydrum.ly"'): continue
			if line.startswith(u'\\include'):
				# Rewrite includes to absolute location of file
				incline = line.replace('\\include', '').strip('"\' ')
				if not incline.startswith('\\'): #already absolute
					incline = os.path.join(os.path.abspath(os.path.dirname(file)), incline)
					line = "\\include \""+incline+"\""
			if printit:
				out_file.write(line.replace('\r', '')+'\n')
		out_file.close()

		# Return tmp_file_path
		return tmp_file
MakeDrum();