|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.IO;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Azure;
using Azure.Core;
namespace Microsoft.DotNet.Helix.Client
{
public partial interface IStorage
{
Task<IImmutableList<Models.ContainerInformation>> ListAsync(
bool? getSasTokens = default,
CancellationToken cancellationToken = default
);
Task<Models.ContainerInformation> NewAsync(
Models.ContainerCreationRequest body,
CancellationToken cancellationToken = default
);
Task<Models.ContainerInformation> ExtendExpirationAsync(
Models.ContainerExtensionRequest body,
CancellationToken cancellationToken = default
);
}
internal partial class Storage : IServiceOperations<HelixApi>, IStorage
{
public Storage(HelixApi client)
{
Client = client ?? throw new ArgumentNullException(nameof(client));
}
public HelixApi Client { get; }
partial void HandleFailedRequest(RestApiException ex);
partial void HandleFailedListRequest(RestApiException ex);
public async Task<IImmutableList<Models.ContainerInformation>> ListAsync(
bool? getSasTokens = default,
CancellationToken cancellationToken = default
)
{
const string apiVersion = "2019-06-17";
var _baseUri = Client.Options.BaseUri;
var _url = new RequestUriBuilder();
_url.Reset(_baseUri);
_url.AppendPath(
"/api/storage",
false);
if (getSasTokens != default(bool?))
{
_url.AppendQuery("getSasTokens", Client.Serialize(getSasTokens));
}
_url.AppendQuery("api-version", Client.Serialize(apiVersion));
using (var _req = Client.Pipeline.CreateRequest())
{
_req.Uri = _url;
_req.Method = RequestMethod.Get;
using (var _res = await Client.SendAsync(_req, cancellationToken).ConfigureAwait(false))
{
if (_res.Status < 200 || _res.Status >= 300)
{
await OnListFailed(_req, _res).ConfigureAwait(false);
}
if (_res.ContentStream == null)
{
await OnListFailed(_req, _res).ConfigureAwait(false);
}
using (var _reader = new StreamReader(_res.ContentStream))
{
var _content = await _reader.ReadToEndAsync().ConfigureAwait(false);
var _body = Client.Deserialize<IImmutableList<Models.ContainerInformation>>(_content);
return _body;
}
}
}
}
internal async Task OnListFailed(Request req, Response res)
{
string content = null;
if (res.ContentStream != null)
{
using (var reader = new StreamReader(res.ContentStream))
{
content = await reader.ReadToEndAsync().ConfigureAwait(false);
}
}
var ex = new RestApiException<Models.ApiError>(
req,
res,
content,
Client.Deserialize<Models.ApiError>(content)
);
HandleFailedListRequest(ex);
HandleFailedRequest(ex);
Client.OnFailedRequest(ex);
throw ex;
}
partial void HandleFailedNewRequest(RestApiException ex);
public async Task<Models.ContainerInformation> NewAsync(
Models.ContainerCreationRequest body,
CancellationToken cancellationToken = default
)
{
if (body == default(Models.ContainerCreationRequest))
{
throw new ArgumentNullException(nameof(body));
}
if (!body.IsValid)
{
throw new ArgumentException("The parameter is not valid", nameof(body));
}
const string apiVersion = "2019-06-17";
var _baseUri = Client.Options.BaseUri;
var _url = new RequestUriBuilder();
_url.Reset(_baseUri);
_url.AppendPath(
"/api/storage",
false);
_url.AppendQuery("api-version", Client.Serialize(apiVersion));
using (var _req = Client.Pipeline.CreateRequest())
{
_req.Uri = _url;
_req.Method = RequestMethod.Post;
if (body != default(Models.ContainerCreationRequest))
{
_req.Content = RequestContent.Create(Encoding.UTF8.GetBytes(Client.Serialize(body)));
_req.Headers.Add("Content-Type", "application/json; charset=utf-8");
}
using (var _res = await Client.SendAsync(_req, cancellationToken).ConfigureAwait(false))
{
if (_res.Status < 200 || _res.Status >= 300)
{
await OnNewFailed(_req, _res).ConfigureAwait(false);
}
if (_res.ContentStream == null)
{
await OnNewFailed(_req, _res).ConfigureAwait(false);
}
using (var _reader = new StreamReader(_res.ContentStream))
{
var _content = await _reader.ReadToEndAsync().ConfigureAwait(false);
var _body = Client.Deserialize<Models.ContainerInformation>(_content);
return _body;
}
}
}
}
internal async Task OnNewFailed(Request req, Response res)
{
string content = null;
if (res.ContentStream != null)
{
using (var reader = new StreamReader(res.ContentStream))
{
content = await reader.ReadToEndAsync().ConfigureAwait(false);
}
}
var ex = new RestApiException<Models.ApiError>(
req,
res,
content,
Client.Deserialize<Models.ApiError>(content)
);
HandleFailedNewRequest(ex);
HandleFailedRequest(ex);
Client.OnFailedRequest(ex);
throw ex;
}
partial void HandleFailedExtendExpirationRequest(RestApiException ex);
public async Task<Models.ContainerInformation> ExtendExpirationAsync(
Models.ContainerExtensionRequest body,
CancellationToken cancellationToken = default
)
{
if (body == default(Models.ContainerExtensionRequest))
{
throw new ArgumentNullException(nameof(body));
}
if (!body.IsValid)
{
throw new ArgumentException("The parameter is not valid", nameof(body));
}
const string apiVersion = "2019-06-17";
var _baseUri = Client.Options.BaseUri;
var _url = new RequestUriBuilder();
_url.Reset(_baseUri);
_url.AppendPath(
"/api/storage/renew",
false);
_url.AppendQuery("api-version", Client.Serialize(apiVersion));
using (var _req = Client.Pipeline.CreateRequest())
{
_req.Uri = _url;
_req.Method = RequestMethod.Post;
if (body != default(Models.ContainerExtensionRequest))
{
_req.Content = RequestContent.Create(Encoding.UTF8.GetBytes(Client.Serialize(body)));
_req.Headers.Add("Content-Type", "application/json; charset=utf-8");
}
using (var _res = await Client.SendAsync(_req, cancellationToken).ConfigureAwait(false))
{
if (_res.Status < 200 || _res.Status >= 300)
{
await OnExtendExpirationFailed(_req, _res).ConfigureAwait(false);
}
if (_res.ContentStream == null)
{
await OnExtendExpirationFailed(_req, _res).ConfigureAwait(false);
}
using (var _reader = new StreamReader(_res.ContentStream))
{
var _content = await _reader.ReadToEndAsync().ConfigureAwait(false);
var _body = Client.Deserialize<Models.ContainerInformation>(_content);
return _body;
}
}
}
}
internal async Task OnExtendExpirationFailed(Request req, Response res)
{
string content = null;
if (res.ContentStream != null)
{
using (var reader = new StreamReader(res.ContentStream))
{
content = await reader.ReadToEndAsync().ConfigureAwait(false);
}
}
var ex = new RestApiException<Models.ApiError>(
req,
res,
content,
Client.Deserialize<Models.ApiError>(content)
);
HandleFailedExtendExpirationRequest(ex);
HandleFailedRequest(ex);
Client.OnFailedRequest(ex);
throw ex;
}
}
}
|