N-FIN-12: add button to generate sample data

2025-07-04 20:19:17 +00:00 · 2024-03-11 03:38:30 +01:00 · 2024-03-11 03:38:30 +01:00 · b8a7a5d6f4
commit b8a7a5d6f4
parent 95f36977da
3 changed files with 223 additions and 1 deletions
--- a/src/lib/actions/generateSampleData.ts
+++ b/src/lib/actions/generateSampleData.ts
@ -0,0 +1,190 @@
+import { prismaClient } from '@/prisma';
+import type { Category, Entity } from '@prisma/client';
+import { EntityType } from '@prisma/client';
+import { getUser } from '@/auth';
+import { URL_SIGN_IN } from '@/lib/constants';
+import { ActionResponse } from '@/lib/types/ActionResponse';
+
+export default async function generateSampleData(): Promise<ActionResponse> {
+    'use server';
+
+    const user = await getUser();
+
+    if (!user) {
+        return {
+            type: 'error',
+            message: 'You must be logged in to create/update an category.',
+            redirect: URL_SIGN_IN,
+        };
+    }
+
+    // Categories: create sample data
+    const categories: Category[] = await prismaClient.category.findMany({where: {userId: user.id}});
+    if (await prismaClient.category.count({where: {userId: user.id}}) == 0) {
+
+        console.log('Creating sample categories...');
+
+        categories.push(await prismaClient.category.create({
+            data: {
+                userId: user.id,
+                name: 'Groceries',
+                color: '#FFBEAC',
+            },
+        }));
+
+        categories.push(await prismaClient.category.create({
+            data: {
+                userId: user.id,
+                name: 'Drugstore items',
+                color: '#9CBCFF',
+            },
+        }));
+
+        categories.push(await prismaClient.category.create({
+            data: {
+                userId: user.id,
+                name: 'Going out',
+                color: '#F1ADFF',
+            },
+        }));
+
+        categories.push(await prismaClient.category.create({
+            data: {
+                userId: user.id,
+                name: 'Random stuff',
+                color: '#C1FFA9',
+            },
+        }));
+
+        categories.push(await prismaClient.category.create({
+            data: {
+                userId: user.id,
+                name: 'Salary',
+                color: '#FFF787',
+            },
+        }));
+
+        console.log('Sample categories created.');
+    }
+    console.log(categories);
+
+    // Entities: create sample data
+    const entities: Entity[] = await prismaClient.entity.findMany({where: {userId: user.id}});
+    if (await prismaClient.entity.count({where: {userId: user.id}}) == 0) {
+
+        console.log('Creating sample entities...');
+
+        entities.push(await prismaClient.entity.create({
+            data: {
+                userId: user.id,
+                name: 'Main Account',
+                type: EntityType.Account,
+            },
+        }));
+
+        entities.push(await prismaClient.entity.create({
+            data: {
+                userId: user.id,
+                name: 'Company',
+                type: EntityType.Entity,
+            },
+        }));
+
+        entities.push(await prismaClient.entity.create({
+            data: {
+                userId: user.id,
+                name: 'Supermarket 1',
+                type: EntityType.Entity,
+            },
+        }));
+
+        entities.push(await prismaClient.entity.create({
+            data: {
+                userId: user.id,
+                name: 'Supermarket 2',
+                type: EntityType.Entity,
+            },
+        }));
+
+        entities.push(await prismaClient.entity.create({
+            data: {
+                userId: user.id,
+                name: 'Supermarket 3',
+                type: EntityType.Entity,
+            },
+        }));
+
+        entities.push(await prismaClient.entity.create({
+            data: {
+                userId: user.id,
+                name: 'Supermarket 4',
+                type: EntityType.Entity,
+            },
+        }));
+
+        console.log('Sample entities created.');
+    }
+    console.log(entities);
+
+    // Payments: create sample data
+    console.log('Creating sample payments...');
+
+    if (await prismaClient.payment.count({where: {userId: user.id}}) == 0) {
+        for (let i = 0; i < 4; i++) {
+
+            const date = new Date();
+            date.setDate(1);
+            date.setMonth(date.getMonth() - i);
+
+            await prismaClient.payment.create({
+                data: {
+                    userId: user.id,
+                    amount: 200000,
+                    date: date,
+                    payorId: entities[1].id,
+                    payeeId: entities[0].id,
+                    categoryId: 5,
+                    createdAt: date,
+                    updatedAt: date,
+                },
+            });
+        }
+    }
+
+    let minAmount = 200; // 2€
+    let maxAmount = 3000; // 30€
+    let minPayee = entities[2].id;
+    let maxPayee = entities[entities.length - 1].id;
+    let minCategory = categories[0].id;
+    let maxCategory = categories[categories.length - 1].id;
+    let payments = 196;
+
+    for (let i = 0; i < payments; i++) {
+
+        const date = new Date(
+            new Date().getTime() - Math.floor(Math.random() * 10000000000));
+
+        await prismaClient.payment.create({
+            data: {
+                userId: user.id,
+                amount: Math.floor(
+                    Math.random() * (maxAmount - minAmount) + minAmount),
+                date: date,
+                payorId: 1,
+                payeeId: Math.floor(
+                    Math.random() * (maxPayee - minPayee) + minPayee),
+                categoryId: Math.floor(
+                    Math.random() * (maxCategory - minCategory) + minCategory),
+                createdAt: date,
+                updatedAt: date,
+            },
+        });
+    }
+
+    console.log('Sample payments created.');
+
+    return {
+        type: 'success',
+        message: 'Sample data created',
+    };
+}