mailcatcher-ng 1.4.0 → 1.5.2

data/lib/mail_catcher/integrations.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require "mail_catcher/integrations/mcp_tools"
+require "mail_catcher/integrations/mcp_server"
+
+module MailCatcher
+  # Integration manager for optional features like MCP and Claude Plugins
+  module Integrations
+    extend self
+
+    attr_accessor :mcp_server
+
+    def initialize
+      @mcp_server = nil
+    end
+
+    # Start integrations based on options
+    def start(options = {})
+      $stderr.puts "[Integrations] Starting integrations with options: #{options.inspect}"
+
+      if options[:mcp_enabled]
+        start_mcp_server(options)
+      end
+    end
+
+    # Start the MCP server
+    def start_mcp_server(options = {})
+      $stderr.puts "[Integrations] Starting MCP server"
+      @mcp_server = MCPServer.new(options)
+
+      # Run MCP server in a separate thread
+      Thread.new do
+        begin
+          @mcp_server.run
+        rescue => e
+          $stderr.puts "[Integrations] MCP server error: #{e.message}"
+          $stderr.puts e.backtrace
+        end
+      end
+
+      # Give the server a moment to start
+      sleep 0.1
+      $stderr.puts "[Integrations] MCP server started"
+    end
+
+    # Stop all integrations
+    def stop
+      $stderr.puts "[Integrations] Stopping integrations"
+      @mcp_server&.stop
+      @mcp_server = nil
+    end
+
+    # Check if MCP server is running
+    def mcp_running?
+      @mcp_server&.running
+    end
+
+    # Get available tools
+    def available_tools
+      MCPTools.tool_names
+    end
+  end
+end

data/lib/mail_catcher/mail.rb

@@ -4,7 +4,9 @@ require "eventmachine"
 require "fileutils"
 require "json"
 require "mail"
+require "nokogiri"
 require "sqlite3"
+require "uri"
 
 module MailCatcher::Mail extend self
   def db
@@ -57,8 +59,24 @@ module MailCatcher::Mail extend self
             FOREIGN KEY (message_id) REFERENCES message (id) ON DELETE CASCADE
           )
         SQL
+        db.execute(<<-SQL)
+          CREATE TABLE IF NOT EXISTS websocket_connection (
+            id INTEGER PRIMARY KEY ASC,
+            session_id TEXT NOT NULL,
+            client_ip TEXT,
+            opened_at DATETIME,
+            closed_at DATETIME,
+            last_ping_at DATETIME,
+            last_pong_at DATETIME,
+            ping_count INTEGER DEFAULT 0,
+            pong_count INTEGER DEFAULT 0,
+            created_at DATETIME DEFAULT CURRENT_DATETIME,
+            updated_at DATETIME DEFAULT CURRENT_DATETIME
+          )
+        SQL
         db.execute("CREATE INDEX IF NOT EXISTS idx_smtp_transcript_message_id ON smtp_transcript(message_id)")
         db.execute("CREATE INDEX IF NOT EXISTS idx_smtp_transcript_session_id ON smtp_transcript(session_id)")
+        db.execute("CREATE INDEX IF NOT EXISTS idx_websocket_connection_session_id ON websocket_connection(session_id)")
         db.execute("PRAGMA foreign_keys = ON")
       end
     end
@@ -118,6 +136,62 @@ module MailCatcher::Mail extend self
     end
   end
 
