Projeto desafio DIO

app.ts

@@ -1,9 +1,17 @@
+import { log } from 'console'
 import { CompanyAccount } from './class/CompanyAccount'
 import { PeopleAccount } from './class/PeopleAccount'
+import { SpecialAccount } from './class/SpecialAccount'
 
 const peopleAccount: PeopleAccount = new PeopleAccount(1, 'Nath', 10)
 console.log(peopleAccount)
-peopleAccount.deposit()
+peopleAccount.deposit(100)
+
 const companyAccount: CompanyAccount = new CompanyAccount('DIO', 20)
-companyAccount.deposit()
-console.log(companyAccount)
+companyAccount.deposit(100)
+companyAccount.getLoan(100)
+console.log(companyAccount)
+
+const specialAccount: SpecialAccount = new SpecialAccount('Everton', 30)
+specialAccount.deposit(100)
+console.log(specialAccount);

class/CompanyAccount.ts

@@ -6,7 +6,12 @@ export class CompanyAccount extends DioAccount {
     super(name, accountNumber)
   }
 
-  getLoan = (): void => {
-    console.log('Voce pegou um empréstimo')
+  getLoan = (amount: number): void => {
+    if (this.getStatus()) {
+      this.balance += amount;
+      console.log('Voce pegou um empréstimo')
+    } else{
+      throw new Error("Não é possível pegar um empréstimo com a conta desativada")
+    }
   }
 }

class/DioAccount.ts

@@ -18,13 +18,21 @@ export abstract class DioAccount {
     return this.name
   }
 
-  deposit = (): void => {
+  getStatus = (): boolean => {
+    return this.status;
+  }
+
+  deposit = (amount: number): void => {
     if(this.validateStatus()){
+      this.balance += amount;
       console.log('Voce depositou')
     }
   }
 
-  withdraw = (): void => {
+  withdraw = (amount: number): void => {
+    if (this.balance > amount) {
+      this.balance -= amount;
+    }
     console.log('Voce sacou')
   }
 
@@ -39,4 +47,4 @@ export abstract class DioAccount {
 
     throw new Error('Conta inválida')
   }
-}
+}

class/SpecialAccount.ts

@@ -0,0 +1,8 @@
+import { DioAccount } from "./DioAccount";
+
+export class SpecialAccount extends DioAccount{
+
+    deposit = (amount: number): void => {
+        this.balance += amount + 10;
+    }
+}