advanced voting service

IdanCo · Mar 24, 2019 · 7f00f6e · 7f00f6e
1 parent 4b0df63
commit 7f00f6e
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 25 deletions.
diff --git a/src/app/components/candidate/candidate.component.html b/src/app/components/candidate/candidate.component.html
@@ -12,28 +12,14 @@
         </button>
       </ng-container>
       <ng-template #rating>
-        <button (click)="onVote(1)" mat-mini-fab color="primary" [matBadge]="0" matBadgePosition="above after" matBadgeColor="accent">
+        <button (click)="onVote(1)" mat-mini-fab color="primary" [matBadge]="candidate.votes ? (candidate.votes[auth.user?.uid] || 0) : 0" matBadgePosition="above after" matBadgeColor="accent">
           <mat-icon>thumb_up</mat-icon>
         </button>
         <button (click)="onVote(-1)" mat-mini-fab color="accent">
           <mat-icon>thumb_down</mat-icon>
         </button>
       </ng-template>
     </mat-card-actions>
-
-    <!--<mat-card-actions class="ml-auto pr-2" *ngIf="!(auth.user$ | async)">-->
-      <!--<button (click)="notLoggedIn()" mat-mini-fab>-->
-        <!--<mat-icon>thumb_up</mat-icon>-->
-      <!--</button>-->
-    <!--</mat-card-actions>-->
-    <!--<mat-card-actions class="ml-auto pr-2" *ngIf="candidate.user.uid !== (auth.user$ | async)?.uid && ratingService.rating$ | async as rating">-->
-      <!--<button (click)="onUpVote()" mat-mini-fab color="primary" [matBadge]="rating[candidate.user.uid] || 0" matBadgePosition="above after" matBadgeColor="accent">-->
-        <!--<mat-icon>thumb_up</mat-icon>-->
-      <!--</button>-->
-      <!--<button (click)="onDownVote()" mat-mini-fab color="accent">-->
-        <!--<mat-icon>thumb_down</mat-icon>-->
-      <!--</button>-->
-    <!--</mat-card-actions>-->
   </mat-card-header>
   <img mat-card-image [src]="candidate.imageURL" [alt]="candidate.title">
   <mat-card-content>

diff --git a/src/app/components/candidate/candidate.component.ts b/src/app/components/candidate/candidate.component.ts
@@ -18,7 +18,7 @@ export class CandidateComponent implements OnInit {
   }
 
   onVote(vote: number) {
-    this.voteService.setVote(this.candidate, vote);
+    this.voteService.setVote(this.candidate, this.auth.user, vote);
   }
 
 }
diff --git a/src/app/services/vote.service.ts b/src/app/services/vote.service.ts
@@ -1,6 +1,5 @@
 import { Injectable } from '@angular/core';
 import { MatSnackBar } from '@angular/material';
-import { AuthService } from './auth.service';
 import { AngularFirestore } from '@angular/fire/firestore';
 import { Candidate } from '../types/candidate.model';
 
@@ -10,23 +9,35 @@ import { Candidate } from '../types/candidate.model';
 export class VoteService {
 
   constructor(private db: AngularFirestore,
-              private snackBar: MatSnackBar,
-              private auth: AuthService) { }
+              private snackBar: MatSnackBar) { }
 
-  setVote(candidate: Candidate, vote: number) {
-    if (!this.auth.user) {
+  setVote(candidate: Candidate, user: User, vote: number) {
+    if (!user) {
       this.snackBar.open('Please login to vote');
       return;
     }
 
-    candidate.votes = candidate.votes || {};
+    return Promise.all([
+      this.updateCandidateVotes(candidate, user, vote),
+      this.updateUserVotes(user, vote)
+    ])
+  }
 
-    let currentVote = (candidate.votes[this.auth.user.uid] || 0) + vote;
+  updateCandidateVotes(candidate: Candidate, user: User, vote: number) {
+    candidate.votes = candidate.votes || {};
+    let currentVote = (candidate.votes[user.uid] || 0) + vote;
     if (currentVote < 0) currentVote = 0;
 
-    candidate.votes[this.auth.user.uid] = currentVote;
+    candidate.votes[user.uid] = currentVote;
+    candidate.totalVotes = (candidate.totalVotes || 0) + vote;
+
+    return this.db.doc<Candidate>(`candidates/${candidate.user.uid}`).update(candidate);
+  }
 
-    this.db.doc<Candidate>(`candidates/${candidate.user.uid}`).update(candidate)
+  updateUserVotes(user: User, vote: number) {
+    let userTotalVotes = (user.totalVotes || 0) + vote;
+    if (userTotalVotes < 0) userTotalVotes = 0;
 
+    return this.db.doc<User>(`users/${user.uid}`).update({ totalVotes: userTotalVotes });
   }
 }
diff --git a/src/app/types/user.model.ts b/src/app/types/user.model.ts
@@ -4,4 +4,5 @@ interface User {
   photoURL?: string;
   displayName?: string;
   myCustomData?: string;
+  totalVotes?: number;
 }