aboutsummaryrefslogtreecommitdiffstats
path: root/lib/dbshell/client
diff options
context:
space:
mode:
authorTeddy Wing2017-04-29 21:55:48 +0200
committerTeddy Wing2017-04-29 21:55:48 +0200
commitb700c1bef89a67a64f1040fb6bb03c0320eefe91 (patch)
treec7173b62b8a7d6f33c41544e14139c8b5f4a6a5a /lib/dbshell/client
parentcf14e894685c695c7ecbaa2c119e2ea414afc00a (diff)
downloaddbshell-rails-b700c1bef89a67a64f1040fb6bb03c0320eefe91.tar.bz2
Rename database client files to put them in a `Client` module
Will be modifying the class names in the next commit. Here we change the file names & locations of the database client files. The format changes thusly: <adapter>_client.rb -> client/<adapter>.rb This organisation feels cleaner to me. Wasn't liking the previous structure.
Diffstat (limited to 'lib/dbshell/client')
-rw-r--r--lib/dbshell/client/database.rb24
-rw-r--r--lib/dbshell/client/postgres.rb23
-rw-r--r--lib/dbshell/client/sqlite3.rb15
3 files changed, 62 insertions, 0 deletions
diff --git a/lib/dbshell/client/database.rb b/lib/dbshell/client/database.rb
new file mode 100644
index 0000000..edea777
--- /dev/null
+++ b/lib/dbshell/client/database.rb
@@ -0,0 +1,24 @@
+require 'dbshell/sqlite3_client'
+require 'dbshell/postgres_client'
+
+class DBShell::DatabaseClient
+ def self.handler(connection_params)
+ case connection_params['adapter']
+ when 'sqlite3'
+ DBShell::Sqlite3Client
+ when 'postgresql'
+ DBShell::PostgresClient
+ else
+ raise DBShell::InvalidDatabaseAdapter
+ end
+ end
+
+ def self.runshell(connection_params)
+ self
+ .handler(connection_params)
+ .runshell(connection_params)
+ end
+end
+
+
+class DBShell::InvalidDatabaseAdapter < StandardError; end
diff --git a/lib/dbshell/client/postgres.rb b/lib/dbshell/client/postgres.rb
new file mode 100644
index 0000000..1c1c663
--- /dev/null
+++ b/lib/dbshell/client/postgres.rb
@@ -0,0 +1,23 @@
+class DBShell::PostgresClient
+ EXECUTABLE_NAME = 'psql'
+
+ def self.runshell(db_info)
+ args = self.build_command(db_info)
+ exec(*args)
+ end
+
+ def self.build_command(db_info)
+ args = [EXECUTABLE_NAME]
+
+ host = db_info['host']
+ port = db_info['port']
+ db_name = db_info['database']
+ user = db_info['username']
+ password = db_info['password']
+
+ args.push('-U', user) if user
+ args.push('-h', host) if host
+ args.push('-p', port) if port
+ args.push(db_name)
+ end
+end
diff --git a/lib/dbshell/client/sqlite3.rb b/lib/dbshell/client/sqlite3.rb
new file mode 100644
index 0000000..adc71ef
--- /dev/null
+++ b/lib/dbshell/client/sqlite3.rb
@@ -0,0 +1,15 @@
+class DBShell::Sqlite3Client
+ EXECUTABLE_NAME = 'sqlite3'
+
+ def self.runshell(db_info)
+ args = self.build_command(db_info)
+ exec(*args)
+ end
+
+ def self.build_command(db_info)
+ args = [
+ EXECUTABLE_NAME,
+ db_info['database']
+ ]
+ end
+end