workbench/c/shellcommands/Set.c

   1 /*
   2     Copyright © 1995-2004, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Set CLI command.
   6 */
   7
   8 /*****************************************************************************
   9
  10     NAME
  11
  12         Set
  13
  14     SYNOPSIS
  15
  16         NAME,STRING/F
  17
  18     LOCATION
  19
  20         Workbench:c
  21
  22     FUNCTION
  23
  24         Set a local environment variable in the current shell. If any global
  25         variables have the same name the local variable will be used instead.
  26
  27         This instance the variable is only accessible from within the shell
  28         it was defined.
  29
  30         If no parameters are specified, the current list of local variables
  31         are displayed.
  32
  33     INPUTS
  34
  35         NAME    - The name of the local variable to set.
  36
  37         STRING  - The value of the local variable NAME.
  38
  39     RESULT
  40
  41         Standard DOS error codes.
  42
  43     NOTES
  44
  45     EXAMPLE
  46
  47         Set Jump 5
  48
  49             Sets a local variable called "Jump" to the value of "5".
  50
  51     BUGS
  52
  53     SEE ALSO
  54
  55         Get, Unset
  56
  57     INTERNALS
  58
  59 ******************************************************************************/
  60
  61
  62 #include <proto/dos.h>
  63 #include <proto/exec.h>
  64
  65 #include <dos/dos.h>
  66 #include <dos/dosextens.h>
  67 #include <dos/rdargs.h>
  68 #include <dos/var.h>
  69 #include <exec/lists.h>
  70 #include <exec/nodes.h>
  71 #include <exec/types.h>
  72 #include <aros/shcommands.h>
  73
  74 #define BUFFER_SIZE     160
  75
  76 static void GetNewString(STRPTR, STRPTR, LONG);
  77
  78 AROS_SH2(Set, 41.0,
  79 AROS_SHA(STRPTR, ,NAME,    ,  NULL),
  80 AROS_SHA(STRPTR, ,STRING, /F, NULL))
  81 {
  82     AROS_SHCOMMAND_INIT
  83
  84     struct Process  * SetProc;
  85     struct LocalVar * SetNode;
  86     IPTR              OutArgs[3];
  87     LONG              VarLength;
  88     char              Buffer1[BUFFER_SIZE];
  89     char              Buffer2[BUFFER_SIZE];
  90
  91
  92     if (SHArg(NAME) != NULL || SHArg(STRING) != NULL)
  93     {
  94         /* Make sure we get to here is either arguments are
  95          * provided on the command line.
  96          */
  97         if (SHArg(NAME) != NULL && SHArg(STRING) != NULL)
  98         {
  99             /* Add the new local variable to the list.
 100              */
 101             if
 102             (
 103                 !SetVar(SHArg(NAME),
 104                         SHArg(STRING),
 105                         -1,
 106                         GVF_LOCAL_ONLY)
 107             )
 108             {
 109                 return RETURN_ERROR;
 110             }
 111         }
 112     }
 113     else
 114     {
 115         SetProc = (struct Process *)FindTask(NULL);
 116
 117         ForeachNode(&(SetProc->pr_LocalVars), SetNode)
 118         {
 119             if (SetNode->lv_Node.ln_Type == LV_VAR)
 120             {
 121                 /* Get a clean variable with no excess
 122                  * characters.
 123                  */
 124                 VarLength = -1;
 125                 VarLength = GetVar(SetNode->lv_Node.ln_Name,
 126                                    &Buffer1[0],
 127                                    BUFFER_SIZE,
 128                                    GVF_LOCAL_ONLY
 129                 );
 130                 if (VarLength != -1)
 131                 {
 132                     GetNewString(&Buffer1[0],
 133                                  &Buffer2[0],
 134                                  VarLength
 135                     );
 136
 137                     Buffer2[VarLength] = NULL;
 138
 139                     OutArgs[0] = (IPTR)SetNode->lv_Node.ln_Name;
 140                     OutArgs[1] = (IPTR)&Buffer2[0];
 141                     OutArgs[2] = (IPTR)NULL;
 142                     VPrintf("%-20s\t%-20s\n", &OutArgs[0]);
 143                 }
 144             }
 145         }
 146     }
 147
 148     return RETURN_OK;
 149
 150     AROS_SHCOMMAND_EXIT
 151 } /* main */
 152
 153 static void GetNewString(STRPTR s, STRPTR d, LONG l)
 154 {
 155     int i;
 156     int j;
 157
 158     i = j = 0;
 159
 160     while (i < l)
 161     {
 162         if (s[i] == '*' || s[i] == '\e')
 163         {
 164             d[j] = '*';
 165
 166             i++;
 167             j++;
 168         }
 169         else
 170         {
 171             d[j] = s[i];
 172
 173             i++;
 174             j++;
 175         }
 176     }
 177 } /* GetNewString */