ezii-postgres 0.0.1

data/natural-backend/db/migrate/20180430103015_add_user_ref_to_columns.rb

@@ -0,0 +1,5 @@
+class AddUserRefToColumns < ActiveRecord::Migration[5.1]
+  def change
+    add_reference :columns, :user, foreign_key: true
+  end
+end

data/natural-backend/db/migrate/20180430103024_add_user_ref_to_row_values.rb

@@ -0,0 +1,5 @@
+class AddUserRefToRowValues < ActiveRecord::Migration[5.1]
+  def change
+    add_reference :row_values, :user, foreign_key: true
+  end
+end

data/natural-backend/db/migrate/20180430141537_add_api_token_projects.rb

@@ -0,0 +1,5 @@
+class AddApiTokenProjects < ActiveRecord::Migration[5.1]
+  def change
+    add_column :projects, :api_token, :string
+  end
+end

data/natural-backend/db/schema.rb

@@ -0,0 +1,112 @@
+# This file is auto-generated from the current state of the database. Instead
+# of editing this file, please use the migrations feature of Active Record to
+# incrementally modify your database, and then regenerate this schema definition.
+#
+# Note that this schema.rb definition is the authoritative source for your
+# database schema. If you need to create the application database on another
+# system, you should be using db:schema:load, not running all the migrations
+# from scratch. The latter is a flawed and unsustainable approach (the more migrations
+# you'll amass, the slower it'll run and the greater likelihood for issues).
+#
+# It's strongly recommended that you check this file into your version control system.
+
+ActiveRecord::Schema.define(version: 20180430141537) do
+
+  # These are extensions that must be enabled in order to support this database
+  enable_extension "plpgsql"
+
+  create_table "columns", force: :cascade do |t|
+    t.string "name"
+    t.string "type"
+    t.bigint "table_id"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.bigint "user_id"
+    t.index ["table_id"], name: "index_columns_on_table_id"
+    t.index ["user_id"], name: "index_columns_on_user_id"
+  end
+
+  create_table "databases", force: :cascade do |t|
+    t.string "name"
+    t.bigint "project_id"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.string "database_identifier"
+    t.bigint "user_id"
+    t.index ["database_identifier"], name: "index_databases_on_database_identifier", unique: true
+    t.index ["project_id"], name: "index_databases_on_project_id"
+    t.index ["user_id"], name: "index_databases_on_user_id"
+  end
+
+  create_table "projects", force: :cascade do |t|
+    t.string "name"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.string "db_username"
+    t.string "db_password"
+    t.bigint "user_id"
+    t.string "api_token"
+    t.index ["user_id"], name: "index_projects_on_user_id"
+  end
+
+  create_table "queries", force: :cascade do |t|
+    t.text "request_data"
+    t.text "response_data"
+    t.bigint "database_id"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.index ["database_id"], name: "index_queries_on_database_id"
+  end
+
+  create_table "row_values", force: :cascade do |t|
+    t.bigint "row_id"
+    t.bigint "column_id"
+    t.text "value"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.bigint "user_id"
+    t.index ["column_id"], name: "index_row_values_on_column_id"
+    t.index ["row_id"], name: "index_row_values_on_row_id"
+    t.index ["user_id"], name: "index_row_values_on_user_id"
+  end
+
+  create_table "rows", force: :cascade do |t|
+    t.bigint "table_id"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.integer "db_id"
+    t.bigint "user_id"
+    t.index ["table_id"], name: "index_rows_on_table_id"
+    t.index ["user_id"], name: "index_rows_on_user_id"
+  end
+
+  create_table "tables", force: :cascade do |t|
+    t.string "name"
+    t.bigint "database_id"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.bigint "user_id"
+    t.index ["database_id"], name: "index_tables_on_database_id"
+    t.index ["user_id"], name: "index_tables_on_user_id"
+  end
+
+  create_table "users", force: :cascade do |t|
+    t.string "email"
+    t.string "password_digest"
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+  end
+
+  add_foreign_key "columns", "tables"
+  add_foreign_key "columns", "users"
+  add_foreign_key "databases", "projects"
+  add_foreign_key "databases", "users"
+  add_foreign_key "projects", "users"
+  add_foreign_key "row_values", "columns"
+  add_foreign_key "row_values", "rows"
+  add_foreign_key "row_values", "users"
+  add_foreign_key "rows", "tables"
+  add_foreign_key "rows", "users"
+  add_foreign_key "tables", "databases"
+  add_foreign_key "tables", "users"
+end

