97 lines
2.9 KiB
C#
97 lines
2.9 KiB
C#
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
using AutomationTool;
|
|
using EpicGames.Core;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.IO.Compression;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Text.RegularExpressions;
|
|
using UnrealBuildBase;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Rocket.Automation
|
|
{
|
|
public class CreateComponentZips : BuildCommand
|
|
{
|
|
class Category
|
|
{
|
|
public string Name;
|
|
public FileFilter Filter = new FileFilter();
|
|
}
|
|
|
|
public override void ExecuteBuild()
|
|
{
|
|
DirectoryReference SourceDir = ParseRequiredDirectoryReferenceParam("SourceDir");
|
|
DirectoryReference OutputDir = ParseRequiredDirectoryReferenceParam("OutputDir");
|
|
DirectoryReference.CreateDirectory(OutputDir);
|
|
|
|
FileReference Location = FileReference.Combine(Unreal.EngineDirectory, "Restricted", "NotForLicensees", "Build", "LauncherAttributes.ini");
|
|
|
|
string[] Lines = FileReference.ReadAllLines(Location);
|
|
|
|
Category CurrentCategory = null;
|
|
List<Category> Categories = new List<Category>();
|
|
foreach (string Line in Lines)
|
|
{
|
|
string TrimLine = Line.Trim();
|
|
if (TrimLine.Length > 0)
|
|
{
|
|
if (TrimLine[0] == '[')
|
|
{
|
|
Match Match = Regex.Match(TrimLine, @"^\[LauncherAttribute\.tag:([A-Za-z0-9_]+)\]$");
|
|
if (Match.Success)
|
|
{
|
|
CurrentCategory = new Category { Name = Match.Groups[1].Value };
|
|
Categories.Add(CurrentCategory);
|
|
}
|
|
else
|
|
{
|
|
CurrentCategory = null;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (CurrentCategory != null)
|
|
{
|
|
CurrentCategory.Filter.AddRule(TrimLine);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
List<FileInfo> SourceFiles = SourceDir.ToDirectoryInfo().EnumerateFiles("*", SearchOption.AllDirectories).ToList();
|
|
Logger.LogInformation("Found {Arg0} source files", SourceFiles.Count);
|
|
|
|
FileReference MainOutputFile = FileReference.Combine(OutputDir, $"UnrealEngine.zip");
|
|
CreateZip(SourceDir, SourceFiles, x => !Categories.Any(y => y.Filter.Matches(x)), MainOutputFile);
|
|
|
|
foreach (Category Category in Categories)
|
|
{
|
|
FileReference OutputFile = FileReference.Combine(OutputDir, $"UnrealEngine_{Category.Name}.zip");
|
|
CreateZip(SourceDir, SourceFiles, x => Category.Filter.Matches(x), OutputFile);
|
|
}
|
|
}
|
|
|
|
static void CreateZip(DirectoryReference SourceDir, List<FileInfo> SourceFiles, Predicate<string> Predicate, FileReference OutputFile)
|
|
{
|
|
long TotalSize = 0;
|
|
List<FileReference> SourceFileRefs = new List<FileReference>();
|
|
foreach (FileInfo SourceFile in SourceFiles)
|
|
{
|
|
FileReference SourceFileRef = new FileReference(SourceFile);
|
|
if (Predicate(SourceFileRef.MakeRelativeTo(SourceDir)))
|
|
{
|
|
SourceFileRefs.Add(SourceFileRef);
|
|
TotalSize += SourceFile.Length;
|
|
}
|
|
}
|
|
|
|
Logger.LogInformation("Creating {OutputFile} ({1:n0} files, {2:n1}mb)", OutputFile, SourceFileRefs.Count, TotalSize / (1024.0 * 1024.0));
|
|
CommandUtils.ZipFiles(OutputFile, SourceDir, SourceFileRefs);
|
|
}
|
|
}
|
|
}
|