using System;
using System.Collections.Generic;
using System.IO;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using API.Constants;
using API.Data;
using API.DTOs.Font;
using API.Entities.Enums.Font;
using API.Extensions;
using API.Services;
using API.Services.Tasks;
using API.Services.Tasks.Scanner.Parser;
using AutoMapper;
using Kavita.Common;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using MimeTypes;
namespace API.Controllers;
[Authorize]
public class FontController : BaseApiController
{
private readonly IUnitOfWork _unitOfWork;
private readonly IDirectoryService _directoryService;
private readonly IFontService _fontService;
private readonly IMapper _mapper;
private readonly ILocalizationService _localizationService;
private readonly Regex _fontFileExtensionRegex = new(Parser.FontFileExtensions, RegexOptions.IgnoreCase, Parser.RegexTimeout);
public FontController(IUnitOfWork unitOfWork, IDirectoryService directoryService,
IFontService fontService, IMapper mapper, ILocalizationService localizationService)
{
_unitOfWork = unitOfWork;
_directoryService = directoryService;
_fontService = fontService;
_mapper = mapper;
_localizationService = localizationService;
}
///
/// List out the fonts
///
///
[ResponseCache(CacheProfileName = ResponseCacheProfiles.TenMinute)]
[HttpGet("all")]
public async Task>> GetFonts()
{
return Ok(await _unitOfWork.EpubFontRepository.GetFontDtosAsync());
}
///
/// Returns a font file
///
///
///
///
[HttpGet]
[AllowAnonymous]
public async Task GetFont(int fontId, string apiKey)
{
var userId = await _unitOfWork.UserRepository.GetUserIdByApiKeyAsync(apiKey);
if (userId == 0) return BadRequest();
var font = await _unitOfWork.EpubFontRepository.GetFontAsync(fontId);
if (font == null) return NotFound();
if (font.Provider == FontProvider.System) return BadRequest("System provided fonts are not loaded by API");
var contentType = MimeTypeMap.GetMimeType(Path.GetExtension(font.FileName));
var path = Path.Join(_directoryService.EpubFontDirectory, font.FileName);
return PhysicalFile(path, contentType, true);
}
///
/// Removes a font from the system
///
///
/// If the font is in use by other users and an admin wants it deleted, they must confirm to force delete it. This is prompted in the UI.
///
[HttpDelete]
public async Task DeleteFont(int fontId, bool force = false)
{
if (User.IsInRole(PolicyConstants.ReadOnlyRole)) return BadRequest(await _localizationService.Translate(User.GetUserId(), "denied"));
var forceDelete = User.IsInRole(PolicyConstants.AdminRole) && force;
var fontInUse = await _fontService.IsFontInUse(fontId);
if (!fontInUse || forceDelete)
{
await _fontService.Delete(fontId);
}
return Ok();
}
///
/// Returns if the given font is in use by any other user. System provided fonts will always return true.
///
///
///
[HttpGet("in-use")]
public async Task> IsFontInUse(int fontId)
{
return Ok(await _fontService.IsFontInUse(fontId));
}
///
/// Manual upload
///
///
///
[HttpPost("upload")]
public async Task> UploadFont(IFormFile formFile)
{
if (User.IsInRole(PolicyConstants.ReadOnlyRole)) return BadRequest(await _localizationService.Translate(User.GetUserId(), "denied"));
if (!_fontFileExtensionRegex.IsMatch(Path.GetExtension(formFile.FileName))) return BadRequest("Invalid file");
if (formFile.FileName.Contains("..")) return BadRequest("Invalid file");
var tempFile = await UploadToTemp(formFile);
var font = await _fontService.CreateFontFromFileAsync(tempFile);
return Ok(_mapper.Map(font));
}
[HttpPost("upload-by-url")]
public async Task UploadFontByUrl([FromQuery]string url)
{
if (User.IsInRole(PolicyConstants.ReadOnlyRole)) return BadRequest(await _localizationService.Translate(User.GetUserId(), "denied"));
// Validate url
try
{
var font = await _fontService.CreateFontFromUrl(url);
return Ok(_mapper.Map(font));
}
catch (KavitaException ex)
{
return BadRequest(_localizationService.Translate(User.GetUserId(), ex.Message));
}
}
private async Task UploadToTemp(IFormFile file)
{
var outputFile = Path.Join(_directoryService.TempDirectory, file.FileName);
await using var stream = System.IO.File.Create(outputFile);
await file.CopyToAsync(stream);
stream.Close();
return outputFile;
}
}