#!/usr/bin/python

import sys
import os.path
from distutils.core import setup, Extension
import string
import yaml # PyYAML
import cobbler.sub_process as subprocess
import Cheetah.Template as Template
import time

VERSION = "2.0.1"
SHORT_DESC = "Network Boot and Update Server"
LONG_DESC = """
Cobbler is a network install server.  Cobbler supports PXE, virtualized installs, and reinstalling existing Linux machines.  The last two modes use a helper tool, 'koan', that integrates with cobbler.  Cobbler's advanced features include importing distributions from DVDs and rsync mirrors, kickstart templating, integrated yum mirroring, and built-in DHCP/DNS Management.  Cobbler has a XMLRPC API for integration with other applications.  There is also a web interface.
"""
TEMPLATES_DIR = "installer_templates"
DEFAULTS = os.path.join(TEMPLATES_DIR, "defaults")
MODULES_TEMPLATE = os.path.join(TEMPLATES_DIR, "modules.conf.template")
SETTINGS_TEMPLATE = os.path.join(TEMPLATES_DIR, "settings.template")
OUTPUT_DIR = "config"

# =========================================================        
def templatify(template, answers, output):
    t = Template.Template(file=template, searchList=answers)
    data = t.respond()
    outf = open(output,"w")
    outf.write(data)
    outf.close()

def gen_build_version():
    fd = open(os.path.join(OUTPUT_DIR, "version"),"w+")
    gitdate = "?"
    gitstamp = "?"
    builddate = time.asctime()
    if os.path.exists(".git"): 
       # for builds coming from git, include the date of the last commit
       cmd = subprocess.Popen(["/usr/bin/git","log"],stdout=subprocess.PIPE)
       data = cmd.communicate()[0].strip()
       for line in data.split("\n"):
           if line.startswith("commit"):
               tokens = line.split(" ",1)
               gitstamp = tokens[1].strip()
           if line.startswith("Date:"):
               tokens = line.split(":",1)
               gitdate = tokens[1].strip()
               break
    data = {
       "gitdate" : gitdate,
       "gitstamp"      : gitstamp,
       "builddate"     : builddate,
       "version"       : VERSION,
       "version_tuple" : [ int(x) for x in VERSION.split(".")]
    }
    fd.write(yaml.dump(data))
    fd.close()
    

def gen_config():
    defaults_file = open(DEFAULTS)
    defaults_data = defaults_file.read()
    defaults_file.close() 
    defaults = yaml.load(defaults_data)
    templatify(MODULES_TEMPLATE, defaults, os.path.join(OUTPUT_DIR, "modules.conf"))
    templatify(SETTINGS_TEMPLATE, defaults, os.path.join(OUTPUT_DIR, "settings"))

