BotSharp/src/Infrastructure/BotSharp.OpenAPI/Controllers/PluginController.cs

using BotSharp.Abstraction.Plugins.Models;
using BotSharp.Abstraction.Users.Enums;
using BotSharp.Core.Plugins;

namespace BotSharp.OpenAPI.Controllers;

[Authorize]
[ApiController]
public class PluginController : ControllerBase
{
    private readonly IServiceProvider _services;
    private readonly IUserIdentity _user;
    private readonly PluginSettings _settings;

    public PluginController(IServiceProvider services, IUserIdentity user, PluginSettings settings)
    {
        _services = services;
        _user = user;
        _settings = settings;
    }

    [HttpGet("/plugins")]
    public async Task<PagedItems<PluginDef>> GetPlugins([FromQuery] PluginFilter filter)
    {
        var userService = _services.GetRequiredService<IUserService>();
        var user = await userService.GetUser(_user.Id);
        if (user?.Role != UserRole.Admin)
        {
            return new PagedItems<PluginDef>();
        }

        var loader = _services.GetRequiredService<PluginLoader>();
        return loader.GetPagedPlugins(_services, filter);
    }

    [HttpGet("/plugin/menu")]
    public async Task<List<PluginMenuDef>> GetPluginMenu()
    {
        var menu = new List<PluginMenuDef>
        {
            new PluginMenuDef("Apps", weight: 5)
            {
                IsHeader = true,
            }
        };

        var userService = _services.GetRequiredService<IUserService>();
        var user = await userService.GetUser(_user.Id);
        if (user?.Role == UserRole.Admin)
        {
            menu.Add(new PluginMenuDef("System", weight: 30) { IsHeader = true });
            menu.Add(new PluginMenuDef("Plugins", link: "page/plugin", icon: "bx bx-plug", weight: 31));
            menu.Add(new PluginMenuDef("Settings", link: "page/setting", icon: "bx bx-cog", weight: 32));
        }

        var loader = _services.GetRequiredService<PluginLoader>();
        foreach (var plugin in loader.GetPlugins(_services))
        {
            if (!plugin.Enabled)
            {
                continue;
            }
            plugin.Module.AttachMenu(menu);
        }
        menu = menu.OrderBy(x => x.Weight).ToList();
        return menu;
    }

    [HttpPost("/plugin/{id}/install")]
    public PluginDef InstallPlugin([FromRoute] string id)
    {
        var loader = _services.GetRequiredService<PluginLoader>();
        return loader.UpdatePluginStatus(_services, id, true);
    }

    [HttpPost("/plugin/{id}/remove")]
    public PluginDef RemovePluginStats([FromRoute] string id)
    {
        var loader = _services.GetRequiredService<PluginLoader>();
        return loader.UpdatePluginStatus(_services, id, false);
    }
}