flightlesssomething/benchmarks.go

package flightlesssomething

import (
	"fmt"
	"net/http"
	"strconv"
	"strings"
	"sync"

	"github.com/gin-contrib/sessions"
	"github.com/gin-gonic/gin"
)

const BENCHMARKS_PER_PAGE = 10

func getBenchmarks(c *gin.Context) {
	session := sessions.Default(c)

	// Get "query" value
	query := c.Query("query")

	// Get "page" value
	page := c.DefaultQuery("page", "1")
	pageInt, _ := strconv.Atoi(page)
	if pageInt < 1 {
		pageInt = 1
	}

	// Get benchmarks according to query
	var benchmarks []Benchmark
	tx := db.
		Preload("User").
		Order("created_at DESC").
		Offset((pageInt - 1) * BENCHMARKS_PER_PAGE).
		Limit(BENCHMARKS_PER_PAGE)
	if query != "" {
		tx = tx.Where("title LIKE ?", "%"+query+"%").Or("description LIKE ?", "%"+query+"%")
	}
	result := tx.Find(&benchmarks)
	if result.Error != nil {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Error occurred while fetching benchmarks",
		})
		return
	}

	// Get total number of benchmarks matching the query
	var benchmarksTotal int64
	tx = db.Model(&Benchmark{})
	if query != "" {
		tx = tx.Where("title LIKE ?", "%"+query+"%").Or("description LIKE ?", "%"+query+"%")
	}
	result = tx.Count(&benchmarksTotal)
	if result.Error != nil {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Error occurred while counting benchmarks",
		})
		return
	}

	// Calculate pagination values
	prevPage := pageInt - 1
	nextPage := pageInt + 1
	totalPages := (int(benchmarksTotal) + BENCHMARKS_PER_PAGE - 1) / BENCHMARKS_PER_PAGE

	c.HTML(http.StatusOK, "benchmarks.tmpl", gin.H{
		"activePage": "benchmarks",
		"username":   session.Get("Username"),
		"userID":     session.Get("ID"),

		"benchmarks":      benchmarks,
		"benchmarksTotal": benchmarksTotal,

		// Query parameters
		"query": query,
		"page":  pageInt,

		// Pagination values
		"prevPage":   prevPage,
		"nextPage":   nextPage,
		"totalPages": totalPages,
	})
}

func getBenchmarkCreate(c *gin.Context) {
	session := sessions.Default(c)
	if session.Get("Username") == "" {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Please authenticate to create a benchmark",
		})
	}

	c.HTML(http.StatusOK, "benchmark_create.tmpl", gin.H{
		"activePage": "benchmark",
		"username":   session.Get("Username"),
		"userID":     session.Get("ID"),
	})
}

func postBenchmarkCreate(c *gin.Context) {
	session := sessions.Default(c)
	if session.Get("Username") == "" {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Please authenticate to create a benchmark",
		})
		return
	}

	title := strings.TrimSpace(c.PostForm("title"))
	if len(title) > 100 || title == "" {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Title must not be empty or exceed 100 characters",
		})
		return
	}

	description := strings.TrimSpace(c.PostForm("description"))
	if len(description) > 500 {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Description must not exceed 500 characters",
		})
		return
	}

	form, err := c.MultipartForm()
	if err != nil {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Error occurred while parsing form data",
		})
		return
	}
	files := form.File["files"]
	if len(files) == 0 {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "No files uploaded",
		})
		return
	}
	if len(files) > 50 {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Too many files uploaded (max 50)",
		})
		return
	}

	// Read CSV files
	// Store to disk only when DB record is created successfully
	csvFiles, err := readBenchmarkFiles(files)
	if err != nil {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Error occurred while reading files: " + err.Error(),
		})
		return
	}

	benchmark := Benchmark{
		UserID:      session.Get("ID").(uint),
		Title:       title,
		Description: description,
	}

	result := db.Create(&benchmark)
	if result.Error != nil {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Error occurred while creating benchmark: " + result.Error.Error(),
		})
		return
	}

	// Store CSV files to disk
	err = storeBenchmarkData(csvFiles, benchmark.ID)
	if err != nil {
		db.Unscoped().Delete(&benchmark) // Hard delete from DB
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Error occurred while storing benchmark data: " + err.Error(),
		})
		return
	}

	if openaiClient != nil || anthroClient != nil {
		go generateSummary(&benchmark, csvFiles)
	}

	// Redirect to the newly created benchmark using GET request
	c.Redirect(http.StatusSeeOther, fmt.Sprintf("/benchmark/%d", benchmark.ID))
}

