mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2025-08-21 16:52:56 +00:00
Ensure deployment of internal plugins
This commit is contained in:
parent
8a4031ed76
commit
c679e17784
@ -1,6 +1,7 @@
|
|||||||
using AIStudio.Dialogs;
|
using AIStudio.Dialogs;
|
||||||
using AIStudio.Settings;
|
using AIStudio.Settings;
|
||||||
using AIStudio.Settings.DataModel;
|
using AIStudio.Settings.DataModel;
|
||||||
|
using AIStudio.Tools.PluginSystem;
|
||||||
using AIStudio.Tools.Rust;
|
using AIStudio.Tools.Rust;
|
||||||
using AIStudio.Tools.Services;
|
using AIStudio.Tools.Services;
|
||||||
|
|
||||||
@ -81,6 +82,9 @@ public partial class MainLayout : LayoutComponentBase, IMessageBusReceiver, IDis
|
|||||||
// Ensure that all settings are loaded:
|
// Ensure that all settings are loaded:
|
||||||
await this.SettingsManager.LoadSettings();
|
await this.SettingsManager.LoadSettings();
|
||||||
|
|
||||||
|
// Ensure that all internal plugins are present:
|
||||||
|
await PluginFactory.EnsureInternalPlugins();
|
||||||
|
|
||||||
// Register this component with the message bus:
|
// Register this component with the message bus:
|
||||||
this.MessageBus.RegisterComponent(this);
|
this.MessageBus.RegisterComponent(this);
|
||||||
this.MessageBus.ApplyFilters(this, [], [ Event.UPDATE_AVAILABLE, Event.USER_SEARCH_FOR_UPDATE, Event.CONFIGURATION_CHANGED, Event.COLOR_THEME_CHANGED ]);
|
this.MessageBus.ApplyFilters(this, [], [ Event.UPDATE_AVAILABLE, Event.USER_SEARCH_FOR_UPDATE, Event.CONFIGURATION_CHANGED, Event.COLOR_THEME_CHANGED ]);
|
||||||
|
@ -0,0 +1,7 @@
|
|||||||
|
namespace AIStudio.Tools.PluginSystem;
|
||||||
|
|
||||||
|
public enum InternalPlugin
|
||||||
|
{
|
||||||
|
LANGUAGE_EN_US,
|
||||||
|
LANGUAGE_DE_DE,
|
||||||
|
}
|
@ -0,0 +1,8 @@
|
|||||||
|
namespace AIStudio.Tools.PluginSystem;
|
||||||
|
|
||||||
|
public readonly record struct InternalPluginData(PluginType Type, Guid Id, string ShortName)
|
||||||
|
{
|
||||||
|
public string ResourcePath => $"{this.Type.GetDirectory()}/{this.ShortName.ToLowerInvariant()}-{this.Id}";
|
||||||
|
|
||||||
|
public string ResourceName => $"{this.ShortName.ToLowerInvariant()}-{this.Id}";
|
||||||
|
}
|
@ -0,0 +1,12 @@
|
|||||||
|
namespace AIStudio.Tools.PluginSystem;
|
||||||
|
|
||||||
|
public static class InternalPluginExtensions
|
||||||
|
{
|
||||||
|
public static InternalPluginData MetaData(this InternalPlugin plugin) => plugin switch
|
||||||
|
{
|
||||||
|
InternalPlugin.LANGUAGE_EN_US => new (PluginType.LANGUAGE, new("97dfb1ba-50c4-4440-8dfa-6575daf543c8"), "en-us"),
|
||||||
|
InternalPlugin.LANGUAGE_DE_DE => new(PluginType.LANGUAGE, new("43065dbc-78d0-45b7-92be-f14c2926e2dc"), "de-de"),
|
||||||
|
|
||||||
|
_ => new InternalPluginData(PluginType.NONE, Guid.Empty, "unknown")
|
||||||
|
};
|
||||||
|
}
|
@ -1,11 +1,90 @@
|
|||||||
using System.Text;
|
using System.Reflection;
|
||||||
|
|
||||||
|
using AIStudio.Settings;
|
||||||
|
|
||||||
using Lua;
|
using Lua;
|
||||||
|
|
||||||
|
using Microsoft.Extensions.FileProviders;
|
||||||
|
|
||||||
namespace AIStudio.Tools.PluginSystem;
|
namespace AIStudio.Tools.PluginSystem;
|
||||||
|
|
||||||
public static class PluginFactory
|
public static class PluginFactory
|
||||||
{
|
{
|
||||||
|
private static readonly ILogger LOG = Program.LOGGER_FACTORY.CreateLogger("PluginFactory");
|
||||||
|
private static readonly string DATA_DIR = SettingsManager.DataDirectory!;
|
||||||
|
|
||||||
|
public static async Task EnsureInternalPlugins()
|
||||||
|
{
|
||||||
|
LOG.LogInformation("Start ensuring internal plugins.");
|
||||||
|
foreach (var plugin in Enum.GetValues<InternalPlugin>())
|
||||||
|
{
|
||||||
|
LOG.LogInformation($"Ensure plugin: {plugin}");
|
||||||
|
await EnsurePlugin(plugin);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static async Task EnsurePlugin(InternalPlugin plugin)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
#if DEBUG
|
||||||
|
var basePath = Path.Join(Environment.CurrentDirectory, "Plugins");
|
||||||
|
var resourceFileProvider = new PhysicalFileProvider(basePath);
|
||||||
|
#else
|
||||||
|
var resourceFileProvider = new ManifestEmbeddedFileProvider(Assembly.GetAssembly(type: typeof(Program))!, "Plugins");
|
||||||
|
#endif
|
||||||
|
|
||||||
|
var metaData = plugin.MetaData();
|
||||||
|
var mainResourcePath = $"{metaData.ResourcePath}/plugin.lua";
|
||||||
|
var resourceInfo = resourceFileProvider.GetFileInfo(mainResourcePath);
|
||||||
|
|
||||||
|
if(!resourceInfo.Exists)
|
||||||
|
{
|
||||||
|
LOG.LogError($"The plugin {plugin} does not exist. This should not happen, since the plugin is an integral part of AI Studio.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure that the additional resources exist:
|
||||||
|
foreach (var content in resourceFileProvider.GetDirectoryContents(metaData.ResourcePath))
|
||||||
|
{
|
||||||
|
if(content.IsDirectory)
|
||||||
|
{
|
||||||
|
LOG.LogError("The plugin contains a directory. This is not allowed.");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
await CopyPluginFile(content, metaData);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
{
|
||||||
|
LOG.LogError($"Was not able to ensure the plugin: {plugin}");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static async Task CopyPluginFile(IFileInfo resourceInfo, InternalPluginData metaData)
|
||||||
|
{
|
||||||
|
await using var inputStream = resourceInfo.CreateReadStream();
|
||||||
|
|
||||||
|
var pluginsRoot = Path.Join(DATA_DIR, "plugins");
|
||||||
|
var pluginTypeBasePath = Path.Join(pluginsRoot, metaData.Type.GetDirectory());
|
||||||
|
|
||||||
|
if (!Directory.Exists(pluginsRoot))
|
||||||
|
Directory.CreateDirectory(pluginsRoot);
|
||||||
|
|
||||||
|
if (!Directory.Exists(pluginTypeBasePath))
|
||||||
|
Directory.CreateDirectory(pluginTypeBasePath);
|
||||||
|
|
||||||
|
var pluginPath = Path.Join(pluginTypeBasePath, metaData.ResourceName);
|
||||||
|
if (!Directory.Exists(pluginPath))
|
||||||
|
Directory.CreateDirectory(pluginPath);
|
||||||
|
|
||||||
|
var pluginFilePath = Path.Join(pluginPath, resourceInfo.Name);
|
||||||
|
|
||||||
|
await using var outputStream = File.Create(pluginFilePath);
|
||||||
|
await inputStream.CopyToAsync(outputStream);
|
||||||
|
}
|
||||||
|
|
||||||
public static async Task LoadAll()
|
public static async Task LoadAll()
|
||||||
{
|
{
|
||||||
|
|
||||||
|
@ -11,4 +11,14 @@ public static class PluginTypeExtensions
|
|||||||
|
|
||||||
_ => "Unknown plugin type",
|
_ => "Unknown plugin type",
|
||||||
};
|
};
|
||||||
|
|
||||||
|
public static string GetDirectory(this PluginType type) => type switch
|
||||||
|
{
|
||||||
|
PluginType.LANGUAGE => "languages",
|
||||||
|
PluginType.ASSISTANT => "assistants",
|
||||||
|
PluginType.CONFIGURATION => "configurations",
|
||||||
|
PluginType.THEME => "themes",
|
||||||
|
|
||||||
|
_ => "unknown",
|
||||||
|
};
|
||||||
}
|
}
|
Loading…
Reference in New Issue
Block a user