Added RegisterRuntime hook

engine/default.go

@@ -37,6 +37,10 @@ func RegisterMounter(runtime, name string, mounter runtime.Mounter) {
 	defaultEngine.RegisterMounter(runtime, name, mounter)
 }
 
+func RegisterRuntime(rt runtime.Runtime) {
+	defaultEngine.RegisterRuntime(rt)
+}
+
 func RegisterDatastoreProvider(name string, provider datastore.Provider) {
 	defaultEngine.RegisterDatastoreProvider(name, provider)
 }

engine/engine.go

@@ -50,6 +50,7 @@ type Engine struct {
 	broker      mq.Broker
 	ds          datastore.Datastore
 	mounters    map[string]*runtime.MultiMounter
+	runtime     runtime.Runtime
 	coordinator *coordinator.Coordinator
 	worker      *worker.Worker
 	dsProviders map[string]datastore.Provider
@@ -281,6 +282,16 @@ func (e *Engine) RegisterMounter(rt string, name string, mounter runtime.Mounter
 	mounters.RegisterMounter(name, mounter)
 }
 
+func (e *Engine) RegisterRuntime(rt runtime.Runtime) {
+	e.mu.Lock()
+	defer e.mu.Unlock()
+	e.mustState(StateIdle)
+	if e.runtime != nil {
+		panic("engine: RegisterRuntime called twice")
+	}
+	e.runtime = rt
+}
+
 func (e *Engine) RegisterDatastoreProvider(name string, provider datastore.Provider) {
 	e.mu.Lock()
 	defer e.mu.Unlock()

engine/engine_test.go

@@ -10,6 +10,7 @@ import (
 	"github.com/runabol/tork"
 	"github.com/runabol/tork/datastore"
 	"github.com/runabol/tork/input"
+	"github.com/runabol/tork/runtime/shell"
 
 	"github.com/runabol/tork/mq"
 	"github.com/runabol/tork/runtime"
@@ -224,6 +225,20 @@ func TestRegisterMounter(t *testing.T) {
 	assert.NoError(t, err)
 }
 
+func TestRegisterRuntime(t *testing.T) {
+	eng := New(Config{Mode: ModeStandalone})
+	assert.Equal(t, StateIdle, eng.state)
+
+	eng.RegisterRuntime(shell.NewShellRuntime(shell.Config{}))
+
+	err := eng.Start()
+	assert.NoError(t, err)
+	assert.Equal(t, StateRunning, eng.state)
+
+	err = eng.Terminate()
+	assert.NoError(t, err)
+}
+
 func TestRegisterDatastoreProvider(t *testing.T) {
 	eng := New(Config{Mode: ModeStandalone})
 	assert.Equal(t, StateIdle, eng.state)

engine/worker.go

@@ -38,6 +38,9 @@ func (e *Engine) initWorker() error {
 }
 
 func (e *Engine) initRuntime() (runtime.Runtime, error) {
+	if e.runtime != nil {
+		return e.runtime, nil
+	}
 	runtimeType := conf.StringDefault("runtime.type", runtime.Docker)
 	switch runtimeType {
 	case runtime.Docker:

runtime/shell/shell.go

@@ -52,6 +52,12 @@ func NewShellRuntime(cfg Config) *ShellRuntime {
 	if cfg.Rexec == nil {
 		cfg.Rexec = reexec.Command
 	}
+	if cfg.UID == "" {
+		cfg.UID = DEFAULT_UID
+	}
+	if cfg.GID == "" {
+		cfg.GID = DEFAULT_GID
+	}
 	return &ShellRuntime{
 		cmds:   new(syncx.Map[string, *exec.Cmd]),
 		shell:  cfg.CMD,