Pull down IResourceDestination.
This commit is contained in:
196
Assets/Cryville/Crtr/Browsing/Legacy/LegacyResourceStore.cs
Normal file
196
Assets/Cryville/Crtr/Browsing/Legacy/LegacyResourceStore.cs
Normal file
@@ -0,0 +1,196 @@
|
||||
using Cryville.Crtr.Extension;
|
||||
using Cryville.Crtr.Extensions.Umg;
|
||||
using Cryville.Crtr.Extensions;
|
||||
using Cryville.Crtr.UI;
|
||||
using Newtonsoft.Json;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System;
|
||||
using System.Linq;
|
||||
using Cryville.Crtr.Browsing.Actions;
|
||||
|
||||
namespace Cryville.Crtr.Browsing.Legacy {
|
||||
internal class LegacyResourceStore : IResourceDestination {
|
||||
public string RootPath { get; private set; }
|
||||
public IResourceAction ImportAction { get; private set; }
|
||||
public event Action ResourceImported;
|
||||
|
||||
static bool _init;
|
||||
public LegacyResourceStore(string rootPath) {
|
||||
RootPath = rootPath;
|
||||
ImportAction = new ImportResourceAction(this);
|
||||
if (!_init) {
|
||||
_init = true;
|
||||
ExtensionManager.Init(rootPath);
|
||||
}
|
||||
}
|
||||
|
||||
public bool CanImport(Uri uri) {
|
||||
if (!uri.IsFile) return false;
|
||||
return CanImportFile(new FileInfo(uri.LocalPath));
|
||||
}
|
||||
protected bool CanImportFile(FileInfo file) {
|
||||
return !file.FullName.StartsWith(new DirectoryInfo(RootPath).FullName);
|
||||
}
|
||||
public bool ImportFrom(Uri uri) {
|
||||
var file = new FileInfo(uri.LocalPath);
|
||||
IEnumerable<ResourceConverter> converters;
|
||||
if (!ExtensionManager.TryGetConverters(file.Extension, out converters)) return false;
|
||||
foreach (var converter in converters) {
|
||||
var resources = new List<Resource>();
|
||||
var ses = new ConversionSession {
|
||||
OnResourceAdd = res => resources.Add(res)
|
||||
};
|
||||
try {
|
||||
converter.Convert(file, ses);
|
||||
}
|
||||
catch (Exception ex) {
|
||||
LogAndPopupExtra(4, ex, "Failed to import resource: {0}", ex.Message);
|
||||
return false;
|
||||
}
|
||||
foreach (var res in resources) {
|
||||
if (!res.Valid) {
|
||||
LogAndPopup(3, "Attempt to import invalid resource: {0}", res);
|
||||
}
|
||||
else if (res is RawChartResource) {
|
||||
var tres = (RawChartResource)res;
|
||||
var dir = new DirectoryInfo(Path.Combine(RootPath, "charts", res.Name));
|
||||
if (!dir.Exists) dir.Create();
|
||||
using (var writer = new StreamWriter(Path.Combine(dir.FullName, ".json"))) {
|
||||
writer.Write(JsonConvert.SerializeObject(ConvertChartData(tres.Main), Game.GlobalJsonSerializerSettings));
|
||||
}
|
||||
using (var writer = new StreamWriter(Path.Combine(dir.FullName, ".umgc"))) {
|
||||
tres.Meta.data = "";
|
||||
writer.Write(JsonConvert.SerializeObject(tres.Meta, Game.GlobalJsonSerializerSettings));
|
||||
}
|
||||
if (tres.Meta.cover != null) {
|
||||
var coverFile = new FileInfo(Path.Combine(file.Directory.FullName, tres.Meta.cover));
|
||||
if (coverFile.Exists)
|
||||
coverFile.CopyTo(Path.Combine(dir.FullName, tres.Meta.cover), true);
|
||||
}
|
||||
}
|
||||
else if (res is FileResource) {
|
||||
var tres = (FileResource)res;
|
||||
DirectoryInfo dest;
|
||||
bool singleFileFlag = false;
|
||||
if (res is ChartResource)
|
||||
dest = new DirectoryInfo(Path.Combine(RootPath, "charts", res.Name));
|
||||
else if (res is ExtensionResource) {
|
||||
dest = new DirectoryInfo(Path.Combine(RootPath, "extensions"));
|
||||
singleFileFlag = true;
|
||||
Popup.Create("Please restart the game to reload the extensions");
|
||||
}
|
||||
else if (res is RulesetResource)
|
||||
dest = new DirectoryInfo(Path.Combine(RootPath, "rulesets", res.Name));
|
||||
else if (res is SkinResource)
|
||||
dest = new DirectoryInfo(Path.Combine(RootPath, "skins", (res as SkinResource).RulesetName, res.Name));
|
||||
else if (res is SongResource)
|
||||
dest = new DirectoryInfo(Path.Combine(RootPath, "songs", res.Name));
|
||||
else {
|
||||
LogAndPopup(3, "Attempt to import unsupported file resource: {0}", res);
|
||||
continue;
|
||||
}
|
||||
if (singleFileFlag || !dest.Exists) {
|
||||
dest.Create();
|
||||
tres.Master.CopyTo(Path.Combine(dest.FullName, singleFileFlag ? tres.Master.Name : tres.Master.Extension));
|
||||
foreach (var attachment in tres.Attachments) {
|
||||
if (singleFileFlag) throw new InvalidOperationException("Internal error");
|
||||
attachment.CopyTo(Path.Combine(dest.FullName, attachment.Name));
|
||||
}
|
||||
}
|
||||
else Game.MainLogger.Log(1, "Resource", "Resource already exists: {0}", res);
|
||||
}
|
||||
else {
|
||||
LogAndPopup(3, "Attempt to import unsupported resource: {0}", res);
|
||||
}
|
||||
}
|
||||
ResourceImported?.Invoke();
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void LogAndPopup(int level, string format, params object[] args) {
|
||||
var msg = string.Format(format, args);
|
||||
Game.MainLogger.Log(level, "Resource", msg);
|
||||
Popup.Create(msg);
|
||||
}
|
||||
|
||||
void LogAndPopupExtra(int level, object extraLog, string format, params object[] args) {
|
||||
var msg = string.Format(format, args);
|
||||
Game.MainLogger.Log(level, "Resource", "{0}\n{1}", msg, extraLog);
|
||||
Popup.Create(msg);
|
||||
}
|
||||
|
||||
static Chart ConvertChartData(ChartData i) {
|
||||
return new Chart {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
format = i.format,
|
||||
groups = ConvertGroups(i.groups),
|
||||
motions = ConvertMotions(i.motions),
|
||||
ruleset = i.ruleset,
|
||||
sigs = ConvertSignatures(i.sigs),
|
||||
sounds = ConvertSounds(i.sounds),
|
||||
time = ConvertBeatTime(i.time),
|
||||
};
|
||||
}
|
||||
static BeatTime? ConvertBeatTime(Extension.BeatTime? value) {
|
||||
if (value == null) return null;
|
||||
return new BeatTime(value.Value.b, value.Value.n, value.Value.d);
|
||||
}
|
||||
static List<Chart.Group> ConvertGroups(List<ChartData.Group> l) {
|
||||
return l.Select(i => new Chart.Group {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
motions = ConvertMotions(i.motions),
|
||||
notes = ConvertNotes(i.notes),
|
||||
time = ConvertBeatTime(i.time),
|
||||
tracks = ConvertTracks(i.tracks),
|
||||
}).ToList();
|
||||
}
|
||||
static List<Chart.Judge> ConvertJudges(List<ChartData.Judge> l) {
|
||||
return l.Select(i => new Chart.Judge {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
name = i.name,
|
||||
time = ConvertBeatTime(i.time),
|
||||
}).ToList();
|
||||
}
|
||||
static List<Chart.Motion> ConvertMotions(List<ChartData.Motion> l) {
|
||||
return l.Select(i => new Chart.Motion {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
motion = i.motion,
|
||||
sumfix = i.sumfix,
|
||||
time = ConvertBeatTime(i.time),
|
||||
}).ToList();
|
||||
}
|
||||
static List<Chart.Note> ConvertNotes(List<ChartData.Note> l) {
|
||||
return l.Select(i => new Chart.Note {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
judges = ConvertJudges(i.judges),
|
||||
motions = ConvertMotions(i.motions),
|
||||
time = ConvertBeatTime(i.time),
|
||||
}).ToList();
|
||||
}
|
||||
static List<Chart.Signature> ConvertSignatures(List<ChartData.Signature> l) {
|
||||
return l.Select(i => new Chart.Signature {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
tempo = i.tempo,
|
||||
time = ConvertBeatTime(i.time),
|
||||
}).ToList();
|
||||
}
|
||||
static List<Chart.Sound> ConvertSounds(List<ChartData.Sound> l) {
|
||||
return l.Select(i => new Chart.Sound {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
id = i.id,
|
||||
offset = i.offset,
|
||||
time = ConvertBeatTime(i.time),
|
||||
}).ToList();
|
||||
}
|
||||
static List<Chart.Track> ConvertTracks(List<ChartData.Track> l) {
|
||||
return l.Select(i => new Chart.Track {
|
||||
endtime = ConvertBeatTime(i.endtime),
|
||||
motions = ConvertMotions(i.motions),
|
||||
time = ConvertBeatTime(i.time),
|
||||
}).ToList();
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user