roughed in store implementation for organizations and memberships (#537)

This commit is contained in:
Michael Quigley 2024-12-09 11:46:13 -05:00
parent d6c8f206b2
commit 842f3e8c20
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
3 changed files with 90 additions and 1 deletions

View File

@ -135,7 +135,6 @@ func (str *Store) UpdateFrontend(fe *Frontend, tx *sqlx.Tx) error {
}
func (str *Store) DeleteFrontend(id int, tx *sqlx.Tx) error {
stmt, err := tx.Prepare("update frontends set updated_at = current_timestamp, deleted = true where id = $1")
if err != nil {
return errors.Wrap(err, "error preparing frontends delete statement")

View File

@ -0,0 +1,36 @@
package store
import (
"github.com/jmoiron/sqlx"
"github.com/pkg/errors"
)
type Organization struct {
Model
Token string
Description string
}
func (str *Store) CreateOrganization(org *Organization, trx *sqlx.Tx) (int, error) {
stmt, err := trx.Prepare("insert into organizations (token, description) values ($1, $2) returning id")
if err != nil {
return 0, errors.Wrap(err, "error preparing organizations insert statement")
}
var id int
if err := stmt.QueryRow(org.Token, org.Description).Scan(&id); err != nil {
return 0, errors.Wrap(err, "error executing organizations insert statement")
}
return id, nil
}
func (str *Store) DeleteOrganization(id int, trx *sqlx.Tx) error {
stmt, err := trx.Prepare("update organizations set updated_at = current_timestamp, deleted = true where id = $1")
if err != nil {
return errors.Wrap(err, "error preparing organizations delete statement")
}
_, err = stmt.Exec(id)
if err != nil {
return errors.Wrap(err, "error executing organizations delete statement")
}
return nil
}

View File

@ -0,0 +1,54 @@
package store
import (
"github.com/jmoiron/sqlx"
"github.com/pkg/errors"
)
func (str *Store) AddAccountToOrganization(acctId, orgId int, trx *sqlx.Tx) error {
stmt, err := trx.Prepare("insert into organization_members (organization_id, account_id) values ($1, $2)")
if err != nil {
return errors.Wrap(err, "error preparing organization_members insert statement")
}
_, err = stmt.Exec(acctId, orgId)
if err != nil {
return errors.Wrap(err, "error executing organization_members insert statement")
}
return nil
}
func (str *Store) IsAccountInOrganization(acctId, orgId int, trx *sqlx.Tx) (bool, error) {
stmt, err := trx.Prepare("select count(0) from organization_members where organization_id = $1 and account_id = $2")
if err != nil {
return false, errors.Wrap(err, "error preparing organization_members count statement")
}
var count int
if err := stmt.QueryRow(acctId, orgId).Scan(&count); err != nil {
return false, errors.Wrap(err, "error executing organization_members count statement")
}
return count > 0, nil
}
func (str *Store) IsAccountAdminOfOrganization(acctId, orgId int, trx *sqlx.Tx) (bool, error) {
stmt, err := trx.Prepare("select count(0) from organization_members where organization_id = $1 and account_id = $2 and admin")
if err != nil {
return false, errors.Wrap(err, "error preparing organization_members count statement")
}
var count int
if err := stmt.QueryRow(acctId, orgId).Scan(&count); err != nil {
return false, errors.Wrap(err, "error executing organization_members count statement")
}
return count > 0, nil
}
func (str *Store) RemoveAccountFromOrganization(acctId, orgId int, trx *sqlx.Tx) error {
stmt, err := trx.Prepare("delete from organization_members where organization_id = $1 and account_id = $2")
if err != nil {
return errors.Wrap(err, "error preparing organization_members delete statement")
}
_, err = stmt.Exec(acctId, orgId)
if err != nil {
return errors.Wrap(err, "error executing organization_members delete statement")
}
return nil
}