-
Notifications
You must be signed in to change notification settings - Fork 2
/
GetUrl.cs
178 lines (153 loc) · 7.33 KB
/
GetUrl.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
using System.Collections.Concurrent;
using System.Diagnostics;
using System.Management.Automation;
using System.Net;
using System.Text.RegularExpressions;
using Microsoft.Azure.Functions.Worker;
using Microsoft.Azure.Functions.Worker.Http;
using Microsoft.Extensions.Logging;
namespace bcaup;
public partial class GetUrl(ILoggerFactory loggerFactory)
{
private readonly ILogger _logger = loggerFactory.CreateLogger<GetUrl>();
private static readonly ConcurrentDictionary<string, CacheEntry> URL_CACHE = new();
private const string VERSION = "1.4.2";
private static readonly string? BCCH_VERSION = Environment.GetEnvironmentVariable("BCCH_VERSION");
[GeneratedRegex("\\u001b\\[[0-9]{1,3}m")]
private static partial Regex FixedRegex();
[Function("bca-url")]
public async Task<HttpResponseData> Run(
[HttpTrigger(AuthorizationLevel.Anonymous, "get", Route = "bca-url/{type?}/{country?}/{version?}")] HttpRequestData req,
string type,
string version,
string country,
string select,
string after,
string before,
string storageAccount,
string accept_insiderEula,
string doNotCheckPlatform,
string doNotRedirect,
string cacheExpiration)
{
_logger.LogInformation("C# HTTP trigger function processed a request.");
var response = req.CreateResponse(HttpStatusCode.OK);
response.Headers.Add("Content-Type", "text/plain; charset=utf-8");
response.Headers.Add("X-bccontainerhelper-version", BCCH_VERSION ?? "Unknown");
response.Headers.Add("X-bcaup-version", VERSION);
var bcchCommand = string.Empty;
try
{
if (string.IsNullOrEmpty(type))
type = "Sandbox";
var typeParam = ValidateTextParam(nameof(type), type);
var countryParam = ValidateTextParam(nameof(country), country);
var versionParam = ValidateTextParam(nameof(version), version);
var selectParam = ValidateTextParam(nameof(select), select);
var afterParam = ValidateTextParam(nameof(after), after);
var beforeParam = ValidateTextParam(nameof(before), before);
var storageAccountParam = ValidateTextParam(nameof(storageAccount), storageAccount);
var accept_insiderEulaParam = GetAcceptInsiderEulaParam(accept_insiderEula);
var doNotCheckPlatformParam = string.Empty;
if (IsValidParamSet(doNotCheckPlatform))
doNotCheckPlatformParam = " -doNotCheckPlatform";
DateTimeOffset expiredAfter = GetExpiredAfter(cacheExpiration);
bcchCommand = $"Get-BCArtifactUrl{typeParam}{countryParam}{versionParam}{selectParam}{afterParam}{beforeParam}{storageAccountParam}{accept_insiderEulaParam}{doNotCheckPlatformParam}";
response.Headers.Add("X-bccontainerhelper-command", bcchCommand);
var url = string.Empty;
if (URL_CACHE.TryGetValue(bcchCommand.ToLower(), out CacheEntry cachedUrl) && cachedUrl.createdOn > expiredAfter)
{
response.Headers.Add("X-bcaup-from-cache", "true");
response.Headers.Add("X-bcaup-cache-timestamp", cachedUrl.createdOn.ToUnixTimeMilliseconds().ToString());
url = cachedUrl.url;
}
else
{
response.Headers.Add("X-bcaup-from-cache", "false");
url = await GetUrlFromBackend(bcchCommand);
var ce = new CacheEntry
{
url = url,
createdOn = DateTime.Now
};
URL_CACHE.AddOrUpdate(bcchCommand.ToLower(), ce, (key, oldValue) => ce);
}
if (doNotRedirect == "true")
await response.WriteStringAsync(url);
else
{
response.StatusCode = HttpStatusCode.Redirect;
response.Headers.Add("Location", url);
}
}
catch (Exception ex)
{
response.StatusCode = HttpStatusCode.BadRequest;
await response.WriteStringAsync(ex.Message);
if (!string.IsNullOrEmpty(bcchCommand))
await response.WriteStringAsync($"Command was {bcchCommand}");
}
return response;
}
private async Task<string> GetUrlFromBackend(string bcchCommand)
{
var psCommmand = $". ./HelperFunctions.ps1; . ./Get-BCArtifactUrl.ps1; {bcchCommand}";
_logger.LogDebug(psCommmand);
var psCommandBytes = System.Text.Encoding.Unicode.GetBytes(psCommmand);
var psCommandBase64 = Convert.ToBase64String(psCommandBytes);
var startInfo = new ProcessStartInfo()
{
FileName = "pwsh",
Arguments = $"-NoProfile -ExecutionPolicy unrestricted -EncodedCommand {psCommandBase64}",
UseShellExecute = false,
RedirectStandardOutput = true,
RedirectStandardError = true
};
using var process = Process.Start(startInfo) ?? throw new Exception("Failed to start the pwsh process!");
await process.WaitForExitAsync();
var err = await process.StandardError.ReadToEndAsync();
if (!string.IsNullOrWhiteSpace(err))
{
if (err.StartsWith("#< CLIXML\n"))
{
err = err.Replace("#< CLIXML\n", "");
var deserialized = PSSerializer.DeserializeAsList(err);
err = string.Join("", deserialized);
err = FixedRegex().Replace(err, "");
throw new Exception(err);
}
}
return await process.StandardOutput.ReadToEndAsync();
}
private static string ValidateTextParam(string paramName, string paramValue)
{
return !string.IsNullOrEmpty(paramValue) ? $" -{paramName} \"{paramValue}\"" : string.Empty;
}
private static bool IsValidParamSet(string? param)
{
// Accept parameter without setting explicit to true and with format "parameter=true"
return param is not null && (param.All(char.IsWhiteSpace) || param.Equals("true", StringComparison.CurrentCultureIgnoreCase));
}
private static string GetAcceptInsiderEulaParam(string accept_insiderEula)
{
var accept_insiderEulaParam = " -accept_insiderEula";
if (IsValidParamSet(accept_insiderEula))
return accept_insiderEulaParam;
return string.Empty;
}
private static DateTimeOffset GetExpiredAfter(string cacheExpiration)
{
if (string.IsNullOrEmpty(cacheExpiration))
return DateTimeOffset.Now.AddHours(-1);
if (!int.TryParse(cacheExpiration, out int parsedCacheExpiration))
throw new Exception(string.Format("The provided cache expiration value {0} is not a valid integer.", cacheExpiration));
if (parsedCacheExpiration < 900) // minimum allowed is 15 minutes
throw new ArgumentOutOfRangeException(string.Format("The provided cache expiration value {0} is invalid. It must be 900 or higher.", cacheExpiration));
return DateTimeOffset.Now.AddSeconds(-parsedCacheExpiration);
}
private struct CacheEntry
{
public string url;
public DateTimeOffset createdOn;
}
}