workbench/c/shellcommands/PathPart.c

   1 /*
   2     Copyright © 2011, The AROS Development Team. All rights reserved.
   3     $Id:$
   4
   5     Desc:
   6     Lang: English
   7 */
   8
   9 /******************************************************************************
  10
  11     NAME
  12
  13         PathPart
  14
  15     TEMPLATE
  16
  17         DIR/K,FILE/K,ADD/K/M
  18
  19     LOCATION
  20
  21         C:
  22
  23     FUNCTION
  24
  25         Extracts directory or file name from a path, or assembles a path
  26
  27     FORMAT
  28
  29         PATHPART [DIR <path name>] [FILE <path name>] [ADD {<device name | directory name | file name>}]
  30
  31     RESULT
  32
  33         Standard DOS return codes.
  34
  35
  36 ******************************************************************************/
  37
  38 #include <proto/exec.h>
  39 #include <proto/dos.h>
  40
  41 //#define DEBUG 1
  42 #include <aros/debug.h>
  43
  44 #include <aros/shcommands.h>
  45
  46 #define MAX_PATH_LEN 2048 /* Same as in C:Delete */
  47
  48 AROS_SH3H(PathPart,50.1, "extract directory or file name from a path, or assemble a path\n",
  49 AROS_SHAH(STRPTR  ,,DIR ,/K  ,NULL,"Path from which to extract the directory name"),
  50 AROS_SHAH(STRPTR  ,,FILE,/K  ,NULL,"Path from which to extract the file name"),
  51 AROS_SHAH(STRPTR *,,ADD ,/K/M,NULL,"Device and/or directory(ies) and/or file names from\n"
  52                                  "\twhich to build a path (order matters!)\n") )
  53 {
  54     AROS_SHCOMMAND_INIT
  55
  56     int    rc     = RETURN_FAIL;
  57     STRPTR outstr = NULL, buf = NULL;
  58     UBYTE  buf2[MAX_PATH_LEN];
  59     LONG   size;
  60
  61     if (    (SHArg(ADD) && SHArg(DIR))
  62          || (SHArg(ADD) && SHArg(FILE))
  63          || (SHArg(DIR) && SHArg(FILE))
  64        )
  65     {
  66         SetIoErr(ERROR_BAD_TEMPLATE);
  67     }
  68     else if (SHArg(DIR))
  69     {
  70         size  = strlen((char *)SHArg(DIR));
  71         if ((buf = AllocVec(size, MEMF_ANY)))
  72         {
  73             CopyMem (SHArg(DIR), buf, size);
  74             buf[PathPart(buf) - buf] = '\0';
  75             outstr = buf;
  76             rc = RETURN_OK;
  77         }
  78     }
  79     else if (SHArg(FILE))
  80     {
  81         outstr = FilePart(SHArg(FILE));
  82         rc = RETURN_OK;
  83     }
  84     else if (SHArg(ADD))
  85     {
  86         buf2[0]='\0';
  87         while (*SHArg(ADD))
  88         {
  89             if (!AddPart(buf2 ,*SHArg(ADD)++ ,MAX_PATH_LEN))
  90             {
  91                 /* rc == RETURN_FAIL until this point... */
  92                 rc = RETURN_ERROR;
  93                 break;
  94             }
  95         }
  96         if (rc != RETURN_ERROR)
  97         {
  98             outstr = buf2;
  99             rc = RETURN_OK;
 100         }
 101     }
 102     else
 103         SetIoErr(ERROR_REQUIRED_ARG_MISSING);
 104
 105     if (rc == RETURN_OK)
 106     {
 107         if (outstr)
 108         {
 109             PutStr(outstr);
 110             FPutC(Output(), '\n');
 111         }
 112     }
 113     else
 114         PrintFault(IoErr(), (CONST_STRPTR)"PathPart");
 115
 116     if (buf)
 117         FreeVec(buf);
 118
 119     return rc;
 120
 121     AROS_SHCOMMAND_EXIT
 122 }
 123
 124