From 05fe3ef2813013ac506f27331ccb07bfa95c6939 Mon Sep 17 00:00:00 2001 From: Michael Quigley Date: Fri, 9 Sep 2022 14:03:47 -0400 Subject: [PATCH] store.AccountRequest and friends (#50, #51) --- controller/store/account_request.go | 47 +++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 controller/store/account_request.go diff --git a/controller/store/account_request.go b/controller/store/account_request.go new file mode 100644 index 00000000..43b3dc46 --- /dev/null +++ b/controller/store/account_request.go @@ -0,0 +1,47 @@ +package store + +import ( + "database/sql" + "github.com/jmoiron/sqlx" + "github.com/pkg/errors" +) + +type AccountRequest struct { + Model + Token string + Email string + Verified bool + SourceAddress string +} + +func (self *Store) CreateAccountRequest(ar *AccountRequest, tx *sql.Tx) (int, error) { + stmt, err := tx.Prepare("insert into account_requests (token, email, source_address) values (?, ?, ?)") + if err != nil { + return 0, errors.Wrap(err, "error preparing account_requests insert statement") + } + res, err := stmt.Exec(ar.Token, ar.Email, ar.SourceAddress) + if err != nil { + return 0, errors.Wrap(err, "error executing account_requests insert statement") + } + id, err := res.LastInsertId() + if err != nil { + return 0, errors.Wrap(err, "error retrieving last account_requests insert id") + } + return int(id), nil +} + +func (self *Store) GetAccountRequest(id int, tx *sqlx.Tx) (*AccountRequest, error) { + ar := &AccountRequest{} + if err := tx.QueryRowx("select * from account_requests where id = ?", id).StructScan(ar); err != nil { + return nil, errors.Wrap(err, "error selecting account_request by id") + } + return ar, nil +} + +func (self *Store) FindAccountRequestWithToken(token string, tx *sqlx.Tx) (*AccountRequest, error) { + ar := &AccountRequest{} + if err := tx.QueryRowx("select * from account_requests where token = ?", token).StructScan(ar); err != nil { + return nil, errors.Wrap(err, "error selecting account_request by token") + } + return ar, nil +}