osu-lazer/osu.Game/Overlays/Settings/Sections/Maintenance/MigrationRunScreen.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System.IO;
using System.Threading.Tasks;
using osu.Framework.Allocation;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Logging;
using osu.Framework.Screens;
using osu.Game.Graphics;
using osu.Game.Graphics.Sprites;
using osu.Game.Graphics.UserInterface;
using osu.Game.Screens;

namespace osu.Game.Overlays.Settings.Sections.Maintenance
{
    public class MigrationRunScreen : OsuScreen
    {
        private readonly DirectoryInfo destination;

        [Resolved]
        private OsuGame game { get; set; }

        public override bool AllowBackButton => false;

        public override bool AllowExternalScreenChange => false;

        public override bool DisallowExternalBeatmapRulesetChanges => true;

        private Task migrationTask;

        public MigrationRunScreen(DirectoryInfo destination)
        {
            this.destination = destination;
        }

        protected override void LoadComplete()
        {
            base.LoadComplete();

            InternalChildren = new Drawable[]
            {
                new FillFlowContainer
                {
                    AutoSizeAxes = Axes.Both,
                    Direction = FillDirection.Vertical,
                    Anchor = Anchor.Centre,
                    Origin = Anchor.Centre,
                    Children = new Drawable[]
                    {
                        new OsuSpriteText
                        {
                            Anchor = Anchor.Centre,
                            Origin = Anchor.Centre,
                            Text = "Migration in progress",
                            Font = OsuFont.Default.With(size: 48)
                        },
                        new LoadingSpinner(true)
                        {
                            State = { Value = Visibility.Visible }
                        }
                    }
                },
            };

            Beatmap.Value = Beatmap.Default;

            migrationTask = Task.Run(() => game.Migrate(destination.FullName))
                                .ContinueWith(t =>
                                {
                                    if (t.IsFaulted)
                                        Logger.Log($"Error during migration: {t.Exception?.Message}", level: LogLevel.Error);

                                    Schedule(this.Exit);
                                });
        }

        public override bool OnExiting(IScreen next)
        {
            // block until migration is finished
            if (migrationTask?.IsCompleted == false)
                return true;

            return base.OnExiting(next);
        }
    }
}