24 hr availability fix (#595)

* Update productfiles.go

Fixes #475

Author: oskarhurst

api/models/productfiles.go

@@ -45,29 +45,47 @@ func GetProductFileAvailability(db *pgxpool.Pool, ID uuid.UUID, interval string,
 	avail := make([]ProductfileAvailability, 0)
 	startTime := time.Date(d.Year(), d.Month(), d.Day(), 0, 0, 0, 0, d.Location()).Format(time.RFC3339)
 	endTime := time.Date(d.Year(), d.Month(), d.Day(), 23, 0, 0, 0, d.Location()).Format(time.RFC3339)
-	if err := pgxscan.Select(context.Background(), db, &avail,
-		`WITH hours AS (
-				SELECT generate_series(
-						$1::timestamp,  -- Start of the interval
-						$2::timestamp,  -- End of the interval
-						$3::interval
-				) AS hour
-		)
-		SELECT 
-				h.hour as datetime,
-				CASE 
-						WHEN pf.datetime IS NOT NULL THEN TRUE
-						ELSE FALSE
-				END AS is_available
-		FROM 
-				hours h
-		LEFT JOIN (
-			select * from productfile where product_id = $4 and datetime >= $5 and datetime <= $6
-		) pf ON date_trunc('hour', pf.datetime) = h.hour
-		ORDER BY 
-				h.hour`, startTime, endTime, interval, ID, startTime, endTime,
-	); err != nil {
-		return make([]ProductfileAvailability, 0), err
+
+	if interval == "24 Hour" {
+		if err := pgxscan.Select(context.Background(), db, &avail,
+			`SELECT datetime, TRUE as is_available
+			FROM productfile
+			WHERE product_id = $1 AND datetime >= $2 AND datetime <= $3
+			ORDER BY datetime`, ID, startTime, endTime,
+		); err != nil {
+			return make([]ProductfileAvailability, 0), err
+		}
+		if len(avail) == 0 {
+			avail = append(avail, ProductfileAvailability{
+				Datetime:    time.Date(d.Year(), d.Month(), d.Day(), 0, 0, 0, 0, d.Location()),
+				IsAvailable: false,
+			})
+		}
+	} else {
+		if err := pgxscan.Select(context.Background(), db, &avail,
+			`WITH hours AS (
+					SELECT generate_series(
+							$1::timestamp,  -- Start of the interval
+							$2::timestamp,  -- End of the interval
+							$3::interval
+					) AS hour
+			)
+			SELECT
+					h.hour as datetime,
+					CASE
+							WHEN pf.datetime IS NOT NULL THEN TRUE
+							ELSE FALSE
+					END AS is_available
+			FROM
+					hours h
+			LEFT JOIN (
+				select * from productfile where product_id = $4 and datetime >= $5 and datetime <= $6
+			) pf ON date_trunc('hour', pf.datetime) = h.hour
+			ORDER BY
+					h.hour`, startTime, endTime, interval, ID, startTime, endTime,
+		); err != nil {
+			return make([]ProductfileAvailability, 0), err
+		}
 	}
 	return avail, nil
 }