#!/usr/bin/python

import glob
import optparse
import os
import re
import sys
import tempfile

sys.path.insert (0, os.path.split (sys.argv[0])[0] + '/..')

from gub import repository
from gub import misc

def parse_options ():
    p = optparse.OptionParser ()
    p.usage = "dist-check.py BUILD-DIR"
    p.add_option ('--repository',
		  action="store",
		  dest="repository",
                  default="",
		  help="Repository of lilypond.")
    
    p.add_option ('--branch',
		  action="store",
		  dest="branch",
                  default="",
		  help="Branch of lilypond.")
    
    (o,a) = p.parse_args ()
    if len (a) < 1:
	p.print_help()
	sys.exit (2)

    
    o.repository = os.path.abspath (o.repository)
    
    return (o, a)

def check_cvs ():
    pass

def system (s):
    print s
    if os.system (s):
        raise Exception ('failed')

def popen (s):
    print s
    return os.popen (s)

def get_config_dict (dir):
    d = misc.grok_sh_variables (dir + '/config.make')

    version_dict = misc.grok_sh_variables (d['configure-srcdir'] + '/VERSION')
    d.update (version_dict)

    d['builddir'] = d['configure-builddir']
    d['srcdir'] = d['configure-srcdir']
    
    return d

def check_files (tarball, repo):
    error_found = False

    tarball = os.path.abspath (tarball)
    tarball_dirname = re.sub ('\.tar.*', '', os.path.split (tarball)[1])

    dir = tempfile.mkdtemp ()
    
    files = popen ('cd %(dir)s && tar xzvf %(tarball)s' % locals ()).readlines ()
    files = [f.strip () for f in files]

    ## .ly files
    ly_files  = [f for f in files
                 if re.search (r'\.ly$', f)]


    ly_file_str = ' '.join (ly_files)
    
    no_version = popen (r"cd %(dir)s && grep '\\version' -L %(ly_file_str)s" % locals ()).readlines ()
    if no_version:
        print 'Files without \\version: '
        print '\n'.join (no_version)
        error_found = True


    ## tarball <-> CVS
    file_dict = dict ((f, 1) for f in files)
    
    entries = repo.all_files ()
    exceptions = ['.cvsignore', 'stepmake/.cvsignore']

    for e in entries:
        filename = e

        if filename in exceptions:
            continue

        filename = os.path.join (tarball_dirname, filename)
        if not file_dict.has_key (filename):
            print ('file from CVS not distributed: %s' % filename)
            error_found = True

    system ('rm -rf %(dir)s' % locals ())
    if error_found:
        raise Exception ('dist error found')

def main ():
    (options, args) = parse_options ()
    
    builddir = args[0]
    config = get_config_dict (builddir)

    system ('cd %(builddir)s/ && make DOCUMENTATION=yes dist' % locals ())
    tarball = '%(builddir)s/out/lilypond-%(MAJOR_VERSION)s.%(MINOR_VERSION)s.%(PATCH_LEVEL)s.tar.gz' % config


    repo = repository.Git (options.repository,
                                     branch=options.branch)

    check_files (tarball, repo)

    # uploading is done by makefile.
    
if __name__ == '__main__':
    main ()