data/natural-backend/db/seeds.rb

@@ -0,0 +1,7 @@
+# This file should contain all the record creation needed to seed the database with its default values.
+# The data can then be loaded with the rails db:seed command (or created alongside the database with db:setup).
+#
+# Examples:
+#
+#   movies = Movie.create([{ name: 'Star Wars' }, { name: 'Lord of the Rings' }])
+#   Character.create(name: 'Luke', movie: movies.first)

data/natural-backend/lib/database_manager/database_manager.rb

@@ -0,0 +1,59 @@
+module Natural
+  class DatabaseManager
+    attr_reader :connection
+
+    def initialize
+      establish_default_connection
+    end
+
+    def connect_to_database(database_name)
+      @connection.close
+      @connection.database = ::Natural::Database.new(database_name)
+      @connection.establish_connection
+      @connection.database.connection = @connection
+      @connection
+    end
+
+    def establish_default_connection
+      @connection = ::Natural::Connection.new
+      @connection.load_rails_database_config
+      @connection.establish_connection
+    end
+
+    def database_user(username, password = nil)
+      database_user = ::Natural::DatabaseUser.new(username, password)
+      database_user.connection = connection
+      database_user
+    end
+
+    def create_user(username, password)
+      database_user(username, password).create
+    end
+
+    def destroy_user(username)
+      database_user(username).destroy
+    end
+
+    def user_exists?(username)
+      database_user(username).exists?
+    end
+
+    def database(identifier)
+      database = ::Natural::Database.new(identifier)
+      database.connection = connection
+      database
+    end
+
+    def create_database(identifier)
+      database(identifier).create
+    end
+
+    def destroy_database(identifier)
+      database(identifier).destroy
+    end
+
+    def database_exists?(identifier)
+      database(identifier).exists?
+    end
+  end
+end

data/natural-backend/lib/database_manager/lib/connection.rb

@@ -0,0 +1,49 @@
+module Natural
+  class Connection
+    attr_accessor :db_user
+    attr_accessor :database
+
+    def close
+      @connection.close
+    end
+
+    def establish_connection
+      options_hash = {
+        user: @db_user.username,
+        password: @db_user.password
+      }
+
+      if database.present?
+        options_hash.merge!({dbname: database.identifier})
+      end
+      
+      @connection = PG.connect(options_hash)
+
+    end
+
+    def load_rails_database_config
+      config = Rails.configuration.database_configuration[Rails.env]
+
+      @db_user = ::Natural::DatabaseUser.new(config['username'],
+                                            config['password'])
+    end
+
+    def exec(*args, &block)
+      @connection.exec(*args, &block)
+    end
+
+    def clone_with_database(database)
+      clone = self.class.new
+      clone.db_user = self.db_user
+      clone.database = database
+      clone.establish_connection
+      clone
+    end
+  end
+
+  module ConnectionProvidable
+    def self.included(base)
+      attr_accessor :connection
+    end
+  end
+end

data/natural-backend/lib/database_manager/lib/database.rb

@@ -0,0 +1,65 @@
+require_relative './connection.rb'
+
+module Natural
+  class Database
+    include ::Natural::ConnectionProvidable
+
+    attr_reader :identifier
+
+    def initialize(identifier)
+      @identifier = identifier
+    end
+
+    def table(table_identifier)
+      table = Table.new(table_identifier)
+      table.connection = connection.clone_with_database(self)
+      table.database = self
+      table
+    end
+
+    def create_table(table_identifier)
+      table(table_identifier).create
+    end
+
+    def destroy_table(table_identifier)
+      table(table_identifier).destroy
+    end
+
+    def table_exists?(table_identifier)
+      table(table_identifier).exists?
+    end
+
+    def tables(associated_db_user)
+      connection.exec(
+        """
+        SELECT tablename FROM pg_catalog.pg_tables WHERE tableowner = \'#{associated_db_user}\';
+        """
+      )
+    end
+
+    def create
+      connection.exec(
+        """
+        CREATE DATABASE \"#{@identifier}\"
+        """
+      )
+      self
+    end
+
+    def destroy
+      connection.exec(
+        """
+        DROP DATABASE \"#{@identifier}\";
+        """
+      )
+    end
+
+    def exists?
+      '1' == connection.exec(
+        """
+        SELECT 1 FROM pg_database WHERE pg_database.datname = \'#{@identifier}\';
+        """
+      ).values[0].try(:[], 0)
+    end
+  end
+end

