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; } }