pg_ctrl 0.0.8

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f29124242193bf5f9f5c2c2768f924f0a2f3a0252daf53eda8394892dbfa0058
+  data.tar.gz: bdeeee78f5f2d68177d98956508127b5c92555ac999da7319cdf54bf7b7ca1a1
+SHA512:
+  metadata.gz: ce9a19a47a13f8cf7d5ec8d755482e4093197b5ea4dff35d1ff1e7d165317b0a2b36f68ee17c0500b9ecb3c5cf4e7844a711c44f5d095b221fef4fc0d70e8bd5
+  data.tar.gz: 9431c871a9fc6001a486d1411b1b97431fcf5dfd7cc8461e942e80f2da792731f41cae2635038bd0cdf00821a12f9e60f8001ecaa9a5d7e9cbeecbeca3e3b95e

data/.gitignore

@@ -0,0 +1,9 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+.*.swp
+t
+tt
+ttt
+t.*

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--format progress

data/.ruby_version

@@ -0,0 +1 @@
+ruby-2.7.1

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in pg_ctrl.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/TODO

@@ -0,0 +1,3 @@
+
++ encoding, lc_collate, and lc_ctype from psql
+

data/lib/pg_ctrl/db.rb

@@ -0,0 +1,78 @@
+
+module PgCtrl
+  module Utils; end
+
+  class RDBMSObject
+    include Utils
+    attr_reader :rdbms
+    attr_reader :oid
+    attr_reader :name
+    attr_reader :owner
+    def initialize(rdbms, oid, name, owner, opts = {})
+      @rdbms, @oid, @name, @owner = rdbms, oid, name, owner
+    end
+    def to_s() @name end
+  end
+
+  class User < RDBMSObject
+    def superuser?() @is_superuser end
+    def can_create_db?() @can_create_db end
+    def can_create_role?() @can_create_role end
+    def can_login?() @can_login end
+    def initialize(rdbms, oid, name, opts = {})
+      super(rdbms, oid, name, self)
+      set_opts(opts, :is_superuser, :can_create_db, :can_create_role, 
+          :can_login)
+    end
+  end
+
+  class Database < RDBMSObject
+    attr_reader :encoding
+    attr_reader :lc_collate
+    attr_reader :lc_ctype
+    def template?() @is_template end
+    def can_connect?() @can_connect end
+    attr_reader :acl
+    def initialize(rdbms, oid, name, owner, opts = {}) 
+      super(rdbms, oid, name, owner)
+      set_opts(opts, :encoding, :lc_collate, :lc_ctype, :is_template,
+          :can_connect, :acl)
+    end
+  end
+
+  class RDBMS
+    include Utils
+
+    attr_reader :id
+    
+    attr_reader :connection
+    def ip?() !host.nil? end
+    def host() @connection.host end
+    def port() @connection.port.to_i end
+
+    # Users and databases indexed by name
+    attr_reader :users
+    attr_reader :databases
+
+    # Current user and database
+    attr_reader :user
+    attr_reader :database
+
+    # Arguments
+    #     initialize(PGconn args)
+    #     initialize(PGconn object)
+    #
+    def initialize(*args)
+      if args.size == 1 && PGconn === args[0]
+        @connection = args[0]
+      else
+        @connection = PGconn.new(args[0])
+      end
+      @id = (!host || host == "127.0.0.1" ? "localhost" : host)
+      load_data
+    end
+
+    def to_s() @id end
+  end
+end
+

data/lib/pg_ctrl/load.rb

@@ -0,0 +1,80 @@
+
+module PgCtrl
+  module Utils
+    def set_opts(opts, *attrs) 
+      attrs.map { |a| self.instance_variable_set("@#{a}", opts[a]) }
+    end
+  end
+
+  class RDBMSObjectSet
+    def initialize
+      @by_name = {}
+      @by_oid = {}
+    end
+
+    def size() @by_name.size end
+    def empty?() @by_name.empty? end
+
+    def [](key) Fixnum === key ? @by_oid[key] : @by_name[key] end
+    def add(value) @by_oid[value.oid] = @by_name[value.name] = value end
+
+    def keys() @by_name.keys end
+    def values() @by_name.values end
+
+    def each(&block) @by_name.each(&block) end
+    def map(&block) @by_name.map(&block) end
+  end
+
+  class RDBMS
+    LOAD_USERS_QUERY = %(
+      select  oid as OID,
+              rolname as NAME,
+              rolsuper as IS_SUPERUSER,
+              rolcreaterole as CAN_CREATE_ROLE,
+              rolcreatedb as CAN_CREATE_DB,
+              rolcanlogin as CAN_LOGIN,
+              case when rolname = current_user then true else false end as
+                  IS_CURRENT_USER
+      from    pg_roles
+    )
+
+    def load_users
+      @connection.each_hash(LOAD_USERS_QUERY) { |r|
+        @users.add User.new(self, r[:oid], r[:name], r)
+        @user = @users[r[:name]] if r[:is_current_user]
+      }
+    end
+
+    LOAD_DATABASES_QUERY = %(
+      select  pg_database.oid as OID,
+              datname as NAME,
+              rolname as OWNER,
+              pg_encoding_to_char(encoding) as ENCODING,
+              datcollate as LC_COLLATE,
+              datctype as LC_CTYPE,
+              datistemplate as IS_TEMPLATE,
+              datallowconn as CAN_CONNECT,
+              datacl as ACL,
+              case when datname = current_database() then true else false end as
+                  IS_CURRENT_DATABASE
+      from    pg_database
+      join    pg_roles on datdba = pg_roles.oid
+    )
+
+    def load_databases
+      @connection.each_hash(LOAD_DATABASES_QUERY) { |r|
+        @databases.add \
+            Database.new(self, r[:oid], r[:name], @users[r[:owner]], r)
+        @database = @databases[r[:name]] if r[:is_current_database]
+      }
+    end
+
+    def load_data
+      @users = RDBMSObjectSet.new
+      @databases = RDBMSObjectSet.new
+      load_users
+      load_databases
+    end
+  end
+end
+

