This library is the core library of ContainerSSH and simultaneously serves as a library to integrate with ContainerSSH. This readme outlines the basics of using this library, for more detailed documentation please head to containerssh.io.
You can fully embed ContainerSSH into your own application. First, you will need to create the configuration structure:
cfg := config.AppConfig{}
// Set the default configuration:
cfg.Default()
You can then populate this config with your options and create a ContainerSSH instance like this:
pool, lifecycle, err := containerssh.New(cfg, loggerFactory)
if err != nil {
return err
}
You will receive a service pool and a lifecycle as a response. You can use these to start the service pool of ContainerSSH. This will block execution until ContainerSSH stops.
err := lifecycle.Run()
This will run ContainerSSH in the current Goroutine. You can also use the lifecycle to add hooks to lifecycle states of ContainerSSH. You must do this before you call Run()
. For example:
lifecycle.OnStarting(
func(s service.Service, l service.Lifecycle) {
print("ContainerSSH is starting...")
},
)
You can also have ContainerSSH stop gracefully by using the Stop()
function on the lifecycle. This takes a context as an argument, which is taken as a timeout for the graceful shutdown.
Finally, you can use the returned pool
variable to rotate the logs. This will trigger all ContainerSSH services to close and reopen their log files.
pool.RotateLogs()
The configuration webhook lets you dynamically configure ContainerSSH. This library contains the tools to create a tiny webserver to serve these webhook requests.
First, you need to fetch this library as a dependency using go modules:
go get github.com/containerssh/libcontainerssh
Next, you will have to write an implementation for the following interface:
package main
import (
"github.com/containerssh/libcontainerssh/config"
)
type ConfigRequestHandler interface {
OnConfig(request config.Request) (config.AppConfig, error)
}
The best way to do this is creating a struct and adding a method with a receiver:
type myConfigReqHandler struct {
}
func (m *myConfigReqHandler) OnConfig(
request configuration.ConfigRequest,
) (config configuration.AppConfig, err error) {
// We recommend using an IDE to discover the possible options here.
if request.Username == "foo" {
config.Docker.Config.ContainerConfig.Image = "yourcompany/yourimage"
}
return config, err
}
Warning! Your OnConfig
method should only return an error if it can genuinely not serve the request. This should not be used as a means to reject users. This should be done using the authentication server. If you return an error ContainerSSH will retry the request several times in an attempt to work around network failures.
Once you have your handler implemented you must decide which method you want to use for integration.
This method is useful if you don't want to run anything else on the webserver, only the config endpoint. You can create a new server like this:
package main
import (
"signal"
"github.com/containerssh/libcontainerssh/config"
"github.com/containerssh/libcontainerssh/config/webhook"
"github.com/containerssh/libcontainerssh/log"
"github.com/containerssh/libcontainerssh/service"
)
func main() {
logger := log.NewLogger(&config.LogConfig{
// Add logging configuration here
})
// Create the webserver service
srv, err := webhook.NewServer(
config.HTTPServerConfiguration{
Listen: "0.0.0.0:8080",
},
&myConfigReqHandler{},
logger,
)
if err != nil {
panic(err)
}
// Set up the lifecycle handler
lifecycle := service.NewLifecycle(srv)
// Launch the webserver in the background
go func() {
//Ignore error, handled later.
_ = lifecycle.Run()
}()
// Handle signals and terminate webserver gracefully when needed.
signals := make(chan os.Signal, 1)
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
go func() {
if _, ok := <-signals; ok {
// ok means the channel wasn't closed, let's trigger a shutdown.
// The context given is the timeout for the shutdown.
lifecycle.Stop(
context.WithTimeout(
context.Background(),
20 * time.Second,
),
)
}
}()
// Wait for the service to terminate.
lastError := lifecycle.Wait()
// We are already shutting down, ignore further signals
signal.Ignore(syscall.SIGINT, syscall.SIGTERM)
// close signals channel so the signal handler gets terminated
close(signals)
if lastError != nil {
// Exit with a non-zero signal
fmt.Fprintf(
os.Stderr,
"an error happened while running the server (%v)",
lastError,
)
os.Exit(1)
}
os.Exit(0)
}
Note: We recommend securing client-server communication with certificates. Please see the Securing webhooks section below/
Use this method if you want to integrate your handler with an existing Go HTTP server. This is rather simple:
handler, err := configuration.NewHandler(&myConfigReqHandler{}, logger)
You can now use the handler
variable as a handler for the http
package or a MUX like gorilla/mux.
This library also contains the components to call the configuration server in a simplified fashion. To create a client simply call the following method:
client, err := configuration.NewClient(
configuration.ClientConfig{
http.ClientConfiguration{
URL: "http://your-server/config-endpoint/"
}
},
logger,
metricsCollector,
)
The logger
is a logger from the log library, the metricsCollector
is supplied by the metrics library.
You can now use the client
variable to fetch the configuration specific to a connecting client:
connectionID := "0123456789ABCDEF"
appConfig, err := client.Get(
ctx,
"my-name-is-trinity",
net.TCPAddr{
IP: net.ParseIP("127.0.0.1"),
Port: 2222,
},
connectionID,
) (AppConfig, error)
Now you have the client-specific configuration in appConfig
.
Note: We recommend securing client-server communication with certificates. The details about securing your HTTP requests are documented in the HTTP library.
This library also provides simplified methods for reading the configuration from an io.Reader
and writing it to an io.Writer
.
file, err := os.Open("file.yaml")
// ...
loader, err := configuration.NewReaderLoader(
file,
logger,
configuration.FormatYAML,
)
// Read global config
appConfig := &configuration.AppConfig{}
err := loader.Load(ctx, appConfig)
// Read connection-specific config:
err := loader.LoadConnection(
ctx,
"my-name-is-trinity",
net.TCPAddr{
IP: net.ParseIP("127.0.0.1"),
Port: 2222,
},
connectionID,
appConfig,
)
As you can see these loaders are designed to be chained together. For example, you could add an HTTP loader after the file loader:
httpLoader, err := configuration.NewHTTPLoader(clientConfig, logger)
This HTTP loader calls the HTTP client described above.
Conversely, you can write the configuration to a YAML format:
saver, err := configuration.NewWriterSaver(
os.Stdout,
logger,
configuration.FormatYAML,
)
err := saver.Save(appConfig)