1 files changed, 86 insertions, 0 deletions
diff --git a/plugins/coverage/src/com/intellij/coverage/JavaCoverageRunner.java b/plugins/coverage/src/com/intellij/coverage/JavaCoverageRunner.java
new file mode 100644
index 000000000000..314e07c23a8f
--- /dev/null
+++ b/plugins/coverage/src/com/intellij/coverage/JavaCoverageRunner.java
@@ -0,0 +1,86 @@
+package com.intellij.coverage;
+
+import com.intellij.execution.configurations.SimpleJavaParameters;
+import com.intellij.openapi.application.PathManager;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.util.SystemInfo;
+import com.intellij.openapi.util.io.FileUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.io.File;
+import java.io.IOException;
+
+/**
+ * @author Roman.Chernyatchik
+ */
+public abstract class JavaCoverageRunner extends CoverageRunner {
+  private static final Logger LOG = Logger.getInstance("#" + JavaCoverageRunner.class.getName());
+  private static final String COVERAGE_AGENT_PATH = "coverage.lib.path";
+
+  public boolean isJdk7Compatible() {
+    return true;
+  }
+  
+  @Override
+  public boolean acceptsCoverageEngine(@NotNull CoverageEngine engine) {
+    return engine instanceof JavaCoverageEngine;
+  }
+
+  public abstract void appendCoverageArgument(final String sessionDataFilePath, @Nullable final String[] patterns, final SimpleJavaParameters parameters,
+                                              final boolean collectLineInfo, final boolean isSampling);
+
+  protected static String handleSpacesInPath(String agentPath) {
+    final String userDefined = System.getProperty(COVERAGE_AGENT_PATH);
+    if (userDefined != null && new File(userDefined).exists()) {
+      agentPath = userDefined;
+    } else {
+      agentPath = new File(agentPath).getParent();
+    }
+    if (!SystemInfo.isWindows && agentPath.contains(" ")) {
+      File dir = new File(PathManager.getSystemPath(), "coverageJars");
+      if (dir.getAbsolutePath().contains(" ")) {
+        try {
+          dir = FileUtil.createTempDirectory("coverage", "jars");
+          if (dir.getAbsolutePath().contains(" ")) {
+            LOG.info("Coverage agent not used since the agent path contains spaces: " + agentPath + "\n" +
+                     "One can move the agent libraries to a directory with no spaces in path and specify its path in idea.properties as " + COVERAGE_AGENT_PATH + "=<path>");
+            return agentPath;
+          }
+        }
+        catch (IOException e) {
+          LOG.info(e);
+          return agentPath;
+        }
+      }
+
+      try {
+        LOG.info("Coverage jars were copied to " + dir.getPath());
+        FileUtil.copyDir(new File(agentPath), dir);
+        return dir.getPath();
+      }
+      catch (IOException e) {
+        LOG.info(e);
+      }
+    }
+    return agentPath;
+  }
+
+   protected static void write2file(File tempFile, String arg) throws IOException {
+    FileUtil.writeToFile(tempFile, (arg + "\n").getBytes("UTF-8"), true);
+  }
+
+  protected static File createTempFile() throws IOException {
+    File tempFile = FileUtil.createTempFile("coverage", "args");
+    if (!SystemInfo.isWindows && tempFile.getAbsolutePath().contains(" ")) {
+      tempFile = FileUtil.createTempFile(new File(PathManager.getSystemPath(), "coverage"), "coverage", "args", true);
+      if (tempFile.getAbsolutePath().contains(" ")) {
+        final String userDefined = System.getProperty(COVERAGE_AGENT_PATH);
+        if (userDefined != null && new File(userDefined).isDirectory()) {
+          tempFile = FileUtil.createTempFile(new File(userDefined), "coverage", "args", true);
+        }
+      }
+    }
+    return tempFile;
+  }
+}