1 files changed, 74 insertions, 11 deletions
diff --git a/ufund-ui/src/app/components/funding-basket/funding-basket.component.ts b/ufund-ui/src/app/components/funding-basket/funding-basket.component.ts
index c44aa27..1dcebc5 100644
--- a/ufund-ui/src/app/components/funding-basket/funding-basket.component.ts
+++ b/ufund-ui/src/app/components/funding-basket/funding-basket.component.ts
@@ -1,6 +1,9 @@
-import {Component, OnInit} from '@angular/core';
-import {Router} from '@angular/router';
-import {UsersService} from '../../services/users.service';
+import { Component, OnInit } from '@angular/core';
+import {User} from '../../models/User';
+import { UsersService } from '../../services/users.service';
+import { Need } from '../../models/Need';
+import { NeedListComponent } from '../need-list/need-list.component';
+import { Router } from '@angular/router';
 
 @Component({
   selector: 'app-funding-basket',
@@ -8,16 +11,76 @@ import {UsersService} from '../../services/users.service';
   templateUrl: './funding-basket.component.html',
   styleUrl: './funding-basket.component.css'
 })
-export class FundingBasketComponent implements OnInit{
-    constructor(
-       private router: Router,
-       private userService: UsersService
-    ) {}
+export class FundingBasketComponent implements
+  OnInit {
+  user!: User;
+  basket: Need[] = [];
+  needCount = 0;
+  need_quantity: {[key: number]: number} = {};
 
-    ngOnInit() {
-        if (!this.userService.getCurrentUser()) {
-            this.router.navigate(['/login'], {queryParams: {redir: this.router.url}})
+  constructor(
+    private router: Router,
+      private usersService: UsersService
+  ) {}
+    
+  ngOnInit(): void {
+    if (!this.usersService.getCurrentUser()) {
+      this.router.navigate(['/login'], {queryParams: {redir: this.router.url}});
+      return;
+    }
+    this.usersService.getCurrentUser()?.subscribe(user => {
+      this.user = user;
+      this.getBasketNeeds();
+    });
+  }
+
+  getBasketNeeds(): void {
+    if (this.user && this.user.cupboard) {
+      this.user.cupboard.forEach(need => {
+        if (this.isInBasket(need)){
+          this.basket.push(need);
         }
+      });
+    }
+  }
+
+  isInBasket(need: Need): boolean {
+    return this.basket.some(n => n.id == need.id);
+  }
+  
+  addNeed(need: Need, quantity: number=1): void {
+    if (this.user && !this.isInBasket(need)) {
+      this.basket.push(need);
+      this.need_quantity[need.id] = quantity;
+    } if (this.isInBasket(need)) {
+        this.need_quantity[need.id] += quantity;
     }
+      this.needCount++;
+    }
+  
+  removeNeed(need: Need, quantity:number=1): void {
+    if (this.user && this.isInBasket(need)) {
+      this.need_quantity[need.id] -= quantity;
+      if (this.need_quantity[need.id] === 0) {
+        this.basket = this.basket.filter(n => need.id !== n.id);
+      }
+      this.needCount--;
+    }
+  }
+
+  removeAllNeeds(): void {
+    this.basket.forEach(need => {
+      this.need_quantity = [];
+    });
+    this.basket = [];
+    this.needCount = 0;
+  }
+
+  isBasketEmpty(): boolean {
+    return this.needCount === 0;
+  }
 
+  checkout(): void {
+    this.removeAllNeeds();
+  }
 }