cmd/cli: add internal logging

So in case of no logging enabled, useful data could be sent to ControlD
server for further troubleshooting.
This commit is contained in:
Cuong Manh Le
2024-12-09 23:17:00 +07:00
committed by Cuong Manh Le
parent a63a30c76b
commit cd5619a05b
6 changed files with 326 additions and 14 deletions

View File

@@ -2,6 +2,7 @@ package cli
import (
"context"
"encoding/base64"
"encoding/json"
"net"
"net/http"
@@ -25,6 +26,8 @@ const (
deactivationPath = "/deactivation"
cdPath = "/cd"
ifacePath = "/iface"
viewLogsPath = "/logs/view"
sendLogsPath = "/logs/send"
)
type controlServer struct {
@@ -211,6 +214,49 @@ func (p *prog) registerControlServerHandler() {
}
w.WriteHeader(http.StatusBadRequest)
}))
p.cs.register(viewLogsPath, http.HandlerFunc(func(w http.ResponseWriter, request *http.Request) {
data, err := p.logContent()
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
if len(data) == 0 {
w.WriteHeader(http.StatusMovedPermanently)
return
}
if err := json.NewEncoder(w).Encode(&logViewResponse{Data: string(data)}); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
}))
p.cs.register(sendLogsPath, http.HandlerFunc(func(w http.ResponseWriter, request *http.Request) {
if time.Since(p.internalLogSent) < logSentInterval {
w.WriteHeader(http.StatusServiceUnavailable)
return
}
data, err := p.logContent()
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
if len(data) == 0 {
w.WriteHeader(http.StatusMovedPermanently)
return
}
logFile := base64.StdEncoding.EncodeToString(data)
req := &controld.LogsRequest{
UID: cdUID,
LogFile: logFile,
}
mainLog.Load().Debug().Msg("sending log file to ControlD server")
if err := controld.SendLogs(req, cdDev); err != nil {
mainLog.Load().Error().Msgf("could not send log file to ControlD server: %v", err)
http.Error(w, err.Error(), http.StatusInternalServerError)
} else {
mainLog.Load().Debug().Msg("sending log file successfully")
}
p.internalLogSent = time.Now()
}))
}
func jsonResponse(next http.Handler) http.Handler {