using System;
using System.Collections.Generic;
using System.IO;
using System.Threading.Tasks;
using API.DTOs.ReadingLists.CBL;
using API.Extensions;
using API.Services;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
namespace API.Controllers;
///
/// Responsible for the CBL import flow
///
public class CblController : BaseApiController
{
private readonly IReadingListService _readingListService;
private readonly IDirectoryService _directoryService;
public CblController(IReadingListService readingListService, IDirectoryService directoryService)
{
_readingListService = readingListService;
_directoryService = directoryService;
}
///
/// The first step in a cbl import. This validates the cbl file that if an import occured, would it be successful.
/// If this returns errors, the cbl will always be rejected by Kavita.
///
/// FormBody with parameter name of cbl
///
[HttpPost("validate")]
public async Task> ValidateCbl([FromForm(Name = "cbl")] IFormFile file)
{
var userId = User.GetUserId();
try
{
var cbl = await SaveAndLoadCblFile(file);
var importSummary = await _readingListService.ValidateCblFile(userId, cbl);
importSummary.FileName = file.FileName;
return Ok(importSummary);
}
catch (ArgumentNullException)
{
return Ok(new CblImportSummaryDto()
{
FileName = file.FileName,
Success = CblImportResult.Fail,
Results = new List()
{
new CblBookResult()
{
Reason = CblImportReason.InvalidFile
}
}
});
}
catch (InvalidOperationException)
{
return Ok(new CblImportSummaryDto()
{
FileName = file.FileName,
Success = CblImportResult.Fail,
Results = new List()
{
new CblBookResult()
{
Reason = CblImportReason.InvalidFile
}
}
});
}
}
///
/// Performs the actual import (assuming dryRun = false)
///
/// FormBody with parameter name of cbl
/// If true, will only emulate the import but not perform. This should be done to preview what will happen
///
[HttpPost("import")]
public async Task> ImportCbl([FromForm(Name = "cbl")] IFormFile file, [FromForm(Name = "dryRun")] bool dryRun = false)
{
try
{
var userId = User.GetUserId();
var cbl = await SaveAndLoadCblFile(file);
var importSummary = await _readingListService.CreateReadingListFromCbl(userId, cbl, dryRun);
importSummary.FileName = file.FileName;
return Ok(importSummary);
} catch (ArgumentNullException)
{
return Ok(new CblImportSummaryDto()
{
FileName = file.FileName,
Success = CblImportResult.Fail,
Results = new List()
{
new CblBookResult()
{
Reason = CblImportReason.InvalidFile
}
}
});
}
catch (InvalidOperationException)
{
return Ok(new CblImportSummaryDto()
{
FileName = file.FileName,
Success = CblImportResult.Fail,
Results = new List()
{
new CblBookResult()
{
Reason = CblImportReason.InvalidFile
}
}
});
}
}
private async Task SaveAndLoadCblFile(IFormFile file)
{
var filename = Path.GetRandomFileName();
var outputFile = Path.Join(_directoryService.TempDirectory, filename);
await using var stream = System.IO.File.Create(outputFile);
await file.CopyToAsync(stream);
stream.Close();
return ReadingListService.LoadCblFromPath(outputFile);
}
}