subscriptions w/o payment

ManasMadan · ManasMadan · commit 2064a4eb5a9b · 2024-09-02T22:10:19.000+05:30
diff --git a/Govt-Billing-React-Ad-Subscriptions/src/components/Login/Login.tsx b/Govt-Billing-React-Ad-Subscriptions/src/components/Login/Login.tsx
@@ -23,7 +23,7 @@ const Login: React.FC = () => {
   };
   const doSignUp = async (email: string, password: string) => {
     try {
-      await signUpWithEmailAndPassword(email, password);
+      await signUpWithEmailAndPassword(email, password, "bronze");
       closeLoginModal();
     } catch {
       console.error("Something Went Wrong");
diff --git a/Govt-Billing-React-Ad-Subscriptions/src/components/Menu/Menu.tsx b/Govt-Billing-React-Ad-Subscriptions/src/components/Menu/Menu.tsx
@@ -1,4 +1,4 @@
-import React, { useState } from "react";
+import React, { useEffect, useState } from "react";
 import * as AppGeneral from "../socialcalc/index.js";
 import { File, Local } from "../Storage/LocalStorage";
 import { isPlatform, IonToast } from "@ionic/react";
@@ -8,6 +8,13 @@ import { IonActionSheet, IonAlert } from "@ionic/react";
 import { saveOutline, save, mail, print } from "ionicons/icons";
 import { APP_NAME } from "../../app-data.js";
 
+import {
+  canUserPerformAction,
+  updateUserQuota,
+  updateUserSubscription,
+} from "../../firebase/firestore.js";
+import useUser from "../../hooks/useUser.js";
+
 const Menu: React.FC<{
   showM: boolean;
   setM: Function;
@@ -22,6 +29,28 @@ const Menu: React.FC<{
   const [showAlert4, setShowAlert4] = useState(false);
   const [showToast1, setShowToast1] = useState(false);
   const [toastMessage, setToastMessage] = useState("");
+  const [canPrint, setCanPrint] = useState(false);
+  const [canEmail, setCanEmail] = useState(false);
+  const { user, isLoading } = useUser();
+  const [showUpgradeAlert, setShowUpgradeAlert] = useState(false);
+
+  const handleUpgradeSubscription = async (tier: string) => {
+    if (user) {
+      try {
+        await updateUserSubscription(user.uid, tier);
+        setToastMessage(`Successfully upgraded to ${tier} tier!`);
+        setShowToast1(true);
+        checkUserPermissions();
+      } catch (error) {
+        console.error("Error upgrading subscription:", error);
+        setToastMessage("Failed to upgrade subscription. Please try again.");
+        setShowToast1(true);
+      }
+    } else {
+      setToastMessage("Please log in to upgrade your subscription.");
+      setShowToast1(true);
+    }
+  };
   /* Utility functions */
   const _validateName = async (filename) => {
     filename = filename.trim();
@@ -55,19 +84,52 @@ const Menu: React.FC<{
     }
     return filename;
   };
+  useEffect(() => {
+    checkUserPermissions();
+  }, [isLoading]);
+
+  const checkUserPermissions = async () => {
+    if (user) {
+      const printPermission = await canUserPerformAction(user.uid, "print");
+      const emailPermission = await canUserPerformAction(user.uid, "email");
+      setCanPrint(printPermission);
+      setCanEmail(emailPermission);
+    }
+  };
+
+  const doPrint = async () => {
+    if (!canPrint) {
+      setToastMessage("You've reached your print quota limit.");
+      setShowToast1(true);
+      return;
+    }
+
+    if (user) {
+      const updated = await updateUserQuota(user.uid, "print");
+      if (!updated) {
+        setToastMessage(
+          "Failed to update quota. You may have reached your limit."
+        );
+        setShowToast1(true);
+        return;
+      }
+    }
 
-  const doPrint = () => {
     if (isPlatform("hybrid")) {
       const printer = Printer;
       printer.print(AppGeneral.getCurrentHTMLContent());
     } else {
       const content = AppGeneral.getCurrentHTMLContent();
-      // useReactToPrint({ content: () => content });
       const printWindow = window.open("/printwindow", "Print Invoice");
       printWindow.document.write(content);
       printWindow.print();
     }
+
+    setToastMessage("Print job sent successfully.");
+    setShowToast1(true);
+    checkUserPermissions(); // Update permissions after printing
   };
+
   const doSave = () => {
     if (props.file === "default") {
       setShowAlert1(true);
@@ -113,7 +175,24 @@ const Menu: React.FC<{
     }
   };
 
-  const sendEmail = () => {
+  const sendEmail = async () => {
+    if (!canEmail) {
+      setToastMessage("You've reached your email quota limit.");
+      setShowToast1(true);
+      return;
+    }
+
+    if (user) {
+      const updated = await updateUserQuota(user.uid, "email");
+      if (!updated) {
+        setToastMessage(
+          "Failed to update quota. You may have reached your limit."
+        );
+        setShowToast1(true);
+        return;
+      }
+    }
+
     if (isPlatform("hybrid")) {
       const content = AppGeneral.getCurrentHTMLContent();
       const base64 = btoa(content);
@@ -127,9 +206,15 @@ const Menu: React.FC<{
         subject: `${APP_NAME} attached`,
         isHtml: true,
       });
+
+      setToastMessage("Email sent successfully.");
+      setShowToast1(true);
     } else {
-      alert("This Functionality works on Anroid/IOS devices");
+      setToastMessage("This functionality works on Android/iOS devices only.");
+      setShowToast1(true);
     }
+
+    checkUserPermissions();
   };
 
   return (
@@ -172,6 +257,14 @@ const Menu: React.FC<{
               console.log("Email clicked");
             },
           },
+          {
+            text: "Upgrade Subscription",
+            icon: "arrow-up-circle-outline",
+            handler: () => {
+              setShowUpgradeAlert(true);
+              console.log("Upgrade Subscription clicked");
+            },
+          },
         ]}
       />
       <IonAlert
@@ -225,6 +318,31 @@ const Menu: React.FC<{
         }
         buttons={["Ok"]}
       />
+      <IonAlert
+        animated
+        isOpen={showUpgradeAlert}
+        onDidDismiss={() => setShowUpgradeAlert(false)}
+        header="Upgrade Subscription"
+        message="Choose a subscription tier to upgrade:"
+        buttons={[
+          {
+            text: "Bronze ($150/month)",
+            handler: () => handleUpgradeSubscription("bronze"),
+          },
+          {
+            text: "Silver ($200/month)",
+            handler: () => handleUpgradeSubscription("silver"),
+          },
+          {
+            text: "Gold ($250/month)",
+            handler: () => handleUpgradeSubscription("gold"),
+          },
+          {
+            text: "Cancel",
+            role: "cancel",
+          },
+        ]}
+      />
       <IonToast
         animated
         isOpen={showToast1}
diff --git a/Govt-Billing-React-Ad-Subscriptions/src/firebase/auth.ts b/Govt-Billing-React-Ad-Subscriptions/src/firebase/auth.ts
@@ -4,15 +4,23 @@ import {
   signOut,
 } from "firebase/auth";
 import { auth } from "./index";
+import { initializeUserSubscription, subscriptionTiers } from "./firestore";
 
-const signUpWithEmailAndPassword = async (email: string, password: string) => {
+const signUpWithEmailAndPassword = async (
+  email: string,
+  password: string,
+  tier: keyof typeof subscriptionTiers
+) => {
   try {
     const userCredential = await createUserWithEmailAndPassword(
       auth,
       email,
       password
     );
     const user = userCredential.user;
+
+    await initializeUserSubscription(user.uid, tier);
+
     return user;
   } catch (error) {
     const errorCode = error.code;
diff --git a/Govt-Billing-React-Ad-Subscriptions/src/firebase/firestore.ts b/Govt-Billing-React-Ad-Subscriptions/src/firebase/firestore.ts
@@ -9,6 +9,7 @@ import {
   where,
   getDoc,
   deleteDoc,
+  updateDoc,
 } from "firebase/firestore";
 import { Local } from "../components/Storage/LocalStorage";
 
@@ -129,10 +130,97 @@ const deleteFileFromFirebase = async (userId, key, onSuccess) => {
     alert("Something went wrong");
   }
 };
+const subscriptionTiers = {
+  bronze: { prints: 10, emails: 10, price: 150 },
+  silver: { prints: 20, emails: 20, price: 200 },
+  gold: { prints: 30, emails: 30, price: 250 },
+};
+
+interface UserSubscription {
+  tier: keyof typeof subscriptionTiers;
+  remainingPrints: number;
+  remainingEmails: number;
+  expirationDate: Date;
+}
+
+const getUserSubscription = async (
+  userId: string
+): Promise<UserSubscription | null> => {
+  const docRef = doc(db, "subscriptions", userId);
+  const docSnap = await getDoc(docRef);
+
+  if (docSnap.exists()) {
+    return docSnap.data() as UserSubscription;
+  } else {
+    console.log("No subscription found for this user");
+    return null;
+  }
+};
+
+const initializeUserSubscription = async (
+  userId: string,
+  tier: keyof typeof subscriptionTiers
+) => {
+  const subscriptionData: UserSubscription = {
+    tier,
+    remainingPrints: subscriptionTiers[tier].prints,
+    remainingEmails: subscriptionTiers[tier].emails,
+    expirationDate: new Date(Date.now() + 30 * 24 * 60 * 60 * 1000), // 30 days from now
+  };
+
+  await setDoc(doc(db, "subscriptions", userId), subscriptionData);
+};
+
+const updateUserQuota = async (
+  userId: string,
+  action: "print" | "email"
+): Promise<boolean> => {
+  const subscription = await getUserSubscription(userId);
+  if (!subscription) return false;
+
+  const quotaField = action === "print" ? "remainingPrints" : "remainingEmails";
+  if (subscription[quotaField] > 0) {
+    await updateDoc(doc(db, "subscriptions", userId), {
+      [quotaField]: subscription[quotaField] - 1,
+    });
+    return true;
+  }
+  return false;
+};
+
+const canUserPerformAction = async (
+  userId: string,
+  action: "print" | "email"
+): Promise<boolean> => {
+  const subscription = await getUserSubscription(userId);
+  if (!subscription) return false;
+
+  const quotaField = action === "print" ? "remainingPrints" : "remainingEmails";
+  return subscription[quotaField] > 0;
+};
+
+const updateUserSubscription = async (userId: string, tier: string) => {
+  const subscriptionData = {
+    tier,
+    remainingPrints: subscriptionTiers[tier].prints,
+    remainingEmails: subscriptionTiers[tier].emails,
+    expirationDate: new Date(Date.now() + 30 * 24 * 60 * 60 * 1000), // 30 days from now
+  };
+
+  await setDoc(doc(db, "subscriptions", userId), subscriptionData, {
+    merge: true,
+  });
+};
 
 export {
   uploadFileToCloud,
   getFilesKeysFromFirestore,
   downloadFileFromFirebase,
   deleteFileFromFirebase,
+  getUserSubscription,
+  updateUserQuota,
+  canUserPerformAction,
+  initializeUserSubscription,
+  subscriptionTiers,
+  updateUserSubscription,
 };
