mailkite 0.4.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cf95721d75808a7648a0d5c5528a5b24cebe649e4e7b4f2aa7cf96309492c4df
-  data.tar.gz: fd40da8370ad8ad24b538613a29a8b623226901a8db8e5bc6a4b0b3ebd1ec709
+  metadata.gz: 79f557c742e7377043c6fbbfc62c788bde380e6c75f647a354c4182346275a4d
+  data.tar.gz: be02421caccae364a81433fd239d043e19aeb546817c0acbf3c9727850894785
 SHA512:
-  metadata.gz: 96e4a3c87afd08886338448d1edf8bfd83d7433733bba220cb892647249c923d46dde87df440ce78c84f1301b734fec480b89d9ee2e54f8eef2e528fec4af4dc
-  data.tar.gz: d7b4ccb09e09a382e7d9bef520115c50aa06c635646763b73cea18bc133326b6080371a56037b3f05ccb5461b53b9d2369d4f5bfde84dedd1e5770d0fe0443d2
+  metadata.gz: 94b4ec4ac99c756f5887d9f5526ac931df163675e0dcd7a460a9703847b253a5c2852d744f227704f847791bb183f6a1226178ee0714b26eb6fc4249d68a3e17
+  data.tar.gz: 501cce3bfee6edd4123a24f337325668e2a4009523814b5d3989fa197a4d4fc2185587dfd3bdf694f2ceed3ba0330e14ceeeac54f351f4f8e30c91b4310c7a97

data/lib/mailkite.rb

@@ -15,7 +15,7 @@ require "base64"
 require "cgi"
 
 module Mailkite
-  VERSION = "0.4.0"
+  VERSION = "0.6.0"
   DEFAULT_BASE_URL = "https://api.mailkite.dev"
   # Reject webhook events older than this (ms) to block replays. Pass 0 to disable.
   DEFAULT_TOLERANCE_MS = 5 * 60 * 1000
@@ -61,6 +61,24 @@ module Mailkite
     '{"status":"ok"}'
   end
 
+  # Control-mode reply telling MailKite to mark the message as spam.
+  # Returns the exact string `{"status":"spam"}`.
+  def self.reply_spam
+    '{"status":"spam"}'
+  end
+
+  # Control-mode reply telling MailKite to drop (discard) the message.
+  # Returns the exact string `{"status":"drop"}`.
+  def self.reply_drop
+    '{"status":"drop"}'
+  end
+
+  # Control-mode reply telling MailKite to block the sender.
+  # Returns the exact string `{"status":"ok","actions":[{"type":"block-sender"}]}`.
+  def self.reply_block_sender
+    '{"status":"ok","actions":[{"type":"block-sender"}]}'
+  end
+
   # Encrypt a UTF-8 string to a customer RSA public key (SPKI/PEM), producing the
   # MailKite at-rest envelope as a compact JSON string. Hybrid scheme: a fresh
   # AES-256-GCM content key encrypts the data, then the raw key is wrapped with
@@ -147,6 +165,27 @@ module Mailkite
         req.body = JSON.generate(body)
       end
 
+      perform(req, uri)
+    end
+
+    # Raw-binary request used by uploadAttachment for `bytes`/`path` uploads:
+    # the body is the file bytes themselves (not JSON, not multipart) and the
+    # Content-Type names the file's media type. Same auth + response parsing.
+    def request_binary(bytes, filename:, content_type:, retention_days: nil)
+      query = { "filename" => filename }
+      query["retentionDays"] = retention_days unless retention_days.nil?
+      qs = query.map { |k, v| "#{CGI.escape(k)}=#{CGI.escape(v.to_s)}" }.join("&")
+      uri = URI("#{@base_url}/v1/attachments?#{qs}")
+      req = Net::HTTP::Post.new(uri)
+      req["Authorization"] = "Bearer #{@api_key}"
+      req["Content-Type"] = content_type
+      req.body = bytes
+      perform(req, uri)
+    end
+
+    # Send a prepared Net::HTTP request and parse the JSON response (shared by
+    # `request` and `request_binary`).
+    def perform(req, uri)
       res = Net::HTTP.start(uri.hostname, uri.port, use_ssl: uri.scheme == "https") { |http| http.request(req) }
       text = res.body
       data = text && !text.empty? ? JSON.parse(text) : nil
