| 1 |
using System; |
| 2 |
using System.Collections.Generic; |
| 3 |
using System.IO; |
| 4 |
using Oni.Collections; |
| 5 |
|
| 6 |
namespace Oni |
| 7 |
{ |
| 8 |
internal sealed class InstanceFileManager |
| 9 |
{ |
| 10 |
private readonly List<string> searchPaths = new List<string>(); |
| 11 |
private readonly Dictionary<string, InstanceFile> loadedFiles = new Dictionary<string, InstanceFile>(StringComparer.OrdinalIgnoreCase); |
| 12 |
private Dictionary<string, string> files; |
| 13 |
|
| 14 |
public InstanceFile OpenFile(string filePath) |
| 15 |
{ |
| 16 |
InstanceFile file; |
| 17 |
|
| 18 |
if (!loadedFiles.TryGetValue(filePath, out file)) |
| 19 |
{ |
| 20 |
try |
| 21 |
{ |
| 22 |
file = InstanceFile.Read(this, filePath); |
| 23 |
} |
| 24 |
catch (Exception ex) |
| 25 |
{ |
| 26 |
Console.Error.WriteLine("Error opening file {0}: {1}", filePath, ex.Message); |
| 27 |
throw; |
| 28 |
} |
| 29 |
|
| 30 |
loadedFiles.Add(filePath, file); |
| 31 |
} |
| 32 |
|
| 33 |
return file; |
| 34 |
} |
| 35 |
|
| 36 |
public List<InstanceFile> OpenDirectories(string[] dirPaths) |
| 37 |
{ |
| 38 |
var files = new List<InstanceFile>(); |
| 39 |
var seenFileNames = new Set<string>(StringComparer.Ordinal); |
| 40 |
|
| 41 |
Array.Reverse(dirPaths); |
| 42 |
|
| 43 |
foreach (string dirPath in dirPaths) |
| 44 |
{ |
| 45 |
var filePaths = FindFiles(dirPath); |
| 46 |
|
| 47 |
foreach (string filePath in filePaths) |
| 48 |
{ |
| 49 |
if (!seenFileNames.Contains(Path.GetFileName(filePath))) |
| 50 |
files.Add(OpenFile(filePath)); |
| 51 |
} |
| 52 |
|
| 53 |
foreach (string filePath in filePaths) |
| 54 |
seenFileNames.Add(Path.GetFileName(filePath)); |
| 55 |
} |
| 56 |
|
| 57 |
return files; |
| 58 |
} |
| 59 |
|
| 60 |
public List<InstanceFile> OpenDirectory(string dirPath) |
| 61 |
{ |
| 62 |
var files = new List<InstanceFile>(); |
| 63 |
|
| 64 |
foreach (string filePath in FindFiles(dirPath)) |
| 65 |
files.Add(OpenFile(filePath)); |
| 66 |
|
| 67 |
return files; |
| 68 |
} |
| 69 |
|
| 70 |
public void AddSearchPath(string path) |
| 71 |
{ |
| 72 |
if (Directory.Exists(path)) |
| 73 |
searchPaths.Add(path); |
| 74 |
} |
| 75 |
|
| 76 |
public InstanceFile FindInstance(string instanceName) |
| 77 |
{ |
| 78 |
if (files == null) |
| 79 |
{ |
| 80 |
files = new Dictionary<string, string>(StringComparer.Ordinal); |
| 81 |
|
| 82 |
foreach (string searchPath in searchPaths) |
| 83 |
{ |
| 84 |
foreach (string filePath in FindFiles(searchPath)) |
| 85 |
{ |
| 86 |
string name = Importer.DecodeFileName(filePath); |
| 87 |
|
| 88 |
if (!files.ContainsKey(name)) |
| 89 |
files[name] = filePath; |
| 90 |
} |
| 91 |
} |
| 92 |
} |
| 93 |
|
| 94 |
string instanceFilePath; |
| 95 |
|
| 96 |
if (!files.TryGetValue(instanceName, out instanceFilePath)) |
| 97 |
return null; |
| 98 |
|
| 99 |
var file = OpenFile(instanceFilePath); |
| 100 |
|
| 101 |
if (file == null) |
| 102 |
return null; |
| 103 |
|
| 104 |
var descriptor = file.Descriptors[0]; |
| 105 |
|
| 106 |
if (file.Header.Version != InstanceFileHeader.Version32) |
| 107 |
return null; |
| 108 |
|
| 109 |
if (descriptor == null || !descriptor.HasName || descriptor.FullName != instanceName) |
| 110 |
return null; |
| 111 |
|
| 112 |
return file; |
| 113 |
} |
| 114 |
|
| 115 |
public InstanceFile FindInstance(string instanceName, InstanceFile baseFile) |
| 116 |
{ |
| 117 |
if (files == null) |
| 118 |
{ |
| 119 |
files = new Dictionary<string, string>(StringComparer.Ordinal); |
| 120 |
|
| 121 |
foreach (string filePath in FindFiles(Path.GetDirectoryName(baseFile.FilePath))) |
| 122 |
files[Importer.DecodeFileName(filePath)] = filePath; |
| 123 |
|
| 124 |
foreach (string searchPath in searchPaths) |
| 125 |
{ |
| 126 |
foreach (string filePath in FindFiles(searchPath)) |
| 127 |
{ |
| 128 |
string name = Importer.DecodeFileName(filePath); |
| 129 |
|
| 130 |
if (!files.ContainsKey(name)) |
| 131 |
files[name] = filePath; |
| 132 |
} |
| 133 |
} |
| 134 |
} |
| 135 |
|
| 136 |
string instanceFilePath; |
| 137 |
|
| 138 |
if (!files.TryGetValue(instanceName, out instanceFilePath)) |
| 139 |
{ |
| 140 |
if (instanceName.Length > 4) |
| 141 |
instanceName = instanceName.Substring(4); |
| 142 |
|
| 143 |
if (!files.TryGetValue(instanceName, out instanceFilePath)) |
| 144 |
{ |
| 145 |
string level0FilePath = Path.Combine(Path.GetDirectoryName(baseFile.FilePath), "level0_Final.dat"); |
| 146 |
|
| 147 |
if (!File.Exists(level0FilePath)) |
| 148 |
return null; |
| 149 |
|
| 150 |
return OpenFile(level0FilePath); |
| 151 |
} |
| 152 |
} |
| 153 |
|
| 154 |
var file = OpenFile(instanceFilePath); |
| 155 |
|
| 156 |
if (file == null || file == baseFile) |
| 157 |
return null; |
| 158 |
|
| 159 |
var descriptor = file.Descriptors[0]; |
| 160 |
|
| 161 |
if (file.Header.Version == InstanceFileHeader.Version32) |
| 162 |
return file; |
| 163 |
|
| 164 |
if (descriptor == null || !descriptor.HasName || descriptor.FullName != instanceName) |
| 165 |
return null; |
| 166 |
|
| 167 |
return file; |
| 168 |
} |
| 169 |
|
| 170 |
private static List<string> FindFiles(string dirPath) |
| 171 |
{ |
| 172 |
var files = new List<string>(); |
| 173 |
|
| 174 |
if (Directory.Exists(dirPath)) |
| 175 |
FindFilesRecursive(dirPath, files); |
| 176 |
|
| 177 |
return files; |
| 178 |
} |
| 179 |
|
| 180 |
private static void FindFilesRecursive(string dirPath, List<string> files) |
| 181 |
{ |
| 182 |
files.AddRange(Directory.GetFiles(dirPath, "*.oni")); |
| 183 |
|
| 184 |
foreach (string childDirPath in Directory.GetDirectories(dirPath)) |
| 185 |
{ |
| 186 |
string name = Path.GetFileName(childDirPath); |
| 187 |
|
| 188 |
if (!string.Equals(name, "_noimport", StringComparison.OrdinalIgnoreCase) |
| 189 |
&& !string.Equals(name, "noimport", StringComparison.OrdinalIgnoreCase)) |
| 190 |
{ |
| 191 |
FindFilesRecursive(childDirPath, files); |
| 192 |
} |
| 193 |
} |
| 194 |
} |
| 195 |
} |
| 196 |
} |
