implement 'cloud template restore' with plan support

Author: dwburke

README.md

@@ -64,9 +64,13 @@ The Cloud features you can use in manage.liquidweb.com on your Cloud Servers you
 ## Plans
 
 A plan is a pre-defined yaml with optional template variables that can be used to
-repeate specific tasks.
+repeate specific tasks.  Fields in the yaml file match the params you would send
+to the command.
 
-Currently only "lw cloud server create" is implemented.
+Current commands supported in a `plan` file:
+
+- cloud server create
+- cloud template restore
 
 Example:
 

cmd/cloudTemplate.go

@@ -0,0 +1,38 @@
+/*
+Copyright © LiquidWeb
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+package cmd
+
+import (
+	"os"
+
+	"github.com/spf13/cobra"
+)
+
+var cloudTemplateCmd = &cobra.Command{
+	Use:   "template",
+	Short: "Cloud template specific operations",
+	Long: `Cloud template specific operations.
+
+For a full list of capabilities, please refer to the "Available Commands" section.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		cmd.Help()
+		os.Exit(1)
+	},
+}
+
+func init() {
+	cloudCmd.AddCommand(cloudTemplateCmd)
+}

cmd/cloudTemplateRestore.go

@@ -0,0 +1,54 @@
+/*
+Copyright © LiquidWeb
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+
+	"github.com/liquidweb/liquidweb-cli/instance"
+)
+
+var cloudTemplateRestoreCmd = &cobra.Command{
+	Use:   "restore",
+	Short: "Restore a Cloud Template on a Cloud Server",
+	Long:  `Restore a Cloud Template on a Cloud Server.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		params := &instance.CloudTemplateRestoreParams{}
+
+		params.UniqId, _ = cmd.Flags().GetString("uniq-id")
+		params.Template, _ = cmd.Flags().GetString("template")
+
+		result, err := lwCliInst.CloudTemplateRestore(params)
+		if err != nil {
+			lwCliInst.Die(err)
+		}
+
+		fmt.Printf("Restoring template! %s\n", result)
+		fmt.Printf("\tcheck progress with 'cloud server status --uniq-id %s'\n", params.UniqId)
+	},
+}
+
+func init() {
+	cloudTemplateCmd.AddCommand(cloudTemplateRestoreCmd)
+
+	cloudTemplateRestoreCmd.Flags().String("uniq-id", "", "uniq-id of Cloud Server")
+	cloudTemplateRestoreCmd.Flags().String("template", "", "name of template to restore")
+
+	cloudTemplateRestoreCmd.MarkFlagRequired("uniq-id")
+	cloudTemplateRestoreCmd.MarkFlagRequired("template")
+}

instance/cloudTemplateRestore.go

@@ -0,0 +1,45 @@
+/*
+Copyright © LiquidWeb
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+package instance
+
+import (
+	"github.com/liquidweb/liquidweb-cli/types/api"
+	"github.com/liquidweb/liquidweb-cli/validate"
+)
+
+type CloudTemplateRestoreParams struct {
+	Template string `yaml:"template"`
+	UniqId   string `yaml:"uniq-id"`
+}
+
+func (ci *Client) CloudTemplateRestore(params *CloudTemplateRestoreParams) (string, error) {
+	validateFields := map[interface{}]interface{}{
+		params.UniqId: "UniqId",
+	}
+	if err := validate.Validate(validateFields); err != nil {
+		return "", err
+	}
+
+	apiArgs := map[string]interface{}{"template": params.Template, "uniq_id": params.UniqId}
+
+	var details apiTypes.CloudTemplateRestoreResponse
+	err := ci.CallLwApiInto("bleed/storm/template/restore", apiArgs, &details)
+	if err != nil {
+		return "", err
+	}
+
+	return details.Reimaged, nil
+}

instance/plan.go

@@ -25,13 +25,18 @@ type Plan struct {
 }
 
 type PlanCloud struct {
-	Server *PlanCloudServer
+	Server   *PlanCloudServer
+	Template *PlanCloudTemplate
 }
 
 type PlanCloudServer struct {
 	Create []CloudServerCreateParams
 }
 
+type PlanCloudTemplate struct {
+	Restore []CloudTemplateRestoreParams
+}
+
 func (ci *Client) ProcessPlan(plan *Plan) error {
 
 	if plan.Cloud != nil {
@@ -51,6 +56,12 @@ func (ci *Client) processPlanCloud(cloud *PlanCloud) error {
 		}
 	}
 
+	if cloud.Template != nil {
+		if err := ci.processPlanCloudTemplate(cloud.Template); err != nil {
+			return err
+		}
+	}
+
 	return nil
 }
 
@@ -80,3 +91,29 @@ func (ci *Client) processPlanCloudServerCreate(params *CloudServerCreateParams)
 		uniqId, uniqId)
 	return nil
 }
+
+func (ci *Client) processPlanCloudTemplate(template *PlanCloudTemplate) error {
+
+	if template.Restore != nil {
+		for _, c := range template.Restore {
+			if err := ci.processPlanCloudTemplateRestore(&c); err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func (ci *Client) processPlanCloudTemplateRestore(params *CloudTemplateRestoreParams) error {
+
+	result, err := ci.CloudTemplateRestore(params)
+	if err != nil {
+		ci.Die(err)
+	}
+
+	fmt.Printf("Restoring template! %s\n", result)
+	fmt.Printf("\tcheck progress with 'cloud server status --uniq-id %s'\n", params.UniqId)
+
+	return nil
+}

types/api/cloud.go

@@ -255,6 +255,10 @@ type CloudImageDeleteResponse struct {
 	Deleted int64 `json:"deleted" mapstructure:"deleted"`
 }
 
+type CloudTemplateRestoreResponse struct {
+	Reimaged string `json:"reimaged" mapstructure:"reimaged"`
+}
+
 type CloudServerIsBlockStorageOptimized struct {
 	IsOptimized bool `json:"is_optimized" mapstructure:"is_optimized"`
 }