at_protocol 0.0.4.1 → 0.0.4.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9079f38f8b754c7aab1c556dd3b79e98a5f95ffa5091ce0a36f3f3da483b69d
-  data.tar.gz: b3423fa8c0a4595f7afb81cbb823c38a58a6d5e609dff38ebe297b85dfd4c84b
+  metadata.gz: e0dd450b9e79e59d5df52330a9e0fd00135db4e2a8eddce654780133835ca69f
+  data.tar.gz: bda2354c8c63d85eee0075c5fda834e1018e31e9c8b5daba221695644998e338
 SHA512:
-  metadata.gz: 066a7e0ba5ea613209e4b180be05c25c82251968fad5e7b611fa3200ae820ceb41268c1759d0ec77eb733aa410c8878cbd6890c61e7bf47ebd1171c364ca84ed
-  data.tar.gz: b106124acf97f5d6443aae45adcf65eac04319d923f92e08e3404bb87c8912ab59ff7feb739ef9b0a377c9b23e83fa0789bed8ba8128dd5e8ad9be4d4a7c0dad
+  metadata.gz: 19eec8bd3383a55ae752ed3561bcced63c5a438fe1060dcd1a08fe46249e56d469b7ded6a026c73d82a84a78ef4fd2af35572ccd610fa854287b1babf43e3e8f
+  data.tar.gz: e8d0dc32c9e1a4101fb8dc30b0ca16a49955f3d096fbc75d36ce20486590cefb67df66c83268dabde43716495247706b6f8a7599a6b530a67b75d1b096c621db

data/ext/at_protocol/extconf.rb

@@ -0,0 +1,3 @@
+require "mkmf"
+
+create_makefile("at_protocol/tid")

data/ext/at_protocol/tid.c

@@ -0,0 +1,130 @@
+#include <ruby.h>
+#include <time.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <math.h>
+
+static VALUE atproto_module;
+static VALUE tid_class;
+const char b32_chars[32] = "234567abcdefghijklmnopqrstuvwxyz";
+
+typedef struct {
+  uint64_t timestamp;
+  uint64_t clock_identifier;
+} TID;
+
+static VALUE tid_alloc(VALUE klass) {
+  TID *tid = ALLOC(TID);
+  return Data_Wrap_Struct(klass, 0, xfree, tid);
+}
+
+static VALUE tid_initialize(int argc, VALUE *argv, VALUE self) {
+  VALUE time_arg = Qnil;
+  rb_scan_args(argc, argv, "01", &time_arg);
+
+  TID *tid;
+  Data_Get_Struct(self, TID, tid);
+
+  if (NIL_P(time_arg)) {
+    struct timeval tv;
+    gettimeofday(&tv, NULL);
+    tid->timestamp = (uint64_t)tv.tv_sec * 1000000 + (uint64_t)tv.tv_usec;
+  } else {
+    if (!rb_obj_is_kind_of(time_arg, rb_cTime)) {
+      rb_raise(rb_eTypeError, "Argument must be a Time object or nil");
+      return Qnil;
+    }
+
+    struct timeval tv;
+    tv.tv_sec = NUM2LONG(rb_funcall(time_arg, rb_intern("tv_sec"), 0));
+    tv.tv_usec = NUM2LONG(rb_funcall(time_arg, rb_intern("tv_usec"), 0));
+    tid->timestamp = (uint64_t)tv.tv_sec * 1000000 + (uint64_t)tv.tv_usec;
+  }
+
+  tid->clock_identifier = rand() & 0x3FF;
+
+  return self;
+}
+
+static VALUE tid_to_time(VALUE self) {
+  TID *tid;
+  Data_Get_Struct(self, TID, tid);
+
+  VALUE sec = LL2NUM(tid->timestamp / 1000000);
+  VALUE usec = LL2NUM(tid->timestamp % 1000000);
+
+  return rb_funcall(rb_cTime, rb_intern("at"), 2, sec, usec);
+}
+
+static VALUE tid_to_s(VALUE self) {
+  TID *tid;
+  Data_Get_Struct(self, TID, tid);
+
+  char tid_str[14];
+  for (int i = 0; i < 11; i++) {
+    int index = (int)((tid->timestamp >> (50 - i * 5)) & 0x1F);
+    tid_str[i] = b32_chars[index];
+  }
+  tid_str[11] = b32_chars[(int)(tid->clock_identifier >> 6) & 0x1F];
+  tid_str[12] = b32_chars[(int)tid->clock_identifier & 0x1F];
+  tid_str[13] = '\0';
+
+  return rb_str_new_cstr(tid_str);
+}
+
+static VALUE tid_from_string(VALUE klass, VALUE str) {
+  Check_Type(str, T_STRING);
+
+  TID *tid;
+  VALUE tid_obj = Data_Make_Struct(klass, TID, NULL, xfree, tid);
+
+  if (RSTRING_LEN(str) != 13) {
+    rb_raise(rb_eArgError, "TID string must be 13 characters long");
+    return Qnil;
+  }
+
+  char tid_str[14];
+  strncpy(tid_str, StringValueCStr(str), sizeof(tid_str));
+  tid_str[13] = '\0';
+
+  // Convert the TID string back to a TID object
+  uint64_t timestamp = 0;
+  for (int i = 0; i < 11; i++) {
+    char c = tid_str[i];
+    const char *pos = strchr(b32_chars, c);
+    if (pos == NULL) {
+      rb_raise(rb_eArgError, "Invalid character in TID string");
+      return Qnil;
+    }
+    int index = (int)(pos - b32_chars);
+    timestamp = (timestamp << 5) | (index & 0x1F);
+  }
+
+  uint64_t clock_identifier = 0;
+  for (int i = 11; i < 13; i++) {
+    char c = tid_str[i];
+    const char *pos = strchr(b32_chars, c);
+    if (pos == NULL) {
+      rb_raise(rb_eArgError, "Invalid character in TID string");
+      return Qnil;
+    }
+    int index = (int)(pos - b32_chars);
+    clock_identifier = (clock_identifier << 5) | (index & 0x1F);
+  }
+
+  tid->timestamp = timestamp;
+  tid->clock_identifier = clock_identifier;
+
+  return tid_obj;
+}
+
+void Init_tid(void) {
+  atproto_module = rb_define_module("ATProto");
+
+  tid_class = rb_define_class_under(atproto_module, "TID", rb_cObject);
+  rb_define_alloc_func(tid_class, tid_alloc);
+  rb_define_method(tid_class, "initialize", tid_initialize, -1);
+  rb_define_method(tid_class, "to_time", tid_to_time, 0);
+  rb_define_method(tid_class, "to_s", tid_to_s, 0);
+  rb_define_singleton_method(tid_class, "from_string", tid_from_string, 1);
+}

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/tid_ext.rb

