Added load resource stream hook

amos402 · amos402 · commit cd73445835f1 · 2018-06-29T15:31:44.000+08:00
diff --git a/src/runtime/pythonengine.cs b/src/runtime/pythonengine.cs
@@ -1,4 +1,4 @@
-﻿using System;
+using System;
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
@@ -7,11 +7,18 @@
 
 namespace Python.Runtime
 {
+    public class ResolveStreamResouceArgs : EventArgs
+    {
+        public string Name { get; set; }
+    }
+
     /// <summary>
     /// This class provides the public interface of the Python runtime.
     /// </summary>
     public class PythonEngine : IDisposable
     {
+        public static event Func<ResolveStreamResouceArgs, Stream> StreamResouceResolve;
+
         private static DelegateManager delegateManager;
         private static bool initialized;
         private static IntPtr _pythonHome = IntPtr.Zero;
@@ -202,8 +209,7 @@ public static void Initialize(IEnumerable<string> args, bool setSysArgv = true)
                     IntPtr builtins = Runtime.PyEval_GetBuiltins();
                     Runtime.PyDict_SetItemString(module_globals, "__builtins__", builtins);
 
-                    Assembly assembly = Assembly.GetExecutingAssembly();
-                    using (Stream stream = assembly.GetManifestResourceStream("clr.py"))
+                    using (Stream stream = LoadResource("clr.py"))
                     using (var reader = new StreamReader(stream))
                     {
                         // add the contents of clr.py to the module
@@ -534,6 +540,26 @@ internal static PyObject RunString(string code, IntPtr? globals, IntPtr? locals,
                 }
             }
         }
+
+        private static Stream LoadResource(string name)
+        {
+            var args = new ResolveStreamResouceArgs()
+            {
+                Name = name
+            };
+            Stream stream;
+            foreach (Func<ResolveStreamResouceArgs, Stream> resolver in StreamResouceResolve.GetInvocationList())
+            {
+                stream = resolver(args);
+                if (stream != null)
+                {
+                    return stream;
+                }
+            }
+            Assembly assembly = Assembly.GetExecutingAssembly();
+            stream = assembly.GetManifestResourceStream("clr.py");
+            return stream;
+        }
     }
 
     public enum RunFlagType
