client.go RawHistoryFolder

1package docker
2
3import (
4	"context"
5	"fmt"
6	"os"
7	"time"
8
9	"github.com/docker/docker/api/types"
10	"github.com/docker/docker/client"
11)
12
13type ContainerEngineClient interface {
14	InitializeClient(*Config) error
15	// Add other methods that both Docker and Podman should implement
16}
17type Config struct {
18	Sock         string
19	PodmanEnable bool
20}
21
22var (
23	dockerCli     *client.Client
24	currentConfig *Config
25)
26
27// DockerClient implements the ContainerEngineClient interface for Docker
28type DockerClient struct{}
29
30func (d *DockerClient) InitializeClient(config *Config) error {
31	// Validate if the Sock field is not empty
32	if config.Sock == "" {
33		return fmt.Errorf("sock field in Config is empty")
34	}
35
36	// Check if the socket file exists if not return an error
37	if _, err := os.Stat(config.Sock); os.IsNotExist(err) {
38		return err
39	}
40
41	// Initialize Docker client
42	cli, err := client.NewClientWithOpts(
43		client.WithAPIVersionNegotiation(),
44		client.WithHost("unix://"+config.Sock),
45	)
46	if err != nil {
47		return fmt.Errorf("error initializing Docker client: %s", err)
48	}
49	dockerCli = cli
50	currentConfig = config
51	return nil
52}
53
54// CheckIfInitialized checks if the Docker client is initialized
55func CheckIfInitialized() error {
56	// Check if there is a socket file
57	if currentConfig == nil || currentConfig.Sock == "" {
58		return fmt.Errorf("docker client is not initialized or configuration is missing")
59	}
60
61	socketPath := currentConfig.Sock // Use the package-level variable
62	if _, err := os.Stat(socketPath); os.IsNotExist(err) {
63		return fmt.Errorf("sock file does not exist: %s", socketPath)
64	}
65
66	// Then we check if the dockerCli is nil
67	if dockerCli == nil {
68		return fmt.Errorf("docker client is not initialized")
69	}
70
71	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
72	defer cancel()
73
74	// More stringent check: try to list containers
75	_, err := dockerCli.ContainerList(ctx, types.ContainerListOptions{})
76	if err != nil {
77		return fmt.Errorf("cannot connect to Docker daemon: %s", err)
78	}
79	return nil
80}