pgai 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6ec24e1bc611522aea3c490acbb0f94b02085584544c979cd56ae9a3bd45f61b
-  data.tar.gz: 7351d32aa09341ec0081e0d4c45466d83034d1ec3ec776b74bef7e05c455997e
+  metadata.gz: 03fa1355e1b4ddb4b4377cc88c8a402085c525693719c36a92cc61b76b0c1738
+  data.tar.gz: fcf92ab74e0a897fd8b0d59ef06cc71963946f648e33462b2f1835f1afd118a2
 SHA512:
-  metadata.gz: 82761cb5d303b13edc9a8b9541e11a9c5d2c06b3765879ede4d6cf4176a8d11f1bcd24563da88481ff6648b308f265f8cc0a2b56eb01ac2cd5888417ab485062
-  data.tar.gz: 527e1976b374a68409c6ca8591b6ab6cb043ed374f69a3daa87d76116bed7d0d04d518d00d6df38bf1305965f9db71cdf4e8aac9c2e4e01fa9a31aa7a00723be
+  metadata.gz: 406edf0028a55c3ee1f86bc4e8702bdac35af844fde8aa6401897ce9f5475296b80255a59f16c4fbf6ceb59d1fdcd0c3c361bb05ed5323bc762e61c94032965d
+  data.tar.gz: 6d4cbaeecbdea0c3354dd83973192450cdca24e4b7ffa4a3b07d699efa2b8c9fd341c0c025eed1b689821c734b2cd85ba0f05f2aa2d40d9f28768afe8a2f2017

data/lib/pgai/cli/main.rb

@@ -2,7 +2,7 @@ module Pgai::Cli
   class Main < Base
     desc "config", "Configure CLI options"
     method_option :prefix, aliases: "-p", desc: "Specify prefix name to be used for clones", required: true
-    method_option :dbname, aliases: "-n", desc: "Specify database name to connect to by default", required: true
+    method_option :dbname, aliases: "-n", desc: "Specify database name to connect to by default", required: true, default: "postgres"
     method_option :proxy, aliases: "-x", desc: "Specify proxy host name", required: true
     method_option :exe, aliases: "-e", desc: "dblab binary path"
     def config

data/lib/pgai/clone_manager.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'securerandom'
+require "securerandom"
 
 module Pgai
   class CloneManager
@@ -76,7 +76,7 @@ module Pgai
         exec("psql #{clone.connection_string}")
       end
 
-      Signal.trap("INT") { }
+      Signal.trap("INT") {}
       start_caffeinate(psql_pid)
       Process.wait(psql_pid)
     ensure

data/lib/pgai/dblab.rb

@@ -18,7 +18,7 @@ module Pgai
     end
 
     def configure_env(port:, token:, id:)
-      dblab("init", "--url", "http://#{hostname}:#{port}", "--token", token, "--environment-id", id, silence: true)
+      dblab("init", "--url", "http://#{hostname}:#{port}", "--token", token, "--environment-id", id, silence: true, raw: true)
       dblab("config", "switch", id, raw: true)
     end
 
@@ -62,8 +62,7 @@ module Pgai
     end
 
     def download_dblab_to(location)
-      platform = Gem::Platform.local
-      uri = URI(DBLAB_BINARY_URL % {os: platform.os, cpu: platform.cpu})
+      uri = URI(DBLAB_BINARY_URL % {os: platform.os, cpu: cpu_platform})
 
       Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
         http.request Net::HTTP::Get.new(uri) do |response|
@@ -74,5 +73,15 @@ module Pgai
         end
       end
     end
+
+    def platform
+      Gem::Platform.local
+    end
+
+    def cpu_platform
+      cpu = platform.cpu
+      return "arm64" if cpu == "aarch64"
+      cpu
+    end
   end
 end

data/lib/pgai/version.rb

@@ -1,3 +1,3 @@
 module Pgai
-  VERSION = "0.1.6"
+  VERSION = "0.1.8"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pgai
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Marius Bobin