Merge pull request #9 from deepfence/add-newruntime

Add NewRuntime util function

autodetect.go

@@ -3,17 +3,20 @@ package vessel
 import (
 	"context"
 	"fmt"
+	"net"
+	"net/url"
+	"strings"
+
 	"github.com/containerd/containerd"
 	"github.com/containerd/containerd/namespaces"
 	"github.com/deepfence/vessel/constants"
+	self_containerd "github.com/deepfence/vessel/containerd"
+	"github.com/deepfence/vessel/docker"
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/client"
 	"github.com/pkg/errors"
 	"github.com/sirupsen/logrus"
 	"google.golang.org/grpc"
-	"net"
-	"net/url"
-	"strings"
 )
 
 func init() {
@@ -187,3 +190,20 @@ func isContainerdRunning(host string) (bool, error) {
 	}
 	return false, nil
 }
+
+// Auto detect and returns the runtime available for the current system
+func NewRuntime() (Runtime, error) {
+
+	runtime, _, err := AutoDetectRuntime()
+	if err != nil {
+		return nil, err
+	}
+
+	if runtime == constants.DOCKER {
+		return docker.New(), nil
+	} else if runtime == constants.CONTAINERD {
+		return self_containerd.New(), nil
+	}
+
+	return nil, errors.New("Unknown runtime")
+}