python/src/com/jetbrains/python/sdk/flavors/WinPythonSdkFlavor.java

   1 /*
   2  * Copyright 2000-2013 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.jetbrains.python.sdk.flavors;
  17
  18 import com.google.common.collect.ImmutableMap;
  19 import com.intellij.openapi.util.io.FileUtil;
  20 import com.intellij.openapi.util.io.WindowsRegistryUtil;
  21 import com.intellij.openapi.util.text.StringUtil;
  22 import com.intellij.openapi.vfs.LocalFileSystem;
  23 import com.intellij.openapi.vfs.VirtualFile;
  24 import com.intellij.openapi.vfs.newvfs.NewVirtualFile;
  25 import com.jetbrains.python.PythonHelpersLocator;
  26
  27 import java.io.File;
  28 import java.util.*;
  29
  30 /**
  31  * @author yole
  32  */
  33 public class WinPythonSdkFlavor extends CPythonSdkFlavor {
  34   public static WinPythonSdkFlavor INSTANCE = new WinPythonSdkFlavor();
  35   private static Map<String, String> ourRegistryMap =
  36     ImmutableMap.of("HKEY_LOCAL_MACHINE\\SOFTWARE\\Python\\PythonCore", "python.exe",
  37                     "HKEY_LOCAL_MACHINE\\SOFTWARE\\Wow6432Node\\Python\\PythonCore", "python.exe",
  38                     "HKEY_LOCAL_MACHINE\\SOFTWARE\\IronPython", "ipy.exe");
  39
  40   private WinPythonSdkFlavor() {
  41   }
  42
  43   @Override
  44   public Collection<String> suggestHomePaths() {
  45     Set<String> candidates = new TreeSet<String>();
  46     findInCandidatePaths(candidates, "python.exe", "jython.bat", "pypy.exe");
  47     findInstallations(candidates, "python.exe", PythonHelpersLocator.getHelpersRoot().getParent());
  48     return candidates;
  49   }
  50
  51   private static void findInCandidatePaths(Set<String> candidates, String... exe_names) {
  52     for (String name : exe_names) {
  53       findInstallations(candidates, name, "C:\\", "C:\\Program Files\\");
  54       findInPath(candidates, name);
  55       findInRegistry(candidates);
  56     }
  57   }
  58
  59   private static void findInstallations(Set<String> candidates, String exe_name, String... roots) {
  60     for (String root : roots) {
  61       findSubdirInstallations(candidates, root, FileUtil.getNameWithoutExtension(exe_name), exe_name);
  62     }
  63   }
  64
  65   public static void findInPath(Collection<String> candidates, String exeName) {
  66     final String path = System.getenv("PATH");
  67     if (path == null) return;
  68     for (String pathEntry : StringUtil.split(path, ";")) {
  69       if (pathEntry.startsWith("\"") && pathEntry.endsWith("\"")) {
  70         if (pathEntry.length() < 2) continue;
  71         pathEntry = pathEntry.substring(1, pathEntry.length() - 1);
  72       }
  73       File f = new File(pathEntry, exeName);
  74       if (f.exists()) {
  75         candidates.add(FileUtil.toSystemDependentName(f.getPath()));
  76       }
  77     }
  78   }
  79
  80   public static void findInRegistry(Collection<String> candidates) {
  81     for (Map.Entry<String, String> entry : ourRegistryMap.entrySet()) {
  82       final String prefix = entry.getKey();
  83       final String exePath = entry.getValue();
  84       List<String> strings = WindowsRegistryUtil.readRegistryBranch(prefix);
  85       for (String string : strings) {
  86         final String path =
  87           WindowsRegistryUtil.readRegistryDefault(prefix + "\\" + string +
  88                                                   "\\InstallPath");
  89         if (path != null) {
  90           File f = new File(path, exePath);
  91           if (f.exists()) {
  92             candidates.add(FileUtil.toSystemDependentName(f.getPath()));
  93           }
  94         }
  95       }
  96     }
  97   }
  98
  99   private static void findSubdirInstallations(Collection<String> candidates, String rootDir, String dir_prefix, String exe_name) {
 100     VirtualFile rootVDir = LocalFileSystem.getInstance().findFileByPath(rootDir);
 101     if (rootVDir != null) {
 102       if (rootVDir instanceof NewVirtualFile) {
 103         ((NewVirtualFile)rootVDir).markDirty();
 104       }
 105       rootVDir.refresh(true, false);
 106       for (VirtualFile dir : rootVDir.getChildren()) {
 107         if (dir.isDirectory() && dir.getName().toLowerCase().startsWith(dir_prefix)) {
 108           VirtualFile python_exe = dir.findChild(exe_name);
 109           if (python_exe != null) candidates.add(FileUtil.toSystemDependentName(python_exe.getPath()));
 110         }
 111       }
 112     }
 113   }
 114 }