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.
79 lines
1.8 KiB
79 lines
1.8 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"syscall"
|
|
|
|
"github.com/adrg/xdg"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
|
|
)
|
|
|
|
// The purpose of this binary is to act as the entrypoint of the isle
|
|
// process. It processes the command-line arguments which are passed in, and
|
|
// then passes execution along to an appropriate binary housed in AppDir/bin
|
|
// (usually a bash script, which is more versatile than a go program).
|
|
|
|
func getAppDirPath() string {
|
|
appDirPath := os.Getenv("APPDIR")
|
|
if appDirPath == "" {
|
|
appDirPath = "."
|
|
}
|
|
return appDirPath
|
|
}
|
|
|
|
var (
|
|
envAppDirPath = getAppDirPath()
|
|
envRuntimeDirPath = filepath.Join(xdg.RuntimeDir, "cryptic-net")
|
|
envDataDirPath = filepath.Join(xdg.DataHome, "cryptic-net")
|
|
)
|
|
|
|
func binPath(name string) string {
|
|
return filepath.Join(envAppDirPath, "bin", name)
|
|
}
|
|
|
|
func main() {
|
|
|
|
logger := mlog.NewLogger(&mlog.LoggerOpts{
|
|
MessageHandler: newLogMsgHandler(),
|
|
MaxLevel: mlog.LevelInfo.Int(),
|
|
})
|
|
defer logger.Close()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
signalCh := make(chan os.Signal, 2)
|
|
signal.Notify(signalCh, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
sig := <-signalCh
|
|
cancel()
|
|
|
|
ctx := mctx.Annotate(ctx, "signal", sig.String())
|
|
logger.Info(ctx, "got signal, exiting gracefully")
|
|
|
|
sig = <-signalCh
|
|
|
|
ctx = mctx.Annotate(ctx, "signal", sig.String())
|
|
logger.FatalString(ctx, "second signal received, force quitting, there may be zombie children left behind, good luck!")
|
|
}()
|
|
|
|
err := subCmdCtx{
|
|
args: os.Args[1:],
|
|
ctx: ctx,
|
|
logger: logger,
|
|
}.doSubCmd(
|
|
subCmdAdmin,
|
|
subCmdDaemon,
|
|
subCmdGarage,
|
|
subCmdHosts,
|
|
subCmdVersion,
|
|
)
|
|
|
|
if err != nil {
|
|
logger.Fatal(ctx, "error running command", err)
|
|
}
|
|
}
|
|
|