From f631c29fb2d5ac32dd6031ddcb9cb603ed32e606 Mon Sep 17 00:00:00 2001
From: Emily <adharaultimasheireen@gmail.com>
Date: Wed, 16 Apr 2025 17:13:19 +0200
Subject: [PATCH] fix bugs

---
 dashboard/components/settings/General.vue      |  2 +-
 dashboard/pages/dashboard/events.vue           | 17 +----------------
 dashboard/pages/dashboard/visits.vue           | 18 ++----------------
 dashboard/pages/register.vue                   |  7 ++++---
 dashboard/server/api/auth/google_login.post.ts | 10 +++++-----
 dashboard/server/api/project/generate_csv.ts   | 17 ++++-------------
 .../server/api/user/delete_account.delete.ts   |  3 ++-
 .../server/services/PaymentServiceHelper.ts    |  4 ++++
 payments/src/index.ts                          | 14 +++++++++++++-
 payments/src/routers/PaymentRouter.ts          | 15 +++++++++++++++
 scripts/consumer/deploy.ts                     |  6 ++++--
 scripts/producer/deploy.ts                     |  6 ++++--
 12 files changed, 59 insertions(+), 60 deletions(-)

diff --git a/dashboard/components/settings/General.vue b/dashboard/components/settings/General.vue
index 653ecc4..2497fb5 100644
--- a/dashboard/components/settings/General.vue
+++ b/dashboard/components/settings/General.vue
@@ -215,7 +215,7 @@ function copyProjectId() {
             </div>
         </template>
         <template #pdelete>
-            <div class="flex lg:justify-end" v-if="!isGuest">
+            <div class="flex lg:justify-end pb-30" v-if="!isGuest">
                 <LyxUiButton type="danger" @click="deleteProject()">
                     Delete project
                 </LyxUiButton>
diff --git a/dashboard/pages/dashboard/events.vue b/dashboard/pages/dashboard/events.vue
index 589c73e..c9df173 100644
--- a/dashboard/pages/dashboard/events.vue
+++ b/dashboard/pages/dashboard/events.vue
@@ -6,14 +6,6 @@ definePageMeta({ layout: 'dashboard' });
 
 const { project } = useProject();
 
-const { isPremium } = useLoggedUser();
-
-const selfhosted = useSelfhosted();
-const canDownload = computed(() => {
-    if (selfhosted) return true;
-    return isPremium.value;
-});
-
 const metricsInfo = ref<number>(0);
 
 const columns = [
@@ -111,17 +103,10 @@ function goToUpgrade() {
                 }" v-model="selectedTimeFrom" :options="options"></USelectMenu>
             </div>
 
-            <div v-if="canDownload" @click="downloadCSV()"
+            <div @click="downloadCSV()"
                 class="bg-[#57c78fc0] hover:bg-[#57c78fab] cursor-pointer text-text poppins font-semibold px-8 py-1 rounded-lg">
                 Download CSV
             </div>
-
-            <div v-if="!canDownload" @click="goToUpgrade()"
-                class="bg-[#57c78f46] hover:bg-[#57c78f42] flex gap-4 items-center cursor-pointer text-text poppins font-semibold px-8 py-2 rounded-lg">
-                <i class="far fa-lock"></i>
-                Upgrade plan for CSV
-            </div>
-
         </div>
 
 
diff --git a/dashboard/pages/dashboard/visits.vue b/dashboard/pages/dashboard/visits.vue
index 4f10277..e65c1d8 100644
--- a/dashboard/pages/dashboard/visits.vue
+++ b/dashboard/pages/dashboard/visits.vue
@@ -6,14 +6,6 @@ definePageMeta({ layout: 'dashboard' });
 
 const { project } = useProject();
 
-const { isPremium } = useLoggedUser();
-
-const selfhosted = useSelfhosted();
-const canDownload = computed(() => {
-    if (selfhosted) return true;
-    return isPremium.value;
-});
-
 const metricsInfo = ref<number>(0);
 
 const columns = [
@@ -116,17 +108,11 @@ function goToUpgrade() {
                 }" v-model="selectedTimeFrom" :options="options"></USelectMenu>
             </div>
 
-            <div v-if="canDownload" @click="downloadCSV()"
+            <div @click="downloadCSV()"
                 class="bg-[#57c78fc0] hover:bg-[#57c78fab] cursor-pointer text-text poppins font-semibold px-8 py-1 rounded-lg">
                 Download CSV
             </div>
-
-            <div v-if="!canDownload" @click="goToUpgrade()"
-                class="bg-[#57c78f46] hover:bg-[#57c78f42] flex gap-4 items-center cursor-pointer text-text poppins font-semibold px-8 py-2 rounded-lg">
-                <i class="far fa-lock"></i>
-                Upgrade plan for CSV
-            </div>
-
+            
         </div>
 
 
diff --git a/dashboard/pages/register.vue b/dashboard/pages/register.vue
index 0af53a6..f6b46ed 100644
--- a/dashboard/pages/register.vue
+++ b/dashboard/pages/register.vue
@@ -61,7 +61,8 @@ async function registerAccount() {
                     Sign up
                 </div>
 
-                <div class="text-lyx-lightmode-text dark:text-lyx-text/80 text-[1.2rem] font-light text-center w-[70%] poppins mt-2">
+                <div
+                    class="text-lyx-lightmode-text dark:text-lyx-text/80 text-[1.2rem] font-light text-center w-[70%] poppins mt-2">
                     Track web analytics and custom events
                     with extreme simplicity in under 30 sec.
                     <br>
@@ -88,8 +89,8 @@ async function registerAccount() {
                             Password must be at least 6 chars long
                         </div>
                         <div class="flex justify-center mt-4 z-[100]">
-                            <LyxUiButton :disabled="!canRegister" @click="registerAccount()" class="text-center"
-                                type="primary">
+                            <LyxUiButton :disabled="!canRegister" @click="canRegister ? registerAccount() : ''"
+                                class="text-center" type="primary">
                                 Sign up
                             </LyxUiButton>
                         </div>
diff --git a/dashboard/server/api/auth/google_login.post.ts b/dashboard/server/api/auth/google_login.post.ts
index 841456d..36d146d 100644
--- a/dashboard/server/api/auth/google_login.post.ts
+++ b/dashboard/server/api/auth/google_login.post.ts
@@ -4,6 +4,7 @@ import { createUserJwt } from '~/server/AuthManager';
 import { UserModel } from '@schema/UserSchema';
 import { EmailService } from '@services/EmailService';
 import { EmailServiceHelper } from '~/server/services/EmailServiceHelper';
+import { PaymentServiceHelper } from '~/server/services/PaymentServiceHelper';
 
 const { GOOGLE_AUTH_CLIENT_SECRET, GOOGLE_AUTH_CLIENT_ID } = useRuntimeConfig()
 
@@ -38,7 +39,6 @@ export default defineEventHandler(async event => {
     const user = await UserModel.findOne({ email: payload.email });
 
     if (user) {
-        user.google_tokens = tokens as any;
         await user.save();
         return {
             error: false,
@@ -46,28 +46,28 @@ export default defineEventHandler(async event => {
         }
     }
 
-
     const newUser = new UserModel({
         email: payload.email,
         given_name: payload.given_name,
         name: payload.name,
         locale: payload.locale,
         picture: payload.picture,
-        google_tokens: tokens,
         created_at: Date.now()
     });
 
     const savedUser = await newUser.save();
 
+    const [ok, error] = await PaymentServiceHelper.create_customer(savedUser.id);
+    if (!ok) throw error;
 
     setImmediate(() => {
-        const emailData = EmailService.getEmailServerInfo('brevolist_add', { email: payload.email as string });
+        if (!payload.email) return;
+        const emailData = EmailService.getEmailServerInfo('brevolist_add', { email: payload.email });
         EmailServiceHelper.sendEmail(emailData);
     });
 
 
     setImmediate(() => {
-        console.log('SENDING WELCOME EMAIL TO', payload.email);
         if (!payload.email) return;
         const emailData = EmailService.getEmailServerInfo('welcome', { target: payload.email });
         EmailServiceHelper.sendEmail(emailData);
diff --git a/dashboard/server/api/project/generate_csv.ts b/dashboard/server/api/project/generate_csv.ts
index a85e8ba..80d2214 100644
--- a/dashboard/server/api/project/generate_csv.ts
+++ b/dashboard/server/api/project/generate_csv.ts
@@ -4,21 +4,12 @@ import { UserModel } from "@schema/UserSchema";
 import { VisitModel } from "@schema/metrics/VisitSchema";
 import { EventModel } from "~/shared/schema/metrics/EventSchema";
 
-const { SELFHOSTED } = useRuntimeConfig();
-
 export default defineEventHandler(async event => {
 
-    const data = await getRequestDataOld(event, { requireSchema: false });
+    const data = await getRequestData(event, [], []);
     if (!data) return;
 
-    const { project, project_id, user } = data;
-
-
-    if (SELFHOSTED.toString() !== 'TRUE' && SELFHOSTED.toString() !== 'true') {
-        const PREMIUM_TYPE = project.premium_type;
-        if (PREMIUM_TYPE === 0) return setResponseStatus(event, 400, 'Project not premium');
-    }
-
+    const { project_id } = data;
 
     const { mode, slice } = getQuery(event);
 
@@ -82,12 +73,12 @@ export default defineEventHandler(async event => {
         const csvHeader = [
             "name",
             "session",
-            "metadata",            
+            "metadata",
             "website",
             "created_at",
         ];
 
-      
+
         const lines: any[] = [];
         eventsReportData.forEach(line => lines.push(line.toJSON()));
 
diff --git a/dashboard/server/api/user/delete_account.delete.ts b/dashboard/server/api/user/delete_account.delete.ts
index 8b63602..f58333d 100644
--- a/dashboard/server/api/user/delete_account.delete.ts
+++ b/dashboard/server/api/user/delete_account.delete.ts
@@ -14,6 +14,7 @@ import { BotTrafficOptionModel } from "~/shared/schema/shields/BotTrafficOptionS
 import { TeamMemberModel } from "~/shared/schema/TeamMemberSchema";
 import { PasswordModel } from "~/shared/schema/PasswordSchema";
 import { PremiumModel } from "~/shared/schema/PremiumSchema";
+import { PaymentServiceHelper } from "~/server/services/PaymentServiceHelper";
 
 export default defineEventHandler(async event => {
 
@@ -35,7 +36,7 @@ export default defineEventHandler(async event => {
     const limitdeletation = await UserLimitModel.deleteMany({ user_id: userData.id });
     const notifiesDeletation = await LimitNotifyModel.deleteMany({ user_id: userData.id });
 
-    // await StripeService.deleteCustomer(premium.customer_id);
+    await PaymentServiceHelper.delete_customer(premium.customer_id);
 
     for (const project of projects) {
         const project_id = project._id;
diff --git a/dashboard/server/services/PaymentServiceHelper.ts b/dashboard/server/services/PaymentServiceHelper.ts
index 63b9e8c..c554bf5 100644
--- a/dashboard/server/services/PaymentServiceHelper.ts
+++ b/dashboard/server/services/PaymentServiceHelper.ts
@@ -47,4 +47,8 @@ export class PaymentServiceHelper {
         return await this.send('/update_customer_info', { user_id, address });
     }
 
+    static async delete_customer(customer_id: string): PaymentServiceResponse<{ ok: true }> {
+        return await this.send('/delete_customer', { customer_id });
+    }
+
 }
\ No newline at end of file
diff --git a/payments/src/index.ts b/payments/src/index.ts
index 6f60568..7067c01 100644
--- a/payments/src/index.ts
+++ b/payments/src/index.ts
@@ -17,11 +17,23 @@ console.log('Stripe started in', STRIPE_TESTMODE ? 'TESTMODE' : 'LIVEMODE');
 
 const app = express();
 
+const TOKEN = process.env.TOKEN;
+
+if (!TOKEN || TOKEN.length == 0) {
+    console.log('TOKEN not set');
+    process.exit();
+}
 
 app.use((req, res, next) => {
+    const token = req.header('x-litlyx-token');
+    if (token != TOKEN) {
+        res.status(403).json({ error: 'token not valid' });
+        return;
+    }
     console.log(req.path);
     next();
-})
+});
+
 
 app.use('/webhook', webhookRouter);
 app.use('/payment', paymentRouter);
diff --git a/payments/src/routers/PaymentRouter.ts b/payments/src/routers/PaymentRouter.ts
index 1d8fc5f..63bc452 100644
--- a/payments/src/routers/PaymentRouter.ts
+++ b/payments/src/routers/PaymentRouter.ts
@@ -146,4 +146,19 @@ paymentRouter.post('/update_customer_info', json(), async (req, res) => {
     } catch (ex) {
         res.status(500).json({ error: ex.message });
     }
+});
+
+
+export const ZBodyDeleteCustomer = z.object({
+    customer_id: z.string(),
+});
+
+paymentRouter.post('/delete_customer', json(), async (req, res) => {
+    try {
+        const deleteCustomerData = ZBodyDeleteCustomer.parse(req.body);
+        await StripeService.deleteCustomer(deleteCustomerData.customer_id);
+        return sendJson(res, 200, { ok: true });
+    } catch (ex) {
+        res.status(500).json({ error: ex.message });
+    }
 });
\ No newline at end of file
diff --git a/scripts/consumer/deploy.ts b/scripts/consumer/deploy.ts
index 06c17de..4373abe 100644
--- a/scripts/consumer/deploy.ts
+++ b/scripts/consumer/deploy.ts
@@ -38,13 +38,15 @@ async function main() {
         const ecosystemContent = fs.readFileSync(LOCAL_PATH + '/ecosystem.config.js', 'utf8');
         const devContent = ecosystemContent
             .replace("$REDIS_URL$", `${REDIS_URL_TESTMODE}`)
-            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_TESTMODE}`);
+            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_TESTMODE}`)
+            .replace("$DEV_MODE$", `true`);
         archive.append(Buffer.from(devContent), { name: '/ecosystem.config.js' });
     } else {
         const ecosystemContent = fs.readFileSync(LOCAL_PATH + '/ecosystem.config.js', 'utf8');
         const devContent = ecosystemContent
             .replace("$REDIS_URL$", `${REDIS_URL_PRODUCTION}`)
-            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_PRODUCTION}`);
+            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_PRODUCTION}`)
+            .replace("$DEV_MODE$", `false`);
         archive.append(Buffer.from(devContent), { name: '/ecosystem.config.js' });
     }
 
diff --git a/scripts/producer/deploy.ts b/scripts/producer/deploy.ts
index 25a301b..dd223db 100644
--- a/scripts/producer/deploy.ts
+++ b/scripts/producer/deploy.ts
@@ -38,13 +38,15 @@ async function main() {
         const ecosystemContent = fs.readFileSync(LOCAL_PATH + '/ecosystem.config.js', 'utf8');
         const devContent = ecosystemContent
             .replace("$REDIS_URL$", `${REDIS_URL_TESTMODE}`)
-            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_TESTMODE}`);
+            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_TESTMODE}`)
+            .replace("$DEV_MODE$", `true`);
         archive.append(Buffer.from(devContent), { name: '/ecosystem.config.js' });
     } else {
         const ecosystemContent = fs.readFileSync(LOCAL_PATH + '/ecosystem.config.js', 'utf8');
         const devContent = ecosystemContent
             .replace("$REDIS_URL$", `${REDIS_URL_PRODUCTION}`)
-            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_PRODUCTION}`);
+            .replace("$MONGO_CONNECTION_STRING$", `${DATABASE_CONNECTION_STRING_PRODUCTION}`)
+            .replace("$DEV_MODE$", `true`);
         archive.append(Buffer.from(devContent), { name: '/ecosystem.config.js' });
     }