relay: remove enforceFilters function
This commit is contained in:
parent
11474c5cbd
commit
5f73232418
|
@ -4,29 +4,19 @@ import {
|
||||||
NostrClientEVENT,
|
NostrClientEVENT,
|
||||||
NostrClientMsg,
|
NostrClientMsg,
|
||||||
NostrClientREQ,
|
NostrClientREQ,
|
||||||
NostrEvent,
|
NostrRelayMsg,
|
||||||
NostrFilter,
|
|
||||||
NSchema as n,
|
NSchema as n,
|
||||||
} from '@nostrify/nostrify';
|
} from '@nostrify/nostrify';
|
||||||
|
|
||||||
|
import { AppController } from '@/app.ts';
|
||||||
import { relayInfoController } from '@/controllers/nostr/relay-info.ts';
|
import { relayInfoController } from '@/controllers/nostr/relay-info.ts';
|
||||||
import * as pipeline from '@/pipeline.ts';
|
import * as pipeline from '@/pipeline.ts';
|
||||||
import { RelayError } from '@/RelayError.ts';
|
import { RelayError } from '@/RelayError.ts';
|
||||||
import { Storages } from '@/storages.ts';
|
import { Storages } from '@/storages.ts';
|
||||||
|
|
||||||
import type { AppController } from '@/app.ts';
|
|
||||||
import { Conf } from '@/config.ts';
|
|
||||||
|
|
||||||
/** Limit of initial events returned for a subscription. */
|
/** Limit of initial events returned for a subscription. */
|
||||||
const FILTER_LIMIT = 100;
|
const FILTER_LIMIT = 100;
|
||||||
|
|
||||||
/** NIP-01 relay to client message. */
|
|
||||||
type RelayMsg =
|
|
||||||
| ['EVENT', string, NostrEvent]
|
|
||||||
| ['NOTICE', string]
|
|
||||||
| ['EOSE', string]
|
|
||||||
| ['OK', string, boolean, string]
|
|
||||||
| ['COUNT', string, { count: number; approximate?: boolean }];
|
|
||||||
|
|
||||||
/** Set up the Websocket connection. */
|
/** Set up the Websocket connection. */
|
||||||
function connectStream(socket: WebSocket) {
|
function connectStream(socket: WebSocket) {
|
||||||
const controllers = new Map<string, AbortController>();
|
const controllers = new Map<string, AbortController>();
|
||||||
|
@ -65,17 +55,15 @@ function connectStream(socket: WebSocket) {
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Handle REQ. Start a subscription. */
|
/** Handle REQ. Start a subscription. */
|
||||||
async function handleReq([_, subId, ...rest]: NostrClientREQ): Promise<void> {
|
async function handleReq([_, subId, ...filters]: NostrClientREQ): Promise<void> {
|
||||||
const filters = prepareFilters(rest);
|
|
||||||
|
|
||||||
const controller = new AbortController();
|
const controller = new AbortController();
|
||||||
controllers.get(subId)?.abort();
|
controllers.get(subId)?.abort();
|
||||||
controllers.set(subId, controller);
|
controllers.set(subId, controller);
|
||||||
|
|
||||||
const db = await Storages.db();
|
const store = await Storages.db();
|
||||||
const pubsub = await Storages.pubsub();
|
const pubsub = await Storages.pubsub();
|
||||||
|
|
||||||
for (const event of await db.query(filters, { limit: FILTER_LIMIT })) {
|
for (const event of await store.query(filters, { limit: FILTER_LIMIT })) {
|
||||||
send(['EVENT', subId, event]);
|
send(['EVENT', subId, event]);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -118,30 +106,20 @@ function connectStream(socket: WebSocket) {
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Handle COUNT. Return the number of events matching the filters. */
|
/** Handle COUNT. Return the number of events matching the filters. */
|
||||||
async function handleCount([_, subId, ...rest]: NostrClientCOUNT): Promise<void> {
|
async function handleCount([_, subId, ...filters]: NostrClientCOUNT): Promise<void> {
|
||||||
const store = await Storages.db();
|
const store = await Storages.db();
|
||||||
const { count } = await store.count(prepareFilters(rest));
|
const { count } = await store.count(filters);
|
||||||
send(['COUNT', subId, { count, approximate: false }]);
|
send(['COUNT', subId, { count, approximate: false }]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Send a message back to the client. */
|
/** Send a message back to the client. */
|
||||||
function send(msg: RelayMsg): void {
|
function send(msg: NostrRelayMsg): void {
|
||||||
if (socket.readyState === WebSocket.OPEN) {
|
if (socket.readyState === WebSocket.OPEN) {
|
||||||
socket.send(JSON.stringify(msg));
|
socket.send(JSON.stringify(msg));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Enforce the filters with certain criteria. */
|
|
||||||
function prepareFilters(filters: NostrClientREQ[2][]): NostrFilter[] {
|
|
||||||
return filters.map((filter) => {
|
|
||||||
const narrow = Boolean(filter.ids?.length || filter.authors?.length);
|
|
||||||
const search = narrow ? filter.search : `domain:${Conf.url.host} ${filter.search ?? ''}`;
|
|
||||||
// Return only local events unless the query is already narrow.
|
|
||||||
return { ...filter, search };
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
const relayController: AppController = (c, next) => {
|
const relayController: AppController = (c, next) => {
|
||||||
const upgrade = c.req.header('upgrade');
|
const upgrade = c.req.header('upgrade');
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue