Skip to content
Snippets Groups Projects
Commit f41692a6 authored by Jonah Husson's avatar Jonah Husson
Browse files

Add username field to users table for raw username

parent 8c5d3d34
No related branches found
No related tags found
3 merge requests!50Revert "update deps",!48Release,!42Add username field to users table for raw username
...@@ -33,15 +33,21 @@ func registerUser(msg *pb.UDBUserRegistration, permPublicKey *rsa.PublicKey, ...@@ -33,15 +33,21 @@ func registerUser(msg *pb.UDBUserRegistration, permPublicKey *rsa.PublicKey,
} }
// Parse the username and UserID // Parse the username and UserID
username := msg.IdentityRegistration.Username username := msg.IdentityRegistration.Username // TODO: this & msg.Frs.Fact seems redundant
uid, err := id.Unmarshal(msg.UID) uid, err := id.Unmarshal(msg.UID)
if err != nil { if err != nil {
return &messages.Ack{}, errors.New("Could not parse UID sent over. " + return &messages.Ack{}, errors.New("Could not parse UID sent over. " +
"Please try again") "Please try again")
} }
flatten := func(s string) string {
return s
}
// TODO: flatten username
flattened := flatten(username)
// Check if username is taken // Check if username is taken
err = store.CheckUser(username, uid) err = store.CheckUser(flattened, uid) // TODO: this should take flattened
if err != nil { if err != nil {
return &messages.Ack{}, errors.Errorf("Username %s is already taken. "+ return &messages.Ack{}, errors.Errorf("Username %s is already taken. "+
"Please try again", username) "Please try again", username)
...@@ -84,7 +90,7 @@ func registerUser(msg *pb.UDBUserRegistration, permPublicKey *rsa.PublicKey, ...@@ -84,7 +90,7 @@ func registerUser(msg *pb.UDBUserRegistration, permPublicKey *rsa.PublicKey,
f := storage.Fact{ f := storage.Fact{
Hash: hashedFact, Hash: hashedFact,
UserId: msg.UID, UserId: msg.UID,
Fact: msg.Frs.Fact.Fact, Fact: flattened,
Type: uint8(msg.Frs.Fact.FactType), Type: uint8(msg.Frs.Fact.FactType),
Signature: msg.Frs.FactSig, Signature: msg.Frs.FactSig,
Verified: true, Verified: true,
...@@ -94,6 +100,7 @@ func registerUser(msg *pb.UDBUserRegistration, permPublicKey *rsa.PublicKey, ...@@ -94,6 +100,7 @@ func registerUser(msg *pb.UDBUserRegistration, permPublicKey *rsa.PublicKey,
// Create the user to insert into the database // Create the user to insert into the database
u := &storage.User{ u := &storage.User{
Id: msg.UID, Id: msg.UID,
Username: username,
RsaPub: msg.RSAPublicPem, RsaPub: msg.RSAPublicPem,
DhPub: msg.IdentityRegistration.DhPubKey, DhPub: msg.IdentityRegistration.DhPubKey,
Salt: msg.IdentityRegistration.Salt, Salt: msg.IdentityRegistration.Salt,
......
...@@ -59,6 +59,7 @@ type MapImpl struct { ...@@ -59,6 +59,7 @@ type MapImpl struct {
// Struct defining the users table for the database // Struct defining the users table for the database
type User struct { type User struct {
Id []byte `gorm:"primaryKey"` Id []byte `gorm:"primaryKey"`
Username string `gorm:"not null;unique"`
RsaPub string `gorm:"not null;unique"` RsaPub string `gorm:"not null;unique"`
DhPub []byte `gorm:"not null;unique"` DhPub []byte `gorm:"not null;unique"`
Salt []byte `gorm:"not null"` Salt []byte `gorm:"not null"`
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment