Typescript CSV Generator Class

csvGenerator.ts

interface ICsvRow {
  items: Array<string>
}

interface ICsvData {
  rows: Array<ICsvRow>
}

class CsvGenerator {
  public headings: Array<string> = []
  public data: ICsvData = { rows: [] }
  public fileName?: string
  public fileDate?: boolean = true

  private finalCsvData = ''

  constructor (headings: Array<string>, csvData: ICsvData, fileName?: string, includeDate?: boolean) {
    this.headings = headings
    this.data = csvData
    this.fileName = fileName ?? ''
    this.fileDate = includeDate ?? true

    this.generateCsv()
  }

  private generateDatedFileName (): string {
    const date = new Date(Date.now())
    const dateString = date.getDate() + '/' + date.getMonth() + '/' + date.getFullYear()
    return dateString
  }

  private generateCsv (): void {
    const csvMeta = 'data:text/csv;charset=utf-8,'

    const csvHeadings = this.headings.join(',') + '\n'

    let csvRows = ''
    this.data.rows.map((row) => {
      csvRows += row.items.join(',') + '\n'
    })

    if (this.fileDate) {
      this.fileName = this.fileName + '_' + this.generateDatedFileName()
    }

    this.finalCsvData = csvMeta + csvHeadings + csvRows
  }

  downloadCsv (): void {
    const encodedUri = encodeURI(this.finalCsvData)
    const link = document.createElement('a')

    link.setAttribute('href', encodedUri)
    link.setAttribute('download', this.fileName + '.csv')
    document.body.appendChild(link)

    link.click()
  }
}

export { ICsvRow, ICsvData, CsvGenerator }