76 lines
2.8 KiB
C#
76 lines
2.8 KiB
C#
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
using EpicGames.Horde;
|
|
using EpicGames.Horde.Secrets;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Net.Http;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
#nullable enable
|
|
|
|
namespace AutomationTool
|
|
{
|
|
/// <summary>
|
|
/// Helper methods for Horde functionality
|
|
/// </summary>
|
|
public partial class CommandUtils
|
|
{
|
|
static readonly ConcurrentDictionary<SecretId, IReadOnlyDictionary<string, string>> _cachedSecrets = new ConcurrentDictionary<SecretId, IReadOnlyDictionary<string, string>>();
|
|
|
|
/// <summary>
|
|
/// Reads a secret from Horde
|
|
/// </summary>
|
|
/// <param name="secretId">Identifier of the secret to retrieve</param>
|
|
/// <param name="cancellationToken">Cancellation token for the operation</param>
|
|
/// <returns>Values configured for the given secret</returns>
|
|
public static async Task<IReadOnlyDictionary<string, string>> GetHordeSecretAsync(SecretId secretId, CancellationToken cancellationToken = default)
|
|
{
|
|
IReadOnlyDictionary<string, string>? secret;
|
|
if (!_cachedSecrets.TryGetValue(secretId, out secret))
|
|
{
|
|
IHordeClient hordeClient = ServiceProvider.GetRequiredService<IHordeClient>();
|
|
|
|
using HordeHttpClient hordeHttpClient = hordeClient.CreateHttpClient();
|
|
|
|
GetSecretResponse response;
|
|
try
|
|
{
|
|
response = await hordeHttpClient.GetSecretAsync(secretId);
|
|
}
|
|
catch (HttpRequestException ex) when (ex.StatusCode == System.Net.HttpStatusCode.NotFound)
|
|
{
|
|
throw new AutomationException(ex, $"Secret '{secretId}' was not found on {hordeClient.ServerUrl}");
|
|
}
|
|
catch (HttpRequestException ex) when (ex.StatusCode == System.Net.HttpStatusCode.Forbidden)
|
|
{
|
|
throw new AutomationException(ex, $"User does not have permissions to read '{secretId}' on {hordeClient.ServerUrl}");
|
|
}
|
|
|
|
secret = response.Data;
|
|
_cachedSecrets.TryAdd(secretId, secret);
|
|
}
|
|
return secret;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Reads a secret from Horde
|
|
/// </summary>
|
|
/// <param name="secretId">Identifier of the secret to retrieve</param>
|
|
/// <param name="propertyName">Name of the property within the secret</param>
|
|
/// <param name="cancellationToken">Cancellation token for the operation</param>
|
|
/// <returns>Values configured for the given secret</returns>
|
|
public static async Task<string> GetHordeSecretAsync(SecretId secretId, string propertyName, CancellationToken cancellationToken = default)
|
|
{
|
|
IReadOnlyDictionary<string, string> secrets = await GetHordeSecretAsync(secretId, cancellationToken);
|
|
if (!secrets.TryGetValue(propertyName, out string? propertyValue))
|
|
{
|
|
throw new AutomationException("Secret '{secretId}' does not contain a value for '{propertyName}'");
|
|
}
|
|
return propertyValue;
|
|
}
|
|
}
|
|
}
|