chat_sdk-x 0.4.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 263d6f652c3e638bf81b76b31d42cbcd629cd545fa905aa12d703b359ad7fd16
-  data.tar.gz: 00c9a4d7394037ec77d1cdbd3fc6f9462009a61c01c8f15acc349fb872b63682
+  metadata.gz: b3d2500b4890fc426c2f4bf0ae98c4d20309818ddfe00c66518013481a6aa436
+  data.tar.gz: 51c114854c79621bdd093b06bc0d1f7a040533ea38f943ce7f6b9bfae056b51c
 SHA512:
-  metadata.gz: b0b4b384141b90f9a2870d7e2f3d16a78ae0a5f02093615053a981d0bb3a8d8d724b7803967992e386c46dff2711f5762ddc867cb1d2e640bd4ce9df87ce5ab5
-  data.tar.gz: e3954e97efc0e99e52787e9934f26d4f0e36df7827352b2a960da04cbfe37346b6a6647c4e8bd5d11cd43692149f3fafe8ea2881b92cd4665cec6c676cefaabc
+  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
+      include ChatSDK::Adapter::MediaTypes
+
+      capabilities :direct_messages, :reactions, :delete_messages, :message_history, :file_uploads
 
       attr_reader :client
 
@@ -83,20 +85,53 @@ 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"]
+          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
 
-          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
-          )
+      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
+
+      def fetch_messages(channel_id:, thread_id: nil, cursor: nil, limit: 50) # rubocop:disable Lint/UnusedMethodArgument
+        if thread_id&.start_with?("x:dm:")
+          participant_id = thread_id.delete_prefix("x:dm:")
+          result = @client.get_dm_events(participant_id: participant_id, cursor: cursor, limit: limit)
+          data = result["data"] || []
+          messages = data.map do |dm|
+            ChatSDK::Message.new(
+              id: dm["id"]&.to_s,
+              text: dm["text"] || "",
+              author: ChatSDK::Author.new(
+                id: dm["sender_id"] || "unknown",
+                name: dm["sender_id"] || "unknown",
+                platform: :x,
+                bot: false
+              ),
+              thread_id: thread_id,
+              channel_id: channel_id,
+              platform: :x,
+              raw: dm
+            )
+          end
+          next_cursor = result.dig("meta", "next_token")
+          [messages, next_cursor]
+        else
+          # X does not provide a robust thread-fetching API for tweets
+          [[], nil]
         end
       end
 
@@ -108,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
@@ -119,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
@@ -27,6 +59,20 @@ module ChatSDK
         request(:delete, "/2/users/#{user_id}/likes/#{tweet_id}")
       end
 
+      def delete_tweet(tweet_id:)
+        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
+        request(:get, "/2/dm_conversations/with/#{participant_id}/dm_events?#{query}")
+      end
+
       private
 
       def base_url

data/lib/chat_sdk/x/event_parser.rb

@@ -10,6 +10,7 @@ module ChatSDK
           events = []
           events.concat(parse_mentions(payload, bot_user_id))
           events.concat(parse_direct_messages(payload, bot_user_id))
+          events.concat(parse_favorites(payload, bot_user_id))
           events
         end
 
@@ -81,6 +82,32 @@ module ChatSDK
             )
           end
         end
+
+        def parse_favorites(payload, bot_user_id)
+          fav_events = payload["favorite_events"] || []
+          return [] unless fav_events.is_a?(Array)
+
+          fav_events.filter_map do |data|
+            user = data["user"] || {}
+            user_id = user["id_str"] || user["id"]&.to_s
+            next if bot_user_id && user_id == bot_user_id
+
+            tweet = data["favorited_status"] || {}
+            tweet_id = tweet["id_str"] || tweet["id"]&.to_s
+
+            ChatSDK::Events::Reaction.new(
+              emoji: "heart",
+              user_id: user_id || "unknown",
+              message_id: tweet_id || "unknown",
+              thread_id: "x:post:#{tweet_id}",
+              channel_id: user_id,
+              added: true,
+              platform: :x,
+              adapter_name: :x,
+              raw: data
+            )
+          end
+        end
       end
     end
   end

metadata

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