PowerToys/WinAlfred/PluginLoader/PythonPluginWrapper.cs

using System;
using System.Collections.Generic;
using System.Runtime.InteropServices;
using System.Windows.Documents;
using Newtonsoft.Json;
using Python.Runtime;
using WinAlfred.Plugin;

namespace WinAlfred.PluginLoader
{
    public class PythonPluginWrapper : IPlugin
    {
        private PluginMetadata metadata;

        public PythonPluginWrapper(PluginMetadata metadata)
        {
            this.metadata = metadata;
        }

        public List<Result> Query(Query query)
        {
            try
            {
                string s = InvokeFunc(metadata.PluginDirecotry, metadata.ExecuteFileName.Replace(".py", ""),"query",query.RawQuery);
                if (string.IsNullOrEmpty(s))
                {
                    return new List<Result>();
                }

                List<PythonResult> o = JsonConvert.DeserializeObject<List<PythonResult>>(s);
                List<Result> r = new List<Result>();
                foreach (PythonResult pythonResult in o)
                {
                    PythonResult ps = pythonResult;
                    if (!string.IsNullOrEmpty(ps.ActionName))
                    {
                        ps.Action = () => InvokeFunc(metadata.PluginDirecotry, metadata.ExecuteFileName.Replace(".py", ""), ps.ActionName, ps.ActionPara);
                    }
                    r.Add(ps);
                }
                return r;
            }
            catch (Exception)
            {

                throw;
            }

        }

        private string InvokeFunc(string path, string moduleName,string func, string para)
        {
            IntPtr gs = PythonEngine.AcquireLock();

            IntPtr pyStrPtr = Runtime.PyString_FromString(path);
            IntPtr SysDotPath = Runtime.PySys_GetObject("path");
            Runtime.PyList_Append(SysDotPath, pyStrPtr);

            PyObject module = PythonEngine.ImportModule(moduleName);
            module = PythonEngine.ReloadModule(module);
            PyObject res = module.InvokeMethod(func, new PyString(para));
            string json = Runtime.GetManagedString(res.Handle);

            PythonEngine.ReleaseLock(gs);

            return json;
        }

        public void Init(PluginInitContext context)
        {
           
        }
    }
}