63cebf07ab
- Updated file permissions in certificate_service_test.go and log_service_test.go to use octal notation. - Added a new doc.go file to document the services package. - Enhanced error handling in docker_service.go, log_service.go, notification_service.go, proxyhost_service.go, remoteserver_service.go, update_service.go, and uptime_service.go by logging errors when closing resources. - Improved log_service.go to simplify log file processing and deduplication. - Introduced CRUD tests for notification templates in notification_service_template_test.go. - Removed the obsolete python_compile_check.sh script. - Updated notification_service.go to improve template management functions. - Added tests for uptime service notifications in uptime_service_notification_test.go.
122 lines
3.1 KiB
Go
122 lines
3.1 KiB
Go
package handlers
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/Wikid82/charon/backend/internal/logger"
|
|
"github.com/Wikid82/charon/backend/internal/models"
|
|
"github.com/Wikid82/charon/backend/internal/services"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type LogsHandler struct {
|
|
service *services.LogService
|
|
}
|
|
|
|
func NewLogsHandler(service *services.LogService) *LogsHandler {
|
|
return &LogsHandler{service: service}
|
|
}
|
|
|
|
func (h *LogsHandler) List(c *gin.Context) {
|
|
logs, err := h.service.ListLogs()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to list logs"})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, logs)
|
|
}
|
|
|
|
func (h *LogsHandler) Read(c *gin.Context) {
|
|
filename := c.Param("filename")
|
|
|
|
// Parse query parameters
|
|
limit, _ := strconv.Atoi(c.DefaultQuery("limit", "50"))
|
|
offset, _ := strconv.Atoi(c.DefaultQuery("offset", "0"))
|
|
|
|
filter := models.LogFilter{
|
|
Search: c.Query("search"),
|
|
Host: c.Query("host"),
|
|
Status: c.Query("status"),
|
|
Level: c.Query("level"),
|
|
Limit: limit,
|
|
Offset: offset,
|
|
Sort: c.DefaultQuery("sort", "desc"),
|
|
}
|
|
|
|
logs, total, err := h.service.QueryLogs(filename, filter)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Log file not found"})
|
|
return
|
|
}
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read log"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"filename": filename,
|
|
"logs": logs,
|
|
"total": total,
|
|
"limit": limit,
|
|
"offset": offset,
|
|
})
|
|
}
|
|
|
|
func (h *LogsHandler) Download(c *gin.Context) {
|
|
filename := c.Param("filename")
|
|
path, err := h.service.GetLogPath(filename)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "invalid filename") {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Log file not found"})
|
|
return
|
|
}
|
|
|
|
// Create a temporary file to serve a consistent snapshot
|
|
// This prevents Content-Length mismatches if the live log file grows during download
|
|
tmpFile, err := os.CreateTemp("", "charon-log-*.log")
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create temp file"})
|
|
return
|
|
}
|
|
defer func() {
|
|
if err := os.Remove(tmpFile.Name()); err != nil {
|
|
logger.Log().WithError(err).Warn("failed to remove temp file")
|
|
}
|
|
}()
|
|
|
|
srcFile, err := os.Open(path)
|
|
if err != nil {
|
|
if err := tmpFile.Close(); err != nil {
|
|
logger.Log().WithError(err).Warn("failed to close temp file")
|
|
}
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to open log file"})
|
|
return
|
|
}
|
|
defer func() {
|
|
if err := srcFile.Close(); err != nil {
|
|
logger.Log().WithError(err).Warn("failed to close source log file")
|
|
}
|
|
}()
|
|
|
|
if _, err := io.Copy(tmpFile, srcFile); err != nil {
|
|
if err := tmpFile.Close(); err != nil {
|
|
logger.Log().WithError(err).Warn("failed to close temp file after copy error")
|
|
}
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to copy log file"})
|
|
return
|
|
}
|
|
if err := tmpFile.Close(); err != nil {
|
|
logger.Log().WithError(err).Warn("failed to close temp file after copy")
|
|
}
|
|
|
|
c.Header("Content-Disposition", "attachment; filename="+filename)
|
|
c.File(tmpFile.Name())
|
|
}
|