package utils

import (
	"bytes"
	"github.com/tealeg/xlsx"
)

func CreateExcel(headers []string, data map[int]string) (*bytes.Buffer, error) {
	file := xlsx.NewFile()
	sheet, err := file.AddSheet("sheet1")
	if err != nil {
		return nil, err
	}

	headerRow := sheet.AddRow()
	for _, header := range headers {
		cell := headerRow.AddCell()
		cell.Value = header
	}

	dataRow := sheet.AddRow()
	for i := 0; i < len(headers); i++ {
		cell := dataRow.AddCell()
		cell.Value = data[i]
	}

	buffer := new(bytes.Buffer)
	if err := file.Write(buffer); err != nil {
		return nil, err
	}

	return buffer, nil
}