func deleteBenchmark(c *gin.Context) {
	session := sessions.Default(c)
	if session.Get("Username") == "" {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Please authenticate to create a benchmark",
		})
		return
	}

	// Get benchmark ID from the path
	id := c.Param("id")

	// Check if user owns the benchmark
	var benchmark Benchmark
	result := db.First(&benchmark, id)
	if result.Error != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Internal server error occurred: " + result.Error.Error(),
		})
		return
	}
	if benchmark.UserID != session.Get("ID") {
		c.HTML(http.StatusUnauthorized, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "You do not own this benchmark",
		})
		return
	}

	// Delete benchmark from DB
	result = db.Unscoped().Delete(&benchmark)
	if result.Error != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Internal server error occurred: " + result.Error.Error(),
		})
		return
	}

	// Delete benchmark data from disk
	err := deleteBenchmarkData(benchmark.ID)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Internal server error occurred: " + err.Error(),
		})
		return
	}

	// Redirect to the benchmarks page
	c.Header("HX-Redirect", "/benchmarks")
	c.JSON(http.StatusOK, gin.H{
		"message": "Benchmark deleted successfully",
	})
}

func getBenchmark(c *gin.Context) {
	session := sessions.Default(c)

	// Get benchmark ID from the path
	id := c.Param("id")

	// Get benchmark details
	intID, err := strconv.Atoi(id)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Internal server error occurred: " + err.Error(),
		})
		return
	}

	var benchmark Benchmark
	benchmark.ID = uint(intID)

	var benchmarkDatas []*BenchmarkData
	var errCSV, errDB error
	errHTTPStatus := http.StatusInternalServerError

	wg := sync.WaitGroup{}
	wg.Add(2)

	go func() {
		defer wg.Done()
		benchmarkDatas, errCSV = retrieveBenchmarkData(benchmark.ID)
	}()

	go func() {
		defer wg.Done()
		result := db.Preload("User").First(&benchmark, id)
		if result.Error != nil {
			errDB = result.Error
			return
		}
		if result.RowsAffected == 0 {
			errDB = fmt.Errorf("Benchmark not found")
			errHTTPStatus = http.StatusNotFound
			return
		}
	}()

	wg.Wait()

	err = errDB
	if err == nil {
		err = errCSV
	}
	if err != nil {
		c.HTML(errHTTPStatus, "error.tmpl", gin.H{
			"activePage":   "error",
			"username":     session.Get("Username"),
			"userID":       session.Get("ID"),
			"errorMessage": "Error occurred: " + errDB.Error(),
		})
		return
	}

	if benchmark.AiSummary == "" && openaiClient != nil {
		go generateSummary(&benchmark, benchmarkDatas)
		benchmark.AiSummary = "AI summary is being generated... Refresh the page later."
	}

	c.HTML(http.StatusOK, "benchmark.tmpl", gin.H{
		"activePage": "benchmark",
		"username":   session.Get("Username"),
		"userID":     session.Get("ID"),

		"benchmark":     benchmark,
		"benchmarkData": benchmarkDatas,
	})
}

func getBenchmarkDownload(c *gin.Context) {
	session := sessions.Default(c)

	// Get benchmark ID from the path
	id := c.Param("id")

	// Get benchmark details
	intID, err := strconv.Atoi(id)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage": "error",
			"username":   session.Get("Username"),
			"userID":     session.Get("ID"),

			"errorMessage": "Internal server error occurred: " + err.Error(),
		})
		return
	}

	var benchmark Benchmark
	benchmark.ID = uint(intID)

	benchmarkDatas, err := retrieveBenchmarkData(benchmark.ID)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage":   "error",
			"username":     session.Get("Username"),
			"userID":       session.Get("ID"),
			"errorMessage": "Error occurred: " + err.Error(),
		})
		return
	}

	content, err := createZipFromBenchmarkData(benchmarkDatas)
	if err != nil {
		c.HTML(http.StatusInternalServerError, "error.tmpl", gin.H{
			"activePage":   "error",
			"username":     session.Get("Username"),
			"userID":       session.Get("ID"),
			"errorMessage": "Error occurred: " + err.Error(),
		})
		return
	}

	fileName := "benchmark_" + id + ".zip"
	c.Header("Content-Type", "application/zip")
	c.Header("Content-Disposition", "attachment; filename="+fileName)
	c.Data(http.StatusOK, "application/zip", content.Bytes())
}