mleku
59247400dc
Some checks failed
Go / build-and-release (push) Has been cancelled
Dgraph-related functionality, configuration, and benchmarks have been removed from the project. This streamlines the codebase to focus on supported backends, specifically eliminating Dgraph references in favor of Neo4j and other implementations. Version bumped to reflect the changes.
110 lines
4.5 KiB
Go
110 lines
4.5 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"time"
|
|
|
|
"next.orly.dev/pkg/database/indexes/types"
|
|
"git.mleku.dev/mleku/nostr/encoders/event"
|
|
"git.mleku.dev/mleku/nostr/encoders/filter"
|
|
"git.mleku.dev/mleku/nostr/encoders/tag"
|
|
"next.orly.dev/pkg/interfaces/store"
|
|
)
|
|
|
|
// Database defines the interface that all database implementations must satisfy.
|
|
// This allows switching between different storage backends (badger, neo4j, etc.)
|
|
type Database interface {
|
|
// Core lifecycle methods
|
|
Path() string
|
|
Init(path string) error
|
|
Sync() error
|
|
Close() error
|
|
Wipe() error
|
|
SetLogLevel(level string)
|
|
Ready() <-chan struct{} // Returns a channel that closes when database is ready to serve requests
|
|
|
|
// Event storage and retrieval
|
|
SaveEvent(c context.Context, ev *event.E) (exists bool, err error)
|
|
GetSerialsFromFilter(f *filter.F) (serials types.Uint40s, err error)
|
|
WouldReplaceEvent(ev *event.E) (bool, types.Uint40s, error)
|
|
|
|
QueryEvents(c context.Context, f *filter.F) (evs event.S, err error)
|
|
QueryAllVersions(c context.Context, f *filter.F) (evs event.S, err error)
|
|
QueryEventsWithOptions(c context.Context, f *filter.F, includeDeleteEvents bool, showAllVersions bool) (evs event.S, err error)
|
|
QueryDeleteEventsByTargetId(c context.Context, targetEventId []byte) (evs event.S, err error)
|
|
QueryForSerials(c context.Context, f *filter.F) (serials types.Uint40s, err error)
|
|
QueryForIds(c context.Context, f *filter.F) (idPkTs []*store.IdPkTs, err error)
|
|
|
|
CountEvents(c context.Context, f *filter.F) (count int, approximate bool, err error)
|
|
|
|
FetchEventBySerial(ser *types.Uint40) (ev *event.E, err error)
|
|
FetchEventsBySerials(serials []*types.Uint40) (events map[uint64]*event.E, err error)
|
|
|
|
GetSerialById(id []byte) (ser *types.Uint40, err error)
|
|
GetSerialsByIds(ids *tag.T) (serials map[string]*types.Uint40, err error)
|
|
GetSerialsByIdsWithFilter(ids *tag.T, fn func(ev *event.E, ser *types.Uint40) bool) (serials map[string]*types.Uint40, err error)
|
|
GetSerialsByRange(idx Range) (serials types.Uint40s, err error)
|
|
|
|
GetFullIdPubkeyBySerial(ser *types.Uint40) (fidpk *store.IdPkTs, err error)
|
|
GetFullIdPubkeyBySerials(sers []*types.Uint40) (fidpks []*store.IdPkTs, err error)
|
|
|
|
// Event deletion
|
|
DeleteEvent(c context.Context, eid []byte) error
|
|
DeleteEventBySerial(c context.Context, ser *types.Uint40, ev *event.E) error
|
|
DeleteExpired()
|
|
ProcessDelete(ev *event.E, admins [][]byte) error
|
|
CheckForDeleted(ev *event.E, admins [][]byte) error
|
|
|
|
// Import/Export
|
|
Import(rr io.Reader)
|
|
Export(c context.Context, w io.Writer, pubkeys ...[]byte)
|
|
ImportEventsFromReader(ctx context.Context, rr io.Reader) error
|
|
ImportEventsFromStrings(ctx context.Context, eventJSONs []string, policyManager interface{ CheckPolicy(action string, ev *event.E, pubkey []byte, remote string) (bool, error) }) error
|
|
|
|
// Relay identity
|
|
GetRelayIdentitySecret() (skb []byte, err error)
|
|
SetRelayIdentitySecret(skb []byte) error
|
|
GetOrCreateRelayIdentitySecret() (skb []byte, err error)
|
|
|
|
// Markers (metadata key-value storage)
|
|
SetMarker(key string, value []byte) error
|
|
GetMarker(key string) (value []byte, err error)
|
|
HasMarker(key string) bool
|
|
DeleteMarker(key string) error
|
|
|
|
// Subscriptions (payment-based access control)
|
|
GetSubscription(pubkey []byte) (*Subscription, error)
|
|
IsSubscriptionActive(pubkey []byte) (bool, error)
|
|
ExtendSubscription(pubkey []byte, days int) error
|
|
RecordPayment(pubkey []byte, amount int64, invoice, preimage string) error
|
|
GetPaymentHistory(pubkey []byte) ([]Payment, error)
|
|
ExtendBlossomSubscription(pubkey []byte, tier string, storageMB int64, daysExtended int) error
|
|
GetBlossomStorageQuota(pubkey []byte) (quotaMB int64, err error)
|
|
IsFirstTimeUser(pubkey []byte) (bool, error)
|
|
|
|
// NIP-43 Invite-based ACL
|
|
AddNIP43Member(pubkey []byte, inviteCode string) error
|
|
RemoveNIP43Member(pubkey []byte) error
|
|
IsNIP43Member(pubkey []byte) (isMember bool, err error)
|
|
GetNIP43Membership(pubkey []byte) (*NIP43Membership, error)
|
|
GetAllNIP43Members() ([][]byte, error)
|
|
StoreInviteCode(code string, expiresAt time.Time) error
|
|
ValidateInviteCode(code string) (valid bool, err error)
|
|
DeleteInviteCode(code string) error
|
|
PublishNIP43MembershipEvent(kind int, pubkey []byte) error
|
|
|
|
// Migrations (version tracking for schema updates)
|
|
RunMigrations()
|
|
|
|
// Query cache methods
|
|
GetCachedJSON(f *filter.F) ([][]byte, bool)
|
|
CacheMarshaledJSON(f *filter.F, marshaledJSON [][]byte)
|
|
GetCachedEvents(f *filter.F) (event.S, bool)
|
|
CacheEvents(f *filter.F, events event.S)
|
|
InvalidateQueryCache()
|
|
|
|
// Utility methods
|
|
EventIdsBySerial(start uint64, count int) (evs []uint64, err error)
|
|
}
|