From c3613d9e41c2ad9dc954de68434a6fc5b5df12a2 Mon Sep 17 00:00:00 2001
From: weipengzhe <3389096668@qq.com>
Date: Thu, 17 Sep 2026 16:17:50 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B7=B2=E5=8A=A0=E5=85=A5=E5=AF=B9=E6=9C=BA?=
 =?UTF-8?q?=E6=A2=B0=E8=87=82=E5=8F=8D=E9=A6=88=E5=80=BC=E5=92=8C=E6=8E=A7?=
 =?UTF-8?q?=E5=88=B6=E9=87=8F=E7=9A=84=E5=8F=82=E6=95=B0=E5=BD=95=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/lbot_arm_control.sh | 120 ++++++++++++++++++++++++++++++++++++
 1 file changed, 120 insertions(+)

diff --git a/scripts/lbot_arm_control.sh b/scripts/lbot_arm_control.sh
index ca10ac5..a0bb1b9 100755
--- a/scripts/lbot_arm_control.sh
+++ b/scripts/lbot_arm_control.sh
@@ -6,7 +6,10 @@ WS_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"
 STATE_DIR="/tmp/lbot_arm_control"
 PID_FILE="${STATE_DIR}/lbot_driver.pid"
 PGID_FILE="${STATE_DIR}/lbot_driver.pgid"
+BAG_PID_FILE="${STATE_DIR}/rosbag.pid"
+BAG_PGID_FILE="${STATE_DIR}/rosbag.pgid"
 LOG_FILE="${WS_DIR}/log/lbot_arm_control.log"
