reforger_crawler_main/controllers/feController.go

package controllers

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

	"gitea.tbdevent.eu/TBD/reforger_crawler_main/initializers"
	"gitea.tbdevent.eu/TBD/reforger_crawler_main/models"
	"github.com/gin-gonic/gin"
)

// convertToSimplifiedAddon converts a full Addon to SimplifiedAddon
func convertToSimplifiedAddon(addon models.Addon) models.SimplifiedAddon {
	var simplifiedFiles []models.SimplifiedAddonFile
	for _, file := range addon.AddonFiles {
		simplifiedFiles = append(simplifiedFiles, models.SimplifiedAddonFile{
			Path:    file.Path,
			Hash:    file.Hash,
			Version: file.Version,
			AddonID: file.AddonID,
		})
	}

	return models.SimplifiedAddon{
		ID:                   addon.ID,
		Name:                 addon.Name,
		Type:                 addon.Type,
		Summary:              addon.Summary,
		SubscriberCount:      addon.SubscriberCount,
		CurrentVersionNumber: addon.CurrentVersionNumber,
		Preview:              addon.Preview,
		Author:               addon.Author,
		AddonFiles:           simplifiedFiles,
	}
}

// convertToSimplifiedAddonWithoutFiles converts a full Addon to SimplifiedAddon without files
func convertToSimplifiedAddonWithoutFiles(addon models.Addon) models.SimplifiedAddon {
	return models.SimplifiedAddon{
		ID:                   addon.ID,
		Name:                 addon.Name,
		Type:                 addon.Type,
		Summary:              addon.Summary,
		SubscriberCount:      addon.SubscriberCount,
		CurrentVersionNumber: addon.CurrentVersionNumber,
		Preview:              addon.Preview,
		Author:               addon.Author,
		AddonFiles:           []models.SimplifiedAddonFile{}, // Empty slice
	}
}