data/lib/pg_ctrl/version.rb

@@ -0,0 +1,3 @@
+module PgCtrl
+  VERSION = "0.0.8"
+end

data/lib/pg_ctrl.rb

@@ -0,0 +1,10 @@
+require 'pg'
+
+require "pg_ctrl/version"
+
+require "pg_ctrl/db.rb"
+require "pg_ctrl/load.rb"
+
+module PgCtrl
+  # Your code goes here...
+end

data/main

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby
+
+PRJ_DIR = "/home/clr/prj"
+
+$: << "#{ENV['PWD']}/lib" << "#{PRJ_DIR}/pg_util/lib"
+
+require 'pg_util.rb'
+require 'pg_ctrl.rb'
+
+
+rdbms = PgCtrl::RDBMS.new
+
+print "Databases\n"
+rdbms.databases.values.each { |d|
+  print "   #{d}\n"
+  print "      OID: #{d.oid}\n"
+  print "      Owner: #{d.owner}\n"
+  print "      Encoding: #{d.encoding}\n"
+  print "      LC Collate: #{d.lc_collate}\n"
+  print "      LC CType: #{d.lc_ctype}\n"
+  print "      Template: #{d.template?}\n"
+  print "      Can connect: #{d.can_connect?}\n"
+}
+print "\n"
+
+print "Users\n"
+rdbms.users.values.each { |u|
+  print "   #{u}\n"
+  print "      OID: #{u.oid}\n"
+  print "      Superuser: #{u.superuser?}\n"
+  print "      Can create DB: #{u.can_create_db?}\n"
+  print "      Can create role: #{u.can_create_role?}\n"
+  print "      Can login: #{u.can_login?}\n"
+
+}
+print "\n"
+
+print "RDBMS #{rdbms}\n"
+print "   host: #{rdbms.host}\n"
+print "   port: #{rdbms.port}\n"
+print "   user: #{rdbms.user}\n"
+print "   database: #{rdbms.database}\n"
+print "\n"
+

data/pg_ctrl.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+require_relative "lib/pg_ctrl/version"
+
+Gem::Specification.new do |s|
+  s.name        = "pg_ctrl"
+  s.version     = PgCtrl::VERSION
+  s.authors     = ["Claus Rasmussen"]
+  s.email       = ["clr@veda.dk"]
+  s.homepage    = ""
+  s.summary     = "Control PostgreSQL server"
+  s.description = "PgCtrl provide an interface for querying and controlling a PostgreSQL server"
+
+  s.rubyforge_project = "pg_ctrl"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+end

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper.rb"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+end

metadata

@@ -0,0 +1,57 @@
+--- !ruby/object:Gem::Specification
+name: pg_ctrl
+version: !ruby/object:Gem::Version
+  version: 0.0.8
+platform: ruby
+authors:
+- Claus Rasmussen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2022-04-24 00:00:00.000000000 Z
+dependencies: []
+description: PgCtrl provide an interface for querying and controlling a PostgreSQL
+  server
+email:
+- clr@veda.dk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".ruby_version"
+- Gemfile
+- Rakefile
+- TODO
+- lib/pg_ctrl.rb
+- lib/pg_ctrl/db.rb
+- lib/pg_ctrl/load.rb
+- lib/pg_ctrl/version.rb
+- main
+- pg_ctrl.gemspec
+- spec/spec_helper.rb
+homepage: ''
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.4
+signing_key: 
+specification_version: 4
+summary: Control PostgreSQL server
+test_files:
+- spec/spec_helper.rb