dotnet-installer/build_projects/dotnet-cli-build/CreateAzureContainer.cs
2017-03-02 20:35:20 -08:00

167 lines
6.5 KiB
C#

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System;
using System.Globalization;
using System.IO;
using System.Net;
using System.Threading.Tasks;
using System.Net.Http;
using Microsoft.Build.Framework;
using Task = Microsoft.Build.Utilities.Task;
namespace Microsoft.DotNet.Build.CloudTestTasks
{
public sealed class CreateAzureContainer : Task
{
/// <summary>
/// The Azure account key used when creating the connection string.
/// </summary>
[Required]
public string AccountKey { get; set; }
/// <summary>
/// The Azure account name used when creating the connection string.
/// </summary>
[Required]
public string AccountName { get; set; }
/// <summary>
/// The name of the container to create. The specified name must be in the correct format, see the
/// following page for more info. https://msdn.microsoft.com/en-us/library/azure/dd135715.aspx
/// </summary>
[Required]
public string ContainerName { get; set; }
/// <summary>
/// When false, if the specified container already exists get a reference to it.
/// When true, if the specified container already exists the task will fail.
/// </summary>
public bool FailIfExists { get; set; }
/// <summary>
/// The read-only SAS token created when ReadOnlyTokenDaysValid is greater than zero.
/// </summary>
[Output]
public string ReadOnlyToken { get; set; }
/// <summary>
/// The number of days for which the read-only token should be valid.
/// </summary>
public int ReadOnlyTokenDaysValid { get; set; }
/// <summary>
/// The URI of the created container.
/// </summary>
[Output]
public string StorageUri { get; set; }
/// <summary>
/// The write-only SAS token create when WriteOnlyTokenDaysValid is greater than zero.
/// </summary>
[Output]
public string WriteOnlyToken { get; set; }
/// <summary>
/// The number of days for which the write-only token should be valid.
/// </summary>
public int WriteOnlyTokenDaysValid { get; set; }
public override bool Execute()
{
return ExecuteAsync().GetAwaiter().GetResult();
}
public async Task<bool> ExecuteAsync()
{
Log.LogMessage(
MessageImportance.High,
"Creating container named '{0}' in storage account {1}.",
ContainerName,
AccountName);
string url = string.Format(
"https://{0}.blob.core.windows.net/{1}?restype=container",
AccountName,
ContainerName);
StorageUri = string.Format(
"https://{0}.blob.core.windows.net/{1}/",
AccountName,
ContainerName);
Log.LogMessage(MessageImportance.Low, "Sending request to create Container");
using (HttpClient client = new HttpClient())
{
Func<HttpRequestMessage> createRequest = () =>
{
DateTime dt = DateTime.UtcNow;
var req = new HttpRequestMessage(HttpMethod.Put, url);
req.Headers.Add(AzureHelper.DateHeaderString, dt.ToString("R", CultureInfo.InvariantCulture));
req.Headers.Add(AzureHelper.VersionHeaderString, AzureHelper.StorageApiVersion);
req.Headers.Add(AzureHelper.AuthorizationHeaderString, AzureHelper.AuthorizationHeader(
AccountName,
AccountKey,
"PUT",
dt,
req));
byte[] bytestoWrite = new byte[0];
int bytesToWriteLength = 0;
Stream postStream = new MemoryStream();
postStream.Write(bytestoWrite, 0, bytesToWriteLength);
req.Content = new StreamContent(postStream);
return req;
};
Func<HttpResponseMessage, bool> validate = (HttpResponseMessage response) =>
{
// the Conflict status (409) indicates that the container already exists, so
// if FailIfExists is set to false and we get a 409 don't fail the task.
return response.IsSuccessStatusCode || (!FailIfExists && response.StatusCode == HttpStatusCode.Conflict);
};
using (HttpResponseMessage response = await AzureHelper.RequestWithRetry(Log, client, createRequest, validate))
{
try
{
Log.LogMessage(
MessageImportance.Low,
"Received response to create Container {0}: Status Code: {1} {2}",
ContainerName, response.StatusCode, response.Content.ToString());
// specifying zero is valid, it means "I don't want a token"
if (ReadOnlyTokenDaysValid > 0)
{
ReadOnlyToken = AzureHelper.CreateContainerSasToken(
AccountName,
ContainerName,
AccountKey,
AzureHelper.SasAccessType.Read,
ReadOnlyTokenDaysValid);
}
// specifying zero is valid, it means "I don't want a token"
if (WriteOnlyTokenDaysValid > 0)
{
WriteOnlyToken = AzureHelper.CreateContainerSasToken(
AccountName,
ContainerName,
AccountKey,
AzureHelper.SasAccessType.Write,
WriteOnlyTokenDaysValid);
}
}
catch (Exception e)
{
Log.LogErrorFromException(e, true);
return false;
}
}
}
return true;
}
}
}