litlyx/dashboard/server/api/project/generate_csv.ts

import { UserModel } from "@schema/UserSchema";
import { VisitModel } from "@schema/metrics/VisitSchema";

import { google } from 'googleapis';

const { GOOGLE_AUTH_CLIENT_SECRET, GOOGLE_AUTH_CLIENT_ID } = useRuntimeConfig()

async function exportToGoogle(data: string, user_id: string) {

    const user = await UserModel.findOne({ _id: user_id }, { google_tokens: true });

    const authClient = new google.auth.OAuth2({
        clientId: GOOGLE_AUTH_CLIENT_ID,
        clientSecret: GOOGLE_AUTH_CLIENT_SECRET
    })

    authClient.setCredentials({ access_token: user?.google_tokens?.access_token, refresh_token: user?.google_tokens?.refresh_token });

    const sheets = google.sheets({ version: 'v4', auth: authClient });

    try {
        const createSheetResponse = await sheets.spreadsheets.create({
            requestBody: {
                properties: {
                    title: 'Text export'
                }
            }
        });

        const spreadsheetId = createSheetResponse.data.spreadsheetId;

        await sheets.spreadsheets.values.update({
            spreadsheetId: spreadsheetId as string,
            range: 'Sheet1!A1',
            requestBody: {
                values: data.split('\n').map(e => {
                    return e.split(',')
                })
            }
        });

        return { ok: true }

    } catch (error: any) {

        console.error('Error creating Google Sheet from CSV:', error);

        if (error.response && error.response.status === 401) {
            return { error: 'Auth error, try to logout and login again' }
        }

        return { error: error.message.toString() }

    }
}

const { SELFHOSTED } = useRuntimeConfig();

export default defineEventHandler(async event => {

    const data = await getRequestDataOld(event, { requireSchema: false });
    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 { mode, slice } = getQuery(event);

    let timeSub = 1000 * 60 * 60 * 24;

    if (slice == '0') {
        timeSub = 1000 * 60 * 60 * 24
    } else if (slice == '1') {
        timeSub = 1000 * 60 * 60 * 24 * 7
    } else if (slice == '2') {
        timeSub = 1000 * 60 * 60 * 24 * 7 * 30
    } else if (slice == '3') {
        timeSub = 1000 * 60 * 60 * 24 * 7 * 30 * 12 * 2
    }

    if (mode === 'visits') {

        const visistsReportData = await VisitModel.find({
            project_id,
            created_at: {
                $gt: Date.now() - timeSub
            }
        });

        const csvHeader = [
            "browser",
            "os",
            "continent",
            "country",
            "device",
            "website",
            "page",
            "referrer",
            "created_at",
        ];


        const lines: any[] = [];
        visistsReportData.forEach(line => lines.push(line.toJSON()));

        const result = csvHeader.join(',') + '\n' + lines.map(element => {
            const content: string[] = [];
            for (const key of csvHeader) {
                content.push(element[key]);
            }
            return content.join(',');
        }).join('\n');



        const isGoogle = getHeader(event, 'x-google-export');

        if (isGoogle === 'true') {
            const data = await exportToGoogle(result, user.id);
            return data;
        }

        return result;
    } else {
        return '';
    }


});