main.c

   1 /*
   2  * main.c
   3  * Created: Sun Feb 25 20:11:31 2001 by tek@wiw.org
   4  * Revised: Sun Apr 22 19:09:44 2001 by tek@wiw.org
   5  * Copyright 2001 Julian E. C. Squires (tek@wiw.org)
   6  * This program comes with ABSOLUTELY NO WARRANTY.
   7  * $Id: main.c,v 1.5 2001/04/22 21:41:23 tek Exp $
   8  *
   9  */
  10
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include <string.h>
  14
  15 #include "noose.h"
  16
  17 typedef struct noose_s {
  18     int verbose;
  19     void *nh;
  20     char *newsrc;
  21 } noose_t;
  22
  23 void usage(char *msg);
  24 int do_check(noose_t *ndat, char *group);
  25 int checksingle(noose_t *ndat, char *group);
  26 int do_read(noose_t *ndat, char *group, int article);
  27 int setnewsrc(noose_t *ndat);
  28
  29 int main(int argc, char **argv)
  30 {
  31     enum { nocommand, check, read } command = nocommand;
  32     char *group, *server;
  33     noose_t ndat;
  34     int i, j, article, ret;
  35
  36     ndat.verbose = 1;
  37     ndat.newsrc = NULL;
  38     server = getenv("NNTPSERVER");
  39
  40     for(i = 1; i < argc; i++) {
  41         if(strcmp(argv[i], "-n") == 0) {
  42             i++;
  43             if(i == argc)
  44                 usage("Option `n' requires an argument.");
  45             server = argv[i];
  46
  47         } else if(strcmp(argv[i], "-r") == 0) {
  48             i++;
  49             if(i == argc)
  50                 usage("Option `r' requires an argument.");
  51             ndat.newsrc = malloc(strlen(argv[i])+1);
  52             strcpy(ndat.newsrc, argv[i]);
  53
  54         } else if(argv[i][0] == '-') {
  55             for(j = 1; j < strlen(argv[i]); j++)
  56                 switch(argv[i][j]) {
  57                 case 'q':
  58                     ndat.verbose--;
  59                     break;
  60
  61                 case 'v':
  62                     ndat.verbose++;
  63                     break;
  64
  65                 default:
  66                     usage("Bad option.");
  67                 }
  68         } else if(strcmp(argv[i], "check") == 0) {
  69             command = check;
  70             i++;
  71             if(i < argc) {
  72                 group = argv[i];
  73             } else
  74                 group = NULL;
  75
  76         } else if(strcmp(argv[i], "read") == 0) {
  77             command = read;
  78             i++;
  79             if(i+1 >= argc) usage("Too few arguments for `read'.");
  80             group = argv[i];
  81             i++;
  82             article = atoi(argv[i]);
  83             if(article == 0) usage("Invalid article number.");
  84
  85         } else
  86             usage("Bad argument.");
  87     }
  88     if(command == nocommand) {
  89         usage("No command specified.");
  90     }
  91
  92     if(server == NULL)
  93         server = DEFNNTPSERVER;
  94
  95     ndat.nh = nntp_connect(server);
  96     if(ndat.nh == NULL) {
  97         if(ndat.verbose >= 0)
  98             fprintf(stderr, "%s: Failed to connect to `%s'.\n", PROGNAME,
  99                     server);
 100         exit(EXIT_FAILURE);
 101     }
 102
 103     switch(command) {
 104     case check:
 105         ret = do_check(&ndat, group);
 106         break;
 107
 108     case read:
 109         ret = do_read(&ndat, group, article);
 110         break;
 111
 112     default:
 113         fprintf(stderr, "Command unsupported.\n");
 114         exit(EXIT_FAILURE);
 115     }
 116
 117     if(ndat.newsrc) free(ndat.newsrc);
 118     nntp_disconnect(ndat.nh);
 119     exit((ret == 0)?EXIT_SUCCESS:EXIT_FAILURE);
 120 }
 121
 122 int setnewsrc(noose_t *ndat)
 123 {
 124     char *home;
 125     char *newsrcs[] = { "/.jnewsrc", "/.newsrc", NULL };
 126     int i;
 127     FILE *fp;
 128
 129     home = getenv("HOME");
 130     if(home == NULL) {
 131         fprintf(stderr, "%s: Please set HOME in your environment.\n",
 132                 PROGNAME);
 133         return -1;
 134     }
 135     for(i = 0; newsrcs[i] != NULL; i++) {
 136         ndat->newsrc = malloc(strlen(home)+strlen(newsrcs[i]));
 137         strcpy(ndat->newsrc, home);
 138         strcat(ndat->newsrc, newsrcs[i]);
 139         if((fp = fopen(ndat->newsrc, "r")) != NULL) {
 140             fclose(fp);
 141             break;
 142         }
 143         free(ndat->newsrc);
 144         ndat->newsrc = NULL;
 145     }
 146     if(newsrcs[i] == NULL) {
 147         fprintf(stderr, "%s: Please specify a newsrc file to use.\n",
 148                 PROGNAME);
 149         return -1;
 150     }
 151     return 0;
 152 }
 153
 154 int do_check(noose_t *ndat, char *group)
 155 {
 156     int nsubbedgroups, i;
 157     char **subbedgroups;
 158
 159     if(ndat->newsrc == NULL)
 160         if(setnewsrc(ndat) != 0) return -1;
 161
 162     if(group == NULL) {
 163         i = newsrc_getsubscribedgroups(ndat->newsrc, &nsubbedgroups,
 164                                        &subbedgroups);
 165         if(i != 0) {
 166             if(ndat->verbose >= 0)
 167                 fprintf(stderr, "%s: Couldn't read subscribed groups from "
 168                         "your newsrc.\n", PROGNAME);
 169             return -1;
 170         }
 171         for(i = 0; i < nsubbedgroups; i++) {
 172             checksingle(ndat, subbedgroups[i]);
 173             free(subbedgroups[i]);
 174         }
 175     } else {
 176         checksingle(ndat, group);
 177     }
 178
 179     return 0;
 180 }
 181
 182 int checksingle(noose_t *ndat, char *group)
 183 {
 184     rangelist_t *rl, *p;
 185     int i, narticles;
 186
 187     rl = rl_new(0, 0, NULL);
 188
 189     if(nntp_cmd_group(ndat->nh, group, NULL, &rl->begin, &rl->end) != 0) {
 190         if(ndat->verbose >= 0)
 191             fprintf(stderr, "%s: nntp_cmd_group failed.\n", PROGNAME);
 192         return -1;
 193     }
 194
 195     newsrc_filter(ndat->newsrc, group, &rl);
 196
 197     narticles = 0;
 198     p = rl;
 199     while(p != NULL) {
 200         for(i = p->begin; i > 0 && i <= p->end; i++) {
 201             if(nntp_cmd_article(ndat->nh, i, NULL, NULL) == 0) {
 202                 narticles++;
 203             }
 204         }
 205         p = p->next;
 206     }
 207
 208     if(narticles > 0 || ndat->verbose > 0) {
 209         printf("There %s %d unread article%s in newsgroup %s.\n",
 210                (narticles == 1)?"is":"are", narticles,
 211                (narticles == 1)?"":"s", group);
 212     }
 213
 214     rl_delete(rl);
 215     return 0;
 216 }
 217
 218 int do_read(noose_t *ndat, char *group, int article)
 219 {
 220     char *artbody, *arthead;
 221
 222     if(nntp_cmd_group(ndat->nh, group, NULL, NULL, NULL) == -1) {
 223         if(ndat->verbose >= 0)
 224             fprintf(stderr, "%s: nntp_cmd_group failed.\n", PROGNAME);
 225         return -1;
 226     }
 227
 228     if(nntp_cmd_article(ndat->nh, article, &arthead, &artbody) == -1) {
 229         if(ndat->verbose >= 0)
 230             fprintf(stderr, "%s: nntp_cmd_article failed.\n", PROGNAME);
 231         return -1;
 232     }
 233
 234     printf("%s%s", arthead, artbody);
 235     free(arthead);
 236     free(artbody);
 237     return 0;
 238 }
 239
 240 void usage(char *msg)
 241 {
 242     printf("%s: %s\n\n", PROGNAME, msg);
 243     printf("usage: %s [options...] <command>\n\n"
 244 "Valid options are:\n"
 245 "  -q                    Quiet mode. This makes `check' only report groups\n"
 246 "                        with more than zero new messages in them.\n"
 247 "  -v                    Verbose. The inverse of quiet.\n"
 248 "  -n server             Specify NNTP server.\n"
 249 "  -r rcfile             Specify newsrc file.\n"
 250 "\n"
 251 "Valid commands are:\n"
 252 "  check [g]             Return how many unread posts are in all subscribed\n"
 253 "                        groups, or just group g if specified.\n"
 254 "  read <g> <n>          Grab the entirity of article n from group g.\n",
 255            PROGNAME);
 256     exit(EXIT_FAILURE);
 257 }
 258
 259 /* EOF main.c */