mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package neko
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
|
|
"n.eko.moe/neko/internal/config"
|
|
"n.eko.moe/neko/internal/structs"
|
|
api "n.eko.moe/neko/internal/http"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
//
|
|
buildDate = ""
|
|
//
|
|
gitCommit = ""
|
|
//
|
|
gitVersion = ""
|
|
//
|
|
gitState = ""
|
|
// Major version when you make incompatible API changes,
|
|
major = "0"
|
|
// Minor version when you add functionality in a backwards-compatible manner, and
|
|
minor = "0"
|
|
// Patch version when you make backwards-compatible bug fixeneko.
|
|
patch = "0"
|
|
)
|
|
|
|
var Service *Neko
|
|
|
|
func init() {
|
|
Service = &Neko{
|
|
Version: &structs.Version{
|
|
Major: major,
|
|
Minor: minor,
|
|
Patch: patch,
|
|
GitVersion: gitVersion,
|
|
GitCommit: gitCommit,
|
|
GitTreeState: gitState,
|
|
BuildDate: buildDate,
|
|
GoVersion: runtime.Version(),
|
|
Compiler: runtime.Compiler,
|
|
Platform: fmt.Sprintf("%s/%s", runtime.GOOS, runtime.GOARCH),
|
|
},
|
|
Root: &config.Root{},
|
|
Serve: &config.Serve{},
|
|
}
|
|
}
|
|
|
|
type Neko struct {
|
|
Version *structs.Version
|
|
Root *config.Root
|
|
Serve *config.Serve
|
|
Logger zerolog.Logger
|
|
http *http.Server
|
|
}
|
|
|
|
func (neko *Neko) Preflight() {
|
|
neko.Logger = log.With().Str("service", "neko").Logger()
|
|
}
|
|
|
|
func (neko *Neko) Start() {
|
|
server := api.New(neko.Serve.Bind, neko.Serve.Password, neko.Serve.Static)
|
|
|
|
if neko.Serve.Cert != "" && neko.Serve.Key != "" {
|
|
go func() {
|
|
if err := server.ListenAndServeTLS(neko.Serve.Cert, neko.Serve.Key); err != http.ErrServerClosed {
|
|
neko.Logger.Panic().Err(err).Msg("Unable to start https server")
|
|
}
|
|
}()
|
|
neko.Logger.Info().Msgf("HTTPS listening on %s", server.Addr)
|
|
} else {
|
|
go func() {
|
|
if err := server.ListenAndServe(); err != http.ErrServerClosed {
|
|
neko.Logger.Panic().Err(err).Msg("Unable to start http server")
|
|
}
|
|
}()
|
|
neko.Logger.Warn().Msgf("HTTP listening on %s", server.Addr)
|
|
}
|
|
|
|
neko.http = server
|
|
}
|
|
|
|
func (neko *Neko) Shutdown() {
|
|
if neko.http != nil {
|
|
if err := neko.http.Shutdown(context.Background()); err != nil {
|
|
neko.Logger.Err(err).Msg("HTTP server shutdown with an error")
|
|
} else {
|
|
neko.Logger.Debug().Msg("HTTP server shutdown")
|
|
}
|
|
}
|
|
}
|
|
|
|
func (neko *Neko) ServeCommand(cmd *cobra.Command, args []string) {
|
|
neko.Logger.Info().Msg("Starting HTTP/S server")
|
|
neko.Start()
|
|
|
|
neko.Logger.Info().Msg("Service ready")
|
|
|
|
quit := make(chan os.Signal)
|
|
signal.Notify(quit, os.Interrupt)
|
|
sig := <-quit
|
|
|
|
neko.Logger.Warn().Msgf("Received %s, attempting graceful shutdown: \n", sig)
|
|
|
|
neko.Shutdown()
|
|
neko.Logger.Info().Msg("Shutting down complete")
|
|
}
|