- Add before/after hooks

Author: DannyBen

lib/bashly/concerns/renderable.rb

@@ -10,6 +10,7 @@ def strings
       @strings ||= MessageStrings.new
     end
 
+    # Outputs a comment that describes the view unless in production mode
     def view_marker(id = nil)
       id ||= ":#{caller_locations(1..1).first.path}"
       "# #{id}" unless Settings.production?
@@ -31,6 +32,9 @@ def load_user_file(file, placeholder: true)
       Settings.production? ? content : "#{view_marker path}\n#{content}"
     end
 
+    # Returns a path to a file in the user's source_dir. The file argument
+    # should either be without exteneion, or with the user's configured
+    # partials_extension.
     def user_file_path(file)
       path = "#{Settings.source_dir}/#{file}"
       ext = ".#{Settings.partials_extension}"
@@ -39,6 +43,11 @@ def user_file_path(file)
       "#{path}#{ext}"
     end
 
+    # Returns true if the user's source file exists
+    def user_file_exist?(file)
+      File.exist? user_file_path(file)
+    end
+
   private
 
     def view_path(view)

lib/bashly/views/command/master_script.gtx

@@ -8,6 +8,7 @@
 = render :user_lib if user_lib.any?
 = render :command_functions
 = render :parse_requirements
+= render :user_hooks
 = render :initialize
 = render :run
 

lib/bashly/views/command/run.gtx

@@ -7,26 +7,26 @@
 >   declare -a input=()
 >   normalize_input "$@"
 >   parse_requirements "${input[@]}"
+if user_file_exist?('before')
+  >   before_hook
+end
 > 
 >   case "$action" in
 
 deep_commands.each do |command|
-  >     "{{ command.action_name }}")
-  >       if [[ ${args['--help']:-} ]]; then
-  >         long_usage=yes
-  >         {{ command.function_name }}_usage
-  >       else
-  >         {{ command.function_name }}_command
-  >       fi
-  >       ;;
-  >
+  >     "{{ command.action_name }}") {{ command.function_name }}_command ;;
 end
 
 if commands.empty?
-  >     "root")
-  >       root_command
-  >       ;;
+  >     "root") root_command ;;
 end
->
 >   esac
+
+if user_file_exist?('after')
+  >
+  >   after_hook
+end
+
 > }
+
+

lib/bashly/views/command/user_hooks.gtx

@@ -0,0 +1,17 @@
+if user_file_exist?('before') || user_file_exist?('after')
+  = view_marker
+end
+
+if user_file_exist?('before')
+  > before_hook() {
+  = load_user_file("before").indent 2
+  > }
+  >
+end
+
+if user_file_exist?('after')
+  > after_hook() {
+  = load_user_file("after").indent 2
+  > }
+  >
+end

spec/bashly/script/command_spec.rb

@@ -353,6 +353,26 @@
     end
   end
 
+  describe '#user_file_exist?' do
+    before { FileUtils.mkdir_p 'spec/tmp/src' }
+
+    context 'when the file exists in the user source path' do
+      before { FileUtils.touch 'spec/tmp/src/test.sh' }
+
+      it 'returns true' do
+        expect(subject.user_file_exist?('test')).to be true
+      end
+    end
+
+    context 'when the file does not in the user source path' do
+      before { FileUtils.rm_f 'spec/tmp/src/test.sh' }
+
+      it 'returns false' do
+        expect(subject.user_file_exist?('test')).to be false
+      end
+    end
+  end
+
   describe '#required_args' do
     it 'returns an array of only the required Argument objects' do
       expect(subject.required_args.size).to eq 1