pgai 0.1.5 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a31b2fc8c0e25d189cd3f3191a8e1833de2cbe8f21dfe5e4322793fea1aa4efd
-  data.tar.gz: 6061e6095d92912eb7d5a2f64fe95740314494ff23aefab32afe4e9b15d77d3b
+  metadata.gz: '028a01e72ff3ca2ffa39e2cf32e400b2e375d79da46f65c51f746d5667dc0719'
+  data.tar.gz: 8f2ea632ff8d19a4382af4b5647a322ef9489da49ccdfa4f462779eb9d184456
 SHA512:
-  metadata.gz: ee38e79b9be45e6a848659a830550ce0cf7ad3cf272fc2c6d6e9ddb95214595ae9fd506a4726d3fffe36719e56a4d4bf439ff8543099ee457feed13e5a6ebbc6
-  data.tar.gz: b3d34902baa84fc445e10d1874455b3168fe3896c78fd83d41b84c24d78ee1bdc9ccf02282b984d848cdf2984bc8f4ca136b43ff367c9a0857857d164b554358
+  metadata.gz: 9b561b90c9a9d9fd230534eb08e0c2c713302f8789a16a4536639a941355f1d6234240d93b2baf55f4845e23ad3377b62f1b075d03a2b9ff1ca26b17cdca4b88
+  data.tar.gz: e46cb2fa48461f6986ee5e41d00aa6d7022d91906f14abf78e0620b85c83ae77c3a6ed4f60392de922b91f79e2c569792efbedf058edc5017d203d38bbfcde31

data/lib/pgai/clone_manager.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require "securerandom"
+
 module Pgai
   class CloneManager
     HOSTNAME = "127.0.0.1"
@@ -71,9 +73,11 @@ module Pgai
       port_forward.start(clone.port)
 
       psql_pid = fork do
-        start_caffeinate(Process.pid)
         exec("psql #{clone.connection_string}")
       end
+
+      Signal.trap("INT") {}
+      start_caffeinate(psql_pid)
       Process.wait(psql_pid)
     ensure
       port_forward.conditionally_stop(clone.port, [psql_pid])

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.5"
+  VERSION = "0.1.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pgai
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.7
 platform: ruby
 authors:
 - Marius Bobin
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-25 00:00:00.000000000 Z
+date: 2023-04-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor