mixtape/zero.Web/Controllers/MediaController.cs

using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.StaticFiles;
using Microsoft.Net.Http.Headers;
using System;
using System.Collections.Generic;
using System.IO;
using System.Net;
using System.Net.Http;
using System.Threading.Tasks;
using zero.Core;
using zero.Core.Api;
using zero.Core.Collections;
using zero.Core.Entities;
using zero.Core.Identity;
using zero.Web.Models;

namespace zero.Web.Controllers
{
  [ZeroAuthorize(Permissions.Sections.Media, PermissionsValue.True)]
  public class MediaController : BackofficeCollectionController<Media, IMediaCollection>
  {
    IMediaFolderApi MediaFolderApi;
    IPaths Paths;

    public MediaController(IMediaCollection collection, IMediaFolderApi mediaFolderApi, IPaths paths) : base(collection)
    {
      MediaFolderApi = mediaFolderApi;
      Paths = paths;
      PreviewTransform = (item, model) => model.Icon = "fth-globe";
    }


    public async Task<ListResult<MediaListItem>> GetListByQuery([FromQuery] MediaListItemQuery query)
    {
      query.IncludeInactive = true;
      return await Collection.GetListByQuery(query);
    }


    [HttpPost]
    public async Task<EntityResult<Media>> Upload(IFormFile file, [FromForm] string folderId) => await Collection.Save(await Collection.Upload(file, folderId));

    [HttpPost]
    public async Task<Media> UploadTemporary(IFormFile file, [FromForm] string folderId) => await Collection.Upload(file, folderId);

    [HttpPost]
    public async Task<EntityResult<Media>> Move([FromBody] ActionCopyModel model) => await Collection.Move(model.Id, model.DestinationId);


    public async Task<MediaListResultModel> GetAll([FromQuery] MediaListQuery query)
    {
      query.IncludeInactive = true;
      ListResult<MediaListModel> items = (await Collection.GetByQuery(query)).MapTo(x => new MediaListModel()
      {
        Id = x.Id,
        IsFolder = false,
        Name = x.Name,
        Size = x.Size,
        Source = x.PreviewSource ?? x.Source,
        Type = x.Type
      });

      IList<MediaFolder> hierarchy = null;
      MediaFolder folder = null;

      if (!String.IsNullOrEmpty(query.FolderId))
      {
        folder = await MediaFolderApi.GetById(query.FolderId);
        hierarchy = await MediaFolderApi.GetHierarchy(query.FolderId);
      }   

      return new MediaListResultModel(items, null, folder, hierarchy);
    }


    [HttpGet]
    public async Task<IActionResult> GetSource([FromQuery] string id, [FromQuery] MediaSourceSize size = MediaSourceSize.Original)
    {
      string path = await Collection.GetSourceById(id, size);

      if (path == null)
      {
        return EmptyImage();
      }

      if (path.StartsWith("url://"))
      {
        path = path.Substring(6);
      }

      var provider = new FileExtensionContentTypeProvider();
      string contentType;
      if (path == null || !provider.TryGetContentType(Path.GetFileName(path), out contentType))
      {
        contentType = "application/octet-stream";
      }

      if (path.StartsWith("http"))
      {
        try
        {
          using HttpClient http = new();
          var content = await http.GetByteArrayAsync(path);
          return File(content, contentType, DateTimeOffset.Now, EntityTagHeaderValue.Any);
        }
        catch
        {
          return EmptyImage();
        }
      }

      string fullPath = Path.Combine(Paths.WebRoot, path?.Trim('/') ?? String.Empty);

      if (path == null || !System.IO.File.Exists(fullPath))
      {
        return EmptyImage();
      }

      return File(path, contentType, DateTimeOffset.Now, EntityTagHeaderValue.Any);
    }


    IActionResult EmptyImage()
    {
      string path = Paths.Map("assets", "empty.png");

      if (!System.IO.File.Exists(path))
      {
        return NotFound();
      }

      return File("assets/empty.png", "image/png", DateTimeOffset.Now, EntityTagHeaderValue.Any);
    }
  }
}