1 files changed, 12 insertions, 1 deletions
diff --git a/pkg/html/service.go b/pkg/html/service.go
index 8a7a163..a39c90d 100644
--- a/pkg/html/service.go
+++ b/pkg/html/service.go
@@ -20,6 +20,7 @@ func (s *service) GenerateHtml(entries []sqlc.GetEntriesRow) (string, error) {
 	var str string
 
 	var currentDate time.Time
+	var group bool
 
 	for _, entry := range entries {
 		date, err := time.Parse(time.DateTime, entry.Timestamp)
@@ -28,7 +29,13 @@ func (s *service) GenerateHtml(entries []sqlc.GetEntriesRow) (string, error) {
 		}
 
 		if currentDate.Year() != date.Year() || currentDate.Month() != date.Month() {
-			str = str + "<h2>" + date.Format("January 2006") + "</h2>"
+			if group {
+				str += "</div>"
+			}
+			str += "<h2>" + date.Format("January 2006") + "</h2>"
+
+			str += "<div class=\"entry-group\">"
+			group = true
 		}
 
 		currentDate = date
@@ -49,5 +56,9 @@ func (s *service) GenerateHtml(entries []sqlc.GetEntriesRow) (string, error) {
 		str += "</span>"
 	}
 
+	if group {
+		str += "</div>"
+	}
+
 	return str, nil
 }