pgai 0.1.8 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 03fa1355e1b4ddb4b4377cc88c8a402085c525693719c36a92cc61b76b0c1738
-  data.tar.gz: fcf92ab74e0a897fd8b0d59ef06cc71963946f648e33462b2f1835f1afd118a2
+  metadata.gz: 7c07781fdcd787b0aaba1d7fe23afcb6e63ee1b5da7900ac454bd9d2e95dde4f
+  data.tar.gz: 3ff5e06bc552e96b04359bca358f83563f11d76eed0cd205d4d8210ab9f86604
 SHA512:
-  metadata.gz: 406edf0028a55c3ee1f86bc4e8702bdac35af844fde8aa6401897ce9f5475296b80255a59f16c4fbf6ceb59d1fdcd0c3c361bb05ed5323bc762e61c94032965d
-  data.tar.gz: 6d4cbaeecbdea0c3354dd83973192450cdca24e4b7ffa4a3b07d699efa2b8c9fd341c0c025eed1b689821c734b2cd85ba0f05f2aa2d40d9f28768afe8a2f2017
+  metadata.gz: 82dc1b7d5aea1922e36753731125e85239c1a4d03e7bcaeb172e37c4db959fdbee24aa55ecaaf82eeb6312ade4b15efcbf827687acf0012eb135487b6b918bd3
+  data.tar.gz: 3308fcc8359891f10b550e20f448ca0e0e32e1ee00753aedbfcf08bd90efe18de7a932fd3de3dd43c841468a45280b663a664d854eb823a98de516a2bb6d7a9d

data/README.md

@@ -25,7 +25,7 @@ An access token will be required and it can be obtained from: https://console.po
 Example:
 
 ```shell
-pgai config --dbname=gitlabhq_dblab --prefix=<gitlab handle> --proxy=<domain> --exe=<optional /path/to/dblab>
+pgai config --dbname=gitlabhq_dblab --prefix=<gitlab handle> --proxy=<alias> --exe=<optional /path/to/dblab>
 ```
 
 To configure environments:
@@ -36,12 +36,13 @@ pgai env add --alias ci --id ci-database --port 12345
 
 The environment id, port, and proxy domain can be found by clicking on the `Connect` button on a database instance page.
 
-Configuring the user and identity file for the proxy domain must be done using the `~/.ssh/config` file. Example:
+Configuring the proxy attributes must be done via the `~/.ssh/config` file. Example:
 
 ```
-Host <domain>
-  IdentityFile ~/.ssh/id_ed25519
+Host <alias>
+  HostName <domain.postgres.ai>
   User <username>
+  IdentityFile ~/.ssh/id_ed25519
 ```
 
 ## dblab binary file

data/lib/pgai/cli/main.rb

@@ -25,6 +25,21 @@ module Pgai::Cli
       Pgai::CloneManager.new(env, config: configuration).cleanup
     end
 
+    desc "reset database-name", "Reset clone's state"
+    def reset(name)
+      env = configuration.enviroments.fetch(name)
+
+      Pgai::CloneManager.new(env, config: configuration).reset
+    end
+
+    desc "info database-name", "Show clone details"
+    def info(name)
+      env = configuration.enviroments.fetch(name)
+
+      data = Pgai::CloneManager.new(env, config: configuration).info
+      print_table data
+    end
+
     desc "env", "Manage environments"
     subcommand "env", Pgai::Cli::Env
   end

data/lib/pgai/clone_manager.rb

@@ -28,6 +28,26 @@ module Pgai
       port_forward.stop
     end
 
+    def reset
+      configure_enviroment
+      return unless find_raw_clone
+
+      dblab.reset_clone(id: clone_id)
+    end
+
+    def info
+      configure_enviroment
+
+      clone = find_clone
+      return {} unless clone
+
+      {
+        connection_string: clone.connection_string,
+        created_at: clone.created_at,
+        data_state_at: clone.data_state_at
+      }
+    end
+
     private
 
     attr_reader :environment, :config, :port_forward, :dblab
@@ -45,7 +65,7 @@ module Pgai
       raw_clone = find_raw_clone
 
       if raw_clone
-        config.find_clone(clone_id) || raise(Pagi::CloneNotFount, "clone not tracked")
+        config.find_clone(clone_id) || raise(Pgai::CloneNotFount, "clone not tracked")
       else
         config.remove_clone(clone_id) && nil
       end
@@ -63,13 +83,16 @@ module Pgai
         password: clone_password,
         user: clone_user,
         host: HOSTNAME,
-        dbname: config.dbname
+        dbname: config.dbname,
+        created_at: raw_clone["createdAt"],
+        data_state_at: raw_clone.dig("snapshot", "dataStateAt")
       }
 
       config.persist_clone(clone_id, attributes)
     end
 
     def psql(clone)
+      puts "Data state at: #{clone.data_state_at}"
       port_forward.start(clone.port)
 
       psql_pid = fork do
@@ -101,7 +124,7 @@ module Pgai
     end
 
     def clone_password
-      @clone_user ||= SecureRandom.hex(16)
+      @clone_password ||= SecureRandom.hex(16)
     end
 
     def start_caffeinate(pid)

data/lib/pgai/config.rb

@@ -6,7 +6,7 @@ require "fileutils"
 
 module Pgai
   class Config
-    Clone = Struct.new(:host, :port, :password, :user, :dbname, keyword_init: true) do
+    Clone = Struct.new(:host, :port, :password, :user, :dbname, :created_at, :data_state_at, keyword_init: true) do
       def connection_string
         "'host=#{host} port=#{port} user=#{user} dbname=#{dbname} password=#{password}'"
       end

data/lib/pgai/dblab.rb

@@ -37,6 +37,10 @@ module Pgai
       dblab("clone", "destroy", id, raw: true)
     end
 
+    def reset_clone(id:)
+      dblab("clone", "reset", id, raw: true)
+    end
+
     private
 
     attr_reader :config, :hostname

data/lib/pgai/version.rb

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

metadata

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