+  def search_messages(query: nil, has_attachments: nil, from_date: nil, to_date: nil)
+    # Build dynamic SQL query with filters
+    sql = "SELECT DISTINCT m.id, m.sender, m.recipients, m.subject, m.size, m.created_at FROM message m"
+    params = []
+    where_clauses = []
+
+    # Determine if we need to join with message_part table
+    needs_join = (query && !query.strip.empty?) || has_attachments.is_a?(TrueClass)
+
+    # Join with message_part table if needed
+    if needs_join
+      sql += " LEFT JOIN message_part mp ON m.id = mp.message_id"
+    end
+
+    # Add search filters - search across subject, sender, and recipients (always available)
+    # Also search body if we have the JOIN
+    if query && !query.strip.empty?
+      q = "%#{query}%"
+      if needs_join
+        where_clauses << "(m.subject LIKE ? OR m.sender LIKE ? OR m.recipients LIKE ? OR mp.body LIKE ?)"
+        params.concat([q, q, q, q])
+      else
+        where_clauses << "(m.subject LIKE ? OR m.sender LIKE ? OR m.recipients LIKE ?)"
+        params.concat([q, q, q])
+      end
+    end
+
+    # Add attachment filter
+    if has_attachments.is_a?(TrueClass)
+      where_clauses << "(mp.is_attachment = 1)"
+    end
+
+    # Add date range filters
+    if from_date
+      where_clauses << "(m.created_at >= ?)"
+      params << from_date
+    end
+
+    if to_date
+      where_clauses << "(m.created_at <= ?)"
+      params << to_date
+    end
+
+    # Combine where clauses
+    sql += " WHERE #{where_clauses.join(' AND ')}" if where_clauses.any?
+
+    sql += " ORDER BY m.created_at, m.id ASC"
+
+    db.prepare(sql).execute(*params).map do |row|
+      columns = ["id", "sender", "recipients", "subject", "size", "created_at"]
+      Hash[columns.zip(row)].tap do |message|
+        message["recipients"] &&= JSON.parse(message["recipients"])
+      end
+    end
+  end
+
   def message(id)
     @message_query ||= db.prepare "SELECT id, sender, recipients, subject, size, type, created_at FROM message WHERE id = ? LIMIT 1"
     row = @message_query.execute(id).next
@@ -354,6 +428,112 @@ module MailCatcher::Mail extend self
     end
   end
 
+  def extract_tokens(id, type:)
+    html_part = message_part_html(id)
+    plain_part = message_part_plain(id)
+
+    content = [html_part&.dig('body'), plain_part&.dig('body')].compact.join("\n")
+
+    case type
+    when 'link' then extract_magic_links(content)
+    when 'otp' then extract_otps(content)
+    when 'token' then extract_reset_tokens(content)
+    else []
+    end
+  end
+
+  def extract_all_links(id)
+    links = []
+
+    if html_part = message_part_html(id)
+      links += extract_links_from_html(html_part['body'])
+    end
+
+    if plain_part = message_part_plain(id)
+      links += extract_links_from_plain(plain_part['body'])
+    end
+
+    links
+  end
+
+  def parse_message_structured(id)
+    # Extract unsubscribe from List-Unsubscribe header
+    source = message_source(id)
+    unsubscribe_link = nil
+
+    if source
+      unsubscribe_header = source.lines.find { |l| l.match?(/^List-Unsubscribe:/i) }
+      unsubscribe_link = unsubscribe_header&.match(/<(https?:\/\/[^>]+)>/)&.[](1)
+    end
+
+    {
+      verification_url: extract_tokens(id, type: 'link').first&.dig(:value),
+      otp_code: extract_tokens(id, type: 'otp').first&.dig(:value),
+      reset_token: extract_tokens(id, type: 'token').first&.dig(:value),
+      unsubscribe_link: unsubscribe_link,
+      all_links: extract_all_links(id)
+    }
+  end
+
+  def accessibility_score(id)
+    html_part = message_part_html(id)
+    return { score: 0, error: 'No HTML part found' } unless html_part
+
+    doc = Nokogiri::HTML(html_part['body'])
+
+    alt_text_data = check_alt_text_detailed(doc)
+    semantic_data = check_semantic_html_detailed(doc)
+    links_data = check_links_detailed(doc)
+
+    scores = {
+      images_with_alt: alt_text_data[:score],
+      semantic_html: semantic_data[:score],
+      links_with_text: links_data[:score]
+    }
+
+    total_score = (scores.values.sum / scores.size.to_f).round
+
+    {
+      score: total_score,
+      breakdown: scores,
+      findings: {
+        images: alt_text_data[:findings],
+        semantic: semantic_data[:findings],
+        links: links_data[:findings]
+      },
+      recommendations: generate_recommendations(scores)
+    }
+  end
+
+  def forward_message(id)
+    return { error: 'SMTP not configured' } unless forward_smtp_configured?
+
+    message = message(id)
+    source = message_source(id)
+    recipients = JSON.parse(message['recipients'])
+
+    require 'net/smtp'
+
+    Net::SMTP.start(
+      MailCatcher.options[:forward_smtp_host],
+      MailCatcher.options[:forward_smtp_port] || 587,
+      'localhost',
+      MailCatcher.options[:forward_smtp_user],
+      MailCatcher.options[:forward_smtp_password],
+      :plain
+    ) do |smtp|
+      smtp.send_message(source, message['sender'], recipients)
+    end
+
+    {
+      success: true,
+      forwarded_to: recipients,
+      forwarded_at: Time.now.utc.iso8601
+    }
+  rescue => e
+    { error: e.message }
+  end
+
   def message_encryption_data(id)
     source = message_source(id)
     return {} unless source
