mirror of
https://github.com/openziti/zrok.git
synced 2025-03-30 09:06:05 +02:00
roughed in store implementation for organizations and memberships (#537)
This commit is contained in:
parent
d6c8f206b2
commit
842f3e8c20
@ -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")
|
||||
|
36
controller/store/organization.go
Normal file
36
controller/store/organization.go
Normal 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
|
||||
}
|
54
controller/store/organizationMember.go
Normal file
54
controller/store/organizationMember.go
Normal 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
|
||||
}
|
Loading…
Reference in New Issue
Block a user