marco.pms.api/Marco.Pms.CacheHelper/SidebarMenu.cs

using Marco.Pms.Model.AppMenu;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Logging;
using MongoDB.Driver;
using MongoDB.Bson;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;

namespace Marco.Pms.CacheHelper
{
    public class SideBarMenu
    {
        private readonly IMongoCollection<MenuSection> _collection;
        private readonly ILogger<SideBarMenu> _logger;

        public SideBarMenu(IConfiguration configuration, ILogger<SideBarMenu> logger)
        {
            _logger = logger;
            var connectionString = configuration["MongoDB:ConnectionMenu"];
            var mongoUrl = new MongoUrl(connectionString);
            var client = new MongoClient(mongoUrl);
            var database = client.GetDatabase(mongoUrl.DatabaseName);
            _collection = database.GetCollection<MenuSection>("Menus");
        }

        public async Task<MenuSection?> CreateMenuSectionAsync(MenuSection section)
        {
            try
            {
                await _collection.InsertOneAsync(section);
                return section;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "Error occurred while adding MenuSection.");
                return null;
            }
        }

        public async Task<MenuItem?> UpdateMenuItemAsync(Guid sectionId, Guid itemId, MenuItem updatedItem)
        {
            try
            {
                var filter = Builders<MenuSection>.Filter.And(
                    Builders<MenuSection>.Filter.Eq(s => s.Id, sectionId),
                    Builders<MenuSection>.Filter.ElemMatch(s => s.Items, i => i.Id == itemId)
                );

                var update = Builders<MenuSection>.Update
                    .Set("Items.$.Text", updatedItem.Text)
                    .Set("Items.$.Icon", updatedItem.Icon)
                    .Set("Items.$.Available", updatedItem.Available)
                    .Set("Items.$.Link", updatedItem.Link);

                var result = await _collection.UpdateOneAsync(filter, update);
                if (result.ModifiedCount > 0)
                {
                    // Re-fetch section and return the updated item
                    var section = await _collection.Find(s => s.Id == sectionId).FirstOrDefaultAsync();
                    return section?.Items.FirstOrDefault(i => i.Id == itemId);
                }

                return null;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "Error updating MenuItem.");
                return null;
            }
        }


        public async Task<SubMenuItem?> UpdateSubmenuItemAsync(Guid sectionId, Guid itemId, Guid subItemId, SubMenuItem updatedSub)
        {
            try
            {
                var filter = Builders<MenuSection>.Filter.Eq(s => s.Id, sectionId);

                var arrayFilters = new List<ArrayFilterDefinition>
        {
            new BsonDocumentArrayFilterDefinition<BsonDocument>(
                new BsonDocument("item._id", itemId.ToString())),
            new BsonDocumentArrayFilterDefinition<BsonDocument>(
                new BsonDocument("sub._id", subItemId.ToString()))
        };

                var update = Builders<MenuSection>.Update
                    .Set("Items.$[item].Submenu.$[sub].Text", updatedSub.Text)
                    .Set("Items.$[item].Submenu.$[sub].Available", updatedSub.Available)
                    .Set("Items.$[item].Submenu.$[sub].Link", updatedSub.Link)
                    .Set("Items.$[item].Submenu.$[sub].PermissionKey", updatedSub.PermissionKey);

                var options = new UpdateOptions { ArrayFilters = arrayFilters };

                var result = await _collection.UpdateOneAsync(filter, update, options);

                if (result.ModifiedCount == 0)
                    return null;

                var updatedSection = await _collection.Find(x => x.Id == sectionId).FirstOrDefaultAsync();

                var subItem = updatedSection?.Items
                    .FirstOrDefault(i => i.Id.ToString() == itemId.ToString())?
                    .Submenu
                    .FirstOrDefault(s => s.Id.ToString() == subItemId.ToString());

                return subItem;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "Error updating SubMenuItem.");
                return null;
            }
        }

    }
}