subplot: Properly support polyglot bindings

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

1 files changed, 10 insertions, 7 deletions

diff --git a/src/bin/subplot.rs b/src/bin/subplot.rs
index 22eda3b..034cbb2 100644
--- a/src/bin/subplot.rs
+++ b/src/bin/subplot.rs
@@ -310,7 +310,9 @@ impl Docgen {
         }
         let mut doc = cli::load_document(&self.input, style)?;
         doc.lint()?;
-        if !doc.check_named_files_exist()? {
+        let template = doc.meta().template_name().unwrap_or("").to_string();
+        if !doc.check_named_files_exist(&template)? || !doc.check_matched_steps_have_impl(&template)
+        {
             eprintln!("Continuing despite warnings");
         }
 
@@ -408,17 +410,18 @@ impl Codegen {
         let _enter = span.enter();
         let mut doc = cli::load_document(&self.filename, Style::default())?;
         doc.lint()?;
-        if !doc.check_named_files_exist()? {
-            eprintln!("Unable to continue");
-            std::process::exit(1);
-        }
-
-        let spec = template_spec(&doc)?;
         let template = doc
             .meta()
             .template_name()
             .ok_or_else(|| anyhow::anyhow!("No template name given"))?
             .to_string();
+        if !doc.check_named_files_exist(&template)? || !doc.check_matched_steps_have_impl(&template)
+        {
+            eprintln!("Unable to continue");
+            std::process::exit(1);
+        }
+
+        let spec = template_spec(&doc)?;
         generate_test_program(&mut doc, &spec, &self.output, &template)?;
 
         if self.run {