scriptcs · adamralph · Mar 17, 2015 · Mar 13, 2015 · Mar 13, 2015 · Mar 17, 2015
diff --git a/src/ScriptCs.Core/ScriptCs.Core.csproj b/src/ScriptCs.Core/ScriptCs.Core.csproj
@@ -94,6 +94,7 @@
     <Compile Include="ScriptExecutor.cs" />
     <Compile Include="ScriptServices.cs" />
     <Compile Include="SessionState.cs" />
+    <Compile Include="ShebangLineProcessor.cs" />
     <Compile Include="StringExtensions.cs" />
     <Compile Include="UsingLineProcessor.cs" />
   </ItemGroup>

diff --git a/src/ScriptCs.Core/ShebangLineProcessor.cs b/src/ScriptCs.Core/ShebangLineProcessor.cs
@@ -0,0 +1,21 @@
+using ScriptCs.Contracts;
+
+namespace ScriptCs
+{
+    public interface IShebangLineProcessor : ILineProcessor
+    {
+    }
+
+    public class ShebangLineProcessor : DirectiveLineProcessor, IShebangLineProcessor
+    {
+        protected override string DirectiveName
+        {
+            get { return "!/usr/bin/env"; }
+        }
+
+        protected override bool ProcessLine(IFileParser parser, FileParserContext context, string line)
+        {
+            return true;
+        }
+    }
+}
diff --git a/src/ScriptCs.Hosting/ScriptServicesRegistration.cs b/src/ScriptCs.Hosting/ScriptServicesRegistration.cs
@@ -62,7 +62,11 @@ protected void RegisterLineProcessors(ContainerBuilder builder)
                 .FirstOrDefault(x => typeof(IReferenceLineProcessor).IsAssignableFrom(x))
                 ?? typeof(ReferenceLineProcessor);
 
-            var processorArray = new[] { loadProcessorType, usingProcessorType, referenceProcessorType }
+            var shebangProcessorType = processorList
+                .FirstOrDefault(x => typeof(IShebangLineProcessor).IsAssignableFrom(x))
+                ?? typeof(ShebangLineProcessor);
+
+            var processorArray = new[] { loadProcessorType, usingProcessorType, referenceProcessorType, shebangProcessorType }
                 .Union(processorList).ToArray();
 
             builder.RegisterTypes(processorArray).As<ILineProcessor>();

diff --git a/src/ScriptCs/ScriptCs.csproj b/src/ScriptCs/ScriptCs.csproj
@@ -125,6 +125,9 @@
       <Generator>SettingsSingleFileGenerator</Generator>
       <LastGenOutput>Settings.Designer.cs</LastGenOutput>
     </None>
+    <None Include="scriptcs">
+    	<CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
   </ItemGroup>
   <ItemGroup>
     <Content Include="..\..\common\Icon.ico">

