package com.jetbrains.python.fixtures;

import com.google.common.collect.Lists;
import com.intellij.execution.ExecutionException;
import com.intellij.execution.Executor;
import com.intellij.execution.configurations.ConfigurationFactory;
import com.intellij.execution.configurations.ConfigurationType;
import com.intellij.execution.configurations.GeneralCommandLine;
import com.intellij.execution.executors.DefaultDebugExecutor;
import com.intellij.execution.executors.DefaultRunExecutor;
import com.intellij.execution.runners.ExecutionEnvironment;
import com.intellij.openapi.project.Project;
import com.jetbrains.python.PythonHelpersLocator;
import com.jetbrains.python.debugger.PyDebugRunner;
import com.jetbrains.python.run.AbstractPythonRunConfiguration;
import com.jetbrains.python.run.PythonCommandLineState;

import java.util.List;

/**
 * @author yole
 */
public abstract class PyCommandLineTestCase extends PyTestCase {
  private static final int PORT = 123;

  protected static int verifyPyDevDParameters(List<String> params) {
    params = Lists.newArrayList(params);
    int debugParam = params.remove("--DEBUG") ? 1 : 0;
    assertEquals(PythonHelpersLocator.getHelperPath("pydev/pydevd.py"), params.get(0));
    assertEquals("--multiproc", params.get(1));
    assertEquals("--client", params.get(2));
    assertEquals("--port", params.get(4));
    assertEquals("" + PORT, params.get(5));
    assertEquals("--file", params.get(6));
    return 7 + debugParam;
  }

  protected <T extends AbstractPythonRunConfiguration> T createConfiguration(final ConfigurationType configurationType, Class<T> cls) {
    final ConfigurationFactory factory = configurationType.getConfigurationFactories()[0];
    final Project project = myFixture.getProject();
    return cls.cast(factory.createTemplateConfiguration(project));
  }

  protected static List<String> buildRunCommandLine(AbstractPythonRunConfiguration configuration) {
    try {
      final Executor executor = DefaultRunExecutor.getRunExecutorInstance();
      final PythonCommandLineState state = (PythonCommandLineState)configuration.getState(executor, new ExecutionEnvironment());
      final GeneralCommandLine generalCommandLine = state.generateCommandLine();
      return generalCommandLine.getParametersList().getList();
    }
    catch (ExecutionException e) {
      throw new RuntimeException(e);
    }
  }

  protected static List<String> buildDebugCommandLine(AbstractPythonRunConfiguration configuration) {
    try {
      final Executor executor = DefaultDebugExecutor.getDebugExecutorInstance();
      final PythonCommandLineState state = (PythonCommandLineState)configuration.getState(executor, new ExecutionEnvironment());
      final GeneralCommandLine generalCommandLine =
        state.generateCommandLine(PyDebugRunner.createCommandLinePatchers(configuration.getProject(), state, configuration, PORT));
      return generalCommandLine.getParametersList().getList();
    }
    catch (ExecutionException e) {
      throw new RuntimeException(e);
    }
  }
}

@@ -541,16 +541,83 @@
 onment;%0A
+import com.intellij.execution.runners.ExecutionEnvironmentBuilder;%0A
 import c
@@ -1892,39 +1892,32 @@
  %7D%0A%0A  protected 
-static 
 List%3CString%3E bui
@@ -2060,32 +2060,169 @@
 utorInstance();%0A
+      ExecutionEnvironment env = new ExecutionEnvironmentBuilder(myFixture.getProject(), executor).setRunProfile(configuration).build();%0A
       final Pyth
@@ -2309,34 +2309,11 @@
 or, 
-new ExecutionEnvironment()
+env
 );%0A 
@@ -2556,23 +2556,16 @@
 otected 
-static 
 List%3CStr
@@ -2726,24 +2726,161 @@
 Instance();%0A
+      ExecutionEnvironment env = new ExecutionEnvironmentBuilder(myFixture.getProject(), executor).setRunProfile(configuration).build();%0A
       final 
@@ -2971,34 +2971,11 @@
 or, 
-new ExecutionEnvironment()
+env
 );%0A 

fix Python run configurations test