1package compute
2
3// Copyright (c) Microsoft and contributors.  All rights reserved.
4//
5// Licensed under the Apache License, Version 2.0 (the "License");
6// you may not use this file except in compliance with the License.
7// You may obtain a copy of the License at
8// http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13//
14// See the License for the specific language governing permissions and
15// limitations under the License.
16//
17// Code generated by Microsoft (R) AutoRest Code Generator.
18// Changes may cause incorrect behavior and will be lost if the code is regenerated.
19
20import (
21	"context"
22	"github.com/Azure/go-autorest/autorest"
23	"github.com/Azure/go-autorest/autorest/azure"
24	"github.com/Azure/go-autorest/tracing"
25	"net/http"
26)
27
28// VirtualMachineScaleSetExtensionsClient is the compute Client
29type VirtualMachineScaleSetExtensionsClient struct {
30	BaseClient
31}
32
33// NewVirtualMachineScaleSetExtensionsClient creates an instance of the VirtualMachineScaleSetExtensionsClient client.
34func NewVirtualMachineScaleSetExtensionsClient(subscriptionID string) VirtualMachineScaleSetExtensionsClient {
35	return NewVirtualMachineScaleSetExtensionsClientWithBaseURI(DefaultBaseURI, subscriptionID)
36}
37
38// NewVirtualMachineScaleSetExtensionsClientWithBaseURI creates an instance of the
39// VirtualMachineScaleSetExtensionsClient client using a custom endpoint.  Use this when interacting with an Azure
40// cloud that uses a non-standard base URI (sovereign clouds, Azure stack).
41func NewVirtualMachineScaleSetExtensionsClientWithBaseURI(baseURI string, subscriptionID string) VirtualMachineScaleSetExtensionsClient {
42	return VirtualMachineScaleSetExtensionsClient{NewWithBaseURI(baseURI, subscriptionID)}
43}
44
45// CreateOrUpdate the operation to create or update an extension.
46// Parameters:
47// resourceGroupName - the name of the resource group.
48// VMScaleSetName - the name of the VM scale set where the extension should be create or updated.
49// vmssExtensionName - the name of the VM scale set extension.
50// extensionParameters - parameters supplied to the Create VM scale set Extension operation.
51func (client VirtualMachineScaleSetExtensionsClient) CreateOrUpdate(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string, extensionParameters VirtualMachineScaleSetExtension) (result VirtualMachineScaleSetExtensionsCreateOrUpdateFuture, err error) {
52	if tracing.IsEnabled() {
53		ctx = tracing.StartSpan(ctx, fqdn+"/VirtualMachineScaleSetExtensionsClient.CreateOrUpdate")
54		defer func() {
55			sc := -1
56			if result.Response() != nil {
57				sc = result.Response().StatusCode
58			}
59			tracing.EndSpan(ctx, sc, err)
60		}()
61	}
62	req, err := client.CreateOrUpdatePreparer(ctx, resourceGroupName, VMScaleSetName, vmssExtensionName, extensionParameters)
63	if err != nil {
64		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "CreateOrUpdate", nil, "Failure preparing request")
65		return
66	}
67
68	result, err = client.CreateOrUpdateSender(req)
69	if err != nil {
70		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "CreateOrUpdate", result.Response(), "Failure sending request")
71		return
72	}
73
74	return
75}
76
77// CreateOrUpdatePreparer prepares the CreateOrUpdate request.
78func (client VirtualMachineScaleSetExtensionsClient) CreateOrUpdatePreparer(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string, extensionParameters VirtualMachineScaleSetExtension) (*http.Request, error) {
79	pathParameters := map[string]interface{}{
80		"resourceGroupName": autorest.Encode("path", resourceGroupName),
81		"subscriptionId":    autorest.Encode("path", client.SubscriptionID),
82		"vmScaleSetName":    autorest.Encode("path", VMScaleSetName),
83		"vmssExtensionName": autorest.Encode("path", vmssExtensionName),
84	}
85
86	const APIVersion = "2019-07-01"
87	queryParameters := map[string]interface{}{
88		"api-version": APIVersion,
89	}
90
91	extensionParameters.Type = nil
92	preparer := autorest.CreatePreparer(
93		autorest.AsContentType("application/json; charset=utf-8"),
94		autorest.AsPut(),
95		autorest.WithBaseURL(client.BaseURI),
96		autorest.WithPathParameters("/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}", pathParameters),
97		autorest.WithJSON(extensionParameters),
98		autorest.WithQueryParameters(queryParameters))
99	return preparer.Prepare((&http.Request{}).WithContext(ctx))
100}
101
102// CreateOrUpdateSender sends the CreateOrUpdate request. The method will close the
103// http.Response Body if it receives an error.
104func (client VirtualMachineScaleSetExtensionsClient) CreateOrUpdateSender(req *http.Request) (future VirtualMachineScaleSetExtensionsCreateOrUpdateFuture, err error) {
105	var resp *http.Response
106	resp, err = client.Send(req, azure.DoRetryWithRegistration(client.Client))
107	if err != nil {
108		return
109	}
110	future.Future, err = azure.NewFutureFromResponse(resp)
111	return
112}
113
114// CreateOrUpdateResponder handles the response to the CreateOrUpdate request. The method always
115// closes the http.Response Body.
116func (client VirtualMachineScaleSetExtensionsClient) CreateOrUpdateResponder(resp *http.Response) (result VirtualMachineScaleSetExtension, err error) {
117	err = autorest.Respond(
118		resp,
119		client.ByInspecting(),
120		azure.WithErrorUnlessStatusCode(http.StatusOK, http.StatusCreated),
121		autorest.ByUnmarshallingJSON(&result),
122		autorest.ByClosing())
123	result.Response = autorest.Response{Response: resp}
124	return
125}
126
127// Delete the operation to delete the extension.
128// Parameters:
129// resourceGroupName - the name of the resource group.
130// VMScaleSetName - the name of the VM scale set where the extension should be deleted.
131// vmssExtensionName - the name of the VM scale set extension.
132func (client VirtualMachineScaleSetExtensionsClient) Delete(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string) (result VirtualMachineScaleSetExtensionsDeleteFuture, err error) {
133	if tracing.IsEnabled() {
134		ctx = tracing.StartSpan(ctx, fqdn+"/VirtualMachineScaleSetExtensionsClient.Delete")
135		defer func() {
136			sc := -1
137			if result.Response() != nil {
138				sc = result.Response().StatusCode
139			}
140			tracing.EndSpan(ctx, sc, err)
141		}()
142	}
143	req, err := client.DeletePreparer(ctx, resourceGroupName, VMScaleSetName, vmssExtensionName)
144	if err != nil {
145		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Delete", nil, "Failure preparing request")
146		return
147	}
148
149	result, err = client.DeleteSender(req)
150	if err != nil {
151		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Delete", result.Response(), "Failure sending request")
152		return
153	}
154
155	return
156}
157
158// DeletePreparer prepares the Delete request.
159func (client VirtualMachineScaleSetExtensionsClient) DeletePreparer(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string) (*http.Request, error) {
160	pathParameters := map[string]interface{}{
161		"resourceGroupName": autorest.Encode("path", resourceGroupName),
162		"subscriptionId":    autorest.Encode("path", client.SubscriptionID),
163		"vmScaleSetName":    autorest.Encode("path", VMScaleSetName),
164		"vmssExtensionName": autorest.Encode("path", vmssExtensionName),
165	}
166
167	const APIVersion = "2019-07-01"
168	queryParameters := map[string]interface{}{
169		"api-version": APIVersion,
170	}
171
172	preparer := autorest.CreatePreparer(
173		autorest.AsDelete(),
174		autorest.WithBaseURL(client.BaseURI),
175		autorest.WithPathParameters("/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}", pathParameters),
176		autorest.WithQueryParameters(queryParameters))
177	return preparer.Prepare((&http.Request{}).WithContext(ctx))
178}
179
180// DeleteSender sends the Delete request. The method will close the
181// http.Response Body if it receives an error.
182func (client VirtualMachineScaleSetExtensionsClient) DeleteSender(req *http.Request) (future VirtualMachineScaleSetExtensionsDeleteFuture, err error) {
183	var resp *http.Response
184	resp, err = client.Send(req, azure.DoRetryWithRegistration(client.Client))
185	if err != nil {
186		return
187	}
188	future.Future, err = azure.NewFutureFromResponse(resp)
189	return
190}
191
192// DeleteResponder handles the response to the Delete request. The method always
193// closes the http.Response Body.
194func (client VirtualMachineScaleSetExtensionsClient) DeleteResponder(resp *http.Response) (result autorest.Response, err error) {
195	err = autorest.Respond(
196		resp,
197		client.ByInspecting(),
198		azure.WithErrorUnlessStatusCode(http.StatusOK, http.StatusAccepted, http.StatusNoContent),
199		autorest.ByClosing())
200	result.Response = resp
201	return
202}
203
204// Get the operation to get the extension.
205// Parameters:
206// resourceGroupName - the name of the resource group.
207// VMScaleSetName - the name of the VM scale set containing the extension.
208// vmssExtensionName - the name of the VM scale set extension.
209// expand - the expand expression to apply on the operation.
210func (client VirtualMachineScaleSetExtensionsClient) Get(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string, expand string) (result VirtualMachineScaleSetExtension, err error) {
211	if tracing.IsEnabled() {
212		ctx = tracing.StartSpan(ctx, fqdn+"/VirtualMachineScaleSetExtensionsClient.Get")
213		defer func() {
214			sc := -1
215			if result.Response.Response != nil {
216				sc = result.Response.Response.StatusCode
217			}
218			tracing.EndSpan(ctx, sc, err)
219		}()
220	}
221	req, err := client.GetPreparer(ctx, resourceGroupName, VMScaleSetName, vmssExtensionName, expand)
222	if err != nil {
223		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Get", nil, "Failure preparing request")
224		return
225	}
226
227	resp, err := client.GetSender(req)
228	if err != nil {
229		result.Response = autorest.Response{Response: resp}
230		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Get", resp, "Failure sending request")
231		return
232	}
233
234	result, err = client.GetResponder(resp)
235	if err != nil {
236		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Get", resp, "Failure responding to request")
237	}
238
239	return
240}
241
242// GetPreparer prepares the Get request.
243func (client VirtualMachineScaleSetExtensionsClient) GetPreparer(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string, expand string) (*http.Request, error) {
244	pathParameters := map[string]interface{}{
245		"resourceGroupName": autorest.Encode("path", resourceGroupName),
246		"subscriptionId":    autorest.Encode("path", client.SubscriptionID),
247		"vmScaleSetName":    autorest.Encode("path", VMScaleSetName),
248		"vmssExtensionName": autorest.Encode("path", vmssExtensionName),
249	}
250
251	const APIVersion = "2019-07-01"
252	queryParameters := map[string]interface{}{
253		"api-version": APIVersion,
254	}
255	if len(expand) > 0 {
256		queryParameters["$expand"] = autorest.Encode("query", expand)
257	}
258
259	preparer := autorest.CreatePreparer(
260		autorest.AsGet(),
261		autorest.WithBaseURL(client.BaseURI),
262		autorest.WithPathParameters("/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}", pathParameters),
263		autorest.WithQueryParameters(queryParameters))
264	return preparer.Prepare((&http.Request{}).WithContext(ctx))
265}
266
267// GetSender sends the Get request. The method will close the
268// http.Response Body if it receives an error.
269func (client VirtualMachineScaleSetExtensionsClient) GetSender(req *http.Request) (*http.Response, error) {
270	return client.Send(req, azure.DoRetryWithRegistration(client.Client))
271}
272
273// GetResponder handles the response to the Get request. The method always
274// closes the http.Response Body.
275func (client VirtualMachineScaleSetExtensionsClient) GetResponder(resp *http.Response) (result VirtualMachineScaleSetExtension, err error) {
276	err = autorest.Respond(
277		resp,
278		client.ByInspecting(),
279		azure.WithErrorUnlessStatusCode(http.StatusOK),
280		autorest.ByUnmarshallingJSON(&result),
281		autorest.ByClosing())
282	result.Response = autorest.Response{Response: resp}
283	return
284}
285
286// List gets a list of all extensions in a VM scale set.
287// Parameters:
288// resourceGroupName - the name of the resource group.
289// VMScaleSetName - the name of the VM scale set containing the extension.
290func (client VirtualMachineScaleSetExtensionsClient) List(ctx context.Context, resourceGroupName string, VMScaleSetName string) (result VirtualMachineScaleSetExtensionListResultPage, err error) {
291	if tracing.IsEnabled() {
292		ctx = tracing.StartSpan(ctx, fqdn+"/VirtualMachineScaleSetExtensionsClient.List")
293		defer func() {
294			sc := -1
295			if result.vmsselr.Response.Response != nil {
296				sc = result.vmsselr.Response.Response.StatusCode
297			}
298			tracing.EndSpan(ctx, sc, err)
299		}()
300	}
301	result.fn = client.listNextResults
302	req, err := client.ListPreparer(ctx, resourceGroupName, VMScaleSetName)
303	if err != nil {
304		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "List", nil, "Failure preparing request")
305		return
306	}
307
308	resp, err := client.ListSender(req)
309	if err != nil {
310		result.vmsselr.Response = autorest.Response{Response: resp}
311		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "List", resp, "Failure sending request")
312		return
313	}
314
315	result.vmsselr, err = client.ListResponder(resp)
316	if err != nil {
317		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "List", resp, "Failure responding to request")
318	}
319
320	return
321}
322
323// ListPreparer prepares the List request.
324func (client VirtualMachineScaleSetExtensionsClient) ListPreparer(ctx context.Context, resourceGroupName string, VMScaleSetName string) (*http.Request, error) {
325	pathParameters := map[string]interface{}{
326		"resourceGroupName": autorest.Encode("path", resourceGroupName),
327		"subscriptionId":    autorest.Encode("path", client.SubscriptionID),
328		"vmScaleSetName":    autorest.Encode("path", VMScaleSetName),
329	}
330
331	const APIVersion = "2019-07-01"
332	queryParameters := map[string]interface{}{
333		"api-version": APIVersion,
334	}
335
336	preparer := autorest.CreatePreparer(
337		autorest.AsGet(),
338		autorest.WithBaseURL(client.BaseURI),
339		autorest.WithPathParameters("/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions", pathParameters),
340		autorest.WithQueryParameters(queryParameters))
341	return preparer.Prepare((&http.Request{}).WithContext(ctx))
342}
343
344// ListSender sends the List request. The method will close the
345// http.Response Body if it receives an error.
346func (client VirtualMachineScaleSetExtensionsClient) ListSender(req *http.Request) (*http.Response, error) {
347	return client.Send(req, azure.DoRetryWithRegistration(client.Client))
348}
349
350// ListResponder handles the response to the List request. The method always
351// closes the http.Response Body.
352func (client VirtualMachineScaleSetExtensionsClient) ListResponder(resp *http.Response) (result VirtualMachineScaleSetExtensionListResult, err error) {
353	err = autorest.Respond(
354		resp,
355		client.ByInspecting(),
356		azure.WithErrorUnlessStatusCode(http.StatusOK),
357		autorest.ByUnmarshallingJSON(&result),
358		autorest.ByClosing())
359	result.Response = autorest.Response{Response: resp}
360	return
361}
362
363// listNextResults retrieves the next set of results, if any.
364func (client VirtualMachineScaleSetExtensionsClient) listNextResults(ctx context.Context, lastResults VirtualMachineScaleSetExtensionListResult) (result VirtualMachineScaleSetExtensionListResult, err error) {
365	req, err := lastResults.virtualMachineScaleSetExtensionListResultPreparer(ctx)
366	if err != nil {
367		return result, autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "listNextResults", nil, "Failure preparing next results request")
368	}
369	if req == nil {
370		return
371	}
372	resp, err := client.ListSender(req)
373	if err != nil {
374		result.Response = autorest.Response{Response: resp}
375		return result, autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "listNextResults", resp, "Failure sending next results request")
376	}
377	result, err = client.ListResponder(resp)
378	if err != nil {
379		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "listNextResults", resp, "Failure responding to next results request")
380	}
381	return
382}
383
384// ListComplete enumerates all values, automatically crossing page boundaries as required.
385func (client VirtualMachineScaleSetExtensionsClient) ListComplete(ctx context.Context, resourceGroupName string, VMScaleSetName string) (result VirtualMachineScaleSetExtensionListResultIterator, err error) {
386	if tracing.IsEnabled() {
387		ctx = tracing.StartSpan(ctx, fqdn+"/VirtualMachineScaleSetExtensionsClient.List")
388		defer func() {
389			sc := -1
390			if result.Response().Response.Response != nil {
391				sc = result.page.Response().Response.Response.StatusCode
392			}
393			tracing.EndSpan(ctx, sc, err)
394		}()
395	}
396	result.page, err = client.List(ctx, resourceGroupName, VMScaleSetName)
397	return
398}
399
400// Update the operation to update an extension.
401// Parameters:
402// resourceGroupName - the name of the resource group.
403// VMScaleSetName - the name of the VM scale set where the extension should be updated.
404// vmssExtensionName - the name of the VM scale set extension.
405// extensionParameters - parameters supplied to the Update VM scale set Extension operation.
406func (client VirtualMachineScaleSetExtensionsClient) Update(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string, extensionParameters VirtualMachineScaleSetExtensionUpdate) (result VirtualMachineScaleSetExtensionsUpdateFuture, err error) {
407	if tracing.IsEnabled() {
408		ctx = tracing.StartSpan(ctx, fqdn+"/VirtualMachineScaleSetExtensionsClient.Update")
409		defer func() {
410			sc := -1
411			if result.Response() != nil {
412				sc = result.Response().StatusCode
413			}
414			tracing.EndSpan(ctx, sc, err)
415		}()
416	}
417	req, err := client.UpdatePreparer(ctx, resourceGroupName, VMScaleSetName, vmssExtensionName, extensionParameters)
418	if err != nil {
419		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Update", nil, "Failure preparing request")
420		return
421	}
422
423	result, err = client.UpdateSender(req)
424	if err != nil {
425		err = autorest.NewErrorWithError(err, "compute.VirtualMachineScaleSetExtensionsClient", "Update", result.Response(), "Failure sending request")
426		return
427	}
428
429	return
430}
431
432// UpdatePreparer prepares the Update request.
433func (client VirtualMachineScaleSetExtensionsClient) UpdatePreparer(ctx context.Context, resourceGroupName string, VMScaleSetName string, vmssExtensionName string, extensionParameters VirtualMachineScaleSetExtensionUpdate) (*http.Request, error) {
434	pathParameters := map[string]interface{}{
435		"resourceGroupName": autorest.Encode("path", resourceGroupName),
436		"subscriptionId":    autorest.Encode("path", client.SubscriptionID),
437		"vmScaleSetName":    autorest.Encode("path", VMScaleSetName),
438		"vmssExtensionName": autorest.Encode("path", vmssExtensionName),
439	}
440
441	const APIVersion = "2019-07-01"
442	queryParameters := map[string]interface{}{
443		"api-version": APIVersion,
444	}
445
446	extensionParameters.Name = nil
447	extensionParameters.Type = nil
448	preparer := autorest.CreatePreparer(
449		autorest.AsContentType("application/json; charset=utf-8"),
450		autorest.AsPatch(),
451		autorest.WithBaseURL(client.BaseURI),
452		autorest.WithPathParameters("/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}", pathParameters),
453		autorest.WithJSON(extensionParameters),
454		autorest.WithQueryParameters(queryParameters))
455	return preparer.Prepare((&http.Request{}).WithContext(ctx))
456}
457
458// UpdateSender sends the Update request. The method will close the
459// http.Response Body if it receives an error.
460func (client VirtualMachineScaleSetExtensionsClient) UpdateSender(req *http.Request) (future VirtualMachineScaleSetExtensionsUpdateFuture, err error) {
461	var resp *http.Response
462	resp, err = client.Send(req, azure.DoRetryWithRegistration(client.Client))
463	if err != nil {
464		return
465	}
466	future.Future, err = azure.NewFutureFromResponse(resp)
467	return
468}
469
470// UpdateResponder handles the response to the Update request. The method always
471// closes the http.Response Body.
472func (client VirtualMachineScaleSetExtensionsClient) UpdateResponder(resp *http.Response) (result VirtualMachineScaleSetExtension, err error) {
473	err = autorest.Respond(
474		resp,
475		client.ByInspecting(),
476		azure.WithErrorUnlessStatusCode(http.StatusOK, http.StatusCreated),
477		autorest.ByUnmarshallingJSON(&result),
478		autorest.ByClosing())
479	result.Response = autorest.Response{Response: resp}
480	return
481}
482