add header/footer to template

Fixes #64

CHANGELOG.md

@@ -11,6 +11,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
   argument. [#24](https://github.com/OSC/ood_core/issues/24)
 - Added `OodCore::Job::Script#shell_path` as an option to all adapters.
   [#82](https://github.com/OSC/ood_core/issues/82)
+- Added `header` and `footer` options to a Batch Connect template.
+  [#64](https://github.com/OSC/ood_core/issues/64)
 
 ### Fixed
 - Replaced `Fixnum` code comments with `Integer`.

lib/ood_core/batch_connect/template.rb

@@ -26,6 +26,10 @@ class Template
       #   looking for available port
       # @option context [#to_i] :passwd_size (32) Length of randomly generated
       #   password
+      # @option context [#to_s] :header ("") Shell code prepended at the top of
+      #   the script body
+      # @option context [#to_s] :footer ("") Shell code appended at the bottom
+      #   of the script body
       # @option context [#to_s] :script_wrapper ("%s") Bash code that wraps
       #   around the body of the template script (use `%s` to interpolate the
       #   body)
@@ -58,11 +62,9 @@ def initialize(context = {})
       # @return [String] rendered template
       def to_s
         <<-EOT.gsub(/^ {10}/, '')
-          #
-          # Generated with ood_core v#{VERSION}
-          #
-
+          #{header}
           #{script_wrapper}
+          #{footer}
         EOT
       end
 
@@ -150,6 +152,16 @@ def bash_helpers
           end.to_s
         end
 
+        # Shell code that is prepended at the top of the script body
+        def header
+          context.fetch(:header, "").to_s
+        end
+
+        # Shell code that is appended at the bottom of the script body
+        def footer
+          context.fetch(:footer, "").to_s
+        end
+
         # Bash code that wraps around the body of the template script (use `%s`
         # to interpolate the body)
         def script_wrapper