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 isValid = await IsValidUser();
        if (!isValid)
        {
            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,
            },
            new PluginMenuDef("System", weight: 30)
            { 
                IsHeader = true,
                Roles = new List<string> { UserRole.Root, UserRole.Admin }
            },
            new PluginMenuDef("Plugins", link: "page/plugin", icon: "bx bx-plug", weight: 31)
            {
                Roles = new List<string> {  UserRole.Root, UserRole.Admin }
            },
            new PluginMenuDef("Settings", link: "page/setting", icon: "bx bx-cog", weight: 32)
            {
                Roles = new List<string> {  UserRole.Root, UserRole.Admin }
            },
            new PluginMenuDef("Roles", link: "page/roles", icon: "bx bx-group", weight: 33)
            {
                Roles = new List<string> {  UserRole.Root, UserRole.Admin }
            },
            new PluginMenuDef("Users", link: "page/users", icon: "bx bx-user", weight: 34)
            {
                Roles = new List<string> {  UserRole.Root, UserRole.Admin }
            }
        };

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

        var userService = _services.GetRequiredService<IUserService>();
        var user = await userService.GetUser(_user.Id);
        menu = loader.GetPluginMenuByRoles(menu, user?.Role);
        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);
    }

    private async Task<bool> IsValidUser()
    {
        var userService = _services.GetRequiredService<IUserService>();
        return await userService.IsAdminUser(_user.Id);
    }
}