pghero 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dc8ca524f848315726b63a987ca3a39ca7768d13
-  data.tar.gz: 230b213962882f7a62615f90fc9bc6b45cc97da5
+  metadata.gz: 408d3e02f0d03c8da95d11eaeea8a6c0fb8be19e
+  data.tar.gz: 557d0b3c72e111fe2355b80eda417ad731d2a345
 SHA512:
-  metadata.gz: 74956a91d498cf2b68d7f3dd261d7b727b7695d84a89851514ab8d31ce617f0c6005e18901a792aeb92c12e718ea4847518553ca74c6304f31f4b9aab7af2483
-  data.tar.gz: 0f63ae5e71393fab956c8d2df185acb4abbe3ce1aaba3b0fed637fc7404011f538b7070c8a0598824968fc7243bdce9845f8cf837b255a64cee3cc7ee5c2eae6
+  metadata.gz: e480e555168eaea71a6eada2f669058ee6097d0ac54e1812648025deeae343727a35f7c30287af8abb5c451d2de6d481fd9c099cf96e76fc92b7becd82740e00
+  data.tar.gz: 8bd10dc018837518f2f623575d2b6f1c2f975b493b1434800fa31a2f999b933bc70318751041431e7ae7d63dc83602c2982b439e6c569dcd0dfd4de7442ef566

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.1.6
+
+- Added methods to create and drop users
+- Added locks
+
 ## 0.1.5
 
 - Added system stats for Amazon RDS

data/README.md

@@ -8,6 +8,8 @@ Postgres insights made easy
 
 Supports PostgreSQL 9.2+
 
+:speech_balloon: Get [handcrafted updates](http://chartkick.us7.list-manage.com/subscribe?u=952c861f99eb43084e0a49f98&id=6ea6541e8e&group[0][32]=true) for new features
+
 For pure SQL, check out [PgHero.sql](https://github.com/ankane/pghero.sql)
 
 A big thanks to [Craig Kerstiens](http://www.craigkerstiens.com/2013/01/10/more-on-postgres-performance/) and [Heroku](https://blog.heroku.com/archives/2013/5/10/more_insight_into_your_database_with_pgextras) for the initial queries :clap:
@@ -41,12 +43,18 @@ PgHero.database_size
 PgHero.relation_sizes
 PgHero.index_hit_rate
 PgHero.table_hit_rate
+```
 
-# kill queries
+Kill queries
+
+```ruby
 PgHero.kill(pid)
 PgHero.kill_all
+```
+
+Query stats
 
-# query stats
+```ruby
 PgHero.query_stats_enabled?
 PgHero.enable_query_stats
 PgHero.disable_query_stats
@@ -55,6 +63,35 @@ PgHero.query_stats
 PgHero.slow_queries
 ```
 
+## Users
+
+Create a user
+
+```ruby
+PgHero.create_user("link")
+# {password: "zbTrNHk2tvMgNabFgCo0ws7T"}
+```
+
+This generates and returns a secure password.  The user has full access to the `public` schema.
+
+Read-only access
+
+```ruby
+PgHero.create_user("epona", readonly: true)
+```
+
+Set the password
+
+```ruby
+PgHero.create_user("zelda", password: "hyrule")
+```
+
+Drop a user
+
+```ruby
+PgHero.drop_user("ganondorf")
+```
+
 ## Security
 
 #### Basic Authentication
@@ -123,7 +160,7 @@ SELECT pg_stat_statements_reset();
 
 ## System Stats
 
-System stats are available for Amazon RDS.  Add these lines to your application’s Gemfile:
+CPU usage is available for Amazon RDS.  Add these lines to your application’s Gemfile:
 
 ```ruby
 gem 'aws-sdk'

data/app/views/layouts/pg_hero/application.html.erb

@@ -346,12 +346,12 @@
           <ul class="nav">
             <!-- poor man's active_link_to -->
             <li class="<%= controller.action_name == "index" ? "active" : "" %>"><%= link_to "Status", root_path %></li>
-            <% if @query_stats_enabled %>
-              <li class="<%= controller.action_name == "query_stats" ? "active" : "" %>"><%= link_to "Queries", query_stats_path %></li>
-            <% end %>
             <% if @system_stats_enabled %>
               <li class="<%= controller.action_name == "system_stats" ? "active" : "" %>"><%= link_to "System", system_stats_path %></li>
             <% end %>
+            <% if @query_stats_enabled %>
+              <li class="<%= controller.action_name == "query_stats" ? "active" : "" %>"><%= link_to "Queries", query_stats_path %></li>
+            <% end %>
             <li class="<%= controller.action_name == "indexes" ? "active" : "" %>"><%= link_to "Indexes", indexes_path %></li>
             <li class="<%= controller.action_name == "space" ? "active" : "" %>"><%= link_to "Space", space_path %></li>
             <li class="<%= controller.action_name == "explain" ? "active" : "" %>"><%= link_to "Explain", explain_path %></li>

data/app/views/pg_hero/home/system_stats.html.erb

@@ -2,5 +2,5 @@
   <h1>CPU Usage</h1>
 
   <%= javascript_include_tag "//www.google.com/jsapi", "chartkick" %>
-  <div style="margin-bottom: 20px;"><%= line_chart @cpu_usage %></div>
+  <div style="margin-bottom: 20px;"><%= line_chart @cpu_usage, max: 101 %></div>
 </div>

data/lib/pghero.rb

@@ -53,6 +53,26 @@ module PgHero
       }
     end
 
+    def locks
+      select_all %Q{
+        SELECT DISTINCT ON (pid)
+          pg_stat_activity.pid,
+          pg_stat_activity.query,
+          age(now(), pg_stat_activity.query_start) AS age
+        FROM
+          pg_stat_activity
+        INNER JOIN
+          pg_locks ON pg_locks.pid = pg_stat_activity.pid
+        WHERE
+          pg_stat_activity.query <> '<insufficient privilege>'
+          AND pg_locks.mode = 'ExclusiveLock'
+          AND pg_stat_activity.pid <> pg_backend_pid()
+        ORDER BY
+          pid,
+          query_start
+      }
+    end
+
     def index_hit_rate
       select_all(%Q{
         SELECT
@@ -297,6 +317,70 @@ module PgHero
       !!(defined?(AWS) && access_key_id && secret_access_key && db_instance_identifier)
     end
 
+    def random_password
+      require "securerandom"
+      SecureRandom.base64(40).delete("+/=")[0...24]
+    end
+
+    def create_user(user, options = {})
+      password = options[:password] || random_password
+      schema = options[:schema] || "public"
+      database = options[:database] || Connection.connection_config[:database]
+
+      commands =
+        [
+          "CREATE ROLE #{user} LOGIN PASSWORD #{Connection.connection.quote(password)}",
+          "GRANT CONNECT ON DATABASE #{database} TO #{user}",
+          "GRANT USAGE ON SCHEMA #{schema} TO #{user}"
+        ]
+      if options[:readonly]
+        commands << "GRANT SELECT ON ALL TABLES IN SCHEMA #{schema} TO #{user}"
+        commands << "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} GRANT SELECT ON TABLES TO #{user}"
+      else
+        commands << "GRANT ALL PRIVILEGES ON ALL TABLES IN SCHEMA #{schema} TO #{user}"
+        commands << "GRANT ALL PRIVILEGES ON ALL SEQUENCES IN SCHEMA #{schema} TO #{user}"
+        commands << "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} GRANT ALL PRIVILEGES ON TABLES TO #{user}"
+        commands << "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} GRANT ALL PRIVILEGES ON SEQUENCES TO #{user}"
+      end
+
+      # run commands
+      Connection.transaction do
+        commands.each do |command|
+          execute command
+        end
+      end
+
+      {password: password}
+    end
+
+    def drop_user(user, options = {})
+      schema = options[:schema] || "public"
+      database = options[:database] || Connection.connection_config[:database]
+
+      # thanks shiftb
+      commands =
+        [
+          "REVOKE CONNECT ON DATABASE #{database} FROM #{user}",
+          "REVOKE USAGE ON SCHEMA #{schema} FROM #{user}",
+          "REVOKE ALL PRIVILEGES ON ALL TABLES IN SCHEMA #{schema} FROM #{user}",
+          "REVOKE ALL PRIVILEGES ON ALL SEQUENCES IN SCHEMA #{schema} FROM #{user}",
+          "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} REVOKE SELECT ON TABLES FROM #{user}",
+          "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} REVOKE SELECT ON SEQUENCES FROM #{user}",
+          "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} REVOKE ALL ON SEQUENCES FROM #{user}",
+          "ALTER DEFAULT PRIVILEGES IN SCHEMA #{schema} REVOKE ALL ON TABLES FROM #{user}",
+          "DROP ROLE #{user}"
+        ]
+
+      # run commands
+      Connection.transaction do
+        commands.each do |command|
+          execute command
+        end
+      end
+
+      true
+    end
+
     def access_key_id
       ENV["PGHERO_ACCESS_KEY_ID"] || ENV["AWS_ACCESS_KEY_ID"]
     end

data/lib/pghero/version.rb

@@ -1,3 +1,3 @@
 module PgHero
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pghero
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-04 00:00:00.000000000 Z
+date: 2014-10-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -141,4 +141,3 @@ summary: Database insights made easy
 test_files:
 - test/pghero_test.rb
 - test/test_helper.rb
-has_rdoc: