FIX brew diy --set-version --set-name

1 files changed, 17 insertions, 5 deletions

diff --git a/Library/Homebrew/brew.h.rb b/Library/Homebrew/brew.h.rb
index ef4561784..a9e0d28e7 100644
--- a/Library/Homebrew/brew.h.rb
+++ b/Library/Homebrew/brew.h.rb
@@ -188,12 +188,24 @@ end
 
 def diy
   path=Pathname.getwd
-  version=path.version
-  path.basename.to_s =~ /(.*?)-?#{version}/
-  name=$1
 
-  raise "Couldn't determine version, try --set-version" if version.nil? or version.empty?
-  raise "Couldn't determine name, try --set-name" if name.nil? or name.empty?
+  if ARGV.include? '--set-version'
+    version=ARGV.next
+  else
+    version=path.version
+    raise "Couldn't determine version, try --set-version" if version.nil? or version.empty?
+  end
+  
+  if ARGV.include? '--set-name'
+    name=ARGV.next
+  else
+    path.basename.to_s =~ /(.*?)-?#{version}/
+    if $1.nil? or $1.empty?
+      name=path.basename
+    else
+      name=$1
+    end
+  end
 
   prefix=HOMEBREW_CELLAR+name+version