prick 0.9.0 → 0.14.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3009c0da5619e908746d98eefe922851d886105d10b6bc49cc0295cf47a50033
-  data.tar.gz: 3ee5cd54a3b29171306c3ad2fe8ea10c2e6d1ad84e1b8446b6314770c6734aef
+  metadata.gz: c155692c268326430a0e9225d4a7c156fedd52533114b62fae7e7ce61da01177
+  data.tar.gz: 4e245212208c3ecc518f313cd9f1af709db5ce41b7281c7247a1078293c486de
 SHA512:
-  metadata.gz: 7ee1ce2cd13293542db575cbbf5d955f542eff9b3cc6e4b87027800e7b0effb12ca134df2665f6a2a240cb0c191cd2fc9e5cca9b943c307591b499e75d27b5db
-  data.tar.gz: 7f81e321f4875bce049b5adf7eb7fe7394a8112135d13b73ebdd48680b014329517e7137df419f504861a2d855243d37dfcafc2f67226fef57c537e28dd27dde
+  metadata.gz: 38ff771885f6a73a20f36985403cb61a63fc2f025c20f92bb1dea4cb2ba047f908e48c4a564a50b707594d32199a87f796e0640135f6f0da5a521fe572c70cdb
+  data.tar.gz: 79641d89e168fe1be4d7493aed420fd40cd71cfb29ac98c359c34f82457bd1b7c265883e83cf661c29ff1ea6437bf1a087a652987409221cc4c439768b480922

data/exe/prick

@@ -195,6 +195,13 @@ begin
     exit 0
   end
 
+  # Change to parent directory containing the Prick version file if not found
+  # in the current directory
+  program.current_directory = Dir.getwd
+  while !Dir.getwd != "/" && !File.exist?(PRICK_VERSION_FILE)
+    Dir.chdir("..")
+  end
+
   # Check prick version
   file = PrickVersion.new
   file.exist? or raise Prick::Error, "Can't find prick version file '#{file.path}'"
@@ -251,7 +258,7 @@ begin
       program.save(version, file)
 
     when :drop
-      program.drop(args.expect(0..1), opts.drop!.all)
+      program.drop(args.expect(0..1), opts.drop!.all?)
 
     when :diff
       mark = opts.diff!.mark

data/lib/prick/builder.rb

@@ -7,18 +7,24 @@ module Prick
   # and is used to match a build file. Build files are looked up in the
   # following order:
   #
-  #   #{name}, #{name}.* executable
+  #   #{name} executable
+  #   #{name}.* executable
   #   #{name}.yml
   #   #{name}.sql
   #   #{name}/
   #
+  # The output from executable objects is expected to be SQL statements that
+  # are fed into postgres
+  #
   # When a resource match a directory, the directory can contain a special
   # default resource ('build' or 'migrate') that takes over the rest of the
   # build process for that directory.  Typically, you'll use the 'build.yml' or
-  # 'migrate.yml' to control the build order of the other resources
+  # 'migrate.yml' to control the build order of the other resources If a
+  # directory doesn't contain a build resource, the resources in the directory
+  # are built in alphabetic order
   #
-  # If a directory doesn't contain a build resource, the resources in the
-  # directory are built in alphabetic order
+  # Build (but not migration) SQL scripts and executables are executed with
+  # search path set to containing schema
   #
   class Builder
     attr_reader :database
@@ -57,11 +63,11 @@ module Prick
       Dir.chdir(dir) { yield(file) }
     end
 