func GetDuplicates(c *gin.Context) {
	id := c.Param("id")
	fmt.Println("Fetching addon with ID:", id)

	// Fetch the source addon with its files
	var sourceAddon models.Addon
	if err := initializers.DB.Preload("AddonFiles").First(&sourceAddon, "id = ?", id).Error; err != nil {
		c.JSON(http.StatusNotFound, gin.H{"error": "Addon not found"})
		return
	}

	// Convert to simplified addon
	simplifiedSourceAddon := convertToSimplifiedAddon(sourceAddon)

	// If the addon has no files, return early
	if len(sourceAddon.AddonFiles) == 0 {
		c.JSON(http.StatusOK, models.DuplicatesResponse{
			SourceAddon:     simplifiedSourceAddon,
			DuplicateAddons: make(map[string]models.AddonWithDuplicates),
		})
		return
	}

	// Get all whitelisted hashes
	var whitelistedHashes []string
	if err := initializers.DB.Model(&models.WhitelistedHash{}).Pluck("hash", &whitelistedHashes).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch whitelisted hashes"})
		return
	}

	// Create a map for faster lookup
	whitelistMap := make(map[string]bool)
	for _, hash := range whitelistedHashes {
		whitelistMap[hash] = true
	}

	// Extract all hashes from source addon files, excluding whitelisted ones
	var sourceHashes []string
	for _, file := range sourceAddon.AddonFiles {
		if file.Hash != "" && !whitelistMap[file.Hash] {
			sourceHashes = append(sourceHashes, file.Hash)
		}
	}

	if len(sourceHashes) == 0 {
		c.JSON(http.StatusOK, models.DuplicatesResponse{
			SourceAddon:     simplifiedSourceAddon,
			DuplicateAddons: make(map[string]models.AddonWithDuplicates),
		})
		return
	}

	// Find all addon files that have matching hashes but belong to different addons
	var duplicateFiles []models.AddonFile
	if err := initializers.DB.Where("hash IN ? AND addon_id != ?", sourceHashes, id).Find(&duplicateFiles).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch duplicate files"})
		return
	}

	// Group duplicate files by addon ID
	duplicatesByAddonID := make(map[string][]models.AddonFile)
	for _, file := range duplicateFiles {
		duplicatesByAddonID[file.AddonID] = append(duplicatesByAddonID[file.AddonID], file)
	}

	// Fetch addon information for each addon that has duplicates
	var addonIDs []string
	for addonID := range duplicatesByAddonID {
		addonIDs = append(addonIDs, addonID)
	}

	var duplicateAddons []models.Addon
	if len(addonIDs) > 0 {
		if err := initializers.DB.Where("id IN ?", addonIDs).Find(&duplicateAddons).Error; err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch duplicate addons"})
			return
		}
	}

	// Set Recheck flag for source addon and all duplicate addons
	if err := initializers.DB.Model(&models.Addon{}).Where("id = ?", id).Update("recheck", true).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update source addon"})
		return
	}

	for _, addon := range duplicateAddons {
		if err := initializers.DB.Model(&models.Addon{}).Where("id = ?", addon.ID).Update("recheck", true).Error; err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update duplicate addon"})
			return
		}
	}

	// Build the response
	response := models.DuplicatesResponse{
		SourceAddon:     simplifiedSourceAddon,
		DuplicateAddons: make(map[string]models.AddonWithDuplicates),
	}

	// Create a map for easy addon lookup
	addonMap := make(map[string]models.Addon)
	for _, addon := range duplicateAddons {
		addonMap[addon.ID] = addon
	}

	// Build the duplicate addons response
	for addonID, files := range duplicatesByAddonID {
		var duplicateFileInfos []models.DuplicateFileInfo
		for _, file := range files {
			duplicateFileInfos = append(duplicateFileInfos, models.DuplicateFileInfo{
				Path:    file.Path,
				Hash:    file.Hash,
				Version: file.Version,
				AddonID: file.AddonID,
			})
		}

		if addon, exists := addonMap[addonID]; exists {
			response.DuplicateAddons[addonID] = models.AddonWithDuplicates{
				Addon:      convertToSimplifiedAddonWithoutFiles(addon),
				Duplicates: duplicateFileInfos,
			}
		}
	}

	c.JSON(http.StatusOK, response)
}

func GetPossibleAddons(c *gin.Context) {
	query := c.Query("q")
	if query == "" {
		c.JSON(http.StatusBadRequest, gin.H{"error": "Query parameter 'q' is required"})
		return
	}

	// Set default limit
	limit := 20
	if limitParam := c.Query("limit"); limitParam != "" {
		if parsedLimit, err := strconv.Atoi(limitParam); err == nil && parsedLimit > 0 && parsedLimit <= 100 {
			limit = parsedLimit
		}
	}

	fmt.Printf("Searching for addons with query: %s (limit: %d)\n", query, limit)

	var addons []models.Addon

	// Search by ID (exact match) or by name (case-insensitive partial match)
	// Using OR condition to search both ID and name fields
	dbQuery := initializers.DB.Where("id = ? OR LOWER(name) LIKE LOWER(?)", query, "%"+query+"%").
		Order("subscriber_count DESC"). // Order by popularity
		Limit(limit)

	if err := dbQuery.Find(&addons).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to search addons"})
		return
	}

	// Convert to search results
	var results []models.AddonSearchResult
	for _, addon := range addons {
		results = append(results, models.AddonSearchResult{
			ID:                   addon.ID,
			Name:                 addon.Name,
			Type:                 addon.Type,
			Summary:              addon.Summary,
			Preview:              addon.Preview,
			SubscriberCount:      addon.SubscriberCount,
			CurrentVersionNumber: addon.CurrentVersionNumber,
			Author:               addon.Author,
		})
	}

	response := models.SearchResponse{
		Query:   query,
		Results: results,
		Total:   len(results),
	}

	c.JSON(http.StatusOK, response)
}