forked from mirrors/pronouns.cc
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
|
|
"emperror.dev/errors"
|
|
"github.com/georgysavva/scany/pgxscan"
|
|
"github.com/jackc/pgx/v4"
|
|
"github.com/rs/xid"
|
|
)
|
|
|
|
type Member struct {
|
|
ID xid.ID
|
|
UserID xid.ID
|
|
Name string
|
|
Bio *string
|
|
AvatarURL *string
|
|
Links []string
|
|
}
|
|
|
|
const ErrMemberNotFound = errors.Sentinel("member not found")
|
|
|
|
func (db *DB) Member(ctx context.Context, id xid.ID) (m Member, err error) {
|
|
sql, args, err := sq.Select("*").From("members").Where("id = ?", id).ToSql()
|
|
if err != nil {
|
|
return m, errors.Wrap(err, "building sql")
|
|
}
|
|
|
|
err = pgxscan.Get(ctx, db, &m, sql, args...)
|
|
if err != nil {
|
|
if errors.Cause(err) == pgx.ErrNoRows {
|
|
return m, ErrMemberNotFound
|
|
}
|
|
|
|
return m, errors.Wrap(err, "retrieving member")
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (db *DB) UserMember(ctx context.Context, userID xid.ID, memberRef string) (m Member, err error) {
|
|
sql, args, err := sq.Select("*").From("members").
|
|
Where("user_id = ? and (id = ? or name = ?)", userID, memberRef, memberRef).ToSql()
|
|
if err != nil {
|
|
return m, errors.Wrap(err, "building sql")
|
|
}
|
|
|
|
err = pgxscan.Get(ctx, db, &m, sql, args...)
|
|
if err != nil {
|
|
if errors.Cause(err) == pgx.ErrNoRows {
|
|
return m, ErrMemberNotFound
|
|
}
|
|
|
|
return m, errors.Wrap(err, "retrieving member")
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (db *DB) UserMembers(ctx context.Context, userID xid.ID) (ms []Member, err error) {
|
|
sql, args, err := sq.Select("*").From("members").Where("user_id = ?", userID).ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "building sql")
|
|
}
|
|
|
|
err = pgxscan.Select(ctx, db, &ms, sql, args...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "retrieving members")
|
|
}
|
|
|
|
if ms == nil {
|
|
ms = make([]Member, 0)
|
|
}
|
|
return ms, nil
|
|
}
|