add "ssh-key get --project-id $ID"

Signed-off-by: Marques Johansson <[email protected]>
equinix · Jul 23, 2022 · 4d7f91a · 4d7f91a
1 parent a5e1018
commit 4d7f91a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 4 deletions.
diff --git a/docs/metal_ssh-key_get.md b/docs/metal_ssh-key_get.md
@@ -18,13 +18,17 @@ metal ssh-key get [-i <SSH-key_UUID>] [flags]
   
   # Returns the details of SSH key 5cb96463-88fd-4d68-94ba-2c9505ff265e:
   metal ssh-key get --id 5cb96463-88fd-4d68-94ba-2c9505ff265e
+
+  # Retrieve all project SSH keys
+  metal ssh-key get --project-id [project_UUID]
 ```
 
 ### Options
 
 ```
-  -h, --help        help for get
-  -i, --id string   The UUID of an SSH key.
+  -h, --help                help for get
+  -i, --id string           The UUID of an SSH key.
+  -p, --project-id string   List SSH Keys for the project identified by Project ID
 ```
 
 ### Options inherited from parent commands

diff --git a/internal/ssh/retrieve.go b/internal/ssh/retrieve.go
@@ -21,6 +21,9 @@
 package ssh
 
 import (
+	"fmt"
+
+	"github.com/packethost/packngo"
 	"github.com/pkg/errors"
 	"github.com/spf13/cobra"
 )
@@ -38,12 +41,28 @@ func (c *Client) Retrieve() *cobra.Command {
   metal ssh-key get
   
   # Returns the details of SSH key 5cb96463-88fd-4d68-94ba-2c9505ff265e:
-  metal ssh-key get --id 5cb96463-88fd-4d68-94ba-2c9505ff265e`,
+  metal ssh-key get --id 5cb96463-88fd-4d68-94ba-2c9505ff265e
 
+  # Retrieve all project SSH keys
+  metal ssh-key get --project-id [project_UUID]`,
 		RunE: func(cmd *cobra.Command, args []string) error {
 			cmd.SilenceUsage = true
 			if sshKeyID == "" {
-				sshKeys, _, err := c.Service.List()
+				projectID, _ := cmd.LocalFlags().GetString("project-id")
+				fmt.Println(cmd.LocalFlags())
+				listFn := func() ([]packngo.SSHKey, *packngo.Response, error) {
+					fmt.Println("foooo")
+					return c.Service.List()
+				}
+
+				if projectID != "" && cmd.Flag("project-id").Changed {
+					listFn = func() ([]packngo.SSHKey, *packngo.Response, error) {
+						fmt.Println("baaar")
+						return c.Service.ProjectList(projectID)
+					}
+				}
+				sshKeys, _, err := listFn()
+
 				if err != nil {
 					return errors.Wrap(err, "Could not list SSH Keys")
 				}
@@ -73,5 +92,6 @@ func (c *Client) Retrieve() *cobra.Command {
 	}
 
 	retrieveSSHKeysCmd.Flags().StringVarP(&sshKeyID, "id", "i", "", "The UUID of an SSH key.")
+	retrieveSSHKeysCmd.Flags().StringP("project-id", "p", "", "List SSH Keys for the project identified by Project ID")
 	return retrieveSSHKeysCmd
 }