add additional configuration functions for users and endpoints

3 files changed, 40 insertions, 4 deletions

diff --git a/db/db.go b/db/db.go
index d9faff3..a9a9a2e 100644
--- a/db/db.go
+++ b/db/db.go
@@ -41,6 +41,7 @@ type Model interface {
 	GetSchemaVersion() (int, error)
 
 	CreateUser(username, password string) (User, error)
+	SetPassword(user User, password string) error
 	AuthenticateUser(username, password string) (User, error)
 	GetById(id string) (User, error)
 	AllUsers() ([]User, error)
@@ -53,6 +54,8 @@ type Model interface {
 
 	CreateEndpoint(name, path, address string) (Endpoint, error)
 	DeleteEndpoint(endpoint Endpoint) error
+	SetEndpointPath(endpoint Endpoint, path string) error
+	SetEndpointAddress(endpoint Endpoint, address string) error
 	GetEndpointByName(name string) (Endpoint, error)
 	GetEndpointByPath(path string) (Endpoint, error)
 	GetEndpointByAddress(address string) (Endpoint, error)
diff --git a/db/endpoint.go b/db/endpoint.go
index f22d77b..cb59de5 100644
--- a/db/endpoint.go
+++ b/db/endpoint.go
@@ -23,7 +23,7 @@ func (p *Phlox) CreateEndpoint(name, path, address string) (Endpoint, error) {
 		Path: path,
 		Address: address,
 	}
-	
+
 	return endpoint, nil
 }
 
@@ -46,6 +46,17 @@ func extractEndpoint(s Scanner) (Endpoint, error) {
 }
 
 
+func (p *Phlox) SetEndpointPath(endpoint Endpoint, path string) error {
+	_, err := p.db.Exec("update endpoints set path=? where endpointid=?", path, endpoint.Id)
+	return err
+}
+
+func (p *Phlox) SetEndpointAddress(endpoint Endpoint, address string) error {
+	_, err := p.db.Exec("update endpoints set address=? where endpointid=?", address, endpoint.Id)
+	return err
+}
+
+
 func queryEndpoint(p *Phlox, query, param string) (Endpoint, error) {
 	row := p.db.QueryRow(query, param)
 	endpoint, err := extractEndpoint(row)
diff --git a/db/user.go b/db/user.go
index 37c0744..c1f2efe 100644
--- a/db/user.go
+++ b/db/user.go
@@ -30,6 +30,17 @@ func saltPassword(password string, salt []byte) []byte {
 }
 
 
+func hashPassword(password string, salt []byte) ([]byte, error) {
+	salted := saltPassword(password, salt)
+	hash, err := bcrypt.GenerateFromPassword(salted, bcrypt.DefaultCost)
+	if err != nil {
+		return []byte{}, err
+	}
+
+	return hash, nil
+}
+
+
 func (p *Phlox) CreateUser(username, password string) (User, error) {
 	user := User{}
 
@@ -44,9 +55,7 @@ func (p *Phlox) CreateUser(username, password string) (User, error) {
 		return user, err
 	}
 
-	salted := saltPassword(password, salt)
-
-	hash, err := bcrypt.GenerateFromPassword(salted, bcrypt.DefaultCost)
+	hash, err := hashPassword(password, salt)
 	if err != nil {
 		return user, err
 	}
@@ -69,6 +78,19 @@ func (p *Phlox) CreateUser(username, password string) (User, error) {
 
 
 
+func (p *Phlox) SetPassword(user User, password string) error {
+	hash, err := hashPassword(password, user.Salt)
+	if err != nil {
+		return err
+	}
+	hash64 := base64.StdEncoding.EncodeToString(hash)
+
+	_, err = p.db.Exec("update users set passwordhash=? where userid=?;", hash64, user.Id)
+	return err
+}
+
+
+
 func extractUser(s Scanner) (User, error) {
 	var userid int
 	var username string