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         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     LONG              VarLength;
  87     char              Buffer1[BUFFER_SIZE];
  88     char              Buffer2[BUFFER_SIZE];
  89
  90
  91     if (SHArg(NAME) != NULL || SHArg(STRING) != NULL)
  92     {
  93         /* Make sure we get to here is either arguments are
  94          * provided on the command line.
  95          */
  96         if (SHArg(NAME) != NULL && SHArg(STRING) != NULL)
  97         {
  98             /* Add the new local variable to the list.
  99              */
 100             if
 101             (
 102                 !SetVar(SHArg(NAME),
 103                         SHArg(STRING),
 104                         -1,
 105                         GVF_LOCAL_ONLY)
 106             )
 107             {
 108                 return RETURN_ERROR;
 109             }
 110         }
 111     }
 112     else
 113     {
 114         SetProc = (struct Process *)FindTask(NULL);
 115
 116         ForeachNode(&(SetProc->pr_LocalVars), SetNode)
 117         {
 118             if (SetNode->lv_Node.ln_Type == LV_VAR)
 119             {
 120                 /* Get a clean variable with no excess
 121                  * characters.
 122                  */
 123                 VarLength = -1;
 124                 VarLength = GetVar(SetNode->lv_Node.ln_Name,
 125                                    &Buffer1[0],
 126                                    BUFFER_SIZE,
 127                                    GVF_LOCAL_ONLY
 128                 );
 129                 if (VarLength != -1)
 130                 {
 131                     GetNewString(&Buffer1[0],
 132                                  &Buffer2[0],
 133                                  VarLength
 134                     );
 135
 136                     Buffer2[VarLength] = 0;
 137
 138                     Printf("%-20s\t%-20s\n", SetNode->lv_Node.ln_Name, Buffer2);
 139                 }
 140             }
 141         }
 142     }
 143
 144     return RETURN_OK;
 145
 146     AROS_SHCOMMAND_EXIT
 147 } /* main */
 148
 149 static void GetNewString(STRPTR s, STRPTR d, LONG l)
 150 {
 151     int i;
 152     int j;
 153
 154     i = j = 0;
 155
 156     while (i < l)
 157     {
 158         if (s[i] == '*' || s[i] == '\e')
 159         {
 160             d[j] = '*';
 161
 162             i++;
 163             j++;
 164         }
 165         else
 166         {
 167             d[j] = s[i];
 168
 169             i++;
 170             j++;
 171         }
 172     }
 173 } /* GetNewString */