293 lines
9.7 KiB
C#
293 lines
9.7 KiB
C#
// (c) Copyright Ascensio System SIA 2010-2022
|
|
//
|
|
// This program is a free software product.
|
|
// You can redistribute it and/or modify it under the terms
|
|
// of the GNU Affero General Public License (AGPL) version 3 as published by the Free Software
|
|
// Foundation. In accordance with Section 7(a) of the GNU AGPL its Section 15 shall be amended
|
|
// to the effect that Ascensio System SIA expressly excludes the warranty of non-infringement of
|
|
// any third-party rights.
|
|
//
|
|
// This program is distributed WITHOUT ANY WARRANTY, without even the implied warranty
|
|
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For details, see
|
|
// the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
|
|
//
|
|
// You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia, EU, LV-1021.
|
|
//
|
|
// The interactive user interfaces in modified source and object code versions of the Program must
|
|
// display Appropriate Legal Notices, as required under Section 5 of the GNU AGPL version 3.
|
|
//
|
|
// Pursuant to Section 7(b) of the License you must retain the original Product logo when
|
|
// distributing the program. Pursuant to Section 7(e) we decline to grant you any rights under
|
|
// trademark law for use of our trademarks.
|
|
//
|
|
// All the Product's GUI elements, including illustrations and icon sets, as well as technical writing
|
|
// content are licensed under the terms of the Creative Commons Attribution-ShareAlike 4.0
|
|
// International. See the License terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
|
|
|
|
namespace ASC.Api.Migration;
|
|
|
|
[Scope]
|
|
[DefaultRoute]
|
|
[ApiController]
|
|
public class MigrationController : ControllerBase
|
|
{
|
|
private const string MigrationCacheKey = "ASC.Migration.Ongoing";
|
|
private readonly CoreBaseSettings _coreBaseSettings;
|
|
private readonly UserManager _userManager;
|
|
private readonly AuthContext _authContext;
|
|
private readonly TempPath _tempPath;
|
|
private readonly StudioNotifyService _studioNotifyService;
|
|
private readonly ICache _cache;
|
|
private readonly IHttpContextAccessor _httpContextAccessor;
|
|
|
|
public MigrationController(
|
|
CoreBaseSettings coreBaseSettings,
|
|
UserManager userManager,
|
|
AuthContext authContext,
|
|
TempPath tempPath,
|
|
StudioNotifyService studioNotifyService,
|
|
ICache cache,
|
|
IHttpContextAccessor httpContextAccessor)
|
|
{
|
|
_coreBaseSettings = coreBaseSettings;
|
|
_userManager = userManager;
|
|
_authContext = authContext;
|
|
_tempPath = tempPath;
|
|
_studioNotifyService = studioNotifyService;
|
|
_cache = cache;
|
|
_httpContextAccessor = httpContextAccessor;
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
[HttpGet("backuptmp")]
|
|
public string GetTmpFolder()
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
var tempFolder = Path.Combine(_tempPath.GetTempPath(), "migration", DateTime.Now.ToString("dd.MM.yyyy_HH_mm"));
|
|
|
|
if (!Directory.Exists(tempFolder))
|
|
{
|
|
Directory.CreateDirectory(tempFolder);
|
|
}
|
|
return tempFolder;
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
[HttpGet("list")]
|
|
public string[] List()
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
return MigrationCore.GetAvailableMigrations();
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="migratorName"></param>
|
|
/// <param name="path"></param>
|
|
[HttpPost("init/{migratorName}")]
|
|
public void UploadAndInit(string migratorName, string path)
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
if (GetOngoingMigration() != null)
|
|
{
|
|
throw new Exception("Migration is already in progress");
|
|
}
|
|
|
|
var migratorMeta = MigrationCore.GetMigrator(migratorName);
|
|
if (migratorMeta == null)
|
|
{
|
|
throw new ItemNotFoundException("No such migration provider");
|
|
}
|
|
var cts = new CancellationTokenSource();
|
|
var migrator = (IMigration)Activator.CreateInstance(migratorMeta.MigratorType);
|
|
try
|
|
{
|
|
migrator.Init(path, cts.Token);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
throw new Exception($"Error while initializing {migratorMeta.MigratorInfo.Name} migrator", ex);
|
|
}
|
|
|
|
var ongoingMigration = new OngoingMigration { Migration = migrator, CancelTokenSource = cts };
|
|
StoreOngoingMigration(ongoingMigration);
|
|
|
|
ongoingMigration.ParseTask = Task.Run(migrator.Parse);
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
[HttpGet("status")]
|
|
public object Status()
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
var ongoingMigration = GetOngoingMigration();
|
|
if (ongoingMigration == null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
if (ongoingMigration.CancelTokenSource.IsCancellationRequested == true)
|
|
{
|
|
var migratorName = ongoingMigration.Migration.GetType().GetCustomAttribute<ApiMigratorAttribute>().Name;
|
|
return migratorName;
|
|
}
|
|
|
|
var result = new MigrationStatus()
|
|
{
|
|
ParseResult = ongoingMigration.ParseTask.IsCompleted ? ongoingMigration.ParseTask.Result : null,
|
|
MigrationEnded = ongoingMigration.MigrationEnded,
|
|
Progress = ongoingMigration.Migration.GetProgress(),
|
|
ProgressStatus = ongoingMigration.Migration.GetProgressStatus()
|
|
};
|
|
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
[HttpPost("cancel")]
|
|
public void Cancel()
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
var ongoingMigration = GetOngoingMigration();
|
|
if (ongoingMigration == null)
|
|
{
|
|
throw new Exception("No migration is in progress");
|
|
}
|
|
ongoingMigration.CancelTokenSource.Cancel();
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="info"></param>
|
|
[HttpPost("migrate")]
|
|
public void Migrate(MigrationApiInfo info)
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
var ongoingMigration = GetOngoingMigration();
|
|
if (ongoingMigration == null)
|
|
{
|
|
throw new Exception("No migration is in progress");
|
|
}
|
|
else if (!ongoingMigration.ParseTask.IsCompleted)
|
|
{
|
|
throw new Exception("Parsing is still in progress");
|
|
}
|
|
|
|
ongoingMigration.MigrationTask = ongoingMigration.Migration.Migrate(info);
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
[HttpGet("logs")]
|
|
public async Task Logs()
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
var ongoingMigration = GetOngoingMigration();
|
|
if (ongoingMigration == null)
|
|
{
|
|
throw new Exception("No migration is in progress");
|
|
}
|
|
|
|
_httpContextAccessor.HttpContext.Response.Headers.Add("Content-Disposition", ContentDispositionUtil.GetHeaderValue("migration.log"));
|
|
_httpContextAccessor.HttpContext.Response.ContentType = "text/plain; charset=UTF-8";
|
|
await ongoingMigration.Migration.GetLogs().CopyToAsync(_httpContextAccessor.HttpContext.Response.Body);
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="isSendWelcomeEmail"></param>
|
|
[HttpPost("finish")]
|
|
public void Finish(bool isSendWelcomeEmail)
|
|
{
|
|
if (!_coreBaseSettings.Standalone || !_userManager.IsDocSpaceAdmin(_authContext.CurrentAccount.ID))
|
|
{
|
|
throw new System.Security.SecurityException();
|
|
}
|
|
|
|
if (isSendWelcomeEmail)
|
|
{
|
|
var ongoingMigration = GetOngoingMigration();
|
|
if (ongoingMigration == null)
|
|
{
|
|
throw new Exception("No migration is in progress");
|
|
}
|
|
var guidUsers = ongoingMigration.Migration.GetGuidImportedUsers();
|
|
foreach (var gu in guidUsers)
|
|
{
|
|
var u = _userManager.GetUsers(gu);
|
|
_studioNotifyService.UserInfoActivation(u);
|
|
}
|
|
}
|
|
ClearCache();
|
|
}
|
|
|
|
// ToDo: Use ASCCache
|
|
private void StoreOngoingMigration(OngoingMigration migration)
|
|
{
|
|
_cache.Insert(MigrationCacheKey, migration, TimeSpan.FromDays(1));
|
|
}
|
|
|
|
private OngoingMigration GetOngoingMigration()
|
|
{
|
|
return _cache.Get<OngoingMigration>(MigrationCacheKey);
|
|
}
|
|
|
|
private void ClearCache()
|
|
{
|
|
_cache.Remove(MigrationCacheKey);
|
|
}
|
|
|
|
private void ClearMigration(CacheEntryRemovedArguments arguments)
|
|
{
|
|
if (typeof(OngoingMigration).IsAssignableFrom(arguments.CacheItem.Value.GetType()))
|
|
{
|
|
var ongoingMigration = (OngoingMigration)arguments.CacheItem.Value;
|
|
ongoingMigration.Migration.Dispose();
|
|
}
|
|
}
|
|
}
|