feat: enhance budget management in Vault class

- Updated budget handling to differentiate between new and existing budgets.
- Implemented logic to register budgets as dirty (update) or new (insert) based on their existence.
- Modified budget deletion queries in VaultDrizzleRepository to ensure correct vault association during updates and deletions.

Author: isaacbatst

src/vault/domain/vault.ts

@@ -174,11 +174,22 @@ export class Vault {
     if (amount < 0) {
       return left('O valor do orçamento não pode ser negativo');
     }
+    const existingBudget = this.budgets.get(category.id);
     this.budgets.set(category.id, { category, amount });
-    this.budgetsTracker.registerNew({
-      category,
-      amount,
-    });
+    
+    if (existingBudget) {
+      // Budget already exists, register as dirty (update)
+      this.budgetsTracker.registerDirty({
+        category,
+        amount,
+      });
+    } else {
+      // New budget, register as new (insert)
+      this.budgetsTracker.registerNew({
+        category,
+        amount,
+      });
+    }
     return right(true);
   }
 

src/vault/repositories/drizzle/vault-drizzle.repository.ts

@@ -1,5 +1,5 @@
 import { Inject, Injectable, Logger } from '@nestjs/common';
-import { eq, inArray } from 'drizzle-orm';
+import { and, eq, inArray } from 'drizzle-orm';
 import { BatchItem } from 'drizzle-orm/batch';
 import {
   DRIZZLE_DATABASE,
@@ -108,7 +108,14 @@ export class VaultDrizzleRepository extends VaultRepository {
     // Delete removed budgets
     for (const b of budgetChanges.deleted) {
       queries.push(
-        this.db.delete(budget).where(eq(budget.categoryId, b.category.id)),
+        this.db
+          .delete(budget)
+          .where(
+            and(
+              eq(budget.vaultId, vaultEntity.id),
+              eq(budget.categoryId, b.category.id),
+            ),
+          ),
       );
     }
 
@@ -118,12 +125,16 @@ export class VaultDrizzleRepository extends VaultRepository {
         this.db
           .update(budget)
           .set({ amount: b.amount })
-          .where(eq(budget.categoryId, b.category.id)),
+          .where(
+            and(
+              eq(budget.vaultId, vaultEntity.id),
+              eq(budget.categoryId, b.category.id),
+            ),
+          ),
       );
     }
 
     // Execute all queries in a batch
-    console.log('queries', queries);
     if (queries.length > 0) {
       const [first, ...rest] = queries;
       await this.db.batch([first, ...rest]);

src/vault/vault.service.ts

@@ -399,8 +399,11 @@ export class VaultService {
       input.vaultId,
     );
     for (const budget of input.budgets) {
+      // Support lookup by both ID and code for backwards compatibility
+      // (web UI sends ID, telegram bot sends code)
       const category = categories.find(
-        (cat) => cat.code === budget.categoryCode,
+        (cat) =>
+          cat.id === budget.categoryCode || cat.code === budget.categoryCode,
       );
       if (!category) {
         this.logger.warn(