feat: support to install docker via dnf (#337)

Co-authored-by: rick <[email protected]>

Author: LinuxSuRen

pkg/os/dnf/docker.go

@@ -0,0 +1,86 @@
+package dnf
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/linuxsuren/http-downloader/pkg/exec"
+)
+
+// dockerInstallerInFedora is the installer of Docker in Fedora
+type dockerInstallerInFedora struct {
+	Execer exec.Execer
+	count  int
+}
+
+// Available check if support current platform
+func (d *dockerInstallerInFedora) Available() (ok bool) {
+	if d.Execer.OS() == "linux" {
+		_, err := d.Execer.LookPath("dnf")
+		ok = err == nil
+	}
+	return
+}
+
+// Install installs the Docker
+func (d *dockerInstallerInFedora) Install() (err error) {
+	if err = d.Execer.RunCommand("dnf", "-y", "install", "dnf-plugins-core"); err != nil {
+		return
+	}
+
+	if err = d.Execer.RunCommand("dnf", "config-manager",
+		"--add-repo", "https://download.docker.com/linux/fedora/docker-ce.repo"); err != nil {
+		return
+	}
+
+	err = d.Execer.RunCommand("dnf", "install", "docker-ce", "docker-ce-cli", "containerd.io", "docker-compose-plugin")
+	return
+}
+
+// Uninstall uninstalls the Docker
+func (d *dockerInstallerInFedora) Uninstall() (err error) {
+	err = d.Execer.RunCommand("dnf", "remove", "docker",
+		"docker-client", "docker-client-latest", "docker-common", "docker-latest",
+		"docker-latest-logrotate",
+		"docker-logrotate",
+		"docker-selinux",
+		"docker-engine-selinux",
+		"docker-engine")
+	return
+}
+
+// WaitForStart waits for the service be started
+func (d *dockerInstallerInFedora) WaitForStart() (ok bool, err error) {
+	var result string
+	if result, err = d.Execer.RunCommandAndReturn("systemctl", "", "status", "docker"); err != nil {
+		return
+	} else if strings.Contains(result, "Unit docker.service could not be found") {
+		err = fmt.Errorf("unit docker.service could not be found")
+	} else if strings.Contains(result, "Active: active") {
+		ok = true
+	} else {
+		if d.count > 0 {
+			fmt.Println("waiting for Docker service start")
+		} else if d.count > 9 {
+			return
+		}
+
+		d.count++
+		time.Sleep(time.Second * 1)
+		return d.WaitForStart()
+	}
+	return
+}
+
+// Start starts the Docker service
+func (d *dockerInstallerInFedora) Start() (err error) {
+	err = d.Execer.RunCommand("systemctl", "start", "docker")
+	return
+}
+
+// Stop stops the Docker service
+func (d *dockerInstallerInFedora) Stop() (err error) {
+	err = d.Execer.RunCommand("systemctl", "stop", "docker")
+	return
+}

pkg/os/dnf/init.go

@@ -0,0 +1,11 @@
+package dnf
+
+import (
+	"github.com/linuxsuren/http-downloader/pkg/exec"
+	"github.com/linuxsuren/http-downloader/pkg/os/core"
+)
+
+// SetInstallerRegistry sets the installer of registry
+func SetInstallerRegistry(registry core.InstallerRegistry, defaultExecer exec.Execer) {
+	registry.Registry("docker", &dockerInstallerInFedora{Execer: defaultExecer})
+}

pkg/os/dnf/init_test.go

@@ -0,0 +1,48 @@
+package dnf
+
+import (
+	"errors"
+	"testing"
+
+	"github.com/linuxsuren/http-downloader/pkg/exec"
+	"github.com/linuxsuren/http-downloader/pkg/os/core"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCommonCase(t *testing.T) {
+	registry := &core.FakeRegistry{}
+	SetInstallerRegistry(registry, exec.FakeExecer{
+		ExpectOS: "linux",
+	})
+
+	registry.Walk(func(s string, i core.Installer) {
+		t.Run(s, func(t *testing.T) {
+			assert.True(t, i.Available())
+			assert.Nil(t, i.Uninstall())
+			assert.Nil(t, i.Install())
+			assert.Nil(t, i.Start())
+			assert.Nil(t, i.Stop())
+		})
+	})
+
+	errRegistry := &core.FakeRegistry{}
+	SetInstallerRegistry(errRegistry, exec.FakeExecer{
+		ExpectLookPathError: errors.New("error"),
+		ExpectError:         errors.New("error"),
+		ExpectOS:            "linux",
+	})
+	errRegistry.Walk(func(s string, i core.Installer) {
+		t.Run(s, func(t *testing.T) {
+			assert.False(t, i.Available())
+			assert.NotNil(t, i.Uninstall())
+			if s != "docker" {
+				assert.NotNil(t, i.Install())
+				assert.Nil(t, i.Start())
+				assert.Nil(t, i.Stop())
+				ok, err := i.WaitForStart()
+				assert.True(t, ok)
+				assert.Nil(t, err)
+			}
+		})
+	})
+}

pkg/os/installer.go

@@ -8,6 +8,7 @@ import (
 	"github.com/linuxsuren/http-downloader/pkg/exec"
 	"github.com/linuxsuren/http-downloader/pkg/os/apt"
 	"github.com/linuxsuren/http-downloader/pkg/os/core"
+	"github.com/linuxsuren/http-downloader/pkg/os/dnf"
 	"github.com/linuxsuren/http-downloader/pkg/os/docker"
 	"github.com/linuxsuren/http-downloader/pkg/os/yum"
 	"github.com/mitchellh/go-homedir"
@@ -28,6 +29,7 @@ func init() {
 	yum.SetInstallerRegistry(defaultInstallerRegistry, defaultExecer)
 	apt.SetInstallerRegistry(defaultInstallerRegistry, defaultExecer)
 	docker.SetInstallerRegistry(defaultInstallerRegistry, defaultExecer)
+	dnf.SetInstallerRegistry(defaultInstallerRegistry, defaultExecer)
 
 	var userHome string
 	var err error