data/natural-backend/lib/database_manager/lib/database_user.rb

@@ -0,0 +1,46 @@
+require_relative './connection.rb'
+
+module Natural
+  class DatabaseUser
+    include ::Natural::ConnectionProvidable
+    attr_reader :username, :password
+
+    def initialize(username, password = nil)
+      @username = username
+      @password = password
+    end
+
+    def create
+      connection.exec(
+        """
+        CREATE USER \"#{@username}\" WITH PASSWORD \'#{@password}\'
+        """
+      )
+      self
+    end
+
+    def destroy
+      connection.exec(
+        """
+        DROP USER \"#{@username}\";
+        """
+      )
+    end
+
+    def exists?
+      '1' == connection.exec(
+        """
+        SELECT 1 FROM pg_roles WHERE rolname='#{@username}'
+        """
+      ).values[0].try(:[], 0)
+    end
+
+    def grant(database)
+      connection.exec(
+        """
+        GRANT ALL PRIVILEGES ON DATABASE \"#{database.identifier}\" TO \"#{@username}\";
+        """
+      )
+    end
+  end
+end

data/natural-backend/lib/database_manager/lib/table.rb

@@ -0,0 +1,129 @@
+require_relative './connection.rb'
+
+module Natural
+  class Table
+    TYPE_IDENTIFIER_MAPPING = {
+      'varchar_255' => 'varchar(255)',
+    }
+
+    include ::Natural::ConnectionProvidable
+
+    attr_reader :identifier
+    attr_accessor :database
+
+    def initialize(identifier)
+      @identifier = identifier
+    end
+
+    def create
+      connection.exec(
+        """
+        CREATE TABLE \"#{@identifier}\" (
+          id SERIAL
+        );
+        """
+      )
+      self
+    end
+
+    def destroy
+      connection.exec(
+        """
+        DROP TABLE \"#{@identifier}\";
+        """
+      )
+    end
+
+    def exists?
+      '1' == connection.exec(
+        """
+        SELECT 1
+        FROM pg_tables
+        WHERE tablename = '#{@identifier}'
+        """
+      ).values[0].try(:[], 0)
+    end
+
+    def add_column(name)
+      connection.exec(
+        """
+        ALTER TABLE \"#{@identifier}\"
+          ADD COLUMN \"#{name}\" varchar(255);
+        """
+      )
+    end
+
+    def update_column_type(name, type_identifier)
+      type = TYPE_IDENTIFIER_MAPPING[type_identifier] || type_identifier
+
+      connection.exec(
+        """
+        ALTER TABLE \"#{@identifier}\"
+          ALTER COLUMN \"#{name}\" TYPE #{type} USING (trim(\"#{name}\" )::#{type});
+        """
+      )
+    end
+
+    def destroy_column(name)
+      connection.exec(
+        """
+        ALTER TABLE \"#{@identifier}\"
+          DROP COLUMN \"#{name}\" RESTRICT;
+        """
+      )
+    end
+
+    def insert_value(column_name, value)
+      connection.exec(
+        """
+        INSERT INTO \"#{@identifier}\" (\"#{column_name}\")
+        VALUES ('#{single_quote_escape(value)}')
+        RETURNING ID;
+        """
+      ).values[0][0]
+    end
+
+    def update_value(column_name, id, value)
+      connection.exec(
+        """
+        UPDATE \"#{@identifier}\"
+        SET \"#{column_name}\" = \'#{single_quote_escape(value)}\'
+        WHERE id=#{id};
+        """
+      )
+    end
+
+    def single_quote_escape(string)
+      string.gsub("'", %q(''))
+    end
+
+    def delete_value(column_name, id)
+      connection.exec(
+        """
+        UPDATE \"#{@identifier}\"
+        SET \"#{column_name}\" = NULL
+        WHERE id=#{id};
+        """
+      )
+    end
+
+    def has_row?(id)
+      connection.exec(
+        """
+        SELECT 1
+        FROM \"#{@identifier}\"
+        WHERE id=#{id};
+        """
+      ).values[0][0] == '1'
+    end
+
+    def delete_row(id)
+      connection.exec(
+        """
+        DELETE FROM \"#{@identifier}\"
+        WHERE id=#{id};
+        """
+      )
+    end
+  end
+end