/*******************************************************************************
 * Copyright (c) 2007, 2014 Massimiliano Ziccardi
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *******************************************************************************/
package it.jnrpe.plugin;

import it.jnrpe.ICommandLine;
import it.jnrpe.ReturnValue;
import it.jnrpe.Status;
import it.jnrpe.plugins.PluginBase;
import it.jnrpe.utils.StreamManager;
import it.jnrpe.utils.StringUtils;

import java.io.BufferedReader;
import java.io.File;
import java.io.InputStreamReader;

/**
 * Executes external (executable) plugins.
 *
 * @author Massimiliano Ziccardi
 */
public class CNativePlugin extends PluginBase {

    /**
     * The first parameter must be the full path to the executable.
     *
     * The rest of the array is sent to the executable as commands parameters
     * 
     * @param cl
     *            The parsed command line
     * @return The return value of the plugin
     */
    public final ReturnValue execute(final ICommandLine cl) {
        File fProcessFile = new File(cl.getOptionValue("executable"));
        StreamManager streamMgr = new StreamManager();

        if (!fProcessFile.exists()) {
            return new ReturnValue(Status.UNKNOWN, "Could not exec executable : " + fProcessFile.getAbsolutePath());
        }

        try {
            String[] vsParams = StringUtils.split(cl.getOptionValue("args", ""), false);
            String[] vCommand = new String[vsParams.length + 1];
            vCommand[0] = cl.getOptionValue("executable");
            System.arraycopy(vsParams, 0, vCommand, 1, vsParams.length);
            Process p = Runtime.getRuntime().exec(vCommand);
            BufferedReader br = (BufferedReader) streamMgr.handle(new BufferedReader(new InputStreamReader(p.getInputStream())));
            StringBuilder msg = new StringBuilder();
            
            for (String line = br.readLine(); line != null; line = br.readLine()) {
                if (msg.length() != 0) {
                    msg.append(System.lineSeparator());
                }
                msg.append(line);
            }
            
            int iReturnCode = p.waitFor();

            return new ReturnValue(Status.fromIntValue(iReturnCode), msg.toString());
        } catch (Exception e) {
            String message = e.getMessage();
            
            LOG.warn(getContext(), "Error executing the native plugin : " + message, e);
            return new ReturnValue(Status.UNKNOWN, "Could not exec executable : " + fProcessFile.getName() + " - ERROR : " + message);
        } finally {
            streamMgr.closeAll();
        }
    }

    @Override
    protected String getPluginName() {
        return "NATIVE";
    }
}