/***************************************************************************
                          zeta_main.cpp  -  description
                             -------------------
    begin                : Mon Mar 17 05:35:56 2003
    copyright            : (C) 2002 by Cavalli Andrea
    email                : cavalli@bioc.unizh.ch
**************************************************************************/
/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#include <getopt.h>
#include <iostream>
#include <config.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <zeta/zargs.h>
#include <signal.h>
#include <stack>
#include <zeta/zobj.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#ifdef ALM_MPI
#include <mpi/almostmpi.h>
Almost::MPI * Almost::MPI::mpi_ = NULL;   
#endif

using namespace std;
extern int interactive_mode;
extern int check_mode;
extern FILE * zeta_in;
extern ZArgs args_;
extern "C" void restart();
extern int interrupt;


//extern stack<string> file_name;
//extern YY_BUFFER_STATE *include_stack;


extern "C" {
  void init_();
  void init_modules();
  void zeta_parse();
  void finalize_();
  void push_file_name(const char * name);
  void pop_file_name();
}



void zeta_term_(int){
  Almost::aout<<flush<<"\nkeyboard interrupt\n>>> "<<flush;
  interrupt = 1;
  restart();
}



void print_banner();


void set_bin_name(char * argv){
  string a = argv;
  int pos = 0;
  pos = a.find_last_of("/");
  if(pos==string::npos) ZArgs::set_bin(a);
  else {
    ZArgs::set_bin(string(a.begin()+pos+1,a.end()));
  }
}

void open_file(string file){
  FILE * f = fopen(file.c_str(),"r");
  if(f==NULL){
    Almost::aout<<"can't open file: "<<file<<endl;
    exit(-1);
  }
  push_file_name(file.c_str());
  zeta_in = f;
}

int nativeSelect(int timeout, int socketDescriptor, bool selectForRead)
{
    fd_set fds;
    FD_ZERO(&fds);
    FD_SET(socketDescriptor, &fds);

    struct timeval tv;
    tv.tv_sec = timeout / 1000;
    tv.tv_usec = (timeout % 1000) * 1000;

    if (selectForRead)
        return select(socketDescriptor + 1, &fds, 0, 0, timeout < 0 ? 0 : &tv);
    else
        return select(socketDescriptor + 1, 0, &fds, 0, timeout < 0 ? 0 : &tv);
}


void do_server(){
  Almost::aout<<"Starting Almost Server"<<endl;
  int serverfd;
  int clientfd;
  sockaddr_in server_addr;
  sockaddr_in client_addr;

  serverfd = socket(AF_INET,SOCK_STREAM,0);
  
  server_addr.sin_family = AF_INET;
  server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
  server_addr.sin_port = htons(30869);
  bind(serverfd,(sockaddr*)&server_addr,sizeof(server_addr));
  listen(serverfd,5);

  signal(SIGCHLD,SIG_IGN);
  //  close(0);
  //loop
  interactive_mode = 0;
  int client_len = sizeof(client_addr);
  close(0);
  clientfd = accept(serverfd,(sockaddr*)&server_addr,(socklen_t*)&client_len);
  close(1);
  dup(clientfd);
  time_t t = time(NULL);
  printf("almost server accepted connection on %s:%d at %s\n",
	 inet_ntoa(server_addr.sin_addr),
	 ntohs(server_addr.sin_port),ctime(&t));
  fprintf(stderr,"almost server accepted connection on %s:%d at %s\n",
	  inet_ntoa(server_addr.sin_addr),
	  ntohs(server_addr.sin_port),ctime(&t));

  fflush(stdin);
  while(1){
    //Avaible ?
    int nbytes = 0;
    int available = 0;
    while(1){
    int ret = nativeSelect(1000,clientfd,true);
    if(ret!=0) break;
    }
    if (::ioctl(clientfd, FIONREAD, (char *) &nbytes) >= 0)
        available = (int) *((int *) &nbytes);
    if(available==0){
      cerr<<"Connection reset by peer\n";
      exit(0);
    }
    char *buff = new char[available+1];
    int nread = 0;

    nread = read(clientfd,buff,available);
    if(nread==0) break;
    buff[nread] = '\0';
    //    Almost::aout<<buff<<endl;

    FILE *q = fopen("tmp","w+");
    fprintf(q,"%s",buff);
    fclose(q);
    delete [] buff;
    open_file("tmp");
    zeta_parse();
    finalize_();
    //    exit(0);
  }
}

void print_banner(){
  cerr<<"almost "<<VERSION<<" ";
#if defined __GNUC__
  cerr<<"GNU C++ version "<< __VERSION__<<"\n";
#endif
#if defined __ICC
  cerr<<"Intel C++ version "<<__ICC<<"\n";
#endif
  cerr<<"Copyright (c) 2001-2004, Andrea Cavalli.\n";
  cerr<<"This program is free software; you can redistribute it and/or\n"
      <<"modify it under the terms of the GNU General Public License\n"
      <<"as published by the Free Software Foundation; either version 2\n"
      <<"of the License, or (at your option) any later version.\n";
}






int main(int argc, char ** argv){


  string file;
  bool server=false;
  bool check =false;
  int c;
  int option_index = 0;
  static option long_options[] = {
    {"interavtive",0,0,'i'},
    {"file",1,0,'f'},
    {"server",0,0,'s'},
    {"check",0,0,'c'}
  };

  
  //install sighandler
  //  siginterrupt(SIGINT, 1);

#ifdef ALM_MPI
  Almost::MPI mpi(&argc,&argv);
  Almost::aout<<"\t>> Starting MPI version"<<endl;
  Almost::aout<<"\t>> Rank "<<mpi.rank()<<endl;
#endif

  string opt;
  while(1){
    c = getopt_long (argc, argv, "if:sc",
		     long_options, &option_index);

    if (c == -1)
      break;

    switch(c){
    case 'i':
      interactive_mode = 1;
      break;
    case 'f':
      file = optarg;
      break;
    case 's':
      server = true;
      break;
    case 'c':
      check = true;
      break;
    }
  }



  //  GC_dont_gc = 1;//until it works ....
  init_();
  init_modules();
  ZObj::set_collectable(true);
  //install sighandler
  //  siginterrupt(SIGINT, 1);





  //not getopt options
  //  Almost::aout<<"Adding args "<<argv[0]<<endl;
  ZArgs::init();
  ZArgs::add_arg(argv[0]);
  for(int i=optind;i<argc;i++){
    //Almost::aout<<"Adding args "<<argv[0]<<endl;
    ZArgs::add_arg(argv[i]);
  }
  set_bin_name(argv[0]);
  //  Almost::aout<<"Argc "<<ZArgs::argc()<<endl;


  do_server();

}