From c793e8218d0195a669d545c810fa4b57c9eab44c Mon Sep 17 00:00:00 2001
From: Huw Walters <huw.walters@gmail.com>
Date: Sat, 31 Aug 2024 20:35:37 +0100
Subject: [PATCH] Comment shebang line before passing to tokenizer.

---
 src/compiler.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/compiler.rs b/src/compiler.rs
index 7155eb3b..bb2b20d6 100644
--- a/src/compiler.rs
+++ b/src/compiler.rs
@@ -9,7 +9,6 @@ use crate::{rules, Cli};
 use chrono::prelude::*;
 use colored::Colorize;
 use heraclitus_compiler::prelude::*;
-use itertools::Itertools;
 use std::env;
 use std::fs;
 use std::fs::File;
@@ -35,12 +34,12 @@ impl AmberCompiler {
             path,
             cli_opts,
         }
-        .load_code(AmberCompiler::strip_off_shebang(code))
+        .load_code(AmberCompiler::comment_shebang(code))
     }
 
-    fn strip_off_shebang(code: String) -> String {
+    fn comment_shebang(code: String) -> String {
         if code.starts_with("#!") {
-            code.split('\n').skip(1).collect_vec().join("\n")
+            String::from("// ") + &code
         } else {
             code
         }