mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2025-04-28 07:59:47 +00:00
Add hot reloading support for plugins
This commit is contained in:
parent
1c7188aa7f
commit
ad4d42f554
@ -94,6 +94,9 @@ public partial class MainLayout : LayoutComponentBase, IMessageBusReceiver, IDis
|
|||||||
// Load (but not start) all plugins, without waiting for them:
|
// Load (but not start) all plugins, without waiting for them:
|
||||||
var pluginLoadingTimeout = new CancellationTokenSource(TimeSpan.FromSeconds(5));
|
var pluginLoadingTimeout = new CancellationTokenSource(TimeSpan.FromSeconds(5));
|
||||||
_ = PluginFactory.LoadAll(pluginLoadingTimeout.Token);
|
_ = PluginFactory.LoadAll(pluginLoadingTimeout.Token);
|
||||||
|
|
||||||
|
// Set up hot reloading for plugins:
|
||||||
|
PluginFactory.SetUpHotReloading();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Register this component with the message bus:
|
// Register this component with the message bus:
|
||||||
|
@ -5,7 +5,7 @@ using Microsoft.AspNetCore.Components;
|
|||||||
|
|
||||||
namespace AIStudio.Pages;
|
namespace AIStudio.Pages;
|
||||||
|
|
||||||
public partial class Plugins : ComponentBase
|
public partial class Plugins : ComponentBase, IMessageBusReceiver
|
||||||
{
|
{
|
||||||
private const string GROUP_ENABLED = "Enabled";
|
private const string GROUP_ENABLED = "Enabled";
|
||||||
private const string GROUP_DISABLED = "Disabled";
|
private const string GROUP_DISABLED = "Disabled";
|
||||||
@ -23,6 +23,9 @@ public partial class Plugins : ComponentBase
|
|||||||
|
|
||||||
protected override async Task OnInitializedAsync()
|
protected override async Task OnInitializedAsync()
|
||||||
{
|
{
|
||||||
|
this.MessageBus.RegisterComponent(this);
|
||||||
|
this.MessageBus.ApplyFilters(this, [], [ Event.PLUGINS_RELOADED ]);
|
||||||
|
|
||||||
this.groupConfig = new TableGroupDefinition<IPluginMetadata>
|
this.groupConfig = new TableGroupDefinition<IPluginMetadata>
|
||||||
{
|
{
|
||||||
Expandable = true,
|
Expandable = true,
|
||||||
@ -53,4 +56,27 @@ public partial class Plugins : ComponentBase
|
|||||||
await this.SettingsManager.StoreSettings();
|
await this.SettingsManager.StoreSettings();
|
||||||
await this.MessageBus.SendMessage<bool>(this, Event.CONFIGURATION_CHANGED);
|
await this.MessageBus.SendMessage<bool>(this, Event.CONFIGURATION_CHANGED);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#region Implementation of IMessageBusReceiver
|
||||||
|
|
||||||
|
public string ComponentName => nameof(Plugins);
|
||||||
|
|
||||||
|
public Task ProcessMessage<T>(ComponentBase? sendingComponent, Event triggeredEvent, T? data)
|
||||||
|
{
|
||||||
|
switch (triggeredEvent)
|
||||||
|
{
|
||||||
|
case Event.PLUGINS_RELOADED:
|
||||||
|
this.InvokeAsync(this.StateHasChanged);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Task<TResult?> ProcessMessageWithResult<TPayload, TResult>(ComponentBase? sendingComponent, Event triggeredEvent, TPayload? data)
|
||||||
|
{
|
||||||
|
return Task.FromResult<TResult?>(default);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endregion
|
||||||
}
|
}
|
@ -1,5 +1,6 @@
|
|||||||
using AIStudio.Agents;
|
using AIStudio.Agents;
|
||||||
using AIStudio.Settings;
|
using AIStudio.Settings;
|
||||||
|
using AIStudio.Tools.PluginSystem;
|
||||||
using AIStudio.Tools.Services;
|
using AIStudio.Tools.Services;
|
||||||
|
|
||||||
using Microsoft.AspNetCore.Server.Kestrel.Core;
|
using Microsoft.AspNetCore.Server.Kestrel.Core;
|
||||||
@ -209,6 +210,7 @@ internal sealed class Program
|
|||||||
await serverTask;
|
await serverTask;
|
||||||
|
|
||||||
RUST_SERVICE.Dispose();
|
RUST_SERVICE.Dispose();
|
||||||
|
PluginFactory.Dispose();
|
||||||
programLogger.LogInformation("The AI Studio server was stopped.");
|
programLogger.LogInformation("The AI Studio server was stopped.");
|
||||||
}
|
}
|
||||||
}
|
}
|
@ -8,6 +8,7 @@ public enum Event
|
|||||||
STATE_HAS_CHANGED,
|
STATE_HAS_CHANGED,
|
||||||
CONFIGURATION_CHANGED,
|
CONFIGURATION_CHANGED,
|
||||||
COLOR_THEME_CHANGED,
|
COLOR_THEME_CHANGED,
|
||||||
|
PLUGINS_RELOADED,
|
||||||
|
|
||||||
// Update events:
|
// Update events:
|
||||||
USER_SEARCH_FOR_UPDATE,
|
USER_SEARCH_FOR_UPDATE,
|
||||||
|
@ -0,0 +1,33 @@
|
|||||||
|
namespace AIStudio.Tools.PluginSystem;
|
||||||
|
|
||||||
|
public static partial class PluginFactory
|
||||||
|
{
|
||||||
|
public static void SetUpHotReloading()
|
||||||
|
{
|
||||||
|
LOG.LogInformation($"Start hot reloading plugins for path '{HOT_RELOAD_WATCHER.Path}'.");
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var messageBus = Program.SERVICE_PROVIDER.GetRequiredService<MessageBus>();
|
||||||
|
|
||||||
|
HOT_RELOAD_WATCHER.IncludeSubdirectories = true;
|
||||||
|
HOT_RELOAD_WATCHER.NotifyFilter = NotifyFilters.LastWrite | NotifyFilters.FileName;
|
||||||
|
HOT_RELOAD_WATCHER.Filter = "*.lua";
|
||||||
|
HOT_RELOAD_WATCHER.Changed += async (_, args) =>
|
||||||
|
{
|
||||||
|
LOG.LogInformation($"File changed: {args.FullPath}");
|
||||||
|
await LoadAll();
|
||||||
|
await messageBus.SendMessage<bool>(null, Event.PLUGINS_RELOADED);
|
||||||
|
};
|
||||||
|
|
||||||
|
HOT_RELOAD_WATCHER.EnableRaisingEvents = true;
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
LOG.LogError(e, "Error while setting up hot reloading.");
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
LOG.LogInformation("Hot reloading plugins set up.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -16,6 +16,8 @@ public static partial class PluginFactory
|
|||||||
private static readonly string PLUGINS_ROOT = Path.Join(DATA_DIR, "plugins");
|
private static readonly string PLUGINS_ROOT = Path.Join(DATA_DIR, "plugins");
|
||||||
|
|
||||||
private static readonly string INTERNAL_PLUGINS_ROOT = Path.Join(PLUGINS_ROOT, ".internal");
|
private static readonly string INTERNAL_PLUGINS_ROOT = Path.Join(PLUGINS_ROOT, ".internal");
|
||||||
|
|
||||||
|
private static readonly FileSystemWatcher HOT_RELOAD_WATCHER = new(PLUGINS_ROOT);
|
||||||
|
|
||||||
private static readonly List<IPluginMetadata> AVAILABLE_PLUGINS = [];
|
private static readonly List<IPluginMetadata> AVAILABLE_PLUGINS = [];
|
||||||
|
|
||||||
@ -138,4 +140,9 @@ public static partial class PluginFactory
|
|||||||
_ => new NoPlugin("This plugin type is not supported yet. Please try again with a future version of AI Studio.")
|
_ => new NoPlugin("This plugin type is not supported yet. Please try again with a future version of AI Studio.")
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static void Dispose()
|
||||||
|
{
|
||||||
|
HOT_RELOAD_WATCHER.Dispose();
|
||||||
|
}
|
||||||
}
|
}
|
@ -2,5 +2,6 @@
|
|||||||
- Added a feature flag for the plugin system. This flag is disabled by default and can be enabled inside the app settings. Please note that this feature is still in development; there are no plugins available yet.
|
- Added a feature flag for the plugin system. This flag is disabled by default and can be enabled inside the app settings. Please note that this feature is still in development; there are no plugins available yet.
|
||||||
- Added the Lua library we use for the plugin system to the about page.
|
- Added the Lua library we use for the plugin system to the about page.
|
||||||
- Added the plugin overview page. This page shows all installed plugins and allows you to enable or disable them. It is only available when the plugin preview feature is enabled.
|
- Added the plugin overview page. This page shows all installed plugins and allows you to enable or disable them. It is only available when the plugin preview feature is enabled.
|
||||||
|
- Added hot reloading for plugins. When any plugin is changed, the app will automatically reload the plugin without needing to restart the app.
|
||||||
- Fixed the preview tooltip component not showing the correct position when used inside a scrollable container.
|
- Fixed the preview tooltip component not showing the correct position when used inside a scrollable container.
|
||||||
- Upgraded to Rust 1.85.1
|
- Upgraded to Rust 1.85.1
|
Loading…
Reference in New Issue
Block a user