implement reset nkode

core/sqlite_db.go

@@ -95,9 +95,57 @@ VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)
 	return nil
 }
 
+func (d *SqliteDB) UpdateUserNKode(u User) error {
+	tx, err := d.db.Begin()
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err != nil {
+			err = tx.Rollback()
+			if err != nil {
+				log.Fatal(fmt.Sprintf("Write new user won't roll back %+v", err))
+			}
+		}
+	}()
+	updateUser := `
+UPDATE user 
+SET renew = ?, refresh_token = ?, code = ?, mask = ?, attributes_per_key = ?, number_of_keys = ?, alpha_key = ?, set_key = ?, pass_key = ?, mask_key = ?, salt = ?, max_nkode_len = ?, idx_interface = ?, svg_id_interface = ? 
+WHERE username = ? AND customer_id = ?
+`
+	var renew int
+	if u.Renew {
+		renew = 1
+	} else {
+		renew = 0
+	}
+	_, err = tx.Exec(updateUser, renew, u.RefreshToken, u.EncipheredPasscode.Code, u.EncipheredPasscode.Mask, u.Kp.AttrsPerKey, u.Kp.NumbOfKeys, util.Uint64ArrToByteArr(u.CipherKeys.AlphaKey), util.Uint64ArrToByteArr(u.CipherKeys.SetKey), util.Uint64ArrToByteArr(u.CipherKeys.PassKey), util.Uint64ArrToByteArr(u.CipherKeys.MaskKey), u.CipherKeys.Salt, u.CipherKeys.MaxNKodeLen, util.IntArrToByteArr(u.Interface.IdxInterface), util.IntArrToByteArr(u.Interface.SvgId), string(u.Email), uuid.UUID(u.CustomerId))
+
+	if err != nil {
+		return err
+	}
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 func (d *SqliteDB) GetCustomer(id CustomerId) (*Customer, error) {
+	tx, err := d.db.Begin()
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		if err != nil {
+			err = tx.Rollback()
+			if err != nil {
+				log.Fatal(fmt.Sprintf("Write new user won't roll back %+v", err))
+			}
+		}
+	}()
 	selectCustomer := `SELECT max_nkode_len, min_nkode_len, distinct_sets, distinct_attributes, lock_out, expiration, attribute_values, set_values FROM customer WHERE id = ?`
-	rows, err := d.db.Query(selectCustomer, uuid.UUID(id))
+	rows, err := tx.Query(selectCustomer, uuid.UUID(id))
 
 	if !rows.Next() {
 		return nil, errors.New(fmt.Sprintf("no new row for customer %s with err %s", id, rows.Err()))
@@ -131,18 +179,26 @@ func (d *SqliteDB) GetCustomer(id CustomerId) (*Customer, error) {
 		},
 		Attributes: NewCustomerAttributesFromBytes(attributeValues, setValues),
 	}
-
+	err = tx.Commit()
+	if err != nil {
+		return nil, fmt.Errorf("read customer won't commit %w", err)
+	}
 	return &customer, nil
 }
 
 func (d *SqliteDB) GetUser(username Email, customerId CustomerId) (*User, error) {
+	tx, err := d.db.Begin()
+	if err != nil {
+		return nil, err
+	}
+	defer tx.Commit()
 	userSelect := `
 SELECT id, renew, refresh_token, code, mask, attributes_per_key, number_of_keys, alpha_key, set_key, pass_key, mask_key, salt, max_nkode_len, idx_interface, svg_id_interface FROM user 
 WHERE user.username = ? AND user.customer_id = ?
 `
-	rows, err := d.db.Query(userSelect, string(username), uuid.UUID(customerId).String())
+	rows, err := tx.Query(userSelect, string(username), uuid.UUID(customerId).String())
 	if !rows.Next() {
-		return nil, errors.New(fmt.Sprintf("no new rows for user %s of customer %s", string(username), uuid.UUID(customerId).String()))
+		return nil, nil
 	}
 	var id string
 	var renewVal int
@@ -246,7 +302,11 @@ UPDATE customer SET attribute_values = ?, set_values = ? WHERE id = ?;
 	userQuery := `
 SELECT id, alpha_key, set_key, attributes_per_key, number_of_keys FROM user WHERE customer_id = ?
 `
-	rows, err := d.db.Query(userQuery, uuid.UUID(id).String())
+	tx, err := d.db.Begin()
+	if err != nil {
+		return err
+	}
+	rows, err := tx.Query(userQuery, uuid.UUID(id).String())
 	for rows.Next() {
 		var userId string
 		var alphaBytes []byte
@@ -283,8 +343,20 @@ SELECT id, alpha_key, set_key, attributes_per_key, number_of_keys FROM user WHER
 	renewExec += `
 COMMIT;
 `
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
 
+	tx, err = d.db.Begin()
+	if err != nil {
+		return err
+	}
 	_, err = d.db.Exec(renewExec, renewArgs...)
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
 	return err
 }
 
@@ -317,10 +389,15 @@ func (d *SqliteDB) GetSvgStringInterface(idxs SvgIdInterface) ([]string, error)
 }
 
 func (d *SqliteDB) getSvgsById(ids []int) ([]string, error) {
+	tx, err := d.db.Begin()
+	if err != nil {
+		return nil, err
+	}
+	defer tx.Commit()
 	selectId := "SELECT svg FROM svg_icon where id = ?"
 	svgs := make([]string, len(ids))
 	for idx, id := range ids {
-		rows, err := d.db.Query(selectId, id)
+		rows, err := tx.Query(selectId, id)
 		if err != nil {
 			return nil, err
 		}
@@ -336,7 +413,12 @@ func (d *SqliteDB) getSvgsById(ids []int) ([]string, error) {
 }
 
 func (d *SqliteDB) getRandomIds(count int) ([]int, error) {
-	rows, err := d.db.Query("SELECT COUNT(*) as count FROM svg_icon;")
+	tx, err := d.db.Begin()
+	if err != nil {
+		return nil, err
+	}
+	defer tx.Commit()
+	rows, err := tx.Query("SELECT COUNT(*) as count FROM svg_icon;")
 	if err != nil {
 		return nil, err
 	}