GR-26405: ZipImporterBuiltins - add missing find_loader

Author: cosminbasca

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/zipimporter/PZipImporter.java

@@ -273,6 +273,22 @@ protected final ModuleInfo getModuleInfo(String fullname) {
         return ModuleInfo.NOT_FOUND;
     }
 
+    @TruffleBoundary
+    protected boolean isDir(String path) {
+        String dirPath = path + separator;
+        for (Object key : files.keys()) {
+            if (key.equals(dirPath)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    @TruffleBoundary
+    protected String getModulePath(String modPath) {
+        return this.archive + separator + modPath;
+    }
+
     /**
      *
      * @param fullname

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/zipimporter/ZipImporterBuiltins.java

@@ -418,6 +418,41 @@ public Object doit(PZipImporter self, String fullname, @SuppressWarnings("unused
 
     }
 
+    @Builtin(name = "find_loader", minNumOfPositionalArgs = 2, maxNumOfPositionalArgs = 3)
+    @TypeSystemReference(PythonArithmeticTypes.class)
+    @GenerateNodeFactory
+    public abstract static class FindLoaderNode extends PythonTernaryBuiltinNode {
+        @Specialization
+        public Object findLoader(PZipImporter self, String fullname, @SuppressWarnings("unused") Object path) {
+            PZipImporter.ModuleInfo mi = self.getModuleInfo(fullname);
+            if (mi != ModuleInfo.NOT_FOUND) {
+                return makeTuple(self, makeList());
+            }
+
+            String modPath = self.makeFilename(fullname);
+            if (self.isDir(modPath)) {
+                return makeTuple(makeList(self.getModulePath(modPath)));
+            }
+            return makeTuple(makeList());
+        }
+
+        private PTuple makeTuple(Object second) {
+            return makeTuple(null, second);
+        }
+
+        private PTuple makeTuple(Object first, Object second) {
+            return factory().createTuple(new Object[]{first != null ? first : PNone.NONE, second});
+        }
+
+        private PList makeList() {
+            return factory().createList();
+        }
+
+        private PList makeList(Object first) {
+            return factory().createList(new Object[]{first});
+        }
+    }
+
     @Builtin(name = "get_code", minNumOfPositionalArgs = 2)
     @TypeSystemReference(PythonArithmeticTypes.class)
     @GenerateNodeFactory