feat(identity): add identity credentials helpers

identity/identity.go

@@ -216,6 +216,27 @@ func (i *Identity) DeleteCredentialsType(t CredentialsType) {
 	delete(i.Credentials, t)
 }
 
+func (i *Identity) GetCredentialsOr(t CredentialsType, or *Credentials) *Credentials {
+	c, ok := i.GetCredentials(t)
+	if !ok {
+		return or
+	}
+	return c
+}
+
+func (i *Identity) UpsertCredentialsConfig(t CredentialsType, conf []byte) {
+	c, ok := i.GetCredentials(t)
+	if !ok {
+		c = &Credentials{}
+	}
+
+	c.Type = t
+	c.IdentityID = i.ID
+	c.Config = conf
+
+	i.SetCredentials(t, *c)
+}
+
 func (i *Identity) GetCredentials(t CredentialsType) (*Credentials, bool) {
 	i.lock().RLock()
 	defer i.lock().RUnlock()

identity/identity_test.go

@@ -29,6 +29,30 @@ func TestNewIdentity(t *testing.T) {
 	assert.True(t, i.IsActive())
 }
 
+func TestIdentityCredentialsOr(t *testing.T) {
+	i := NewIdentity(config.DefaultIdentityTraitsSchemaID)
+	i.Credentials = nil
+
+	expected := &Credentials{ID: x.NewUUID(), Type: CredentialsTypePassword}
+	assert.Equal(t, expected, i.GetCredentialsOr(CredentialsTypePassword, expected))
+
+	expected = &Credentials{ID: x.NewUUID(), Type: CredentialsTypeWebAuthn}
+	i.SetCredentials(CredentialsTypeWebAuthn, *expected)
+
+	assert.Equal(t, expected, i.GetCredentialsOr(CredentialsTypeWebAuthn, nil))
+}
+
+func TestIdentityCredentialsOrCreate(t *testing.T) {
+	i := NewIdentity(config.DefaultIdentityTraitsSchemaID)
+	i.Credentials = nil
+
+	expected := &Credentials{Config: []byte("true"), IdentityID: i.ID, Type: CredentialsTypePassword}
+	i.UpsertCredentialsConfig(CredentialsTypePassword, []byte("true"))
+	actual, ok := i.GetCredentials(CredentialsTypePassword)
+	assert.True(t, ok)
+	assert.Equal(t, expected, actual)
+}
+
 func TestIdentityCredentials(t *testing.T) {
 	i := NewIdentity(config.DefaultIdentityTraitsSchemaID)
 	i.Credentials = nil