UnrealEngineUWP/Engine/Source/Programs/AutomationTool/AutomationUtils/P4WriteConfig.cs

// Copyright Epic Games, Inc. All Rights Reserved.

using System;
using System.Text;
using System.IO;
using EpicGames.Core;
using System.Collections.Generic;
using System.Linq;
using UnrealBuildBase;

namespace AutomationTool
{
	[Help("Auto-detects P4 settings based on the current path and creates a p4config file with the relevant settings.")]
	[Help("setignore", "Adds a P4IGNORE to the default file (Engine/Extras/Perforce/p4ignore)")]
	[Help("path=<path>", "Write to a path other than the current directory")]
	[Help("p4port=<server:port>", "Optional hint/override of the server to use during lookup")]
	[Help("p4user=<server:port>", "Optional hint/override of the username to use during lookup")]
	public class P4WriteConfig : BuildCommand
	{
		public override ExitCode Execute()
		{
			Log.TraceInformation("Setting up Perforce environment.");

			// User can specify these to help auto detection
			string Port = ParseParamValue("p4port", "");
			string User = ParseParamValue("p4user", "");

			bool SetIgnore = ParseParam("setignore");
			bool ListOnly = ParseParam("listonly");

			string OutputPath = ParseParamValue("path", "");

			// apply any hints
			if (!string.IsNullOrEmpty(Port))
			{
				Environment.SetEnvironmentVariable(EnvVarNames.P4Port, Port);
			}

			if (!string.IsNullOrEmpty(User))
			{
				Environment.SetEnvironmentVariable(EnvVarNames.User, User);
			}

			// try to init P4
			try
			{
				CommandUtils.InitP4Environment();
				CommandUtils.InitDefaultP4Connection();
			}
			catch (Exception Ex)
			{
				Log.TraceError(
					"Unable to find matching Perforce info. If the below does not help try P4WriteConfig -p4port=<server:port> and -p4user=<username> to supply more info");
				Log.TraceError("{0}", Ex.Message);
				return ExitCode.Error_Arguments;
			}

			// store all our settings
			StringBuilder P4Config = new StringBuilder();

			P4Config.AppendLine($"P4PORT={P4Env.ServerAndPort}");
			P4Config.AppendLine($"P4USER={P4Env.User}");
			P4Config.AppendLine($"P4CLIENT={P4Env.Client}");

			if (SetIgnore)
			{
				string IgnorePath = Path.Combine(Unreal.EngineDirectory.ToString(), "Extras", "Perforce", "p4ignore");
				P4Config.AppendLine($"P4IGNORE={IgnorePath}");
			}

			string P4Settings = P4Config.ToString();

			if (!string.IsNullOrEmpty(OutputPath))
			{
				if (!DirectoryExists(OutputPath) && !FileExists(OutputPath))
				{
					throw new AutomationException("Path {0} does not exist.", OutputPath);
				}
			}
			else
			{
				OutputPath = Environment.CurrentDirectory;
			}

			if (!File.Exists(OutputPath))
			{
				OutputPath = Path.Combine(OutputPath, "p4config.txt");
			}

			Console.WriteLine("***\nWriting\n{0}to - {1}\n***", P4Settings, OutputPath);

			if (!ListOnly)
			{
				File.WriteAllText(OutputPath, P4Settings);

				string OutputFile = Path.GetFileName(OutputPath);

				Log.TraceInformation("Wrote P4 settings to {0}", OutputPath);

				P4.P4(string.Format("set P4CONFIG={0}", OutputFile));
				Log.TraceInformation("set P4CONFIG={0}", OutputFile);
			}
			else
			{
				Log.TraceInformation("Skipped write");
			}

			return ExitCode.Success;
		}
	}
}