DocSpace-buildtools/common/ASC.Data.Backup.Core/Service/ProgressItems/RestoreProgressItem.cs

// (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

/*
 *
 * (c) Copyright Ascensio System Limited 2010-2020
 *
 * This program is freeware. You can redistribute it and/or modify it under the terms of the GNU 
 * General Public License (GPL) version 3 as published by the Free Software Foundation (https://www.gnu.org/copyleft/gpl.html). 
 * In accordance with Section 7(a) of the GNU GPL 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 more details, see GNU GPL at https://www.gnu.org/copyleft/gpl.html
 *
 * You can contact Ascensio System SIA by email at sales@onlyoffice.com
 *
 * The interactive user interfaces in modified source and object code versions of ONLYOFFICE must display 
 * Appropriate Legal Notices, as required under Section 5 of the GNU GPL version 3.
 *
 * Pursuant to Section 7 § 3(b) of the GNU GPL you must retain the original ONLYOFFICE logo which contains 
 * relevant author attributions when distributing the software. If the display of the logo in its graphic 
 * form is not reasonably feasible for technical reasons, you must include the words "Powered by ONLYOFFICE" 
 * in every copy of the program you distribute. 
 * Pursuant to Section 7 § 3(e) we decline to grant you any rights under trademark law for use of our trademarks.
 *
*/

namespace ASC.Data.Backup.Services;

[Transient]
public class RestoreProgressItem : BaseBackupProgressItem
{
    private TenantManager _tenantManager;
    private BackupStorageFactory _backupStorageFactory;
    private readonly NotifyHelper _notifyHelper;
    private BackupRepository _backupRepository;
    private RestorePortalTask _restorePortalTask;
    private readonly CoreBaseSettings _coreBaseSettings;

    private string _currentRegion;
    private string _upgradesPath;
    private Dictionary<string, string> _configPaths;

    public RestoreProgressItem(
        ILog logger,
        IServiceScopeFactory serviceScopeFactory,
        NotifyHelper notifyHelper,
        CoreBaseSettings coreBaseSettings)
        : base(logger, serviceScopeFactory)
    {

        _notifyHelper = notifyHelper;
        _coreBaseSettings = coreBaseSettings;

        BackupProgressItemEnum = BackupProgressItemEnum.Restore;
    }

    public BackupStorageType StorageType { get; set; }
    public string StoragePath { get; set; }
    public bool Notify { get; set; }
    public Dictionary<string, string> StorageParams { get; set; }
    public string TempFolder { get; set; }

    public void Init(StartRestoreRequest request, string tempFolder, string upgradesPath, string currentRegion, Dictionary<string, string> configPaths)
    {
        TenantId = request.TenantId;
        Notify = request.NotifyAfterCompletion;
        StoragePath = request.FilePathOrId;
        StorageType = request.StorageType;
        TempFolder = tempFolder;
        _upgradesPath = upgradesPath;
        _currentRegion = currentRegion;
        _configPaths = configPaths;
    }

    protected override void DoJob()
    {
        Tenant tenant = null;

        var tempFile = PathHelper.GetTempFileName(TempFolder);

        try
        {

            using var scope = _serviceScopeProvider.CreateScope();

            _tenantManager = scope.ServiceProvider.GetService<TenantManager>();
            _backupStorageFactory = scope.ServiceProvider.GetService<BackupStorageFactory>();
            _backupRepository = scope.ServiceProvider.GetService<BackupRepository>();

            tenant = _tenantManager.GetTenant(TenantId);
            _tenantManager.SetCurrentTenant(tenant);
            _notifyHelper.SendAboutRestoreStarted(tenant, Notify);
            tenant.SetStatus(TenantStatus.Restoring);
            _tenantManager.SaveTenant(tenant);

            _restorePortalTask = scope.ServiceProvider.GetService<RestorePortalTask>();

            var storage = _backupStorageFactory.GetBackupStorage(StorageType, TenantId, StorageParams);

            storage.Download(StoragePath, tempFile);

            if (!_coreBaseSettings.Standalone)
            {
                var backupHash = BackupWorker.GetBackupHash(tempFile);
                var record = _backupRepository.GetBackupRecord(backupHash, TenantId);

                if (record == null)
                {
                    throw new Exception(BackupResource.BackupNotFound);
                }
            }

            Percentage = 10;


            var columnMapper = new ColumnMapper();
            columnMapper.SetMapping("tenants_tenants", "alias", tenant.Alias, Guid.Parse(Id).ToString("N"));
            columnMapper.Commit();

            var restoreTask = _restorePortalTask;
            restoreTask.Init(_configPaths[_currentRegion], tempFile, TenantId, columnMapper, _upgradesPath);
            restoreTask.ProgressChanged += (sender, args) =>
            {
                Percentage = Percentage = 10d + 0.65 * args.Progress;
                PublishChanges();
            };
            restoreTask.RunJob();

            Tenant restoredTenant = null;

            if (restoreTask.Dump)
            {
                AscCacheNotify.OnClearCache();

                if (Notify)
                {
                    var tenants = _tenantManager.GetTenants();
                    foreach (var t in tenants)
                    {
                        _notifyHelper.SendAboutRestoreCompleted(t, Notify);
                    }
                }
            }
            else
            {
                _tenantManager.RemoveTenant(tenant.Id);

                restoredTenant = _tenantManager.GetTenant(columnMapper.GetTenantMapping());
                restoredTenant.SetStatus(TenantStatus.Active);
                restoredTenant.Alias = tenant.Alias;
                restoredTenant.PaymentId = string.Empty;

                if (string.IsNullOrEmpty(restoredTenant.MappedDomain) && !string.IsNullOrEmpty(tenant.MappedDomain))
                {
                    restoredTenant.MappedDomain = tenant.MappedDomain;
                }

                _tenantManager.SaveTenant(restoredTenant);
                _tenantManager.SetCurrentTenant(restoredTenant);
                // sleep until tenants cache expires
                Thread.Sleep(TimeSpan.FromMinutes(2));

                _notifyHelper.SendAboutRestoreCompleted(restoredTenant, Notify);
            }

            Percentage = 75;

            PublishChanges();

            File.Delete(tempFile);

            Percentage = 100;
            PublishChanges();
        }
        catch (Exception error)
        {
            Logger.Error(error);
            Exception = error;

            if (tenant != null)
            {
                tenant.SetStatus(TenantStatus.Active);
                _tenantManager.SaveTenant(tenant);
            }
        }
        finally
        {
            try
            {
                PublishChanges();
            }
            catch (Exception error)
            {
                Logger.Error("publish", error);
            }

            if (File.Exists(tempFile))
            {
                File.Delete(tempFile);
            }
        }
    }

    public override object Clone()
    {
        return MemberwiseClone();
    }
}