workbench/c/shellcommands/If.c

   1 /*
   2     Copyright © 1995-2001, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Desc:
   6     Lang: English
   7 */
   8 #include <aros/debug.h>
   9 /******************************************************************************
  10
  11
  12     NAME
  13
  14         If
  15
  16     SYNOPSIS
  17
  18         NOT/S,WARN/S,ERROR/S,FAIL/S,,EQ/K,GT/K,GE/K,VAL/S,EXISTS/K
  19
  20     LOCATION
  21
  22         Workbench:C
  23
  24     FUNCTION
  25
  26         Carry out all the commands in a block if a given conditional is true.
  27         (A block is a run of command lines ended with an Else or EndIf
  28         command.) For every If command there must be a corresponding EndIf.
  29         If the condition is false, command execution will skip to the
  30         corresponding Else of EndIf command.
  31
  32     INPUTS
  33
  34         NOT               --  Negates the value of the condition
  35
  36         WARN              --  True if the previous return code was greater
  37                               than or equal to 5.
  38         ERROR             --  True if the previous return code was greater
  39                               than or equal to 10.
  40         FAIL              --  True if the previous return code was greater
  41                               than or equal to 20.
  42
  43         EQ, GE, GT        --  True if the first value is equal, greater than
  44                               or equal respectively greater than the second.
  45
  46         VAL               --  Indicate that the comparison should treat the
  47                               strings as numerical values.
  48
  49         EXISTS  <string>  --  True if the file or directory <string> exists.
  50
  51
  52     RESULT
  53
  54     NOTES
  55
  56         ERROR and FAIL will only be appropriate if the fail level of the
  57         script is set via FailAt (the standard fail level is 10 and if any
  58         return code exceeds or equals this value, the script will be aborted).
  59
  60     EXAMPLE
  61
  62         If 500 GT 200 VAL
  63             echo "500 is greater than 200"
  64         Else
  65             If EXISTS S:User-Startup
  66                 echo "User-Startup script found in S:"
  67                 Execute S:User-Startup
  68             EndIf
  69         EndIf
  70
  71     BUGS
  72
  73     SEE ALSO
  74
  75         Else, EndIf, FailAt
  76
  77     INTERNALS
  78
  79     HISTORY
  80
  81     10.01.2000  SDuvan   implemented
  82
  83 ******************************************************************************/
  84
  85 #define  DEBUG  0
  86 #include <aros/debug.h>
  87
  88 #include <dos/dos.h>
  89 #include <dos/dosextens.h>
  90 #include <dos/rdargs.h>
  91 #include <dos/stdio.h>
  92 #include <proto/dos.h>
  93 #include <proto/utility.h>
  94 #include <proto/exec.h>
  95 #include <dos_commanderrors.h>
  96
  97 #define SH_GLOBAL_DOSBASE 1
  98 struct UtilityBase *UtilityBase;
  99
 100 #include <aros/shcommands.h>
 101
 102 static BOOL doeval(STRPTR arg1, STRPTR arg2, BYTE op, IPTR numeric);
 103
 104 AROS_SH10(If, 41.1,
 105 AROS_SHA(BOOL, ,NOT,/S, FALSE),
 106 AROS_SHA(BOOL, ,WARN,/S,FALSE),
 107 AROS_SHA(BOOL, ,ERROR,/S, FALSE),
 108 AROS_SHA(BOOL, ,FAIL,/S,FALSE),
 109 AROS_SHA(STRPTR, , , ,NULL),
 110 AROS_SHA(STRPTR, ,EQ,/K,NULL),
 111 AROS_SHA(STRPTR, ,GT,/K,NULL),
 112 AROS_SHA(STRPTR, ,GE,/K,NULL),
 113 AROS_SHA(BOOL, ,VAL,/S,FALSE),
 114 AROS_SHA(STRPTR, ,EXISTS,/K,NULL))
 115 {
 116
 117     AROS_SHCOMMAND_INIT
 118
 119     BOOL result = FALSE;
 120
 121     struct CommandLineInterface *cli = Cli();
 122
 123     UtilityBase = (struct UtilityBase *)OpenLibrary("utility.library", 39);
 124     if (!UtilityBase)
 125         return RETURN_FAIL;
 126
 127
 128     if((cli != NULL) && (cli->cli_CurrentInput != cli->cli_StandardInput))
 129     {
 130         D(bug("Current input = %p, Standard input = %p\n",
 131               cli->cli_CurrentInput, cli->cli_StandardInput));
 132
 133         if(SHArg(WARN))
 134         {
 135             if(cli->cli_ReturnCode >= RETURN_WARN)
 136                 result = TRUE;
 137         }
 138         else if(SHArg(ERROR))
 139         {
 140             if(cli->cli_ReturnCode >= RETURN_ERROR)
 141                 result = TRUE;
 142         }
 143         else if(SHArg(FAIL))
 144         {
 145             if(cli->cli_ReturnCode >= RETURN_FAIL)
 146                 result = TRUE;
 147         }
 148         else if(SHArg(EQ))
 149         {
 150             result = doeval(SHArg( ), SHArg(EQ), 0, SHArg(VAL));
 151         }
 152         else if (SHArg(GT))
 153         {
 154             result = doeval(SHArg( ), SHArg(GT), 1, SHArg(VAL));
 155         }
 156         else if (SHArg(GE))
 157         {
 158             result = doeval(SHArg( ), SHArg(GE), 2, SHArg(VAL));
 159         }
 160         else if(SHArg(EXISTS))
 161         {
 162             BPTR lock = Lock(SHArg(EXISTS), SHARED_LOCK);
 163
 164             if(lock != NULL)
 165                 result = TRUE;
 166
 167             UnLock(lock);
 168         }
 169
 170         if(SHArg(NOT))                 /* NOT */
 171             result = !result;
 172
 173
 174         /* We have determined the result -- now we've got to act on it. */
 175
 176         if(!result)
 177         {
 178             char a;
 179             char buffer[256];
 180             int  level = 1;         /* If block level */
 181             BOOL found = FALSE; /* Have we found a matching Else or
 182                                    EndIF? */
 183
 184             SelectInput(cli->cli_CurrentInput);
 185
 186             while(!found)
 187             {
 188                 LONG status;
 189
 190                 status = ReadItem(buffer, sizeof(buffer), NULL);
 191
 192                 if(status == ITEM_ERROR)
 193                     break;
 194
 195                 if(status == ITEM_NOTHING)
 196                 {
 197                     if(FGetC(Input()) == ENDSTREAMCH)
 198                         break;
 199                 }
 200
 201                 switch(FindArg("IF,ELSE,ENDIF", buffer))
 202                 {
 203                 case 0:
 204                     level++;
 205                     //                  printf("Found If\n");
 206                     break;
 207
 208                 case 1:
 209                     if(level == 1)
 210                         found = TRUE;
 211                     break;
 212
 213                 case 2:
 214                     level--;
 215
 216                     if(level == 0)
 217                         found = TRUE;
 218                     break;
 219                 }
 220
 221                 /* Take care of long lines */
 222                 do
 223                 {
 224                     a = FGetC(Input());
 225                 } while (a != '\n' && a != ENDSTREAMCH);
 226             }
 227
 228             if(!found)
 229                 PrintFault(ERROR_NO_MATCHING_ELSEENDIF, "If");
 230         }
 231     }
 232     else
 233     {
 234         Flush(Output());
 235         PrintFault(ERROR_SCRIPT_ONLY, "If");
 236     }
 237
 238     CloseLibrary((struct Library *)UtilityBase);
 239
 240     return RETURN_OK;
 241
 242     AROS_SHCOMMAND_EXIT
 243 }
 244
 245 static BOOL doeval(STRPTR arg1, STRPTR arg2, BYTE op, IPTR numeric)
 246 {
 247     STRPTR s1 = (STRPTR)arg1;
 248     STRPTR s2 = (STRPTR)arg2;
 249     BOOL result = FALSE;
 250
 251     if (s1 && s2)
 252     {
 253         if (numeric)
 254         {
 255             LONG val1, val2;
 256
 257             StrToLong(s1, &val1);
 258             StrToLong(s2, &val2);
 259
 260             switch(op)
 261             {
 262                 case 0:
 263                     result = (val1 == val2);
 264                     break;
 265
 266                 case 1:
 267                     result = (val1 > val2);
 268                     break;
 269
 270                 case 2:
 271                     result = (val1 >= val2);
 272                     break;
 273             }
 274
 275         } /* if (numeric) */
 276         else
 277         {
 278             LONG match;
 279
 280             match = Stricmp(s1, s2);
 281
 282             switch(op)
 283             {
 284                 case 0:
 285                     result = (match == 0);
 286                     break;
 287
 288                 case 1:
 289                     result = (match > 0);
 290                     break;
 291
 292                 case 2:
 293                     result = (match >= 0);
 294                     break;
 295             }
 296
 297         }
 298
 299     } /* if (s1 && s2) */
 300
 301     return result;
 302 }
 303