fix: validate key names dont have spaces (#1449) (#1490)

- validate key names dont have spaces
- migrate existing keys with spaces
- validate key name in CLI before create

api/access_key.go

@@ -20,7 +20,7 @@ type ListAccessKeysRequest struct {
 
 type CreateAccessKeyRequest struct {
 	IdentityID        uid.ID   `json:"identityID" validate:"required"`
-	Name              string   `json:"name" validate:"required"`
+	Name              string   `json:"name" validate:"required,excludes= "`
 	TTL               Duration `json:"ttl" validate:"required" note:"maximum time valid"`
 	ExtensionDeadline Duration `json:"extensionDeadline,omitempty" validate:"required" note:"How long the key is active for before it needs to be renewed. The access key must be used within this amount of time to renew validity"`
 }

internal/access/setup.go

@@ -57,7 +57,7 @@ func Setup(c *gin.Context) (string, *models.AccessKey, error) {
 	}
 
 	key := &models.AccessKey{
-		Name:      fmt.Sprintf("%s access key", name),
+		Name:      fmt.Sprintf("%s-access-key", name),
 		IssuedFor: identity.ID,
 		ExpiresAt: time.Now().Add(math.MaxInt64).UTC(),
 	}

internal/cmd/keys.go

@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"fmt"
+	"strings"
 	"time"
 
 	"github.com/spf13/cobra"
@@ -53,6 +54,10 @@ infra keys create main wall-e 12h --extension-deadline=1h
 			keyName := args[0]
 			machineName := args[1]
 
+			if strings.Contains(keyName, " ") {
+				return fmt.Errorf("key name cannot contain spaces")
+			}
+
 			client, err := defaultAPIClient()
 			if err != nil {
 				return err

internal/server/config.go

@@ -546,7 +546,7 @@ func (s *Server) importAccessKeys() error {
 			}
 		}
 
-		name := fmt.Sprintf("default %s access key", k)
+		name := fmt.Sprintf("default-%s-access-key", k)
 
 		accessKey, err := data.GetAccessKey(s.db, data.ByIssuedFor(machine.ID))
 		if err != nil {

internal/server/data/migrations.go

@@ -289,6 +289,31 @@ func migrate(db *gorm.DB) error {
 			},
 			// this change cannot be rolled back
 		},
+		// #1449: access key name can't have whitespace
+		{
+			ID: "202204061643",
+			Migrate: func(tx *gorm.DB) error {
+				if tx.Migrator().HasTable("access_keys") {
+					keys, err := ListAccessKeys(db)
+					if err != nil {
+						return err
+					}
+
+					for i := range keys {
+						if strings.Contains(keys[i].Name, " ") {
+							keys[i].Name = strings.ReplaceAll(keys[i].Name, " ", "-")
+							err := SaveAccessKey(db, &keys[i])
+							if err != nil {
+								return err
+							}
+						}
+					}
+				}
+
+				return nil
+			},
+			// context lost, cannot roll back
+		},
 		// next one here
 	})
 

internal/server/models/access_key.go

@@ -14,7 +14,7 @@ var (
 // AccessKey is a session token presented to the Infra server as proof of authentication
 type AccessKey struct {
 	Model
-	Name      string `gorm:"uniqueIndex:,where:deleted_at is NULL"`
+	Name      string `gorm:"uniqueIndex:,where:deleted_at is NULL" validate:"excludes= "`
 	IssuedFor uid.ID `validate:"required"`
 
 	ExpiresAt         time.Time     `validate:"required"`

internal/server/server_test.go

@@ -875,7 +875,7 @@ func TestImportAccessKeysUpdate(t *testing.T) {
 	err = s.importAccessKeys()
 	assert.NilError(t, err)
 
-	accessKey, err := data.GetAccessKey(s.db, data.ByName("default admin access key"))
+	accessKey, err := data.GetAccessKey(s.db, data.ByName("default-admin-access-key"))
 	assert.NilError(t, err)
 	assert.Equal(t, accessKey.KeyID, "EKoHADINYX")
 }