chat_sdk-x 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: df8df52963fbcdfc02f6457a2470c9708aa276d08443d5ec55a2174194db06c3
-  data.tar.gz: 012ca08f40e7e0b86638879fdc4db880ec46e43f95e3e6b7632bd51212872db2
+  metadata.gz: b3d2500b4890fc426c2f4bf0ae98c4d20309818ddfe00c66518013481a6aa436
+  data.tar.gz: 51c114854c79621bdd093b06bc0d1f7a040533ea38f943ce7f6b9bfae056b51c
 SHA512:
-  metadata.gz: 9762fc0118566399c42d729c675246cb326548083d8d8ed3d930503f892a450ad0cb64a026af4c573ebbad7873137a52df32e667a68d02a975e2981f275d9d62
-  data.tar.gz: 77e54b2060e5ecdc190466eaf666d4a7ed20db53ee7ca382d46b2a99f87ade839b00ab5084cb9b4b8827a84a77d9fc0296990bb33991678493a63f990c4589dc
+  metadata.gz: 4f096900b3e142d8aba30b804c8ad49309755392876dd3741bb7824d92879bd09fc5b5aad4f888b6185556d34c001664eedc3af3fcdc83a1adaad6d3d2b19493
+  data.tar.gz: 7665fbb3b51567a21a4abf0007853845ed13a902f3cf1483e4b08eba2c5f23408e31001ebbb16f870f09af5d5de99ae8d01e113932bd977bd634802acd8809e2

data/lib/chat_sdk/x/adapter.rb

@@ -7,7 +7,9 @@ require "rack/utils"
 module ChatSDK
   module X
     class Adapter < ChatSDK::Adapter::Base
-      capabilities :direct_messages, :reactions, :delete_messages, :message_history
+      include ChatSDK::Adapter::MediaTypes
+
+      capabilities :direct_messages, :reactions, :delete_messages, :message_history, :file_uploads
 
       attr_reader :client
 
@@ -83,23 +85,23 @@ module ChatSDK
             raw: result
           )
         else
-          payload = {text: text}
-          payload[:reply] = {in_reply_to_tweet_id: channel_id} if channel_id && thread_id
-          result = @client.create_tweet(**payload)
-          tweet_id = result.dig("data", "id") || result["id"]
-
-          ChatSDK::Message.new(
-            id: tweet_id,
-            text: text,
-            author: ChatSDK::Author.new(id: @user_id || "bot", name: "bot", platform: :x, bot: true),
-            thread_id: thread_id || "x:post:#{tweet_id}",
-            channel_id: channel_id,
-            platform: :x,
-            raw: result
-          )
+          reply_to = (channel_id if channel_id && thread_id)
+          result = @client.create_tweet(text: text, reply_to: reply_to)
+          parse_tweet_message(result, channel_id, thread_id: thread_id)
         end
       end
 
+      def upload_file(channel_id:, io:, filename:, thread_id: nil, comment: nil) # rubocop:disable Lint/UnusedMethodArgument
+        content_type = detect_content_type(filename)
+        bytes = io.respond_to?(:size) ? io.size : io.read.bytesize.tap { io.rewind }
+
+        media_id = @client.upload_media(io: io, content_type: content_type, total_bytes: bytes)
+
+        text = comment || ""
+        result = @client.create_tweet(text: text, media_ids: [media_id])
+        parse_tweet_message(result, channel_id)
+      end
+
       def delete_message(channel_id:, message_id:) # rubocop:disable Lint/UnusedMethodArgument
         @client.delete_tweet(tweet_id: message_id)
       end
@@ -141,6 +143,19 @@ module ChatSDK
         @client.unlike_tweet(user_id: @user_id, tweet_id: message_id)
       end
 
+      def get_user(user_id)
+        data = @client.get_user(user_id)
+        return nil unless data&.dig("data", "id")
+
+        ChatSDK::Author.new(
+          id: data.dig("data", "id"),
+          name: data.dig("data", "username"),
+          platform: :x,
+          bot: false,
+          raw: data
+        )
+      end
+
       def open_dm(user_id)
         user_id
       end
@@ -152,6 +167,22 @@ module ChatSDK
       def render(postable_message)
         postable_message.text
       end
+
+      private
+
+      def parse_tweet_message(result, channel_id, thread_id: nil)
+        tweet_id = result.dig("data", "id") || result["id"]
+
+        ChatSDK::Message.new(
+          id: tweet_id,
+          text: result.dig("data", "text") || "",
+          author: ChatSDK::Author.new(id: @user_id || "bot", name: "bot", platform: :x, bot: true),
+          thread_id: thread_id || "x:post:#{tweet_id}",
+          channel_id: channel_id,
+          platform: :x,
+          raw: result
+        )
+      end
     end
   end
 end

data/lib/chat_sdk/x/api_client.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require "base64"
+
 module ChatSDK
   module X
     class ApiClient < ChatSDK::ApiClient::Base
@@ -9,12 +11,42 @@ module ChatSDK
         @access_token = access_token
       end
 
-      def create_tweet(text:, reply: nil)
+      def create_tweet(text:, reply_to: nil, media_ids: nil)
         body = {"text" => text}
-        body["reply"] = reply if reply
+        body["reply"] = {"in_reply_to_tweet_id" => reply_to} if reply_to
+        body["media"] = {"media_ids" => media_ids} if media_ids&.any?
         request(:post, "/2/tweets", body)
       end
 
+      def upload_media(io:, content_type:, total_bytes:)
+        # INIT
+        init_result = request(:post, "/2/media/upload", {
+          command: "INIT",
+          total_bytes: total_bytes,
+          media_type: content_type
+        })
+        media_id = init_result.dig("data", "id") || init_result["media_id_string"]
+
+        # APPEND (single chunk, works for images <5MB)
+        response = upload_connection.post("/2/media/upload") do |req|
+          req.body = {
+            command: "APPEND",
+            media_id: media_id,
+            segment_index: 0,
+            media_data: Base64.strict_encode64(io.read)
+          }
+        end
+        handle_response(response)
+
+        # FINALIZE
+        request(:post, "/2/media/upload", {
+          command: "FINALIZE",
+          media_id: media_id
+        })
+
+        media_id
+      end
+
       def send_dm(participant_id:, text:)
         request(:post, "/2/dm_conversations/with/#{participant_id}/messages", {"text" => text})
       end
@@ -31,6 +63,10 @@ module ChatSDK
         request(:delete, "/2/tweets/#{tweet_id}")
       end
 
+      def get_user(user_id)
+        request(:get, "/2/users/#{user_id}?user.fields=name,username")
+      end
+
       def get_dm_events(participant_id:, cursor: nil, limit: 50)
         query = "max_results=#{[limit, 100].min}&dm_event.fields=id,text,sender_id,created_at"
         query += "&pagination_token=#{cursor}" if cursor

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chat_sdk-x
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Quentin Rousseau