forked from getfider/fider
-
Notifications
You must be signed in to change notification settings - Fork 0
/
csv.go
85 lines (75 loc) · 1.65 KB
/
csv.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package csv
import (
"bytes"
gocsv "encoding/csv"
"strconv"
"strings"
"time"
"github.com/getfider/fider/app/models"
)
//FromIdeas return a byte array of CSV file containing all ideas
func FromIdeas(ideas []*models.Idea) ([]byte, error) {
buffer := &bytes.Buffer{}
writer := gocsv.NewWriter(buffer)
header := []string{
"number",
"title",
"description",
"created_on",
"created_by",
"total_supporters",
"total_comments",
"status",
"responded_by",
"responded_on",
"response",
"original_number",
"original_title",
"tags",
}
if err := writer.Write(header); err != nil {
return nil, err
}
for _, idea := range ideas {
var (
originalNumber string
originalTitle string
respondedBy string
respondedOn string
response string
)
if idea.Response != nil {
respondedBy = idea.Response.User.Name
respondedOn = idea.Response.RespondedOn.Format(time.RFC3339)
response = idea.Response.Text
if idea.Response.Original != nil {
originalNumber = strconv.Itoa(idea.Response.Original.Number)
originalTitle = idea.Response.Original.Title
}
}
record := []string{
strconv.Itoa(idea.Number),
idea.Title,
idea.Description,
idea.CreatedOn.Format(time.RFC3339),
idea.User.Name,
strconv.Itoa(idea.TotalSupporters),
strconv.Itoa(idea.TotalComments),
models.GetIdeaStatusName(idea.Status),
respondedBy,
respondedOn,
response,
originalNumber,
originalTitle,
strings.Join(idea.Tags, ", "),
}
if err := writer.Write(record); err != nil {
return nil, err
}
}
writer.Flush()
if err := writer.Error(); err != nil {
return nil, err
}
return buffer.Bytes(), nil
}