From 1f3975027d26e055ed537e363108938107d689b3 Mon Sep 17 00:00:00 2001
From: Aritro Saha <29025984+AritroSaha10@users.noreply.github.com>
Date: Wed, 27 Dec 2023 16:18:17 -0500
Subject: [PATCH] feat(backend): closes #69

---
 .../cmd/util/import_tickets_from_csv/main.go  | 29 +++++++++++++++++--
 backend/middleware/sentry.go                  |  2 +-
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/backend/cmd/util/import_tickets_from_csv/main.go b/backend/cmd/util/import_tickets_from_csv/main.go
index ef59fc0..f272fc9 100644
--- a/backend/cmd/util/import_tickets_from_csv/main.go
+++ b/backend/cmd/util/import_tickets_from_csv/main.go
@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"strconv"
 	"strings"
 	"sync"
 	"sync/atomic"
@@ -100,10 +101,20 @@ func main() {
 	}
 	studentNumberColNum-- // Move to 0-based index
 
+	var localMaxScanCountColNum int // -1 -> no given column number
+	fmt.Printf("What column # stores the maximum scan count for a ticket? [1-%d, 0 for none]? ", csvReader.FieldsPerRecord)
+	if n, err := fmt.Scanf("%d", &localMaxScanCountColNum); err != nil || n != 1 {
+		log.Fatal().Err(err).Msg("could not parse local max scan count")
+	}
+	if localMaxScanCountColNum < 0 || localMaxScanCountColNum > int(csvReader.FieldsPerRecord) {
+		log.Fatal().Msg("column # given is outside accepted range")
+	}
+	localMaxScanCountColNum-- // Move to 0-based index
+
 	var globalMaxScanCount int
-	fmt.Printf("What should the maximum scan count be per ticket [>=0, 0 for infinite]? ")
+	fmt.Printf("What should the default maximum scan count be per ticket [>=0, 0 for infinite]? ")
 	if n, err := fmt.Scanf("%d", &globalMaxScanCount); err != nil || n != 1 {
-		log.Fatal().Err(err).Msg("could not parse max scan count")
+		log.Fatal().Err(err).Msg("could not parse global max scan count")
 	}
 	if globalMaxScanCount < 0 {
 		log.Fatal().Msg("global max scan count below 0")
@@ -133,6 +144,18 @@ func main() {
 			rawFullName := rec[studentNameColNum]
 			studentNumber := rec[studentNumberColNum]
 
+			maxScanCount := globalMaxScanCount
+			// Try parsing the local max scan count from DB if column
+			// number exists, otherwise defaulting to global
+			if localMaxScanCountColNum != -1 {
+				tmpScanCount, err := strconv.Atoi(rec[localMaxScanCountColNum])
+				if err != nil {
+					log.Warn().Err(err).Str("rawLocalMaxScanCount", rec[localMaxScanCountColNum]).Msg("could not parse local max scan count")
+				} else {
+					maxScanCount = tmpScanCount
+				}
+			}
+
 			nameSplit := strings.Split(rawFullName, ", ")
 			if len(nameSplit) != 2 {
 				err := fmt.Errorf("more than one comma in name: %s", rawFullName)
@@ -146,7 +169,7 @@ func main() {
 			queuedTicket := models.QueuedTicket{
 				StudentNumber:  studentNumber,
 				EventID:        eventID,
-				MaxScanCount:   globalMaxScanCount,
+				MaxScanCount:   maxScanCount,
 				FullNameUpdate: fullName,
 			}
 
diff --git a/backend/middleware/sentry.go b/backend/middleware/sentry.go
index dd20eff..3c343fb 100644
--- a/backend/middleware/sentry.go
+++ b/backend/middleware/sentry.go
@@ -19,7 +19,7 @@ func CreateNewSentryMiddleware() *sentryhttp.Handler {
 		log.Error().Err(err).Msg("Sentry initialization failed")
 	}
 	defer sentry.Flush(time.Second)
-	log.Debug().Msg("Sentry initialized")
+	log.Debug().Msg("sentry initialized")
 
 	return sentryhttp.New(sentryhttp.Options{
 		Repanic: true,