//
//  Compile with Borland C/C++ v3.1 : bcc -v- -ml cl386.c
//  Compile with Watcom C/C++ v10.6 : wcl -y -ox -ml cl386.c
//
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dos.h>
#include <process.h>
#ifdef __TURBOC__
  #include <dir.h>
#else
  #include <direct.h>
#endif

#define VERSION "1.0"

static char usage[] = /* Usage */ \
"MK386 Version " VERSION " (c) Kirill Joss. Compile " __DATE__ "\n"
"Usage: MK386.EXE [options] files\n"
"Options:       (default parameters if first)\n"
"/l[-|+]  dump listing file        /w-xxx   disable a warning       \n"
"/i[-|+]  dump preprocessed file   /Dxxx    define something        \n"
"/e[+|-]  dump errors to file      /Ennn    max number of errors    \n"
"/k[+|-]  keep generate files      /Ipath   specify include path    \n"
"/K[-|+]  keep response files      /Lpath   specify .LIB & .OBJ dir \n"
"/A[+|-]  disable extensions       /Cxxx    codegen parameters      \n"
"/v[-|+]  debug info               /Oxxx    optimzer parameters     \n"
"/m[-|+]  generate .MAP file       /Tp      use TLINK & PMODE       \n"
"/n[-|+]  no default .LIB & .OBJ   /Tl      use TLINK & PMODE(low)  \n"
"/M[+|-]  run MAKE                 /Tw      use WLINK & PMODE/W     \n"
"/B[-|+]  build all                /Td      use WLINK & DOS/4GW     \n"
"/a       generate .ASM file       /c       generate .OBJ file      \n"
"/ename   set name of .EXE file    /opath   set out path .OBJ files \n"
"/?,/h,/H this help                @fname   specify response file   \n";


#define FALSE 0
#define TRUE  1

#define MAX_STR 256 /* Max len of string in .CFG file */

// is param ?
#define isparam(s) ( *(s) == '-' || *(s) == '/' || *(s) == '+' )
// check for NOT NULL !!!
#define chkNULL(exp, msg )  if ( !(exp) ) error( ERR_FATAL, msg )
// empty string ?
#define empty_string(s)  ( (s) == NULL || *(s) == '\0' )

typedef struct _Item {
    struct _Item *next;
    char data[1];
} Item;

typedef struct _List {
    Item *head;
    Item *tail;
    unsigned count;
} List;

Item *ItemCreate( char *dat );
#define ItemDelete(item) free(item)

void ListAdd( List *list, char *data );
void ListDelete( List *list );
int  ListFind( List *list, char *data );
void ListIter( List *list, void (*iter)( Item *) );

#define FileAdd(list, file) ListAdd((list),(file))
#define ParamAdd(list, param) ListAdd((list),(param));

List files; // list of all files

List l_inc; // list of INCLUDE path
List l_def; // list of DEFINE var
List l_cc386; // list of CC386 options

int _compile  = TRUE;
int _assemble = TRUE;
int _link     = TRUE;

int _dump_cpp = FALSE; // dump preprocessed file
int _dump_err =  TRUE; // dump error file
int _dump_lst = FALSE; // dump listing file
int _ansi     =  TRUE; // ANSI

int _map_file = FALSE; // map file
int _nodef_lib= FALSE; // no default lib
int _debug    = FALSE; // debug info

int _keep_rsp = FALSE; // keep response files
int _keep_gen = TRUE;  // keep generate files

int _make_run = TRUE;
int _build_all = FALSE;

int _output = 'p';     // Output format, same /Tx, can be 'p', 'l', 'w', 'd'

char path_lib[ _MAX_PATH ]; // path for default LIB & OBJ
char path_obj[ _MAX_PATH ]; // path for output .OBJ files
char name_exe[ _MAX_PATH ]; // name of .EXE file

 // For error()
