Add set_password func

codenize-tools · Nov 2, 2015 · aec3716 · aec3716
1 parent bcd4eab
commit aec3716
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/lib/gratan/driver.rb b/lib/gratan/driver.rb
@@ -113,6 +113,25 @@ def identify(user, host, identifier)
     ]
 
     update(sql)
+
+    if (identifier || '').empty?
+      set_password(user, host, identifier)
+    end
+  end
+
+  def set_password(user, host, password, options = {})
+    password ||= ''
+
+    unless options[:hash]
+      password = "PASSWORD('#{escape(password)}')"
+    end
+
+    sql = 'SET PASSWORD FOR %s = %s' % [
+      quote_user(user, host),
+      password,
+    ]
+
+    update(sql)
   end
 
   def set_require(user, host, required)