Class: MailMCP::ImapClient

Inherits:
Object
  • Object
show all
Defined in:
lib/mail_mcp/imap_client.rb

Defined Under Namespace

Classes: AuthError, ConnectionError

Constant Summary collapse

OPEN_TIMEOUT = 
10
IDLE_TIMEOUT = 
30

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(imap) ⇒ ImapClient

Returns a new instance of ImapClient.



13
14
15
 
# File 'lib/mail_mcp/imap_client.rb', line 13

def initialize(imap)
  @imap = imap
end

Instance Attribute Details

#imap ⇒ Object (readonly)

Returns the value of attribute imap.



11
12
13
 
# File 'lib/mail_mcp/imap_client.rb', line 11

def imap
  @imap
end

Class Method Details

.connect(config) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
 
# File 'lib/mail_mcp/imap_client.rb', line 32

def self.connect(config)
  conn = open_connection(config)
  conn.(config[:username], config[:password])
  client = new(conn)
  yield client
rescue Net::IMAP::NoResponseError, Net::IMAP::BadResponseError => e
  MailMCP.logger.warn { "IMAP authentication failed user=#{config[:username]}: #{e.message}" }
  raise AuthError, "IMAP authentication failed: #{e.message}"
rescue StandardError => e
  MailMCP.logger.error do
    "IMAP connection failed host=#{config[:host]}:#{config[:port]} ssl=#{config[:ssl]}: #{e.class}: #{e.message}"
  end
  raise ConnectionError, "IMAP connection failed: #{e.message}"
ensure
  begin
    conn&.logout
  rescue StandardError
    nil
  end
  begin
    conn&.disconnect
  rescue StandardError
    nil
  end
end

.validate!(config) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/mail_mcp/imap_client.rb', line 17

def self.validate!(config)
  conn = open_connection(config)
  conn.(config[:username], config[:password])
  conn.logout
  conn.disconnect
rescue Net::IMAP::NoResponseError, Net::IMAP::BadResponseError => e
  MailMCP.logger.warn { "IMAP authentication failed user=#{config[:username]}: #{e.message}" }
  raise AuthError, "IMAP authentication failed: #{e.message}"
rescue StandardError => e
  MailMCP.logger.error do
    "IMAP connection failed host=#{config[:host]}:#{config[:port]} ssl=#{config[:ssl]}: #{e.class}: #{e.message}"
  end
  raise ConnectionError, "IMAP connection failed: #{e.message}"
end

Instance Method Details

#append_message(folder:, raw_message:, flags: [:Seen]) ⇒ Object 



127
128
129
130
131
132
 
# File 'lib/mail_mcp/imap_client.rb', line 127

def append_message(folder:, raw_message:, flags: [:Seen])
  MailMCP.logger.info do
    "IMAP append_message folder=#{folder.inspect} flags=#{flags.inspect} bytes=#{raw_message.bytesize}"
  end
  @imap.append(folder, raw_message, flags, Time.now)
end

#delete_message(folder:, uid:) ⇒ Object 



98
99
100
101
102
103
 
# File 'lib/mail_mcp/imap_client.rb', line 98

def delete_message(folder:, uid:)
  MailMCP.logger.info { "IMAP delete_message folder=#{folder.inspect} uid=#{uid}" }
  @imap.select(folder)
  @imap.uid_store(uid.to_i, "+FLAGS", [:Deleted])
  @imap.expunge
end

#get_message(folder:, uid:) ⇒ Object 



78
79
80
81
82
83
84
85
86
87
88
 
# File 'lib/mail_mcp/imap_client.rb', line 78

def get_message(folder:, uid:)
  MailMCP.logger.info { "IMAP get_message folder=#{folder.inspect} uid=#{uid}" }
  @imap.examine(folder)
  data = @imap.uid_fetch([uid.to_i], %w[RFC822 FLAGS]).first
  unless data
    MailMCP.logger.warn { "IMAP get_message not found folder=#{folder.inspect} uid=#{uid}" }
    return nil
  end

  format_message(uid: uid, parsed: Mail.new(data.attr["RFC822"]), flags: data.attr["FLAGS"])
end

#list_mailboxes ⇒ Object 



58
59
60
61
 
# File 'lib/mail_mcp/imap_client.rb', line 58

def list_mailboxes
  MailMCP.logger.info { "IMAP list_mailboxes" }
  @imap.list("", "*").map(&:name)
end

#list_messages(folder:, page: 1, per_page: 20) ⇒ Object 



63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/mail_mcp/imap_client.rb', line 63

def list_messages(folder:, page: 1, per_page: 20)
  MailMCP.logger.info { "IMAP list_messages folder=#{folder.inspect} page=#{page} per_page=#{per_page}" }
  @imap.examine(folder)
  uids = @imap.uid_search(["ALL"]).reverse
  total = uids.length
  offset = (page - 1) * per_page
  page_uids = uids[offset, per_page] || []
  MailMCP.logger.debug { "IMAP list_messages folder=#{folder.inspect} total=#{total} returned=#{page_uids.size}" }
  return { messages: [], total: total } if page_uids.empty?

  envelopes = @imap.uid_fetch(page_uids, ["ENVELOPE", "FLAGS", "RFC822.SIZE"])
  messages = (envelopes || []).map { |msg| format_envelope(msg) }
  { messages: messages, total: total, page: page, per_page: per_page }
end

#move_message(folder:, uid:, destination:) ⇒ Object 



105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/mail_mcp/imap_client.rb', line 105

def move_message(folder:, uid:, destination:)
  MailMCP.logger.info { "IMAP move_message folder=#{folder.inspect} uid=#{uid} destination=#{destination.inspect}" }
  @imap.select(folder)
  if @imap.capability.include?("MOVE")
    @imap.uid_move(uid.to_i, destination)
  else
    MailMCP.logger.debug { "IMAP move_message falling back to copy+delete (server lacks MOVE)" }
    @imap.uid_copy(uid.to_i, destination)
    @imap.uid_store(uid.to_i, "+FLAGS", [:Deleted])
    @imap.expunge
  end
end

#search_messages(folder:, query:) ⇒ Object 



90
91
92
93
94
95
96
 
# File 'lib/mail_mcp/imap_client.rb', line 90

def search_messages(folder:, query:)
  MailMCP.logger.info { "IMAP search_messages folder=#{folder.inspect} query=#{query.inspect}" }
  @imap.examine(folder)
  results = @imap.search(query.split)
  MailMCP.logger.debug { "IMAP search_messages matched=#{results.size}" }
  results
end

#update_flags(folder:, uid:, add: [], remove: []) ⇒ Object 



118
119
120
121
122
123
124
125
 
# File 'lib/mail_mcp/imap_client.rb', line 118

def update_flags(folder:, uid:, add: [], remove: [])
  MailMCP.logger.info do
    "IMAP update_flags folder=#{folder.inspect} uid=#{uid} add=#{add.inspect} remove=#{remove.inspect}"
  end
  @imap.select(folder)
  @imap.uid_store(uid.to_i, "+FLAGS", add) unless add.empty?
  @imap.uid_store(uid.to_i, "-FLAGS", remove) unless remove.empty?
end