at_protocol 0.0.4.1 → 0.0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9079f38f8b754c7aab1c556dd3b79e98a5f95ffa5091ce0a36f3f3da483b69d
-  data.tar.gz: b3423fa8c0a4595f7afb81cbb823c38a58a6d5e609dff38ebe297b85dfd4c84b
+  metadata.gz: bac3005062f2494c16a3fcabc3af9f45daf3b127b81624872ab71603de1f9fbc
+  data.tar.gz: 28088b1b416163feab91d06e42245279216673cf2ae1d077eebc78bfb79a0b1c
 SHA512:
-  metadata.gz: 066a7e0ba5ea613209e4b180be05c25c82251968fad5e7b611fa3200ae820ceb41268c1759d0ec77eb733aa410c8878cbd6890c61e7bf47ebd1171c364ca84ed
-  data.tar.gz: b106124acf97f5d6443aae45adcf65eac04319d923f92e08e3404bb87c8912ab59ff7feb739ef9b0a377c9b23e83fa0789bed8ba8128dd5e8ad9be4d4a7c0dad
+  metadata.gz: eeb00d26cead39126432d2be52540da02f5d9734eebe9c5c4735dd4e4e556ae88a0886fe8455c98a72969b7728e2213ad0f957317a1d89852cdbcb687e7d44f7
+  data.tar.gz: deb75bae3fe96f9eb2831f0e61ce48f23aa9030b5fe1714123cb0c7293e7378ac3a13cef0850f3b653e5467376b19fbf07de5a9d50e21d2a775527ce16b507ba

data/lib/at_protocol/at_uri.rb

@@ -13,6 +13,10 @@ module ATProto
 
     module_function :at_uri
   end
+
+  def self.AtUri(str)
+    RequestUtils.at_uri(str)
+  end
 end
 
 module ATProto

data/lib/at_protocol/collection.rb

@@ -9,22 +9,10 @@ module ATProto
       const(:repo, ATProto::Repo)
       const(:collection, String)
 
-      sig { params(limit: Integer).returns(T::Array[ATProto::Record]) }
-
-      def list(limit = 10)
-        self.repo.xrpc
-          .get.com_atproto_repo_listRecords(
-            repo: self.repo.did,
-            collection: self.collection,
-            limit: limit,
-          )["records"]
-          .map { |record|
-          ATProto::Record.from_hash(record)
-        }
-      end
+      sig { params(count: T.nilable(Integer)).returns(T::Array[ATProto::Record]) }
 
-      def list_all()
-        T.must(get_paginated_data(self.repo, :com_atproto_repo_listRecords.to_s, key: "records", params: { repo: self.repo.to_s, collection: self.to_s }, cursor: nil) do |record|
+      def list(count = nil)
+        T.must(get_paginated_data(self.repo, :com_atproto_repo_listRecords.to_s, key: "records", params: { repo: self.repo.to_s, collection: self.to_s }, count: count, cursor: nil) do |record|
           ATProto::Record.from_hash(record)
         end)
       end

data/lib/at_protocol/requests.rb

@@ -78,6 +78,9 @@ module ATProto
         cursor: T.nilable(
           String
         ),
+        count: T.nilable(
+          Integer
+        ),
         map_block: T.nilable(Proc),
       )
         .returns(T
@@ -86,7 +89,7 @@ module ATProto
           ))
     }
 
-    def get_paginated_data(session, method, key:, params:, cursor: nil, &map_block)
+    def get_paginated_data(session, method, key:, params:, cursor: nil, count: nil, &map_block)
       params.merge({ limit: 100, cursor: cursor }).then do |send_data|
         session.xrpc.get.public_send(method, **send_data).then do |response|
           response.dig(key).then do |data|
@@ -94,26 +97,19 @@ module ATProto
               return []
             end
 
-            if block_given?
-              data.map(&map_block).then do |results|
-                response.dig("cursor").then do |next_cursor|
-                  if next_cursor.nil?
-                    return results
-                  else
-                    get_paginated_data(session, method, key: key, params: params, cursor: next_cursor, &map_block).then do |next_results|
-                      return results + next_results
-                    end
-                  end
-                end
+            results = if block_given?
+                data.map(&map_block)
+              else
+                data
               end
-            else
-              response.dig("cursor").then do |next_cursor|
-                if next_cursor.nil?
-                  return data
-                else
-                  get_paginated_data(session, method, key: key, params: params, cursor: next_cursor, &map_block).then do |next_results|
-                    return data + next_results
-                  end
+
+            response.dig("cursor").then do |next_cursor|
+              if next_cursor.nil? || (count && results.length >= count)
+                return results[0..count]
+              else
+                remaining_count = count ? count - results.length : nil
+                get_paginated_data(session, method, key: key, params: params, cursor: next_cursor, count: remaining_count, &map_block).then do |next_results|
+                  return (results + next_results)
                 end
               end
             end
@@ -123,3 +119,26 @@ module ATProto
     end
   end
 end
+
+module ATProto
+  module RequestUtils
+    class XRPCResponseCode < T::Enum
+      enums do
+        Unknown = new(1)
+        InvalidResponse = new(2)
+        Success = new(200)
+        InvalidRequest = new(400)
+        AuthRequired = new(401)
+        Forbidden = new(403)
+        XRPCNotSupported = new(404)
+        PayloadTooLarge = new(413)
+        RateLimitExceeded = new(429)
+        InternalServerError = new(500)
+        MethodNotImplemented = new(501)
+        UpstreamFailure = new(502)
+        NotEnoughResouces = new(503)
+        UpstreamTimeout = new(504)
+      end
+    end
+  end
+end

data/lib/at_protocol/session.rb

@@ -66,6 +66,10 @@ module ATProto
         raise UnauthorizedError
       end
     end
+
+    def inspect
+      "#<ATProto::Session(did: #{did}, access_token: #{access_token})>"
+    end
   end
 end
 

data/lib/at_protocol/version.rb

@@ -1,4 +1,4 @@
 # typed: strict
 module ATProto
-  VERSION = "0.0.4.1"
+  VERSION = "0.0.4.2"
 end

data/lib/at_protocol/writes.rb

@@ -1,5 +1,7 @@
 # typed: true
 module ATProto
+  extend T::Sig
+
   class Writes < T::Struct
     class Write < T::Struct
       class Action < T::Enum
@@ -101,14 +103,13 @@ module ATProto
 
     class << self
       extend T::Sig
-      sig { params(block: Proc).returns(T::Array[Write]) }
-
-      def generate(&block)
-        Collector.new.instance_eval(&block)
-      end
     end
   end
 
+  def self.Writes(session, &block)
+    Writes.new(writes: Writes::Collector.new.instance_eval(&block), session: session, repo: Repo.new(session.did))
+  end
+
   class Writes
     class Collector
       include RequestUtils

data/lib/at_protocol.rb

@@ -15,6 +15,18 @@ class Class
   end
 end
 
+module ATProto
+  class << self
+    def method_missing(method_name, *args, &block)
+      if const_defined?(method_name)
+        Object.const_get(method_name).new(*args, &block)
+      else
+        super
+      end
+    end
+  end
+end
+
 require "skyfall/cid"
 require "xrpc"
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: at_protocol
 version: !ruby/object:Gem::Version
-  version: 0.0.4.1
+  version: 0.0.4.2
 platform: ruby
 authors:
 - Shreyan Jain
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-09-01 00:00:00.000000000 Z
+date: 2023-09-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: xrpc