Merge pull request #10838 from pchote/map-saving-packages
Rework map loading and saving.
This commit is contained in:
@@ -34,17 +34,6 @@ namespace OpenRA.FileSystem
|
||||
|
||||
Cache<string, List<IReadOnlyPackage>> fileIndex = new Cache<string, List<IReadOnlyPackage>>(_ => new List<IReadOnlyPackage>());
|
||||
|
||||
public IReadWritePackage CreatePackage(string filename)
|
||||
{
|
||||
var content = new Dictionary<string, byte[]>();
|
||||
if (filename.EndsWith(".zip", StringComparison.InvariantCultureIgnoreCase))
|
||||
return new ZipFile(this, filename, content);
|
||||
if (filename.EndsWith(".oramap", StringComparison.InvariantCultureIgnoreCase))
|
||||
return new ZipFile(this, filename, content);
|
||||
|
||||
return new Folder(filename, content);
|
||||
}
|
||||
|
||||
public IReadOnlyPackage OpenPackage(string filename)
|
||||
{
|
||||
if (filename.EndsWith(".mix", StringComparison.InvariantCultureIgnoreCase))
|
||||
@@ -75,6 +64,32 @@ namespace OpenRA.FileSystem
|
||||
return new Folder(Platform.ResolvePath(filename));
|
||||
}
|
||||
|
||||
public IReadOnlyPackage OpenPackage(string filename, IReadOnlyPackage parent)
|
||||
{
|
||||
// HACK: limit support to zip and folder until we generalize the PackageLoader support
|
||||
if (parent is Folder)
|
||||
{
|
||||
var path = Path.Combine(parent.Name, filename);
|
||||
|
||||
// HACK: work around SharpZipLib's lack of support for writing to in-memory files
|
||||
if (filename.EndsWith(".zip", StringComparison.InvariantCultureIgnoreCase))
|
||||
return new ZipFile(this, path);
|
||||
if (filename.EndsWith(".oramap", StringComparison.InvariantCultureIgnoreCase))
|
||||
return new ZipFile(this, path);
|
||||
|
||||
var subFolder = Platform.ResolvePath(path);
|
||||
if (Directory.Exists(subFolder))
|
||||
return new Folder(subFolder);
|
||||
}
|
||||
|
||||
if (filename.EndsWith(".zip", StringComparison.InvariantCultureIgnoreCase))
|
||||
return new ZipFile(this, filename, parent.GetStream(filename));
|
||||
if (filename.EndsWith(".oramap", StringComparison.InvariantCultureIgnoreCase))
|
||||
return new ZipFile(this, filename, parent.GetStream(filename));
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
public IReadWritePackage OpenWritablePackage(string filename)
|
||||
{
|
||||
if (filename.EndsWith(".zip", StringComparison.InvariantCultureIgnoreCase))
|
||||
@@ -144,7 +159,13 @@ namespace OpenRA.FileSystem
|
||||
packagesForFile.RemoveAll(p => p == package);
|
||||
|
||||
mountedPackages.Remove(package);
|
||||
explicitMounts.Remove(package.Name);
|
||||
var explicitKeys = explicitMounts.Where(kv => kv.Value == package)
|
||||
.Select(kv => kv.Key)
|
||||
.ToList();
|
||||
|
||||
foreach (var key in explicitKeys)
|
||||
explicitMounts.Remove(key);
|
||||
|
||||
package.Dispose();
|
||||
}
|
||||
else
|
||||
|
||||
@@ -18,16 +18,6 @@ namespace OpenRA.FileSystem
|
||||
{
|
||||
readonly string path;
|
||||
|
||||
// Create a new folder package
|
||||
public Folder(string path, Dictionary<string, byte[]> contents)
|
||||
{
|
||||
this.path = path;
|
||||
if (Directory.Exists(path))
|
||||
Directory.Delete(path, true);
|
||||
|
||||
Write(contents);
|
||||
}
|
||||
|
||||
public Folder(string path)
|
||||
{
|
||||
this.path = path;
|
||||
@@ -43,6 +33,8 @@ namespace OpenRA.FileSystem
|
||||
{
|
||||
foreach (var filename in Directory.GetFiles(path, "*", SearchOption.TopDirectoryOnly))
|
||||
yield return Path.GetFileName(filename);
|
||||
foreach (var filename in Directory.GetDirectories(path))
|
||||
yield return Path.GetFileName(filename);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -54,18 +46,31 @@ namespace OpenRA.FileSystem
|
||||
|
||||
public bool Contains(string filename)
|
||||
{
|
||||
return File.Exists(Path.Combine(path, filename));
|
||||
var combined = Path.Combine(path, filename);
|
||||
return combined.StartsWith(path) && File.Exists(combined);
|
||||
}
|
||||
|
||||
public void Write(Dictionary<string, byte[]> contents)
|
||||
public void Update(string filename, byte[] contents)
|
||||
{
|
||||
if (!Directory.Exists(path))
|
||||
Directory.CreateDirectory(path);
|
||||
|
||||
foreach (var file in contents)
|
||||
using (var dataStream = File.Create(Path.Combine(path, file.Key)))
|
||||
using (var writer = new BinaryWriter(dataStream))
|
||||
writer.Write(file.Value);
|
||||
using (var s = File.Create(Path.Combine(path, filename)))
|
||||
s.Write(contents, 0, contents.Length);
|
||||
}
|
||||
|
||||
public void Delete(string filename)
|
||||
{
|
||||
// HACK: ZipFiles can't be loaded as read-write from a stream, so we are
|
||||
// forced to bypass the parent package and load them with their full path
|
||||
// in FileSystem.OpenPackage. Their internal name therefore contains the
|
||||
// full parent path too. We need to be careful to not add a second path
|
||||
// prefix to these hacked packages.
|
||||
var filePath = filename.StartsWith(path) ? filename : Path.Combine(path, filename);
|
||||
if (Directory.Exists(filePath))
|
||||
Directory.Delete(filePath, true);
|
||||
else if (File.Exists(filePath))
|
||||
File.Delete(filePath);
|
||||
}
|
||||
|
||||
public void Dispose() { }
|
||||
|
||||
@@ -25,6 +25,7 @@ namespace OpenRA.FileSystem
|
||||
|
||||
public interface IReadWritePackage : IReadOnlyPackage
|
||||
{
|
||||
void Write(Dictionary<string, byte[]> contents);
|
||||
void Update(string filename, byte[] contents);
|
||||
void Delete(string filename);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -27,31 +27,24 @@ namespace OpenRA.FileSystem
|
||||
ZipConstants.DefaultCodePage = Encoding.UTF8.CodePage;
|
||||
}
|
||||
|
||||
public ZipFile(FileSystem context, string filename)
|
||||
public ZipFile(FileSystem context, string filename, Stream stream, bool createOrClearContents = false)
|
||||
{
|
||||
Name = filename;
|
||||
|
||||
try
|
||||
{
|
||||
// Pull the file into memory, don't keep it open.
|
||||
pkg = new SZipFile(new MemoryStream(File.ReadAllBytes(filename)));
|
||||
}
|
||||
catch (ZipException e)
|
||||
{
|
||||
Log.Write("debug", "Couldn't load zip file: {0}", e.Message);
|
||||
}
|
||||
if (createOrClearContents)
|
||||
pkg = SZipFile.Create(stream);
|
||||
else
|
||||
pkg = new SZipFile(stream);
|
||||
}
|
||||
|
||||
// Create a new zip with the specified contents.
|
||||
public ZipFile(FileSystem context, string filename, Dictionary<string, byte[]> contents)
|
||||
public ZipFile(IReadOnlyFileSystem context, string filename, bool createOrClearContents = false)
|
||||
{
|
||||
Name = filename;
|
||||
|
||||
if (File.Exists(filename))
|
||||
File.Delete(filename);
|
||||
|
||||
if (createOrClearContents)
|
||||
pkg = SZipFile.Create(filename);
|
||||
Write(contents);
|
||||
else
|
||||
pkg = new SZipFile(filename);
|
||||
}
|
||||
|
||||
public Stream GetStream(string filename)
|
||||
@@ -83,19 +76,18 @@ namespace OpenRA.FileSystem
|
||||
return pkg.GetEntry(filename) != null;
|
||||
}
|
||||
|
||||
public void Write(Dictionary<string, byte[]> contents)
|
||||
public void Update(string filename, byte[] contents)
|
||||
{
|
||||
// TODO: Clear existing content?
|
||||
pkg.Close();
|
||||
pkg = SZipFile.Create(Name);
|
||||
pkg.BeginUpdate();
|
||||
|
||||
foreach (var kvp in contents)
|
||||
pkg.Add(new StaticMemoryDataSource(kvp.Value), kvp.Key);
|
||||
|
||||
pkg.Add(new StaticMemoryDataSource(contents), filename);
|
||||
pkg.CommitUpdate();
|
||||
}
|
||||
|
||||
public void Delete(string filename)
|
||||
{
|
||||
pkg.BeginUpdate();
|
||||
pkg.Delete(filename);
|
||||
pkg.CommitUpdate();
|
||||
pkg.Close();
|
||||
pkg = new SZipFile(new MemoryStream(File.ReadAllBytes(Name)));
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
|
||||
@@ -490,9 +490,9 @@ namespace OpenRA
|
||||
return rules.Value;
|
||||
}
|
||||
|
||||
public void Save(IReadWritePackage toContainer)
|
||||
public void Save(IReadWritePackage toPackage)
|
||||
{
|
||||
MapFormat = 8;
|
||||
MapFormat = SupportedMapFormat;
|
||||
|
||||
var root = new List<MiniYamlNode>();
|
||||
var fields = new[]
|
||||
@@ -534,31 +534,19 @@ namespace OpenRA
|
||||
root.Add(new MiniYamlNode("Notifications", null, NotificationDefinitions));
|
||||
root.Add(new MiniYamlNode("Translations", null, TranslationDefinitions));
|
||||
|
||||
var entries = new Dictionary<string, byte[]>();
|
||||
entries.Add("map.bin", SaveBinaryData());
|
||||
var s = root.WriteToString();
|
||||
entries.Add("map.yaml", Encoding.UTF8.GetBytes(s));
|
||||
|
||||
// Add any custom assets
|
||||
if (Package != null)
|
||||
{
|
||||
// Saving to a new package: copy over all the content from the map
|
||||
if (Package != null && toPackage != Package)
|
||||
foreach (var file in Package.Contents)
|
||||
{
|
||||
if (file == "map.bin" || file == "map.yaml")
|
||||
continue;
|
||||
toPackage.Update(file, Package.GetStream(file).ReadAllBytes());
|
||||
|
||||
entries.Add(file, Package.GetStream(file).ReadAllBytes());
|
||||
}
|
||||
}
|
||||
|
||||
// Saving the map to a new location
|
||||
Package = toContainer;
|
||||
|
||||
// Update existing package
|
||||
toContainer.Write(entries);
|
||||
// Update the package with the new map data
|
||||
var s = root.WriteToString();
|
||||
toPackage.Update("map.yaml", Encoding.UTF8.GetBytes(s));
|
||||
toPackage.Update("map.bin", SaveBinaryData());
|
||||
Package = toPackage;
|
||||
|
||||
// Update UID to match the newly saved data
|
||||
Uid = ComputeUID(toContainer);
|
||||
Uid = ComputeUID(toPackage);
|
||||
}
|
||||
|
||||
public CellLayer<TerrainTile> LoadMapTiles()
|
||||
|
||||
@@ -26,6 +26,8 @@ namespace OpenRA
|
||||
public sealed class MapCache : IEnumerable<MapPreview>, IDisposable
|
||||
{
|
||||
public static readonly MapPreview UnknownMap = new MapPreview(null, MapGridType.Rectangular, null);
|
||||
public readonly IReadOnlyDictionary<IReadOnlyPackage, MapClassification> MapLocations;
|
||||
|
||||
readonly Cache<string, MapPreview> previews;
|
||||
readonly ModData modData;
|
||||
readonly SheetBuilder sheetBuilder;
|
||||
@@ -41,6 +43,36 @@ namespace OpenRA
|
||||
var gridType = Exts.Lazy(() => modData.Manifest.Get<MapGrid>().Type);
|
||||
previews = new Cache<string, MapPreview>(uid => new MapPreview(uid, gridType.Value, this));
|
||||
sheetBuilder = new SheetBuilder(SheetType.BGRA);
|
||||
|
||||
// Enumerate map directories
|
||||
var mapLocations = new Dictionary<IReadOnlyPackage, MapClassification>();
|
||||
foreach (var kv in modData.Manifest.MapFolders)
|
||||
{
|
||||
var name = kv.Key;
|
||||
var classification = string.IsNullOrEmpty(kv.Value)
|
||||
? MapClassification.Unknown : Enum<MapClassification>.Parse(kv.Value);
|
||||
|
||||
IReadOnlyPackage package;
|
||||
var optional = name.StartsWith("~");
|
||||
if (optional)
|
||||
name = name.Substring(1);
|
||||
|
||||
try
|
||||
{
|
||||
package = modData.ModFiles.OpenPackage(name);
|
||||
}
|
||||
catch
|
||||
{
|
||||
if (optional)
|
||||
continue;
|
||||
|
||||
throw;
|
||||
}
|
||||
|
||||
mapLocations.Add(package, classification);
|
||||
}
|
||||
|
||||
MapLocations = new ReadOnlyDictionary<IReadOnlyPackage, MapClassification>(mapLocations);
|
||||
}
|
||||
|
||||
public void LoadMaps()
|
||||
@@ -49,32 +81,36 @@ namespace OpenRA
|
||||
if (!modData.Manifest.Contains<MapGrid>())
|
||||
return;
|
||||
|
||||
// Expand the dictionary (dir path, dir type) to a dictionary of (map path, dir type)
|
||||
var mapPaths = modData.Manifest.MapFolders.SelectMany(kv =>
|
||||
FindMapsIn(modData.ModFiles, kv.Key).ToDictionary(p => p, p => string.IsNullOrEmpty(kv.Value)
|
||||
? MapClassification.Unknown : Enum<MapClassification>.Parse(kv.Value)));
|
||||
|
||||
var mapGrid = modData.Manifest.Get<MapGrid>();
|
||||
foreach (var path in mapPaths)
|
||||
var mapGrid = Game.ModData.Manifest.Get<MapGrid>();
|
||||
foreach (var kv in MapLocations)
|
||||
{
|
||||
foreach (var map in kv.Key.Contents)
|
||||
{
|
||||
IReadOnlyPackage mapPackage = null;
|
||||
try
|
||||
{
|
||||
using (new Support.PerfTimer(path.Key))
|
||||
using (new Support.PerfTimer(map))
|
||||
{
|
||||
var package = modData.ModFiles.OpenPackage(path.Key);
|
||||
var uid = Map.ComputeUID(package);
|
||||
previews[uid].UpdateFromMap(package, path.Value, modData.Manifest.MapCompatibility, mapGrid.Type);
|
||||
mapPackage = modData.ModFiles.OpenPackage(map, kv.Key);
|
||||
if (mapPackage == null)
|
||||
continue;
|
||||
|
||||
var uid = Map.ComputeUID(mapPackage);
|
||||
previews[uid].UpdateFromMap(mapPackage, kv.Key, kv.Value, modData.Manifest.MapCompatibility, mapGrid.Type);
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
Console.WriteLine("Failed to load map: {0}", path);
|
||||
if (mapPackage != null)
|
||||
mapPackage.Dispose();
|
||||
Console.WriteLine("Failed to load map: {0}", map);
|
||||
Console.WriteLine("Details: {0}", e);
|
||||
Log.Write("debug", "Failed to load map: {0}", path);
|
||||
Log.Write("debug", "Failed to load map: {0}", map);
|
||||
Log.Write("debug", "Details: {0}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void QueryRemoteMapDetails(IEnumerable<string> uids)
|
||||
{
|
||||
@@ -119,31 +155,6 @@ namespace OpenRA
|
||||
new Download(url, _ => { }, onInfoComplete);
|
||||
}
|
||||
|
||||
public static IEnumerable<string> FindMapsIn(FileSystem.FileSystem context, string dir)
|
||||
{
|
||||
string[] noMaps = { };
|
||||
|
||||
// Ignore optional flag
|
||||
if (dir.StartsWith("~"))
|
||||
dir = dir.Substring(1);
|
||||
|
||||
// HACK: We currently only support maps loaded from Folders
|
||||
// This is a temporary workaround that resolves the filesystem paths to a system directory
|
||||
IReadOnlyPackage package;
|
||||
string filename;
|
||||
if (context.TryGetPackageContaining(dir, out package, out filename))
|
||||
dir = Path.Combine(package.Name, filename);
|
||||
else if (Directory.Exists(Platform.ResolvePath(dir)))
|
||||
dir = Platform.ResolvePath(dir);
|
||||
else
|
||||
return noMaps;
|
||||
|
||||
var dirsWithMaps = Directory.GetDirectories(dir)
|
||||
.Where(d => Directory.GetFiles(d, "map.yaml").Any() && Directory.GetFiles(d, "map.bin").Any());
|
||||
|
||||
return dirsWithMaps.Concat(Directory.GetFiles(dir, "*.oramap"));
|
||||
}
|
||||
|
||||
void LoadAsyncInternal()
|
||||
{
|
||||
Log.Write("debug", "MapCache.LoadAsyncInternal started");
|
||||
@@ -269,6 +280,9 @@ namespace OpenRA
|
||||
return;
|
||||
}
|
||||
|
||||
foreach (var p in previews.Values)
|
||||
p.Dispose();
|
||||
|
||||
// We need to let the loader thread exit before we can dispose our sheet builder.
|
||||
// Ideally we should dispose our resources before returning, but we don't to block waiting on the loader thread to exit.
|
||||
// Instead, we'll queue disposal to be run once it has exited.
|
||||
|
||||
@@ -53,13 +53,14 @@ namespace OpenRA
|
||||
public readonly bool downloading;
|
||||
}
|
||||
|
||||
public class MapPreview
|
||||
public class MapPreview : IDisposable
|
||||
{
|
||||
static readonly CPos[] NoSpawns = new CPos[] { };
|
||||
MapCache cache;
|
||||
|
||||
public readonly string Uid;
|
||||
public IReadOnlyPackage Package { get; private set; }
|
||||
IReadOnlyPackage parentPackage;
|
||||
|
||||
public string Title { get; private set; }
|
||||
public string Type { get; private set; }
|
||||
@@ -116,7 +117,7 @@ namespace OpenRA
|
||||
Visibility = MapVisibility.Lobby;
|
||||
}
|
||||
|
||||
public void UpdateFromMap(IReadOnlyPackage p, MapClassification classification, string[] mapCompatibility, MapGridType gridType)
|
||||
public void UpdateFromMap(IReadOnlyPackage p, IReadOnlyPackage parent, MapClassification classification, string[] mapCompatibility, MapGridType gridType)
|
||||
{
|
||||
Dictionary<string, MiniYaml> yaml;
|
||||
using (var yamlStream = p.GetStream("map.yaml"))
|
||||
@@ -128,6 +129,7 @@ namespace OpenRA
|
||||
}
|
||||
|
||||
Package = p;
|
||||
parentPackage = parent;
|
||||
GridType = gridType;
|
||||
Class = classification;
|
||||
|
||||
@@ -270,19 +272,22 @@ namespace OpenRA
|
||||
return;
|
||||
|
||||
Status = MapStatus.Downloading;
|
||||
var baseMapPath = Platform.ResolvePath("^", "maps", Game.ModData.Manifest.Mod.Id);
|
||||
|
||||
// Create the map directory if it doesn't exist
|
||||
if (!Directory.Exists(baseMapPath))
|
||||
Directory.CreateDirectory(baseMapPath);
|
||||
var installLocation = cache.MapLocations.FirstOrDefault(p => p.Value == MapClassification.User);
|
||||
if (installLocation.Key == null || !(installLocation.Key is IReadWritePackage))
|
||||
{
|
||||
Log.Write("debug", "Map install directory not found");
|
||||
Status = MapStatus.DownloadError;
|
||||
return;
|
||||
}
|
||||
|
||||
var mapInstallPackage = installLocation.Key as IReadWritePackage;
|
||||
var modData = Game.ModData;
|
||||
new Thread(() =>
|
||||
{
|
||||
// Request the filename from the server
|
||||
// Run in a worker thread to avoid network delays
|
||||
var mapUrl = Game.Settings.Game.MapRepository + Uid;
|
||||
var mapPath = string.Empty;
|
||||
var mapFilename = string.Empty;
|
||||
try
|
||||
{
|
||||
var request = WebRequest.Create(mapUrl);
|
||||
@@ -296,11 +301,11 @@ namespace OpenRA
|
||||
return;
|
||||
}
|
||||
|
||||
mapPath = System.IO.Path.Combine(baseMapPath, res.Headers["Content-Disposition"].Replace("attachment; filename = ", ""));
|
||||
mapFilename = res.Headers["Content-Disposition"].Replace("attachment; filename = ", "");
|
||||
}
|
||||
|
||||
Action<DownloadProgressChangedEventArgs> onDownloadProgress = i => { DownloadBytes = i.BytesReceived; DownloadPercentage = i.ProgressPercentage; };
|
||||
Action<AsyncCompletedEventArgs, bool> onDownloadComplete = (i, cancelled) =>
|
||||
Action<DownloadDataCompletedEventArgs, bool> onDownloadComplete = (i, cancelled) =>
|
||||
{
|
||||
download = null;
|
||||
|
||||
@@ -313,15 +318,16 @@ namespace OpenRA
|
||||
return;
|
||||
}
|
||||
|
||||
Log.Write("debug", "Downloaded map to '{0}'", mapPath);
|
||||
mapInstallPackage.Update(mapFilename, i.Result);
|
||||
Log.Write("debug", "Downloaded map to '{0}'", mapFilename);
|
||||
Game.RunAfterTick(() =>
|
||||
{
|
||||
using (var package = modData.ModFiles.OpenPackage(mapPath))
|
||||
UpdateFromMap(package, MapClassification.User, null, GridType);
|
||||
var package = modData.ModFiles.OpenPackage(mapFilename, mapInstallPackage);
|
||||
UpdateFromMap(package, mapInstallPackage, MapClassification.User, null, GridType);
|
||||
});
|
||||
};
|
||||
|
||||
download = new Download(mapUrl, mapPath, onDownloadProgress, onDownloadComplete);
|
||||
download = new Download(mapUrl, onDownloadProgress, onDownloadComplete);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
@@ -344,5 +350,22 @@ namespace OpenRA
|
||||
{
|
||||
Status = MapStatus.Unavailable;
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
if (Package != null)
|
||||
{
|
||||
Package.Dispose();
|
||||
Package = null;
|
||||
}
|
||||
}
|
||||
|
||||
public void Delete()
|
||||
{
|
||||
Invalidate();
|
||||
var deleteFromPackage = parentPackage as IReadWritePackage;
|
||||
if (deleteFromPackage != null)
|
||||
deleteFromPackage.Delete(Package.Name);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -14,6 +14,7 @@ using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using OpenRA.FileSystem;
|
||||
using OpenRA.Graphics;
|
||||
using OpenRA.Mods.Common.FileFormats;
|
||||
using OpenRA.Mods.Common.Traits;
|
||||
@@ -95,7 +96,7 @@ namespace OpenRA.Mods.Common.UtilityCommands
|
||||
Map.FixOpenAreas(Rules);
|
||||
|
||||
var dest = Path.GetFileNameWithoutExtension(args[1]) + ".oramap";
|
||||
var package = modData.ModFiles.CreatePackage(dest);
|
||||
var package = new ZipFile(modData.ModFiles, dest, true);
|
||||
Map.Save(package);
|
||||
Console.WriteLine(dest + " saved.");
|
||||
}
|
||||
|
||||
@@ -785,17 +785,7 @@ namespace OpenRA.Mods.Common.UtilityCommands
|
||||
|
||||
yaml.Nodes.First(n => n.Key == "MapFormat").Value = new MiniYaml(Map.SupportedMapFormat.ToString());
|
||||
|
||||
var entries = new Dictionary<string, byte[]>();
|
||||
entries.Add("map.yaml", Encoding.UTF8.GetBytes(yaml.Nodes.WriteToString()));
|
||||
foreach (var file in package.Contents)
|
||||
{
|
||||
if (file == "map.yaml")
|
||||
continue;
|
||||
|
||||
entries.Add(file, package.GetStream(file).ReadAllBytes());
|
||||
}
|
||||
|
||||
package.Write(entries);
|
||||
package.Update("map.yaml", Encoding.UTF8.GetBytes(yaml.Nodes.WriteToString()));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -28,6 +28,20 @@ namespace OpenRA.Mods.Common.Widgets.Logic
|
||||
public string UiLabel;
|
||||
}
|
||||
|
||||
class SaveDirectory
|
||||
{
|
||||
public readonly Folder Folder;
|
||||
public readonly string DisplayName;
|
||||
public readonly MapClassification Classification;
|
||||
|
||||
public SaveDirectory(Folder folder, MapClassification classification)
|
||||
{
|
||||
Folder = folder;
|
||||
DisplayName = Platform.UnresolvePath(Folder.Name);
|
||||
Classification = classification;
|
||||
}
|
||||
}
|
||||
|
||||
[ObjectCreator.UseCtor]
|
||||
public SaveMapLogic(Widget widget, ModData modData, Action<string> onSave, Action onExit,
|
||||
Map map, List<MiniYamlNode> playerDefinitions, List<MiniYamlNode> actorDefinitions)
|
||||
@@ -60,58 +74,57 @@ namespace OpenRA.Mods.Common.Widgets.Logic
|
||||
visibilityDropdown.ShowDropDown("LABEL_DROPDOWN_TEMPLATE", 210, mapVisibility, setupItem);
|
||||
}
|
||||
|
||||
Func<string, string> makeMapDirectory = dir =>
|
||||
{
|
||||
if (dir.StartsWith("~"))
|
||||
dir = dir.Substring(1);
|
||||
var writableDirectories = new List<SaveDirectory>();
|
||||
SaveDirectory selectedDirectory = null;
|
||||
|
||||
IReadOnlyPackage package;
|
||||
string f;
|
||||
if (modData.ModFiles.TryGetPackageContaining(dir, out package, out f))
|
||||
dir = Path.Combine(package.Name, f);
|
||||
|
||||
return Platform.UnresolvePath(dir);
|
||||
};
|
||||
|
||||
var mapDirectories = modData.Manifest.MapFolders
|
||||
.ToDictionary(kv => makeMapDirectory(kv.Key), kv => Enum<MapClassification>.Parse(kv.Value));
|
||||
|
||||
var mapPath = map.Package != null ? map.Package.Name : null;
|
||||
var directoryDropdown = widget.Get<DropDownButtonWidget>("DIRECTORY_DROPDOWN");
|
||||
{
|
||||
Func<string, ScrollItemWidget, ScrollItemWidget> setupItem = (option, template) =>
|
||||
Func<SaveDirectory, ScrollItemWidget, ScrollItemWidget> setupItem = (option, template) =>
|
||||
{
|
||||
var item = ScrollItemWidget.Setup(template,
|
||||
() => directoryDropdown.Text == option,
|
||||
() => directoryDropdown.Text = option);
|
||||
item.Get<LabelWidget>("LABEL").GetText = () => option;
|
||||
() => selectedDirectory == option,
|
||||
() => selectedDirectory = option);
|
||||
item.Get<LabelWidget>("LABEL").GetText = () => option.DisplayName;
|
||||
return item;
|
||||
};
|
||||
|
||||
// TODO: This won't work for maps inside oramod packages
|
||||
var mapDirectory = mapPath != null ? Platform.UnresolvePath(Path.GetDirectoryName(mapPath)) : null;
|
||||
var initialDirectory = mapDirectories.Keys.FirstOrDefault(f => f == mapDirectory);
|
||||
foreach (var kv in modData.MapCache.MapLocations)
|
||||
{
|
||||
var folder = kv.Key as Folder;
|
||||
if (folder == null)
|
||||
continue;
|
||||
|
||||
try
|
||||
{
|
||||
using (var fs = File.Create(Path.Combine(folder.Name, ".testwritable"), 1, FileOptions.DeleteOnClose))
|
||||
{
|
||||
// Do nothing: we just want to test whether we can create the file
|
||||
}
|
||||
|
||||
writableDirectories.Add(new SaveDirectory(folder, kv.Value));
|
||||
}
|
||||
catch
|
||||
{
|
||||
// Directory is not writable
|
||||
}
|
||||
}
|
||||
|
||||
if (map.Package != null)
|
||||
selectedDirectory = writableDirectories.FirstOrDefault(k => k.Folder.Contains(map.Package.Name));
|
||||
|
||||
// Prioritize MapClassification.User directories over system directories
|
||||
if (initialDirectory == null)
|
||||
initialDirectory = mapDirectories.OrderByDescending(kv => kv.Value).First().Key;
|
||||
if (selectedDirectory == null)
|
||||
selectedDirectory = writableDirectories.OrderByDescending(kv => kv.Classification).First();
|
||||
|
||||
directoryDropdown.Text = initialDirectory;
|
||||
directoryDropdown.GetText = () => selectedDirectory == null ? "" : selectedDirectory.DisplayName;
|
||||
directoryDropdown.OnClick = () =>
|
||||
directoryDropdown.ShowDropDown("LABEL_DROPDOWN_TEMPLATE", 210, mapDirectories.Keys, setupItem);
|
||||
directoryDropdown.ShowDropDown("LABEL_DROPDOWN_TEMPLATE", 210, writableDirectories, setupItem);
|
||||
}
|
||||
|
||||
var mapIsUnpacked = false;
|
||||
|
||||
// TODO: This won't work for maps inside oramod packages
|
||||
if (mapPath != null)
|
||||
{
|
||||
var attr = File.GetAttributes(mapPath);
|
||||
mapIsUnpacked = attr.HasFlag(FileAttributes.Directory);
|
||||
}
|
||||
var mapIsUnpacked = map.Package != null && map.Package is Folder;
|
||||
|
||||
var filename = widget.Get<TextFieldWidget>("FILENAME");
|
||||
filename.Text = mapIsUnpacked ? Path.GetFileName(mapPath) : Path.GetFileNameWithoutExtension(mapPath);
|
||||
filename.Text = map.Package == null ? "" : mapIsUnpacked ? Path.GetFileName(map.Package.Name) : Path.GetFileNameWithoutExtension(map.Package.Name);
|
||||
var fileType = mapIsUnpacked ? MapFileType.Unpacked : MapFileType.OraMap;
|
||||
|
||||
var fileTypes = new Dictionary<MapFileType, MapFileTypeInfo>()
|
||||
@@ -159,22 +172,33 @@ namespace OpenRA.Mods.Common.Widgets.Logic
|
||||
|
||||
map.RequiresMod = modData.Manifest.Mod.Id;
|
||||
|
||||
// Create the map directory if required
|
||||
Directory.CreateDirectory(Platform.ResolvePath(directoryDropdown.Text));
|
||||
|
||||
// TODO: This won't work for maps inside oramod packages
|
||||
var combinedPath = Platform.ResolvePath(Path.Combine(directoryDropdown.Text, filename.Text + fileTypes[fileType].Extension));
|
||||
var combinedPath = Platform.ResolvePath(Path.Combine(selectedDirectory.Folder.Name, filename.Text + fileTypes[fileType].Extension));
|
||||
|
||||
// Invalidate the old map metadata
|
||||
if (map.Uid != null && combinedPath == mapPath)
|
||||
if (map.Uid != null && map.Package != null && map.Package.Name == combinedPath)
|
||||
modData.MapCache[map.Uid].Invalidate();
|
||||
|
||||
var package = modData.ModFiles.CreatePackage(combinedPath);
|
||||
try
|
||||
{
|
||||
var package = map.Package as IReadWritePackage;
|
||||
if (package == null || package.Name != combinedPath)
|
||||
{
|
||||
selectedDirectory.Folder.Delete(combinedPath);
|
||||
if (fileType == MapFileType.OraMap)
|
||||
package = new ZipFile(modData.DefaultFileSystem, combinedPath, true);
|
||||
else
|
||||
package = new Folder(combinedPath);
|
||||
}
|
||||
|
||||
map.Save(package);
|
||||
}
|
||||
catch
|
||||
{
|
||||
Console.WriteLine("Failed to save map at {0}", combinedPath);
|
||||
}
|
||||
|
||||
// Update the map cache so it can be loaded without restarting the game
|
||||
var classification = mapDirectories[directoryDropdown.Text];
|
||||
modData.MapCache[map.Uid].UpdateFromMap(map.Package, classification, null, map.Grid.Type);
|
||||
modData.MapCache[map.Uid].UpdateFromMap(map.Package, selectedDirectory.Folder, selectedDirectory.Classification, null, map.Grid.Type);
|
||||
|
||||
Console.WriteLine("Saved current map at {0}", combinedPath);
|
||||
Ui.CloseWindow();
|
||||
|
||||
@@ -288,22 +288,15 @@ namespace OpenRA.Mods.Common.Widgets.Logic
|
||||
|
||||
string DeleteMap(string map)
|
||||
{
|
||||
var path = modData.MapCache[map].Package.Name;
|
||||
try
|
||||
{
|
||||
if (File.Exists(path))
|
||||
File.Delete(path);
|
||||
else if (Directory.Exists(path))
|
||||
Directory.Delete(path, true);
|
||||
|
||||
modData.MapCache[map].Invalidate();
|
||||
|
||||
modData.MapCache[map].Delete();
|
||||
if (selectedUid == map)
|
||||
selectedUid = WidgetUtils.ChooseInitialMap(tabMaps[currentTab].Select(mp => mp.Uid).FirstOrDefault());
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Game.Debug("Failed to delete map '{0}'. See the debug.log file for details.", path);
|
||||
Game.Debug("Failed to delete map '{0}'. See the debug.log file for details.", map);
|
||||
Log.Write("debug", ex.ToString());
|
||||
}
|
||||
|
||||
|
||||
@@ -11,6 +11,7 @@
|
||||
|
||||
using System;
|
||||
using System.IO;
|
||||
using OpenRA.FileSystem;
|
||||
|
||||
namespace OpenRA.Mods.D2k.UtilityCommands
|
||||
{
|
||||
@@ -37,7 +38,7 @@ namespace OpenRA.Mods.D2k.UtilityCommands
|
||||
return;
|
||||
|
||||
var dest = Path.GetFileNameWithoutExtension(args[1]) + ".oramap";
|
||||
var package = modData.ModFiles.CreatePackage(dest);
|
||||
var package = new ZipFile(modData.DefaultFileSystem, dest, true);
|
||||
map.Save(package);
|
||||
Console.WriteLine(dest + " saved.");
|
||||
}
|
||||
|
||||
@@ -179,7 +179,7 @@ namespace OpenRA.Mods.TS.UtilityCommands
|
||||
map.PlayerDefinitions = mapPlayers.ToMiniYaml();
|
||||
|
||||
var dest = Path.GetFileNameWithoutExtension(args[1]) + ".oramap";
|
||||
var package = modData.ModFiles.CreatePackage(dest);
|
||||
var package = new ZipFile(modData.DefaultFileSystem, dest, true);
|
||||
map.Save(package);
|
||||
Console.WriteLine(dest + " saved.");
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user