UmputunandGitHub
4adbca22b2
* fresh vendor * enforce vendor flag with env * move tree to service package * move controllers routing to rest.go, remove separate subrouters * cleanup/formatting * missing vendor packages * move rss, img proxy, robots and getstarted to top-level routes * simplify logger middleware usage * randomize server test ports * check for non-admin in admin tests * lint: minor warns * admin test to jwt admin
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
"syscall"
|
|
|
|
log "github.com/go-pkgz/lgr"
|
|
"github.com/jessevdk/go-flags"
|
|
|
|
"github.com/umputun/remark/backend/app/cmd"
|
|
)
|
|
|
|
// Opts with all cli commands and flags
|
|
type Opts struct {
|
|
ServerCmd cmd.ServerCommand `command:"server"`
|
|
ImportCmd cmd.ImportCommand `command:"import"`
|
|
BackupCmd cmd.BackupCommand `command:"backup"`
|
|
RestoreCmd cmd.RestoreCommand `command:"restore"`
|
|
AvatarCmd cmd.AvatarCommand `command:"avatar"`
|
|
CleanupCmd cmd.CleanupCommand `command:"cleanup"`
|
|
|
|
RemarkURL string `long:"url" env:"REMARK_URL" required:"true" description:"url to remark"`
|
|
SharedSecret string `long:"secret" env:"SECRET" required:"true" description:"shared secret key"`
|
|
|
|
Dbg bool `long:"dbg" env:"DEBUG" description:"debug mode"`
|
|
}
|
|
|
|
var revision = "unknown"
|
|
|
|
func main() {
|
|
fmt.Printf("remark42 %s\n", revision)
|
|
|
|
var opts Opts
|
|
p := flags.NewParser(&opts, flags.Default)
|
|
p.CommandHandler = func(command flags.Commander, args []string) error {
|
|
setupLog(opts.Dbg)
|
|
// commands implements CommonOptionsCommander to allow passing set of extra options defined for all commands
|
|
c := command.(cmd.CommonOptionsCommander)
|
|
c.SetCommon(cmd.CommonOpts{
|
|
RemarkURL: opts.RemarkURL,
|
|
SharedSecret: opts.SharedSecret,
|
|
Revision: revision,
|
|
})
|
|
err := c.Execute(args)
|
|
if err != nil {
|
|
log.Printf("[ERROR] failed with %+v", err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
if _, err := p.Parse(); err != nil {
|
|
if flagsErr, ok := err.(*flags.Error); ok && flagsErr.Type == flags.ErrHelp {
|
|
os.Exit(0)
|
|
} else {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
}
|
|
|
|
func setupLog(dbg bool) {
|
|
if dbg {
|
|
log.Setup(log.Debug, log.CallerFile, log.CallerFunc, log.Msec, log.LevelBraces)
|
|
return
|
|
}
|
|
log.Setup(log.Msec, log.LevelBraces)
|
|
}
|
|
|
|
// getDump reads runtime stack and returns as a string
|
|
func getDump() string {
|
|
maxSize := 5 * 1024 * 1024
|
|
stacktrace := make([]byte, maxSize)
|
|
length := runtime.Stack(stacktrace, true)
|
|
if length > maxSize {
|
|
length = maxSize
|
|
}
|
|
return string(stacktrace[:length])
|
|
}
|
|
|
|
func init() {
|
|
// catch SIGQUIT and print stack traces
|
|
sigChan := make(chan os.Signal)
|
|
go func() {
|
|
for range sigChan {
|
|
log.Printf("[INFO] SIGQUIT detected, dump:\n%s", getDump())
|
|
}
|
|
}()
|
|
signal.Notify(sigChan, syscall.SIGQUIT)
|
|
}
|