Update Distribution Functionality

package.json

@@ -38,6 +38,7 @@
   "dependencies": {
     "bcryptjs": "^2.4.3",
     "cors": "^2.8.5",
+    "decimal.js": "^10.6.0",
     "express": "^4.21.1",
     "express-rate-limit": "^7.5.0",
     "helmet": "^7.2.0",

src/components/v1/distribution/distribution.controller.ts

@@ -2,7 +2,7 @@ import type { Request, Response } from "express"
 import AppDataSource from "../../../config/persistence/data-source"
 import { DistributionEntity } from "./distribution.entity"
 import { DistributionService } from "./distribution.service"
-import type { ApiResponse, DistributionResponseDto, CreateDistributionDto } from "./distribution.dto"
+import type { ApiResponse, DistributionResponseDto, CreateDistributionDto, UpdateDistributionDto } from "./distribution.dto"
 
 const getDistributionService = () => {
   if (!AppDataSource.isInitialized) {
@@ -41,6 +41,35 @@ export const createDistribution = async (req: Request, res: Response): Promise<v
   }
 }
 
+export const updateDistribution = async (req: Request, res: Response): Promise<void> => {
+  try {
+    const distributionService = getDistributionService()
+    const { id } = req.params
+    const validatedData = req.body as UpdateDistributionDto
+
+    const distribution = await distributionService.updateDistribution(id, validatedData)
+
+    const response: ApiResponse<DistributionResponseDto> = {
+      data: distribution,
+      success: true,
+      message: "Distribution updated successfully",
+    }
+
+    res.status(200).json(response)
+  } catch (error) {
+    console.error("Error in updateDistribution:", error)
+
+    const isNotFound = error instanceof Error && error.message === "Distribution not found"
+    const status = isNotFound ? 404 : 500
+    const errorResponse: ApiResponse<null> = {
+      data: null,
+      success: false,
+      message: isNotFound && error instanceof Error ? error.message : "Internal server error",
+    }
+    res.status(status).json(errorResponse)
+  }
+}
+
 export const listDistributions = async (_req: Request, res: Response): Promise<void> => {
   try {
     const distributionService = getDistributionService()

src/components/v1/distribution/distribution.dto.ts

@@ -1,7 +1,8 @@
 import type { DistributionType, Network } from "../../../types/enums"
-import type { CreateDistributionInput } from "./distribution.validation"
+import type { CreateDistributionInput, UpdateDistributionInput } from "./distribution.validation"
 
 export type CreateDistributionDto = CreateDistributionInput
+export type UpdateDistributionDto = UpdateDistributionInput
 
 export interface DistributionResponseDto {
   id: string

src/components/v1/distribution/distribution.service.ts

@@ -1,6 +1,7 @@
+import { Decimal } from "decimal.js"
 import type { Repository } from "typeorm"
 import type { DistributionEntity } from "./distribution.entity"
-import type { CreateDistributionDto, DistributionResponseDto } from "./distribution.dto"
+import type { CreateDistributionDto, DistributionResponseDto, UpdateDistributionDto } from "./distribution.dto"
 import { DistributionStatus, Network } from "../../../types/enums"
 
 export class DistributionService {
@@ -20,6 +21,47 @@ export class DistributionService {
     }
   }
 
+  async updateDistribution(id: string, updateData: UpdateDistributionDto): Promise<DistributionResponseDto> {
+    try {
+      const distribution = await this.distributionRepository.findOne({ where: { id } })
+      if (!distribution) {
+        throw new Error("Distribution not found")
+      }
+
+      const updatedFields: Partial<DistributionEntity> = { ...updateData }
+
+      if (updateData.userAddress !== undefined) {
+        updatedFields.userAddress = updateData.userAddress.toLowerCase()
+      }
+      if (updateData.tokenAddress !== undefined) {
+        updatedFields.tokenAddress = updateData.tokenAddress.toLowerCase()
+      }
+      if (updateData.tokenSymbol !== undefined) {
+        updatedFields.tokenSymbol = updateData.tokenSymbol.toUpperCase()
+      }
+
+      if (updateData.totalAmount || updateData.usdRate) {
+        const totalAmount = updateData.totalAmount ?? distribution.totalAmount
+        const usdRate = updateData.usdRate ?? distribution.usdRate
+        updatedFields.totalUsdAmount = this.calculateTotalUsdAmount(totalAmount, usdRate)
+      }
+
+      if (updateData.metadata === null) {
+        updatedFields.metadata = null
+      } else if (updateData.metadata !== undefined) {
+        updatedFields.metadata = this.processMetadata(updateData.metadata)
+      }
+
+      Object.assign(distribution, updatedFields)
+      const savedDistribution = await this.distributionRepository.save(distribution)
+
+      return this.formatDistributionResponse(savedDistribution)
+    } catch (error) {
+      console.error("Error updating distribution:", error)
+      throw error instanceof Error ? error : new Error("Failed to update distribution")
+    }
+  }
+
   async listDistributions(limit = 50): Promise<DistributionResponseDto[]> {
     try {
       const distributions = await this.distributionRepository.find({
@@ -69,10 +111,9 @@ export class DistributionService {
 
   private calculateTotalUsdAmount(totalAmount: string, usdRate: string): string {
     try {
-      const amount = Number.parseFloat(totalAmount)
-      const rate = Number.parseFloat(usdRate)
-      const totalUsd = amount * rate
-      return totalUsd.toString()
+      const amount = new Decimal(totalAmount)
+      const rate = new Decimal(usdRate)
+      return amount.mul(rate).toString()
     } catch (error) {
       console.warn("Error calculating total USD amount:", error)
       return "0"

src/components/v1/distribution/distribution.validation.ts

@@ -1,5 +1,5 @@
 import { z } from "zod"
-import { DistributionType, Network } from "../../../types/enums"
+import { DistributionType, DistributionStatus, Network } from "../../../types/enums"
 
 export class ValidationError extends Error {
   constructor(message: string) {
@@ -46,7 +46,18 @@ export const createDistributionSchema = z.object({
     })
     .optional(),
 
-  metadata: z.record(z.any()).optional(),
+  metadata: z.record(z.any()).nullable().optional(),
+})
+
+export const updateDistributionSchema = createDistributionSchema.partial().extend({
+  status: z.nativeEnum(DistributionStatus, {
+    errorMap: () => ({ message: "status must be a valid DistributionStatus" }),
+  }).optional(),
+})
+
+export const updateDistributionParamsSchema = z.object({
+  id: z.string().uuid("id must be a valid UUID"),
 })
 
 export type CreateDistributionInput = z.infer<typeof createDistributionSchema>
+export type UpdateDistributionInput = z.infer<typeof updateDistributionSchema>

src/components/v1/distribution/distrubtion.routes.ts

@@ -1,14 +1,25 @@
 import EnhancedRouter from "../../../utils/enhancedRouter"
 import policyMiddleware from "../../../appMiddlewares/policy.middleware"
-import { createDistributionSchema } from "./distribution.validation"
+import {
+  createDistributionSchema,
+  updateDistributionSchema,
+  updateDistributionParamsSchema,
+} from "./distribution.validation"
 import {
   createDistribution,
+  updateDistribution,
   listDistributions,
 } from "./distribution.controller"
 
 const distributionRouter = new EnhancedRouter()
 
 distributionRouter.get("/", listDistributions)
 distributionRouter.post("/", policyMiddleware(createDistributionSchema), createDistribution)
+distributionRouter.patch(
+  "/:id",
+  policyMiddleware(updateDistributionParamsSchema, "params"),
+  policyMiddleware(updateDistributionSchema),
+  updateDistribution,
+)
 
 export default distributionRouter.getRouter()