Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Renamed AlreadyUsed to Used

Co-authored-by: Kristoffer Dalby <kradalby@kradalby.no>
  • Loading branch information
juanfont and kradalby authored Oct 13, 2021
1 parent 30788e1 commit 93517aa
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 7 deletions.
2 changes: 1 addition & 1 deletion api.go
Original file line number Diff line number Diff line change
Expand Up @@ -395,7 +395,7 @@ func (h *Headscale) handleAuthKey(c *gin.Context, db *gorm.DB, idKey wgkey.Key,
m.RegisterMethod = "authKey"
db.Save(&m)

pak.AlreadyUsed = true
pak.Used = true
db.Save(&pak)

resp.MachineAuthorized = true
Expand Down
4 changes: 2 additions & 2 deletions cmd/headscale/cli/preauthkeys.go
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ var listPreAuthKeys = &cobra.Command{
return
}

d := pterm.TableData{{"ID", "Key", "Reusable", "Ephemeral", "AlreadyUsed", "Expiration", "Created"}}
d := pterm.TableData{{"ID", "Key", "Reusable", "Ephemeral", "Used", "Expiration", "Created"}}
for _, k := range *keys {
expiration := "-"
if k.Expiration != nil {
Expand All @@ -76,7 +76,7 @@ var listPreAuthKeys = &cobra.Command{
k.Key,
reusable,
strconv.FormatBool(k.Ephemeral),
fmt.Sprintf("%v", k.AlreadyUsed),
fmt.Sprintf("%v", k.Used),
expiration,
k.CreatedAt.Format("2006-01-02 15:04:05"),
})
Expand Down
4 changes: 2 additions & 2 deletions preauth_keys.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ type PreAuthKey struct {
Namespace Namespace
Reusable bool
Ephemeral bool `gorm:"default:false"`
AlreadyUsed bool `gorm:"default:false"`
Used bool `gorm:"default:false"`

CreatedAt *time.Time
Expiration *time.Time
Expand Down Expand Up @@ -111,7 +111,7 @@ func (h *Headscale) checkKeyValidity(k string) (*PreAuthKey, error) {
return nil, err
}

if len(machines) != 0 || pak.AlreadyUsed {
if len(machines) != 0 || pak.Used {
return nil, errorAuthKeyNotReusableAlreadyUsed
}

Expand Down
4 changes: 2 additions & 2 deletions preauth_keys_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -181,13 +181,13 @@ func (*Suite) TestExpirePreauthKey(c *check.C) {
c.Assert(p, check.IsNil)
}

func (*Suite) TestNotReusableMarkedAsAlreadyUsed(c *check.C) {
func (*Suite) TestNotReusableMarkedAsUsed(c *check.C) {
n, err := h.CreateNamespace("test6")
c.Assert(err, check.IsNil)

pak, err := h.CreatePreAuthKey(n.Name, false, false, nil)
c.Assert(err, check.IsNil)
pak.AlreadyUsed = true
pak.Used = true
h.db.Save(&pak)

_, err = h.checkKeyValidity(pak.Key)
Expand Down

0 comments on commit 93517aa

Please sign in to comment.