cgit.sxemacs.org Git - gnus/blob - lisp/nnimap.el

   1 ;;; nnimap.el --- IMAP interface for Gnus
   2
   3 ;; Copyright (C) 2010 Free Software Foundation, Inc.
   4
   5 ;; Author: Lars Magne Ingebrigtsen <larsi@gnus.org>
   6 ;;         Simon Josefsson <simon@josefsson.org>
   7
   8 ;; This file is part of GNU Emacs.
   9
  10 ;; GNU Emacs is free software: you can redistribute it and/or modify
  11 ;; it under the terms of the GNU General Public License as published by
  12 ;; the Free Software Foundation, either version 3 of the License, or
  13 ;; (at your option) any later version.
  14
  15 ;; GNU Emacs is distributed in the hope that it will be useful,
  16 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 ;; GNU General Public License for more details.
  19
  20 ;; You should have received a copy of the GNU General Public License
  21 ;; along with GNU Emacs.  If not, see <http://www.gnu.org/licenses/>.
  22
  23 ;;; Commentary:
  24
  25 ;; nnimap interfaces Gnus with IMAP servers.
  26
  27 ;;; Code:
  28
  29 (eval-and-compile
  30   (require 'nnheader))
  31
  32 (eval-when-compile
  33   (require 'cl))
  34
  35 (require 'nnheader)
  36 (require 'gnus-util)
  37 (require 'gnus)
  38 (require 'nnoo)
  39 (require 'netrc)
  40 (require 'parse-time)
  41
  42 (nnoo-declare nnimap)
  43
  44 (defvoo nnimap-address nil
  45   "The address of the IMAP server.")
  46
  47 (defvoo nnimap-server-port nil
  48   "The IMAP port used.
  49 If nnimap-stream is `ssl', this will default to `imaps'.  If not,
  50 it will default to `imap'.")
  51
  52 (defvoo nnimap-stream 'ssl
  53   "How nnimap will talk to the IMAP server.
  54 Values are `ssl' and `network'.")
  55
  56 (defvoo nnimap-shell-program (if (boundp 'imap-shell-program)
  57                                  (if (listp imap-shell-program)
  58                                      (car imap-shell-program)
  59                                    imap-shell-program)
  60                                "ssh %s imapd"))
  61
  62 (defvoo nnimap-inbox nil
  63   "The mail box where incoming mail arrives and should be split out of.")
  64
  65 (defvoo nnimap-authenticator nil
  66   "How nnimap authenticate itself to the server.
  67 Possible choices are nil (use default methods) or `anonymous'.")
  68
  69 (defvoo nnimap-fetch-partial-articles nil
  70   "If non-nil, nnimap will fetch partial articles.
  71 If t, nnimap will fetch only the first part.  If a string, it
  72 will fetch all parts that have types that match that string.  A
  73 likely value would be \"text/\" to automatically fetch all
  74 textual parts.")
  75
  76 (defvoo nnimap-expunge t
  77   "If non-nil, expunge articles after deleting them.
  78 This is always done if the server supports UID EXPUNGE, but it's
  79 not done by default on servers that doesn't support that command.")
  80
  81
  82 (defvoo nnimap-connection-alist nil)
  83
  84 (defvoo nnimap-current-infos nil)
  85
  86 (defvar nnimap-process nil)
  87
  88 (defvar nnimap-status-string "")
  89
  90 (defvar nnimap-split-download-body-default nil
  91   "Internal variable with default value for `nnimap-split-download-body'.")
  92
  93 (defstruct nnimap
  94   group process commands capabilities select-result newlinep)
  95
  96 (defvar nnimap-object nil)
  97
  98 (defvar nnimap-mark-alist
  99   '((read "\\Seen")
 100     (tick "\\Flagged")
 101     (reply "\\Answered")
 102     (expire "gnus-expire")
 103     (dormant "gnus-dormant")
 104     (score "gnus-score")
 105     (save "gnus-save")
 106     (download "gnus-download")
 107     (forward "gnus-forward")))
 108
 109 (defvar nnimap-split-methods nil)
 110
 111 (defun nnimap-buffer ()
 112   (nnimap-find-process-buffer nntp-server-buffer))
 113
 114 (deffoo nnimap-retrieve-headers (articles &optional group server fetch-old)
 115   (with-current-buffer nntp-server-buffer
 116     (erase-buffer)
 117     (when (nnimap-possibly-change-group group server)
 118       (with-current-buffer (nnimap-buffer)
 119         (nnimap-send-command "SELECT %S" (utf7-encode group t))
 120         (erase-buffer)
 121         (nnimap-wait-for-response
 122          (nnimap-send-command
 123           "UID FETCH %s %s"
 124           (nnimap-article-ranges (gnus-compress-sequence articles))
 125           (format "(UID RFC822.SIZE BODYSTRUCTURE %s)"
 126                   (format
 127                    (if (member "IMAP4REV1"
 128                                (nnimap-capabilities nnimap-object))
 129                        "BODY.PEEK[HEADER.FIELDS %s]"
 130                      "RFC822.HEADER.LINES %s")
 131                    (append '(Subject From Date Message-Id
 132                                      References In-Reply-To Xref)
 133                            nnmail-extra-headers))))
 134          t)
 135         (nnimap-transform-headers))
 136       (insert-buffer-substring
 137        (nnimap-find-process-buffer (current-buffer))))
 138     t))
 139
 140 (defun nnimap-transform-headers ()
 141   (goto-char (point-min))
 142   (let (article bytes lines size)
 143     (block nil
 144       (while (not (eobp))
 145         (while (not (looking-at "^\\* [0-9]+ FETCH.*UID \\([0-9]+\\)"))
 146           (delete-region (point) (progn (forward-line 1) (point)))
 147           (when (eobp)
 148             (return)))
 149         (setq article (match-string 1)
 150               bytes (nnimap-get-length)
 151               lines nil)
 152         (beginning-of-line)
 153         (setq size
 154               (and (re-search-forward "RFC822.SIZE \\([0-9]+\\)"
 155                                       (line-end-position)
 156                                       t)
 157                    (match-string 1)))
 158         (beginning-of-line)
 159         (when (search-forward "BODYSTRUCTURE" (line-end-position) t)
 160           (let ((structure (ignore-errors (read (current-buffer)))))
 161             (while (and (consp structure)
 162                         (not (stringp (car structure))))
 163               (setq structure (car structure)))
 164             (setq lines (nth 7 structure))))
 165         (delete-region (line-beginning-position) (line-end-position))
 166         (insert (format "211 %s Article retrieved." article))
 167         (forward-line 1)
 168         (when size
 169           (insert (format "Chars: %s\n" size)))
 170         (when lines
 171           (insert (format "Lines: %s\n" lines)))
 172         (re-search-forward "^\r$")
 173         (delete-region (line-beginning-position) (line-end-position))
 174         (insert ".")
 175         (forward-line 1)))))
 176
 177 (defun nnimap-get-length ()
 178   (and (re-search-forward "{\\([0-9]+\\)}" (line-end-position) t)
 179        (string-to-number (match-string 1))))
 180
 181 (defun nnimap-article-ranges (ranges)
 182   (let (result)
 183     (cond
 184      ((numberp ranges)
 185       (number-to-string ranges))
 186      ((numberp (cdr ranges))
 187       (format "%d:%d" (car ranges) (cdr ranges)))
 188      (t
 189       (dolist (elem ranges)
 190         (push
 191          (if (consp elem)
 192              (format "%d:%d" (car elem) (cdr elem))
 193            (number-to-string elem))
 194          result))
 195       (mapconcat #'identity (nreverse result) ",")))))
 196
 197 (deffoo nnimap-open-server (server &optional defs)
 198   (if (nnimap-server-opened server)
 199       t
 200     (unless (assq 'nnimap-address defs)
 201       (setq defs (append defs (list (list 'nnimap-address server)))))
 202     (nnoo-change-server 'nnimap server defs)
 203     (or (nnimap-find-connection nntp-server-buffer)
 204         (nnimap-open-connection nntp-server-buffer))))
 205
 206 (defun nnimap-make-process-buffer (buffer)
 207   (with-current-buffer
 208       (generate-new-buffer (format "*nnimap %s %s %s*"
 209                                    nnimap-address nnimap-server-port
 210                                    (gnus-buffer-exists-p buffer)))
 211     (mm-disable-multibyte)
 212     (buffer-disable-undo)
 213     (gnus-add-buffer)
 214     (set (make-local-variable 'after-change-functions) nil)
 215     (set (make-local-variable 'nnimap-object) (make-nnimap))
 216     (push (list buffer (current-buffer)) nnimap-connection-alist)
 217     (current-buffer)))
 218
 219 (defun nnimap-open-shell-stream (name buffer host port)
 220   (let ((process-connection-type nil))
 221     (start-process name buffer shell-file-name
 222                    shell-command-switch
 223                    (format-spec
 224                     nnimap-shell-program
 225                     (format-spec-make
 226                      ?s host
 227                      ?p port)))))
 228
 229 (defun nnimap-credentials (address ports)
 230   (let (port credentials)
 231     ;; Request the credentials from all ports, but only query on the
 232     ;; last port if all the previous ones have failed.
 233     (while (and (null credentials)
 234                 (setq port (pop ports)))
 235       (setq credentials
 236             (auth-source-user-or-password
 237              '("login" "password") address port nil (null ports))))
 238     credentials))
 239
 240 (defun nnimap-open-connection (buffer)
 241   (with-current-buffer (nnimap-make-process-buffer buffer)
 242     (let* ((coding-system-for-read 'binary)
 243            (coding-system-for-write 'binary)
 244            (ports
 245             (cond
 246              ((eq nnimap-stream 'network)
 247               (open-network-stream
 248                "*nnimap*" (current-buffer) nnimap-address
 249                (or nnimap-server-port
 250                    (if (netrc-find-service-number "imap")
 251                        "imap"
 252                      "143")))
 253               '("143" "imap"))
 254              ((eq nnimap-stream 'shell)
 255               (nnimap-open-shell-stream
 256                "*nnimap*" (current-buffer) nnimap-address
 257                (or nnimap-server-port "imap"))
 258               '("imap"))
 259              ((eq nnimap-stream 'ssl)
 260               (open-tls-stream
 261                "*nnimap*" (current-buffer) nnimap-address
 262                (or nnimap-server-port
 263                    (if (netrc-find-service-number "imaps")
 264                        "imaps"
 265                      "993")))
 266               '("143" "993" "imap" "imaps"))))
 267            connection-result login-result credentials)
 268       (setf (nnimap-process nnimap-object)
 269             (get-buffer-process (current-buffer)))
 270       (when (and (nnimap-process nnimap-object)
 271                  (memq (process-status (nnimap-process nnimap-object))
 272                        '(open run)))
 273         (gnus-set-process-query-on-exit-flag (nnimap-process nnimap-object) nil)
 274         (when (setq connection-result (nnimap-wait-for-connection))
 275           (unless (equal connection-result "PREAUTH")
 276             (if (not (setq credentials
 277                            (if (eq nnimap-authenticator 'anonymous)
 278                                (list "anonymous"
 279                                      (message-make-address))
 280                              (nnimap-credentials
 281                               nnimap-address
 282                               (if nnimap-server-port
 283                                   (cons (format "%s" nnimap-server-port) ports)
 284                                 ports)))))
 285                 (setq nnimap-object nil)
 286               (setq login-result (nnimap-command "LOGIN %S %S"
 287                                                  (car credentials)
 288                                                  (cadr credentials)))
 289               (unless (car login-result)
 290                 (delete-process (nnimap-process nnimap-object))
 291                 (setq nnimap-object nil))))
 292           (when nnimap-object
 293             (setf (nnimap-capabilities nnimap-object)
 294                   (mapcar
 295                    #'upcase
 296                    (or (nnimap-find-parameter "CAPABILITY" (cdr login-result))
 297                        (nnimap-find-parameter
 298                         "CAPABILITY" (cdr (nnimap-command "CAPABILITY"))))))
 299             (when (member "QRESYNC" (nnimap-capabilities nnimap-object))
 300               (nnimap-command "ENABLE QRESYNC"))
 301             t))))))
 302
 303 (defun nnimap-find-parameter (parameter elems)
 304   (let (result)
 305     (dolist (elem elems)
 306       (cond
 307        ((equal (car elem) parameter)
 308         (setq result (cdr elem)))
 309        ((and (equal (car elem) "OK")
 310              (consp (cadr elem))
 311              (equal (caadr elem) parameter))
 312         (setq result (cdr (cadr elem))))))
 313     result))
 314
 315 (deffoo nnimap-close-server (&optional server)
 316   t)
 317
 318 (deffoo nnimap-request-close ()
 319   t)
 320
 321 (deffoo nnimap-server-opened (&optional server)
 322   (and (nnoo-current-server-p 'nnimap server)
 323        nntp-server-buffer
 324        (gnus-buffer-live-p nntp-server-buffer)
 325        (nnimap-find-connection nntp-server-buffer)))
 326
 327 (deffoo nnimap-status-message (&optional server)
 328   nnimap-status-string)
 329
 330 (deffoo nnimap-request-article (article &optional group server to-buffer)
 331   (with-current-buffer nntp-server-buffer
 332     (let ((result (nnimap-possibly-change-group group server))
 333           parts)
 334       (when (stringp article)
 335         (setq article (nnimap-find-article-by-message-id group article)))
 336       (when (and result
 337                  article)
 338         (erase-buffer)
 339         (with-current-buffer (nnimap-buffer)
 340           (erase-buffer)
 341           (when nnimap-fetch-partial-articles
 342             (if (eq nnimap-fetch-partial-articles t)
 343                 (setq parts '(1))
 344               (nnimap-command "UID FETCH %d (BODYSTRUCTURE)" article)
 345               (goto-char (point-min))
 346               (when (re-search-forward "FETCH.*BODYSTRUCTURE" nil t)
 347                 (let ((structure (ignore-errors (read (current-buffer)))))
 348                   (setq parts (nnimap-find-wanted-parts structure))))))
 349           (setq result
 350                 (nnimap-command
 351                  (if (member "IMAP4REV1" (nnimap-capabilities nnimap-object))
 352                      "UID FETCH %d BODY.PEEK[]"
 353                    "UID FETCH %d RFC822.PEEK")
 354                  article))
 355           ;; Check that we really got an article.
 356           (goto-char (point-min))
 357           (unless (looking-at "\\* [0-9]+ FETCH")
 358             (setq result nil)))
 359         (let ((buffer (nnimap-find-process-buffer (current-buffer))))
 360           (when (car result)
 361             (with-current-buffer (or to-buffer nntp-server-buffer)
 362               (insert-buffer-substring buffer)
 363               (goto-char (point-min))
 364               (let ((bytes (nnimap-get-length)))
 365                 (delete-region (line-beginning-position)
 366                                (progn (forward-line 1) (point)))
 367                 (goto-char (+ (point) bytes))
 368                 (delete-region (point) (point-max))
 369                 (nnheader-ms-strip-cr))
 370               (cons group article))))))))
 371
 372 (defun nnimap-find-wanted-parts (structure)
 373   (message-flatten-list (nnimap-find-wanted-parts-1 structure "")))
 374
 375 (defun nnimap-find-wanted-parts-1 (structure prefix)
 376   (let ((num 1)
 377         parts)
 378     (while (consp (car structure))
 379       (let ((sub (pop structure)))
 380         (if (consp (car sub))
 381             (push (nnimap-find-wanted-parts-1
 382                    sub (if (string= prefix "")
 383                            (number-to-string num)
 384                          (format "%s.%s" prefix num)))
 385                   parts)
 386           (let ((type (format "%s/%s" (nth 0 sub) (nth 1 sub))))
 387             (when (string-match nnimap-fetch-partial-articles type)
 388               (push (if (string= prefix "")
 389                         (number-to-string num)
 390                       (format "%s.%s" prefix num))
 391                     parts)))
 392           (incf num))))
 393     (nreverse parts)))
 394
 395 (deffoo nnimap-request-group (group &optional server dont-check info)
 396   (let ((result (nnimap-possibly-change-group group server))
 397         articles active marks high low)
 398     (with-current-buffer nntp-server-buffer
 399       (when result
 400         (if (and dont-check
 401                  (setq active (nth 2 (assoc group nnimap-current-infos))))
 402             (insert (format "211 %d %d %d %S\n"
 403                             (- (cdr active) (car active))
 404                             (car active)
 405                             (cdr active)
 406                             group))
 407           (with-current-buffer (nnimap-buffer)
 408             (erase-buffer)
 409             (let ((group-sequence
 410                    (nnimap-send-command "SELECT %S" (utf7-encode group t)))
 411                   (flag-sequence
 412                    (nnimap-send-command "UID FETCH 1:* FLAGS")))
 413               (nnimap-wait-for-response flag-sequence)
 414               (setq marks
 415                     (nnimap-flags-to-marks
 416                      (nnimap-parse-flags
 417                       (list (list group-sequence flag-sequence 1 group)))))
 418               (when info
 419                 (nnimap-update-infos marks (list info)))
 420               (goto-char (point-max))
 421               (cond
 422                (marks
 423                 (setq high (nth 3 (car marks))
 424                       low (nth 4 (car marks))))
 425                ((re-search-backward "UIDNEXT \\([0-9]+\\)" nil t)
 426                 (setq high (1- (string-to-number (match-string 1)))
 427                       low 1)))))
 428           (erase-buffer)
 429           (insert
 430            (format
 431             "211 %d %d %d %S\n" (1+ (- high low)) low high group)))
 432         t))))
 433
 434 (deffoo nnimap-request-create-group (group &optional server args)
 435   (when (nnimap-possibly-change-group nil server)
 436     (with-current-buffer (nnimap-buffer)
 437       (car (nnimap-command "CREATE %S" (utf7-encode group t))))))
 438
 439 (deffoo nnimap-request-delete-group (group &optional force server)
 440   (when (nnimap-possibly-change-group nil server)
 441     (with-current-buffer (nnimap-buffer)
 442       (car (nnimap-command "DELETE %S" (utf7-encode group t))))))
 443
 444 (deffoo nnimap-request-expunge-group (group &optional server)
 445   (when (nnimap-possibly-change-group group server)
 446     (with-current-buffer (nnimap-buffer)
 447       (car (nnimap-command "EXPUNGE")))))
 448
 449 (defun nnimap-get-flags (spec)
 450   (let ((articles nil)
 451         elems)
 452     (with-current-buffer (nnimap-buffer)
 453       (erase-buffer)
 454       (nnimap-wait-for-response (nnimap-send-command
 455                                  "UID FETCH %s FLAGS" spec))
 456       (goto-char (point-min))
 457       (while (re-search-forward "^\\* [0-9]+ FETCH (\\(.*\\))" nil t)
 458         (setq elems (nnimap-parse-line (match-string 1)))
 459         (push (cons (string-to-number (cadr (member "UID" elems)))
 460                     (cadr (member "FLAGS" elems)))
 461               articles)))
 462     (nreverse articles)))
 463
 464 (deffoo nnimap-close-group (group &optional server)
 465   t)
 466
 467 (deffoo nnimap-request-move-article (article group server accept-form
 468                                              &optional last internal-move-group)
 469   (with-temp-buffer
 470     (when (nnimap-request-article article group server (current-buffer))
 471       ;; If the move is internal (on the same server), just do it the easy
 472       ;; way.
 473       (let ((message-id (message-field-value "message-id")))
 474         (if internal-move-group
 475             (let ((result
 476                    (with-current-buffer (nnimap-buffer)
 477                      (nnimap-command "UID COPY %d %S"
 478                                      article
 479                                      (utf7-encode internal-move-group t)))))
 480               (when (car result)
 481                 (nnimap-delete-article article)
 482                 (cons internal-move-group
 483                       (nnimap-find-article-by-message-id
 484                        internal-move-group message-id))))
 485           ;; Move the article to a different method.
 486           (let ((result (eval accept-form)))
 487             (when result
 488               (nnimap-delete-article article)
 489               result)))))))
 490
 491 (deffoo nnimap-request-expire-articles (articles group &optional server force)
 492   (cond
 493    ((null articles)
 494     nil)
 495    ((not (nnimap-possibly-change-group group server))
 496     articles)
 497    ((and force
 498          (eq nnmail-expiry-target 'delete))
 499     (unless (nnimap-delete-article articles)
 500       (message "Article marked for deletion, but not expunged."))
 501     nil)
 502    (t
 503     (let ((deletable-articles
 504            (if force
 505                articles
 506              (gnus-sorted-intersection
 507               articles
 508               (nnimap-find-expired-articles group)))))
 509       (if (null deletable-articles)
 510           articles
 511         (if (eq nnmail-expiry-target 'delete)
 512             (nnimap-delete-article deletable-articles)
 513           (setq deletable-articles
 514                 (nnimap-process-expiry-targets
 515                  deletable-articles group server)))
 516         ;; Return the articles we didn't delete.
 517         (gnus-sorted-complement articles deletable-articles))))))
 518
 519 (defun nnimap-process-expiry-targets (articles group server)
 520   (let ((deleted-articles nil))
 521     (dolist (article articles)
 522       (let ((target nnmail-expiry-target))
 523         (with-temp-buffer
 524           (when (nnimap-request-article article group server (current-buffer))
 525             (message "Expiring article %s:%d" group article)
 526             (when (functionp target)
 527               (setq target (funcall target group)))
 528             (when (and target
 529                        (not (eq target 'delete)))
 530               (if (or (gnus-request-group target t)
 531                       (gnus-request-create-group target))
 532                   (nnmail-expiry-target-group target group)
 533                 (setq target nil)))
 534             (when target
 535               (push article deleted-articles))))))
 536     ;; Change back to the current group again.
 537     (nnimap-possibly-change-group group server)
 538     (setq deleted-articles (nreverse deleted-articles))
 539     (nnimap-delete-article deleted-articles)
 540     deleted-articles))
 541
 542 (defun nnimap-find-expired-articles (group)
 543   (let ((cutoff (nnmail-expired-article-p group nil nil)))
 544     (with-current-buffer (nnimap-buffer)
 545       (let ((result
 546              (nnimap-command
 547               "UID SEARCH SENTBEFORE %s"
 548               (format-time-string
 549                (format "%%d-%s-%%Y"
 550                        (upcase
 551                         (car (rassoc (nth 4 (decode-time cutoff))
 552                                      parse-time-months))))
 553                cutoff))))
 554         (and (car result)
 555              (delete 0 (mapcar #'string-to-number
 556                                (cdr (assoc "SEARCH" (cdr result))))))))))
 557
 558
 559 (defun nnimap-find-article-by-message-id (group message-id)
 560   (when (nnimap-possibly-change-group group nil)
 561     (with-current-buffer (nnimap-buffer)
 562       (let ((result
 563              (nnimap-command "UID SEARCH HEADER Message-Id %S" message-id))
 564             article)
 565         (when (car result)
 566           ;; Select the last instance of the message in the group.
 567           (and (setq article
 568                      (car (last (assoc "SEARCH" (cdr result)))))
 569                (string-to-number article)))))))
 570
 571 (defun nnimap-delete-article (articles)
 572   (with-current-buffer (nnimap-buffer)
 573     (nnimap-command "UID STORE %s +FLAGS.SILENT (\\Deleted)"
 574                     (nnimap-article-ranges articles))
 575     (cond
 576      ((member "UIDPLUS" (nnimap-capabilities nnimap-object))
 577       (nnimap-command "UID EXPUNGE %s"
 578                       (nnimap-article-ranges articles))
 579       t)
 580      (nnimap-expunge
 581       (nnimap-command "EXPUNGE")
 582       t)
 583      (t (gnus-message 7 (concat "nnimap: nnimap-expunge is not set and the "
 584                                 "server doesn't support UIDPLUS, so we won't "
 585                                 "delete this article now"))))))
 586
 587 (deffoo nnimap-request-scan (&optional group server)
 588   (when (and (nnimap-possibly-change-group nil server)
 589              (equal group nnimap-inbox)
 590              nnimap-inbox
 591              nnimap-split-methods)
 592     (nnimap-split-incoming-mail)))
 593
 594 (defun nnimap-marks-to-flags (marks)
 595   (let (flags flag)
 596     (dolist (mark marks)
 597       (when (setq flag (cadr (assq mark nnimap-mark-alist)))
 598         (push flag flags)))
 599     flags))
 600
 601 (deffoo nnimap-request-set-mark (group actions &optional server)
 602   (when (nnimap-possibly-change-group group server)
 603     (let (sequence)
 604       (with-current-buffer (nnimap-buffer)
 605         ;; Just send all the STORE commands without waiting for
 606         ;; response.  If they're successful, they're successful.
 607         (dolist (action actions)
 608           (destructuring-bind (range action marks) action
 609             (let ((flags (nnimap-marks-to-flags marks)))
 610               (when flags
 611                 (setq sequence (nnimap-send-command
 612                                 "UID STORE %s %sFLAGS.SILENT (%s)"
 613                                 (nnimap-article-ranges range)
 614                                 (if (eq action 'del)
 615                                     "-"
 616                                   "+")
 617                                 (mapconcat #'identity flags " ")))))))
 618         ;; Wait for the last command to complete to avoid later
 619         ;; syncronisation problems with the stream.
 620         (when sequence
 621           (nnimap-wait-for-response sequence))))))
 622
 623 (deffoo nnimap-request-accept-article (group &optional server last)
 624   (when (nnimap-possibly-change-group nil server)
 625     (nnmail-check-syntax)
 626     (let ((message (buffer-string))
 627           (message-id (message-field-value "message-id"))
 628           sequence)
 629       (with-current-buffer (nnimap-buffer)
 630         (setq sequence (nnimap-send-command
 631                         "APPEND %S {%d}" (utf7-encode group t)
 632                         (length message)))
 633         (process-send-string (get-buffer-process (current-buffer)) message)
 634         (process-send-string (get-buffer-process (current-buffer))
 635                              (if (nnimap-newlinep nnimap-object)
 636                                  "\n"
 637                                "\r\n"))
 638         (let ((result (nnimap-get-response sequence)))
 639           (when result
 640             (cons group
 641                   (nnimap-find-article-by-message-id group message-id))))))))
 642
 643 (defun nnimap-add-cr ()
 644   (goto-char (point-min))
 645   (while (re-search-forward "\r?\n" nil t)
 646     (replace-match "\r\n" t t)))
 647
 648 (defun nnimap-get-groups ()
 649   (let ((result (nnimap-command "LIST \"\" \"*\""))
 650         groups)
 651     (when (car result)
 652       (dolist (line (cdr result))
 653         (when (and (equal (car line) "LIST")
 654                    (not (and (caadr line)
 655                              (string-match "noselect" (caadr line)))))
 656           (push (car (last line)) groups)))
 657       (nreverse groups))))
 658
 659 (deffoo nnimap-request-list (&optional server)
 660   (nnimap-possibly-change-group nil server)
 661   (with-current-buffer nntp-server-buffer
 662     (erase-buffer)
 663     (let ((groups
 664            (with-current-buffer (nnimap-buffer)
 665              (nnimap-get-groups)))
 666           sequences responses)
 667       (when groups
 668         (setf (nnimap-group nnimap-object) nil)
 669         (with-current-buffer (nnimap-buffer)
 670           (dolist (group groups)
 671             (push (list (nnimap-send-command "EXAMINE %S" (utf7-encode group t))
 672                         group)
 673                   sequences))
 674           (nnimap-wait-for-response (caar sequences))
 675           (setq responses
 676                 (nnimap-get-responses (mapcar #'car sequences))))
 677         (dolist (response responses)
 678           (let* ((sequence (car response))
 679                  (response (cadr response))
 680                  (group (cadr (assoc sequence sequences))))
 681             (when (and group
 682                        (equal (caar response) "OK"))
 683               (let ((uidnext (nnimap-find-parameter "UIDNEXT" response))
 684                     highest exists)
 685                 (dolist (elem response)
 686                   (when (equal (cadr elem) "EXISTS")
 687                     (setq exists (string-to-number (car elem)))))
 688                 (when uidnext
 689                   (setq highest (1- (string-to-number (car uidnext)))))
 690                 (cond
 691                  ((null highest)
 692                   (insert (format "%S 0 1 y\n" (utf7-decode group t))))
 693                  ((zerop exists)
 694                   ;; Empty group.
 695                   (insert (format "%S %d %d y\n"
 696                                   (utf7-decode group t) highest (1+ highest))))
 697                  (t
 698                   ;; Return the widest possible range.
 699                   (insert (format "%S %d 1 y\n" (utf7-decode group t)
 700                                   (or highest exists)))))))))
 701         t))))
 702
 703 (deffoo nnimap-retrieve-group-data-early (server infos)
 704   (when (nnimap-possibly-change-group nil server)
 705     (with-current-buffer (nnimap-buffer)
 706       ;; QRESYNC handling isn't implemented.
 707       (let ((qresyncp (member "notQRESYNC" (nnimap-capabilities nnimap-object)))
 708             marks groups sequences)
 709         ;; Go through the infos and gather the data needed to know
 710         ;; what and how to request the data.
 711         (dolist (info infos)
 712           (setq marks (gnus-info-marks info))
 713           (push (list (gnus-group-real-name (gnus-info-group info))
 714                       (cdr (assq 'active marks))
 715                       (cdr (assq 'uid marks)))
 716                 groups))
 717         ;; Then request the data.
 718         (erase-buffer)
 719         (setf (nnimap-group nnimap-object) nil)
 720         (dolist (elem groups)
 721           (if (and qresyncp
 722                    (nth 2 elem))
 723               (push
 724                (list 'qresync
 725                      (nnimap-send-command "EXAMINE %S (QRESYNC (%s %s))"
 726                                           (car elem)
 727                                           (car (nth 2 elem))
 728                                           (cdr (nth 2 elem)))
 729                      nil
 730                      (car elem))
 731                sequences)
 732             (let ((start
 733                    (if (nth 1 elem)
 734                        ;; Fetch the last 100 flags.
 735                        (max 1 (- (cdr (nth 1 elem)) 100))
 736                      1)))
 737               (push (list (nnimap-send-command "EXAMINE %S" (car elem))
 738                           (nnimap-send-command "UID FETCH %d:* FLAGS" start)
 739                           start
 740                           (car elem))
 741                     sequences))))
 742         sequences))))
 743
 744 (deffoo nnimap-finish-retrieve-group-infos (server infos sequences)
 745   (when (and sequences
 746              (nnimap-possibly-change-group nil server))
 747     (with-current-buffer (nnimap-buffer)
 748       ;; Wait for the final data to trickle in.
 749       (nnimap-wait-for-response (cadar sequences))
 750       ;; Now we should have all the data we need, no matter whether
 751       ;; we're QRESYNCING, fetching all the flags from scratch, or
 752       ;; just fetching the last 100 flags per group.
 753       (nnimap-update-infos (nnimap-flags-to-marks
 754                             (nnimap-parse-flags
 755                              (nreverse sequences)))
 756                            infos)
 757       ;; Finally, just return something resembling an active file in
 758       ;; the nntp buffer, so that the agent can save the info, too.
 759       (with-current-buffer nntp-server-buffer
 760         (erase-buffer)
 761         (dolist (info infos)
 762           (let* ((group (gnus-info-group info))
 763                  (active (gnus-active group)))
 764             (when active
 765               (insert (format "%S %d %d y\n"
 766                               (gnus-group-real-name group)
 767                               (cdr active)
 768                               (car active))))))))))
 769
 770 (defun nnimap-update-infos (flags infos)
 771   (dolist (info infos)
 772     (let ((group (gnus-group-real-name (gnus-info-group info))))
 773       (nnimap-update-info info (cdr (assoc group flags))))))
 774
 775 (defun nnimap-update-info (info marks)
 776   (when marks
 777     (destructuring-bind (existing flags high low uidnext start-article) marks
 778       (let ((group (gnus-info-group info))
 779             (completep (and start-article
 780                             (= start-article 1))))
 781         ;; First set the active ranges based on high/low.
 782         (if (or completep
 783                 (not (gnus-active group)))
 784             (gnus-set-active group
 785                              (if high
 786                                  (cons low high)
 787                                ;; No articles in this group.
 788                                (cons (1- uidnext) uidnext)))
 789           (setcdr (gnus-active group) high))
 790         ;; Then update the list of read articles.
 791         (let* ((unread
 792                 (gnus-compress-sequence
 793                  (gnus-set-difference
 794                   (gnus-set-difference
 795                    existing
 796                    (cdr (assoc "\\Seen" flags)))
 797                   (cdr (assoc "\\Flagged" flags)))))
 798                (read (gnus-range-difference
 799                       (cons start-article high) unread)))
 800           (when (> start-article 1)
 801             (setq read
 802                   (gnus-range-nconcat
 803                    (if (> start-article 1)
 804                        (gnus-sorted-range-intersection
 805                         (cons 1 (1- start-article))
 806                         (gnus-info-read info))
 807                      (gnus-info-read info))
 808                    read)))
 809           (gnus-info-set-read info read)
 810           ;; Update the marks.
 811           (setq marks (gnus-info-marks info))
 812           ;; Note the active level for the next run-through.
 813           (let ((active (assq 'active marks)))
 814             (if active
 815                 (setcdr active (gnus-active group))
 816               (push (cons 'active (gnus-active group)) marks)))
 817           (dolist (type (cdr nnimap-mark-alist))
 818             (let ((old-marks (assoc (car type) marks))
 819                   (new-marks (gnus-compress-sequence
 820                               (cdr (assoc (cadr type) flags)))))
 821               (setq marks (delq old-marks marks))
 822               (pop old-marks)
 823               (when (and old-marks
 824                          (> start-article 1))
 825                 (setq old-marks (gnus-range-difference
 826                                  old-marks
 827                                  (cons start-article high)))
 828                 (setq new-marks (gnus-range-nconcat old-marks new-marks)))
 829               (when new-marks
 830                 (push (cons (car type) new-marks) marks)))
 831             (gnus-info-set-marks info marks t)
 832             (nnimap-store-info info (gnus-active group))))))))
 833
 834 (defun nnimap-store-info (info active)
 835   (let* ((group (gnus-group-real-name (gnus-info-group info)))
 836          (entry (assoc group nnimap-current-infos)))
 837     (if entry
 838         (setcdr entry (list info active))
 839       (push (list group info active) nnimap-current-infos))))
 840
 841 (defun nnimap-flags-to-marks (groups)
 842   (let (data group totalp uidnext articles start-article mark)
 843     (dolist (elem groups)
 844       (setq group (car elem)
 845             uidnext (cadr elem)
 846             start-article (caddr elem)
 847             articles (cdddr elem))
 848       (let ((high (caar articles))
 849             marks low existing)
 850         (dolist (article articles)
 851           (setq low (car article))
 852           (push (car article) existing)
 853           (dolist (flag (cdr article))
 854             (setq mark (assoc flag marks))
 855             (if (not mark)
 856                 (push (list flag (car article)) marks)
 857               (setcdr mark (cons (car article) (cdr mark)))))
 858           (push (list group existing marks high low uidnext start-article)
 859                 data))))
 860     data))
 861
 862 (defun nnimap-parse-flags (sequences)
 863   (goto-char (point-min))
 864   (let (start end articles groups uidnext elems)
 865     (dolist (elem sequences)
 866       (destructuring-bind (group-sequence flag-sequence totalp group) elem
 867         ;; The EXAMINE was successful.
 868         (when (and (search-forward (format "\n%d OK " group-sequence) nil t)
 869                    (progn
 870                      (forward-line 1)
 871                      (setq start (point))
 872                      (if (re-search-backward "UIDNEXT \\([0-9]+\\)"
 873                                                (or end (point-min)) t)
 874                          (setq uidnext (string-to-number (match-string 1)))
 875                        (setq uidnext nil))
 876                      (goto-char start))
 877                    ;; The UID FETCH FLAGS was successful.
 878                    (search-forward (format "\n%d OK " flag-sequence) nil t))
 879           (setq end (point))
 880           (goto-char start)
 881           (while (re-search-forward "^\\* [0-9]+ FETCH (\\(.*\\))" end t)
 882             (setq elems (nnimap-parse-line (match-string 1)))
 883             (push (cons (string-to-number (cadr (member "UID" elems)))
 884                         (cadr (member "FLAGS" elems)))
 885                   articles))
 886           (push (nconc (list group uidnext totalp) articles) groups)
 887           (setq articles nil))))
 888     groups))
 889
 890 (defun nnimap-find-process-buffer (buffer)
 891   (cadr (assoc buffer nnimap-connection-alist)))
 892
 893 (deffoo nnimap-request-post (&optional server)
 894   (setq nnimap-status-string "Read-only server")
 895   nil)
 896
 897 (defun nnimap-possibly-change-group (group server)
 898   (let ((open-result t))
 899     (when (and server
 900                (not (nnimap-server-opened server)))
 901       (setq open-result (nnimap-open-server server)))
 902     (cond
 903      ((not open-result)
 904       nil)
 905      ((not group)
 906       t)
 907      (t
 908       (with-current-buffer (nnimap-buffer)
 909         (if (equal group (nnimap-group nnimap-object))
 910             t
 911           (let ((result (nnimap-command "SELECT %S" (utf7-encode group t))))
 912             (when (car result)
 913               (setf (nnimap-group nnimap-object) group
 914                     (nnimap-select-result nnimap-object) result)
 915               result))))))))
 916
 917 (defun nnimap-find-connection (buffer)
 918   "Find the connection delivering to BUFFER."
 919   (let ((entry (assoc buffer nnimap-connection-alist)))
 920     (when entry
 921       (if (and (buffer-name (cadr entry))
 922                (get-buffer-process (cadr entry))
 923                (memq (process-status (get-buffer-process (cadr entry)))
 924                      '(open run)))
 925           (get-buffer-process (cadr entry))
 926         (setq nnimap-connection-alist (delq entry nnimap-connection-alist))
 927         nil))))
 928
 929 (defvar nnimap-sequence 0)
 930
 931 (defun nnimap-send-command (&rest args)
 932   (process-send-string
 933    (get-buffer-process (current-buffer))
 934    (nnimap-log-command
 935     (format "%d %s%s\n"
 936             (incf nnimap-sequence)
 937             (apply #'format args)
 938             (if (nnimap-newlinep nnimap-object)
 939                 ""
 940               "\r"))))
 941   nnimap-sequence)
 942
 943 (defun nnimap-log-command (command)
 944   (with-current-buffer (get-buffer-create "*imap log*")
 945     (goto-char (point-max))
 946     (insert (format-time-string "%H:%M:%S") " " command))
 947   command)
 948
 949 (defun nnimap-command (&rest args)
 950   (erase-buffer)
 951   (let* ((sequence (apply #'nnimap-send-command args))
 952          (response (nnimap-get-response sequence)))
 953     (if (equal (caar response) "OK")
 954         (cons t response)
 955       (nnheader-report 'nnimap "%s"
 956                        (mapconcat (lambda (a)
 957                                     (format "%s" a))
 958                                   (car response) " "))
 959       nil)))
 960
 961 (defun nnimap-get-response (sequence)
 962   (nnimap-wait-for-response sequence)
 963   (nnimap-parse-response))
 964
 965 (defun nnimap-wait-for-connection ()
 966   (let ((process (get-buffer-process (current-buffer))))
 967     (goto-char (point-min))
 968     (while (and (memq (process-status process)
 969                       '(open run))
 970                 (not (re-search-forward "^\\* .*\n" nil t)))
 971       (nnheader-accept-process-output process)
 972       (goto-char (point-min)))
 973     (forward-line -1)
 974     (and (looking-at "\\* \\([A-Z0-9]+\\)")
 975          (match-string 1))))
 976
 977 (defun nnimap-wait-for-response (sequence &optional messagep)
 978   (let ((process (get-buffer-process (current-buffer))))
 979     (goto-char (point-max))
 980     (while (and (memq (process-status process)
 981                       '(open run))
 982                 (not (re-search-backward (format "^%d .*\n" sequence)
 983                                          (max (point-min) (- (point) 500))
 984                                          t)))
 985       (when messagep
 986         (message "Read %dKB" (/ (buffer-size) 1000)))
 987       (nnheader-accept-process-output process)
 988       (goto-char (point-max)))))
 989
 990 (defun nnimap-parse-response ()
 991   (let ((lines (split-string (nnimap-last-response-string) "\r\n" t))
 992         result)
 993     (dolist (line lines)
 994       (push (cdr (nnimap-parse-line line)) result))
 995     ;; Return the OK/error code first, and then all the "continuation
 996     ;; lines" afterwards.
 997     (cons (pop result)
 998           (nreverse result))))
 999
1000 ;; Parse an IMAP response line lightly.  They look like
1001 ;; "* OK [UIDVALIDITY 1164213559] UIDs valid", typically, so parse
1002 ;; the lines into a list of strings and lists of string.
1003 (defun nnimap-parse-line (line)
1004   (let (char result)
1005     (with-temp-buffer
1006       (insert line)
1007       (goto-char (point-min))
1008       (while (not (eobp))
1009         (if (eql (setq char (following-char)) ? )
1010             (forward-char 1)
1011           (push
1012            (cond
1013             ((eql char ?\[)
1014              (split-string (buffer-substring
1015                             (1+ (point)) (1- (search-forward "]")))))
1016             ((eql char ?\()
1017              (split-string (buffer-substring
1018                             (1+ (point)) (1- (search-forward ")")))))
1019             ((eql char ?\")
1020              (forward-char 1)
1021              (buffer-substring (point) (1- (search-forward "\""))))
1022             (t
1023              (buffer-substring (point) (if (search-forward " " nil t)
1024                                            (1- (point))
1025                                          (goto-char (point-max))))))
1026            result)))
1027       (nreverse result))))
1028
1029 (defun nnimap-last-response-string ()
1030   (save-excursion
1031     (forward-line 1)
1032     (let ((end (point)))
1033       (forward-line -1)
1034       (when (not (bobp))
1035         (forward-line -1)
1036         (while (and (not (bobp))
1037                     (eql (following-char) ?*))
1038           (forward-line -1))
1039         (unless (eql (following-char) ?*)
1040           (forward-line 1)))
1041       (buffer-substring (point) end))))
1042
1043 (defun nnimap-get-responses (sequences)
1044   (let (responses)
1045     (dolist (sequence sequences)
1046       (goto-char (point-min))
1047       (when (re-search-forward (format "^%d " sequence) nil t)
1048         (push (list sequence (nnimap-parse-response))
1049               responses)))
1050     responses))
1051
1052 (defvar nnimap-incoming-split-list nil)
1053
1054 (defun nnimap-fetch-inbox (articles)
1055   (erase-buffer)
1056   (nnimap-wait-for-response
1057    (nnimap-send-command
1058     "UID FETCH %s %s"
1059     (nnimap-article-ranges articles)
1060     (format "(UID %s%s)"
1061             (format
1062              (if (member "IMAP4REV1"
1063                          (nnimap-capabilities nnimap-object))
1064                  "BODY.PEEK[HEADER] BODY.PEEK"
1065                "RFC822.PEEK"))
1066             (if nnimap-split-download-body-default
1067                 "[]"
1068               "[1]")))
1069    t))
1070
1071 (defun nnimap-split-incoming-mail ()
1072   (with-current-buffer (nnimap-buffer)
1073     (let ((nnimap-incoming-split-list nil)
1074           (nnmail-split-methods nnimap-split-methods)
1075           (nnmail-inhibit-default-split-group t)
1076           (groups (nnimap-get-groups))
1077           new-articles)
1078       (erase-buffer)
1079       (nnimap-command "SELECT %S" nnimap-inbox)
1080       (setq new-articles (nnimap-new-articles (nnimap-get-flags "1:*")))
1081       (when new-articles
1082         (nnimap-fetch-inbox new-articles)
1083         (nnimap-transform-split-mail)
1084         (nnheader-ms-strip-cr)
1085         (nnmail-cache-open)
1086         (nnmail-split-incoming (current-buffer)
1087                                #'nnimap-save-mail-spec
1088                                nil nil
1089                                #'nnimap-dummy-active-number)
1090         (when nnimap-incoming-split-list
1091           (let ((specs (nnimap-make-split-specs nnimap-incoming-split-list))
1092                 sequences)
1093             ;; Create any groups that doesn't already exist on the
1094             ;; server first.
1095             (dolist (spec specs)
1096               (unless (member (car spec) groups)
1097                 (nnimap-command "CREATE %S" (utf7-encode (car spec) t))))
1098             ;; Then copy over all the messages.
1099             (erase-buffer)
1100             (dolist (spec specs)
1101               (let ((group (car spec))
1102                     (ranges (cdr spec)))
1103                 (push (list (nnimap-send-command "UID COPY %s %S"
1104                                                  (nnimap-article-ranges ranges)
1105                                                  (utf7-encode group t))
1106                             ranges)
1107                       sequences)))
1108             ;; Wait for the last COPY response...
1109             (when sequences
1110               (nnimap-wait-for-response (caar sequences))
1111               ;; And then mark the successful copy actions as deleted,
1112               ;; and possibly expunge them.
1113               (nnimap-mark-and-expunge-incoming
1114                (nnimap-parse-copied-articles sequences)))))))))
1115
1116 (defun nnimap-mark-and-expunge-incoming (range)
1117   (when range
1118     (setq range (nnimap-article-ranges range))
1119     (let ((sequence
1120            (nnimap-send-command
1121             "UID STORE %s +FLAGS.SILENT (\\Deleted)" range)))
1122       (cond
1123        ;; If the server supports it, we now delete the message we have
1124        ;; just copied over.
1125        ((member "UIDPLUS" (nnimap-capabilities nnimap-object))
1126         (setq sequence (nnimap-send-command "UID EXPUNGE %s" range)))
1127        ;; If it doesn't support UID EXPUNGE, then we only expunge if the
1128        ;; user has configured it.
1129        (nnimap-expunge
1130         (setq sequence (nnimap-send-command "EXPUNGE"))))
1131       (nnimap-wait-for-response sequence))))
1132
1133 (defun nnimap-parse-copied-articles (sequences)
1134   (let (sequence copied range)
1135     (goto-char (point-min))
1136     (while (re-search-forward "^\\([0-9]+\\) OK " nil t)
1137       (setq sequence (string-to-number (match-string 1)))
1138       (when (setq range (cadr (assq sequence sequences)))
1139         (push (gnus-uncompress-range range) copied)))
1140     (gnus-compress-sequence (sort (apply #'nconc copied) #'<))))
1141
1142 (defun nnimap-new-articles (flags)
1143   (let (new)
1144     (dolist (elem flags)
1145       (when (or (null (cdr elem))
1146                 (and (not (member "\\Deleted" (cdr elem)))
1147                      (not (member "\\Seen" (cdr elem)))))
1148         (push (car elem) new)))
1149     (gnus-compress-sequence (nreverse new))))
1150
1151 (defun nnimap-make-split-specs (list)
1152   (let ((specs nil)
1153         entry)
1154     (dolist (elem list)
1155       (destructuring-bind (article spec) elem
1156         (dolist (group (delete nil (mapcar #'car spec)))
1157           (unless (setq entry (assoc group specs))
1158             (push (setq entry (list group)) specs))
1159           (setcdr entry (cons article (cdr entry))))))
1160     (dolist (entry specs)
1161       (setcdr entry (gnus-compress-sequence (sort (cdr entry) #'<))))
1162     specs))
1163
1164 (defun nnimap-transform-split-mail ()
1165   (goto-char (point-min))
1166   (let (article bytes)
1167     (block nil
1168       (while (not (eobp))
1169         (while (not (looking-at "^\\* [0-9]+ FETCH.*UID \\([0-9]+\\)"))
1170           (delete-region (point) (progn (forward-line 1) (point)))
1171           (when (eobp)
1172             (return)))
1173         (setq article (match-string 1)
1174               bytes (nnimap-get-length))
1175         (delete-region (line-beginning-position) (line-end-position))
1176         ;; Insert MMDF separator, and a way to remember what this
1177         ;; article UID is.
1178         (insert (format "\^A\^A\^A\^A\n\nX-nnimap-article: %s" article))
1179         (forward-char (1+ bytes))
1180         (setq bytes (nnimap-get-length))
1181         (delete-region (line-beginning-position) (line-end-position))
1182         (forward-char (1+ bytes))
1183         (delete-region (line-beginning-position) (line-end-position))))))
1184
1185 (defun nnimap-dummy-active-number (group &optional server)
1186   1)
1187
1188 (defun nnimap-save-mail-spec (group-art &optional server full-nov)
1189   (let (article)
1190     (goto-char (point-min))
1191     (if (not (re-search-forward "X-nnimap-article: \\([0-9]+\\)" nil t))
1192         (error "Invalid nnimap mail")
1193       (setq article (string-to-number (match-string 1))))
1194     (push (list article group-art)
1195           nnimap-incoming-split-list)))
1196
1197 (provide 'nnimap)
1198
1199 ;;; nnimap.el ends here