posgra 0.1.0

data/lib/posgra/dsl/grants.rb

@@ -0,0 +1,53 @@
+class Posgra::DSL::Grants
+  include Posgra::Logger::Helper
+  include Posgra::TemplateHelper
+  include Posgra::Utils::Helper
+
+  def self.eval(dsl, path, options = {})
+    self.new(path, options) do
+      eval(dsl, binding, path)
+    end
+  end
+
+  attr_reader :result
+
+  def initialize(path, options = {}, &block)
+    @path = path
+    @options = options
+    @result = {}
+
+    @context = Hashie::Mash.new(
+      :path => path,
+      :options => options,
+      :templates => {},
+    )
+
+    instance_eval(&block)
+  end
+
+  private
+
+  def template(name, &block)
+    @context.templates[name.to_s] = block
+  end
+
+  def require(file)
+    pgrantfile = (file =~ %r|\A/|) ? file : File.expand_path(File.join(File.dirname(@path), file))
+
+    if File.exist?(pgrantfile)
+      instance_eval(File.read(pgrantfile), pgrantfile)
+    elsif File.exist?(pgrantfile + '.rb')
+      instance_eval(File.read(pgrantfile + '.rb'), pgrantfile + '.rb')
+    else
+      Kernel.require(file)
+    end
+  end
+
+  def role(name, &block)
+    name = name.to_s
+
+    if matched?(name, @options[:include_role], @options[:exclude_role])
+      @result[name] = Posgra::DSL::Grants::Role.new(@context, name, @options, &block).result
+    end
+  end
+end

data/lib/posgra/dsl/grants/role.rb

@@ -0,0 +1,23 @@
+class Posgra::DSL::Grants::Role
+  include Posgra::Logger::Helper
+  include Posgra::TemplateHelper
+  include Posgra::Utils::Helper
+
+  attr_reader :result
+
+  def initialize(context, role, options, &block)
+    @role = role
+    @options = options
+    @context = context.merge(:role => role)
+    @result = {}
+    instance_eval(&block)
+  end
+
+  def schema(name, &block)
+    name = name.to_s
+
+    if matched?(name, @options[:include_schema], @options[:exclude_schema])
+      @result[name] = Posgra::DSL::Grants::Role::Schema.new(@context, name, @options, &block).result
+    end
+  end
+end

data/lib/posgra/dsl/grants/role/schema.rb

@@ -0,0 +1,22 @@
+class Posgra::DSL::Grants::Role::Schema
+  include Posgra::Logger::Helper
+  include Posgra::TemplateHelper
+
+  attr_reader :result
+
+  def initialize(context, schema, options, &block)
+    @schema = schema
+    @options = options
+    @context = context.merge(:schema => schema)
+    @result = {}
+    instance_eval(&block)
+  end
+
+  def on(name, &block)
+    unless name.is_a?(Regexp)
+      name = name.to_s
+    end
+
+    @result[name] = Posgra::DSL::Grants::Role::Schema::On.new(@context, name, @options, &block).result
+  end
+end

data/lib/posgra/dsl/grants/role/schema/on.rb

@@ -0,0 +1,22 @@
+class Posgra::DSL::Grants::Role::Schema::On
+  include Posgra::Logger::Helper
+  include Posgra::TemplateHelper
+
+  attr_reader :result
+
+  def initialize(context, object, options, &block)
+    @object = object
+    @options = options
+    @context = context.merge(:object => object)
+    @result = {}
+    instance_eval(&block)
+  end
+
+  def grant(name, options = {})
+    name = name.to_s
+
+    @result[name] = {
+      'is_grantable' => !!options[:grantable]
+    }
+  end
+end

data/lib/posgra/dsl/roles.rb

