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.Middleware; 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.GetUserIdByAuthKeyAsync(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] [DisallowRole(PolicyConstants.ReadOnlyRole)] public async Task DeleteFont(int fontId, bool force = false) { 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")] [DisallowRole(PolicyConstants.ReadOnlyRole)] public async Task> UploadFont(IFormFile formFile) { 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")] [DisallowRole(PolicyConstants.ReadOnlyRole)] public async Task UploadFontByUrl([FromQuery]string url) { // Validate url try { var font = await _fontService.CreateFontFromUrl(url); return Ok(_mapper.Map(font)); } catch (KavitaException ex) { return BadRequest(_localizationService.Translate(UserId, 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; } }