Nico Golde:
[apps/madmutt.git] / nntp / newsrc.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 1998 Brandon Long <blong@fiction.net>
4  * Copyright (C) 1999 Andrej Gritsenko <andrej@lucky.net>
5  * Copyright (C) 2000-2001 Vsevolod Volkov <vvv@mutt.org.ua>
6  *
7  * This file is part of mutt-ng, see http://www.muttng.org/.
8  * It's licensed under the GNU General Public License,
9  * please see the file GPL in the top level source directory.
10  */
11
12 #ifdef HAVE_CONFIG_H
13 #include <config.h>
14 #endif
15
16 #include "mutt.h"
17 #include "mutt_curses.h"
18 #include "sort.h"
19 #include "mx.h"
20 #include "mime.h"
21 #include "nntp.h"
22 #include "rfc822.h"
23 #include "rfc1524.h"
24 #include "rfc2047.h"
25
26 #include "lib/mem.h"
27 #include "lib/str.h"
28 #include "lib/intl.h"
29 #include "lib/debug.h"
30
31 #include <unistd.h>
32 #include <string.h>
33 #include <ctype.h>
34 #include <stdlib.h>
35 #include <sys/stat.h>
36
37 void nntp_add_to_list (NNTP_SERVER * s, NNTP_DATA * d)
38 {
39   LIST *l;
40
41   if (!s || !d)
42     return;
43
44   l = safe_calloc (1, sizeof (LIST));
45   if (s->list)
46     s->tail->next = l;
47   else
48     s->list = l;
49   s->tail = l;
50   l->data = (void *) d;
51 }
52
53 static int nntp_parse_newsrc_line (NNTP_SERVER * news, char *line)
54 {
55   NNTP_DATA *data;
56   char group[LONG_STRING];
57   int x = 1;
58   char *p = line, *b, *h;
59   size_t len;
60
61   while (*p) {
62     if (*p++ == ',')
63       x++;
64   }
65
66   p = line;
67   while (*p && (*p != ':' && *p != '!'))
68     p++;
69   if (!*p)
70     return -1;
71   len = p + 1 - line;
72   if (len > sizeof (group))
73     len = sizeof (group);
74   strfcpy (group, line, len);
75   if ((data = (NNTP_DATA *) hash_find (news->newsgroups, group)) == NULL) {
76     data =
77       (NNTP_DATA *) safe_calloc (1, sizeof (NNTP_DATA) + mutt_strlen (group) + 1);
78     data->group = (char *) data + sizeof (NNTP_DATA);
79     strcpy (data->group, group);
80     data->nserv = news;
81     data->deleted = 1;
82     if (news->newsgroups->nelem < news->newsgroups->curnelem * 2)
83       news->newsgroups =
84         hash_resize (news->newsgroups, news->newsgroups->nelem * 2);
85     hash_insert (news->newsgroups, data->group, data, 0);
86     nntp_add_to_list (news, data);
87   }
88   else
89     FREE ((void **) &data->entries);
90
91   data->rc = 1;
92   data->entries = safe_calloc (x * 2, sizeof (NEWSRC_ENTRY));
93   data->max = x * 2;
94
95   if (*p == ':')
96     data->subscribed = 1;
97   else
98     data->subscribed = 0;
99
100   p++;
101   b = p;
102   x = 0;
103   while (*b) {
104     while (*p && *p != ',' && *p != '\n')
105       p++;
106     if (*p) {
107       *p = '\0';
108       p++;
109     }
110     if ((h = strchr (b, '-'))) {
111       *h = '\0';
112       h++;
113       data->entries[x].first = atoi (b);
114       data->entries[x].last = atoi (h);
115     }
116     else {
117       data->entries[x].first = atoi (b);
118       data->entries[x].last = data->entries[x].first;
119     }
120     b = p;
121     if (data->entries[x].last != 0)
122       x++;
123   }
124   if (x && !data->lastMessage)
125     data->lastMessage = data->entries[x - 1].last;
126   data->num = x;
127   mutt_newsgroup_stat (data);
128   debug_print (2, ("Newsgroup %s\n", data->group));
129
130   return 0;
131 }
132
133 static int slurp_newsrc (NNTP_SERVER * news)
134 {
135   FILE *fp;
136   char *buf;
137   struct stat sb;
138
139   news->stat = stat (news->newsrc, &sb);
140   news->size = sb.st_size;
141   news->mtime = sb.st_mtime;
142
143   if ((fp = safe_fopen (news->newsrc, "r")) == NULL)
144     return -1;
145   /* hmm, should we use dotlock? */
146   if (mx_lock_file (news->newsrc, fileno (fp), 0, 0, 1)) {
147     fclose (fp);
148     return -1;
149   }
150
151   buf = safe_malloc (sb.st_size + 1);
152   while (fgets (buf, sb.st_size + 1, fp))
153     nntp_parse_newsrc_line (news, buf);
154   FREE (&buf);
155
156   mx_unlock_file (news->newsrc, fileno (fp), 0);
157   fclose (fp);
158   return 0;
159 }
160
161 void nntp_cache_expand (char *dst, const char *src)
162 {
163   snprintf (dst, _POSIX_PATH_MAX, "%s/%s", NewsCacheDir, src);
164   mutt_expand_path (dst, _POSIX_PATH_MAX);
165 }
166
167 /* Loads $news_cache_dir/.index into memory, loads newsserver data
168  * and newsgroup cache names */
169 static int nntp_parse_cacheindex (NNTP_SERVER * news)
170 {
171   struct stat st;
172   char buf[HUGE_STRING], *cp;
173   char dir[_POSIX_PATH_MAX], file[_POSIX_PATH_MAX];
174   FILE *index;
175   NNTP_DATA *data;
176   int l, m, t;
177
178   /* check is server name defined or not */
179   if (!news || !news->conn || !news->conn->account.host)
180     return -1;
181   unset_option (OPTNEWSCACHE);
182   if (!NewsCacheDir || !*NewsCacheDir)
183     return 0;
184
185   strfcpy (dir, NewsCacheDir, sizeof (dir));
186   mutt_expand_path (dir, sizeof (dir));
187
188   if (lstat (dir, &st) || (st.st_mode & S_IFDIR) == 0) {
189     snprintf (buf, sizeof (buf), _("Directory %s not exist. Create it?"),
190               dir);
191     if (mutt_yesorno (buf, M_YES) != M_YES
192         || mkdir (dir, (S_IRWXU + S_IRWXG + S_IRWXO))) {
193       mutt_error _("Cache directory not created!");
194
195       return -1;
196     }
197     mutt_clear_error ();
198   }
199
200   set_option (OPTNEWSCACHE);
201
202   FREE (&news->cache);
203   snprintf (buf, sizeof (buf), "%s/.index", dir);
204   if (!(index = safe_fopen (buf, "a+")))
205     return 0;
206   rewind (index);
207   while (fgets (buf, sizeof (buf), index)) {
208     buf[mutt_strlen (buf) - 1] = 0;  /* strip ending '\n' */
209     if (!safe_strncmp (buf, "#: ", 3) &&
210         !safe_strcasecmp (buf + 3, news->conn->account.host))
211       break;
212   }
213   while (fgets (buf, sizeof (buf), index)) {
214     cp = buf;
215     while (*cp && *cp != ' ')
216       cp++;
217     if (!*cp)
218       continue;
219     cp[0] = 0;
220     if (!mutt_strcmp (buf, "#:"))
221       break;
222     sscanf (cp + 1, "%s %d %d", file, &l, &m);
223     if (!mutt_strcmp (buf, "ALL")) {
224       news->cache = safe_strdup (file);
225       news->newgroups_time = m;
226     }
227     else if (news->newsgroups) {
228       if ((data = (NNTP_DATA *) hash_find (news->newsgroups, buf)) == NULL) {
229         data =
230           (NNTP_DATA *) safe_calloc (1,
231                                      sizeof (NNTP_DATA) + mutt_strlen (buf) + 1);
232         data->group = (char *) data + sizeof (NNTP_DATA);
233         strcpy (data->group, buf);
234         data->nserv = news;
235         data->deleted = 1;
236         if (news->newsgroups->nelem < news->newsgroups->curnelem * 2)
237           news->newsgroups =
238             hash_resize (news->newsgroups, news->newsgroups->nelem * 2);
239         hash_insert (news->newsgroups, data->group, data, 0);
240         nntp_add_to_list (news, data);
241       }
242       data->cache = safe_strdup (file);
243       t = 0;
244       if (!data->firstMessage || data->lastMessage < m)
245         t = 1;
246       if (!data->firstMessage)
247         data->firstMessage = l;
248       if (data->lastMessage < m)
249         data->lastMessage = m;
250       data->lastCached = m;
251       if (t || !data->unread)
252         mutt_newsgroup_stat (data);
253     }
254   }
255   fclose (index);
256   return 0;
257 }
258
259 const char *nntp_format_str (char *dest, size_t destlen, char op,
260                              const char *src, const char *fmt,
261                              const char *ifstring, const char *elsestring,
262                              unsigned long data, format_flag flags)
263 {
264   char fn[SHORT_STRING], tmp[SHORT_STRING];
265
266   switch (op) {
267   case 's':
268     strncpy (fn, NewsServer, sizeof (fn) - 1);
269     str_tolower (fn);
270     snprintf (tmp, sizeof (tmp), "%%%ss", fmt);
271     snprintf (dest, destlen, tmp, fn);
272     break;
273   }
274   return (src);
275 }
276
277 /* nntp_parse_url: given an NNPT URL, return host, port,
278  * username, password and newsgroup will recognise. */
279 int nntp_parse_url (const char *server, ACCOUNT * acct,
280                     char *group, size_t group_len)
281 {
282   ciss_url_t url;
283   char *c;
284   int ret = -1;
285
286   /* Defaults */
287   acct->flags = 0;
288   acct->port = NNTP_PORT;
289   acct->type = M_ACCT_TYPE_NNTP;
290
291   c = safe_strdup (server);
292   url_parse_ciss (&url, c);
293
294   if (url.scheme == U_NNTP || url.scheme == U_NNTPS) {
295     if (url.scheme == U_NNTPS) {
296       acct->flags |= M_ACCT_SSL;
297       acct->port = NNTP_SSL_PORT;
298     }
299
300     *group = '\0';
301     if (url.path)
302       strfcpy (group, url.path, group_len);
303
304     ret = mutt_account_fromurl (acct, &url);
305   }
306
307   FREE (&c);
308   return ret;
309 }
310
311 void nntp_expand_path (char *line, size_t len, ACCOUNT * acct)
312 {
313   ciss_url_t url;
314
315   url.path = safe_strdup (line);
316   mutt_account_tourl (acct, &url);
317   url_ciss_tostring (&url, line, len, 0);
318   FREE (&url.path);
319 }
320
321 /*
322  * Automatically loads a newsrc into memory, if necessary.
323  * Checks the size/mtime of a newsrc file, if it doesn't match, load
324  * again.  Hmm, if a system has broken mtimes, this might mean the file
325  * is reloaded every time, which we'd have to fix.
326  *
327  * a newsrc file is a line per newsgroup, with the newsgroup, then a 
328  * ':' denoting subscribed or '!' denoting unsubscribed, then a 
329  * comma separated list of article numbers and ranges.
330  */
331 NNTP_SERVER *mutt_select_newsserver (char *server)
332 {
333   char file[_POSIX_PATH_MAX];
334   char *buf, *p;
335   LIST *list;
336   ACCOUNT acct;
337   NNTP_SERVER *serv;
338   CONNECTION *conn;
339
340   if (!server || !*server) {
341     mutt_error _("No newsserver defined!");
342
343     return NULL;
344   }
345
346   buf = p = safe_calloc (mutt_strlen (server) + 10, sizeof (char));
347   if (url_check_scheme (server) == U_UNKNOWN) {
348     strcpy (buf, "nntp://");
349     p = strchr (buf, '\0');
350   }
351   strcpy (p, server);
352
353   if ((nntp_parse_url (buf, &acct, file, sizeof (file))) < 0 || *file) {
354     FREE (&buf);
355     mutt_error (_("%s is an invalid newsserver specification!"), server);
356     return NULL;
357   }
358   FREE (&buf);
359
360   conn = mutt_conn_find (NULL, &acct);
361   if (!conn)
362     return NULL;
363
364   mutt_FormatString (file, sizeof (file), NONULL (NewsRc), nntp_format_str, 0,
365                      0);
366   mutt_expand_path (file, sizeof (file));
367
368   serv = (NNTP_SERVER *) conn->data;
369   if (serv) {
370     struct stat sb;
371
372     /* externally modified? */
373     if (serv->stat != stat (file, &sb) || (!serv->stat &&
374                                            (serv->size != sb.st_size
375                                             || serv->mtime != sb.st_mtime))) {
376       for (list = serv->list; list; list = list->next) {
377         NNTP_DATA *data = (NNTP_DATA *) list->data;
378
379         if (data) {
380           data->subscribed = 0;
381           data->rc = 0;
382           data->num = 0;
383         }
384       }
385       slurp_newsrc (serv);
386       nntp_clear_cacheindex (serv);
387     }
388
389     if (serv->status == NNTP_BYE)
390       serv->status = NNTP_NONE;
391     nntp_check_newgroups (serv, 0);
392     return serv;
393   }
394
395   /* New newsserver */
396   serv = safe_calloc (1, sizeof (NNTP_SERVER));
397   serv->conn = conn;
398   serv->newsrc = safe_strdup (file);
399   serv->newsgroups = hash_create (1009);
400   slurp_newsrc (serv);          /* load .newsrc */
401   nntp_parse_cacheindex (serv); /* load .index */
402   if (option (OPTNEWSCACHE) && serv->cache && nntp_get_cache_all (serv) >= 0)
403     nntp_check_newgroups (serv, 1);
404   else if (nntp_get_active (serv) < 0) {
405     hash_destroy (&serv->newsgroups, nntp_delete_data);
406     for (list = serv->list; list; list = list->next)
407       list->data = NULL;
408     mutt_free_list (&serv->list);
409     FREE (&serv->newsrc);
410     FREE (&serv->cache);
411     FREE (&serv);
412     return NULL;
413   }
414   nntp_clear_cacheindex (serv);
415   conn->data = (void *) serv;
416
417   return serv;
418 }
419
420 /* 
421  * full status flags are not supported by nntp, but we can fake some
422  * of them.  This is how:
423  * Read = a read message number is in the .newsrc
424  * New = a message is new since we last read this newsgroup
425  * Old = anything else
426  * So, Read is marked as such in the newsrc, old is anything that is 
427  * "skipped" in the newsrc, and new is anything not in the newsrc nor
428  * in the cache. By skipped, I mean before the last unread message
429  */
430 void nntp_get_status (CONTEXT * ctx, HEADER * h, char *group, int article)
431 {
432   NNTP_DATA *data = (NNTP_DATA *) ctx->data;
433   int x;
434
435   if (group)
436     data = (NNTP_DATA *) hash_find (data->nserv->newsgroups, group);
437
438   if (!data) {
439 #ifdef DEBUG
440     if (group)
441       debug_print (3, ("newsgroup %s not found\n", group));
442 #endif
443     return;
444   }
445
446   for (x = 0; x < data->num; x++) {
447     if ((article >= data->entries[x].first) &&
448         (article <= data->entries[x].last)) {
449       /* we cannot use mutt_set_flag() because mx_update_context()
450          didn't called yet */
451       h->read = 1;
452       return;
453     }
454   }
455   /* If article was not cached yet, it is new! :) */
456   if (!data->cache || article > data->lastCached)
457     return;
458   /* Old articles are articles which aren't read but an article after them
459    * has been cached */
460   if (option (OPTMARKOLD))
461     h->old = 1;
462 }
463
464 void mutt_newsgroup_stat (NNTP_DATA * data)
465 {
466   int i;
467   unsigned int first, last;
468
469   data->unread = 0;
470   if (data->lastMessage == 0 || data->firstMessage > data->lastMessage)
471     return;
472
473   data->unread = data->lastMessage - data->firstMessage + 1;
474   for (i = 0; i < data->num; i++) {
475     first = data->entries[i].first;
476     if (first < data->firstMessage)
477       first = data->firstMessage;
478     last = data->entries[i].last;
479     if (last > data->lastMessage)
480       last = data->lastMessage;
481     if (first <= last)
482       data->unread -= last - first + 1;
483   }
484 }
485
486 static int puti (char *line, int num)
487 {
488   char *p, s[32];
489
490   for (p = s; num;) {
491     *p++ = '0' + num % 10;
492     num /= 10;
493   }
494   while (p > s)
495     *line++ = *--p, num++;
496   *line = '\0';
497   return num;
498 }
499
500 static void nntp_create_newsrc_line (NNTP_DATA * data, char **buf,
501                                      char **pline, size_t * buflen)
502 {
503   char *line = *pline;
504   size_t len = *buflen - (*pline - *buf);
505   int x, i;
506
507   if (len < LONG_STRING * 10) {
508     len += *buflen;
509     *buflen *= 2;
510     line = *buf;
511     safe_realloc (buf, *buflen);
512     line = *buf + (*pline - line);
513   }
514   strcpy (line, data->group);
515   len -= mutt_strlen (line) + 1;
516   line += mutt_strlen (line);
517   *line++ = data->subscribed ? ':' : '!';
518   *line++ = ' ';
519   *line = '\0';
520
521   for (x = 0; x < data->num; x++) {
522     if (len < LONG_STRING) {
523       len += *buflen;
524       *buflen *= 2;
525       *pline = line;
526       line = *buf;
527       safe_realloc (buf, *buflen);
528       line = *buf + (*pline - line);
529     }
530     if (x) {
531       *line++ = ',';
532       len--;
533     }
534
535 #if 0
536     if (data->entries[x].first == data->entries[x].last)
537       snprintf (line, len, "%d%n", data->entries[x].first, &i);
538     else
539       snprintf (line, len, "%d-%d%n",
540                 data->entries[x].first, data->entries[x].last, &i);
541     len -= i;
542     line += i;
543 #else
544     i = puti (line, data->entries[x].first);
545     line += i;
546     len -= i;
547     if (data->entries[x].first != data->entries[x].last) {
548       *line++ = '-';
549       len--;
550       i = puti (line, data->entries[x].last);
551       line += i;
552       len -= i;
553     }
554 #endif
555   }
556   *line++ = '\n';
557   *line = '\0';
558   *pline = line;
559 }
560
561 void newsrc_gen_entries (CONTEXT * ctx)
562 {
563   NNTP_DATA *data = (NNTP_DATA *) ctx->data;
564   int series, x;
565   unsigned int last = 0, first = 1;
566   int save_sort = SORT_ORDER;
567
568   if (Sort != SORT_ORDER) {
569     save_sort = Sort;
570     Sort = SORT_ORDER;
571     mutt_sort_headers (ctx, 0);
572   }
573
574   if (!data->max) {
575     data->entries = safe_calloc (5, sizeof (NEWSRC_ENTRY));
576     data->max = 5;
577   }
578
579   /*
580    * Set up to fake initial sequence from 1 to the article before the 
581    * first article in our list
582    */
583   data->num = 0;
584   series = 1;
585
586   for (x = 0; x < ctx->msgcount; x++) {
587     if (series) {               /* search for first unread */
588       /*
589        * We don't actually check sequential order, since we mark 
590        * "missing" entries as read/deleted
591        */
592       last = ctx->hdrs[x]->article_num;
593       if (last >= data->firstMessage && !ctx->hdrs[x]->deleted &&
594           !ctx->hdrs[x]->read) {
595         if (data->num >= data->max) {
596           data->max = data->max * 2;
597           safe_realloc (&data->entries, data->max * sizeof (NEWSRC_ENTRY));
598         }
599         data->entries[data->num].first = first;
600         data->entries[data->num].last = last - 1;
601         data->num++;
602         series = 0;
603       }
604     }
605     else {                      /* search for first read */
606
607       if (ctx->hdrs[x]->deleted || ctx->hdrs[x]->read) {
608         first = last + 1;
609         series = 1;
610       }
611       last = ctx->hdrs[x]->article_num;
612     }
613   }
614   if (series && first <= data->lastLoaded) {
615     if (data->num >= data->max) {
616       data->max = data->max * 2;
617       safe_realloc (&data->entries, data->max * sizeof (NEWSRC_ENTRY));
618     }
619     data->entries[data->num].first = first;
620     data->entries[data->num].last = data->lastLoaded;
621     data->num++;
622   }
623
624   if (save_sort != Sort) {
625     Sort = save_sort;
626     mutt_sort_headers (ctx, 0);
627   }
628 }
629
630 int mutt_newsrc_update (NNTP_SERVER * news)
631 {
632   char *buf, *line;
633   NNTP_DATA *data;
634   LIST *tmp;
635   int r = -1;
636   size_t len, llen;
637
638   if (!news)
639     return -1;
640   llen = len = 10 * LONG_STRING;
641   line = buf = safe_calloc (1, len);
642   /* we will generate full newsrc here */
643   for (tmp = news->list; tmp; tmp = tmp->next) {
644     data = (NNTP_DATA *) tmp->data;
645     if (!data || !data->rc)
646       continue;
647     nntp_create_newsrc_line (data, &buf, &line, &llen);
648     debug_print (2, ("Added to newsrc: %s\n", line));
649     line += mutt_strlen (line);
650   }
651   /* newrc being fully rewritten */
652   if (news->newsrc &&
653       (r = mutt_update_list_file (news->newsrc, NULL, "", buf)) == 0) {
654     struct stat st;
655
656     stat (news->newsrc, &st);
657     news->size = st.st_size;
658     news->mtime = st.st_mtime;
659   }
660   FREE (&buf);
661   return r;
662 }
663
664 static FILE *mutt_mkname (char *s)
665 {
666   char buf[_POSIX_PATH_MAX], *pc;
667   int fd;
668   FILE *fp;
669
670   nntp_cache_expand (buf, s);
671   if ((fp = safe_fopen (buf, "w")))
672     return fp;
673
674   nntp_cache_expand (buf, "cache-XXXXXX");
675   pc = buf + mutt_strlen (buf) - 12; /* positioning to "cache-XXXXXX" */
676   if ((fd = mkstemp (buf)) == -1)
677     return NULL;
678   strcpy (s, pc);               /* generated name */
679   return fdopen (fd, "w");
680 }
681
682 /* Updates info into .index file: ALL or about selected newsgroup */
683 static int nntp_update_cacheindex (NNTP_SERVER * serv, NNTP_DATA * data)
684 {
685   char buf[LONG_STRING], *key = "ALL";
686   char file[_POSIX_PATH_MAX];
687
688   if (!serv || !serv->conn || !serv->conn->account.host)
689     return -1;
690
691   if (data && data->group) {
692     key = data->group;
693     snprintf (buf, sizeof (buf), "%s %s %d %d", key, data->cache,
694               data->firstMessage, data->lastLoaded);
695   }
696   else {
697     strfcpy (file, serv->cache, sizeof (file));
698     snprintf (buf, sizeof (buf), "ALL %s 0 %d", file,
699               (int) serv->newgroups_time);
700   }
701   nntp_cache_expand (file, ".index");
702   return mutt_update_list_file (file, serv->conn->account.host, key, buf);
703 }
704
705 /* Remove cache files of unsubscribed newsgroups */
706 void nntp_clear_cacheindex (NNTP_SERVER * news)
707 {
708   NNTP_DATA *data;
709   LIST *tmp;
710
711   if (option (OPTSAVEUNSUB) || !news)
712     return;
713
714   for (tmp = news->list; tmp; tmp = tmp->next) {
715     data = (NNTP_DATA *) tmp->data;
716     if (!data || data->subscribed || !data->cache)
717       continue;
718     nntp_delete_cache (data);
719     debug_print (2, ("Removed from .index: %s\n", data->group));
720   }
721   return;
722 }
723
724 int nntp_save_cache_index (NNTP_SERVER * news)
725 {
726   char buf[HUGE_STRING];
727   char file[_POSIX_PATH_MAX];
728   NNTP_DATA *d;
729   FILE *f;
730   LIST *l;
731
732   if (!news || !news->newsgroups)
733     return -1;
734   if (!option (OPTNEWSCACHE))
735     return 0;
736
737   if (news->cache) {
738     nntp_cache_expand (file, news->cache);
739     unlink (file);
740     f = safe_fopen (file, "w");
741   }
742   else {
743     strfcpy (buf, news->conn->account.host, sizeof (buf));
744     f = mutt_mkname (buf);
745     news->cache = safe_strdup (buf);
746     nntp_cache_expand (file, buf);
747   }
748   if (!f)
749     return -1;
750
751   for (l = news->list; l; l = l->next) {
752     if ((d = (NNTP_DATA *) l->data) && !d->deleted) {
753       if (d->desc)
754         snprintf (buf, sizeof (buf), "%s %d %d %c %s\n", d->group,
755                   d->lastMessage, d->firstMessage, d->allowed ? 'y' : 'n',
756                   d->desc);
757       else
758         snprintf (buf, sizeof (buf), "%s %d %d %c\n", d->group,
759                   d->lastMessage, d->firstMessage, d->allowed ? 'y' : 'n');
760       if (fputs (buf, f) == EOF) {
761         fclose (f);
762         unlink (file);
763         return -1;
764       }
765     }
766   }
767   fclose (f);
768
769   if (nntp_update_cacheindex (news, NULL)) {
770     unlink (file);
771     return -1;
772   }
773   return 0;
774 }
775
776 int nntp_save_cache_group (CONTEXT * ctx)
777 {
778   char buf[HUGE_STRING], addr[STRING];
779   char file[_POSIX_PATH_MAX];
780   FILE *f;
781   HEADER *h;
782   struct tm *tm;
783   int i = 0, save = SORT_ORDER;
784   int prev = 0;
785
786   if (!option (OPTNEWSCACHE))
787     return 0;
788   if (!ctx || !ctx->data || ctx->magic != M_NNTP)
789     return -1;
790
791   if (((NNTP_DATA *) ctx->data)->cache) {
792     nntp_cache_expand (file, ((NNTP_DATA *) ctx->data)->cache);
793     unlink (file);
794     f = safe_fopen (file, "w");
795   }
796   else {
797     snprintf (buf, sizeof (buf), "%s-%s",
798               ((NNTP_DATA *) ctx->data)->nserv->conn->account.host,
799               ((NNTP_DATA *) ctx->data)->group);
800     f = mutt_mkname (buf);
801     ((NNTP_DATA *) ctx->data)->cache = safe_strdup (buf);
802     nntp_cache_expand (file, buf);
803   }
804   if (!f)
805     return -1;
806
807   if (Sort != SORT_ORDER) {
808     save = Sort;
809     Sort = SORT_ORDER;
810     mutt_sort_headers (ctx, 0);
811   }
812
813   /* Save only $nntp_context messages... */
814   ((NNTP_DATA *) ctx->data)->lastCached = 0;
815   if (NntpContext && ctx->msgcount > NntpContext)
816     i = ctx->msgcount - NntpContext;
817   for (; i < ctx->msgcount; i++) {
818     if (!ctx->hdrs[i]->deleted && ctx->hdrs[i]->article_num != prev) {
819       h = ctx->hdrs[i];
820       addr[0] = 0;
821       rfc822_write_address (addr, sizeof (addr), h->env->from, 0);
822       tm = gmtime (&h->date_sent);
823       snprintf (buf, sizeof (buf),
824                 "%d\t%s\t%s\t%d %s %d %02d:%02d:%02d GMT\t%s\t",
825                 h->article_num, h->env->subject, addr, tm->tm_mday,
826                 Months[tm->tm_mon], tm->tm_year + 1900, tm->tm_hour,
827                 tm->tm_min, tm->tm_sec, h->env->message_id);
828       fputs (buf, f);
829       if (h->env->references)
830         mutt_write_references (h->env->references, f);
831       snprintf (buf, sizeof (buf), "\t%ld\t%d\tXref: %s\n",
832                 h->content->length, h->lines, NONULL (h->env->xref));
833       if (fputs (buf, f) == EOF) {
834         fclose (f);
835         unlink (file);
836         return -1;
837       }
838     }
839     prev = ctx->hdrs[i]->article_num;
840   }
841
842   if (save != Sort) {
843     Sort = save;
844     mutt_sort_headers (ctx, 0);
845   }
846   fclose (f);
847
848   if (nntp_update_cacheindex (((NNTP_DATA *) ctx->data)->nserv,
849                               (NNTP_DATA *) ctx->data)) {
850     unlink (file);
851     return -1;
852   }
853   ((NNTP_DATA *) ctx->data)->lastCached =
854     ((NNTP_DATA *) ctx->data)->lastLoaded;
855   return 0;
856 }
857
858 void nntp_delete_cache (NNTP_DATA * data)
859 {
860   char buf[_POSIX_PATH_MAX];
861
862   if (!option (OPTNEWSCACHE) || !data || !data->cache || !data->nserv)
863     return;
864
865   nntp_cache_expand (buf, data->cache);
866   unlink (buf);
867   FREE (&data->cache);
868   data->lastCached = 0;
869   nntp_cache_expand (buf, ".index");
870   mutt_update_list_file (buf, data->nserv->conn->account.host, data->group,
871                          NULL);
872 }
873
874 NNTP_DATA *mutt_newsgroup_subscribe (NNTP_SERVER * news, char *group)
875 {
876   NNTP_DATA *data;
877
878   if (!news || !news->newsgroups || !group || !*group)
879     return NULL;
880   if (!(data = (NNTP_DATA *) hash_find (news->newsgroups, group))) {
881     data =
882       (NNTP_DATA *) safe_calloc (1, sizeof (NNTP_DATA) + mutt_strlen (group) + 1);
883     data->group = (char *) data + sizeof (NNTP_DATA);
884     strcpy (data->group, group);
885     data->nserv = news;
886     data->deleted = 1;
887     if (news->newsgroups->nelem < news->newsgroups->curnelem * 2)
888       news->newsgroups =
889         hash_resize (news->newsgroups, news->newsgroups->nelem * 2);
890     hash_insert (news->newsgroups, data->group, data, 0);
891     nntp_add_to_list (news, data);
892   }
893   if (!data->subscribed) {
894     data->subscribed = 1;
895     data->rc = 1;
896   }
897   return data;
898 }
899
900 NNTP_DATA *mutt_newsgroup_unsubscribe (NNTP_SERVER * news, char *group)
901 {
902   NNTP_DATA *data;
903
904   if (!news || !news->newsgroups || !group || !*group ||
905       !(data = (NNTP_DATA *) hash_find (news->newsgroups, group)))
906     return NULL;
907   if (data->subscribed) {
908     data->subscribed = 0;
909     if (!option (OPTSAVEUNSUB))
910       data->rc = 0;
911   }
912   return data;
913 }
914
915 NNTP_DATA *mutt_newsgroup_catchup (NNTP_SERVER * news, char *group)
916 {
917   NNTP_DATA *data;
918
919   if (!news || !news->newsgroups || !group || !*group ||
920       !(data = (NNTP_DATA *) hash_find (news->newsgroups, group)))
921     return NULL;
922   if (!data->max) {
923     data->entries = safe_calloc (5, sizeof (NEWSRC_ENTRY));
924     data->max = 5;
925   }
926   data->num = 1;
927   data->entries[0].first = 1;
928   data->unread = 0;
929   data->entries[0].last = data->lastMessage;
930   if (Context && Context->data == data) {
931     int x;
932
933     for (x = 0; x < Context->msgcount; x++)
934       mutt_set_flag (Context, Context->hdrs[x], M_READ, 1);
935   }
936   return data;
937 }
938
939 NNTP_DATA *mutt_newsgroup_uncatchup (NNTP_SERVER * news, char *group)
940 {
941   NNTP_DATA *data;
942
943   if (!news || !news->newsgroups || !group || !*group ||
944       !(data = (NNTP_DATA *) hash_find (news->newsgroups, group)))
945     return NULL;
946   if (!data->max) {
947     data->entries = safe_calloc (5, sizeof (NEWSRC_ENTRY));
948     data->max = 5;
949   }
950   data->num = 1;
951   data->entries[0].first = 1;
952   data->entries[0].last = data->firstMessage - 1;
953   if (Context && Context->data == data) {
954     int x;
955
956     data->unread = Context->msgcount;
957     for (x = 0; x < Context->msgcount; x++)
958       mutt_set_flag (Context, Context->hdrs[x], M_READ, 0);
959   }
960   else
961     data->unread = data->lastMessage - data->entries[0].last;
962   return data;
963 }
964
965 /* this routine gives the first newsgroup with new messages */
966 void nntp_buffy (char *s)
967 {
968   LIST *list;
969
970   for (list = CurrentNewsSrv->list; list; list = list->next) {
971     NNTP_DATA *data = (NNTP_DATA *) list->data;
972
973     if (data && data->subscribed && data->unread) {
974       if (Context && Context->magic == M_NNTP &&
975           !mutt_strcmp (data->group, ((NNTP_DATA *) Context->data)->group)) {
976         unsigned int i, unread = 0;
977
978         for (i = 0; i < Context->msgcount; i++)
979           if (!Context->hdrs[i]->read && !Context->hdrs[i]->deleted)
980             unread++;
981         if (!unread)
982           continue;
983       }
984       strcpy (s, data->group);
985       break;
986     }
987   }
988 }