90 lines
3.6 KiB
C#
90 lines
3.6 KiB
C#
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using EpicGames.Core;
|
|
using UnrealBuildBase;
|
|
using UnrealBuildTool;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace AutomationTool
|
|
{
|
|
[Help("Copy all the binaries for a target into a different folder. Can be restored using the UnstashTarget command. Useful for A/B testing.")]
|
|
[Help("-Name", "Name of the target")]
|
|
[Help("-Platform", "Platform that the target was built for")]
|
|
[Help("-Configuration", "Architecture that the target was built for")]
|
|
[Help("-Architecture", "Architecture that the target was built for")]
|
|
[Help("-Project", "Project file for the target")]
|
|
[Help("-To", "Output directory to store the stashed binaries")]
|
|
public class StashTarget : BuildCommand
|
|
{
|
|
public override void ExecuteBuild()
|
|
{
|
|
// Parse all the arguments
|
|
string TargetName = ParseRequiredStringParam("Name");
|
|
string PlatformName = ParseOptionalStringParam("Platform");
|
|
UnrealTargetPlatform Platform;
|
|
if (UnrealTargetPlatform.TryParse(PlatformName, out Platform))
|
|
{
|
|
Platform = HostPlatform.Current.HostEditorPlatform;
|
|
}
|
|
UnrealTargetConfiguration Configuration = ParseOptionalEnumParam<UnrealTargetConfiguration>("Configuration") ?? UnrealTargetConfiguration.Development;
|
|
string ArchitectureString = ParseOptionalStringParam("Architecture");
|
|
UnrealArchitectures Architecture = UnrealArchitectures.FromString(ArchitectureString, Platform);
|
|
FileReference ProjectFile = ParseOptionalFileReferenceParam("Project");
|
|
DirectoryReference ToDir = ParseRequiredDirectoryReferenceParam("To");
|
|
|
|
// Read the receipt
|
|
FileReference ReceiptFile = TargetReceipt.GetDefaultPath(DirectoryReference.FromFile(ProjectFile) ?? Unreal.EngineDirectory, TargetName, Platform, Configuration, Architecture);
|
|
if(!FileReference.Exists(ReceiptFile))
|
|
{
|
|
throw new AutomationException("Unable to find '{0}'", ReceiptFile);
|
|
}
|
|
|
|
TargetReceipt Receipt = TargetReceipt.Read(ReceiptFile);
|
|
|
|
// Enumerate all the files we want to move
|
|
List<FileReference> FilesToMove = new List<FileReference>();
|
|
FilesToMove.Add(ReceiptFile);
|
|
FilesToMove.AddRange(Receipt.BuildProducts.Select(x => x.Path));
|
|
|
|
// Move all the files to the output folder
|
|
DirectoryReference.CreateDirectory(ToDir);
|
|
CommandUtils.DeleteDirectoryContents(ToDir.FullName);
|
|
foreach(FileReference SourceFile in FilesToMove)
|
|
{
|
|
FileReference TargetFile = FileReference.Combine(ToDir, SourceFile.MakeRelativeTo(Unreal.RootDirectory));
|
|
Logger.LogInformation("Copying {SourceFile} to {TargetFile}", SourceFile, TargetFile);
|
|
CommandUtils.CopyFile(SourceFile.FullName, TargetFile.FullName);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Help("Copy all the binaries from a target back into the root directory. Use in combination with the StashTarget command.")]
|
|
[Help("-From", "Directory to copy from")]
|
|
public class UnstashTarget : BuildCommand
|
|
{
|
|
public override void ExecuteBuild()
|
|
{
|
|
// Parse the arguments
|
|
DirectoryReference FromDir = ParseRequiredDirectoryReferenceParam("From");
|
|
if(!DirectoryReference.Exists(FromDir))
|
|
{
|
|
throw new AutomationException("Source directory '{0}' does not exist", FromDir);
|
|
}
|
|
|
|
// Just copy all the files into place
|
|
foreach(FileReference SourceFile in DirectoryReference.EnumerateFiles(FromDir, "*", SearchOption.AllDirectories))
|
|
{
|
|
FileReference TargetFile = FileReference.Combine(Unreal.RootDirectory, SourceFile.MakeRelativeTo(FromDir));
|
|
Logger.LogInformation("Copying {SourceFile} to {TargetFile}", SourceFile, TargetFile);
|
|
CopyFile(SourceFile.FullName, TargetFile.FullName);
|
|
}
|
|
}
|
|
}
|
|
}
|