86 lines
2.6 KiB
C#
86 lines
2.6 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Threading.Tasks;
|
|
using Content.Shared.Input;
|
|
using Robust.Client.Interfaces.Graphics;
|
|
using Robust.Client.Interfaces.Input;
|
|
using Robust.Shared.Input;
|
|
using Robust.Shared.Interfaces.Resources;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Utility;
|
|
using SixLabors.ImageSharp;
|
|
|
|
namespace Content.Client
|
|
{
|
|
internal class ScreenshotHook : IScreenshotHook
|
|
{
|
|
private static readonly ResourcePath BaseScreenshotPath = new ResourcePath("/Screenshots");
|
|
|
|
[Dependency] private readonly IInputManager _inputManager = default;
|
|
[Dependency] private readonly IClyde _clyde = default;
|
|
[Dependency] private readonly IResourceManager _resourceManager = default;
|
|
|
|
public void Initialize()
|
|
{
|
|
_inputManager.SetInputCommand(ContentKeyFunctions.TakeScreenshot, InputCmdHandler.FromDelegate(_ =>
|
|
{
|
|
Take(ScreenshotType.AfterUI);
|
|
}));
|
|
|
|
_inputManager.SetInputCommand(ContentKeyFunctions.TakeScreenshotNoUI, InputCmdHandler.FromDelegate(_ =>
|
|
{
|
|
Take(ScreenshotType.BeforeUI);
|
|
}));
|
|
}
|
|
|
|
private async void Take(ScreenshotType type)
|
|
{
|
|
var screenshot = await _clyde.ScreenshotAsync(type);
|
|
|
|
var time = DateTime.Now.ToString("yyyy-M-dd_HH.mm.ss");
|
|
|
|
if (!_resourceManager.UserData.IsDir(BaseScreenshotPath))
|
|
{
|
|
_resourceManager.UserData.CreateDir(BaseScreenshotPath);
|
|
}
|
|
|
|
for (var i = 0; i < 5; i++)
|
|
{
|
|
try
|
|
{
|
|
var filename = time;
|
|
|
|
if (i != 0)
|
|
{
|
|
filename = $"{filename}-{i}";
|
|
}
|
|
|
|
await using var file =
|
|
_resourceManager.UserData.Open(BaseScreenshotPath / $"{filename}.png", FileMode.CreateNew);
|
|
|
|
await Task.Run(() =>
|
|
{
|
|
// Saving takes forever, so don't hang the game on it.
|
|
screenshot.SaveAsPng(file);
|
|
});
|
|
|
|
Logger.InfoS("screenshot", "Screenshot taken as {0}.png", filename);
|
|
return;
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
Logger.WarningS("screenshot", "Failed to save screenshot, retrying?:\n{0}", e);
|
|
}
|
|
}
|
|
|
|
Logger.ErrorS("screenshot", "Unable to save screenshot.");
|
|
}
|
|
}
|
|
|
|
public interface IScreenshotHook
|
|
{
|
|
void Initialize();
|
|
}
|
|
}
|