-    def do_sql(path)
+    def do_sql(path, schema: nil)
 #     puts "do_sql(#{path})"
       do_dir(path) { |file| 
         if @execute
-          Rdbms.exec_file(database.name, file, user: database.user) 
+          Rdbms.exec_file(database.name, file, user: database.user, schema: schema) 
         else
           @lines += IO.readlines(file).map(&:chomp)
         end
@@ -69,12 +75,12 @@ module Prick
       true
     end
 
-    def do_exe(path)
+    def do_exe(path, schema: nil)
 #     puts "do_exe(#{path})"
       do_dir(path) { |file| 
-        lines = Command.command "#{file} #{database.name} #{database.user}" # FIXME Security
+        lines = Command.command "./#{file} #{database.name} #{database.user}" # FIXME Security
         if @execute
-          Rdbms.exec_sql(database.name, lines.join("\n"), database.user)
+          Rdbms.exec_sql(database.name, lines.join("\n"), user: database.user, schema: schema)
         else
           @lines += lines
         end
@@ -159,6 +165,17 @@ module Prick
     end
 
     def self.yml_file(directory) File.join(directory, "build") + ".yml" end
+
+  protected
+    def do_sql(path)
+      schema = Dir.getwd.sub(/^.*schema\/([^\/]+)(?:\/.*)?$/, '\1')
+      super(path, schema: schema)
+    end
+
+    def do_exe(path)
+      schema = Dir.getwd.sub(/^.*schema\/([^\/]+)(?:\/.*)?$/, '\1')
+      super(path, schema: schema)
+    end
   end
 end
 

data/lib/prick/database.rb

@@ -59,7 +59,23 @@ module Prick
       @version = nil 
     end
 
-    def clean() recreate end # TODO: Find solution that doesn't involve dropping the database
+    # Hollow-out a database without dropping it. This is useful compared to a
+    # simple drop database since it wont block on other sessions wont block
+    def clean() 
+      if exist?
+        schemas = Rdbms.select_values(
+                      "postgres", 
+                      "select nspname from pg_namespace where nspowner != 10")
+        for schema in schemas
+          Rdbms.exec_sql(name, "drop schema \"#{schema}\" cascade")
+        end
+        Rdbms.exec_sql(name, "drop schema if exists public cascade")
+        Rdbms.exec_sql(name, "create schema public authorization postgres")
+        Rdbms.exec_sql(name, "grant usage, create on schema public to public")
+      else
+        create
+      end
+    end
 
     def to_s() name end
   end

data/lib/prick/program.rb

@@ -10,8 +10,10 @@ module Prick
 
     attr_accessor :quiet
     attr_accessor :verbose
+    attr_accessor :current_directory # Path to the directory where prick was called from
 
-    def initialize(quiet: false, verbose: false)
+    def initialize(current_directory = Dir.getwd, quiet: false, verbose: false)
+      @current_directory = current_directory
       @quiet = quiet
       @verbose = verbose
     end
@@ -95,8 +97,9 @@ module Prick
         database = database ? Database.new(database, project.user) : project.database(version)
         project.load(database, version: version)
         mesg "Loaded v#{version}", into_mesg, "from cache"
-      else file = file_or_version
-        project.load(database, file: file)
+      else 
+        file = file_or_version
+        project.load(database, file: norm_path(file))
         mesg "Loaded #{File.basename(file)}", into_mesg
       end
     end
@@ -104,9 +107,8 @@ module Prick
     def save(version, file)
       database = project.database(version)
       database.exist? or raise "Can't find database '#{database}'"
-      subj_mesg = file ? file : "cache"
-      project.save(database, file: file)
-      mesg "Saved #{database} to #{subj_mesg}"
+      project.save(database, file: norm_path(file))
+      mesg "Saved #{database} to #{file || 'cache'}"
     end
 
     def drop(database, all)
@@ -230,15 +232,16 @@ module Prick
 
     # TODO: Create a Backup class and a Project#backup_store object
     def backup(file = nil) 
-      file = file || File.join(SPOOL_DIR, "#{project.name}-#{Time.now.utc.strftime("%Y%m%d-%H%M%S")}.sql.gz")
-      project.save(file: file) 
+      path = norm_path(file) || 
+          File.join(SPOOL_DIR, "#{project.name}-#{Time.now.utc.strftime("%Y%m%d-%H%M%S")}.sql.gz")
+      project.save(file: path) 
       mesg "Backed-up database to #{file}"
     end
 
-    def restore(file_arg = nil)
-      file = file_arg || Dir.glob(File.join(SPOOL_DIR, "#{name}-*.sql.gz")).sort.last
-      File.exist?(file) or raise Error, "Can't find #{file_arg || "any backup file"}"
-      project.load(file: file)
+    def restore(file = nil)
+      path =  norm_path(file) || Dir.glob(File.join(SPOOL_DIR, "#{name}-*.sql.gz")).sort.last
+      File.exist?(path) or raise Error, "Can't find #{file || "any backup file"}"
+      project.load(file: path)
       mesg "Restored database from #{file}"
     end
 
@@ -249,6 +252,16 @@ module Prick
       project_name == project.name or raise Error, "Database not owned by this prick project: #{database}"
     end
 
+    # Expand relative path with respect to the directory where prick was called
+    # from. Return nil if file is nil
+    def norm_path(file)
+      if file && !file.start_with?("/")
+        File.join(current_directory, file)
+      else
+        file
+      end
+    end
+
     def mesg(*args) puts args.compact.grep(/\S/).join(' ') if !quiet end
     def verb(*args) puts args.compact.grep(/\S/).join(' ') if verbose end
   end

data/lib/prick/rdbms.rb

@@ -4,16 +4,15 @@ require 'csv'
 
 module Prick
   module Rdbms
-#   extend Ensure
-
     ### EXECUTE SQL
 
     # Execute the SQL statement and return stdout as an array of tuples
-    def self.exec_sql(db, sql, user: ENV['USER'])
+    def self.exec_sql(db, sql, user: ENV['USER'], schema: nil)
+      schema ||= "public"
       stdout = Command.command %(
         {
           echo "set role #{user};"
-          echo "set search_path to public;"
+          echo "set search_path to #{schema};"
           cat <<'EOF'
           #{sql}
 EOF
@@ -23,8 +22,8 @@ EOF
     end
 
     # Execute the given file and return stdout as an array of tuples
-    def self.exec_file(db, file, user: ENV['USER'])
-      self.exec_sql(db, File.read(file), user: user)
+    def self.exec_file(db, file, user: ENV['USER'], schema: nil)
+      self.exec_sql(db, File.read(file), user: user, schema: schema)
     end
 
     # Execute the SQL statement and return the result as an array of record tuples

data/lib/prick/version.rb

@@ -4,7 +4,7 @@
 
 # Required by gem
 module Prick
-  VERSION = "0.9.0"
+  VERSION = "0.14.0"
 end
 
 # Project related code starts here

data/share/schema/schema.sql

@@ -1,5 +1,3 @@
 
 drop schema if exists [<SCHEMA>] cascade;
 create schema [<SCHEMA>];
-
-set search_path to [<SCHEMA>];

data/share/schema/schema/prick/schema.sql

@@ -1,5 +1,3 @@
 
 drop schema if exists prick cascade;
 create schema prick;
-
-set search_path to prick;

data/share/schema/schema/public/build.yml

@@ -1,7 +1,6 @@
 # Default schema files
 #
 ---
-- schema 
 - roles 
 - types 
 - tables

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: prick
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.14.0
 platform: ruby
 authors:
 - Claus Rasmussen
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-02-10 00:00:00.000000000 Z
+date: 2021-03-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: shellopts
@@ -133,7 +133,6 @@ files:
 - share/schema/schema/prick/tables.sql
 - share/schema/schema/public/.keep
 - share/schema/schema/public/build.yml
-- share/schema/schema/public/schema.sql
 - test_assorted
 - test_feature
 - test_refactor
@@ -156,7 +155,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.1.2
 signing_key: 
 specification_version: 4
 summary: A release control and management system for postgresql

data/share/schema/schema/public/schema.sql

@@ -1,3 +0,0 @@
-set search_path to public;
-
-