lib/cmd.php

   1 <?php
   2
   3         function getCmdOpt($aArg, $aSpec, &$aResult, $bExitOnError = false, $bExitOnUnknown = false)
   4         {
   5                 $aQuick = array();
   6                 $aCounts = array();
   7
   8                 foreach($aSpec as $aLine)
   9                 {
  10                         if (is_array($aLine))
  11                         {
  12                                 if ($aLine[0]) $aQuick['--'.$aLine[0]] = $aLine;
  13                                 if ($aLine[1]) $aQuick['-'.$aLine[1]] = $aLine;
  14                                 $aCounts[$aLine[0]] = 0;
  15                         }
  16                 }
  17
  18                 $aResult = array();
  19                 $bUnknown = false;
  20                 $iSize = sizeof($aArg);
  21                 for ($i = 1; $i < $iSize; $i++)
  22                 {
  23                         if (isset($aQuick[$aArg[$i]]))
  24                         {
  25                                 $aLine = $aQuick[$aArg[$i]];
  26                                 $aCounts[$aLine[0]]++;
  27                                 $xVal = null;
  28                                 if ($aLine[4] == $aLine[5])
  29                                 {
  30                                         if ($aLine[4])
  31                                         {
  32                                                 $xVal = array();
  33                                                 for($n = $aLine[4]; $i < $iSize && $n; $n--)
  34                                                 {
  35                                                         $i++;
  36                                                         if ($i >= $iSize || $aArg[$i][0] == '-') showUsage($aSpec, $bExitOnError, 'Parameter of  \''.$aLine[0].'\' is missing');
  37
  38                                                         switch ($aLine[6])
  39                                                         {
  40                                                         case 'realpath':
  41                                                                 $xVal[] = realpath($aArg[$i]);
  42                                                                 break;
  43                                                         case 'realdir':
  44                                                                 $sPath = realpath(dirname($aArg[$i]));
  45                                                                 if ($sPath)
  46                                                                         $xVal[] = $sPath . '/' . basename($aArg[$i]);
  47                                                                 else
  48                                                                         $xVal[] = $sPath;
  49                                                                 break;
  50                                                         case 'bool':
  51                                                                 $xVal[] = (bool)$aArg[$i];
  52                                                                 break;
  53                                                         case 'int':
  54                                                                 $xVal[] = (int)$aArg[$i];
  55                                                                 break;
  56                                                         case 'float':
  57                                                                 $xVal[] = (float)$aArg[$i];
  58                                                                 break;
  59                                                         default:
  60                                                                 $xVal[] = $aArg[$i];
  61                                                                 break;
  62                                                         }
  63                                                 }
  64                                                 if ($aLine[4] == 1) $xVal = $xVal[0];
  65                                         }
  66                                         else
  67                                         {
  68                                                 $xVal = true;
  69                                         }
  70                                 }
  71                                 else
  72                                 {
  73                                         fail('Variable numbers of params not yet supported');
  74                                 }
  75
  76                                 if ($aLine[3] > 1)
  77                                 {
  78                                         if (!array_key_exists($aLine[0], $aResult)) $aResult[$aLine[0]] = array();
  79                                         $aResult[$aLine[0]][] = $xVal;
  80                                 }
  81                                 else
  82                                 {
  83                                         $aResult[$aLine[0]] = $xVal;
  84                                 }
  85                         }
  86                         else
  87                         {
  88                                 $bUnknown = $aArg[$i];
  89                         }
  90                 }
  91
  92                 if (array_key_exists('help', $aResult)) showUsage($aSpec);
  93                 if ($bUnknown && $bExitOnUnknown) showUsage($aSpec, $bExitOnError, 'Unknown option \''.$bUnknown.'\'');
  94
  95                 foreach($aSpec as $aLine)
  96                 {
  97                         if (is_array($aLine))
  98                         {
  99                                 if ($aCounts[$aLine[0]] < $aLine[2]) showUsage($aSpec, $bExitOnError, 'Option \''.$aLine[0].'\' is missing');
 100                                 if ($aCounts[$aLine[0]] > $aLine[3]) showUsage($aSpec, $bExitOnError, 'Option \''.$aLine[0].'\' is pressent too many times');
 101                                 switch ($aLine[6])
 102                                 {
 103                                 case 'bool':
 104                                         if (!array_key_exists($aLine[0], $aResult))
 105                                                 $aResult[$aLine[0]] = false;
 106                                         break;
 107                                 }
 108                         }
 109                 }
 110                 return $bUnknown;
 111         }
 112
 113         function showUsage($aSpec, $bExit = false, $sError = false)
 114         {
 115                 if ($sError)
 116                 {
 117                         echo basename($_SERVER['argv'][0]).': '.$sError."\n";
 118                         echo 'Try `'.basename($_SERVER['argv'][0]).' --help` for more information.'."\n";
 119                         exit;
 120                 }
 121                 echo "Usage: ".basename($_SERVER['argv'][0])."\n";
 122                 $bFirst = true;
 123                 foreach($aSpec as $aLine)
 124                 {
 125                         if (is_array($aLine))
 126                         {
 127                                 if ($bFirst)
 128                                 {
 129                                         $bFirst = false;
 130                                         echo "\n";
 131                                 }
 132                                 $aNames = array();
 133                                 if ($aLine[1]) $aNames[] = '-'.$aLine[1];
 134                                 if ($aLine[0]) $aNames[] = '--'.$aLine[0];
 135                                 $sName = join(', ',$aNames);
 136                                 echo '  '.$sName.str_repeat(' ',30-strlen($sName)).$aLine[7]."\n";
 137                         }
 138                         else
 139                         {
 140                                 echo $aLine."\n";
 141                         }
 142                 }
 143                 echo "\n";
 144                 exit;
 145         }
 146
 147         function chksql($oSql, $sMsg = false)
 148         {
 149                 if (PEAR::isError($oSql))
 150                 {
 151                         fail($sMsg || $oSql->getMessage(), $oSql->userinfo);
 152                 }
 153
 154                 return $oSql;
 155         }