mirror of https://github.com/curusarn/resh
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.3 KiB
100 lines
3.3 KiB
package main
|
|
|
|
import (
|
|
"github.com/curusarn/resh/internal/histcli"
|
|
"github.com/curusarn/resh/internal/syncconnector"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/curusarn/resh/internal/cfg"
|
|
"github.com/curusarn/resh/internal/histfile"
|
|
"github.com/curusarn/resh/internal/recordint"
|
|
"github.com/curusarn/resh/internal/sesswatch"
|
|
"github.com/curusarn/resh/internal/signalhandler"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// TODO: turn server and handlers into package
|
|
|
|
type Server struct {
|
|
sugar *zap.SugaredLogger
|
|
config cfg.Config
|
|
|
|
reshHistoryPath string
|
|
bashHistoryPath string
|
|
zshHistoryPath string
|
|
}
|
|
|
|
func (s *Server) Run() {
|
|
var recordSubscribers []chan recordint.Collect
|
|
var sessionInitSubscribers []chan recordint.SessionInit
|
|
var sessionDropSubscribers []chan string
|
|
var signalSubscribers []chan os.Signal
|
|
|
|
shutdown := make(chan string)
|
|
|
|
history := histcli.New(s.sugar)
|
|
|
|
// histfile
|
|
histfileRecords := make(chan recordint.Collect)
|
|
recordSubscribers = append(recordSubscribers, histfileRecords)
|
|
histfileSessionsToDrop := make(chan string)
|
|
sessionDropSubscribers = append(sessionDropSubscribers, histfileSessionsToDrop)
|
|
histfileSignals := make(chan os.Signal)
|
|
signalSubscribers = append(signalSubscribers, histfileSignals)
|
|
maxHistSize := 10000 // lines
|
|
minHistSizeKB := 2000 // roughly lines
|
|
histfile.New(s.sugar, histfileRecords, histfileSessionsToDrop,
|
|
s.reshHistoryPath, s.bashHistoryPath, s.zshHistoryPath,
|
|
maxHistSize, minHistSizeKB,
|
|
histfileSignals, shutdown, history)
|
|
|
|
// sesswatch
|
|
sesswatchRecords := make(chan recordint.Collect)
|
|
// TODO: add sync connector subscriber
|
|
recordSubscribers = append(recordSubscribers, sesswatchRecords)
|
|
sesswatchSessionsToWatch := make(chan recordint.SessionInit)
|
|
sessionInitSubscribers = append(sessionInitSubscribers, sesswatchSessionsToWatch)
|
|
sesswatch.Go(
|
|
s.sugar,
|
|
sesswatchSessionsToWatch,
|
|
sesswatchRecords,
|
|
sessionDropSubscribers,
|
|
s.config.SessionWatchPeriodSeconds,
|
|
)
|
|
|
|
// handlers
|
|
mux := http.NewServeMux()
|
|
mux.Handle("/status", &statusHandler{sugar: s.sugar})
|
|
mux.Handle("/record", &recordHandler{sugar: s.sugar, subscribers: recordSubscribers})
|
|
mux.Handle("/session_init", &sessionInitHandler{sugar: s.sugar, subscribers: sessionInitSubscribers})
|
|
mux.Handle("/dump", &dumpHandler{sugar: s.sugar, history: history})
|
|
|
|
server := &http.Server{
|
|
Addr: "localhost:" + strconv.Itoa(s.config.Port),
|
|
Handler: mux,
|
|
ReadTimeout: 1 * time.Second,
|
|
WriteTimeout: 1 * time.Second,
|
|
ReadHeaderTimeout: 1 * time.Second,
|
|
IdleTimeout: 30 * time.Second,
|
|
}
|
|
go server.ListenAndServe()
|
|
|
|
s.sugar.Infow("", "sync_addr", s.config.SyncConnectorAddress)
|
|
if s.config.SyncConnectorAddress != nil {
|
|
sc, err := syncconnector.New(s.sugar, *s.config.SyncConnectorAddress, s.config.SyncConnectorAuthToken, s.config.SyncConnectorPullPeriodSeconds, s.config.SyncConnectorSendPeriodSeconds, history)
|
|
if err != nil {
|
|
s.sugar.Errorw("Sync Connector init failed", "error", err)
|
|
} else {
|
|
s.sugar.Infow("Initialized Sync Connector", "Sync Connector", sc)
|
|
}
|
|
// TODO: load sync connector data
|
|
// TODO: load sync connector data
|
|
// TODO: send connector data periodically (record by record / or batch)
|
|
}
|
|
|
|
// signalhandler - takes over the main goroutine so when signal handler exists the whole program exits
|
|
signalhandler.Run(s.sugar, signalSubscribers, shutdown, server)
|
|
}
|
|
|