instance_destroy.go

// Copyright (c) 2016 Pulcy.
//
// 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 main

import (
	"fmt"

	"github.com/spf13/cobra"

	"github.com/pulcy/quark/providers"
)

var (
	cmdDestroyInstance = &cobra.Command{
		Short: "Destroy a single instance",
		Long:  "Destroy a single instance",
		Use:   "destroy",
		Run:   destroyInstance,
	}

	destroyInstanceFlags providers.ClusterInstanceInfo
)

func init() {
	cmdDestroyInstance.Flags().StringVar(&destroyInstanceFlags.Domain, "domain", defaultDomain(), "Cluster domain")
	cmdDestroyInstance.Flags().StringVar(&destroyInstanceFlags.Name, "name", "", "Cluster name")
	cmdDestroyInstance.Flags().StringVar(&destroyInstanceFlags.Prefix, "prefix", "", "Instance prefix name")
	cmdInstance.AddCommand(cmdDestroyInstance)
}

func destroyInstance(cmd *cobra.Command, args []string) {
	requireProfile := false
	loadArgumentsFromCluster(cmd.Flags(), requireProfile)
	clusterInstanceInfoFromArgs(&destroyInstanceFlags, args)

	provider := newProvider()
	destroyInstanceFlags.ClusterInfo = provider.ClusterDefaults(destroyInstanceFlags.ClusterInfo)

	if destroyInstanceFlags.Domain == "" {
		Exitf("Please specify a domain\n")
	}
	if destroyInstanceFlags.Name == "" {
		Exitf("Please specify a name\n")
	}
	if destroyInstanceFlags.Prefix == "" {
		Exitf("Please specify a prefix\n")
	}
	if err := confirm(fmt.Sprintf("Are you sure you want to destroy %s?", destroyInstanceFlags.String())); err != nil {
		Exitf("%v\n", err)
	}

	// Remove instance from etcd
	instances, err := provider.GetInstances(destroyInstanceFlags.ClusterInfo)
	if err != nil {
		Exitf("Failed to list instances: %v\n", err)
	}
	toRemove, err := instances.InstanceByName(destroyInstanceFlags.String())
	if err != nil {
		Exitf("Failed to find instance '%s'\n", destroyInstanceFlags.String())
	}
	machineID, err := toRemove.GetMachineID(log)
	if err != nil {
		Exitf("Failed to query machine id for instance: %#v", err)
	}
	isEtcdProxy, err := toRemove.IsEtcdProxy(log)
	if err != nil {
		Exitf("Failed to query etcd mode for instance: %#v", err)
	}
	if !isEtcdProxy {
		remainingInstances := instances.Except(toRemove)
		if err := remainingInstances.RemoveEtcdMember(log, toRemove.Name, toRemove.ClusterIP); err != nil {
			log.Errorf("Failed to remove instance '%s' from ETCD", destroyInstanceFlags.String())
		}
	}

	if err := provider.DeleteInstance(destroyInstanceFlags, newDnsProvider()); err != nil {
		Exitf("Failed to destroy instance: %v\n", err)
	}

	// Update existing members
	if err := providers.UpdateClusterMembers(log, destroyInstanceFlags.ClusterInfo, false, nil, provider); err != nil {
		Exitf("Failed to update cluster members: %v\n", err)
	}

	// Remove machine from vault
	if err := newVaultProvider().RemoveMachine(machineID); err != nil {
		log.Warningf("Failed to remove machine from vault: %#v", err)
	}

	Infof("Destroyed instance %s\n", destroyInstanceFlags)
}