github.com/mattyw/juju@v0.0.0-20140610034352-732aecd63861/cmd/juju/authorizedkeys_delete.go (about)

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