1 files changed, 224 insertions, 1 deletions

diff --git a/src/optparser.cpp b/src/optparser.cpp
index 2046792..f99dd85 100644
--- a/src/optparser.cpp
+++ b/src/optparser.cpp
@@ -1,4 +1,8 @@
 #include "bu/optparser.h"
+#include "bu/sio.h"
+using namespace Bu;
+
+#include <stdlib.h>
 
 Bu::OptParser::OptParser()
 {
@@ -8,6 +12,81 @@ Bu::OptParser::~OptParser()
 {
 }
 
+void Bu::OptParser::parse( int argc, char **argv )
+{
+        for( int j = 1; j < argc; j++ )
+        {
+                if( argv[j][0] == '-' )
+                {
+                        // Now we're on to something, which kind is it?
+                        if( argv[j][1] == '-' )
+                        {
+                                // Long param, cool, that's easy, first search for =
+                                int iEPos;
+                                for( iEPos = 2; argv[j][iEPos] != '\0' &&
+                                         argv[j][iEPos] != '='; iEPos++ ) { }
+
+                                Bu::FString sOpt;
+                                int iCount = argc-j;
+                                if( argv[j][iEPos] == '=' )
+                                {
+                                        sOpt.set( argv[j]+2, iEPos-2 );
+                                        iCount++;
+                                }
+                                else
+                                {
+                                        sOpt.set( argv[j]+2 );
+                                }
+                                Option *pOpt = hlOption.get( sOpt );
+                                Bu::StrArray aParams( iCount );
+                                aParams.append( sOpt );
+                                if( argv[j][iEPos] == '=' )
+                                {
+                                        aParams.append( argv[j]+iEPos+1 );
+                                }
+                                for( int k = j+1; k < argc; k++ )
+                                {
+                                        aParams.append( argv[k] );
+                                }
+                                if( pOpt->sUsed )
+                                        j += pOpt->sUsed( aParams );
+                        }
+                        else
+                        {
+                                int iCPos;
+                                for( iCPos = 1; argv[j][iCPos] != '\0'; iCPos++ )
+                                {
+                                        Option *pOpt = hsOption.get( argv[j][iCPos] );
+                                        Bu::StrArray aParams( argc-j+1 );
+                                        char buf[2] = {argv[j][iCPos], '\0'};
+                                        aParams.append( buf );
+                                        if( pOpt->bShortHasParams )
+                                        {
+                                                if( argv[j][iCPos+1] != '\0' )
+                                                        aParams.append( argv[j]+iCPos+1 );
+                                                for( int k = j+1; k < argc; k++ )
+                                                {
+                                                        aParams.append( argv[k] );
+                                                }
+                                                if( pOpt->sUsed )
+                                                {
+                                                        j += pOpt->sUsed( aParams );
+                                                }
+                                                break;
+                                        }
+                                        else
+                                        {
+                                                pOpt->sUsed( aParams );
+                                        }
+                                }
+                        }
+                }
+                else
+                {
+                }
+        }
+}
+
 void Bu::OptParser::addOption( const Option &opt )
 {
         lOption.append( opt );
@@ -15,13 +94,155 @@ void Bu::OptParser::addOption( const Option &opt )
                 hsOption.insert( opt.cOpt, &lOption.last() );
         if( opt.sOpt )
                 hlOption.insert( opt.sOpt, &lOption.last() );
+}
 
+void Bu::OptParser::addHelpOption( char c, const Bu::FString &s, const Bu::FString &sHelp )
+{
+        Option o;
+        o.sUsed = slot( this, &OptParser::optHelp );
+        o.cOpt = c;
+        o.sOpt = s;
+        o.sHelp = sHelp;
+        o.bShortHasParams = false;
+        addOption( o );
+}
+
+int Bu::OptParser::optHelp( StrArray aParams )
+{
+        bool bHasShort = false;
+        int iMaxWidth = 0;
+        int iScrWidth = 80;
+        char *env = getenv("COLUMNS");
+        if( env )
+                iScrWidth = strtol( env, NULL, 10 );
+        for( OptionList::iterator i = lOption.begin(); i; i++ )
+        {
+                if( (*i).cOpt != '\0' )
+                        bHasShort = true;
+                if( (*i).sOpt && iMaxWidth < (*i).sOpt.getSize() )
+                        iMaxWidth = (*i).sOpt.getSize();
+        }
+        int iIndent = 4;
+        if( bHasShort )
+                iIndent += 4;
+        if( iMaxWidth > 0 )
+                iIndent += 4 + iMaxWidth;
+        for( OptionList::iterator i = lOption.begin(); i; i++ )
+        {
+                sio << "    ";
+                if( bHasShort )
+                {
+                        if( (*i).cOpt == '\0' )
+                                sio << "  ";
+                        else
+                                sio << "-" << (*i).cOpt;
+                        sio << "  ";
+                }
+                if( iMaxWidth > 0 )
+                {
+                        if( (*i).sOpt )
+                        {
+                                sio << "--" << Fmt(iMaxWidth, Fmt::Left) << (*i).sOpt;
+                        }
+                        else
+                        {
+                                sio << "  " << Fmt(iMaxWidth) << "";
+                        }
+                        sio << "  ";
+                }
+                sio << format( (*i).sHelp, iScrWidth-iIndent-1, iIndent );
+                sio << sio.nl;
+        }
+        exit( 0 );
+        return 0;
 }
 
 Bu::FString Bu::OptParser::format( const Bu::FString &sIn, int iWidth,
                 int iIndent )
 {
+        Bu::FString sOut;
+        Bu::FString sIndent;
+        for( int j = 0; j < iIndent; j++ )
+                sIndent.append(" ", 1);
+        bool bFirst = true;
+        int iSpaceCount = 0;
+        bool bSpace = false;
+        int iPrevLineLen;
+        int iLineLen = 0;
+        char c;
+        Bu::FString::const_iterator iLastSpace, iStart;
+        for( Bu::FString::const_iterator i = iLastSpace = iStart = sIn.begin(); i; i++ )
+        {
+                c = *i;
+                if( *i == ' ' )
+                {
+                        if( bSpace == false )
+                        {
+                                iLastSpace = i;
+                                iSpaceCount++;
+                                bSpace = true;
+                                iPrevLineLen = iLineLen;
+                        }
+                }
+                else
+                {
+                        bSpace = false;
+                }
+                iLineLen++;
 
+                if( iLineLen >= iWidth )
+                {
+                        iSpaceCount--;
+                        if( bFirst == true )
+                                bFirst = false;
+                        else
+                                sOut += sIndent;
+                        int iExtraSpaces = iWidth-iPrevLineLen;
+                        bSpace = false;
+                        float fFill = 0.0;
+                        int iSubSpaceCount = 0;
+                        float fAdd = ((float)iExtraSpaces/(float)iSpaceCount);
+                        for( Bu::FString::const_iterator k = iStart; k != iLastSpace; k++ )
+                        {
+                                sOut += *k;
+                                if( *k == ' ' )
+                                {
+                                        if( bSpace == false && iExtraSpaces > 0 )
+                                        {
+                                                bSpace = true;
+                                                fFill += fAdd;
+                                                iSubSpaceCount++;
+                                                for( int sp = 0; sp < (int)(fFill); sp++ )
+                                                {
+                                                        sOut += ' ';
+                                                        iExtraSpaces--;
+                                                }
+                                                fFill -= (int)fFill;
+                                                if( iSubSpaceCount == iSpaceCount && iExtraSpaces > 0 )
+                                                {
+                                                        for(; iExtraSpaces > 0; iExtraSpaces-- )
+                                                        {
+                                                                sOut += ' ';
+                                                        }
+                                                }
+                                        }
+                                }
+                                else
+                                        bSpace = false;
+                        }
+                        //sOut.append( iStart, iLastSpace );
+                        sOut.append("\n");
+                        for(; iLastSpace && *iLastSpace == ' '; iLastSpace++ ) { }
+                        iStart = i = iLastSpace;
+                        bSpace = false;
+                        iLineLen = 1;
+                        iSpaceCount = 0;
+                }
+        }
+        if( !bFirst )
+                sOut += sIndent;
+        sOut.append( iStart );
+        return sOut;
 }
 
 
@@ -29,7 +250,9 @@ Bu::FString Bu::OptParser::format( const Bu::FString &sIn, int iWidth,
 // Code for Bu::OptParser::Option
 //
 
-Bu::OptParser::Option::Option()
+Bu::OptParser::Option::Option() :
+        cOpt( '\0' ),
+        bShortHasParams( false )
 {
 }