github.com/cloud-green/juju@v0.0.0-20151002100041-a00291338d3d/cmd/juju/commands/authorizedkeys_delete.go (about)

     1  // Copyright 2013 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package commands
     5  
     6  import (
     7  	"errors"
     8  	"fmt"
     9  
    10  	"github.com/juju/cmd"
    11  	"launchpad.net/gnuflag"
    12  
    13  	"github.com/juju/juju/cmd/juju/block"
    14  )
    15  
    16  var deleteKeysDoc = `
    17  Delete existing authorized ssh keys to remove ssh access for the holder of those keys.
    18  The keys to delete are found by specifying either the "comment" portion of the ssh key,
    19  typically something like "user@host", or the key fingerprint found by using ssh-keygen.
    20  `
    21  
    22  // DeleteKeysCommand is used to delete authorised ssh keys for a user.
    23  type DeleteKeysCommand struct {
    24  	AuthorizedKeysBase
    25  	user   string
    26  	keyIds []string
    27  }
    28  
    29  func (c *DeleteKeysCommand) Info() *cmd.Info {
    30  	return &cmd.Info{
    31  		Name:    "delete",
    32  		Args:    "<ssh key id> [...]",
    33  		Doc:     deleteKeysDoc,
    34  		Purpose: "delete authorized ssh keys for a Juju user",
    35  	}
    36  }
    37  
    38  func (c *DeleteKeysCommand) Init(args []string) error {
    39  	switch len(args) {
    40  	case 0:
    41  		return errors.New("no ssh key id specified")
    42  	default:
    43  		c.keyIds = args
    44  	}
    45  	return nil
    46  }
    47  
    48  func (c *DeleteKeysCommand) SetFlags(f *gnuflag.FlagSet) {
    49  	f.StringVar(&c.user, "user", "admin", "the user for which to delete the keys")
    50  }
    51  
    52  func (c *DeleteKeysCommand) Run(context *cmd.Context) error {
    53  	client, err := c.NewKeyManagerClient()
    54  	if err != nil {
    55  		return err
    56  	}
    57  	defer client.Close()
    58  
    59  	results, err := client.DeleteKeys(c.user, c.keyIds...)
    60  	if err != nil {
    61  		return block.ProcessBlockedError(err, block.BlockChange)
    62  	}
    63  	for i, result := range results {
    64  		if result.Error != nil {
    65  			fmt.Fprintf(context.Stderr, "cannot delete key id %q: %v\n", c.keyIds[i], result.Error)
    66  		}
    67  	}
    68  	return nil
    69  }