generated from padok-team/yatas-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
categoriesBlock.go
124 lines (115 loc) · 2.87 KB
/
categoriesBlock.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main
import (
"strconv"
"github.com/jomei/notionapi"
"github.com/stangirard/yatas/plugins/commons"
)
func CalculatePercent(success int, failure int) string {
total := success + failure
if total == 0 {
return "0"
}
return strconv.Itoa((success * 100) / total)
}
func contains(s []string, e string) bool {
for _, a := range s {
if a == e {
return true
}
}
return false
}
func createCategoriesBlock(test commons.Tests) notionapi.Block {
// Find the categories
categories := []string{}
categoriesSuccess := map[string]int{}
categoriesFailure := map[string]int{}
for _, check := range test.Checks {
for _, category := range check.Categories {
if !contains(categories, category) {
categories = append(categories, category)
categoriesSuccess[category] = 0
categoriesFailure[category] = 0
}
if check.Status == "OK" {
categoriesSuccess[category]++
} else {
categoriesFailure[category]++
}
}
}
// Calculate the completion scores
scores := []string{}
for _, category := range categories {
scores = append(scores, CalculatePercent(categoriesSuccess[category], categoriesFailure[category])+" %")
}
// Write the categories
block := createCategorieTable(categories, scores)
return block
}
func createCategorieTable(categories, scores []string) notionapi.Block {
table := notionapi.TableBlock{
BasicBlock: notionapi.BasicBlock{
Type: notionapi.BlockType("table"),
Object: notionapi.ObjectType("block"),
},
Table: notionapi.Table{
TableWidth: 2,
HasColumnHeader: true,
Children: createCategorieTableRows(categories, scores),
},
}
return table
}
func createCategorieTableRows(categories, scores []string) []notionapi.Block {
rows := []notionapi.Block{}
rows = append(rows, createCategorieTableHeader())
for key, category := range categories {
score := scores[key]
r := createCategorieTableRow(category, score)
rows = append(rows, r)
}
return rows
}
func createCategorieTableHeader() notionapi.TableRowBlock {
category := notionapi.Text{
Content: "Category",
}
completion := notionapi.Text{
Content: "Completion",
}
row := notionapi.TableRowBlock{
BasicBlock: notionapi.BasicBlock{
Type: notionapi.BlockType("table_row"),
Object: notionapi.ObjectType("block"),
},
TableRow: notionapi.TableRow{
Cells: [][]notionapi.RichText{
{{Text: &category}},
{{Text: &completion}},
},
},
}
return row
}
func createCategorieTableRow(category, score string) notionapi.TableRowBlock {
cat := notionapi.Text{
Content: category,
}
completion := notionapi.Text{
Content: score,
}
row := notionapi.TableRowBlock{
BasicBlock: notionapi.BasicBlock{
Type: notionapi.BlockType("table_row"),
Object: notionapi.ObjectType("block"),
},
TableRow: notionapi.TableRow{
Cells: [][]notionapi.RichText{
{{Text: &cat}},
{{Text: &completion}},
},
},
}
return row
}