// Copyright 2015 ThoughtWorks, Inc.

// This file is part of getgauge/Intellij-plugin.

// getgauge/Intellij-plugin 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.

// getgauge/Intellij-plugin 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 getgauge/Intellij-plugin.  If not, see <http://www.gnu.org/licenses/>.

package com.thoughtworks.gauge.execution;

import com.intellij.debugger.impl.GenericDebuggerRunner;
import com.intellij.execution.ExecutionException;
import com.intellij.execution.configurations.ConfigurationFactory;
import com.intellij.execution.configurations.GeneralCommandLine;
import com.intellij.execution.executors.DefaultDebugExecutor;
import com.intellij.execution.impl.RunManagerImpl;
import com.intellij.execution.impl.RunnerAndConfigurationSettingsImpl;
import com.intellij.execution.process.ColoredProcessHandler;
import com.intellij.execution.process.ProcessTerminatedListener;
import com.intellij.execution.remote.RemoteConfiguration;
import com.intellij.execution.remote.RemoteConfigurationType;
import com.intellij.execution.runners.ExecutionEnvironment;
import com.intellij.ide.util.ProjectPropertiesComponentImpl;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.application.ModalityState;
import com.intellij.openapi.project.Project;

public class GaugeRunProcessHandler extends ColoredProcessHandler {

    private GaugeRunProcessHandler(Process process, String commandLineString) {
        super(process, commandLineString);
    }

    public static GaugeRunProcessHandler runCommandLine(final GeneralCommandLine commandLine, GaugeRunConfiguration.DebugInfo debugInfo, Project project) throws ExecutionException {
        final GaugeRunProcessHandler gaugeRunProcess = new GaugeRunProcessHandler(commandLine.createProcess(), commandLine.getCommandLineString());
        ProcessTerminatedListener.attach(gaugeRunProcess);
        if (debugInfo.shouldDebug()) {
            launchDebugger(project, debugInfo);
        }
        return gaugeRunProcess;
    }

    private static void launchDebugger(final Project project, final GaugeRunConfiguration.DebugInfo debugInfo) {
        Runnable runnable = () -> {
            final long startTime = System.currentTimeMillis();
            GenericDebuggerRunner basicProgramRunner = new GenericDebuggerRunner();
            ProjectPropertiesComponentImpl propertiesComponent = new ProjectPropertiesComponentImpl();
            RunManagerImpl manager = new RunManagerImpl(project, propertiesComponent);
            ConfigurationFactory configFactory = RemoteConfigurationType.getInstance().getConfigurationFactories()[0];
            RemoteConfiguration remoteConfig = new RemoteConfiguration(project, configFactory);
            remoteConfig.PORT = debugInfo.getPort();
            remoteConfig.HOST = debugInfo.getHost();
            remoteConfig.USE_SOCKET_TRANSPORT = true;
            remoteConfig.SERVER_MODE = false;
            RunnerAndConfigurationSettingsImpl configuration = new RunnerAndConfigurationSettingsImpl(manager, remoteConfig, false);
            ExecutionEnvironment environment = new ExecutionEnvironment(new DefaultDebugExecutor(), basicProgramRunner, configuration, project);

            boolean debuggerConnected = false;
            // Trying to connect to gauge java for 25 secs. The sleep is because it may take a few seconds for gauge to launch the java process and the jvm to load after that
            while (!debuggerConnected && ((System.currentTimeMillis() - startTime) < 25000)) {
                try {
                    Thread.sleep(5000);
                    basicProgramRunner.execute(environment);
                    debuggerConnected = true;
                } catch (Exception e) {
                    System.err.println("Failed to connect debugger. Retrying... : " + e.getMessage());
                }
            }
        };

        ApplicationManager.getApplication().invokeAndWait(runnable, ModalityState.any());
    }
}