at_protocol 0.0.4 → 0.0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88028815424d12616955d54125fb23ad627812290c2c52a8d491bb108ebd54fa
-  data.tar.gz: d68b5af3850c75174ef573809f2d87a4012c707d82079601cdd3351da5c317ed
+  metadata.gz: bac3005062f2494c16a3fcabc3af9f45daf3b127b81624872ab71603de1f9fbc
+  data.tar.gz: 28088b1b416163feab91d06e42245279216673cf2ae1d077eebc78bfb79a0b1c
 SHA512:
-  metadata.gz: 37cdd260697f6461ed9215f4d06b932988a292a5fc53afde8d555a1dc04c5355819026742528997d5c2d476a9618ddf396a972f794c73cab11784e6fff39c520
-  data.tar.gz: 5cd9f10e044b3edfdc37ce71abd01ae53fd66bfe4fa9669157d0a2f507e5d6190f3ed4b89490fb012dc9868ac2c7e2722fa35967428934cf1efec697c6def618
+  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/record/strongref.rb

@@ -0,0 +1,22 @@
+# typed: true
+module ATProto
+  class Record
+    class StrongRef < T::Struct
+      include RequestUtils
+      extend T::Sig
+      const :uri, ATProto::AtUri
+      prop :cid, Skyfall::CID
+
+      def to_json
+        {
+          "uri" => uri.to_s,
+          "cid" => cid.to_s,
+        }
+      end
+
+      def inspect
+        self.to_json.to_json
+      end
+    end
+  end
+end

data/lib/at_protocol/record.rb

@@ -46,7 +46,7 @@ module ATProto
           record: content_hash,
         }
         params[:rkey] = rkey unless rkey.nil?
-        from_uri(at_uri(session.xrpc.post.com_atproto_repo_createRecord(params)["uri"]))
+        from_uri(at_uri(session.xrpc.post.com_atproto_repo_createRecord(**params)["uri"]))
       end
     end
     dynamic_attr_reader(:to_uri) { "at://#{self.uri.repo}/#{self.uri.collection}/#{self.uri.rkey}" }
@@ -72,19 +72,19 @@ module ATProto
 
     def to_write(type = :delete)
       ATProto::Writes::Write.new(
-        {
+        **({
           action: Writes::Write::Action.deserialize(type),
           value: (self.content if type == :update),
           collection: self.uri.collection,
           rkey: self.uri.rkey,
-        }.compact
+        }.compact),
       )
     end
 
     sig { params(other: ATProto::Record).returns(T::Boolean) }
 
     def ==(other)
-      self.cid.to_s == other.cid.to_s
+      self.cid.to_s == other.cid.to_s && self.uri.to_s == other.uri.to_s
     end
   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"
+  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,13 +15,21 @@ class Class
   end
 end
 
-require :"skyfall/cid".to_s
+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"
-require "at_protocol/requests"
-require "at_protocol/session"
-require "at_protocol/repo"
-require "at_protocol/collection"
-require "at_protocol/at_uri"
-require "at_protocol/writes"
-require "at_protocol/record"
-require "at_protocol/helpers/strongref"
+
+%w(requests session repo collection at_uri writes record record/strongref).each do |name|
+  require "at_protocol/#{name}"
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: at_protocol
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.4.2
 platform: ruby
 authors:
 - Shreyan Jain
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-08-16 00:00:00.000000000 Z
+date: 2023-09-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: xrpc
@@ -36,6 +36,7 @@ files:
 - "./lib/at_protocol/at_uri.rb"
 - "./lib/at_protocol/collection.rb"
 - "./lib/at_protocol/record.rb"
+- "./lib/at_protocol/record/strongref.rb"
 - "./lib/at_protocol/repo.rb"
 - "./lib/at_protocol/requests.rb"
 - "./lib/at_protocol/session.rb"