/* Le programme "banque"                                                 
   Synopsis :
     banque <numeroCompte> <typeOperation> [<montant>]
   Parametres :
     <numeroCompte> contient un numero de compte (compris entre 0 et 9)
     <typeOperation> contient 'L' pour Lire le compte <numeroCompte>
                              'M' pour Modifier le compte <numeroCompte> de
                                  <montant> (NB: pour crediter un compte,
                                  on met 'M' et un <montant> negatif
     <montant> n'est utilise que quand <typeOperation> vaut 'M'. Il correspond
       au montant a debiter
   Exemples :
     banque 5 L         affiche le solde du compte numéro 5
     banque 3 M 50.5    modifie en debitant le compte numero 3 de 50.5
*/

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define NOMFICHIER "./banque.dat"
#define NBCOMPTES 10

#define FATAL(msg) {                     \
   fprintf(stderr,"%s:%s %s:%d:%s\n",    \
           strerror(errno), msg,         \
           __FILE__, __LINE__, __func__);\
    abort();                             \
}

int main(int argc, char *argv[]){
  int numCompte;
  char typeOp;
  double montant;
  double solde;

  /* Declarations de variables a ajouter */
  int fd;
  int nbR, nbW;
  int i;
  int rc;

  /* Lecture des parametres */
  if ((argc < 3) || (argc>4)){
    fputs("USAGE : banque <numeroCompte> <typeOperation> [<montant>]\n",
	  stderr);
    return EXIT_FAILURE;
  }

  numCompte = atoi(argv[1]);
  if ((numCompte<0) || (numCompte>=NBCOMPTES)){
    fputs("USAGE : banque <numeroCompte> <typeOperation> [<montant>]\n",
	  stderr);
    fputs("<numCompte> doit etre compris entre 0 et 9 (inclus)\n",
	  stderr);
    return EXIT_FAILURE;
  }

  typeOp = toupper(argv[2][0]);
  if ((typeOp != 'L') && (typeOp != 'M')){
    fputs("USAGE : banque <numeroCompte> <typeOperation> [<montant>]\n",
	  stderr);
    fputs("<typeOperation> doit valoir 'L' ou 'M'\n",
	  stderr);
    return EXIT_FAILURE;
  }
      
  if (typeOp == 'M'){
    if (argc != 4){
      fputs("USAGE : banque <numeroCompte> <typeOperation> [<montant>]\n",
	    stderr);
      fputs("<montant> est obligatoire quand <typeOp> vaut 'M'\n",
	    stderr);
      return EXIT_FAILURE;
    }else{
      montant = atof(argv[3]);
    }
  }

  /* Ouvrir le fichier "compte.data" (si ce fichier n'existe pas, on
     le cree de sorte qu'il contienne NBCOMPTES comptes (sous la forme
     de NBCOMPTES soldes à 0)                                         */
  fd = open(NOMFICHIER, O_RDWR, 0);
  if (fd < 0){
    if (errno == ENOENT){
      /* Le fichier n'existe pas : on le cree donc */
      fd = open(NOMFICHIER, O_CREAT|O_RDWR, S_IRUSR|S_IWUSR );
      if (fd < 0){
	FATAL("open fichier");
      }
      /* On le remplit */
      solde = 0;
      for (i=0 ; i<NBCOMPTES ; i++){
	nbW = write(fd, &solde, sizeof(solde));
	if (nbW != sizeof(solde)){
	  FATAL("pb initialisation fichier");
	}
      }
    }else{
      FATAL("open fichier");
    }
  }

  /* On lit le solde du compte
   */
  rc = lseek(fd, numCompte*sizeof(double), SEEK_SET);
  if (rc < 0){
    FATAL("lseek");
  }
  nbR = read(fd, &solde, sizeof(solde));
  if (nbR != sizeof(double)){
    FATAL("read");
  }

  /* On l'affiche */
  printf("Solde initial du compte %d = %10.2f\n", numCompte, solde);

  /* On dort un peu (pour laisser le temps a d'autres processus de
     s'executer                                                   */
  sleep(5);

  /* On effectue l'operation */
  if (typeOp == 'L'){
    /* Nouvelle consultation du solde de ce compte dans le fichier */
    rc = lseek(fd, numCompte*sizeof(double), SEEK_SET);
    if (rc < 0){
      FATAL("lseek");
    }
    nbR = read(fd, &solde, sizeof(solde));
    if (nbR != sizeof(double)){
      FATAL("read");
    }
    /* Nouvel affichage */
    printf("Solde du compte %d = %10.2f\n", numCompte, solde);
  }else{
    /* Debit */
    solde -= montant;
    /* On ecrit le nouveau solde dans le fichier */
    rc = lseek(fd, numCompte*sizeof(double), SEEK_SET);
    if (rc < 0){
      FATAL("lseek");
    }
    nbW = write(fd, &solde, sizeof(double));
    if (nbW != sizeof(solde)){
      FATAL("pb write solde");
    }
  }

  /* On referme le fichier */
  rc = close(fd);
  if (rc < 0){
    FATAL("close");
  }

  return EXIT_SUCCESS;
}