@@ -0,0 +1,67 @@
+class Posgra::DSL::Roles
+  include Posgra::Logger::Helper
+  include Posgra::TemplateHelper
+  include Posgra::Utils::Helper
+
+  def self.eval(dsl, path, options = {})
+    self.new(path, options) do
+      eval(dsl, binding, path)
+    end
+  end
+
+  def result
+    @result.fetch(:users).uniq
+    @result
+  end
+
+  def initialize(path, options = {}, &block)
+    @path = path
+    @options = options
+    @result = {
+      :users => [],
+      :users_by_group => {},
+    }
+
+    @context = Hashie::Mash.new(
+      :path => path,
+      :options => options,
+      :templates => {},
+    )
+
+    instance_eval(&block)
+  end
+
+  private
+
+  def template(name, &block)
+    @context.templates[name.to_s] = block
+  end
+
+  def require(file)
+    pgrantfile = (file =~ %r|\A/|) ? file : File.expand_path(File.join(File.dirname(@path), file))
+
+    if File.exist?(pgrantfile)
+      instance_eval(File.read(pgrantfile), pgrantfile)
+    elsif File.exist?(pgrantfile + '.rb')
+      instance_eval(File.read(pgrantfile + '.rb'), pgrantfile + '.rb')
+    else
+      Kernel.require(file)
+    end
+  end
+
+  def user(name, &block)
+    name = name.to_s
+
+    if matched?(name, @options[:include_role], @options[:exclude_role])
+      @result[:users] << name
+    end
+  end
+
+  def group(name, &block)
+    name = name.to_s
+
+    if matched?(name, @options[:include_role], @options[:exclude_role])
+      @result[:users_by_group][name] = Posgra::DSL::Roles::Group.new(@context, name, @options, &block).result
+    end
+  end
+end

data/lib/posgra/dsl/roles/group.rb

@@ -0,0 +1,24 @@
+class Posgra::DSL::Roles::Group
+  include Posgra::Logger::Helper
+  include Posgra::TemplateHelper
+  include Posgra::Utils::Helper
+
+  attr_reader :result
+
+  def initialize(context, group, options, &block)
+    @group = group
+    @options = options
+    @context = context.merge(:group => group)
+    @result = []
+    instance_eval(&block) if block
+  end
+
+  def user(name)
+    name = name.to_s
+
+    if matched?(name, @options[:include_role], @options[:exclude_role])
+      name = name.kind_of?(Regexp) ? name : name.to_s
+      @result << name
+    end
+  end
+end

data/lib/posgra/exporter.rb

@@ -0,0 +1,25 @@
+class Posgra::Exporter
+  def self.export_roles(driver, options = {}, &block)
+    self.new(driver, options).export_roles(&block)
+  end
+
+  def self.export_grants(driver, options = {}, &block)
+    self.new(driver, options).export_grants(&block)
+  end
+
+  def initialize(driver, options = {})
+    @driver = driver
+    @options = options
+  end
+
+  def export_roles
+    {
+      :users => @driver.describe_users.keys,
+      :users_by_group => @driver.describe_groups,
+    }
+  end
+
+  def export_grants
+    @driver.describe_grants
+  end
+end

data/lib/posgra/ext/string_ext.rb

