import { MiddlewareHandler } from '@hono/hono';
import { NostrSigner, NStore } from '@nostrify/nostrify';

import { UserStore } from '@/storages/UserStore.ts';
import { Storages } from '@/storages.ts';

export const requireStore: MiddlewareHandler<{ Variables: { store: NStore } }> = async (c, next) => {
  if (!c.get('store')) {
    throw new Error('Store is required');
  }
  await next();
};

/** Store middleware. */
export const storeMiddleware: MiddlewareHandler<{ Variables: { signer?: NostrSigner; store: NStore } }> = async (
  c,
  next,
) => {
  const pubkey = await c.get('signer')?.getPublicKey();

  if (pubkey) {
    const store = new UserStore(pubkey, await Storages.admin());
    c.set('store', store);
  } else {
    c.set('store', await Storages.admin());
  }
  await next();
};