#define ERR_NO        0  /* no error ! */
#define ERR_FATAL   128  /* fatal error */
#define ERR_RUNTIME 129  /* stdlib error - see 'errno' */

 // output errmsg & exit()
void error( int exitcode, char *msg );
 // output help & exit(0)
void help( void );
 // get parameters from <name>.CFG, CL386, cmdline ...
void get_param( char *exe_name, char *env_var, int argc, char *argv[] );
 // get parameters from file
int get_param_file( char *file );
 // get parameters from string
void get_param_str( char *str );
 // check ONE parameter
void get_one_param( char *param );
 // get ONE options
void get_option( char *param );
 // get ONE file ( can '*' & '?' )
void get_source( char *filename );
 // get ONE file without '*' & '?'
void get_one_source( char *filename );
 // find source files
void find_source( char *fpath, char *fname );
 // clear all list
void cleanup( void );
 // generate MAKEFILE.
void gen_makefile( void );
 // run overlay MAKE
void make( char *maker );


void main( int argc, char *argv[] ) {

    if ( argc == 1 ) // no argument
        help();
    get_param( argv[0], "CL386", argc - 1, argv + 1 );
    atexit( cleanup );  // set exit function
    if ( files.count == 0 ) // no files
        help();
    gen_makefile();
    cleanup();
    if ( _make_run )
        make( "MAKE.EXE" );
}


void help( void ) {
    printf( usage ); exit( 0 );
}


static char _file[ _MAX_PATH  ];
static char fdisk[ _MAX_DRIVE ];
static char fpath[ _MAX_DIR   ];
static char fname[ _MAX_FNAME ];
static char fext [ _MAX_EXT   ];


void get_param( char *exe_name, char *env_var, int argc, char *argv[] ) {
    static char _cfg[] = ".CFG";
    int i;
    char *env;

    _splitpath( exe_name, fdisk, fpath, fname, fext );
    _makepath( _file, NULL, NULL, fname, _cfg );
    if ( ! get_param_file( _file ) ) {
        _makepath( _file, fdisk, fpath, fname, _cfg );
        get_param_file( _file );
    }

    if ( (env = getenv( env_var ) ) != NULL )
        get_param_str( env );

    for ( i = 0; i < argc; i++ )
        get_one_param( argv[ i ] );
}


int get_param_file( char *file ) {
    static char str[ MAX_STR ];
    FILE *f = fopen( file, "rt" );

    if ( f == NULL )
        return FALSE;

    while( fgets( str, MAX_STR - 1, f) != NULL )
        get_param_str( str );

    fclose( f );
    return TRUE;
}


void get_param_str( char *s ) {
    char arg[ MAX_STR ];
    int  len;

    for ( len = 0; *s; s++ ) {
        if ( *s == ' ' || *s == '\t' || *s == '\n' ) {
            if ( len > 0 ) {
                arg[ len ] = '\0';
                get_one_param( arg );
            }
            arg[ len = 0 ] = '\0';
        }
        else {
            arg[ len++ ] = *s;
        }
    }

    if ( len > 0 ) {
        arg[ len ] = '\0';
        get_one_param( arg );
    }
}


void get_one_param( char *arg ) {
    if ( isparam( arg ) )                 // -X , +X or /X
        get_option( arg );
    else if ( *arg == '@' ) {               // @filename
        if ( ! get_param_file( arg + 1 )  )
            error( ERR_RUNTIME, arg + 1 );
    }
    else
        get_source( arg );             // file.ext
}


void error( int exitcode, char *msg ) {
    switch ( exitcode ) {
        case ERR_NO: break;

        case ERR_FATAL: fprintf( stderr, "Fatal : %s\n", msg );
            break;

        case ERR_RUNTIME: fprintf( stderr, "%s : %s", msg, strerror( errno ) );
            break;

        default: fprintf( stderr, "Program error : %s !\n", msg );
            break;
    }
    exit( exitcode );
}


