mirror of
https://github.com/opencloud-eu/opencloud.git
synced 2025-12-31 17:30:29 -06:00
124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package command
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
gofig "github.com/gookit/config/v2"
|
|
gooyaml "github.com/gookit/config/v2/yaml"
|
|
ociscfg "github.com/owncloud/ocis/ocis-pkg/config"
|
|
"github.com/owncloud/ocis/ocis-pkg/log"
|
|
"github.com/owncloud/ocis/ocis-pkg/version"
|
|
"github.com/owncloud/ocis/proxy/pkg/config"
|
|
"github.com/thejerf/suture/v4"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
// Execute is the entry point for the ocis-proxy command.
|
|
func Execute(cfg *config.Config) error {
|
|
app := &cli.App{
|
|
Name: "ocis-proxy",
|
|
Version: version.String,
|
|
Usage: "proxy for oCIS",
|
|
Compiled: version.Compiled(),
|
|
|
|
Authors: []*cli.Author{
|
|
{
|
|
Name: "ownCloud GmbH",
|
|
Email: "support@owncloud.com",
|
|
},
|
|
},
|
|
|
|
//Flags: flagset.RootWithConfig(cfg),
|
|
|
|
Before: func(c *cli.Context) error {
|
|
cfg.Service.Version = version.String
|
|
return nil
|
|
},
|
|
|
|
Commands: []*cli.Command{
|
|
Server(cfg),
|
|
Health(cfg),
|
|
PrintVersion(cfg),
|
|
},
|
|
}
|
|
|
|
cli.HelpFlag = &cli.BoolFlag{
|
|
Name: "help,h",
|
|
Usage: "Show the help",
|
|
}
|
|
|
|
cli.VersionFlag = &cli.BoolFlag{
|
|
Name: "version,v",
|
|
Usage: "Print the version",
|
|
}
|
|
|
|
return app.Run(os.Args)
|
|
}
|
|
|
|
// NewLogger initializes a service-specific logger instance.
|
|
func NewLogger(cfg *config.Config) log.Logger {
|
|
return log.NewLogger(
|
|
log.Name("proxy"),
|
|
log.Level(cfg.Log.Level),
|
|
log.Pretty(cfg.Log.Pretty),
|
|
log.Color(cfg.Log.Color),
|
|
log.File(cfg.Log.File),
|
|
)
|
|
}
|
|
|
|
// ParseConfig loads proxy configuration from known paths.
|
|
func ParseConfig(c *cli.Context, cfg *config.Config) error {
|
|
// create a new config and load files and env values onto it since this needs to be thread-safe.
|
|
cnf := gofig.NewWithOptions("proxy", gofig.ParseEnv)
|
|
|
|
// TODO(refs) add ENV + toml + json
|
|
cnf.AddDriver(gooyaml.Driver)
|
|
|
|
// TODO(refs) load from expected locations with the expected name
|
|
err := cnf.LoadFiles("/Users/aunger/code/owncloud/ocis/proxy/pkg/command/proxy_example_config.yaml")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// bind all keys to cfg, as we expect an entire proxy.[yaml, toml...] to define all keys and not only sub values.
|
|
err = cnf.BindStruct("", cfg)
|
|
|
|
return nil
|
|
}
|
|
|
|
// SutureService allows for the proxy command to be embedded and supervised by a suture supervisor tree.
|
|
type SutureService struct {
|
|
cfg *config.Config
|
|
}
|
|
|
|
// NewSutureService creates a new proxy.SutureService
|
|
func NewSutureService(cfg *ociscfg.Config) suture.Service {
|
|
inheritLogging(cfg)
|
|
if cfg.Mode == 0 {
|
|
cfg.Proxy.Supervised = true
|
|
}
|
|
return SutureService{
|
|
cfg: cfg.Proxy,
|
|
}
|
|
}
|
|
|
|
func (s SutureService) Serve(ctx context.Context) error {
|
|
s.cfg.Context = ctx
|
|
if err := Execute(s.cfg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// inheritLogging is a poor man's global logging state tip-toeing around circular dependencies. It sets the logging
|
|
// of the service to whatever is in the higher config (in this case coming from ocis.yaml) and sets them as defaults,
|
|
// being overwritten when the extension parses its config file / env variables.
|
|
func inheritLogging(cfg *ociscfg.Config) {
|
|
cfg.Proxy.Log.File = cfg.Log.File
|
|
cfg.Proxy.Log.Color = cfg.Log.Color
|
|
cfg.Proxy.Log.Pretty = cfg.Log.Pretty
|
|
cfg.Proxy.Log.Level = cfg.Log.Level
|
|
}
|