mail_mcp 1.0.0 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f3aaf842d07440eb53310d1d076f7503945be59d985e769f5581be72217c1ea
-  data.tar.gz: cd6cca99c6a414586cf85f7c6b5702fd0ef9564a567d8fbc5bb47f686caefcaa
+  metadata.gz: caf27cd17b2adac73fae31883b7f0ae1e20b7c2be61e583180d9fe7d230e3c23
+  data.tar.gz: bab6dfc1f67f681fd30689a4c38eb4ab4cac6ec0abe485755e74f25333e1c761
 SHA512:
-  metadata.gz: 96b7d8751afc7f21a085b4e69a4d33163256cb05423f756c879a90732f0e262df584a09a19f7edd37a8cb862ffb8f28ac35240f074b3babdffd6beaaeef796ef
-  data.tar.gz: 05cc473e8ecb80d69d8408db74e8c9a7272ed618f2434c053211a215d4d57832f401b9054fddfac2f07df4d58150b21e46c46af78bd98699eda0a105fa02e8a1
+  metadata.gz: bad25881c7462f141971105d5455b9cec71af498aa3dadc7617caea6ba2aa5802cfa91b4f899ef637bafcbf8d23c7d902a8f3af3c942b9e83c1f26e324433fc1
+  data.tar.gz: 438d165458bf0b62c1a66d45ad65766f99c9517e5d0d986b1be4e370968a53e0d81254a83a17152eecdc7117f19cdca473df4d8bfffffa365c0398ff4548f62e

data/lib/mail_mcp/app.rb

@@ -144,7 +144,7 @@ module MailMCP
 
     def resolve_mcp_context
       auth = request.env["HTTP_AUTHORIZATION"]
-      return [nil, nil] unless auth&.start_with?("Bearer ")
+      return [nil, unauthorized_response("missing bearer token")] unless auth&.start_with?("Bearer ")
 
       creds = JwtService.verify(auth[7..])
       context = CredentialContext.new(
@@ -161,12 +161,16 @@ module MailMCP
       )
       [context, nil]
     rescue JwtService::Error => e
-      error = [
+      [nil, unauthorized_response(e.message)]
+    end
+
+    # 401 challenge that tells an MCP client where to run the OAuth flow.
+    def unauthorized_response(description)
+      [
         401,
         { "Content-Type" => "application/json", "WWW-Authenticate" => mcp_www_authenticate },
-        [JSON.generate({ error: "invalid_token", error_description: e.message })]
+        [JSON.generate({ error: "invalid_token", error_description: description })]
       ]
-      [nil, error]
     end
 
     def exchange_code(params, client_id)

data/lib/mail_mcp/imap_client.rb

@@ -190,12 +190,15 @@ module MailMCP
 
     def extract_attachments(mail)
       mail.attachments.map do |att|
+        # Mail#decoded does not memoize — it base64-decodes into a new String on
+        # every call, so decode once and reuse for both the upload and the size.
+        content = att.decoded
         url = AttachmentStore.upload(
-          content: att.decoded,
+          content: content,
           filename: att.filename || "attachment",
           content_type: att.content_type
         )
-        { filename: att.filename, content_type: att.content_type, size: att.decoded.bytesize, url: url }
+        { filename: att.filename, content_type: att.content_type, size: content.bytesize, url: url }
       end
     end
   end

data/lib/mail_mcp/tools/update_mail_message_flags_tool.rb

@@ -10,25 +10,56 @@ module MailMCP
       open_world_hint: true
     )
 
+    # IMAP allows only these six backslash-prefixed system flags; anything else
+    # must be sent as a keyword (a bare atom with no leading backslash).
+    SYSTEM_FLAGS = %w[Seen Answered Flagged Deleted Draft Recent].freeze
+
+    # Friendly color names map to Mozilla/Thunderbird $labelN keywords, the
+    # de-facto cross-client convention for colored flags.
+    COLOR_KEYWORDS = {
+      "red" => "$label1",
+      "orange" => "$label2",
+      "green" => "$label3",
+      "blue" => "$label4",
+      "purple" => "$label5"
+    }.freeze
+
     input_schema(
       type: "object",
       properties: {
         folder: { type: "string" },
         uid: { type: "integer" },
         add: { type: "array", items: { type: "string" },
-               description: "Flags to add, e.g. ['\\\\Seen', '\\\\Flagged']" },
+               description: "Flags to add. Settable system flags: '\\\\Seen', '\\\\Answered', " \
+                            "'\\\\Flagged', '\\\\Deleted', '\\\\Draft' (\\\\Recent is recognized " \
+                            "but cannot be set by clients). Color names " \
+                            "('red', 'orange', 'green', 'blue', 'purple') map to " \
+                            "$labelN keywords. Any other value is sent as a custom keyword." },
         remove: { type: "array", items: { type: "string" }, description: "Flags to remove" }
       },
       required: %w[folder uid]
     )
 
     def self.call(folder:, uid:, server_context:, add: [], remove: [])
-      add_flags    = add.map(&:to_sym)
-      remove_flags = remove.map(&:to_sym)
+      add_flags    = add.map { |f| normalize_flag(f) }
+      remove_flags = remove.map { |f| normalize_flag(f) }
       ImapClient.connect(server_context.imap_config) do |c|
         c.update_flags(folder: folder, uid: uid, add: add_flags, remove: remove_flags)
       end
       MCP::Tool::Response.new([{ type: "text", text: "Flags updated for message #{uid}" }])
     end
+
+    # System flags become Symbols (net/imap renders them with a leading
+    # backslash). Custom keywords stay Strings (rendered as bare atoms).
+    # Friendly color names are translated to their $labelN keyword first.
+    def self.normalize_flag(flag)
+      name = flag.to_s.delete_prefix("\\")
+      return COLOR_KEYWORDS.fetch(name.downcase) if COLOR_KEYWORDS.key?(name.downcase)
+
+      # System flags are case-insensitive per RFC 3501; normalize to the
+      # canonical capitalization so e.g. "seen" / "\\SEEN" still set :Seen.
+      system_flag = SYSTEM_FLAGS.find { |f| f.casecmp?(name) }
+      system_flag ? system_flag.to_sym : name
+    end
   end
 end

data/lib/mail_mcp/version.rb

@@ -1,3 +1,3 @@
 module MailMCP
-  VERSION = "1.0.0".freeze
+  VERSION = "1.0.2".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mail_mcp
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.2
 platform: ruby
 authors:
 - Edgars Beigarts