make polars commands propagate metadata

crates/nu_plugin_polars/src/dataframe/command/datetime/as_date.rs

@@ -182,6 +182,20 @@ impl PluginCommand for AsDate {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl AsDate {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/as_datetime.rs

@@ -247,6 +247,20 @@ impl PluginCommand for AsDateTime {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl AsDateTime {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/convert_time_zone.rs

@@ -136,6 +136,20 @@ impl PluginCommand for ConvertTimeZone {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ConvertTimeZone {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
 

crates/nu_plugin_polars/src/dataframe/command/datetime/datepart.rs

@@ -130,6 +130,20 @@ impl PluginCommand for ExprDatePart {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ExprDatePart {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let part: Spanned<String> = call.req(0)?;
 

crates/nu_plugin_polars/src/dataframe/command/datetime/get_day.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetDay {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetDay {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_hour.rs

@@ -99,6 +99,20 @@ impl PluginCommand for GetHour {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetHour {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_minute.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetMinute {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetMinute {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_month.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetMonth {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetMonth {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_nanosecond.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetNanosecond {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetNanosecond {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_ordinal.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetOrdinal {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetOrdinal {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_second.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetSecond {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetSecond {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_week.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetWeek {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetWeek {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_weekday.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetWeekDay {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetWeekDay {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/get_year.rs

@@ -83,6 +83,20 @@ impl PluginCommand for GetYear {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl GetYear {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/datetime/replace_time_zone.rs

@@ -215,6 +215,20 @@ impl PluginCommand for ReplaceTimeZone {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ReplaceTimeZone {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
 

crates/nu_plugin_polars/src/dataframe/command/datetime/strftime.rs

@@ -93,6 +93,20 @@ impl PluginCommand for StrFTime {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl StrFTime {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/datetime/truncate.rs

@@ -149,6 +149,20 @@ impl PluginCommand for Truncate {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl Truncate {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         command(plugin, engine, call, input).map_err(LabeledError::from)
     }

crates/nu_plugin_polars/src/dataframe/command/integer/to_decimal.rs

@@ -71,6 +71,20 @@ impl PluginCommand for ToDecimal {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ToDecimal {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/integer/to_integer.rs

@@ -65,6 +65,20 @@ impl PluginCommand for ToInteger {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ToInteger {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/list/contains.rs

@@ -115,6 +115,20 @@ impl PluginCommand for ListContains {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ListContains {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/concat_str.rs

@@ -84,6 +84,20 @@ impl PluginCommand for ExprConcatStr {
         plugin: &Self::Plugin,
         engine: &EngineInterface,
         call: &EvaluatedCall,
+        input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ExprConcatStr {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
         _input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let separator: String = call.req(0)?;

crates/nu_plugin_polars/src/dataframe/command/string/contains.rs

@@ -99,6 +99,20 @@ impl PluginCommand for Contains {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl Contains {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/replace.rs

@@ -106,6 +106,20 @@ impl PluginCommand for Replace {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl Replace {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/replace_all.rs

@@ -107,6 +107,20 @@ impl PluginCommand for ReplaceAll {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ReplaceAll {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/str_join.rs

@@ -92,6 +92,20 @@ impl PluginCommand for StrJoin {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl StrJoin {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/str_lengths.rs

@@ -88,6 +88,20 @@ impl PluginCommand for StrLengths {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl StrLengths {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/str_split.rs

@@ -58,6 +58,20 @@ impl PluginCommand for StrSplit {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl StrSplit {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let separator = call.req::<Spanned<Value>>(0).and_then(|sep| {
             let sep_expr = NuExpression::try_from_value(plugin, &sep.item)?;

crates/nu_plugin_polars/src/dataframe/command/string/str_strip_chars.rs

@@ -132,6 +132,20 @@ impl PluginCommand for StrStripChars {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl StrStripChars {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/to_lowercase.rs

@@ -89,6 +89,20 @@ impl PluginCommand for ToLowerCase {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ToLowerCase {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {

crates/nu_plugin_polars/src/dataframe/command/string/to_uppercase.rs

@@ -93,6 +93,20 @@ impl PluginCommand for ToUpperCase {
         engine: &EngineInterface,
         call: &EvaluatedCall,
         input: PipelineData,
+    ) -> Result<PipelineData, LabeledError> {
+        let metadata = input.metadata();
+        self.run_inner(plugin, engine, call, input)
+            .map(|pd| pd.set_metadata(metadata))
+    }
+}
+
+impl ToUpperCase {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let value = input.into_value(call.head)?;
         match PolarsPluginObject::try_from_value(plugin, &value)? {