lib/bozo/packagers/rubygems.rb in bozo-scripts-0.2.0 vs lib/bozo/packagers/rubygems.rb in bozo-scripts-0.2.1

@@ -14,10 +14,22 @@
     end
 
     private
 
     def build_gem(spec)
-      execute_command :rubygems, ['gem', 'build', spec]
+      version_file = File.expand_path(File.dirname(File.realpath(__FILE__)) + '/../../../VERSION')
+
+      if pre_release?
+        File.open(version_file, 'w') { |f| f << "#{version}.pre#{env['GIT_HASH']}" }
+      end
+
+      begin
+        execute_command :rubygems, ['gem', 'build', spec]
+      ensure
+        if pre_release?
+          File.open(version_file, 'w') { |f| f << version.to_s }
+        end
+      end
     end
 
   end
 
 end
\ No newline at end of file