static int setop( char *param, int *option ) {
    if ( param[2] == '-' && param[3] == '\0' )
        *option = FALSE;
    else if ( ( param[2] == '+' && param[3] == '\0' ) || param[2] == '\0' )
        *option = TRUE;
    else
        return FALSE;
    return TRUE;
}


void get_option( char *param ) {
    int len;

    switch( param[ 1 ] ) {

        case 'i' : // dump preprocessed file
            if ( setop( param, &_dump_cpp ) )
                return;
            break;

        case 'e' : // dump errors to file
            if ( setop( param, &_dump_err ) )
                return;
            else
                strupr( strncpy( name_exe, param + 2, _MAX_PATH ) );
            break;

         case 'l' : // dump listing file
            if ( setop( param, &_dump_lst ) )
                return;
            break;

        case 'A' : // disable extensions
            if ( setop( param, &_ansi ) )
                return;
            break;

        case 'm' : // gen map file
            if ( setop( param, &_map_file ) )
                return;
            break;

        case 'n' : // no default lib
            if ( setop( param, &_nodef_lib ) )
                return;
            break;

        case 'v' : // debug info
            if ( setop( param, &_debug ) )
                return;
            break;

        case 'K' : // keep response files
            if ( setop( param, &_keep_rsp ) )
                return;
            break;

        case 'k' : // keep generate files
            if ( setop( param, &_keep_gen ) )
                return;
            break;

        case 'B' : // building all
            if ( setop( param, &_build_all ) )
                return;
            break;

        case 'M' : // run make, else ONLY generate MAKEFILE.
            if ( setop( param, &_make_run ) )
                return;
            break;

        case 'a' :
            if ( param[2] == '\0' ) {
                _compile = TRUE;
                _assemble = _link = FALSE;
                return;
            }
            break;

        case 'c' :
            if ( param[2] == '\0' ) {
                _compile = _assemble = TRUE;
                _link = FALSE;
                return;
            }
            break;

        case '?' :
        case 'h' :
        case 'H' :
            help(); /* Not return ! */

        case 'D' : // define
            ParamAdd( &l_def, param + 2 );
            return;

        case 'I' : // include path
            ParamAdd( &l_inc, param + 2 );
            return;

        case 'L' : // lib path
            strupr( strncpy( path_lib, param + 2, _MAX_PATH ) );
            len = strlen( path_lib );
            if ( len > 0 && path_lib[ len - 1 ] == '\\' )
                path_lib[ len - 1 ] = '\0';
            return;

        case 'o' : // output obj path
            strupr( strncpy( path_obj, param + 2, _MAX_PATH ) );
            len = strlen( path_obj );
            if ( len > 0 && path_obj[ len - 1 ] == '\\' )
                path_obj[ len - 1 ] = '\0';
            return;

        case 'w' :
        case 'C' :
        case 'E' :
        case 'O' :
            ParamAdd( &l_cc386, param );
            return;

        case 'T' :
            if ( param[3] == '\0' )
                if ( param[2] == 'p' || param[2] == 'l' || \
                param[2] == 'w' || param[2] == 'd' ) {
                    _output = param[2]; return;
                }
            break;

    }
    fprintf( stderr, "Warning! Parameter error : %s\n", param );
}


Item *ItemCreate( char *dat ) {
    Item *item;
    int len;

    if ( empty_string( dat ) )
        return NULL;

    len = strlen( dat );

    if ( (item = (Item *)malloc( sizeof( Item ) + len )) != NULL ) {
        item->next = NULL;
        strcpy( item->data, dat );
    }

    return item;
}


void ListDelete( List *list ) {
    Item *p, *q;

    chkNULL( list, "Pointer is NULL !" );

    for( p = list->head; p; p = q ) {
        q = p->next;
        free(p);
    }
    list->head = list->tail = NULL;
    list->count = 0;
}


void ListAdd( List *list, char *file ) {
    Item *item;

    chkNULL( list, "Pointer is NULL !" );

    item = ItemCreate( file );