// Copyright (c) ppy Pty Ltd . Licensed under the MIT Licence. // See the LICENCE file in the repository root for full licence text. #nullable disable using System.IO; using System.Linq; using osu.Framework.Allocation; using osu.Framework.Bindables; using osu.Framework.Graphics; using osu.Framework.Localisation; using osu.Game.Beatmaps; using osu.Game.Overlays; namespace osu.Game.Screens.Edit.Setup { internal class ResourcesSection : SetupSection { private LabelledFileChooser audioTrackChooser; private LabelledFileChooser backgroundChooser; public override LocalisableString Title => "Resources"; [Resolved] private MusicController music { get; set; } [Resolved] private BeatmapManager beatmaps { get; set; } [Resolved] private IBindable working { get; set; } [Resolved] private EditorBeatmap editorBeatmap { get; set; } [Resolved] private SetupScreenHeader header { get; set; } [BackgroundDependencyLoader] private void load() { Children = new Drawable[] { backgroundChooser = new LabelledFileChooser(".jpg", ".jpeg", ".png") { Label = "Background", FixedLabelWidth = LABEL_WIDTH, TabbableContentContainer = this }, audioTrackChooser = new LabelledFileChooser(".mp3", ".ogg") { Label = "Audio Track", FixedLabelWidth = LABEL_WIDTH, TabbableContentContainer = this }, }; if (!string.IsNullOrEmpty(working.Value.Metadata.BackgroundFile)) backgroundChooser.Current.Value = new FileInfo(working.Value.Metadata.BackgroundFile); if (!string.IsNullOrEmpty(working.Value.Metadata.AudioFile)) audioTrackChooser.Current.Value = new FileInfo(working.Value.Metadata.AudioFile); backgroundChooser.Current.BindValueChanged(backgroundChanged); audioTrackChooser.Current.BindValueChanged(audioTrackChanged); updatePlaceholderText(); } public bool ChangeBackgroundImage(FileInfo source) { if (!source.Exists) return false; var set = working.Value.BeatmapSetInfo; var destination = new FileInfo($@"bg{source.Extension}"); // remove the previous background for now. // in the future we probably want to check if this is being used elsewhere (other difficulties?) var oldFile = set.Files.FirstOrDefault(f => f.Filename == working.Value.Metadata.BackgroundFile); using (var stream = source.OpenRead()) { if (oldFile != null) beatmaps.DeleteFile(set, oldFile); beatmaps.AddFile(set, stream, destination.Name); } editorBeatmap.SaveState(); working.Value.Metadata.BackgroundFile = destination.Name; header.Background.UpdateBackground(); return true; } public bool ChangeAudioTrack(FileInfo source) { if (!source.Exists) return false; var set = working.Value.BeatmapSetInfo; var destination = new FileInfo($@"audio{source.Extension}"); // remove the previous audio track for now. // in the future we probably want to check if this is being used elsewhere (other difficulties?) var oldFile = set.Files.FirstOrDefault(f => f.Filename == working.Value.Metadata.AudioFile); using (var stream = source.OpenRead()) { if (oldFile != null) beatmaps.DeleteFile(set, oldFile); beatmaps.AddFile(set, stream, destination.Name); } working.Value.Metadata.AudioFile = destination.Name; editorBeatmap.SaveState(); music.ReloadCurrentTrack(); return true; } private void backgroundChanged(ValueChangedEvent file) { if (!ChangeBackgroundImage(file.NewValue)) backgroundChooser.Current.Value = file.OldValue; updatePlaceholderText(); } private void audioTrackChanged(ValueChangedEvent file) { if (!ChangeAudioTrack(file.NewValue)) audioTrackChooser.Current.Value = file.OldValue; updatePlaceholderText(); } private void updatePlaceholderText() { audioTrackChooser.Text = audioTrackChooser.Current.Value == null ? "Click to select a track" : "Click to replace the track"; backgroundChooser.Text = backgroundChooser.Current.Value == null ? "Click to select a background image" : "Click to replace the background image"; } } }