[GR-19718] Add an import hook to patch pip so we can apply patches to packages

PullRequest: graalpython/738

Author: timfel

graalpython/com.oracle.graal.python.shell/src/com/oracle/graal/python/shell/GraalPythonMain.java

@@ -31,14 +31,17 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.lang.management.ManagementFactory;
+import java.nio.file.Files;
 import java.nio.file.NoSuchFileException;
+import java.nio.file.Paths;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.ListIterator;
 import java.util.Map;
 import java.util.Set;
 
 import com.oracle.truffle.llvm.toolchain.launchers.common.Driver;
+
 import org.graalvm.launcher.AbstractLanguageLauncher;
 import org.graalvm.nativeimage.ImageInfo;
 import org.graalvm.nativeimage.ProcessProperties;
@@ -482,7 +485,18 @@ private void evalNonInteractive(Context context) throws IOException {
             src = Source.newBuilder(getLanguageId(), commandString, "<string>").build();
         } else {
             assert inputFile != null;
-            src = Source.newBuilder(getLanguageId(), new File(inputFile)).mimeType(MIME_TYPE).build();
+            String mimeType = "";
+            try {
+                mimeType = Files.probeContentType(Paths.get(inputFile));
+            } catch (IOException e) {
+            }
+            File f = new File(inputFile);
+            if (f.isFile() && (mimeType == null || !mimeType.equals("application/zip"))) {
+                src = Source.newBuilder(getLanguageId(), f).mimeType(MIME_TYPE).build();
+            } else {
+                String runMod = String.format("import sys; sys.path.insert(0, '%s'); import runpy; runpy._run_module_as_main('__main__', False)", inputFile);
+                src = Source.newBuilder(getLanguageId(), runMod, "<string>").build();
+            }
         }
         context.eval(src);
     }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/Python3Core.java

@@ -249,6 +249,7 @@ private static final String[] initializeCoreFiles() {
                         "pwd",
                         "resource",
                         "_contextvars",
+                        "pip_hook",
                         "_lzma"));
         // must be last
         coreFiles.add("final_patches");