From 7a76e7cbaec2a60e328d2c2d8f105ea3a852c9c3 Mon Sep 17 00:00:00 2001
From: Kamil Klecha <urlikodd@gmail.com>
Date: Sun, 17 Oct 2021 23:52:53 +0200
Subject: [PATCH] Added getting and setting cities list from firestore

---
 src/views/Forecast.vue | 51 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 48 insertions(+), 3 deletions(-)

diff --git a/src/views/Forecast.vue b/src/views/Forecast.vue
index 1fe09cc..0388bdf 100644
--- a/src/views/Forecast.vue
+++ b/src/views/Forecast.vue
@@ -3,7 +3,7 @@
     <v-card-title>Prognoza pogody</v-card-title>
     <v-row justify="space-around">
       <v-col pa-1 v-for="(city, i) in city_ids" :key="i">
-        <City :id="city"/>
+        <City :id="city" />
       </v-col>
     </v-row>
   </v-card>
@@ -12,6 +12,16 @@
 <script lang="ts">
 import { Component, Vue } from "vue-property-decorator";
 import City from "@/components/City.vue";
+import { nanoid } from "nanoid";
+import {
+  collection,
+  query,
+  doc,
+  where,
+  getDocs,
+  setDoc,
+  getFirestore,
+} from "firebase/firestore";
 
 @Component({
   components: {
@@ -19,10 +29,45 @@ import City from "@/components/City.vue";
   },
 })
 export default class Forecast extends Vue {
+  private db = getFirestore();
+  private users = collection(this.db, "users");
+
+  private created() {
+    this.getFirebaseUserData();
+  }
+
+  private async getFirebaseUserData() {
+    const q = query(this.users, where("uid", "==", this.user.uid));
+
+    const querySnapshot = await getDocs(q);
+
+    if (querySnapshot.size > 0) {
+      this.$data.dataID = querySnapshot.docs[0].id;
+      this.$data.cities = querySnapshot.docs[0].data().cities;
+    } else {
+      this.saveUser(nanoid());
+    }
+  }
+
+  private async saveUser(id: string) {
+    if (this.$data.dataID === undefined) {
+      this.$data.dataID = id;
+    }
+    await setDoc(doc(this.users, id), {
+      uid: this.user.uid,
+      cities: this.$data.cities,
+    });
+  }
+
+  get user() {
+    return this.$store.getters.user;
+  }
+
   private data() {
     return {
-      city_ids: [6695624, 759734, 3094802, 764849]
-    }
+      city_ids: [6695624, 759734, 3094802, 764849],
+      cities: [],
+    };
   }
 }
 </script>
\ No newline at end of file