make polars commands propagate metadata

crates/nu_plugin_polars/src/dataframe/command/aggregation/agg_groups.rs

@@ -60,6 +60,20 @@ impl PluginCommand for ExprAggGroups {
         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 ExprAggGroups {
+    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/aggregation/aggregate.rs

@@ -126,6 +126,20 @@ impl PluginCommand for LazyAggregate {
         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 LazyAggregate {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let vals: Vec<Value> = call.rest(0)?;
         let value = Value::list(vals, call.head);

crates/nu_plugin_polars/src/dataframe/command/aggregation/count.rs

@@ -55,6 +55,20 @@ impl PluginCommand for ExprCount {
         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 ExprCount {
+    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/aggregation/implode.rs

@@ -52,6 +52,20 @@ impl PluginCommand for ExprImplode {
         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 ExprImplode {
+    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/core/summary.rs

@@ -111,6 +111,20 @@ impl PluginCommand for Summary {
         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 Summary {
+    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/data/first.rs

@@ -96,6 +96,20 @@ impl PluginCommand for FirstDF {
         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 FirstDF {
+    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/data/join.rs

@@ -240,6 +240,20 @@ impl PluginCommand for LazyJoin {
         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 LazyJoin {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
+        input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let left = call.has_flag("left")?;
         let full = call.has_flag("full")?;

crates/nu_plugin_polars/src/dataframe/command/data/len.rs

@@ -55,6 +55,20 @@ impl PluginCommand for ExprLen {
         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 ExprLen {
+    fn run_inner(
+        &self,
+        plugin: &PolarsPlugin,
+        engine: &EngineInterface,
+        call: &EvaluatedCall,
         _input: PipelineData,
     ) -> Result<PipelineData, LabeledError> {
         let res: NuExpression = polars::prelude::len().into();

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

@@ -116,6 +116,20 @@ impl PluginCommand for StrSlice {
         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 StrSlice {
+    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)? {