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
     });
 }
Beispiel #4
0
 /// <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);
     }
 }
Beispiel #5
0
 /// <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());
 }
Beispiel #6
0
        /// <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);
        }