/*
    * This file is part of Waarp Project (named also Waarp or GG).
    *
    *  Copyright (c) 2019, Waarp SAS, and individual contributors by the @author
    *  tags. See the COPYRIGHT.txt in the distribution for a full listing of
    * individual contributors.
    *
    *  All Waarp Project is free software: you can redistribute it and/or
    * modify it under the terms of the GNU General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or (at your
   * option) any later version.
   *
   * Waarp 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 General Public License for more details.
   *
   *  You should have received a copy of the GNU General Public License along with
   * Waarp . If not, see <http://www.gnu.org/licenses/>.
   */
  package org.waarp.openr66.context.task;
  
  import org.waarp.common.logging.WaarpLogger;
  import org.waarp.common.logging.WaarpLoggerFactory;
  import org.waarp.common.role.RoleDefault.ROLE;
  import org.waarp.common.utility.WaarpShutdownHook;
  import org.waarp.openr66.context.R66Session;
  import org.waarp.openr66.protocol.configuration.Configuration;
  import org.waarp.openr66.protocol.exception.OpenR66ProtocolSystemException;
  import org.waarp.openr66.protocol.utils.ChannelUtils;
  
  import java.util.concurrent.TimeUnit;
  
  /**
   * Command to Restart the R66 server (for instance after upgrade of jar sent by
   * administrative operations,
   * unzipped in the library directory)
   */
  public class RestartServerTask extends AbstractTask {
  
    /**
     * Internal Logger
     */
    private static final WaarpLogger logger =
        WaarpLoggerFactory.getLogger(RestartServerTask.class);
  
    /**
     * @param argRule
     * @param delay
     * @param argTransfer
     * @param session
     */
    public RestartServerTask(final String argRule, final int delay,
                             final String argTransfer, final R66Session session) {
      super(TaskType.RESTART, delay, argRule, argTransfer, session);
    }
  
    @Override
    public void run() {
      // check if allowed to do restart
      // SYSTEM authorization
      final boolean isAdmin = session.getAuth().isValidRole(ROLE.SYSTEM);
      if (!isAdmin) {
        // not allowed
        logger.error("Shutdown order asked through task but unallowed: " +
                     session.getAuth().getUser());
        futureCompletion.setFailure(new OpenR66ProtocolSystemException(
            "Shutdown order asked through task but unallowed: " +
            session.getAuth().getUser()));
        return;
      }
      if (!Configuration.configuration.isServer()) {
        logger.error(
            "Shutdown order asked through task but this is a client, not a server");
        futureCompletion.setFailure(new OpenR66ProtocolSystemException(
            "Shutdown order asked through task but this is a client, not a server"));
        return;
      }
      // now start the process
      logger.warn("Shutdown order received and going from: " +
                  session.getAuth().getUser());
      WaarpShutdownHook.setRestart(true);
      futureCompletion.setSuccess();
      final Thread thread =
          new Thread(new ChannelUtils(), "R66 Shutdown and Restart Thread");
      thread.setDaemon(true);
      // give time for the task to finish correctly
      Configuration.configuration.launchInFixedDelay(thread, 1000,
                                                     TimeUnit.MILLISECONDS);
    }
  
  }