highperfocused/nostream
1
0
Fork 0
mirror of https://github.com/Cameri/nostream.git synced 2025-03-17 21:31:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: support author/id prefix search

Browse Source
This commit is contained in:
Ricardo Arturo Cabral Mejia 2022-08-13 06:33:37 +00:00
parent bd5360ee7e
commit 92d5688693
No known key found for this signature in database
GPG Key ID: 5931EBF43A650245
1 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/repositories/event-repository.ts
View File

@ -9,6 +9,8 @@ import { isGenericTagQuery } from '../utils/filter'
import { toBuffer, toJSON } from '../utils/transforms'
const evenLengthTruncate = (input: string) => input.substring(0, input.length >> 1 << 1)
export class EventRepository implements IEventRepository {
public constructor(private readonly dbClient: Knex) {}
@ -17,11 +19,36 @@ export class EventRepository implements IEventRepository {
const builder = this.dbClient<DBEvent>('events')
if (Array.isArray(filter.authors)) {
builder.whereIn('event_pubkey', filter.authors.map(toBuffer))
builder.andWhere(function (bd) {
bd.whereIn(
'event_pubkey',
filter.authors.filter((author) => author.length === 64).map(toBuffer)
)
for (const author of filter.authors.filter((author) => author.length < 64)) {
const prefix = evenLengthTruncate(author)
if (prefix.length) {
bd.orWhereRaw('substring("event_pubkey" from 1 for ?) = ?', [prefix.length >> 1, toBuffer(prefix)])
}
}
})
}
if (Array.isArray(filter.ids)) {
builder.whereIn('event_id', filter.ids.map(toBuffer))
builder.andWhere(function (bd) {
bd.whereIn(
'event_id',
filter.ids.filter((id) => id.length === 64).map(toBuffer)
)
for (const id of filter.ids.filter((id) => id.length < 64)) {
const prefix = evenLengthTruncate(id)
if (prefix.length) {
bd.orWhereRaw('substring("event_id" from 1 for ?) = ?', [prefix.length >> 1, toBuffer(prefix)])
}
}
})
}
if (Array.isArray(filter.kinds)) {
@ -60,6 +87,8 @@ export class EventRepository implements IEventRepository {
query.union(subqueries, true)
}
console.log('query', query.toString())
return query.stream()
}