DistributorV2

apps/distributor/src/app.ts

@@ -4,6 +4,7 @@ import { DistributorV1Worker } from './distributor'
 import { StandardMerkleTree } from '@openzeppelin/merkle-tree'
 import { selectAll } from 'app/utils/supabase/selectAll'
 import { supabaseAdmin } from './supabase'
+import { DistributorV2Worker } from './distributorv2'
 
 const logger = pino({
   level: process.env.LOG_LEVEL || 'info',
@@ -12,6 +13,7 @@ const logger = pino({
 
 // Initialize DistributorWorker
 const distributorV1Worker = new DistributorV1Worker(logger)
+const distributorV2Worker = new DistributorV2Worker(logger)
 
 // Initialize Express app
 const app = express()
@@ -32,6 +34,13 @@ distributorRouter.get('/v1', async (req: Request, res: Response) => {
   })
 })
 
+distributorRouter.get('/v2', async (req: Request, res: Response) => {
+  res.json({
+    distributor: true,
+    ...distributorV2Worker.toJSON(),
+  })
+})
+
 // Middleware for checking authorization
 const checkAuthorization = (req: Request, res: Response, next: () => void) => {
   if (!req.headers.authorization?.includes(process.env.SUPABASE_SERVICE_ROLE as string)) {
@@ -117,6 +126,22 @@ distributorRouter.post('/v1', checkAuthorization, async (req, res) => {
   })
 })
 
+distributorRouter.post('/v2', checkAuthorization, async (req, res) => {
+  const { id } = req.body as { id: string }
+  logger.info({ id }, 'Received request to calculate distribution')
+  try {
+    await distributorV2Worker.calculateDistribution(id)
+  } catch (err) {
+    logger.error(err, 'Error while calculating distribution')
+    throw err
+  }
+
+  res.json({
+    distributor: true,
+    id: id,
+  })
+})
+
 app.use('/distributor', distributorRouter)
 
 export default app