pgai 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 123bfba393d6bed50c493b3acd2d934b497c506a09c5680293bc2d68eb942394
-  data.tar.gz: cc04c9ab11322ba25526b3fef520c72c172e9672d52e04aded6b23e37a28490b
+  metadata.gz: 00ac038ff78647c4a3569cb87ebe3ae2dd10b9946be720ccffba5b951fc09f26
+  data.tar.gz: b4581834f1c91edfddff07cff3e05bdd658aa8983038a40adc24b3beaaf68158
 SHA512:
-  metadata.gz: d746ae9b949706ad27f5aeb8089ff928475188a2e43ab996b8ece041feae63b740d70a3ed165f0af3f022309066cec0bcb6200fca8b1c7c1da4a1e7dec7932e1
-  data.tar.gz: 43b5d7e409bca1d28cb75c6fff9a3259c34348bc7365035134acff02639e5e8a7753401f83ef3d3dcbbab2fa6089060e91bf4bfb1406dd27b19e102d63b23a9f
+  metadata.gz: 572343b7a247938db90162ed94f44aa835f0dd7078ca6366f78b0b4151e8c82218b2aacd47f45f8ecb21c0d381e093642a5e40f0b61da37a28014c0a104d80a4
+  data.tar.gz: f7fa39825ffcac44f5130b5e2a98f9348d013928382f8ee5ae73d63417f75c1f6cb2dfc85d4ef627c4828656ede3fccee9eea4dcb9efcbdcde9385b44e238ccf

data/lib/pgai/cli/env.rb

@@ -33,10 +33,26 @@ module Pgai::Cli
     desc "list", "List all configured environments"
     def list
       data = Pgai::Resources::Local::Environment.all.map do |env|
-        env.attributes.slice(:id, :alias, :port, :dbname)
+        env.attributes.slice(:id, :alias, :port, :dbname, :snapshot)
       end
 
       say JSON.pretty_generate(data)
     end
+
+    desc "dup env", "Duplicate an environment and override the provided attributes"
+    method_option :alias, aliases: "-a",
+      desc: "This will be used internally for the connect command",
+      required: true
+    method_option :dbname,
+      aliases: "-n",
+      desc: "Specify database name to connect to by default"
+    method_option :snapshot,
+      aliases: "-s",
+      desc: "Specify the snapshot id if you want to create a clone from a specific snapshot"
+    def dup(env)
+      env = Pgai::Resources::Local::Environment.find(env)
+      attrs = env.attributes.merge(options.transform_keys(&:to_sym))
+      Pgai::Resources::Local::Environment.new(attrs).save
+    end
   end
 end

data/lib/pgai/client.rb

@@ -46,6 +46,7 @@ module Pgai
     def parse_reponse(data)
       data = JSON.parse(data)
       transform!(data)
+      handle_unauthorized(data)
       data
     end
 
@@ -65,5 +66,12 @@ module Pgai
         transform_array!(data)
       end
     end
+
+    def handle_unauthorized(data)
+      return unless data.is_a?(Hash)
+      return unless data[:code] == "UNAUTHORIZED"
+
+      raise UnauthorizedError, data[:message]
+    end
   end
 end

data/lib/pgai/create_clone_service.rb

@@ -33,7 +33,7 @@ module Pgai
     def create_clone_resource
       @clone_resource = Resources::Remote::Clone.new(id: id).tap do |resource|
         resource.db_object = Resources::Remote::DbObject.new(db_name: env.dbname)
-        resource.snapshot = Resources::Remote::Snapshot.latest
+        resource.snapshot = find_snapshot
         resource.save
       end
     end
@@ -56,6 +56,14 @@ module Pgai
       @clone_resource ||= create_clone_resource
     end
 
+    def find_snapshot
+      if env.snapshot
+        Resources::Remote::Snapshot.find(env.snapshot)
+      else
+        Resources::Remote::Snapshot.latest
+      end
+    end
+
     def wait_for_clone_to_be_ready
       loop do
         progress_bar.advance(steps)

data/lib/pgai/resources/local/environment.rb

@@ -10,6 +10,7 @@ module Pgai
         attribute :dbname, :string
         attribute :access_token, :string
         attribute :clone_prefix, :string
+        attribute :snapshot, :string
 
         def client
           @client ||= Pgai::Client.new(

data/lib/pgai/resources/remote/snapshot.rb

@@ -24,6 +24,12 @@ module Pgai
           def path
             "snapshots"
           end
+
+          def find(id)
+            record = all.find { |snapshot| snapshot.id.to_s == id.to_s }
+            raise Pgai::ResourceNotFound unless record
+            record
+          end
         end
       end
     end

data/lib/pgai/version.rb

@@ -1,3 +1,3 @@
 module Pgai
-  VERSION = "1.0.1"
+  VERSION = "1.0.2"
 end

data/lib/pgai.rb

@@ -5,8 +5,14 @@ require "zeitwerk"
 loader = Zeitwerk::Loader.for_gem
 loader.setup
 
+require "thor"
+
 module Pgai
   class Error < StandardError; end
 
+  class CliError < ::Thor::Error; end
+
   class ResourceNotFound < Error; end
+
+  class UnauthorizedError < CliError; end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pgai
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Marius Bobin
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-11-14 00:00:00.000000000 Z
+date: 2024-12-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor