public void AssignToSectorEndTest()
{
var firstRange = new SectorRange(0, 1000);
var secondRange = new SectorRange(1000, 2000);
var firstSector = CreateMock <ISector>();
var secondSector = CreateMock <ISector>();
var collection = new SectorCollection();
collection.AssignSector(firstRange, firstSector);
collection.AssignSector(secondRange, secondSector);
var sectorPairs = collection.EnumerateSectorPairs().ToList();
AssertEquals(2, sectorPairs.Count);
AssertEquals(new SectorRange(0, 1000), sectorPairs[0].Key);
AssertEquals(firstSector, sectorPairs[0].Value);
AssertEquals(new SectorRange(1000, 2000), sectorPairs[1].Key);
AssertEquals(secondSector, sectorPairs[1].Value);
VerifyNoMoreInteractions();
}
public void AssignSectorTest()
{
var range = new SectorRange(1000, 2000);
var sector = CreateMock <ISector>();
var collection = new SectorCollection();
collection.AssignSector(range, sector);
var sectors = collection.EnumerateSectors().ToList();
AssertEquals(1, sectors.Count);
AssertEquals(sector, sectors[0]);
VerifyNoMoreInteractions();
}
public CommandList LinkGameModifications(IReadOnlyList <Modification> modifications)
{
var archiveLoader = new RiotArchiveLoader(leagueConfiguration.RadsPath);
Dictionary <uint, List <KeyValuePair <RiotArchive, SectorCollection> > > archivePairsById = new Dictionary <uint, List <KeyValuePair <RiotArchive, SectorCollection> > >();
var riotSolution = riotSolutionLoader.Load(leagueConfiguration.RadsPath, RiotProjectType.GameClient);
var gameProject = riotSolution.ProjectsByType[RiotProjectType.GameClient];
var gameManifest = gameProject.ReleaseManifest;
foreach (var modification in modifications)
{
var resolutionTableComponent = modification.GetComponent <LeagueResolutionTableComponent>();
var resolutionTable = resolutionTableComponent.Table;
var contentPath = Path.Combine(modification.RepositoryPath, "content");
var contentDirectory = fileSystemProxy.GetDirectoryInfo(contentPath);
var contentFiles = contentDirectory.EnumerateFiles("*", SearchOption.AllDirectories);
var contentRelativePaths = contentFiles.Select(f => f.FullName.Substring(contentPath.Length + 1));
var objectsPath = Path.Combine(modification.RepositoryPath, "objects");
foreach (var contentRelativePath in contentRelativePaths)
{
LeagueResolutionTableValue resolutionTableValue;
if (!resolutionTable.TryGetValue(contentRelativePath, out resolutionTableValue))
{
logger?.Warn($"Not linking {contentRelativePath}: resolution table lookup failed");
continue;
}
if (resolutionTableValue.Target != LeagueModificationTarget.Game)
{
logger?.Warn($"Not linking {contentRelativePath}: does not target game client.");
continue;
}
var manifestEntry = gameManifest.Root.GetRelativeOrNull <ReleaseManifestFileEntry>(resolutionTableValue.ResolvedPath);
if (manifestEntry == null)
{
logger?.Warn($"Not linking {contentRelativePath}: could not find {resolutionTableValue.ResolvedPath} in manifest.");
continue;
}
List <KeyValuePair <RiotArchive, SectorCollection> > archivePairs;
if (!archivePairsById.TryGetValue(manifestEntry.ArchiveId, out archivePairs))
{
archivePairs = new List <KeyValuePair <RiotArchive, SectorCollection> >();
IReadOnlyList <RiotArchive> archives;
if (!archiveLoader.TryLoadArchives(manifestEntry.ArchiveId, out archives))
{
logger?.Error($"Not linking {contentRelativePath}: could not load archives for {manifestEntry.ArchiveId}!");
continue;
}
archives.ForEach(archive => {
var datLength = new FileInfo(archive.DatFilePath).Length;
SectorCollection sectorCollection = new SectorCollection();
sectorCollection.AssignSector(new SectorRange(0, datLength), new FileSector(archive.DatFilePath, 0, datLength));
archivePairs.Add(archive.PairValue(sectorCollection));
});
archivePairsById.Add(manifestEntry.ArchiveId, archivePairs);
}
var rafPath = RAFUtil.FormatPathToRAFPath(resolutionTableValue.ResolvedPath);
RAFFileListEntry rafEntry = null;
SectorCollection sectors = null;
foreach (var archivePair in archivePairs)
{
rafEntry = archivePair.Key.GetDirectoryFile().GetFileList().GetFileEntryOrNull(rafPath);
if (rafEntry != null)
{
sectors = archivePair.Value;
break;
}
}
if (rafEntry == null)
{
logger?.Warn($"Not linking {contentRelativePath}: could not find {resolutionTableValue.ResolvedPath} in {manifestEntry.ArchiveId} archives.");
continue;
}
var vfmEndOffset = sectors.EnumerateSectorPairs().Last().Key.endExclusive;
sectors.DeleteRange(rafEntry.FileOffset, rafEntry.FileOffset + rafEntry.FileSize);
var objectsFilePath = Path.Combine(objectsPath, contentRelativePath);
var objectsFileLength = new FileInfo(objectsFilePath).Length;
rafEntry.FileOffset = (uint)vfmEndOffset;
rafEntry.FileSize = (uint)objectsFileLength;
sectors.AssignSector(new SectorRange(vfmEndOffset, vfmEndOffset + objectsFileLength), new FileSector(objectsFilePath, 0, objectsFileLength));
var originalFileLength = new FileInfo(Path.Combine(contentPath, contentRelativePath)).Length;
manifestEntry.CompressedSize = (uint)objectsFileLength;
manifestEntry.DecompressedSize = (uint)originalFileLength;
logger?.Warn("Successfully linked " + resolutionTableValue.ResolvedPath + " in archive " + manifestEntry.ArchiveId + ".");
}
}
var commandList = new DefaultCommandList();
var versionStringUtilities = new VersionStringUtilities();
var tempDir = temporaryFileService.AllocateTemporaryDirectory(TimeSpan.FromMinutes(1));
logger?.Info("Allocated temporary directory " + tempDir);
foreach (var archivePair in archivePairsById)
{
string versionString = versionStringUtilities.GetVersionString(archivePair.Key);
foreach (var archiveData in archivePair.Value)
{
// Get archive name (e.g. archive_2.raf or archive_12930813.raf)
string archiveFileName = archiveData.Key.RAFFilePath.With(x => x.Substring(x.LastIndexOfAny(new[] { '/', '\\' }) + 1));
// Serialize the VFM
var vfmSerializer = new SectorCollectionSerializer();
var vfmFileName = versionString + "/" + archiveFileName + ".dat.vfm";
var vfmPath = temporaryFileService.AllocateTemporaryFile(tempDir, vfmFileName);
using (var vfmFileStream = File.Open(vfmPath, FileMode.Create, FileAccess.Write, FileShare.None))
using (var writer = new BinaryWriter(vfmFileStream)) {
vfmSerializer.Serialize(archiveData.Value, writer);
commandList.Add(commandFactory.CreateFileRemappingCommand(archiveData.Key.DatFilePath, vfmFileStream.Name));
}
logger?.Info("Wrote VFM " + vfmFileName + " to " + tempDir);
// Serialize the RAF
var rafFileName = versionString + "/" + archiveFileName;
var rafPath = temporaryFileService.AllocateTemporaryFile(tempDir, rafFileName);
using (var rafFileStream = File.Open(rafPath, FileMode.Create, FileAccess.Write, FileShare.None))
using (var writer = new BinaryWriter(rafFileStream)) {
writer.Write(archiveData.Key.GetDirectoryFile().GetBytes());
commandList.Add(commandFactory.CreateFileRedirectionCommand(archiveData.Key.RAFFilePath, rafFileStream.Name));
}
logger?.Info("Wrote RAF " + rafFileName + " to " + tempDir);
}
}
// Serialize the Release Manifest
var manifestPath = temporaryFileService.AllocateTemporaryFile(tempDir, "releasemanifest");
using (var manifestFileStream = File.Open(manifestPath, FileMode.Create, FileAccess.Write, FileShare.None))
using (var writer = new BinaryWriter(manifestFileStream)) {
new ReleaseManifestWriter(gameManifest).Save(writer);
commandList.Add(commandFactory.CreateFileRedirectionCommand(gameManifest.Path, manifestFileStream.Name));
}
logger?.Info("Wrote release manifest to " + tempDir);
return(commandList);
}