@@ -0,0 +1,25 @@
+class String
+  @@colorize = false
+
+  class << self
+    def colorize=(value)
+      @@colorize = value
+    end
+
+    def colorize
+      @@colorize
+    end
+  end # of class methods
+
+  Term::ANSIColor::Attribute.named_attributes.map do |attribute|
+    class_eval(<<-EOS, __FILE__, __LINE__ + 1)
+      def #{attribute.name}
+        if @@colorize
+          Term::ANSIColor.send(#{attribute.name.inspect}, self)
+        else
+          self
+        end
+      end
+    EOS
+  end
+end

data/lib/posgra/identifier.rb

@@ -0,0 +1,2 @@
+module Posgra::Identifier
+end

data/lib/posgra/identifier/auto.rb

@@ -0,0 +1,38 @@
+class Posgra::Identifier::Auto
+  def initialize(output, options = {})
+    @output = output
+    @options = options
+  end
+
+  def identify(user)
+    password = mkpasswd
+    puts_password(user, password)
+    password
+  end
+
+  private
+
+  def mkpasswd(len = 8)
+    [*1..9, *'A'..'Z', *'a'..'z'].shuffle.slice(0, len).join
+  end
+
+  def puts_password(user, password)
+    open_output do |f|
+      f.puts("#{user},#{password}")
+    end
+  end
+
+  def open_output
+    return if @options[:dry_run]
+
+    if @output == '-'
+      yield($stdout)
+      $stdout.flush
+    else
+      open(@output, 'a') do |f|
+        yield(f)
+        f.flush
+      end
+    end
+  end
+end

data/lib/posgra/logger.rb

@@ -0,0 +1,33 @@
+class Posgra::Logger < ::Logger
+  include Singleton
+
+  def initialize
+    super($stdout)
+
+    self.formatter = proc do |severity, datetime, progname, msg|
+      "#{msg}\n"
+    end
+
+    self.level = Logger::INFO
+  end
+
+  def set_debug(value)
+    self.level = value ? Logger::DEBUG : Logger::INFO
+  end
+
+  module Helper
+    def log(level, message, options = {})
+      global_options = @options || {}
+      message = "#{@object_identifier}: #{message}" if @object_identifier
+      message = "[#{level.to_s.upcase}] #{message}" unless level == :info
+
+      if global_options[:dry_run] and options[:dry_run] != false
+        message << ' (dry-run)' if global_options[:dry_run]
+      end
+
+      message = message.send(options[:color]) if options[:color]
+      logger = global_options[:logger] || Posgra::Logger.instance
+      logger.send(level, message)
+    end
+  end
+end

data/lib/posgra/template.rb

@@ -0,0 +1,18 @@
+module Posgra::TemplateHelper
+  def include_template(template_name, context = {})
+    tmplt = @context.templates[template_name.to_s]
+
+    unless tmplt
+      raise "Template `#{template_name}` is not defined"
+    end
+
+    context_orig = @context
+    @context = @context.merge(context)
+    instance_eval(&tmplt)
+    @context = context_orig
+  end
+
+  def context
+    @context
+  end
+end

data/lib/posgra/utils.rb

@@ -0,0 +1,17 @@
+class Posgra::Utils
+  module Helper
+    def matched?(name, include_r, exclude_r)
+      result = true
+
+      if exclude_r
+        result &&= name !~ exclude_r
+      end
+
+      if include_r
+        result &&= name =~ include_r
+      end
+
+      result
+    end
+  end
+end

data/lib/posgra/version.rb

@@ -0,0 +1,3 @@
+module Posgra
+  VERSION = "0.1.0"
+end

data/posgra.gemspec

@@ -0,0 +1,32 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'posgra/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'posgra'
+  spec.version       = Posgra::VERSION
+  spec.authors       = ['winebarrel']
+  spec.email         = ['sgwr_dts@yahoo.co.jp']
+
+  spec.summary       = %q{Posgra is a tool to manage PostgreSQL roles/permissions.}
+  spec.description   = %q{Posgra is a tool to manage PostgreSQL roles/permissions.}
+  spec.homepage      = 'https://github.com/winebarrel/posgra'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'pg'
+  spec.add_dependency 'term-ansicolor'
+  spec.add_dependency 'hashie'
+  spec.add_dependency 'thor'
+
+  spec.add_development_dependency 'bundler'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_development_dependency 'coveralls'
+  spec.add_development_dependency 'unindent'
+end

metadata

@@ -0,0 +1,206 @@
+--- !ruby/object:Gem::Specification
+name: posgra
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- winebarrel
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-01-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: term-ansicolor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hashie
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: coveralls
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: unindent
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Posgra is a tool to manage PostgreSQL roles/permissions.
+email:
+- sgwr_dts@yahoo.co.jp
+executables:
+- posgra
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/posgra
+- lib/posgra.rb
+- lib/posgra/cli.rb
+- lib/posgra/cli/app.rb
+- lib/posgra/cli/grant.rb
+- lib/posgra/cli/helper.rb
+- lib/posgra/cli/role.rb
+- lib/posgra/client.rb
+- lib/posgra/driver.rb
+- lib/posgra/dsl.rb
+- lib/posgra/dsl/converter.rb
+- lib/posgra/dsl/grants.rb
+- lib/posgra/dsl/grants/role.rb
+- lib/posgra/dsl/grants/role/schema.rb
+- lib/posgra/dsl/grants/role/schema/on.rb
+- lib/posgra/dsl/roles.rb
+- lib/posgra/dsl/roles/group.rb
+- lib/posgra/exporter.rb
+- lib/posgra/ext/string_ext.rb
+- lib/posgra/identifier.rb
+- lib/posgra/identifier/auto.rb
+- lib/posgra/logger.rb
+- lib/posgra/template.rb
+- lib/posgra/utils.rb
+- lib/posgra/version.rb
+- posgra.gemspec
+homepage: https://github.com/winebarrel/posgra
+licenses:
+- MIT
+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: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Posgra is a tool to manage PostgreSQL roles/permissions.
+test_files: []