mixtape/zero.Core/Api/MediaFolderApi.cs

using FluentValidation;
using Raven.Client.Documents;
using Raven.Client.Documents.Linq;
using Raven.Client.Documents.Session;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using zero.Core.Collections;
using zero.Core.Database.Indexes;
using zero.Core.Entities;
using zero.Core.Extensions;
using zero.Core.Validation;

namespace zero.Core.Api
{
  public class MediaFolderApi : BackofficeApi, IMediaFolderApi
  {
    IValidator<MediaFolder> Validator;


    public MediaFolderApi(ICollectionContext store, IValidator<MediaFolder> validator) : base(store)
    {
      Validator = validator;
    }


    /// <inheritdoc />
    public async Task<MediaFolder> GetById(string id)
    {
      return await GetById<MediaFolder>(id);
    }


    /// <inheritdoc />
    public async Task<IList<MediaFolder>> GetAll(string parentId = null)
    {
      return await Session.Query<MediaFolder>()
        .WhereIf(x => x.ParentId == parentId, !parentId.IsNullOrEmpty(), x => x.ParentId == null)
        .OrderByDescending(x => x.Name)
        .ToListAsync();
    }


    /// <inheritdoc />
    public async Task<IList<TreeItem>> GetAllAsTree(string parentId = null, string activeId = null)
    {
      List<TreeItem> items = new List<TreeItem>();
      string[] openIds = new string[0] { };

      IList<MediaFolder> folders = await Session.Query<MediaFolder>()
        .WhereIf(x => x.ParentId == parentId, !parentId.IsNullOrEmpty(), x => x.ParentId == null)
        .OrderByDescending(x => x.CreatedDate).ThenBy(x => x.Name)
        .ToListAsync();


      // get hierarchy so we know if we should set the folder to open
      if (!activeId.IsNullOrEmpty())
      {
        MediaFolder_ByHierarchy.Result result = await Session.Query<MediaFolder_ByHierarchy.Result, MediaFolder_ByHierarchy>()
          .ProjectInto<MediaFolder_ByHierarchy.Result>()
          .Include<MediaFolder_ByHierarchy.Result, MediaFolder>(x => x.Path.Select(p => p.Id))
          .FirstOrDefaultAsync(x => x.Id == activeId);

        if (result != null)
        {
          openIds = result.Path.Select(x => x.Id).ToArray();
        }
      }


      // get children for all folders
      string[] folderIds = folders.Select(x => x.Id).ToArray();

      IList<MediaFolders_WithChildren.Result> children = await Session.Query<MediaFolders_WithChildren.Result, MediaFolders_WithChildren>()
        .ProjectInto<MediaFolders_WithChildren.Result>()
        .Where(x => x.Id.In(folderIds))
        .ToListAsync();


      foreach (MediaFolder folder in folders)
      {
        int childCount = children.Count(x => x.Id == folder.Id);

        items.Add(new TreeItem()
        {
          Id = folder.Id,
          Name = folder.Name,
          HasChildren = childCount > 0,
          ChildCount = childCount,
          ParentId = folder.ParentId,
          Sort = folder.Sort,
          Icon = "fth-folder",
          IsOpen = openIds.Contains(folder.Id),
          IsInactive = !folder.IsActive,
          HasActions = true,
          Modifier = !folder.IsActive ? new TreeItemModifier()
          {
            Icon = "fth-minus-circle color-yellow",
            Name = "Inactive"
          } : null
        });
      } 

      //if (parentId.IsNullOrEmpty())
      //{
      //  items.Add(new TreeItem()
      //  {
      //    Id = "recyclebin",
      //    ParentId = null,
      //    Sort = 99999,
      //    Name = "@recyclebin.name",
      //    Icon = "fth-trash",
      //    HasChildren = false
      //  });
      //}

      return items;
    }


    /// <inheritdoc />
    public async Task<IList<MediaFolder>> GetHierarchy(string id)
    {
      MediaFolder_ByHierarchy.Result result = await Session.Query<MediaFolder_ByHierarchy.Result, MediaFolder_ByHierarchy>()
        .ProjectInto<MediaFolder_ByHierarchy.Result>()
        .Include<MediaFolder_ByHierarchy.Result, MediaFolder>(x => x.Path.Select(p => p.Id))
        .FirstOrDefaultAsync(x => x.Id == id);

      if (result == null)
      {
        return new List<MediaFolder>();
      }

      List<string> ids = result.Path.Select(x => x.Id).ToList();
      ids.Add(id);

      return (await Session.LoadAsync<MediaFolder>(ids)).Select(x => x.Value).ToList();
    }


    /// <inheritdoc />
    public async Task<EntityResult<MediaFolder>> Save(MediaFolder model)
    {
      model.IsActive = true;
      return await SaveModel(model, Validator);
    }


    /// <inheritdoc />
    public async Task<EntityResult<MediaFolder>> Move(string id, string parentId)
    {
      MediaFolder model = await GetById<MediaFolder>(id);
      MediaFolder parent = await GetById<MediaFolder>(parentId);

      if (model == null || (!parentId.IsNullOrEmpty() && parent == null))
      {
        return EntityResult<MediaFolder>.Fail("@errors.idnotfound");
      }

      model.ParentId = parent?.Id;

      return await Save(model);
    }


    /// <inheritdoc />
    public async Task<EntityResult<MediaFolder>> Delete(string id)
    {
      return await DeleteById<MediaFolder>(id);
    }
  }


  public interface IMediaFolderApi : IBackofficeApi
  {
    /// <summary>
    /// Get application by Id
    /// </summary>
    Task<MediaFolder> GetById(string id);

    /// <summary>
    /// Get hierarchy for a folder
    /// </summary>
    Task<IList<MediaFolder>> GetHierarchy(string id);

    /// <summary>
    /// Get all folders with the specified parent or on root
    /// </summary>
    Task<IList<MediaFolder>> GetAll(string parentId = null);

    /// <summary>
    /// Get all folders with the specified parent or on root for tree output
    /// </summary>
    Task<IList<TreeItem>> GetAllAsTree(string parentId = null, string activeId = null);

    /// <summary>
    /// Creates or updates a folder
    /// </summary>
    Task<EntityResult<MediaFolder>> Save(MediaFolder model);

    /// <summary>
    /// Move a folder to a new parent
    /// </summary>
    Task<EntityResult<MediaFolder>> Move(string id, string parentId);

    /// <summary>
    /// Deletes a folder
    /// </summary>
    Task<EntityResult<MediaFolder>> Delete(string id);
  }
}