* modb-standard.el (modb-standard-economize-entity-size): Changed
[elisp/wanderlust.git] / elmo / modb-standard.el
1 ;;; modb-standard.el --- Standartd Implement of MODB.
2
3 ;; Copyright (C) 2003 Yuuichi Teranishi <teranisi@gohome.org>
4
5 ;; Author: Yuuichi Teranishi <teranisi@gohome.org>
6 ;;      Hiroya Murata <lapis-lazuli@pop06.odn.ne.jp>
7 ;; Keywords: mail, net news
8
9 ;; This file is part of ELMO (Elisp Library for Message Orchestration).
10
11 ;; This program is free software; you can redistribute it and/or modify
12 ;; it under the terms of the GNU General Public License as published by
13 ;; the Free Software Foundation; either version 2, or (at your option)
14 ;; any later version.
15 ;;
16 ;; This program is distributed in the hope that it will be useful,
17 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
18 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19 ;; GNU General Public License for more details.
20 ;;
21 ;; You should have received a copy of the GNU General Public License
22 ;; along with GNU Emacs; see the file COPYING.  If not, write to the
23 ;; Free Software Foundation, Inc., 59 Temple Place - Suite 330,
24 ;; Boston, MA 02111-1307, USA.
25 ;;
26
27 ;;; Commentary:
28 ;;
29
30 ;;; Code:
31 ;;
32 (eval-when-compile (require 'cl))
33
34 (require 'elmo-util)
35 (require 'modb)
36
37 (defcustom modb-standard-divide-number 500
38   "*Standard modb divide entity number."
39   :type '(choice (const :tag "Not divide" nil)
40                  number)
41   :group 'elmo)
42
43 (defcustom modb-standard-economize-entity-size t
44   "*Economize message entity size.
45 When non-nil, redundunt message-id string are not saved."
46   :type 'boolean
47   :group 'elmo)
48
49 (defvar modb-standard-entity-filename "entity"
50   "Message entity database.")
51
52 (defvar modb-standard-flag-filename "flag"
53   "Message number <=> Flag status database.")
54
55 (defvar modb-standard-msgid-filename "msgid"
56   "Message number <=> Message-Id database.")
57
58 (eval-and-compile
59   (luna-define-class modb-standard (modb-generic)
60                      (number-list       ; sorted list of message numbers.
61                       entity-map        ; number, msg-id -> entity mapping.
62                       flag-map          ; number -> flag-list mapping
63                       flag-count        ; list of (FLAG . COUNT)
64                       ))
65   (luna-define-internal-accessors 'modb-standard))
66
67 ;; for internal use only
68 (defsubst modb-standard-key (number)
69   (concat "#" (number-to-string number)))
70
71 (defsubst modb-standard-entity-id (entity)
72   (if (eq 'autoload (car-safe entity))
73       (cddr entity)
74     (elmo-msgdb-message-entity-field
75      (elmo-message-entity-handler entity)
76      entity 'message-id)))
77
78 (defsubst modb-standard-entity-map (modb)
79   (or (modb-standard-entity-map-internal modb)
80       (modb-standard-set-entity-map-internal
81        modb
82        (elmo-make-hash (elmo-msgdb-length modb)))))
83
84 (defsubst modb-standard-flag-map (modb)
85   (or (modb-standard-flag-map-internal modb)
86       (modb-standard-set-flag-map-internal
87        modb
88        (elmo-make-hash (elmo-msgdb-length modb)))))
89
90 (defsubst modb-standard-set-message-modified (modb number)
91   (if modb-standard-divide-number
92       (let ((section (/ number modb-standard-divide-number))
93             (modified (modb-generic-message-modified-internal modb)))
94         (unless (memq section modified)
95           (modb-generic-set-message-modified-internal
96            modb (cons section modified))))
97     (modb-generic-set-message-modified-internal modb t)))
98
99 (defsubst modb-standard-set-flag-modified (modb number)
100   (modb-generic-set-flag-modified-internal modb t))
101
102 (defsubst modb-standard-message-flags (modb number)
103   (cdr (elmo-get-hash-val (modb-standard-key number)
104                           (modb-standard-flag-map-internal modb))))
105
106 (defsubst modb-standard-match-flags (check-flags flags)
107   (catch 'done
108     (while check-flags
109       (when (memq (car check-flags) flags)
110         (throw 'done t))
111       (setq check-flags (cdr check-flags)))))
112
113 (defsubst modb-standard-countup-flags (modb flags &optional delta)
114   (let ((flag-count (modb-standard-flag-count-internal modb))
115         (delta (or delta 1))
116         elem)
117     (dolist (flag flags)
118       (if (setq elem (assq flag flag-count))
119           (setcdr elem (+ (cdr elem) delta))
120         (setq flag-count (cons (cons flag delta) flag-count))))
121     (modb-standard-set-flag-count-internal modb flag-count)))
122
123 ;; save and load functions
124 (defun modb-standard-load-msgid (modb path)
125   (let* ((alist (elmo-object-load
126                  (expand-file-name modb-standard-msgid-filename path)))
127          (table (or (modb-standard-entity-map-internal modb)
128                     (elmo-make-hash (length alist))))
129          numbers info)
130     (dolist (pair alist)
131       (setq info (cons 'autoload pair))
132       (elmo-set-hash-val (modb-standard-key (car pair)) info table)
133       (elmo-set-hash-val (cdr pair) info table)
134       (setq numbers (cons (car pair) numbers)))
135     (modb-standard-set-number-list-internal modb (nreverse numbers))
136     (modb-standard-set-entity-map-internal modb table)))
137
138 (defun modb-standard-save-msgid (modb path)
139   (let ((table (modb-standard-entity-map-internal modb))
140         entity alist)
141     (dolist (number (modb-standard-number-list-internal modb))
142       (setq entity (elmo-get-hash-val (modb-standard-key number) table))
143       (setq alist (cons (cons number (modb-standard-entity-id entity))
144                         alist)))
145     (elmo-object-save
146      (expand-file-name modb-standard-msgid-filename path)
147      (nreverse alist))))
148
149 (defun modb-standard-load-flag (modb path)
150   (let ((table (or (modb-standard-flag-map-internal modb)
151                    (elmo-make-hash (elmo-msgdb-length modb)))))
152     (dolist (info (elmo-object-load
153                    (expand-file-name modb-standard-flag-filename path)))
154       (modb-standard-countup-flags modb (cdr info))
155       (elmo-set-hash-val (modb-standard-key (car info)) info table))
156     (modb-standard-set-flag-map-internal modb table)))
157
158 (defun modb-standard-save-flag (modb path)
159   (let (table flist info)
160     (when (setq table (modb-standard-flag-map-internal modb))
161       (mapatoms
162        (lambda (atom)
163          (setq info (symbol-value atom))
164          (when (cdr info)
165            (setq flist (cons info flist))))
166        table))
167     (elmo-object-save
168      (expand-file-name modb-standard-flag-filename path)
169      flist)))
170
171 (defsubst modb-standard-entity-filename (section)
172   (if section
173       (concat modb-standard-entity-filename
174               "-"
175               (number-to-string section))
176     modb-standard-entity-filename))
177
178 (defsubst modb-standard-loaded-message-id (msgdb number)
179   "Get message-id for autoloaded entity."
180   (let ((ret (elmo-get-hash-val
181               (modb-standard-key number)
182               (modb-standard-entity-map-internal msgdb))))
183     (cond
184      ((and ret (eq (car-safe ret) 'autoload))
185       (cdr (cdr ret))) ; message-id.
186      ((and ret (stringp (car-safe ret)))
187       ;; Already loaded.
188       (car ret))
189      ((null ret)
190       ;; Garbage entity.
191       (elmo-clear-hash-val (modb-standard-key number)
192                            (modb-standard-entity-map-internal msgdb))
193       nil)                              ; return nil.
194      (t (error "Internal error: invalid msgdb status")))))
195
196 (defun modb-standard-load-entity (modb path &optional section)
197   (let ((table (or (modb-standard-entity-map-internal modb)
198                    (elmo-make-hash (elmo-msgdb-length modb))))
199         number msgid)
200     (dolist (entity (elmo-object-load
201                      (expand-file-name
202                       (modb-standard-entity-filename section)
203                       path)))
204       (setq number (elmo-msgdb-message-entity-number
205                     (elmo-message-entity-handler entity)
206                     entity)
207             msgid (modb-standard-loaded-message-id modb number))
208       (when msgid
209         (setcar entity msgid)
210         (elmo-set-hash-val msgid entity table)
211         (elmo-set-hash-val (modb-standard-key number) entity table)))
212     (modb-standard-set-entity-map-internal modb table)))
213
214 (defsubst modb-standard-save-entity-1 (modb path &optional section)
215   (let ((table (modb-standard-entity-map-internal modb))
216         (filename (expand-file-name
217                    (modb-standard-entity-filename section) path))
218         entity entities)
219     (dolist (number (modb-standard-number-list-internal modb))
220       (when (and (or (null section)
221                      (= section (/ number modb-standard-divide-number)))
222                  (setq entity (elmo-msgdb-message-entity modb number)))
223         (when modb-standard-economize-entity-size
224           (when (stringp (car entity))
225             (setq entity (cons t (cdr entity)))))
226         (setq entities (cons entity entities))))
227     (if entities
228         (elmo-object-save filename entities)
229       (ignore-errors (delete-file filename)))))
230
231 (defun modb-standard-save-entity (modb path)
232   (let ((sections (modb-generic-message-modified-internal modb)))
233     (cond ((listp sections)
234            (dolist (section sections)
235              (modb-standard-save-entity-1 modb path section)))
236           (sections
237            (modb-standard-save-entity-1 modb path)))))
238
239 ;;; Implement
240 ;;
241 (luna-define-method elmo-msgdb-load ((msgdb modb-standard))
242   (let ((inhibit-quit t)
243         (path (elmo-msgdb-location msgdb)))
244     (when (file-exists-p (expand-file-name modb-standard-flag-filename path))
245       (modb-standard-load-msgid msgdb path)
246       (modb-standard-load-flag msgdb path)
247       (unless modb-standard-divide-number
248         (modb-standard-load-entity msgdb path))
249       t)))
250
251 (luna-define-method elmo-msgdb-save ((msgdb modb-standard))
252   (let ((path (elmo-msgdb-location msgdb))
253         (inhibit-quit t))
254     (when (elmo-msgdb-message-modified-p msgdb)
255       (modb-standard-save-msgid  msgdb path)
256       (modb-standard-save-entity msgdb path)
257       (modb-generic-set-message-modified-internal msgdb nil))
258     (when (elmo-msgdb-flag-modified-p msgdb)
259       (modb-standard-save-flag msgdb path)
260       (modb-generic-set-flag-modified-internal msgdb nil))))
261
262 (luna-define-method elmo-msgdb-append :around ((msgdb modb-standard)
263                                                msgdb-append)
264   (when (> (elmo-msgdb-length msgdb-append) 0)
265     (if (eq (luna-class-name msgdb-append) 'modb-standard)
266         (let ((numbers (modb-standard-number-list-internal msgdb-append))
267               duplicates)
268           ;; number-list
269           (modb-standard-set-number-list-internal
270            msgdb
271            (nconc (modb-standard-number-list-internal msgdb)
272                   numbers))
273           ;; entity-map
274           (let ((table (modb-standard-entity-map msgdb))
275                 entity msg-id)
276             (dolist (number numbers)
277               (setq entity (elmo-msgdb-message-entity msgdb-append number)
278                     msg-id (modb-standard-entity-id entity))
279               (if (elmo-get-hash-val msg-id table)
280                   (setq duplicates (cons number duplicates))
281                 (elmo-set-hash-val msg-id entity table))
282               (elmo-set-hash-val (modb-standard-key number)
283                                  entity
284                                  table)))
285           ;; flag-map
286           (let ((table (modb-standard-flag-map msgdb)))
287             (mapatoms
288              (lambda (atom)
289                (elmo-set-hash-val (symbol-name atom)
290                                   (symbol-value atom)
291                                   table))
292              (modb-standard-flag-map msgdb-append)))
293           ;; flag-count
294           (dolist (pair (modb-standard-flag-count-internal msgdb-append))
295             (modb-standard-countup-flags msgdb (list (car pair)) (cdr pair)))
296           ;; modification flags
297           (dolist (number (modb-standard-number-list-internal msgdb-append))
298             (modb-standard-set-message-modified msgdb number)
299             (modb-standard-set-flag-modified msgdb number))
300           duplicates)
301       (luna-call-next-method))))
302
303 (luna-define-method elmo-msgdb-clear :after ((msgdb modb-standard))
304   (modb-standard-set-number-list-internal msgdb nil)
305   (modb-standard-set-entity-map-internal msgdb nil)
306   (modb-standard-set-flag-map-internal msgdb nil)
307   (modb-standard-set-flag-count-internal msgdb nil))
308
309 (luna-define-method elmo-msgdb-length ((msgdb modb-standard))
310   (length (modb-standard-number-list-internal msgdb)))
311
312 (luna-define-method elmo-msgdb-flag-available-p ((msgdb modb-standard) flag)
313   t)
314
315 (luna-define-method elmo-msgdb-flags ((msgdb modb-standard) number)
316   (modb-standard-message-flags msgdb number))
317
318 (luna-define-method elmo-msgdb-set-flag ((msgdb modb-standard)
319                                          number flag)
320   (case flag
321     (read
322      (elmo-msgdb-unset-flag msgdb number 'unread))
323     (uncached
324      (elmo-msgdb-unset-flag msgdb number 'cached))
325     (t
326      (let ((cur-flags (modb-standard-message-flags msgdb number))
327            new-flags diff)
328        (unless (memq flag cur-flags)
329          (setq new-flags (cons flag cur-flags))
330          (setq diff (elmo-list-diff new-flags cur-flags))
331          (modb-standard-countup-flags msgdb (car diff))
332          (modb-standard-countup-flags msgdb (cadr diff) -1)
333          (elmo-set-hash-val (modb-standard-key number)
334                             (cons number new-flags)
335                             (modb-standard-flag-map msgdb))
336          (modb-standard-set-flag-modified msgdb number))))))
337
338 (luna-define-method elmo-msgdb-unset-flag ((msgdb modb-standard)
339                                            number flag)
340   (case flag
341     (read
342      (elmo-msgdb-set-flag msgdb number 'unread))
343     (uncached
344      (elmo-msgdb-set-flag msgdb number 'cached))
345     (all
346      (modb-standard-countup-flags msgdb
347                                   (modb-standard-message-flags msgdb number)
348                                   -1)
349      (elmo-clear-hash-val (modb-standard-key number)
350                           (modb-standard-flag-map msgdb)))
351     (t
352      (let ((cur-flags (modb-standard-message-flags msgdb number))
353            new-flags diff)
354        (when (memq flag cur-flags)
355          (setq new-flags (delq flag (copy-sequence cur-flags)))
356          (setq diff (elmo-list-diff new-flags cur-flags))
357          (modb-standard-countup-flags msgdb (car diff))
358          (modb-standard-countup-flags msgdb (cadr diff) -1)
359          (elmo-set-hash-val (modb-standard-key number)
360                             (cons number new-flags)
361                             (modb-standard-flag-map msgdb))
362          (modb-standard-set-flag-modified msgdb number))
363        (when (eq flag 'unread)
364          (elmo-msgdb-unset-flag msgdb number 'new))))))
365
366 (luna-define-method elmo-msgdb-flag-count ((msgdb modb-standard))
367   (modb-standard-flag-count-internal msgdb))
368
369 (luna-define-method elmo-msgdb-list-messages ((msgdb modb-standard))
370   (copy-sequence
371    (modb-standard-number-list-internal msgdb)))
372
373 (luna-define-method elmo-msgdb-list-flagged ((msgdb modb-standard) flag)
374   (let (entry matched)
375     (case flag
376       (read
377        (dolist (number (modb-standard-number-list-internal msgdb))
378          (unless (memq 'unread (modb-standard-message-flags msgdb number))
379            (setq matched (cons number matched)))))
380       (uncached
381        (dolist (number (modb-standard-number-list-internal msgdb))
382          (unless (memq 'cached (modb-standard-message-flags msgdb number))
383            (setq matched (cons number matched)))))
384       (any
385        (mapatoms
386         (lambda (atom)
387           (setq entry (symbol-value atom))
388           (unless (and (eq (length (cdr entry)) 1)
389                        (eq (car (cdr entry)) 'cached))
390             ;; If there is a flag other than cached, then the message
391             ;; matches to `any'.
392             (setq matched (cons (car entry) matched))))
393         (modb-standard-flag-map msgdb)))
394       (digest
395        (let ((flags (append elmo-digest-flags
396                             (elmo-get-global-flags t t))))
397          (mapatoms
398           (lambda (atom)
399             (setq entry (symbol-value atom))
400             (when (modb-standard-match-flags flags (cdr entry))
401               (setq matched (cons (car entry) matched))))
402           (modb-standard-flag-map msgdb))))
403       (t
404        (mapatoms
405         (lambda (atom)
406           (setq entry (symbol-value atom))
407           (when (memq flag (cdr entry))
408             (setq matched (cons (car entry) matched))))
409         (modb-standard-flag-map msgdb))))
410     matched))
411
412 (luna-define-method elmo-msgdb-search ((msgdb modb-standard)
413                                        condition &optional numbers)
414   (if (vectorp condition)
415       (let ((key (elmo-filter-key condition))
416             results)
417         (cond
418          ((and (string= key "flag")
419                (eq (elmo-filter-type condition) 'match))
420           (setq results (elmo-msgdb-list-flagged
421                          msgdb
422                          (intern (elmo-filter-value condition))))
423           (if numbers
424               (elmo-list-filter numbers results)
425             results))
426          ((member key '("first" "last"))
427           (let* ((numbers (or numbers
428                               (modb-standard-number-list-internal msgdb)))
429                  (len (length numbers))
430                  (lastp (string= key "last"))
431                  (value (string-to-number (elmo-filter-value condition))))
432             (when (eq (elmo-filter-type condition) 'unmatch)
433               (setq lastp (not lastp)
434                     value (- len value)))
435             (if lastp
436                 (nthcdr (max (- len value) 0) numbers)
437               (when (> value 0)
438                 (let* ((numbers (copy-sequence numbers))
439                        (last (nthcdr (1- value) numbers)))
440                   (when last
441                     (setcdr last nil))
442                   numbers)))))
443          (t
444           t)))
445     t))
446
447 (luna-define-method elmo-msgdb-append-entity ((msgdb modb-standard)
448                                               entity &optional flags)
449   (when entity
450     (let ((number (elmo-msgdb-message-entity-number
451                    (elmo-message-entity-handler entity) entity))
452           (msg-id (elmo-msgdb-message-entity-field
453                    (elmo-message-entity-handler entity) entity 'message-id))
454           duplicate)
455       (when msg-id
456         ;; number-list
457         (modb-standard-set-number-list-internal
458          msgdb
459          (nconc (modb-standard-number-list-internal msgdb)
460                 (list number)))
461         ;; entity-map
462         (let ((table (modb-standard-entity-map msgdb)))
463           (setq duplicate (elmo-get-hash-val msg-id table))
464           (elmo-set-hash-val (modb-standard-key number) entity table)
465           (elmo-set-hash-val msg-id entity table))
466         ;; modification flags
467         (modb-standard-set-message-modified msgdb number)
468         ;; flag-map
469         (when flags
470           (elmo-set-hash-val
471            (modb-standard-key number)
472            (cons number flags)
473            (modb-standard-flag-map msgdb))
474           (modb-standard-countup-flags msgdb flags)
475           (modb-standard-set-flag-modified msgdb number))
476         duplicate))))
477
478 (luna-define-method elmo-msgdb-delete-messages ((msgdb modb-standard)
479                                                 numbers)
480   (let ((number-list (modb-standard-number-list-internal msgdb))
481         (entity-map (modb-standard-entity-map-internal msgdb))
482         (flag-map (modb-standard-flag-map-internal msgdb))
483         key entity)
484     (dolist (number numbers)
485       (setq key (modb-standard-key number)
486             entity (elmo-get-hash-val key entity-map))
487       (when entity
488         ;; number-list
489         (setq number-list (delq number number-list))
490         ;; entity-map
491         (elmo-clear-hash-val key entity-map)
492         (elmo-clear-hash-val (modb-standard-entity-id entity) entity-map)
493         ;; flag-count (must be BEFORE flag-map)
494         (modb-standard-countup-flags
495          msgdb
496          (modb-standard-message-flags msgdb number)
497          -1)
498         ;; flag-map
499         (elmo-clear-hash-val key flag-map)
500         (modb-standard-set-message-modified msgdb number)
501         (modb-standard-set-flag-modified msgdb number)))
502     (modb-standard-set-number-list-internal msgdb number-list)
503     (modb-standard-set-entity-map-internal msgdb entity-map)
504     (modb-standard-set-flag-map-internal msgdb flag-map)
505     t))
506
507 (luna-define-method elmo-msgdb-sort-entities ((msgdb modb-standard)
508                                               predicate &optional app-data)
509   (message "Sorting...")
510   (let ((numbers (modb-standard-number-list-internal msgdb)))
511     (modb-standard-set-number-list-internal
512      msgdb
513      (sort numbers (lambda (a b)
514                      (funcall predicate
515                               (elmo-msgdb-message-entity msgdb a)
516                               (elmo-msgdb-message-entity msgdb b)
517                               app-data))))
518     (message "Sorting...done")
519     msgdb))
520
521 (defun modb-standard-message-entity (msgdb key load)
522   (let ((ret (elmo-get-hash-val
523               key
524               (modb-standard-entity-map-internal msgdb)))
525         (inhibit-quit t))
526     (if (eq 'autoload (car-safe ret))
527         (when (and load modb-standard-divide-number)
528           (modb-standard-load-entity
529            msgdb
530            (elmo-msgdb-location msgdb)
531            (/ (nth 1 ret) modb-standard-divide-number))
532           (modb-standard-message-entity msgdb key nil))
533       ret)))
534
535 (luna-define-method elmo-msgdb-message-number ((msgdb modb-standard)
536                                                message-id)
537   (let ((ret (elmo-get-hash-val
538               message-id
539               (modb-standard-entity-map-internal msgdb))))
540     (if (eq 'autoload (car-safe ret))
541         ;; Not loaded yet but can return number.
542         (nth 1 ret)
543       (elmo-message-entity-number ret))))
544
545 (luna-define-method elmo-msgdb-message-field ((msgdb modb-standard)
546                                               number field)
547   (let ((ret (elmo-get-hash-val
548               (modb-standard-key number)
549               (modb-standard-entity-map-internal msgdb))))
550     (if (and (eq 'autoload (car-safe ret)) (eq field 'message-id))
551         ;; Not loaded yet but can return message-id
552         (cdr (cdr ret))
553       (elmo-message-entity-field (elmo-msgdb-message-entity
554                                   msgdb (modb-standard-key number))
555                                  field))))
556
557 (luna-define-method elmo-msgdb-message-entity ((msgdb modb-standard) key)
558   (when key
559     (modb-standard-message-entity
560      msgdb
561      (cond ((stringp key) key)
562            ((numberp key) (modb-standard-key key)))
563      'autoload)))
564
565 (require 'product)
566 (product-provide (provide 'modb-standard) (require 'elmo-version))
567
568 ;;; modb-standard.el ends here