// Copyright (C) 2010 David Sugar, Tycho Softworks.
//
// This file is part of GNU uCommon C++.
//
// GNU uCommon C++ is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// GNU uCommon C++ is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with GNU uCommon C++.  If not, see <http://www.gnu.org/licenses/>.

#include <ucommon/ucommon.h>

using namespace UCOMMON_NAMESPACE;

static shell::flagopt helpflag('h',"--help",    _TEXT("display this list"));
static shell::flagopt althelp('?', NULL, NULL);
static shell::charopt delim('d',    "--delim", _TEXT("set deliminter between arguments"));
static shell::flagopt directory('D', "--directory", _TEXT("expand directory into file arguments"));
static shell::flagopt lines('l',    "--lines", _TEXT("list arguments on separate lines"));
static shell::stringopt quote('q',  "--quote",  _TEXT("set quote for each argument"), "string", "");
static shell::flagopt recursive('R', "--recursive", _TEXT("recursive directory scan"));
static shell::flagopt follow('F', "--follow", _TEXT("follow symlinks"));
static shell::flagopt rflag('r',    "--reverse", _TEXT("reverse order of arguments"));

static char prefix[80] = {0, 0};
static char suffix[80] = {0, 0};

static void output(bool middle, const char *arg)
{
    if(is(lines))
        file::cout << prefix << arg << suffix << "\n";
    else if(middle)
        file::cout << *delim << prefix << arg << suffix;
    else
        file::cout << prefix << arg << suffix;
}

static void dirpath(bool middle, String path, bool top = true)
{
    char filename[128];
    string_t subdir;
    dir_t dir(path);
    unsigned count = 0;

    while(is(dir) && dir.read(filename, sizeof(filename))) {
        if(*filename == '.')
            continue;

        ++count;
        subdir = (String)path + (String)"/" + (String)filename;
        output(middle, subdir);
        middle = true;

        if(fsys::is_dir(*subdir)) {
            if(is(follow) || is(recursive)) {
                if(!fsys::is_link(*subdir) || is(follow))
                    dirpath(true, subdir, false);
            }
        }
    }
    if(top && !count)
        output(middle, path);
}

PROGRAM_MAIN(argc, argv)
{
    unsigned count = 0;
    char *ep;
    bool middle = false;

    shell::bind("args");
    shell args(argc, argv);

    if(is(helpflag) || is(althelp)) {
        printf("%s\n", _TEXT("Usage: args [options] arguments..."));
        printf("%s\n\n", _TEXT("Echo command line arguments"));
        printf("%s\n", _TEXT("Options:"));
        shell::help();
        printf("\n%s\n", _TEXT("Report bugs to dyfet@gnu.org"));
        PROGRAM_EXIT(0);
    }

    if(!args())
        PROGRAM_EXIT(0);

    if(quote[0]) {
        if(!quote[1]) {
            prefix[0] = quote[0];
            suffix[0] = quote[0];
        }
        else if(!quote[2]) {
            prefix[0] = quote[0];
            suffix[0] = quote[1];
        }
        else if(quote[0] == '<') {
            String::set(prefix, sizeof(prefix), *quote);
            snprintf(suffix, sizeof(suffix), "</%s", *quote + 1);
        }
        else if(quote[0] == '(') {
            String::set(prefix, sizeof(prefix), quote);
            ep = strchr((char *)*quote, ')');
            if(ep)
                *ep = 0;
            suffix[0] = ')';
        }
        else {
            String::set(prefix, sizeof(prefix), quote);
            String::set(suffix, sizeof(suffix), quote);
        }
    }

    if(is(rflag)) {
        count = args();
        while(count--) {
            if(fsys::is_dir(args[count]) && (is(directory) || is(recursive) || is(follow)))
                dirpath(middle, (String)args[count]);
            else
                output(middle, args[count]);
            middle = true;
        }
    }
    else while(count < args()) {
        if(fsys::is_dir(args[count]) && (is(directory) || is(recursive) || is(follow)))
            dirpath(middle, (String)args[count++]);
        else
            output(middle, (String)args[count++]);
        middle = true;
    }

    if(!lines)
        file::cout << "\n";

    PROGRAM_EXIT(0);
}