2 files changed, 21 insertions, 0 deletions
diff --git a/ufund-ui/src/app/components/need-list/need-list.component.html b/ufund-ui/src/app/components/need-list/need-list.component.html
index 07f6735..fce6377 100644
--- a/ufund-ui/src/app/components/need-list/need-list.component.html
+++ b/ufund-ui/src/app/components/need-list/need-list.component.html
@@ -14,6 +14,7 @@
                 {{need.name}}
             </a>
             <button (click)="delete(need.id)" *ngIf="isManager()">Delete</button>
+            <button (click)="add(need)" *ngIf="isHelper()">Add To Basket</button>
         </div>
     </div>
 </div>
@@ -23,4 +24,5 @@
         {{need.name}}
     </a>
     <button (click)="delete(need.id)" *ngIf="isManager()">Delete</button>
+    <button (click)="add(need)" *ngIf="isHelper()">Add To Basket</button>
 </li>
diff --git a/ufund-ui/src/app/components/need-list/need-list.component.ts b/ufund-ui/src/app/components/need-list/need-list.component.ts
index b21979f..953904c 100644
--- a/ufund-ui/src/app/components/need-list/need-list.component.ts
+++ b/ufund-ui/src/app/components/need-list/need-list.component.ts
@@ -96,6 +96,25 @@ export class NeedListComponent {
     return type === ("MANAGER" as unknown as userType);
   }
 
+  isHelper() {
+    const type = this.usersService.getCurrentUser()?.type;
+    console.log(type);
+    return type === ("HELPER" as unknown as userType);
+  }
+
+  add(need: Need) {
+    const currentUser = this.usersService.getCurrentUser();
+    if (currentUser) {
+      this.usersService.updateUser(currentUser.username).subscribe(() => {
+        const currentUser = this.usersService.getCurrentUser();
+        if (currentUser && currentUser.basket) {
+          currentUser.basket.push(need);
+        }
+      });
+    }
+   
+  }
+
   back() {
     this.searchResults = [];
   }