From a181d4c0a6be41efc5608c6140bdf3d41c7e4890 Mon Sep 17 00:00:00 2001
From: Homayoon Alimohammadi <homayoon.alimohammadi@canonical.com>
Date: Fri, 6 Dec 2024 16:45:38 +0400
Subject: [PATCH] Fix pebble Containerd path (#874)

---
 src/k8s/pkg/snap/pebble.go | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/k8s/pkg/snap/pebble.go b/src/k8s/pkg/snap/pebble.go
index c4a67019e..6da366af6 100644
--- a/src/k8s/pkg/snap/pebble.go
+++ b/src/k8s/pkg/snap/pebble.go
@@ -13,9 +13,10 @@ import (
 )
 
 type PebbleOpts struct {
-	SnapDir       string
-	SnapCommonDir string
-	RunCommand    func(ctx context.Context, command []string, opts ...func(c *exec.Cmd)) error
+	SnapDir           string
+	SnapCommonDir     string
+	RunCommand        func(ctx context.Context, command []string, opts ...func(c *exec.Cmd)) error
+	ContainerdBaseDir string
 }
 
 // pebble implements the Snap interface.
@@ -38,6 +39,15 @@ func NewPebble(opts PebbleOpts) *pebble {
 		},
 	}
 
+	containerdBaseDir := opts.ContainerdBaseDir
+	if containerdBaseDir == "" {
+		containerdBaseDir = "/"
+		if s.Strict() {
+			containerdBaseDir = opts.SnapCommonDir
+		}
+	}
+	s.containerdBaseDir = containerdBaseDir
+
 	return s
 }