2018-11-23 07:03:32 +00:00
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
2021-10-21 20:04:19 +00:00
|
|
|
using Microsoft.AspNetCore.Mvc;
|
2019-06-30 01:38:11 +00:00
|
|
|
using NzbDrone.Common.Crypto;
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
|
|
using NzbDrone.Common.Extensions;
|
2018-11-23 07:03:32 +00:00
|
|
|
using NzbDrone.Core.Backup;
|
|
|
|
using Radarr.Http;
|
2019-06-30 01:38:11 +00:00
|
|
|
using Radarr.Http.REST;
|
2021-10-21 20:04:19 +00:00
|
|
|
using Radarr.Http.REST.Attributes;
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2019-12-03 01:36:18 +00:00
|
|
|
namespace Radarr.Api.V3.System.Backup
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2021-10-21 20:04:19 +00:00
|
|
|
[V3ApiController("system/backup")]
|
|
|
|
public class BackupController : Controller
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
|
|
|
private readonly IBackupService _backupService;
|
2019-06-30 01:38:11 +00:00
|
|
|
private readonly IAppFolderInfo _appFolderInfo;
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2019-06-30 01:38:11 +00:00
|
|
|
private static readonly List<string> ValidExtensions = new List<string> { ".zip", ".db", ".xml" };
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
public BackupController(IBackupService backupService,
|
2019-06-30 01:38:11 +00:00
|
|
|
IAppFolderInfo appFolderInfo,
|
|
|
|
IDiskProvider diskProvider)
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
|
|
|
_backupService = backupService;
|
2019-06-30 01:38:11 +00:00
|
|
|
_appFolderInfo = appFolderInfo;
|
|
|
|
_diskProvider = diskProvider;
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[HttpGet]
|
2018-11-23 07:03:32 +00:00
|
|
|
public List<BackupResource> GetBackupFiles()
|
|
|
|
{
|
|
|
|
var backups = _backupService.GetBackups();
|
|
|
|
|
|
|
|
return backups.Select(b => new BackupResource
|
2019-12-22 22:08:53 +00:00
|
|
|
{
|
|
|
|
Id = GetBackupId(b),
|
|
|
|
Name = b.Name,
|
|
|
|
Path = $"/backup/{b.Type.ToString().ToLower()}/{b.Name}",
|
|
|
|
Type = b.Type,
|
|
|
|
Time = b.Time
|
|
|
|
})
|
2019-06-30 01:38:11 +00:00
|
|
|
.OrderByDescending(b => b.Time)
|
|
|
|
.ToList();
|
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[RestDeleteById]
|
|
|
|
public void DeleteBackup(int id)
|
2019-06-30 01:38:11 +00:00
|
|
|
{
|
|
|
|
var backup = GetBackup(id);
|
|
|
|
var path = GetBackupPath(backup);
|
|
|
|
|
|
|
|
if (!_diskProvider.FileExists(path))
|
|
|
|
{
|
|
|
|
throw new NotFoundException();
|
|
|
|
}
|
|
|
|
|
|
|
|
_diskProvider.DeleteFile(path);
|
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[HttpPost("restore/{id:int}")]
|
2019-08-28 21:43:55 +00:00
|
|
|
public object Restore(int id)
|
2019-06-30 01:38:11 +00:00
|
|
|
{
|
|
|
|
var backup = GetBackup(id);
|
|
|
|
|
|
|
|
if (backup == null)
|
|
|
|
{
|
|
|
|
throw new NotFoundException();
|
|
|
|
}
|
|
|
|
|
|
|
|
var path = GetBackupPath(backup);
|
|
|
|
|
|
|
|
_backupService.Restore(path);
|
|
|
|
|
|
|
|
return new
|
2019-12-22 22:08:53 +00:00
|
|
|
{
|
|
|
|
RestartRequired = true
|
|
|
|
};
|
2019-06-30 01:38:11 +00:00
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[HttpPost("restore/upload")]
|
2019-08-28 21:43:55 +00:00
|
|
|
public object UploadAndRestore()
|
2019-06-30 01:38:11 +00:00
|
|
|
{
|
2021-10-21 20:04:19 +00:00
|
|
|
var files = Request.Form.Files;
|
2019-06-30 01:38:11 +00:00
|
|
|
|
|
|
|
if (files.Empty())
|
|
|
|
{
|
|
|
|
throw new BadRequestException("file must be provided");
|
|
|
|
}
|
|
|
|
|
|
|
|
var file = files.First();
|
2021-10-21 20:04:19 +00:00
|
|
|
var extension = Path.GetExtension(file.FileName);
|
2019-06-30 01:38:11 +00:00
|
|
|
|
|
|
|
if (!ValidExtensions.Contains(extension))
|
|
|
|
{
|
|
|
|
throw new UnsupportedMediaTypeException($"Invalid extension, must be one of: {ValidExtensions.Join(", ")}");
|
|
|
|
}
|
|
|
|
|
|
|
|
var path = Path.Combine(_appFolderInfo.TempFolder, $"radarr_backup_restore{extension}");
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
_diskProvider.SaveStream(file.OpenReadStream(), path);
|
2019-06-30 01:38:11 +00:00
|
|
|
_backupService.Restore(path);
|
|
|
|
|
|
|
|
// Cleanup restored file
|
|
|
|
_diskProvider.DeleteFile(path);
|
|
|
|
|
|
|
|
return new
|
2019-12-22 22:08:53 +00:00
|
|
|
{
|
|
|
|
RestartRequired = true
|
|
|
|
};
|
2019-06-30 01:38:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private string GetBackupPath(NzbDrone.Core.Backup.Backup backup)
|
|
|
|
{
|
|
|
|
return Path.Combine(_backupService.GetBackupFolder(backup.Type), backup.Name);
|
|
|
|
}
|
|
|
|
|
|
|
|
private int GetBackupId(NzbDrone.Core.Backup.Backup backup)
|
|
|
|
{
|
|
|
|
return HashConverter.GetHashInt31($"backup-{backup.Type}-{backup.Name}");
|
|
|
|
}
|
|
|
|
|
|
|
|
private NzbDrone.Core.Backup.Backup GetBackup(int id)
|
|
|
|
{
|
|
|
|
return _backupService.GetBackups().SingleOrDefault(b => GetBackupId(b) == id);
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|