@@ -505,8 +685,208 @@ module MailCatcher::Mail extend self
     end
   end
 
+  def create_websocket_connection(session_id, client_ip)
+    @create_ws_connection_query ||= db.prepare(<<-SQL)
+      INSERT INTO websocket_connection (session_id, client_ip, opened_at, created_at, updated_at)
+      VALUES (?, ?, datetime('now'), datetime('now'), datetime('now'))
+    SQL
+    @create_ws_connection_query.execute(session_id, client_ip)
+    db.last_insert_row_id
+  end
+
+  def close_websocket_connection(session_id)
+    @close_ws_connection_query ||= db.prepare(<<-SQL)
+      UPDATE websocket_connection
+      SET closed_at = datetime('now'), updated_at = datetime('now')
+      WHERE session_id = ? AND closed_at IS NULL
+    SQL
+    @close_ws_connection_query.execute(session_id)
+  end
+
+  def record_websocket_ping(session_id)
+    @record_ping_query ||= db.prepare(<<-SQL)
+      UPDATE websocket_connection
+      SET last_ping_at = datetime('now'), ping_count = ping_count + 1, updated_at = datetime('now')
+      WHERE session_id = ? AND closed_at IS NULL
+    SQL
+    @record_ping_query.execute(session_id)
+  end
+
+  def record_websocket_pong(session_id)
+    @record_pong_query ||= db.prepare(<<-SQL)
+      UPDATE websocket_connection
+      SET last_pong_at = datetime('now'), pong_count = pong_count + 1, updated_at = datetime('now')
+      WHERE session_id = ? AND closed_at IS NULL
+    SQL
+    @record_pong_query.execute(session_id)
+  end
+
   private
 
