workbench/prefs/reqtools/args.c

   1 /*
   2     Copyright © 2013, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <dos/rdargs.h>
   7 #include <workbench/startup.h>
   8
   9 #include <proto/dos.h>
  10 #include <proto/icon.h>
  11
  12 #include "args.h"
  13 #include "misc.h"
  14
  15 /*** Global Variables *******************************************************/
  16 STATIC CONST_STRPTR   TEMPLATE = "FROM,USE/S,SAVE/S";
  17 STATIC IPTR           args[COUNT];
  18 STATIC struct RDArgs *rdargs;
  19 STATIC BPTR           olddir = (BPTR)-1;
  20
  21 /*** Functions **************************************************************/
  22 BOOL ReadArguments(int argc, char **argv)
  23 {
  24     if (argc)
  25     {
  26         rdargs = ReadArgs(TEMPLATE, args, NULL);
  27         return rdargs != NULL;
  28     }
  29     else
  30     {
  31         struct WBStartup *wbmsg = (struct WBStartup *)argv;
  32         struct WBArg *wbarg = wbmsg->sm_ArgList;
  33         struct DiskObject *dobj;
  34         STRPTR *toolarray;
  35         STRPTR tooltype;
  36
  37         if (wbmsg->sm_NumArgs > 1)
  38         {
  39             wbarg++;
  40             if (wbarg->wa_Lock && *wbarg->wa_Name)
  41             {
  42                 olddir = CurrentDir(wbarg->wa_Lock);
  43                 dobj = GetDiskObject(wbarg->wa_Name);
  44                 if (dobj)
  45                 {
  46                     args[FROM] = (IPTR)wbarg->wa_Name;
  47                     toolarray = dobj->do_ToolTypes;
  48                     tooltype = FindToolType(toolarray, "ACTION");
  49                     if (tooltype)
  50                     {
  51                         if (MatchToolValue(tooltype, "USE"))
  52                         {
  53                             args[USE] = TRUE;
  54                         }
  55                         else if (MatchToolValue(tooltype, "SAVE"))
  56                         {
  57                             args[SAVE] = TRUE;
  58                         }
  59                     }
  60                     FreeDiskObject(dobj);
  61                     return TRUE;
  62                 }
  63                 else
  64                 {
  65                     ShowMessage("Couldn't read DiskObject\n");
  66                 }
  67             }
  68         }
  69         else if (wbmsg->sm_NumArgs == 1)
  70         {
  71             return TRUE;
  72         }
  73     }
  74     return FALSE;
  75 }
  76
  77 VOID FreeArguments(VOID)
  78 {
  79     FreeArgs(rdargs);
  80     if (olddir != (BPTR)-1)
  81         CurrentDir(olddir);
  82 }
  83
  84 IPTR GetArgument(enum Argument id)
  85 {
  86     if (id >= 0 && id < COUNT) return args[id];
  87     else                       return 0;
  88 }