2
2
mirror of https://github.com/octoleo/restic.git synced 2024-11-19 03:25:19 +00:00
restic/vendor/github.com/Azure/azure-sdk-for-go/management/virtualmachineimage/client.go

113 lines
3.0 KiB
Go
Raw Normal View History

2017-09-13 12:09:48 +00:00
// +build go1.7
2017-08-05 18:30:20 +00:00
// Package virtualmachineimage provides a client for Virtual Machine Images.
package virtualmachineimage
import (
"encoding/xml"
"fmt"
"net/url"
"github.com/Azure/azure-sdk-for-go/management"
)
const (
azureImageListURL = "services/vmimages"
azureImageDeleteURLformat = "services/vmimages/%s"
azureRoleOperationsURL = "services/hostedservices/%s/deployments/%s/roleinstances/%s/operations"
errParamNotSpecified = "Parameter %s is not specified."
)
//NewClient is used to instantiate a new Client from an Azure client
func NewClient(client management.Client) Client {
return Client{client}
}
//ListVirtualMachineImages lists the available VM images, filtered by the optional parameters.
//See https://msdn.microsoft.com/en-us/library/azure/dn499770.aspx
func (c Client) ListVirtualMachineImages(parameters ListParameters) (ListVirtualMachineImagesResponse, error) {
var imageList ListVirtualMachineImagesResponse
listURL := azureImageListURL
v := url.Values{}
if parameters.Location != "" {
v.Add("location", parameters.Location)
}
if parameters.Publisher != "" {
v.Add("publisher", parameters.Publisher)
}
if parameters.Category != "" {
v.Add("category", parameters.Category)
}
query := v.Encode()
if query != "" {
listURL = listURL + "?" + query
}
response, err := c.SendAzureGetRequest(listURL)
if err != nil {
return imageList, err
}
err = xml.Unmarshal(response, &imageList)
return imageList, err
}
//DeleteVirtualMachineImage deletes the named VM image. If deleteVHDs is specified,
//the referenced OS and data disks are also deleted.
//See https://msdn.microsoft.com/en-us/library/azure/dn499769.aspx
func (c Client) DeleteVirtualMachineImage(name string, deleteVHDs bool) error {
if name == "" {
return fmt.Errorf(errParamNotSpecified, "name")
}
uri := fmt.Sprintf(azureImageDeleteURLformat, name)
if deleteVHDs {
uri = uri + "?comp=media"
}
_, err := c.SendAzureDeleteRequest(uri) // delete is synchronous for this operation
return err
}
type ListParameters struct {
Location string
Publisher string
Category string
}
const CategoryUser = "User"
//Capture captures a VM into a VM image. The VM has to be shut down previously.
//See https://msdn.microsoft.com/en-us/library/azure/dn499768.aspx
func (c Client) Capture(cloudServiceName, deploymentName, roleName string,
name, label string, osState OSState, parameters CaptureParameters) (management.OperationID, error) {
if cloudServiceName == "" {
return "", fmt.Errorf(errParamNotSpecified, "cloudServiceName")
}
if deploymentName == "" {
return "", fmt.Errorf(errParamNotSpecified, "deploymentName")
}
if roleName == "" {
return "", fmt.Errorf(errParamNotSpecified, "roleName")
}
request := CaptureRoleAsVMImageOperation{
VMImageName: name,
VMImageLabel: label,
OSState: osState,
CaptureParameters: parameters,
}
data, err := xml.Marshal(request)
if err != nil {
return "", err
}
return c.SendAzurePostRequest(fmt.Sprintf(azureRoleOperationsURL,
cloudServiceName, deploymentName, roleName), data)
}