@@ -166,6 +205,75 @@ module Mailkite
       request("POST", "/v1/send", message)
     end
 
+    # Extension → MIME map for raw-binary attachment uploads (default
+    # application/octet-stream when an extension is unknown).
+    ATTACHMENT_MIME = {
+      "pdf" => "application/pdf",
+      "png" => "image/png",
+      "jpg" => "image/jpeg",
+      "jpeg" => "image/jpeg",
+      "gif" => "image/gif",
+      "webp" => "image/webp",
+      "svg" => "image/svg+xml",
+      "csv" => "text/csv",
+      "txt" => "text/plain",
+      "html" => "text/html",
+      "json" => "application/json",
+      "zip" => "application/zip",
+      "doc" => "application/msword",
+      "docx" => "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+      "xls" => "application/vnd.ms-excel",
+      "xlsx" => "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+      "ics" => "text/calendar",
+      "ical" => "text/calendar",
+    }.freeze
+
+    # Upload a file and get back a secure, time-limited URL to reference as a
+    # send() attachment ({ filename, url }) or link inline — instead of
+    # base64-inlining large files on every send. Provide the file ONE of four
+    # ways via the `file` hash (priority order):
+    #   1. url            — MailKite fetches & re-hosts the remote file
+    #   2. bytes          — raw binary string, uploaded directly
+    #   3. path           — local file read off disk, then uploaded as bytes
+    #   4. content        — base64 string (the original behavior)
+    # Plus optional filename, contentType, and retentionDays.
+    def uploadAttachment(file)
+      url = file["url"] || file[:url]
+      bytes = file["bytes"] || file[:bytes]
+      path = file["path"] || file[:path]
+      content = file["content"] || file[:content]
+      filename = file["filename"] || file[:filename]
+      content_type = file["contentType"] || file[:contentType]
+      retention_days = file["retentionDays"] || file[:retentionDays]
+
+      if url
+        body = { "url" => url }
+        body["filename"] = filename unless filename.nil?
+        body["contentType"] = content_type unless content_type.nil?
+        body["retentionDays"] = retention_days unless retention_days.nil?
+        return request("POST", "/v1/attachments", body)
+      end
+
+      if bytes || path
+        if path
+          bytes = File.binread(path)
+          filename ||= File.basename(path)
+          content_type ||= ATTACHMENT_MIME[File.extname(path).downcase.delete_prefix(".")]
+        end
+        content_type ||= "application/octet-stream"
+        return request_binary(bytes, filename: filename, content_type: content_type, retention_days: retention_days)
+      end
+
+      if content
+        body = { "content" => content, "filename" => filename }
+        body["contentType"] = content_type unless content_type.nil?
+        body["retentionDays"] = retention_days unless retention_days.nil?
+        return request("POST", "/v1/attachments", body)
+      end
+
+      raise ArgumentError, "uploadAttachment requires one of: path, bytes, url, or content"
+    end
+
     # Run an inbound message through an AI agent. message keys: text (required),
     # plus optional subject/from/html/routeId/address/model.
     def agent(message)
@@ -270,6 +378,18 @@ module Mailkite
       Mailkite.reply_ok
     end
 
+    def reply_spam
+      Mailkite.reply_spam
+    end
+
+    def reply_drop
+      Mailkite.reply_drop
+    end
+
+    def reply_block_sender
+      Mailkite.reply_block_sender
+    end
+
     def encrypt(plaintext, public_key)
       Mailkite.encrypt(plaintext, public_key)
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mailkite
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.6.0
 platform: ruby
 authors:
 - MailKite