Add --user (#2)

CHANGELOG.md

@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.0.2] - 2022-05-14
+
+### Added
+
+- Add `--user` options to force it to secure/undo a specific user rather than the entire `KUBECONFIG`
+
 ## [0.0.1] - 2022-05-12
 
 ### Added

cmd/secure.go

@@ -37,6 +37,7 @@ func FileExists(filename string) (bool, error) {
 
 func init() {
  secureCmd.Flags().StringP("kubeconfig", "c", "", "Kubeconfig path")
+ secureCmd.Flags().StringP("user", "u", "", "Secure specific user instead of all")
 
  rootCmd.AddCommand(secureCmd)
 }
@@ -53,7 +54,20 @@ var secureCmd = &cobra.Command{
  log.Fatal(err)
  }
 
+ specificUser, err := cmd.Flags().GetString("user")
+ if err != nil {
+ log.Fatal(err)
+ }
+ if specificUser != "" {
+ fmt.Printf("Looking up for a specific user %s\n", specificUser)
+ }
+
  for name, user := range cfg.AuthInfos {
+ if specificUser != "" && specificUser != name {
+ fmt.Printf("Skip user %s: not a %s\n", name, specificUser)
+ continue
+ }
+
  if len(user.ClientCertificateData) == 0 && len(user.ClientKeyData) == 0 && user.Username == "" && user.Password == "" {
  fmt.Printf("Skip user %s: nothing to secure\n", name)
  continue

cmd/undo.go

@@ -15,6 +15,7 @@ import (
 
 func init() {
  undoCmd.Flags().StringP("kubeconfig", "c", "", "Kubeconfig path")
+ undoCmd.Flags().StringP("user", "u", "", "Secure specific user instead of all")
 
  rootCmd.AddCommand(undoCmd)
 }
@@ -31,7 +32,20 @@ var undoCmd = &cobra.Command{
  log.Fatal(err)
  }
 
+ specificUser, err := cmd.Flags().GetString("user")
+ if err != nil {
+ log.Fatal(err)
+ }
+ if specificUser != "" {
+ fmt.Printf("Looking up for a specific user %s\n", specificUser)
+ }
+
  for name, user := range cfg.AuthInfos {
+ if specificUser != "" && specificUser != name {
+ fmt.Printf("Skip user %s: not a %s\n", name, specificUser)
+ continue
+ }
+
  fmt.Printf("Found user: %s\n", name)
 
  if user.Exec == nil || !strings.HasSuffix(user.Exec.Command, executable) {