Photon Controller platform vendor code

vendor/github.com/vmware/photon-controller-go-sdk/photon/images.go

@@ -0,0 +1,125 @@
+// Copyright (c) 2016 VMware, Inc. All Rights Reserved.
+//
+// This product is licensed to you under the Apache License, Version 2.0 (the "License").
+// You may not use this product except in compliance with the License.
+//
+// This product may include a number of subcomponents with separate copyright notices and
+// license terms. Your use of these subcomponents is subject to the terms and conditions
+// of the subcomponent's license, as noted in the LICENSE file.
+
+package photon
+
+import (
+	"encoding/json"
+	"io"
+)
+
+// Contains functionality for images API.
+type ImagesAPI struct {
+	client *Client
+}
+
+// Options for GetImage API.
+type ImageGetOptions struct {
+	Name string `urlParam:"name"`
+}
+
+var imageUrl string = "/images"
+
+// Uploads a new image, reading from the specified image path.
+// If options is nil, default options are used.
+func (api *ImagesAPI) CreateFromFile(imagePath string, options *ImageCreateOptions) (task *Task, err error) {
+	params := imageCreateOptionsToMap(options)
+	res, err := api.client.restClient.MultipartUploadFile(api.client.Endpoint+imageUrl, imagePath, params, api.client.options.TokenOptions.AccessToken)
+	if err != nil {
+		return
+	}
+	defer res.Body.Close()
+	result, err := getTask(getError(res))
+	return result, err
+}
+
+// Uploads a new image, reading from the specified io.Reader.
+// Name is a descriptive name of the image, it is used in the filename field of the Content-Disposition header,
+// and does not need to be unique.
+// If options is nil, default options are used.
+func (api *ImagesAPI) Create(reader io.Reader, name string, options *ImageCreateOptions) (task *Task, err error) {
+	params := imageCreateOptionsToMap(options)
+	res, err := api.client.restClient.MultipartUpload(api.client.Endpoint+imageUrl, reader, name, params, api.client.options.TokenOptions.AccessToken)
+	if err != nil {
+		return
+	}
+	defer res.Body.Close()
+	result, err := getTask(getError(res))
+	return result, err
+}
+
+// Gets all images on this photon instance.
+func (api *ImagesAPI) GetAll(options *ImageGetOptions) (images *Images, err error) {
+	uri := api.client.Endpoint + imageUrl
+	if options != nil {
+		uri += getQueryString(options)
+	}
+	res, err := api.client.restClient.GetList(api.client.Endpoint, uri, api.client.options.TokenOptions.AccessToken)
+	if err != nil {
+		return
+	}
+
+	images = &Images{}
+	err = json.Unmarshal(res, images)
+	return
+}
+
+// Gets details of image with the specified ID.
+func (api *ImagesAPI) Get(imageID string) (image *Image, err error) {
+	res, err := api.client.restClient.Get(api.client.Endpoint+imageUrl+"/"+imageID, api.client.options.TokenOptions.AccessToken)
+	if err != nil {
+		return
+	}
+	defer res.Body.Close()
+	res, err = getError(res)
+	if err != nil {
+		return
+	}
+	var result Image
+	err = json.NewDecoder(res.Body).Decode(&result)
+	return &result, nil
+}
+
+// Deletes image with the specified ID.
+func (api *ImagesAPI) Delete(imageID string) (task *Task, err error) {
+	res, err := api.client.restClient.Delete(api.client.Endpoint+imageUrl+"/"+imageID, api.client.options.TokenOptions.AccessToken)
+	if err != nil {
+		return
+	}
+	defer res.Body.Close()
+	result, err := getTask(getError(res))
+	return result, err
+}
+
+// Gets all tasks with the specified image ID, using options to filter the results.
+// If options is nil, no filtering will occur.
+func (api *ImagesAPI) GetTasks(id string, options *TaskGetOptions) (result *TaskList, err error) {
+	uri := api.client.Endpoint + imageUrl + "/" + id + "/tasks"
+	if options != nil {
+		uri += getQueryString(options)
+	}
+
+	res, err := api.client.restClient.GetList(api.client.Endpoint, uri, api.client.options.TokenOptions.AccessToken)
+	if err != nil {
+		return
+	}
+
+	result = &TaskList{}
+	err = json.Unmarshal(res, result)
+	return
+}
+
+func imageCreateOptionsToMap(opts *ImageCreateOptions) map[string]string {
+	if opts == nil {
+		return nil
+	}
+	return map[string]string{
+		"ImageReplication": opts.ReplicationType,
+	}
+}