diff --git a/src/ScriptCs/scriptcs b/src/ScriptCs/scriptcs
@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+## Make sure to chmod +x this file!
+
+DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+
+mono "$DIR/scriptcs.exe" $@
diff --git a/test/ScriptCs.Core.Tests/FileProcessorTests.cs b/test/ScriptCs.Core.Tests/FileProcessorTests.cs
@@ -16,6 +16,7 @@ public class ProcessFileMethod
         {
             private List<string> _file1 = new List<string>
                 {
+                    @"#!/usr/bin/env scriptcs",
                     @"#load ""script2.csx""",
                     @"#load ""script4.csx"";",
                     "using System;",
@@ -184,6 +185,24 @@ public void ShouldNotIncludeReferencesInCode()
                 result.Code.ShouldNotContain("#r");
             }
 
+            [Fact]
+            public void ShouldNotIncludeShebangsInCode()
+            {
+                var file1 = new List<string>
+                    {
+                        @"#!/usr/bin/env scriptcs",
+                        "using System;",
+                        @"Console.WriteLine(""Hi!"");"
+                    };
+
+                _fileSystem.Setup(x => x.ReadFileLines(It.Is<string>(f => f == "script1.csx"))).Returns(file1.ToArray());
+
+                var processor = GetFilePreProcessor();
+                var result = processor.ProcessFile("script1.csx");
+
+                result.Code.ShouldNotContain("#!/usr/bin/env");
+            }
+
             [Fact]
             public void ShouldNotLoadSameFileTwice()
             {
@@ -502,7 +521,8 @@ private IFilePreProcessor GetFilePreProcessor()
                 {
                     new UsingLineProcessor(),
                     new ReferenceLineProcessor(_fileSystem.Object),
-                    new LoadLineProcessor(_fileSystem.Object)
+                    new LoadLineProcessor(_fileSystem.Object),
+                    new ShebangLineProcessor()
                 };
 
                 return new FilePreProcessor(_fileSystem.Object, Mock.Of<ILog>(), lineProcessors);
@@ -565,7 +585,8 @@ private IFilePreProcessor GetFilePreProcessor()
                 {
                     new UsingLineProcessor(),
                     new ReferenceLineProcessor(_fileSystem.Object),
-                    new LoadLineProcessor(_fileSystem.Object)
+                    new LoadLineProcessor(_fileSystem.Object),
+                    new ShebangLineProcessor()
                 };
 
                 return new FilePreProcessor(_fileSystem.Object, Mock.Of<ILog>(), lineProcessors);
@@ -622,6 +643,7 @@ private IFilePreProcessor GetFilePreProcessor(ILineProcessor customDirectiveProc
                         new UsingLineProcessor(),
                         new ReferenceLineProcessor(_fileSystem.Object),
                         loadLineProcessor,
+                        new ShebangLineProcessor(),
                         customDirectiveProcessor
                     };
 

diff --git a/test/ScriptCs.Core.Tests/ScriptCs.Core.Tests.csproj b/test/ScriptCs.Core.Tests/ScriptCs.Core.Tests.csproj
@@ -74,6 +74,7 @@
     <Compile Include="ScriptHostTests.cs" />
     <Compile Include="ScriptPackResolverTests.cs" />
     <Compile Include="ScriptPackSessionTests.cs" />
+    <Compile Include="ShebangLineProcessorTests.cs" />
     <Compile Include="UsingLineProcessorTests.cs" />
   </ItemGroup>
   <ItemGroup>

diff --git a/test/ScriptCs.Core.Tests/ShebangLineProcessorTests.cs b/test/ScriptCs.Core.Tests/ShebangLineProcessorTests.cs
@@ -0,0 +1,38 @@
+using Should;
+using Xunit.Extensions;
+using ScriptCs.Contracts;
+
+namespace ScriptCs.Tests
+{
+    public class ShebangLineProcessorTests
+    {
+        public class TheProcessLineMethod
+        {
+            [Theory, ScriptCsAutoData]
+            public void ShouldReturnTrueOnShebangLine(IFileParser parser, ShebangLineProcessor processor)
+            {
+                // Arrange
+                const string Line = @"#!/usr/bin/env scriptcs";
+
+                // Act
+                var result = processor.ProcessLine(parser, new FileParserContext(), Line, true);
+
+                // Assert
+                result.ShouldBeTrue();
+            }
+
+            [Theory, ScriptCsAutoData]
+            public void ShouldReturnFalseOtherwise(IFileParser parser, ShebangLineProcessor processor)
+            {
+                // Arrange
+                const string Line = @"var x = new Test();";
+
+                // Act
+                var result = processor.ProcessLine(parser, new FileParserContext(), Line, true);
+
+                // Assert
+                result.ShouldBeFalse();
+            }
+        }
+    }
+}
diff --git a/test/ScriptCs.Hosting.Tests/RuntimeServicesTests.cs b/test/ScriptCs.Hosting.Tests/RuntimeServicesTests.cs
@@ -156,6 +156,7 @@ public void ShouldRegisterTheDefaultLineProcessors()
                 processors.Where(p => p is IUsingLineProcessor).ShouldNotBeEmpty();
                 processors.Where(p => p is IReferenceLineProcessor).ShouldNotBeEmpty();
                 processors.Where(p => p is ILoadLineProcessor).ShouldNotBeEmpty();
+                processors.Where(p => p is IShebangLineProcessor).ShouldNotBeEmpty();
             }
 
             [Fact]