"""
Running small pieces of cobbler sync when certain actions are taken,
such that we don't need a time consuming sync when adding new
systems if nothing has changed for systems that have already 
been created.

Copyright 2006, Red Hat, Inc
Michael DeHaan <mdehaan@redhat.com>

This software may be freely redistributed under the terms of the GNU
general public license.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
"""


import os
import os.path
import shutil
import time
import yaml # Howell-Clark version
import sub_process
import sys

import utils
import action_sync
from cexceptions import *
import traceback
import errno

from rhpl.translate import _, N_, textdomain, utf8


class BootLiteSync:
    """
    Handles conversion of internal state to the tftpboot tree layout
    """

    def __init__(self,config):
        """
        Constructor
        """
        self.verbose     = True
        self.config      = config
        self.distros     = config.distros()
        self.profiles    = config.profiles()
        self.systems     = config.systems()
        self.settings    = config.settings()
        self.repos       = config.repos()
        self.sync        = action_sync.BootSync(self.config) 

    def add_single_distro(self, name):
        # get the distro record
        distro = self.distros.find(name=name)
        if distro is None:
            raise CX(_("error in distro lookup: %s") % name)
        # generate YAML file in distros/$name in webdir
        self.sync.write_distro_file(distro)
        # copy image files to images/$name in webdir & tftpboot:
        self.sync.copy_single_distro_files(distro)

    def remove_single_distro(self, name):
        # delete distro YAML file in distros/$name in webdir
        self.sync.rmfile(os.path.join(self.settings.webdir, "distros", name))
        # delete contents of images/$name directory in webdir
        self.sync.rmtree(os.path.join(self.settings.webdir, "images", name))
        # delete contents of images/$name in tftpboot
        self.sync.rmtree(os.path.join(self.settings.tftpboot, "images", name))
        # delete potential symlink to tree in webdir/links
        self.sync.rmfile(os.path.join(self.settings.webdir, "links", name)) 

    def add_single_profile(self, name):
        # get the profile object:
        profile = self.profiles.find(name=name)
        if profile is None:
            raise CX(_("error in profile lookup"))
        # rebuild profile_list YAML file in webdir
        self.sync.write_listings()
        # add profiles/$name YAML file in webdir
        self.sync.write_profile_file(profile)
        # generate kickstart for kickstarts/$name/ks.cfg in webdir
        self.sync.validate_kickstart_for_specific_profile(profile)
    
    def remove_single_profile(self, name):
        # rebuild profile_list YAML file in webdir
        self.sync.write_listings()
        # delete profiles/$name file in webdir
        self.sync.rmfile(os.path.join(self.settings.webdir, "profiles", name))
        # delete contents on kickstarts/$name directory in webdir
        self.sync.rmtree(os.path.join(self.settings.webdir, "kickstarts", name))
    
    def add_single_system(self, name):
        # get the system object:
        system = self.systems.find(name=name)
        if system is None:
            raise CX(_("error in system lookup for %s") % name)
        # rebuild system_list file in webdir
        self.sync.regen_ethers() # /etc/ethers, for dnsmasq & rarpd
        self.sync.regen_hosts()  # /var/lib/cobbler/cobbler_hosts, pretty much for dnsmasq
        self.sync.write_listings()
        # write the PXE and YAML files for the system
        self.sync.write_all_system_files(system)
        # per system kickstarts
        self.sync.validate_kickstart_for_specific_system(system)

    def remove_single_system(self, name):
        system_record = self.systems.find(name=name)
        # rebuild system_list file in webdir
        self.sync.write_listings()
        # delete system YAML file in systems/$name in webdir
        self.sync.rmfile(os.path.join(self.settings.webdir, "systems", name))
        # delete contents of kickstarts_sys/$name in webdir
        system_record = self.systems.find(name=name)
        filename = utils.get_config_filename(system_record)
        self.sync.rmtree(os.path.join(self.settings.webdir, "kickstarts_sys", filename))

        if not system_record.is_pxe_supported():
            # no need to go any further with PXE cleanup
            return
        
        # delete PXE Linux configuration file (which might be in one of two places)
        itanic = False
        profile = self.profiles.find(name=system_record.profile)
        if profile is not None:
            distro = self.distros.find(name=profile.distro)
            if distro is not None and distro in [ "ia64", "IA64"]:
                itanic = True
        if not itanic:
            self.sync.rmfile(os.path.join(self.settings.tftpboot, "pxelinux.cfg", filename))
        else:
            self.sync.rmfile(os.path.join(self.settings.tftpboot, filename))