/*
 *  WB2Argv v1.3, Amiga Workbench argv/argc emulation routines.
 *  Copyright (C) 1995-96 Jens T. Berger Thielemann
 *
 *  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.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 *  Contact the author at:
 *      Jens Berger
 *      Spektrumvn. 4
 *      N-0666 Oslo
 *      Norway
 *      E-mail: <jensthi@ifi.uio.no>
 *
 *  Note: If you use any of these routines in your programs, you have to
 *  mention so in the document.
 *
 *
 */

#ifdef AMIGA

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include <workbench/startup.h>
#include <proto/dos.h>
#include <workbench/workbench.h>
#include <proto/icon.h>
#include <proto/exec.h>

#include "WB2Argv.h"

/* #define WB_DEBUG */

#ifdef WB_DEBUG
char __stdiowin[] = "CON:0/10/640/180/WB2Argv";
char __stdiov37[] = "/AUTO/CLOSE/WAIT";
#endif

static char *ExpandLockName(BPTR Lock, STRPTR Name);

/*
 * Converts a message from workbench into an argv look-a-like. Tooltypes
 * are converted to long-named options, suitable for passing to GNU's
 * getopt_long(). Shift-clicked files will be appended at the end; their
 * tooltypes will _currently_ not be processed. This may happen in a future
 * version (is it desirable?).
 *
 * Flags: Currently these do only control whether we'll convert the
 * case of the options. See WB2Argv.h for options.
 *
 * Returns an array of string-pointers; terminated with a NULL. To get
 * argc, call CountArgv(). Returns NULL if out of memory, or not running
 * v2.04.
 *
 */


char **WB2Argv(struct WBStartup *WBMsg, LONGBITS Flags)
{
    ULONG   c, CurArg, NewArgc;
    char    **NewArgv, **Tools, *Buf, *Orig;
    BOOL    TmpSucc = TRUE, IconOpen = FALSE;
    struct DiskObject *dskobj = NULL;
    BPTR    OldCurDir = 0L;

    if(WBMsg && (DOSBase->dl_lib.lib_Version > 36))
    {
        /*
         * First, count the # of argv entries. We get both from
         * shift-clicked icons + tooltypes.
         */

        NewArgc = WBMsg->sm_NumArgs;

        if(IconBase = OpenLibrary("icon.library", 0))
        {
            OldCurDir = CurrentDir(WBMsg->sm_ArgList->wa_Lock);
            IconOpen = TRUE;
            if(dskobj = GetDiskObject(WBMsg->sm_ArgList->wa_Name))
            {
                Tools = dskobj->do_ToolTypes;

                while(*Tools++)
                    NewArgc++;
            }
        }

        /*
         * Stuff it together...
         */

        if(NewArgv = calloc((size_t) NewArgc + 2, sizeof(char *)))
        {
            CurArg = 0;

            /* Program name */

            TmpSucc = FALSE;

            if(Buf = ExpandLockName(WBMsg->sm_ArgList->wa_Lock,
                                    WBMsg->sm_ArgList->wa_Name))
            {
                NewArgv[CurArg++] = Buf;
                TmpSucc = TRUE;
            }

            if(dskobj && TmpSucc)
            {
                /* Tooltypes becomes longnamed options */
                for(Tools = dskobj->do_ToolTypes;
                    (Orig = *Tools) && TmpSucc;
                    Tools++)
                {
                    Orig = stpblk(Orig);

                    /* To avoid a false `--' */
                    if(*Orig && (*Orig != '(')) /* ) */
                    {
                        TmpSucc = FALSE;

                        if(Buf = malloc(strlen(Orig) + 4))
                        {
                            strcpy(Buf, "--");
                            strcat(Buf, Orig);

                            TmpSucc = TRUE;
                            NewArgv[CurArg++] = Buf;

                            /* Make option lowercase */

                            switch(Flags & W2A_CASEMASK)
                            {
                            case W2A_LOWER:
                                for(;
                                    *Buf && (*Buf != '=');
                                    Buf++)
                                    *Buf = tolower(*Buf);
                                break;
                            case W2A_UPPER:
                                for(;
                                    *Buf && (*Buf != '=');
                                    Buf++)
                                    *Buf = toupper(*Buf);
                                break;
                            case W2A_KEEPCASE:
                                break;
                            }
                        }
                    }
                }
            }

            /* End of options... */
            NewArgv[CurArg++] = "--";

            if(TmpSucc)
            {
                for(c = 1;          /* Skip program name */
                   (c < WBMsg->sm_NumArgs) && TmpSucc;
                    c++)
                {
                    TmpSucc = FALSE;

                    if(Buf = ExpandLockName(WBMsg->sm_ArgList[c].wa_Lock,
                                            WBMsg->sm_ArgList[c].wa_Name))
                    {
                        NewArgv[CurArg++] = Buf;
                        TmpSucc = TRUE;
                    }
                }
            }
        }

        if(dskobj)
            FreeDiskObject(dskobj);

        if(OldCurDir)
            CurrentDir(OldCurDir);

        if(IconOpen)
            CloseLibrary(IconBase);

        if(TmpSucc)
            return(NewArgv);
    }

    return(NULL);
}

ULONG CountArgv(const char **argv)
{
    ULONG argc = 0;

    if(argv)
    {
        while(*argv++)
            argc++;
    }
    return(argc);
}

static char *ExpandLockName(BPTR Lock, STRPTR Name)
{
    UBYTE   Buffer[BUFSIZ];
    char    *Retval = NULL;

    if(NameFromLock(Lock, Buffer, BUFSIZ))
    {
        if(AddPart(Buffer, Name, BUFSIZ))
            Retval = strdup(Buffer);
    }

    return(Retval);
}

#ifdef WB_DEBUG
#  define EXIT_FAILURE 20
void main(int argc, char **argv)
{
    int c;

#ifdef AMIGA
    if(_WBenchMsg)
    {
        if(argv = WB2Argv(_WBenchMsg, W2A_LOWER))
            argc = CountArgv(argv);
        else
            exit(EXIT_FAILURE);
    }
#endif

    for(c = 0; c < argc; c++)
        printf("%s\n", argv[c]);

}

#endif /* WB_DEBUG */

#endif /* AMIGA */