aljaz/ditto
1
0
Fork 0
mirror of https://github.com/aljazceru/ditto.git synced 2025-12-17 13:34:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for db:export filters

Browse Source
This commit is contained in:
Siddharth Singh
2024-08-25 19:41:35 +05:30
parent fb199123be
commit 125dfb54f3
2 changed files with 113 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
scripts/db-export.ts
View File

@@ -2,8 +2,6 @@ import { Storages } from '@/storages.ts';
import { NostrFilter } from '@nostrify/nostrify';
import { Command, InvalidOptionArgumentError } from 'commander';
const store = await Storages.db();
interface ExportFilter {
authors?: string[];
ids?: string[];
@@ -47,11 +45,11 @@ function die(code: number, ...args: any[]) {
function tagFilterShortcut(name: 'd' | 'e' | 'p', value: string) {
const val = [value];
if (findInvalid(val)) die(1, `ERROR: Invalid value supplied for ${name}-tag.`);
if (findInvalid(val)) throw new Error(`ERROR: Invalid value supplied for ${name}-tag.`);
return val;
}
async function exportEvents(args: ExportFilter) {
export function buildFilter(args: ExportFilter) {
const filter: NostrFilter = {};
const { authors, ids, kinds, d, e, limit, p, search, since, until, tags } = args;
if (since) {
@@ -62,18 +60,19 @@ async function exportEvents(args: ExportFilter) {
}
if (authors && authors.length) {
const invalid = findInvalid(authors);
if (invalid) die(1, `ERROR: Invalid pubkey ${invalid} supplied.`);
if (invalid) throw new Error(`ERROR: Invalid pubkey ${invalid} supplied.`);
filter.authors = authors;
}
if (ids) {
const invalid = findInvalid(ids);
if (invalid) die(1, `ERROR: Invalid event ID ${invalid} supplied.`);
if (invalid) throw new Error(`ERROR: Invalid event ID ${invalid} supplied.`);
filter.ids = ids;
}
if (kinds && kinds.length) {
filter.kinds = kinds;
}
if (d) {
filter['#d'] = tagFilterShortcut('d', d);
filter['#d'] = [d];
}
if (e) {
filter['#e'] = tagFilterShortcut('e', e);
@@ -94,6 +93,19 @@ async function exportEvents(args: ExportFilter) {
}
}
return filter;
}
async function exportEvents(args: ExportFilter) {
const store = await Storages.db();
let filter: NostrFilter = {};
try {
filter = buildFilter(args);
} catch (e) {
die(1, e.message || e.toString());
}
let count = 0;
for await (const msg of store.req([filter])) {
if (msg[0] === 'EOSE') {
@@ -144,5 +156,3 @@ if (import.meta.main) {
await exporter.parseAsync(Deno.args, { from: 'user' });
}
Deno.exit();