/**
 * Copyright (c) 2009--2014 Red Hat, Inc.
 *
 * This software is licensed to you under the GNU General Public License,
 * version 2 (GPLv2). There is NO WARRANTY for this software, express or
 * implied, including the implied warranties of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. You should have received a copy of GPLv2
 * along with this software; if not, see
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
 *
 * Red Hat trademarks are not licensed under GPLv2. No permission is
 * granted to use or replicate Red Hat trademarks that are incorporated
 * in this software or its documentation.
 */
package com.redhat.rhn.manager.kickstart.cobbler;

import com.redhat.rhn.common.validator.ValidatorError;
import com.redhat.rhn.domain.kickstart.KickstartableTree;
import com.redhat.rhn.domain.user.User;
import com.redhat.rhn.manager.kickstart.KickstartUrlHelper;

import org.cobbler.CobblerConnection;
import org.cobbler.Distro;

import java.util.HashMap;
import java.util.Map;


/**
 * CobblerProfileComand - class to contain logic to communicate with cobbler
 * @version $Rev$
 */
public class CobblerDistroCommand extends CobblerCommand {

    protected KickstartableTree tree;

    /**
     * @param userIn - user wanting to sync with cobbler
     */
    public CobblerDistroCommand(User userIn) {
        super(userIn);
    }

    /**
     * @param ksTreeIn - KickstartableTree to sync
     * @param userIn - user wanting to sync with cobbler
     */
    public CobblerDistroCommand(KickstartableTree ksTreeIn, User userIn) {
        super(userIn);
        this.tree = ksTreeIn;
    }

    /**
     * @param ksTreeIn - KickstartableTree to sync
     */
    public CobblerDistroCommand(KickstartableTree ksTreeIn) {
        super();
        this.tree = ksTreeIn;
    }

    /**
     * Copy cobbler fields that shouldn't change in cobbler
     */
    protected void updateCobblerFields() {
        CobblerConnection con = getCobblerConnection();
        Distro nonXen = Distro.lookupById(con, tree.getCobblerId());
        Distro xen = Distro.lookupById(con, tree.getCobblerXenId());

        Map ksmeta = new HashMap();
        KickstartUrlHelper helper = new KickstartUrlHelper(this.tree);
        ksmeta.put(KickstartUrlHelper.COBBLER_MEDIA_VARIABLE,
                helper.getKickstartMediaPath());
        if (tree.getOrgId() != null) {
            ksmeta.put("org", tree.getOrg().getId());
        }

        //if the newly edited tree does para virt....
        if (tree.doesParaVirt()) {
            //IT does paravirt so we need to either update the xen distro or create one
            if (xen == null) {
                xen = Distro.create(con, tree.getCobblerXenDistroName(), tree
                        .getKernelXenPath(), tree.getInitrdXenPath(), ksmeta, tree
                        .getInstallType().getCobblerBreed(), tree.getInstallType()
                        .getCobblerOsVersion(), tree.getChannel().getChannelArch()
                        .cobblerArch());
                xen.save();
                tree.setCobblerXenId(xen.getId());
            }
            else {
                xen.setKernel(tree.getKernelXenPath());
                xen.setInitrd(tree.getInitrdXenPath());
                xen.setBreed(tree.getInstallType().getCobblerBreed());
                xen.setOsVersion(tree.getInstallType().getCobblerOsVersion());
                xen.setKsMeta(ksmeta);
                xen.setArch(tree.getChannel().getChannelArch().cobblerArch());
                xen.save();
            }

        }
        else {
            //it doesn't do paravirt, so we need to delete the xen distro
            if (xen != null) {
                xen.remove();
                tree.setCobblerXenId(null);
            }
        }

        if (nonXen != null) {
            nonXen.setInitrd(tree.getInitrdPath());
            nonXen.setKernel(tree.getKernelPath());
            nonXen.setBreed(tree.getInstallType().getCobblerBreed());
            nonXen.setOsVersion(tree.getInstallType().getCobblerOsVersion());
            nonXen.setKsMeta(ksmeta);
            nonXen.setArch(tree.getChannel().getChannelArch().cobblerArch());
            nonXen.save();
        }
    }


    /**
     * {@inheritDoc}
     */
    public ValidatorError store() {
        throw new UnsupportedOperationException();
    }

}