@@ -0,0 +1,19 @@
+require "at_protocol/tid"
+
+class Time
+  def to_tid
+    ATProto::TID.new(self)
+  end
+end
+
+class String
+  def to_tid
+    ATProto::TID.from_string(self)
+  end
+end
+
+class ATProto::TID
+  def inspect
+    "#<ATProto::TID(#{self.to_s})>"
+  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.3"
 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,9 +15,21 @@ 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"
 
-%w(requests session repo collection at_uri writes record record/strongref).each do |name|
+%w(requests session repo collection at_uri writes record record/strongref tid_ext).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.1
+  version: 0.0.4.3
 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-10-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: xrpc
@@ -25,13 +25,30 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.1.5
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A Ruby gem for interacting with AT Protocol
 email:
 - shreyan.jain.9@outlook.com
 executables: []
-extensions: []
+extensions:
+- ext/at_protocol/extconf.rb
 extra_rdoc_files: []
 files:
+- "./ext/at_protocol/extconf.rb"
+- "./ext/at_protocol/tid.c"
 - "./lib/at_protocol.rb"
 - "./lib/at_protocol/at_uri.rb"
 - "./lib/at_protocol/collection.rb"
@@ -40,8 +57,10 @@ files:
 - "./lib/at_protocol/repo.rb"
 - "./lib/at_protocol/requests.rb"
 - "./lib/at_protocol/session.rb"
+- "./lib/at_protocol/tid_ext.rb"
 - "./lib/at_protocol/version.rb"
 - "./lib/at_protocol/writes.rb"
+- ext/at_protocol/extconf.rb
 homepage: https://github.com/ShreyanJain9/at_protocol
 licenses:
 - MIT
@@ -50,6 +69,7 @@ post_install_message:
 rdoc_options: []
 require_paths:
 - lib
+- ext
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="