if __name__ == "__main__":
    gen_build_version()
    gen_config()
        
    # etc configs
    etcpath     = "/etc/cobbler"
    initpath    = "/etc/init.d"
    rotpath       = "/etc/logrotate.d"
    powerpath   = etcpath + "/power"
    pxepath     = etcpath + "/pxe"
    reppath     = etcpath + "/reporting"
    zonepath    = etcpath + "/zone_templates"
        
    # lib paths
    libpath     = "/var/lib/cobbler"
    backpath    = libpath + "/backup"
    trigpath    = libpath + "/triggers"
    snippetpath = libpath + "/snippets"
    kickpath    = libpath + "/kickstarts"
    dbpath      = libpath + "/config"
    loadpath    = libpath + "/loaders"

    # share paths
    sharepath   = "/usr/share/cobbler"
    itemplates  = sharepath + "/installer_templates"
    wwwtmpl     = sharepath + "/webui_templates"
    manpath     = "share/man/man1"
    spool_koan  = "/var/spool/koan"       
 
    # www paths
    wwwpath  = "/var/www/cobbler"
    if os.path.exists("/etc/SuSE-release"):
        wwwconf  = "/etc/apache2/conf.d"
    elif os.path.exists("/etc/debian_version"):
        wwwconf  = "/etc/apache2/conf.d"
    else:
        wwwconf  = "/etc/httpd/conf.d"

    wwwcon   = "/var/www/cobbler_webui_content"
    vw_localmirror = wwwpath + "/localmirror"
    vw_kickstarts  = wwwpath + "/kickstarts"
    vw_kickstarts_sys  = wwwpath + "/kickstarts_sys"
    vw_repomirror = wwwpath + "/repo_mirror"
    vw_ksmirror   = wwwpath + "/ks_mirror"
    vw_ksmirrorc  = wwwpath + "/ks_mirror/config"
    vw_images     = wwwpath + "/images"
    vw_distros    = wwwpath + "/distros"
    vw_systems    = wwwpath + "/systems"
    vw_profiles   = wwwpath + "/profiles"
    vw_links      = wwwpath + "/links"
    vw_aux        = wwwpath + "/aux"
    modpython     = wwwpath + "/web"
    modpythonsvc  = wwwpath + "/svc"
        
    # log paths
    logpath  = "/var/log/cobbler"
    logpath2 = logpath + "/kicklog"
    logpath3 = logpath + "/syslog"
    logpath4 = "/var/log/httpd/cobbler"
    logpath5 = logpath + "/anamon"
    logpath6 = "/var/log/koan"
    logpath7 = logpath + "/tasks"

    # django content
    dj_config    = "/etc/httpd/conf.d/"
    dj_templates = "/usr/share/cobbler/web/cobbler_web/templates"
    dj_webui     = "/usr/share/cobbler/web/cobbler_web"
    dj_webui2    = "/usr/share/cobbler/web/cobbler_web/templatetags"
    dj_webui_proj= "/usr/share/cobbler/web"
    dj_sessions  = "/var/lib/cobbler/webui_sessions"
    dj_js        = "/var/www/cobbler_webui_content/"

    setup(
        name="cobbler",
        version = VERSION,
        author = "Michael DeHaan",
        author_email = "mdehaan@redhat.com",
        url = "http://fedorahosted.org/cobbler/",
        license = "GPL",
        packages = [
            "cobbler",
            "cobbler/modules", 
            "koan"
        ],
        scripts = [
            "scripts/cobbler", 
            "scripts/cobblerd", 
            "scripts/cobbler-ext-nodes", 
            "scripts/koan",
            "scripts/cobbler-register"
        ],
        data_files = [ 
            (modpythonsvc, ['scripts/services.py']),
 
            # miscellaneous config files
            (rotpath,  ['config/cobblerd_rotate']),
            (wwwconf,  ['config/cobbler.conf']),
            (libpath,  ['config/cobbler_hosts']),
            (etcpath,  ['config/modules.conf']),
            (etcpath,  ['config/users.digest']),
            (etcpath,  ['config/rsync.exclude']),
            (etcpath,  ['config/users.conf']),
            (etcpath,  ['config/cheetah_macros']),
            (initpath, ['config/cobblerd']),
            (etcpath,  ['config/settings']),

            # django webui content
            (dj_config,    [ 'config/cobbler_web.conf' ]),
            (dj_templates, [ 'web/cobbler_web/templates/blank.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/empty.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/enoaccess.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/error_page.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/header.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/index.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/item.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/ksfile_edit.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/ksfile_list.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/snippet_edit.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/snippet_list.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/master.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/message.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/paginate.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/settings.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/generic_edit.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/generic_list.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/generic_delete.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/generic_rename.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/events.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/eventlog.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/import.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/task_created.tmpl' ]),
            (dj_templates, [ 'web/cobbler_web/templates/check.tmpl' ]),

            # django code, private to cobbler-web application
            (dj_webui,       [ 'web/cobbler_web/__init__.py' ]),
            (dj_webui_proj,  [ 'web/__init__.py' ]),
            (dj_webui_proj,  [ 'web/urls.py' ]),
            (dj_webui_proj,  [ 'web/manage.py' ]),
            (dj_webui_proj,  [ 'web/settings.py' ]),
            (dj_webui,       [ 'web/cobbler_web/urls.py' ]),
            (dj_webui,       [ 'web/cobbler_web/views.py' ]),
            (dj_webui2,      [ 'web/cobbler_web/templatetags/site.py' ]),
            (dj_webui2,      [ 'web/cobbler_web/templatetags/__init__.py' ]),
            (dj_sessions,  []),

            # backups for upgrades
            (backpath, []),

            # for --version support across distros
            (libpath,  ['config/version']),
     
            # bootloaders and syslinux support files
            # we only package zpxe.rexx because it's source
            # user supplies the others
            (loadpath,  ['scripts/zpxe.rexx']),
                    
            # database/serializer
            (dbpath + "/distros.d",  []),
            (dbpath + "/profiles.d", []),
            (dbpath + "/systems.d",  []),
            (dbpath + "/repos.d",    []),
            (dbpath + "/images.d",   []),

            # sample kickstart files
            (kickpath,  ['kickstarts/legacy.ks']),
            (kickpath,  ['kickstarts/sample.ks']),
            (kickpath,  ['kickstarts/sample_end.ks']),
            (kickpath,  ['kickstarts/default.ks']),
            (kickpath,  ['kickstarts/pxerescue.ks']),
                                
            # seed files for debian
            (kickpath,  ['kickstarts/sample.seed']),
 
            # templates for DHCP, DNS, TFTP, RSYNC
            (etcpath,  ['templates/dhcp.template']),
            (etcpath,  ['templates/dnsmasq.template']),
            (etcpath,  ['templates/named.template']),
            (etcpath,  ['templates/zone.template']),
            (etcpath,  ['templates/rsync.template']),
                                
            # templates for netboot configs
            (pxepath,  ['templates/pxedefault.template']),
            (pxepath,  ['templates/pxesystem.template']),
            (pxepath,  ['templates/pxesystem_s390x.template']),
            (pxepath,  ['templates/pxeprofile_s390x.template']),
            (pxepath,  ['templates/s390x_conf.template']),
            (pxepath,  ['templates/s390x_parm.template']),
            (pxepath,  ['templates/pxesystem_ia64.template']),
            (pxepath,  ['templates/pxesystem_ppc.template']),
            (pxepath,  ['templates/pxeprofile.template']),
            (pxepath,  ['templates/pxelocal.template']),
            (pxepath,  ['templates/pxelocal_ia64.template']),
            (pxepath,  ['templates/pxelocal_s390x.template']),

            # templates for power management
            (powerpath, ['templates/power_apc_snmp.template']), 
            (powerpath, ['templates/power_integrity.template']), 
            (powerpath, ['templates/power_ipmilan.template']),
            (powerpath, ['templates/power_bullpap.template']),     
            (powerpath, ['templates/power_ipmitool.template']),
            (powerpath, ['templates/power_drac.template']),        
            (powerpath, ['templates/power_rsa.template']),
            (powerpath, ['templates/power_ether_wake.template']),  
            (powerpath, ['templates/power_wti.template']),
            (powerpath, ['templates/power_ilo.template']),
            (powerpath, ['templates/power_lpar.template']),        
            (powerpath, ['templates/power_bladecenter.template']),
            (powerpath, ['templates/power_virsh.template']),        

            # templates for reporting
            (reppath,   ['templates/build_report_email.template']), 

            # templates for setup
            (itemplates, ['installer_templates/modules.conf.template']),
            (itemplates, ['installer_templates/settings.template']),
            (itemplates, ['installer_templates/defaults']),

            # useful kickstart snippets that we ship
            (snippetpath, ['snippets/partition_select']),
            (snippetpath, ['snippets/pre_partition_select']),
            (snippetpath, ['snippets/main_partition_select']),
            (snippetpath, ['snippets/post_install_kernel_options']),
            (snippetpath, ['snippets/network_config']),
            (snippetpath, ['snippets/pre_install_network_config']),
            (snippetpath, ['snippets/post_install_network_config']),
            (snippetpath, ['snippets/func_install_if_enabled']),
            (snippetpath, ['snippets/func_register_if_enabled']),
            (snippetpath, ['snippets/download_config_files']),
            (snippetpath, ['snippets/koan_environment']),
            (snippetpath, ['snippets/pre_anamon']),
            (snippetpath, ['snippets/post_anamon']),
            (snippetpath, ['snippets/post_s390_reboot']),
            (snippetpath, ['snippets/post_koan_add_reinstall_entry']),
            (snippetpath, ['snippets/redhat_register']),
            (snippetpath, ['snippets/cobbler_register']),
            (snippetpath, ['snippets/keep_ssh_host_keys']),
            (snippetpath, ['snippets/log_ks_pre']),
            (snippetpath, ['snippets/log_ks_post']),

            # documentation
            (manpath,  ['docs/cobbler.1.gz']),
            (manpath,  ['docs/koan.1.gz']),
            (manpath,  ['docs/cobbler-register.1.gz']),

            # logfiles
            (logpath,  []),
            (logpath2, []),
            (logpath3, []),
            (logpath4, []),
            (logpath5, []),
            (logpath6, []),
            (logpath7, []),
  
            # spoolpaths
            (spool_koan, []),

            # web page directories that we own
            (vw_localmirror,    []),
            (vw_kickstarts,     []),
            (vw_kickstarts_sys, []),
            (vw_repomirror,     []),
            (vw_ksmirror,       []),
            (vw_ksmirrorc,      []),
            (vw_distros,        []),
            (vw_images,         []),
            (vw_systems,        []),
            (vw_profiles,       []),
            (vw_links,          []),
            (vw_aux,            []),

            # zone-specific templates directory
            (zonepath,    []),

            # Web UI templates for object viewing & modification
            # FIXME: other templates to add as they are created.
            # slurp in whole directory?

            # Web UI support files
            (wwwcon,            ['web/content/style.css']),
            (wwwcon,            ['web/content/logo-cobbler.png']),
            (modpython,         ['web/content/index.html']),
            (wwwpath + "/pub",  []),
            (dj_js,             ['web/content/cobbler.js']),
            # FIXME: someday Fedora/EPEL will package these and then we should not embed them then.
            (dj_js,             ['web/content/jquery-1.3.2.js']),
            (dj_js,             ['web/content/jquery-1.3.2.min.js']),
            (dj_js,             ['web/content/jsGrowl_jquery.js']),
            (dj_js,             ['web/content/jsGrowl.js']),
            (dj_js,             ['web/content/jsgrowl_close.png']),
            (dj_js,             ['web/content/jsgrowl_corners.png']),
            (dj_js,             ['web/content/jsgrowl_middle_hover.png']),
            (dj_js,             ['web/content/jsgrowl_corners_hover.png']),
            (dj_js,             ['web/content/jsgrowl_side_hover.png']),
            (dj_js,             ['web/content/jsGrowl.css']),

            # Anamon script
            (vw_aux,            ['aux/anamon', 'aux/anamon.init']),

            # Directories to hold cobbler triggers
            ("%s/add/distro/pre" % trigpath,      []),
            ("%s/add/distro/post" % trigpath,     []),
            ("%s/add/profile/pre" % trigpath,     []),
            ("%s/add/profile/post" % trigpath,    []),
            ("%s/add/system/pre" % trigpath,      []),
            ("%s/add/system/post" % trigpath,     []),
            ("%s/add/repo/pre" % trigpath,        []),
            ("%s/add/repo/post" % trigpath,       []),
            ("%s/delete/distro/pre" % trigpath,   []),
            ("%s/delete/distro/post" % trigpath,  []),
            ("%s/delete/profile/pre" % trigpath,  []),
            ("%s/delete/profile/post" % trigpath, []),
            ("%s/delete/system/pre" % trigpath,   []),
            ("%s/delete/system/post" % trigpath,  []),
            ("%s/delete/repo/pre" % trigpath,     []),
            ("%s/delete/repo/post" % trigpath,    []),
            ("%s/delete/repo/post" % trigpath,    []),
            ("%s/install/pre" % trigpath,         []),
            ("%s/install/post" % trigpath,        []),
            ("%s/sync/pre" % trigpath,            []),
            ("%s/sync/post" % trigpath,           []),
            ("%s/change" % trigpath,              [])
        ],
        description = SHORT_DESC,
        long_description = LONG_DESC
    )