Skip to content

Commit

Permalink
refactor(csv): add small changes to the csv file
Browse files Browse the repository at this point in the history
  • Loading branch information
danvergara committed Sep 6, 2024
1 parent e1cadb0 commit ed4860d
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions pkg/files/documents/csv.go
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,9 @@ func (c *Csv) ConvertTo(fileType, subType string, file io.Reader) (io.Reader, er

reader := csv.NewReader(file)
xlsxFile := xlsx.NewFile()
sheet, err := xlsxFile.AddSheet(strings.TrimSuffix(c.filename, filepath.Ext(c.filename)))
sheet, err := xlsxFile.AddSheet(
strings.TrimSuffix(c.filename, filepath.Ext(c.filename)),
)
if err != nil {
return nil, fmt.Errorf("error creating a xlsx sheet %w", err)
}
Expand All @@ -100,7 +102,12 @@ func (c *Csv) ConvertTo(fileType, subType string, file io.Reader) (io.Reader, er
}
}

xlsxFile.Save(xlsxPath)
if err := xlsxFile.Save(xlsxPath); err != nil {
return nil, fmt.Errorf(
"error at saving the temporary csv file: %w",
err,
)
}

tmpCsvFile, err := os.Open(xlsxPath)
if err != nil {
Expand All @@ -109,7 +116,7 @@ func (c *Csv) ConvertTo(fileType, subType string, file io.Reader) (io.Reader, er
err,
)
}
defer tmpCsvFile.Close()
defer os.Remove(tmpCsvFile.Name())

// Creates the zip file that will be returned.
archive, err := os.Create(zipFileName)
Expand All @@ -119,6 +126,7 @@ func (c *Csv) ConvertTo(fileType, subType string, file io.Reader) (io.Reader, er
err,
)
}
defer os.Remove(archive.Name())

// Creates a Zip Writer to add files later on.
zipWriter := zip.NewWriter(archive)
Expand Down

0 comments on commit ed4860d

Please sign in to comment.