ref to campaign instead of string in votes

Zenika · Jul 30, 2018 · 1fd94c9 · 1fd94c9
1 parent f40abb2
commit 1fd94c9
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/functions/src/cast-vote.ts b/functions/src/cast-vote.ts
@@ -1,9 +1,9 @@
-import * as admin from "firebase-admin";
+import { firestore } from "firebase-admin";
 import * as functions from "firebase-functions";
 import { Config, isEnabled, asNumber, asBoolean } from "./config";
 import { selectCampaign } from "./select-campaign";
 
-const db = admin.firestore();
+const db = firestore();
 const config = functions.config() as Config;
 const requireUniqueVote = asBoolean(
   config.features.voting_campaigns.require_unique_vote
@@ -18,8 +18,8 @@ interface RequestPayload {
 export interface Vote {
   value: string;
   voter: string;
-  campaign: string;
-  recordedAt: admin.firestore.Timestamp;
+  campaign: firestore.DocumentReference;
+  recordedAt: firestore.Timestamp;
 }
 
 export const castVote = functions.https.onCall(
@@ -52,8 +52,8 @@ export const castVote = functions.https.onCall(
       .substr(0, 7);
 
     const vote: Vote = {
-      campaign: campaignId,
-      recordedAt: admin.firestore.Timestamp.fromDate(voteDate),
+      campaign: db.collection("voting-campaign").doc(campaignId),
+      recordedAt: firestore.Timestamp.fromDate(voteDate),
       voter: payload.voter,
       value: payload.vote
     };