mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-14 03:37:10 +08:00
72 lines
2.4 KiB
C#
72 lines
2.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using Python.Runtime;
|
|
using Wox.Helper;
|
|
using Wox.Plugin;
|
|
using Wox.PluginLoader;
|
|
|
|
namespace Wox.Commands
|
|
{
|
|
public class PluginCommand : BaseCommand
|
|
{
|
|
private string currentPythonModulePath = string.Empty;
|
|
private IntPtr GIL;
|
|
|
|
public override void Dispatch(Query q,bool updateView = true)
|
|
{
|
|
PluginPair thirdPlugin = Plugins.AllPlugins.FirstOrDefault(o => o.Metadata.ActionKeyword == q.ActionName);
|
|
if (thirdPlugin != null && !string.IsNullOrEmpty(thirdPlugin.Metadata.ActionKeyword))
|
|
{
|
|
if (thirdPlugin.Metadata.Language == AllowedLanguage.Python)
|
|
{
|
|
SwitchPythonEnv(thirdPlugin);
|
|
}
|
|
ThreadPool.QueueUserWorkItem(t =>
|
|
{
|
|
try
|
|
{
|
|
List<Result> r = thirdPlugin.Plugin.Query(q);
|
|
r.ForEach(o =>
|
|
{
|
|
o.PluginDirectory = thirdPlugin.Metadata.PluginDirecotry;
|
|
o.OriginQuery = q;
|
|
});
|
|
if(updateView) UpdateResultView(r);
|
|
}
|
|
catch (Exception queryException)
|
|
{
|
|
Log.Error(string.Format("Plugin {0} query failed: {1}", thirdPlugin.Metadata.Name,
|
|
queryException.Message));
|
|
#if (DEBUG)
|
|
{
|
|
throw;
|
|
}
|
|
#endif
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
private void SwitchPythonEnv(PluginPair thirdPlugin)
|
|
{
|
|
if (currentPythonModulePath != thirdPlugin.Metadata.PluginDirecotry)
|
|
{
|
|
currentPythonModulePath = thirdPlugin.Metadata.PluginDirecotry;
|
|
|
|
if (GIL != IntPtr.Zero)
|
|
{
|
|
Runtime.PyEval_RestoreThread(GIL);
|
|
PythonEngine.Shutdown();
|
|
}
|
|
PythonEngine.Initialize();
|
|
IntPtr pyStrPtr = Runtime.PyString_FromString(thirdPlugin.Metadata.PluginDirecotry);
|
|
IntPtr sysDotPath = Runtime.PySys_GetObject("path");
|
|
Runtime.PyList_Append(sysDotPath, pyStrPtr);
|
|
GIL = PythonEngine.BeginAllowThreads();
|
|
}
|
|
}
|
|
}
|
|
}
|