allow db schema to be set in config

lib/common/configs.ex

@@ -10,7 +10,8 @@ defmodule Plsm.Common.Configs do
       database_name: Application.get_env(:plsm, :database_name, ""),
       username: Application.get_env(:plsm, :username, ""),
       password: Application.get_env(:plsm, :password, ""),
-      type: Application.get_env(:plsm, :type, :mysql)
+      type: Application.get_env(:plsm, :type, :mysql),
+      schema: Application.get_env(:plsm, :schema, "public")
     }
 
     project_config = %Plsm.Configs.Project{

lib/common/structs.ex

@@ -3,7 +3,13 @@ defmodule Plsm.Configs do
 end
 
 defmodule Plsm.Configs.Database do
-  defstruct server: "", port: "", database_name: "", username: "", password: "", type: :mysql
+  defstruct server: "",
+            port: "",
+            database_name: "",
+            username: "",
+            password: "",
+            type: :mysql,
+            schema: ""
 end
 
 defmodule Plsm.Configs.Project do

lib/config/config.ex

@@ -8,6 +8,7 @@ defmodule Plsm.Config.Config do
 
   def write(file_name) do
     config_exists? = File.exists?(file_name)
+
     case File.open(file_name, [:append]) do
       {:ok, file} -> IO.binwrite(file, output_config(config_exists?))
       _ -> {:error, "Could not open file #{file_name}. Please ensure that it exists."}
@@ -30,15 +31,17 @@ defmodule Plsm.Config.Config do
     current
     |> append_config_item_string("server", "localhost")
     |> append_next_item()
-    |> append_config_item_string("port", "3306")
+    |> append_config_item_string("port", "5432")
     |> append_next_item()
     |> append_config_item_string("database_name", "name of database")
     |> append_next_item()
-    |> append_config_item_string("username", "username")
+    |> append_config_item_string("username", "postgres")
+    |> append_next_item()
+    |> append_config_item_string("password", "postgres")
     |> append_next_item()
-    |> append_config_item_string("password", "password")
+    |> append_config_item_atom("type", "postgres")
     |> append_next_item()
-    |> append_config_item_atom("type", "mysql")
+    |> append_config_item_string("schema", "public")
   end
 
   defp project_config(current) do
@@ -71,6 +74,7 @@ defmodule Plsm.Config.Config do
      #    * username -> The username that is used to connect. Make sure that there is sufficient privileges to be able to connect, query tables as well as query information schemas on the database. The schema information is used to find the index/keys on each table
      #    * password -> This is necessary as there is no default nor is there any handling of a blank password currently.
      #    * type -> This dictates which database vendor you are using. We currently support PostgreSQL and MySQL. If no value is entered then it will default to MySQL. Do note that this is an atom and not a string
+     #    * schema -> The database schema namespace for the target tables.
     """
   end
 

lib/database/postgresql.ex

@@ -4,6 +4,7 @@ defmodule Plsm.Database.PostgreSQL do
             username: "postgres",
             password: "postgres",
             database_name: "db",
+            schema: "public",
             connection: nil
 end
 
@@ -15,7 +16,8 @@ defimpl Plsm.Database, for: Plsm.Database.PostgreSQL do
       port: configs.database.port,
       username: configs.database.username,
       password: configs.database.password,
-      database_name: configs.database.database_name
+      database_name: configs.database.database_name,
+      schema: configs.database.schema
     }
   end
 
@@ -30,13 +32,16 @@ defimpl Plsm.Database, for: Plsm.Database.PostgreSQL do
         database: db.database_name
       )
 
+    Postgrex.query!(conn, "SET search_path TO '#{db.schema}';", [])
+
     %Plsm.Database.PostgreSQL{
       connection: conn,
       server: db.server,
       port: db.port,
       username: db.username,
       password: db.password,
-      database_name: db.database_name
+      database_name: db.database_name,
+      schema: db.schema
     }
   end
 
@@ -46,7 +51,7 @@ defimpl Plsm.Database, for: Plsm.Database.PostgreSQL do
     {_, result} =
       Postgrex.query(
         db.connection,
-        "SELECT table_name FROM information_schema.tables WHERE table_schema = 'public';",
+        "SELECT table_name FROM information_schema.tables WHERE table_schema = '#{db.schema}';",
         []
       )
 

lib/io/export.ex

@@ -58,6 +58,7 @@ defmodule Plsm.IO.Export do
       module_declaration(project_name, table.header.name) <>
         model_inclusion() <>
         primary_key_disable() <>
+        schema_prefix_declaration() <>
         schema_declaration(table.header.name)
 
     trimmed_columns = remove_foreign_keys(table.columns)
@@ -99,6 +100,15 @@ defmodule Plsm.IO.Export do
     two_space("@primary_key false\n")
   end
 
+  defp schema_prefix_declaration do
+    configs = Plsm.Common.Configs.load_configs()
+
+    case configs.database.schema do
+      "public" -> ""
+      prefix -> two_space("@schema_prefix \"#{prefix}\"\n")
+    end
+  end
+
   defp schema_declaration(table_name) do
     two_space("schema \"#{table_name}\" do\n")
   end