2019-07-29 20:49:12 +08:00
|
|
|
// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
|
2019-01-24 16:43:03 +08:00
|
|
|
// See the LICENCE file in the repository root for full licence text.
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
using System;
|
2021-03-31 13:57:28 +08:00
|
|
|
using System.Collections.Generic;
|
2018-04-13 17:19:50 +08:00
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
|
|
|
using System.Reflection;
|
2020-11-20 17:06:08 +08:00
|
|
|
using System.Runtime.Versioning;
|
2018-04-13 17:19:50 +08:00
|
|
|
using System.Threading.Tasks;
|
2020-05-08 09:38:31 +08:00
|
|
|
using Microsoft.Win32;
|
2021-04-27 10:37:08 +08:00
|
|
|
using osu.Desktop.Security;
|
2018-04-13 17:19:50 +08:00
|
|
|
using osu.Desktop.Overlays;
|
|
|
|
using osu.Framework.Platform;
|
|
|
|
using osu.Game;
|
2018-07-06 15:39:27 +08:00
|
|
|
using osu.Desktop.Updater;
|
2018-08-01 01:58:39 +08:00
|
|
|
using osu.Framework;
|
2019-03-30 22:56:38 +08:00
|
|
|
using osu.Framework.Logging;
|
2018-12-27 18:23:58 +08:00
|
|
|
using osu.Framework.Screens;
|
|
|
|
using osu.Game.Screens.Menu;
|
2019-09-24 17:03:01 +08:00
|
|
|
using osu.Game.Updater;
|
2020-07-01 23:15:41 +08:00
|
|
|
using osu.Desktop.Windows;
|
2021-03-31 13:57:28 +08:00
|
|
|
using osu.Framework.Threading;
|
2021-01-25 02:18:16 +08:00
|
|
|
using osu.Game.IO;
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
namespace osu.Desktop
|
|
|
|
{
|
|
|
|
internal class OsuGameDesktop : OsuGame
|
|
|
|
{
|
|
|
|
private readonly bool noVersionOverlay;
|
2018-12-27 18:23:58 +08:00
|
|
|
private VersionManager versionManager;
|
2018-04-13 17:19:50 +08:00
|
|
|
|
|
|
|
public OsuGameDesktop(string[] args = null)
|
|
|
|
: base(args)
|
|
|
|
{
|
|
|
|
noVersionOverlay = args?.Any(a => a == "--no-version-overlay") ?? false;
|
|
|
|
}
|
|
|
|
|
2021-01-25 02:46:10 +08:00
|
|
|
public override StableStorage GetStorageForStableInstall()
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
2019-03-30 22:56:38 +08:00
|
|
|
if (Host is DesktopGameHost desktopHost)
|
2020-05-08 09:38:31 +08:00
|
|
|
{
|
|
|
|
string stablePath = getStableInstallPath();
|
|
|
|
if (!string.IsNullOrEmpty(stablePath))
|
2021-01-25 02:18:16 +08:00
|
|
|
return new StableStorage(stablePath, desktopHost);
|
2020-05-08 09:38:31 +08:00
|
|
|
}
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
2019-07-30 11:44:08 +08:00
|
|
|
catch (Exception)
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2019-07-30 11:44:08 +08:00
|
|
|
Logger.Log("Could not find a stable install", LoggingTarget.Runtime, LogLevel.Important);
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
2019-03-30 22:56:38 +08:00
|
|
|
|
|
|
|
return null;
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
|
2020-05-08 09:38:31 +08:00
|
|
|
private string getStableInstallPath()
|
|
|
|
{
|
2021-06-02 14:13:21 +08:00
|
|
|
static bool checkExists(string p) => Directory.Exists(Path.Combine(p, "Songs")) || File.Exists(Path.Combine(p, "osu!.cfg"));
|
2020-05-08 09:38:31 +08:00
|
|
|
|
|
|
|
string stableInstallPath;
|
|
|
|
|
2020-11-20 17:06:08 +08:00
|
|
|
if (OperatingSystem.IsWindows())
|
2020-05-08 09:38:31 +08:00
|
|
|
{
|
2021-01-15 14:17:38 +08:00
|
|
|
try
|
|
|
|
{
|
|
|
|
stableInstallPath = getStableInstallPathFromRegistry();
|
2020-05-08 09:38:31 +08:00
|
|
|
|
2021-01-15 14:17:38 +08:00
|
|
|
if (!string.IsNullOrEmpty(stableInstallPath) && checkExists(stableInstallPath))
|
|
|
|
return stableInstallPath;
|
|
|
|
}
|
|
|
|
catch { }
|
2020-05-08 09:38:31 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
stableInstallPath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), @"osu!");
|
|
|
|
if (checkExists(stableInstallPath))
|
|
|
|
return stableInstallPath;
|
|
|
|
|
|
|
|
stableInstallPath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".osu");
|
|
|
|
if (checkExists(stableInstallPath))
|
|
|
|
return stableInstallPath;
|
|
|
|
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
2020-11-20 17:06:08 +08:00
|
|
|
[SupportedOSPlatform("windows")]
|
|
|
|
private string getStableInstallPathFromRegistry()
|
|
|
|
{
|
|
|
|
using (RegistryKey key = Registry.ClassesRoot.OpenSubKey("osu"))
|
2020-11-21 07:06:20 +08:00
|
|
|
return key?.OpenSubKey(@"shell\open\command")?.GetValue(string.Empty)?.ToString()?.Split('"')[1].Replace("osu!.exe", "");
|
2020-11-20 17:06:08 +08:00
|
|
|
}
|
|
|
|
|
2020-03-05 12:34:04 +08:00
|
|
|
protected override UpdateManager CreateUpdateManager()
|
|
|
|
{
|
|
|
|
switch (RuntimeInfo.OS)
|
|
|
|
{
|
|
|
|
case RuntimeInfo.Platform.Windows:
|
|
|
|
return new SquirrelUpdateManager();
|
|
|
|
|
|
|
|
default:
|
|
|
|
return new SimpleUpdateManager();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-04-13 17:19:50 +08:00
|
|
|
protected override void LoadComplete()
|
|
|
|
{
|
|
|
|
base.LoadComplete();
|
|
|
|
|
|
|
|
if (!noVersionOverlay)
|
2019-07-29 20:49:12 +08:00
|
|
|
LoadComponentAsync(versionManager = new VersionManager { Depth = int.MinValue }, Add);
|
2018-07-06 15:39:27 +08:00
|
|
|
|
2019-12-18 13:07:32 +08:00
|
|
|
LoadComponentAsync(new DiscordRichPresence(), Add);
|
2020-07-01 23:15:41 +08:00
|
|
|
|
|
|
|
if (RuntimeInfo.OS == RuntimeInfo.Platform.Windows)
|
2020-07-24 15:38:48 +08:00
|
|
|
LoadComponentAsync(new GameplayWinKeyBlocker(), Add);
|
2021-04-27 05:41:04 +08:00
|
|
|
|
2021-04-27 09:05:18 +08:00
|
|
|
LoadComponentAsync(new ElevatedPrivilegesChecker(), Add);
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
|
2019-01-23 19:52:00 +08:00
|
|
|
protected override void ScreenChanged(IScreen lastScreen, IScreen newScreen)
|
2018-12-27 18:23:58 +08:00
|
|
|
{
|
2019-01-23 19:52:00 +08:00
|
|
|
base.ScreenChanged(lastScreen, newScreen);
|
|
|
|
|
2018-12-27 18:23:58 +08:00
|
|
|
switch (newScreen)
|
|
|
|
{
|
2019-07-09 16:59:40 +08:00
|
|
|
case IntroScreen _:
|
2018-12-27 18:23:58 +08:00
|
|
|
case MainMenu _:
|
2019-06-11 13:28:52 +08:00
|
|
|
versionManager?.Show();
|
2018-12-27 18:23:58 +08:00
|
|
|
break;
|
2019-04-01 11:16:05 +08:00
|
|
|
|
2018-12-27 18:23:58 +08:00
|
|
|
default:
|
2019-06-11 13:28:52 +08:00
|
|
|
versionManager?.Hide();
|
2018-12-27 18:23:58 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-04-13 17:19:50 +08:00
|
|
|
public override void SetHost(GameHost host)
|
|
|
|
{
|
|
|
|
base.SetHost(host);
|
2019-04-01 11:16:05 +08:00
|
|
|
|
2020-10-03 11:28:43 +08:00
|
|
|
var iconStream = Assembly.GetExecutingAssembly().GetManifestResourceStream(GetType(), "lazer.ico");
|
|
|
|
|
2021-03-12 17:35:42 +08:00
|
|
|
var desktopWindow = (SDL2DesktopWindow)host.Window;
|
2018-04-13 17:19:50 +08:00
|
|
|
|
2021-03-12 17:35:42 +08:00
|
|
|
desktopWindow.CursorState |= CursorState.Hidden;
|
|
|
|
desktopWindow.SetIconFromStream(iconStream);
|
|
|
|
desktopWindow.Title = Name;
|
|
|
|
desktopWindow.DragDrop += f => fileDrop(new[] { f });
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
|
2021-03-31 13:57:28 +08:00
|
|
|
private readonly List<string> importableFiles = new List<string>();
|
|
|
|
private ScheduledDelegate importSchedule;
|
|
|
|
|
2020-06-12 08:16:21 +08:00
|
|
|
private void fileDrop(string[] filePaths)
|
2018-04-13 17:19:50 +08:00
|
|
|
{
|
2021-03-31 13:57:28 +08:00
|
|
|
lock (importableFiles)
|
|
|
|
{
|
2021-10-27 12:04:41 +08:00
|
|
|
string firstExtension = Path.GetExtension(filePaths.First());
|
2021-03-31 13:57:28 +08:00
|
|
|
|
|
|
|
if (filePaths.Any(f => Path.GetExtension(f) != firstExtension)) return;
|
|
|
|
|
|
|
|
importableFiles.AddRange(filePaths);
|
|
|
|
|
|
|
|
Logger.Log($"Adding {filePaths.Length} files for import");
|
|
|
|
|
|
|
|
// File drag drop operations can potentially trigger hundreds or thousands of these calls on some platforms.
|
|
|
|
// In order to avoid spawning multiple import tasks for a single drop operation, debounce a touch.
|
|
|
|
importSchedule?.Cancel();
|
|
|
|
importSchedule = Scheduler.AddDelayed(handlePendingImports, 100);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private void handlePendingImports()
|
|
|
|
{
|
|
|
|
lock (importableFiles)
|
|
|
|
{
|
|
|
|
Logger.Log($"Handling batch import of {importableFiles.Count} files");
|
2018-04-13 17:19:50 +08:00
|
|
|
|
2021-10-27 12:04:41 +08:00
|
|
|
string[] paths = importableFiles.ToArray();
|
2021-03-31 13:57:28 +08:00
|
|
|
importableFiles.Clear();
|
2018-04-13 17:19:50 +08:00
|
|
|
|
2021-03-31 13:57:28 +08:00
|
|
|
Task.Factory.StartNew(() => Import(paths), TaskCreationOptions.LongRunning);
|
|
|
|
}
|
2018-04-13 17:19:50 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|