+BAG_LOG_FILE="${WS_DIR}/log/lbot_arm_rosbag.log"
 
 usage() {
   echo "Usage: $0 {start|stop|status}"
@@ -20,6 +23,14 @@ is_running() {
   kill -0 "${pid}" 2>/dev/null
 }
 
+is_bag_running() {
+  [[ -f "${BAG_PID_FILE}" ]] || return 1
+  local pid
+  pid="$(cat "${BAG_PID_FILE}")"
+  [[ -n "${pid}" ]] || return 1
+  kill -0 "${pid}" 2>/dev/null
+}
+
 source_workspace() {
   set +u
   if [[ -n "${ROS_DISTRO:-}" && -f "/opt/ros/${ROS_DISTRO}/setup.bash" ]]; then
@@ -43,11 +54,109 @@ source_workspace() {
   set -u
 }
 
+start_bag_recorder() {
+  mkdir -p "${STATE_DIR}" "$(dirname "${BAG_LOG_FILE}")"
+
+  if is_bag_running; then
+    echo "rosbag recorder is already running, pid=$(cat "${BAG_PID_FILE}")"
+    return 0
+  fi
+
+  echo "Starting rosbag recorder..."
+  echo "Rosbag log: ${BAG_LOG_FILE}"
+
+  : > "${BAG_LOG_FILE}"
+  setsid bash -lc "
+    set -euo pipefail
+    set +u
+    source '${WS_DIR}/install/setup.bash'
+    set -u
+    while true; do
+      start_time=\\\$(date +%Y%m%d_%H%M%S)
+      output=/tmp/lbot_arm_\\${start_time}.mcap
+      echo \"Starting rosbag: \\${output}\"
+      set +e
+      timeout 3600 ros2 bag record \
+        --storage mcap \
+        --output \"\\${output}\" \
+        --polling-interval 2 \
+        --include-unpublished-topics \
+        --disable-keyboard-controls \
+        --log-level warn \
+        --topics \
+          /robot1/left_arm/joint_states \
+          /robot1/right_arm/joint_states \
+          /robot1/system_error \
+          /robot1/left_hand/set_l20_joint \
+          /robot1/right_hand/set_l20_joint \
+        --services \
+          /robot1/left_arm/move_joint \
+          /robot1/right_arm/move_joint
+      rc=\\$?
+      set -e
+      if [[ \\${rc} -ne 124 ]]; then
+        exit \\${rc}
+      fi
+    done
+  " >> "${BAG_LOG_FILE}" 2>&1 &
+
+  local pid=$!
+  local pgid
+  pgid="$(ps -o pgid= -p "${pid}" | tr -d ' ')"
+  echo "${pid}" > "${BAG_PID_FILE}"
+  echo "${pgid:-${pid}}" > "${BAG_PGID_FILE}"
+
+  sleep 1
+  if is_bag_running; then
+    echo "rosbag recorder started, pid=${pid}, pgid=$(cat "${BAG_PGID_FILE}")"
+  else
+    echo "rosbag recorder failed to start. Check log: ${BAG_LOG_FILE}"
+    rm -f "${BAG_PID_FILE}" "${BAG_PGID_FILE}"
+    return 1
+  fi
+}
+
+stop_bag_recorder() {
+  if ! [[ -f "${BAG_PID_FILE}" ]]; then
+    return 0
+  fi
+
+  local pid pgid
+  pid="$(cat "${BAG_PID_FILE}")"
+  pgid="${pid}"
+  if [[ -f "${BAG_PGID_FILE}" ]]; then
+    pgid="$(cat "${BAG_PGID_FILE}")"
+  fi
+
+  if ! kill -0 "${pid}" 2>/dev/null; then
+    rm -f "${BAG_PID_FILE}" "${BAG_PGID_FILE}"
+    return 0
+  fi
+
+  echo "Stopping rosbag recorder, pid=${pid}, pgid=${pgid}..."
+  kill -TERM -- "-${pgid}" 2>/dev/null || kill -TERM "${pid}" 2>/dev/null || true
+
+  for _ in {1..30}; do
+    if ! kill -0 "${pid}" 2>/dev/null; then
+      rm -f "${BAG_PID_FILE}" "${BAG_PGID_FILE}"
+      echo "rosbag recorder stopped"
+      return 0
+    fi
+    sleep 0.2
+  done
+
+  echo "rosbag recorder did not exit after SIGTERM, forcing stop..."
+  kill -KILL -- "-${pgid}" 2>/dev/null || kill -KILL "${pid}" 2>/dev/null || true
+  rm -f "${BAG_PID_FILE}" "${BAG_PGID_FILE}"
+  echo "rosbag recorder stopped"
+}
+
 start_driver() {
   mkdir -p "${STATE_DIR}" "$(dirname "${LOG_FILE}")"
 
   if is_running; then
     echo "lbot_driver is already running, pid=$(cat "${PID_FILE}")"
+    start_bag_recorder
     return 0
   fi
 
@@ -69,6 +178,7 @@ start_driver() {
   sleep 1
   if is_running; then
     echo "lbot_driver started, pid=${pid}, pgid=$(cat "${PGID_FILE}")"
+    start_bag_recorder
   else
     echo "lbot_driver failed to start. Check log: ${LOG_FILE}"
     rm -f "${PID_FILE}" "${PGID_FILE}"
@@ -79,6 +189,7 @@ start_driver() {
 stop_driver() {
   if ! [[ -f "${PID_FILE}" ]]; then
     echo "lbot_driver is not running"
+    stop_bag_recorder
     return 0
   fi
 
@@ -92,6 +203,7 @@ stop_driver() {
   if ! kill -0 "${pid}" 2>/dev/null; then
     echo "lbot_driver process is not alive, cleaning state"
     rm -f "${PID_FILE}" "${PGID_FILE}"
+    stop_bag_recorder
     return 0
   fi
 
@@ -102,6 +214,7 @@ stop_driver() {
     if ! kill -0 "${pid}" 2>/dev/null; then
       rm -f "${PID_FILE}" "${PGID_FILE}"
       echo "lbot_driver stopped"
+      stop_bag_recorder
       return 0
     fi
     sleep 0.2
@@ -111,6 +224,7 @@ stop_driver() {
   kill -KILL -- "-${pgid}" 2>/dev/null || kill -KILL "${pid}" 2>/dev/null || true
   rm -f "${PID_FILE}" "${PGID_FILE}"
   echo "lbot_driver stopped"
+  stop_bag_recorder
 }
 
 status_driver() {
@@ -119,6 +233,12 @@ status_driver() {
   else
     echo "lbot_driver is not running"
   fi
+
+  if is_bag_running; then
+    echo "rosbag recorder is running, pid=$(cat "${BAG_PID_FILE}"), pgid=$(cat "${BAG_PGID_FILE}" 2>/dev/null || cat "${BAG_PID_FILE}")"
+  else
+    echo "rosbag recorder is not running"
+  fi
 }
 
 case "${1:-}" in