:RAddParameterNB works

Author: killphi

plugin/refactorings/add_parameter.vim

@@ -8,16 +8,6 @@ function! AddParameter()
     return
   endtry
 
-  " brackets
-  if g:ruby_refactoring_brackets_have_space == 1
-    let bracket_open  = "( "
-    let bracket_close = " )"
-  else
-    let bracket_open  = "("
-    let bracket_close = ")"
-  endif
-
-
   " Save current position
   let cursor_position = getpos(".")
 
@@ -30,6 +20,15 @@ function! AddParameter()
   " remove trailing whitespace
   s/\v\s$//e
 
+  " brackets
+  if g:ruby_refactoring_brackets_have_space == 1
+    let bracket_open  = "( "
+    let bracket_close = " )"
+  else
+    let bracket_open  = "("
+    let bracket_close = ")"
+  endif
+
   " test the line
   let line = getline(".")
 

plugin/refactorings/add_parameter_without_using_brackets.vim

@@ -1,4 +1,3 @@
-
 " Synopsis:
 "   Adds a parameter (or many separated with commas) to a method with no
 "   brackets
@@ -19,7 +18,22 @@ function! AddParameterNB()
     exec "?\\<def\\>"
   endif
 
-  execute "normal A " . name . "\<Esc>"
+  " remove trailing whitespace
+  s/\v\s$//e
+
+  " remove brackets
+  s/\v\s*\( ?/ /e
+  s/\v\s*\)//e
+
+  " test the line
+  let has_parameter = match(getline("."), "\\vdef \\S+\\s+\\S+")
+
+
+  if has_parameter != -1
+    exec "normal A, " . name . "\<Esc>"
+  else
+    exec "normal A " . name . "\<Esc>"
+  endif
 
   " Restore caret position
   call setpos(".", cursor_position)