AUTHUPDATE.java

/**
 * Copyright 2009, Frederic Bregier, and individual contributors by the @author tags. See the
 * COPYRIGHT.txt in the distribution for a full listing of individual contributors.
 * 
 * This is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser
 * General Public License as published by the Free Software Foundation; either version 3.0 of the
 * License, or (at your option) any later version.
 * 
 * This software is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License along with this
 * software; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 * Boston, MA 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.waarp.gateway.ftp.config;

import java.io.File;

import org.waarp.common.command.ReplyCode;
import org.waarp.common.command.exception.CommandAbstractException;
import org.waarp.common.command.exception.Reply500Exception;
import org.waarp.common.command.exception.Reply501Exception;
import org.waarp.common.logging.WaarpLogger;
import org.waarp.common.logging.WaarpLoggerFactory;
import org.waarp.ftp.core.command.AbstractCommand;

/**
 * AUTHTUPDATE command: implements the command that will try to update the authentications from the
 * file given as argument or the original one if no argument is given.<br>
 * Two optional arguments exist:<br>
 * - PURGE: empty first the current authentications before applying the update<br>
 * - SAVE: save the final authentications on the original name given at startup.<br>
 * 
 * @author Frederic Bregier
 * 
 */
public class AUTHUPDATE extends AbstractCommand {
    /**
     * Internal Logger
     */
    private static final WaarpLogger logger = WaarpLoggerFactory
            .getLogger(AUTHUPDATE.class);

    @Override
    public void exec() throws CommandAbstractException {
        if (!getSession().getAuth().isAdmin()) {
            // not admin
            throw new Reply500Exception("Command Not Allowed");
        }
        String filename = null;
        boolean purge = false;
        boolean write = false;
        if (!hasArg()) {
            filename = ((FileBasedConfiguration) getConfiguration()).getAuthenticationFile();
        } else {
            String[] authents = getArgs();
            for (int i = 0; i < authents.length; i++) {
                if (authents[i].equalsIgnoreCase("PURGE")) {
                    purge = true;
                } else if (authents[i].equalsIgnoreCase("SAVE")) {
                    write = true;
                } else if (filename == null) {
                    filename = authents[i];
                }
            }
            if (filename == null) {
                filename = ((FileBasedConfiguration) getConfiguration()).getAuthenticationFile();
            }
            File file = new File(filename);
            if (!file.canRead()) {
                throw new Reply501Exception("Filename given as parameter is not found: " + filename);
            }
        }
        if (!((FileBasedConfiguration) getConfiguration()).initializeAuthent(filename, purge)) {
            throw new Reply501Exception("Filename given as parameter is not correct");
        }
        if (write) {
            if (!((FileBasedConfiguration) getConfiguration()).
                    saveAuthenticationFile(
                    ((FileBasedConfiguration) getConfiguration()).getAuthenticationFile())) {
                throw new Reply501Exception("Update is done but Write operation is not correct");
            }
        }
        logger.warn("Authentication was updated from " + filename);
        getSession().setReplyCode(ReplyCode.REPLY_200_COMMAND_OKAY,
                "Authentication is updated");
    }

}