pronounscc/backend/db/member.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
	AvatarURLs []string `db:"avatar_urls"`
	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
}