57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
|
|
"emperror.dev/errors"
|
|
"git.sleepycat.moe/sam/mercury/internal/database"
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/keegancsmith/sqlf"
|
|
"github.com/oklog/ulid/v2"
|
|
)
|
|
|
|
// PostStore is the interface to posts in the database.
|
|
type PostStore struct {
|
|
q Querier
|
|
}
|
|
|
|
// NewPostStore creates a new PostStore instance.
|
|
func NewPostStore(q Querier) *PostStore {
|
|
return &PostStore{q: q}
|
|
}
|
|
|
|
// ByID gets a post by its ID.
|
|
func (s *PostStore) ByID(ctx context.Context, id ulid.ULID) (p database.Post, err error) {
|
|
q := sqlf.Sprintf("SELECT * FROM posts WHERE id = %s", id)
|
|
|
|
p, err = Get[database.Post](ctx, s.q, q)
|
|
if err != nil {
|
|
if errors.Cause(err) == pgx.ErrNoRows {
|
|
return p, ErrNotFound
|
|
}
|
|
return p, errors.Wrap(err, "getting post")
|
|
}
|
|
return p, nil
|
|
}
|
|
|
|
func (s *PostStore) Create(ctx context.Context, blog database.Blog, content string, visibility database.PostVisibility) (p database.Post, err error) {
|
|
q := sqlf.Sprintf(
|
|
"INSERT INTO posts (id, blog_id, content, source, visibility) VALUES (%s, %s, %s, %s, %s) RETURNING *",
|
|
makeULID(), blog.ID, content, content, visibility)
|
|
|
|
p, err = Get[database.Post](ctx, s.q, q)
|
|
if err != nil {
|
|
return p, errors.Wrap(err, "creating post")
|
|
}
|
|
return p, nil
|
|
}
|
|
|
|
func (s *PostStore) LocalCount(ctx context.Context) (count int64, err error) {
|
|
q := sqlf.Sprintf("SELECT count(*) FROM posts WHERE visibility = %s OR visibility = %s", database.PublicVisibility, database.UnlistedVisibility)
|
|
|
|
err = s.q.QueryRow(ctx, q.Query(sqlf.PostgresBindVar)).Scan(&count)
|
|
if err != nil {
|
|
return count, errors.Wrap(err, "getting count")
|
|
}
|
|
return count, nil
|
|
}
|