/**
 * Copyright (c) 2009--2012 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.
 */
/*
 * Copyright (c) 2010 SUSE LLC
 */
package com.redhat.rhn.frontend.action.kickstart.tree;

import com.redhat.rhn.common.validator.ValidatorError;
import com.redhat.rhn.domain.channel.Channel;
import com.redhat.rhn.domain.channel.ChannelFactory;
import com.redhat.rhn.domain.kickstart.KickstartFactory;
import com.redhat.rhn.domain.kickstart.KickstartInstallType;
import com.redhat.rhn.domain.kickstart.KickstartableTree;
import com.redhat.rhn.frontend.action.BaseEditAction;
import com.redhat.rhn.frontend.struts.RequestContext;
import com.redhat.rhn.manager.PersistOperation;
import com.redhat.rhn.manager.kickstart.tree.BaseTreeEditOperation;

import org.apache.struts.action.DynaActionForm;
import org.apache.struts.util.LabelValueBean;

import java.util.Iterator;
import java.util.List;

import javax.servlet.http.HttpServletRequest;

/**
 * TreeCreate class for creating Kickstart Trees
 * @version $Rev: 1 $
 */
public abstract class BaseTreeAction extends BaseEditAction {

    public static final String INSTALL_TYPE = "installtype";
    public static final String BASE_PATH = "basepath";
    public static final String CHANNEL_ID = "channelid";
    public static final String CHANNELS = "channels";
    public static final String INSTALLTYPES = "installtypes";
    public static final String LABEL = "label";
    public static final String NOCHANNELS = "nochannels";
    public static final String NOINSTALLTYPES = "noinstalltypes";

    public static final String HIDE_SUBMIT = "hidesubmit";
    public static final String KERNEL_OPTS = "kernelopts";
    public static final String POST_KERNEL_OPTS = "postkernelopts";

    protected void processRequestAttributes(RequestContext rctx, PersistOperation opr) {
        BaseTreeEditOperation bte = (BaseTreeEditOperation) opr;
        Iterator i = bte.getKickstartableTreeChannels().iterator();
        if (!i.hasNext()) {
            rctx.getRequest().setAttribute(NOCHANNELS, "true");
            rctx.getRequest().setAttribute(HIDE_SUBMIT, "true");
            return;
        }
        rctx.getRequest().setAttribute(CHANNELS,
                createLabelValueList(i, "getName", "getId"));

        Channel selectedBaseChannel = getSelectedBaseChannel(rctx);
        if (selectedBaseChannel == null) {
            rctx.getRequest().setAttribute(HIDE_SUBMIT, "true");
            return;
        }
        i = KickstartFactory.lookupKickstartInstallTypes().iterator();
        if (!i.hasNext()) {
            rctx.getRequest().setAttribute(NOINSTALLTYPES, "true");
            rctx.getRequest().setAttribute(HIDE_SUBMIT, "true");
        }
        else {
            rctx.getRequest().setAttribute(INSTALLTYPES,
            createLabelValueList(i, "getName", "getLabel"));
        }

    }

    protected ValidatorError processCommandSetters(PersistOperation operation,
                                                            DynaActionForm form,
                                                            HttpServletRequest request) {
        BaseTreeEditOperation bte = (BaseTreeEditOperation) operation;

        String label = form.getString(LABEL);
        if (!label.equals(bte.getTree().getLabel())) {
            KickstartableTree tree = KickstartFactory.lookupKickstartTreeByLabel(
                    label, bte.getUser().getOrg());
            if (tree != null) {
                return new ValidatorError("distribution.tree.exists", tree.getLabel());
            }
        }


        bte.setBasePath(form.getString(BASE_PATH));
        Long channelId = (Long) form.get(CHANNEL_ID);
        Channel c = ChannelFactory.lookupByIdAndUser(channelId, operation.getUser());
        bte.setChannel(c);
        bte.setLabel(form.getString(LABEL));
        KickstartInstallType type = KickstartFactory.
            lookupKickstartInstallTypeByLabel(form.getString(INSTALL_TYPE));
        bte.setInstallType(type);

        if (type.isSUSE()) {
            String kopts = form.getString(POST_KERNEL_OPTS);
            if (kopts.contains("install=")) {
                kopts = kopts + " install=http://" + request.getLocalName() +
                    "/ks/dist/" + form.getString(LABEL);
            }
            bte.setKernelOptions(kopts);
        }
        else {
            bte.setKernelOptions(form.getString(KERNEL_OPTS));
        }
        bte.setPostKernelOptions(form.getString(POST_KERNEL_OPTS));

        return null;

    }

    /**
     * Return the selected base channel, either the existing value for a
     * channel being edited, the previous selection on the form after an
     * error, or the first value in the list.
     * @param rctx Request context to examine.
     * @return Channel selected previously or by default, null if neither exist.
     */
    protected Channel getSelectedBaseChannel(RequestContext rctx) {

        String previousChannelIdSelection = rctx.getParam(CHANNEL_ID, false);
        if (previousChannelIdSelection != null) {
            return ChannelFactory.lookupById(new Long(previousChannelIdSelection));
        }

        KickstartableTree tree = (KickstartableTree)rctx.getRequest().getAttribute(
                RequestContext.KSTREE);
        if (tree != null) {
            // Looks like we're editing an existing tree:
            return tree.getChannel();
        }

        List channelLabels = (List)rctx.getRequest().getAttribute(CHANNELS);
        if (channelLabels != null) {
            String channelId = ((LabelValueBean)channelLabels.get(0)).getValue();
            return ChannelFactory.lookupById(new Long(channelId));
        }

        return null;
    }
}