/*
    Copyright 1995-2008, Felspar Co Ltd. http://fost.3.felspar.com/
    Distributed under the Boost Software License, Version 1.0.
    See accompanying file LICENSE_1_0.txt or copy at
        http://www.boost.org/LICENSE_1_0.txt
*/


#include "fost-cli.hpp"
#include <fost/arguments.hpp>


using namespace fostlib;


fostlib::arguments::arguments( int argc, native_char *argv[] ) {
    load( argc, argv );
}


fostlib::arguments::arguments( int argc, native_char *argv[], native_char *envp[] ) {
    load( argc, argv );
    load( envp );
}


void fostlib::arguments::load( int argc, native_char *argv[] ) {
    for ( int c( 0 ); c != argc; ++c ) {
        if ( argv[ c ][ 0 ] == L'-' || argv[ c ][ 0 ] == L'/' ) {
            if ( argc == c + 1 )
                m_switches[ string( argv[ c ] + 1 ) ] = L"";
            else {
                m_switches[ string( argv[ c ] + 1 ) ] = string( argv[ c + 1 ] );
                ++c;
            }
        } else {
            m_arguments.push_back( string( argv[ c ] ) );
        }
    }
}


void fostlib::arguments::load( native_char *envp[] ) {
    for ( int arg( 0 ); envp[ arg ] != NULL; ++arg ) {
        string setting( envp[ arg ] );
        std::pair< string, nullable< string > > set = partition( setting, L"=" );
        if ( !set.second.isnull() ) m_environment[ set.first ] = set.second.value();
    }
}


string fostlib::arguments::environment( const string &env_name ) {
#ifdef WIN32
    DWORD length = GetEnvironmentVariable(L"windir", NULL, 0);
    if ( length ) {
        boost::scoped_array< wchar_t > windir( new wchar_t[length]);
        GetEnvironmentVariable(L"windir", windir.get(), length);
        return string(windir.get());
    } else
        throw exceptions::null("Environment variable WINDIR not found");
#else
    throw exceptions::not_implemented("fostlib::arguments::environment( const string &env_name )");
#endif
}
void fostlib::arguments::environment( const string &envName, const string &section, const string &name ) {
    if ( m_environment.find( envName ) != m_environment.end() ) {
        m_registered.push_back( boost::shared_ptr< setting< json > >( new setting< json >( L"Environment", section, name, json( m_environment[ envName ] ), false ) ) );
    }
}


void fostlib::arguments::argument( arguments::size_type argument, const string &section, const string &name ) {
    if ( argument >= 0 && argument < size() ) {
        m_registered.push_back( boost::shared_ptr< setting< json > >( new setting< json >( L"Command argument", section, name, json( m_arguments[ argument ] ), false ) ) );
    }
}


nullable< string > fostlib::arguments::commandSwitch( const string &s ) const {
    std::map< string, string >::const_iterator pos( m_switches.find( s ) );
    if ( pos == m_switches.end() )
        return null;
    else
        return (*pos).second;
}


void fostlib::arguments::commandSwitch( const string &theSwitch, const string &section, const string &name ) {
    if ( m_switches.find( theSwitch) != m_switches.end() )
        m_registered.push_back( boost::shared_ptr< setting< json > >( new setting< json >( L"Command switch", section, name, json::parse( m_switches[ theSwitch ], json( m_switches[ theSwitch ] ) ), false ) ) );
}


arguments::size_type fostlib::arguments::size() const {
    return m_arguments.size();
}


nullable< string > fostlib::arguments::operator []( arguments::size_type pos ) const {
    if ( pos >= size() ) {
        return nullable< string >();
    } else {
        return m_arguments[ pos ];
    }
}


ostream &fostlib::arguments::printOn( ostream &o ) const {
    for ( std::map< string, string >::const_iterator it( m_environment.begin() ); it != m_environment.end(); ++it ) {
        o << L"argument,environment," << (*it).first << L"," << (*it).second << std::endl;
    }
    for ( std::vector< string >::const_iterator it( m_arguments.begin() ); it != m_arguments.end(); ++it ) {
        o << L"argument,command line argument," << (*it) << std::endl;
    }
    for ( std::map< string, string >::const_iterator it( m_switches.begin() ); it != m_switches.end(); ++it ) {
        o << L"argument,command line switch," << (*it).first << L"," << (*it).second << std::endl;
    }
    return o;
}