import { convertCsvToWorkbook, createFlatfile } from 'csvtoflatfile'; // or import csvtoflatfile from 'csvtoflatfile';

import { readFile } from "fs/promises";
import { convertCsvToWorkbook, createFlatfile } from 'csvtoflatfile';

const generateFlatfile = async () => {
    const importedCsv: string = await readFile("./yourData.csv", 'utf8');

    // create a workbook
    const { workbook, recordData } = convertCsvToWorkbook({
      workbookName: "Zillow", // the name of your newly created workbook
      csvConfigs: [{
        csv: importedCsv, // your csv file
        hasColumnHeaders: true, // whether or not your csv data includes headers
        sheetName: 'Cool Sheet #1' // the sheet name for your data
      }]
    });


    // optionally, modify/customize your data. Any Flatfile supported record keys are supported.
    // In this case we'll add a warning message when the number of floods exceeds 11.
    const zillowRecords = recordData[0];
    const modifiedRecordData = zillowRecords.map((parsedWeatherData) => {
        return parsedWeatherData.map((weatherData) => {
            // add a warning when floods exceed 11
            const floodsExceedsEleven = (Number(weatherData.value) > 11 && weatherData.header === 'Flood';
            if (floodsExceedsEleven) return {
                ...weatherData, messages: [{
                    type: "warn",
                    message: "That's a lot of flooding!"
                }]
            }
            return { ...weatherData, updatedAt: new Date().toISOString() }; // you can also insert custom record data fields here that will be inserted into the flatile!
        });
    });

    // create the Flatfile
    const flatfile = await createFlatfile({ workbook,
      recordDataList: [modifiedRecordData], flatfileApiKey: process.env?.flatfile ?? ''
    });
    return flatfile; // Flatfile contains the flatfile and record creation response
}

const generateFlatfile = async () => {
    const zillowData = await readFile("./testData/zillow.csv", 'utf8');;
    const nileData = await readFile("./testData/nile.csv", 'utf8');
    const treeData = await readFile("./testData/trees.csv", 'utf8');

    const data = convertCsvToWorkbook({
        workbookName: "Workbook",
        workbookSpaceId: "space1",
        csvConfigs: [
          { csv: zillowData, hasColumnHeaders: true, sheetName: 'Zillow' },
          { csv: nileData, hasColumnHeaders: true, sheetName: 'The Nile' },
          { csv: treeData, hasColumnHeaders: true, sheetName: 'Trees' }
        ]
    });

    const flatfile = await createFlatfile({
      workbook: data.workbook,
      recordDataList: data.recordData,
      flatfileApiKey: process.env?.flatfile ?? ''
    });
    return flatfile;
}