From aa3678c632e5ea94f61886a31ea0a590af897391 Mon Sep 17 00:00:00 2001
From: Julian Vassev <jvassev@gmail.com>
Date: Mon, 13 May 2024 19:58:25 +0300
Subject: [PATCH] Support k3s-agent deployment

---
 k8s/scripts/kubelet-config-helper.sh | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/k8s/scripts/kubelet-config-helper.sh b/k8s/scripts/kubelet-config-helper.sh
index d23f74d..82ec897 100755
--- a/k8s/scripts/kubelet-config-helper.sh
+++ b/k8s/scripts/kubelet-config-helper.sh
@@ -1480,6 +1480,30 @@ function do_config_kubelet() {
 	fi
 }
 
+###############################################################################
+# Scenario 6: k3s worker
+###############################################################################
+function kubelet_k3s_deploymet() {
+	if systemctl is-active --quiet k3s-agent; then
+		return
+	fi
+
+	return 1
+}
+
+function do_config_kubelet_k3s_systemd() {
+	local unit_file="/etc/systemd/system/k3s-agent.service"
+
+	if grep -q 'container-runtime-endpoint=/var/run/crio/crio.sock' $unit_file; then
+		echo "k3s-agent.service already modified"
+		return
+	fi
+
+	sed "s| agent | agent --kubelet-arg=--cgroup-driver=systemd --container-runtime-endpoint=/var/run/crio/crio.sock |" -i $unit_file
+	systemctl daemon-reload
+	systemctl restart k3s-agent
+}
+
 function main() {
 
 	euid=$(id -u)
@@ -1509,6 +1533,8 @@ function main() {
 		do_config_kubelet_rke2
 	elif kubelet_docker_systemd_deployment; then
 		do_config_kubelet_docker_systemd
+	elif kubelet_k3s_deploymet; then
+		do_config_kubelet_k3s_systemd
 	else
 		do_config_kubelet
 	fi