lib/docman/commands/yaml_execute_cmd.rb in docman-0.0.81 vs lib/docman/commands/yaml_execute_cmd.rb in docman-0.0.82

@@ -18,12 +18,12 @@
         Dir.chdir @context['full_build_path']
         if self['environments'].nil? || self['environments'] == 'all' || self['environments'].include?(@context.environment_name)
           if self['providers'].nil? || self['providers'] == 'all' || self['providers'].include?(@context['provider'])
             commands = nil
             if self['source_type'] == 'file'
-              yaml_file_name = self['yaml_file_name'].nil? ? '.build.yml' : self['yaml_file_name']
-              build_file = File.join(@context['full_build_path'], yaml_file_name)
-              if File.file? build_file
+              yaml_file_name = self['yaml_file_name'].nil? ? '{unipipe,.unipipe,drupipe,.drupipe,build,.build}.{yaml,yml}' : self['yaml_file_name']
+              build_file = Dir.glob([File.join(@context['docroot_config'].config_dir, yaml_file_name), File.join(@context['full_build_path'], yaml_file_name)]).first
+              if not build_file.nil?
                 build_file_yaml = YAML::load_file(build_file)
                 commands = build_file_yaml[self['stage']]
                 source = yaml_file_name
               end
             end
\ No newline at end of file