#include "fd.h"
#include "stralloc.h"
#include "readwrite.h"
#include "sgetopt.h"
#include "wait.h"
#include "env.h"
#include "byte.h"
#include "str.h"
#include "alloc.h"
#include "exit.h"
#include "fork.h"
#include "case.h"
#include "subfd.h"
#include "error.h"
#include "substdio.h"
#include "sig.h"

void die(e,s) int e; char *s; { substdio_putsflush(subfderr,s); _exit(e); }
void die_usage() { die(100,"qsmhook: fatal: incorrect usage\n"); }
void die_temp() { die(111,"qsmhook: fatal: temporary problem\n"); }
void die_read() { die(111,"qsmhook: fatal: unable to read message\n"); }
void die_badcmd() { die(100,"qsmhook: fatal: command not found\n"); }

int flagrpline = 0; char *rpline;
int flagufline = 1; char *ufline;
int flagdtline = 0; char *dtline;
char *host;
char *sender;
char *recip;

stralloc newarg = {0};

substdio ssout;
char outbuf[SUBSTDIO_OUTSIZE];
substdio ssin;
char inbuf[SUBSTDIO_INSIZE];

void main(argc,argv)
int argc;
char **argv;
{
 int pid;
 int wstat;
 int pi[2];
 int opt;
 char **arg;
 char *x;
 int i;
 int flagesc;

 sig_pipeignore();

 if (!(dtline = env_get("DTLINE"))) die_usage();
 if (!(rpline = env_get("RPLINE"))) die_usage();
 if (!(ufline = env_get("UFLINE"))) die_usage();
 if (!(recip = env_get("LOCAL"))) die_usage();
 if (!(host = env_get("HOST"))) die_usage();
 if (!(sender = env_get("SENDER"))) die_usage();

 while ((opt = getopt(argc,argv,"DFlMmnPsx:")) != opteof)
   switch(opt)
    {
     case 'D': case 'F': case 'M': break; /* be serious */
     case 'l': flagdtline = 1; break; /* also return-receipt-to? blech */
     case 'm': break; /* we only handle one recipient anyway */
     case 'n': flagufline = 0; break;
     case 's': break; /* could call quote() otherwise, i suppose... */
     case 'P': flagrpline = 1; break;
     case 'x':
       if (case_starts(recip,optarg))
	 recip += str_len(optarg);
       break;
     default:
       _exit(100);
    }
 argc -= optind;
 argv += optind;

 if (!*argv) die_usage();

 for (arg = argv;x = *arg;++arg)
  {
   if (!stralloc_copys(&newarg,"")) die_temp();
   flagesc = 0;
   for (i = 0;x[i];++i)
     if (flagesc)
      {
       switch(x[i])
	{
	 case '%': if (!stralloc_cats(&newarg,"%")) die_temp(); break;
	 case 'g': if (!stralloc_cats(&newarg,sender)) die_temp(); break;
	 case 'h': if (!stralloc_cats(&newarg,host)) die_temp(); break;
	 case 'u': if (!stralloc_cats(&newarg,recip)) die_temp(); break;
	}
       flagesc = 0;
      }
     else
       if (x[i] == '%')
	 flagesc = 1;
       else
	 if (!stralloc_append(&newarg,&x[i])) die_temp();
   if (!stralloc_0(&newarg)) die_temp();
   i = str_len(newarg.s) + 1;
   if (!(x = alloc(i))) die_temp();
   byte_copy(x,i,newarg.s);
   *arg = x;
  }

 if (pipe(pi) == -1) die_temp();

 switch(pid = fork())
  {
   case -1:
     die_temp();
   case 0:
     close(pi[1]);
     if (fd_move(0,pi[0]) == -1) die_temp();
     sig_pipedefault();
     execvp(*argv,argv);
     if (error_temp(errno)) die_temp();
     die_badcmd();
  }
 close(pi[0]);

 substdio_fdbuf(&ssout,write,pi[1],outbuf,sizeof(outbuf));
 substdio_fdbuf(&ssin,read,0,inbuf,sizeof(inbuf));
 if (flagufline) substdio_bputs(&ssout,ufline);
 if (flagrpline) substdio_bputs(&ssout,rpline);
 if (flagdtline) substdio_bputs(&ssout,dtline);
 if (substdio_copy(&ssout,&ssin) == -2) die_read();
 substdio_flush(&ssout);
 close(pi[1]);

 if (wait_pid(&wstat,pid) == -1) die_temp();
 if (wait_crashed(wstat)) die_temp();
 _exit(wait_exitcode(wstat));
}