#include "innbbsconf.h"
#include "bbslib.h"

extern char *optarg;
extern int opterr, optind;

usage(name)
char *name;
{
   fprintf(stderr, "Usage: %s [-p path] commands\n",name);
   fprintf(stderr, " where available commands:\n");
   fprintf(stderr,"  ctlinnbbsd reload   : reload datafiles for innbbsd\n");
   fprintf(stderr,"  ctlinnbbsd shutdown : shutdown innbbsd gracefully\n");
   fprintf(stderr,"  ctlinnbbsd mode     : examine mode of innbbsd\n");
   fprintf(stderr,"  ctlinnbbsd addhist <mid> path: add history\n");
   fprintf(stderr,"  ctlinnbbsd grephist <mid>: query history\n");
   fprintf(stderr,"  ctlinnbbsd verboselog on|off : verboselog on/off\n");
   fprintf(stderr,"  ctlinnbbsd hismaint : maintain history\n");
   fprintf(stderr,"  ctlinnbbsd listnodelist  : list nodelist.bbs\n");
   fprintf(stderr,"  ctlinnbbsd listnewsfeeds : list newsfeeds.bbs\n");
#ifdef GETRUSAGE
   fprintf(stderr,"  ctlinnbbsd getrusage: get resource usage\n");
#endif
#ifdef MALLOCMAP
   fprintf(stderr,"  ctlinnbbsd mallocmap: get malloc map\n");
#endif
}


char *DefaultPath = LOCALDAEMON;
char INNBBSbuffer[4096];

FILE *innbbsin, *innbbsout;
int innbbsfd;

ctlinnbbsd(argc, argv)
int argc;
char **argv;
{
	fgets(INNBBSbuffer, sizeof INNBBSbuffer, innbbsin);
	printf("%s",INNBBSbuffer);
	if (strcasecmp(argv[0], "shutdown")==0   || 
	    strcasecmp(argv[0], "reload")==0     ||
	    strcasecmp(argv[0], "hismaint")==0   ||
#ifdef GETRUSAGE
	    strcasecmp(argv[0], "getrusage")==0  ||
#endif
#ifdef MALLOCMAP
	    strcasecmp(argv[0], "mallocmap")==0  ||
#endif
	    strcasecmp(argv[0], "mode")==0       ||
	    strcasecmp(argv[0], "listnodelist")==0   ||
	    strcasecmp(argv[0], "listnewsfeeds")==0     
	    ) {
	    fprintf( innbbsout, "%s\r\n", argv[0]);
	    fflush( innbbsout);
	    fgets(INNBBSbuffer, sizeof INNBBSbuffer, innbbsin);
	    printf("%s",INNBBSbuffer);
	    if (strcasecmp(argv[0], "mode") ==0
#ifdef GETRUSAGE
		||
	        strcasecmp(argv[0], "getrusage") ==0  
		||
	        strcasecmp(argv[0], "mallocmap") ==0  
#endif
		||
	        strcasecmp(argv[0], "listnodelist")==0   
		||
	        strcasecmp(argv[0], "listnewsfeeds")==0     
	       ) {
		while (fgets(INNBBSbuffer, sizeof INNBBSbuffer, innbbsin) != NULL) {
		       if (strcmp(INNBBSbuffer,".\r\n")==0) {
				break;
		       }
		       printf("%s",INNBBSbuffer);
		}
	    }
	} else if (strcasecmp(argv[0], "grephist")==0 ||
	    strcasecmp(argv[0], "verboselog")==0  ) {
	    if (argc < 2) {
	       usage("ctlinnbbsd");
	    } else {
		fprintf( innbbsout, "%s %s\r\n", argv[0], argv[1]);
		fflush( innbbsout);
		fgets(INNBBSbuffer, sizeof INNBBSbuffer, innbbsin);
		printf("%s\n",INNBBSbuffer);
	    }
	} else if (strcasecmp(argv[0], "addhist")==0) {
	    if (argc < 3) {
	       usage("ctlinnbbsd");
	    } else {
		fprintf( innbbsout, "%s %s %s\r\n", argv[0], argv[1], argv[2]);
		fflush( innbbsout);
		fgets(INNBBSbuffer, sizeof INNBBSbuffer, innbbsin);
		printf("%s",INNBBSbuffer);
	    }
	} else {
	    fprintf(stderr, "invalid command %s\n", argv[0]);
	}
	if (strcasecmp(argv[0],"shutdown") != 0) {
		fprintf( innbbsout, "QUIT\r\n");
		fflush(innbbsout);
		fgets(INNBBSbuffer, sizeof INNBBSbuffer, innbbsin);
	}
}

initsocket()
{
     innbbsfd = unixclient(DefaultPath,"tcp");
     if (innbbsfd < 0) {
        fprintf(stderr, "Connect to %s error. You may not run innbbsd\n", DefaultPath);
        exit(2);
     }
     if ( (innbbsin= fdopen(innbbsfd,"r")) == NULL ||
          (innbbsout= fdopen(innbbsfd,"w"))== NULL ) {
       fprintf( stderr, "fdopen error\n");
       exit(3);
     }
}

closesocket()
{
    if (innbbsin != NULL)
       fclose(innbbsin);
    if (innbbsout != NULL)
       fclose(innbbsout);
    if (innbbsfd >= 0)
       close(innbbsfd);
}

main(argc, argv)
int argc;
char **argv;
{
    int c, errflag=0;

    while ((c = getopt(argc,argv,"p:h?"))!= -1)
      switch (c) {
      case 'p':
       DefaultPath = optarg;
       break;
      case 'h':
      case '?':
      default:
       errflag ++;
       break;
    }
    if (errflag > 0) {
       usage(argv[0]);
       return(1);
    }
    if (argc - optind < 1) {
       usage(argv[0]);
       exit(1);
    }
    initial_bbs(NULL);
    initsocket();
    ctlinnbbsd(argc-optind, argv+optind);
    closesocket();
}