nginx-ui/api/nginx_log/log_list.go

package nginx_log

import (
	"net/http"
	"strings"

	"github.com/0xJacky/Nginx-UI/internal/nginx_log"
	"github.com/gin-gonic/gin"
	"github.com/uozi-tech/cosy/logger"
)

// GetLogList returns a list of Nginx log files with their index status
func GetLogList(c *gin.Context) {
	var filters []func(*nginx_log.NginxLogWithIndex) bool

	if logType := c.Query("type"); logType != "" {
		filters = append(filters, func(entry *nginx_log.NginxLogWithIndex) bool {
			return entry.Type == logType
		})
	}

	if name := c.Query("name"); name != "" {
		filters = append(filters, func(entry *nginx_log.NginxLogWithIndex) bool {
			return strings.Contains(entry.Name, name)
		})
	}

	if path := c.Query("path"); path != "" {
		filters = append(filters, func(entry *nginx_log.NginxLogWithIndex) bool {
			return strings.Contains(entry.Path, path)
		})
	}

	// Add filter for indexed status if requested
	if indexed := c.Query("indexed"); indexed != "" {
		filters = append(filters, func(entry *nginx_log.NginxLogWithIndex) bool {
			switch indexed {
			case "true":
				return entry.IndexStatus == nginx_log.IndexStatusIndexed
			case "false":
				return entry.IndexStatus == nginx_log.IndexStatusNotIndexed
			case "indexing":
				return entry.IndexStatus == nginx_log.IndexStatusIndexing
			default:
				return true
			}
		})
	}

	data := nginx_log.GetAllLogsWithIndexGrouped(filters...)

	orderBy := c.DefaultQuery("sort_by", "name")
	sort := c.DefaultQuery("order", "desc")

	data = nginx_log.SortWithIndex(orderBy, sort, data)

	// Calculate summary statistics
	totalCount := len(data)
	indexedCount := 0
	indexingCount := 0
	var totalDocuments uint64 = 0

	// Try to get total document count from modern indexer if available
	// The indexer is the source of truth for document counts.
	indexer := nginx_log.GetIndexer()
	if indexer != nil {
		stats := indexer.GetStats()
		if stats != nil {
			totalDocuments = stats.TotalDocuments
			logger.Debugf("Retrieved document count from indexer stats: %d", totalDocuments)
		}
	}

	for _, log := range data {
		switch log.IndexStatus {
		case nginx_log.IndexStatusIndexed:
			indexedCount++
			if totalDocuments == 0 {
				// Fallback to summing up document counts from the database records
				// This handles the case where the searcher might not be immediately up-to-date
				var dbCount uint64
				for _, l := range data {
					dbCount += l.DocumentCount
				}
				totalDocuments = dbCount
			}
		case nginx_log.IndexStatusIndexing:
			indexingCount++
		}
	}

	c.JSON(http.StatusOK, LogListResponse{
		Data: data,
		Summary: LogListSummary{
			TotalFiles:    totalCount,
			IndexedFiles:  indexedCount,
			IndexingFiles: indexingCount,
			DocumentCount: int(totalDocuments),
		},
	})
}