# encoding: utf-8

import os
import sys
import codecs
import re


from mk_scipy_paper import tex2pdf, current_dir , copy_files, preamble, \
        render_abstract, addfile, sourcedir, outdir, outfilename


def hack_include_graphics(latex_text, attach_dir):
    """ Replaces all the \includegraphics call with call that impose the
        width to be 0.9\linewidth.
    """
    latex_text = re.sub(r'\\includegraphics(\[.*\])?\{',
                        r'\includegraphics\1{' + attach_dir,
                        latex_text)
    return latex_text


class MyStringIO(object):
    """ An unicode-friendly stringIO-like object.
    """

    def __init__(self):
        self.lines = []

    def write(self, line):
        self.lines.append(line)

    def getvalue(self):
        return u''.join(self.lines)

def mk_booklet_tex(outfilename):
    """ Generate the entire booklet latex file.
    """
    outfile = codecs.open(outfilename, 'w', 'utf-8')
    preamble(outfile)
    copy_files()
    #addfile(outfile, sourcedir + os.sep + 'title.tex')
    addfile(outfile, sourcedir + os.sep + 'introduction.tex')

    #outfile.write(ur'\setcounter{page}{0}' + '\n')

    #from sanum.controllers import Root as Controller
#    abstracts = model.Abstract.select()
#    for abstract in abstracts:
#        if not abstract.approved:
#            continue
#        print abstract.title
#        # Hack: I don't use a stringIO, because it is not unicode-safe.
#        tmpout = MyStringIO()
#        # Hack: I don't wont to be bound to the controller, to be
#        # abstractle to run without cherrypy.
#        #attach_dir = Controller._paper_attach_dir(abstract.id)
#        attach_dir = os.path.abspath(os.sep.join(
#                    (os.path.dirname(sanum.__file__), 'static', 
#                    'papers', '%i' % abstract.id))) + os.sep
#        render_abstract(tmpout, abstract)
#        outstring = hack_include_graphics(tmpout.getvalue(),
#                            attach_dir)
#        outfile.write(outstring)
#        #outfile.write(ur'\fillbreak' + '\n')

    outfile.write(ur'\end{document}' + '\n')




def mk_booklet(outfilename=outfilename):
    """ Generate the entire booklet pdf file.
    """
    name, ext = os.path.splitext(outfilename)
    mk_booklet_tex(name + '.tex')
    return tex2pdf(name, remove_tex=False, timeout=60)

if __name__ == '__main__':
    mk_booklet(outfilename)