+  def extract_magic_links(content)
+    # Extract URLs with common token parameters: token, verify, confirmation, magic
+    # Use non-capturing group (?:...) so scan returns the full match
+    pattern = %r{https?://[^\s<>]+[?&](?:token|verify|confirmation|magic)=[a-zA-Z0-9_\-.~%]+}i
+    content.scan(pattern).compact.map do |url|
+      # Extract surrounding context (50 chars before and after)
+      start_pos = content.index(url)
+      if start_pos
+        context_start = [0, start_pos - 50].max
+        context_end = [content.length, start_pos + url.length + 50].min
+        context = content[context_start...context_end].strip
+
+        {
+          type: 'magic_link',
+          value: url,
+          context: context
+        }
+      end
+    end.compact
+  end
+
+  def extract_otps(content)
+    # Extract 6-digit OTP codes, preferring those near keywords
+    pattern = /\b(\d{6})\b/
+    results = []
+
+    content.scan(pattern).each do |match|
+      otp = match[0]
+      start_pos = content.index(otp)
+      context_start = [0, start_pos - 50].max
+      context_end = [content.length, start_pos + otp.length + 50].min
+      context = content[context_start...context_end].strip
+
+      # Check if near keywords: code, otp, verification, confirm, pin
+      if context.match?(/code|otp|verification|confirm|pin/i)
+        results << {
+          type: 'otp',
+          value: otp,
+          context: context
+        }
+      end
+    end
+
+    results
+  end
+
+  def extract_reset_tokens(content)
+    # Extract reset token URLs
+    pattern = %r{https?://[^\s<>]*reset[^\s<>]*[?&]token=[a-zA-Z0-9_\-.~%]+}i
+    content.scan(pattern).map do |url|
+      start_pos = content.index(url)
+      context_start = [0, start_pos - 50].max
+      context_end = [content.length, start_pos + url.length + 50].min
+      context = content[context_start...context_end].strip
+
+      {
+        type: 'reset_token',
+        value: url,
+        context: context
+      }
+    end
+  end
+
+  def extract_links_from_html(html)
+    doc = Nokogiri::HTML(html)
+    doc.css('a[href]').map do |link|
+      href = link['href']
+      {
+        href: href,
+        text: link.text.strip,
+        is_verification: href.match?(/verify|confirm|token|magic|activate/i),
+        is_unsubscribe: href.match?(/unsubscribe|opt-out|remove/i)
+      }
+    end
+  end
+
+  def extract_links_from_plain(text)
+    text.scan(URI.regexp(['http', 'https'])).map do |match|
+      url = match[0]
+      {
+        href: url,
+        text: url,
+        is_verification: url.match?(/verify|confirm|token|magic|activate/i),
+        is_unsubscribe: url.match?(/unsubscribe|opt-out|remove/i)
+      }
+    end
+  end
+
+  def check_alt_text_detailed(doc)
+    images = doc.css('img')
+    return { score: 100, findings: { total: 0, with_alt: 0, without_alt: [] } } if images.empty?
+
+    with_alt = images.select { |img| img['alt'] && !img['alt'].strip.empty? }
+    without_alt = images.reject { |img| img['alt'] && !img['alt'].strip.empty? }
+
+    findings = {
+      total: images.size,
+      with_alt: with_alt.size,
+      without_alt: without_alt.map { |img| { src: img['src'], alt_missing: img['alt'].nil? } }
+    }
+
+    score = (with_alt.size.to_f / images.size * 100).round
+
+    { score: score, findings: findings }
+  end
+
+  def check_semantic_html_detailed(doc)
+    semantic_tags = doc.css('header, nav, main, article, section, footer, aside')
+    has_semantic = semantic_tags.any?
+
+    findings = {
+      has_semantic_tags: has_semantic,
+      found_tags: semantic_tags.map(&:name).uniq
+    }
+
+    score = has_semantic ? 100 : 50
+
+    { score: score, findings: findings }
+  end
+
+  def check_links_detailed(doc)
+    links = doc.css('a')
+    return { score: 100, findings: { total: 0, with_text: 0, without_text: [] } } if links.empty?
+
+    with_text = links.select { |a| !a.text.strip.empty? || (a['aria-label'] && !a['aria-label'].empty?) }
+    without_text = links.reject { |a| !a.text.strip.empty? || (a['aria-label'] && !a['aria-label'].empty?) }
+
+    findings = {
+      total: links.size,
+      with_text: with_text.size,
+      without_text: without_text.map { |a| { href: a['href'], text_empty: a.text.strip.empty? } }
+    }
+
+    score = (with_text.size.to_f / links.size * 100).round
+
+    { score: score, findings: findings }
+  end
+
+  def check_alt_text(doc)
+    images = doc.css('img')
+    return 100 if images.empty?
+
+    with_alt = images.select { |img| img['alt'] && !img['alt'].strip.empty? }
+    (with_alt.size.to_f / images.size * 100).round
+  end
+
+  def check_semantic_html(doc)
+    semantic_tags = doc.css('header, nav, main, article, section, footer, aside')
+    semantic_tags.any? ? 100 : 50
+  end
+
+  def generate_recommendations(scores)
+    recs = []
+    recs << "Add alt text to all images" if scores[:images_with_alt] < 100
+    recs << "Use semantic HTML tags (header, main, article, section)" if scores[:semantic_html] < 100
+    recs << "Ensure all links have descriptive text or aria-label" if scores[:links_with_text] < 100
+    recs
+  end
+
+  def forward_smtp_configured?
+    MailCatcher.options[:forward_smtp_host] &&
+      MailCatcher.options[:forward_smtp_port]
+  end
+
   def parse_authentication_results(auth_header)
     results = {
       dmarc: nil,

data/lib/mail_catcher/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MailCatcher
-  VERSION = '1.4.0'
+  VERSION = '1.5.2'
 end