public override void ExecuteCmdlet()
{
if (this.InvitationId != null && this.Location != null)
{
try
{
ConsumerInvitation ConsumerInvitation =
this.DataShareManagementClient.ConsumerInvitations.Get(this.Location, this.InvitationId);
this.WriteObject(ConsumerInvitation.ToPsObject());
}
catch (DataShareErrorException ex) when(ex.Response.StatusCode.Equals(HttpStatusCode.NotFound))
{
throw new PSArgumentException(string.Format(Resources.ResourceNotFoundMessage, this.InvitationId));
}
}
else
{
string nextPageLink = null;
List <ConsumerInvitation> invitationList = new List <ConsumerInvitation>();
do
{
IPage <ConsumerInvitation> invitations = string.IsNullOrEmpty(nextPageLink)
? this.DataShareManagementClient.ConsumerInvitations.ListInvitations()
: this.DataShareManagementClient.ConsumerInvitations.ListInvitationsNext(nextPageLink);
invitationList.AddRange(invitations.AsEnumerable());
nextPageLink = invitations.NextPageLink;
} while (nextPageLink != null);
IEnumerable <PSDataShareConsumerInvitation> consumerInvitations = invitationList.Select(invitation => invitation.ToPsObject());
this.WriteObject(consumerInvitations, true);
}
}
public ShareSubscription CreateIfNotExistShareSubscription(Invitation invitation)
{
if (invitation.InvitationStatus == "Accepted")
{
Console.WriteLine(
$"Invitation {invitation.InvitationId} is already accepted. Trying to get Share Subscription...");
try
{
ShareSubscription shareSubscription = this.DataShareClient.ShareSubscriptions.Get(
this.Principal.DataShareResourceGroup,
this.Principal.DataShareAccountName,
this.Principal.DataShareShareSubscriptionName);
Console.WriteLine($"\r\n\r\nFound share subscription {shareSubscription.Id}...");
if (shareSubscription.InvitationId != invitation.InvitationId)
{
Console.WriteLine(
$"{shareSubscription.Id} was not created for InvitationId {invitation.InvitationId}. Make sure the configuration is valid.");
throw new Exception("Invalid configuration.");
}
return(shareSubscription);
}
catch (DataShareErrorException exception)
{
if (exception.Response.StatusCode == HttpStatusCode.NotFound)
{
Console.WriteLine(
$"Share subscription {this.Principal.DataShareShareSubscriptionName} does not exist. Make sure the configuration is valid.");
}
throw;
}
}
ConsumerInvitation consumerInvitation = this.DataShareClient.ConsumerInvitations.Get(
Configuration.Location,
invitation.InvitationId);
var shareSubscriptionPayload = new ShareSubscription
{
InvitationId = invitation.InvitationId,
SourceShareLocation = consumerInvitation.Location
};
return(this.DataShareClient.ShareSubscriptions.Create(
this.Principal.DataShareResourceGroup,
this.Principal.DataShareAccountName,
this.Principal.DataShareShareSubscriptionName,
shareSubscriptionPayload));
}
public static PSDataShareConsumerInvitation ToPsObject(this ConsumerInvitation consumerInvitation)
{
return(new PSDataShareConsumerInvitation
{
Id = consumerInvitation.Id,
Name = consumerInvitation.Name,
Type = consumerInvitation.Type,
InvitationId = consumerInvitation.InvitationId,
InvitationStatus = consumerInvitation.InvitationStatus,
Sender = consumerInvitation.UserEmail ?? consumerInvitation.UserName,
SentAt = consumerInvitation.SentAt,
SenderCompanyName = consumerInvitation.ProviderTenantName,
DataSetCount = consumerInvitation.DataSetCount,
Description = consumerInvitation.Description,
Terms = consumerInvitation.TermsOfUse,
RespondedAt = consumerInvitation.RespondedAt,
Location = consumerInvitation.Location,
ShareName = consumerInvitation.ShareName
});
}
/// <summary>
/// Rejects the invitation identified by invitationId
/// </summary>
/// <remarks>
/// Reject an invitation
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='location'>
/// Location of the invitation
/// </param>
/// <param name='invitation'>
/// An invitation payload
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <ConsumerInvitation> RejectInvitationAsync(this IConsumerInvitationsOperations operations, string location, ConsumerInvitation invitation, CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.RejectInvitationWithHttpMessagesAsync(location, invitation, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
/// <summary>
/// Rejects the invitation identified by invitationId
/// </summary>
/// <remarks>
/// Reject an invitation
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='location'>
/// Location of the invitation
/// </param>
/// <param name='invitation'>
/// An invitation payload
/// </param>
public static ConsumerInvitation RejectInvitation(this IConsumerInvitationsOperations operations, string location, ConsumerInvitation invitation)
{
return(operations.RejectInvitationAsync(location, invitation).GetAwaiter().GetResult());
}
/// <summary>
/// Rejects the invitation identified by invitationId
/// </summary>
/// <remarks>
/// Reject an invitation
/// </remarks>
/// <param name='location'>
/// Location of the invitation
/// </param>
/// <param name='invitation'>
/// An invitation payload
/// </param>
/// <param name='customHeaders'>
/// Headers that will be added to request.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
/// <exception cref="DataShareErrorException">
/// Thrown when the operation returned an invalid status code
/// </exception>
/// <exception cref="SerializationException">
/// Thrown when unable to deserialize the response
/// </exception>
/// <exception cref="ValidationException">
/// Thrown when a required parameter is null
/// </exception>
/// <exception cref="System.ArgumentNullException">
/// Thrown when a required parameter is null
/// </exception>
/// <return>
/// A response object containing the response body and response headers.
/// </return>
public async Task <AzureOperationResponse <ConsumerInvitation> > RejectInvitationWithHttpMessagesAsync(string location, ConsumerInvitation invitation, Dictionary <string, List <string> > customHeaders = null, CancellationToken cancellationToken = default(CancellationToken))
{
if (location == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "location");
}
if (invitation == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "invitation");
}
if (invitation != null)
{
invitation.Validate();
}
if (Client.ApiVersion == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "this.Client.ApiVersion");
}
// Tracing
bool _shouldTrace = ServiceClientTracing.IsEnabled;
string _invocationId = null;
if (_shouldTrace)
{
_invocationId = ServiceClientTracing.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("location", location);
tracingParameters.Add("invitation", invitation);
tracingParameters.Add("cancellationToken", cancellationToken);
ServiceClientTracing.Enter(_invocationId, this, "RejectInvitation", tracingParameters);
}
// Construct URL
var _baseUrl = Client.BaseUri.AbsoluteUri;
var _url = new System.Uri(new System.Uri(_baseUrl + (_baseUrl.EndsWith("/") ? "" : "/")), "providers/Microsoft.DataShare/locations/{location}/rejectInvitation").ToString();
_url = _url.Replace("{location}", System.Uri.EscapeDataString(location));
List <string> _queryParameters = new List <string>();
if (Client.ApiVersion != null)
{
_queryParameters.Add(string.Format("api-version={0}", System.Uri.EscapeDataString(Client.ApiVersion)));
}
if (_queryParameters.Count > 0)
{
_url += (_url.Contains("?") ? "&" : "?") + string.Join("&", _queryParameters);
}
// Create HTTP transport objects
var _httpRequest = new HttpRequestMessage();
HttpResponseMessage _httpResponse = null;
_httpRequest.Method = new HttpMethod("POST");
_httpRequest.RequestUri = new System.Uri(_url);
// Set Headers
if (Client.GenerateClientRequestId != null && Client.GenerateClientRequestId.Value)
{
_httpRequest.Headers.TryAddWithoutValidation("x-ms-client-request-id", System.Guid.NewGuid().ToString());
}
if (Client.AcceptLanguage != null)
{
if (_httpRequest.Headers.Contains("accept-language"))
{
_httpRequest.Headers.Remove("accept-language");
}
_httpRequest.Headers.TryAddWithoutValidation("accept-language", Client.AcceptLanguage);
}
if (customHeaders != null)
{
foreach (var _header in customHeaders)
{
if (_httpRequest.Headers.Contains(_header.Key))
{
_httpRequest.Headers.Remove(_header.Key);
}
_httpRequest.Headers.TryAddWithoutValidation(_header.Key, _header.Value);
}
}
// Serialize Request
string _requestContent = null;
if (invitation != null)
{
_requestContent = Rest.Serialization.SafeJsonConvert.SerializeObject(invitation, Client.SerializationSettings);
_httpRequest.Content = new StringContent(_requestContent, System.Text.Encoding.UTF8);
_httpRequest.Content.Headers.ContentType = System.Net.Http.Headers.MediaTypeHeaderValue.Parse("application/json; charset=utf-8");
}
// Set Credentials
if (Client.Credentials != null)
{
cancellationToken.ThrowIfCancellationRequested();
await Client.Credentials.ProcessHttpRequestAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
}
// Send Request
if (_shouldTrace)
{
ServiceClientTracing.SendRequest(_invocationId, _httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
_httpResponse = await Client.HttpClient.SendAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
if (_shouldTrace)
{
ServiceClientTracing.ReceiveResponse(_invocationId, _httpResponse);
}
HttpStatusCode _statusCode = _httpResponse.StatusCode;
cancellationToken.ThrowIfCancellationRequested();
string _responseContent = null;
if ((int)_statusCode != 200)
{
var ex = new DataShareErrorException(string.Format("Operation returned an invalid status code '{0}'", _statusCode));
try
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
DataShareError _errorBody = Rest.Serialization.SafeJsonConvert.DeserializeObject <DataShareError>(_responseContent, Client.DeserializationSettings);
if (_errorBody != null)
{
ex.Body = _errorBody;
}
}
catch (JsonException)
{
// Ignore the exception
}
ex.Request = new HttpRequestMessageWrapper(_httpRequest, _requestContent);
ex.Response = new HttpResponseMessageWrapper(_httpResponse, _responseContent);
if (_shouldTrace)
{
ServiceClientTracing.Error(_invocationId, ex);
}
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw ex;
}
// Create Result
var _result = new AzureOperationResponse <ConsumerInvitation>();
_result.Request = _httpRequest;
_result.Response = _httpResponse;
if (_httpResponse.Headers.Contains("x-ms-request-id"))
{
_result.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
// Deserialize Response
if ((int)_statusCode == 200)
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
try
{
_result.Body = Rest.Serialization.SafeJsonConvert.DeserializeObject <ConsumerInvitation>(_responseContent, Client.DeserializationSettings);
}
catch (JsonException ex)
{
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw new SerializationException("Unable to deserialize the response.", _responseContent, ex);
}
}
if (_shouldTrace)
{
ServiceClientTracing.Exit(_invocationId, _result);
}
return(_result);
}
