diff --git a/r2r_common/src/lib.rs b/r2r_common/src/lib.rs index 764b91f..127f64f 100644 --- a/r2r_common/src/lib.rs +++ b/r2r_common/src/lib.rs @@ -148,13 +148,16 @@ pub fn get_wanted_messages() -> Vec { get_ros_msgs_files(&dirs) } else { // Else we look for all msgs we can find using the ament prefix path. - let ament_prefix_var = env::var("AMENT_PREFIX_PATH").expect("Source your ROS!"); - let paths = ament_prefix_var - .split(':') - .map(Path::new) - .collect::>(); + if let Ok(ament_prefix_var) = env::var("AMENT_PREFIX_PATH") { + let paths = ament_prefix_var + .split(':') + .map(Path::new) + .collect::>(); - get_ros_msgs(&paths) + get_ros_msgs(&paths) + } else { + vec![] + } }; let msgs = parse_msgs(&msgs);