subplotlib-derive: Add support for location in step macro

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

1 files changed, 3 insertions, 2 deletions

diff --git a/subplotlib-derive/src/lib.rs b/subplotlib-derive/src/lib.rs
index 83e4adf..b10d3b3 100644
--- a/subplotlib-derive/src/lib.rs
+++ b/subplotlib-derive/src/lib.rs
@@ -350,10 +350,11 @@ fn process_step(mut input: ItemFn) -> proc_macro2::TokenStream {
             impl Builder {
                 #(#fieldfns)*
 
-                pub fn build(self, step_text: String) -> ScenarioStep {
+                pub fn build(self, step_text: String, location: &'static str) -> ScenarioStep {
                     ScenarioStep::new(step_text, move |ctx, _defuse_poison|
                         #builder_body,
-                        |scenario| register_contexts(scenario)
+                        |scenario| register_contexts(scenario),
+                        location,
                     )
                 }
             }