//////////////////////////////////////////////////////////////////////////////
//  Module Name:  iv.cpp
//  Programmer:   Alex Martinov, 2:5030/601.11fidonet.org
//  Description:  Increase version utility
//  Date:         15-06-00
//  Last Change:  15-06-00
//////////////////////////////////////////////////////////////////////////////

#include <stdio.h>
#include <stdlib.h>
#include <io.h>
#include <conio.h>
#include <fcntl.h>
#include <sys\stat.h>
#include <string.h>
#include <errno.h>
#include <dir.h>
#include <ctype.h>


#define BUFFER_SIZE 4096


unsigned char buf[BUFFER_SIZE];


int op_q = 0;
int help = 0;
int setval = 0;
int backup = 0;
long valforset = 0;


char *files[2] = { NULL, NULL };

unsigned char StringBuf[512];
char TempName[MAXPATH];



void print_help(void);
void print_copyr(void);





int main (int argc, char* argv[])
{
 char *p;
 int fc = 0;

 int ret_zero = 0;

 strcpy(StringBuf,"");
 strcpy(TempName,"$$IVXXXXXX");

 int i;
 // read command-line configuration
 for (i=1; i<argc; i++)
     {
      p=argv[i];
      if (*p=='/' || *p=='-')
         {
          p++;
          switch(*p){
            case 'h':
            case '?': help = 1; break;
            case 'q': op_q = 1; break;
            case 's': strcpy(StringBuf,(++p)--);
                      break;
            case 'v': valforset = atol((++p)--);
                      setval = 1;
                      break;
            case 'b': backup = 1;
                      break;
        case 'z': ret_zero = 1;
                      break;
            default:  print_copyr();
                      printf("Invalid key: %s\n",--p);
                      print_help();
                      return 0;
          };
         }
      else
         {
          if (fc<2)
             files[fc++] = argv[i];
         }
     }

 if (help)
    {
     print_copyr();
     print_help();
     return 0;
    }

 if (!op_q)
    {
     print_copyr();
    }

 if (files[0]==NULL)
    {
     if (op_q) print_copyr();
     printf("Missing input file name\n");
     print_help();
     return 0;
    }

 if (strlen(StringBuf)==0)
    {
     if (op_q) print_copyr();
     printf("Missing increasing string\n");
     print_help();
     return 0;
    }


 int handle = open(files[0],O_BINARY|O_RDONLY);
 if (handle<0)
    {
     if (op_q) print_copyr();
     printf("Can't open input file: %s\n",files[0]);
     switch(errno)
        {
         case ENOENT: printf("%s\n","No such file or directory"); break;
         case EMFILE: printf("%s\n","Too many open files"); break;
         case EACCES: printf("%s\n","Permission denied"); break;
         case EINVACC: printf("%s\n","Invalid access code"); break;
        };
     return 0;
    }

 mktemp(TempName);
 int outhandle;

 if (!strcmp(files[0],files[1])) // source == destination
    {
     files[1] = NULL;
    }

 if (files[1]==NULL)
    outhandle = open(TempName,O_RDWR|O_CREAT|O_BINARY,S_IREAD|S_IWRITE);
 else
    outhandle = open(files[1],O_RDWR|O_CREAT|O_BINARY,S_IREAD|S_IWRITE);

 if (outhandle<0)
    {
     if (op_q) print_copyr();
     printf("Can't open output file: %s\n",(files[1]==NULL ? TempName : files[1]));
     switch(errno)
        {
         case ENOENT: printf("%s\n","No such file or directory"); break;
         case EMFILE: printf("%s\n","Too many open files"); break;
         case EACCES: printf("%s\n","Permission denied"); break;
         case EINVACC: printf("%s\n","Invalid access code"); break;
        };
     close(handle);
     return 0;
    }

 FILE *fin = fdopen(handle, "rb");
 if (fin==NULL)
    {
     if (op_q) print_copyr();
     printf("fdopen failed\n");
    }


 char *ptr = fgets((char*)buf, BUFFER_SIZE, fin);
 char *end_var;
 char numbuf[64];
 long int val;
 int occurence = 0;
 char wschar = ' ';
 while(ptr!=NULL)
      {
       end_var = strstr((char*)buf, (char*)StringBuf);
       if (end_var!=NULL)
          {
           occurence++;
           end_var += strlen(StringBuf);
           int bsize = end_var - buf;
           write(outhandle,buf,bsize); // save the begin of string

           if (isspace(*end_var) || !*end_var)
              {
               write(outhandle,&wschar,1); //optionaly add whitespaces
               if (*end_var) end_var++;
              }

           val = atol(end_var);
           while (*end_var && isdigit(*end_var)) end_var++; // scan for number end
           if (setval)
              {
               val = valforset;
              }
           else
              {
               val++; // this line is most significant
              }

           ltoa(val, numbuf, 10); // convert new num to string
           write(outhandle,numbuf,strlen(numbuf)); // writes it to file
           write(outhandle,end_var,strlen(end_var)); // write end to file
          }
       else
          {
           write(outhandle,buf,strlen(buf));
          }
       ptr = fgets((char*)buf, BUFFER_SIZE, fin);
      }
 fclose(fin);
 close(outhandle);

 if (!op_q)
    {
     printf("Found %d occurence of string '%s' in file '%s'\n",occurence,StringBuf,files[0]);
    }


 if (files[1]==NULL)
    { // if source == destination
     if (backup)
        {
         char *point;
         char backupbuf[MAXPATH];
         strcpy(backupbuf,files[0]);
         point = strrchr(backupbuf,'.');
         if (point!=NULL) // found extention delimiter
            {
             *point = 0;
             strcat(backupbuf,".bak"); // generate backup file name
             unlink(backupbuf); // remove prev bak
             rename(files[0],backupbuf); // create backup copy
            }
        }
     else
        {
         unlink(files[0]);  //remove source
        }
     rename(TempName, files[0]);  // rename temporary file
     unlink(TempName); // remove temp
    }



 if (ret_zero) return 0;
 return val;
}



void print_copyr(void)
{
 char *prgname_str = "Increase Version ver. 2.0\n";
 char *copyright_str = "(c)1999-2000 by Alex Martinov, 2:5030/601.11fidonet.org\n";
 printf("%s%s",prgname_str,copyright_str);
}


void print_help(void)
{
 char *h1 = " Usage: ver [options] input_file [output_file]\n\
     options:  -h,-?      ('/' is also allowed)\n\
                          prints this help screen.\n\
               -sstr      str - string for increase it's number.\n\
               -vN        force set value to N.\n\
           -b         create backup file (used only if source==destination).\n\
           -z         always return nul.\n\
               -q         operate quetly.\n";

 char *h2 = "     input_file  - file, which contain number for increase.\n\
     output_file - file for save results.\n";

 printf("%s%s",h1,h2);

}