mediatag/main.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#if __DCL_WINDOWS
#include <windows.h>
#endif
 
#include <locale.h>
#include <stdio.h>
#include <wchar.h>  // wprintf
 
#include <dcl/Charset.h>
#include <dcl/Array.h>
#include <dcl/DateTime.h>
#include <dcl/FileWriter.h>
#include <dcl/Files.h>
#include <dcl/Arguments.h>
 
#include "MediaInfo.h"
 
#include "main.h"
#include "TagCounter.h"
#include "TagReader.h"
 
#if __DCL_DEBUG
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("media/main.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
static Arguments::Option __options__[] =
{
    { L"summary", L's', NULL,      0, L"Summary of tag items"   },
    { L"verbose", L'v', NULL,      0, L"Produce verbose output" },
    { NULL,       0,    NULL,      0, L"The following options should be grouped together:" },
    { L"database", L'd', L"STRING", 0, L"Database connection."
            "\n\t\t\t    DRIVER=DCLFirebird;USER=MT100;PASSWORD=pass;SERVER=g02u24kn2/32055;DATABASE=MT100"
            "\n\t\t\t    DRIVER=DCLInformix;USER=MT100;PASSWORD=pass;SERVER=informix02;DATABASE=MT100"
            "\n\t\t\t    DRIVER=DCLMariaDB; USER=MT100;PASSWORD=pass;SERVER=g02u24kn2;PORT=32011;DATABASE=MT100"
            "\n\t\t\t    DRIVER=DCLOracleDb;USER=MT100;PASSWORD=pass;DATABASE=g01r08od21:1521/AL32UTF8"
            "\n\t\t\t    DRIVER=DCLPostgreSQL;USER=MT100;PASSWORD=pass;DATABASE=tcp:postgresql://g02u24kn2:32017/MT100" },
    { L"dry-run",   L'n',   NULL,       0,  L"perform a trial run with no changes made" },
    { NULL }
};
 
MainArguments::MainArguments(Writer& _output, Writer& _errout)
    : Arguments(
        _output, _errout,
        L"mediatag 1.0",
        L"daejung@gowoonsoft.com",
        L"PATH",
        L"Collects mp3 tags",
        __options__
    )
{
     __dryrun = __verbose = __summary = false;
}
 
String MainArguments::toString() const
{
    StringBuilder sb;
    sb.append(value0())
        .append(L" --verbose=").append(String::valueOf(__verbose))
        .append(L" --summary=").append(String::valueOf(__summary))
        .append(L" --database=").append(__database)
        .append(L" --dry-run=").append(String::valueOf(__dryrun))
        .append(L" values[").append(values().toString())
        .append(L"]");
 
    return sb;
}
 
void MainArguments::onOption(int _key, const String& _arg)
        __DCL_THROWS1(Exception*)
{
    switch (_key) {
    case L's':
        __summary = true;
        break;
    case L'v':
        __verbose = true;
        break;
    case L'd':
        __database = _arg;
        break;
    case L'n':
        __dryrun = true;
        break;
    }
}
 
String MainArguments::onValidate()
{
    return Arguments::onValidate();
}
 
void __main(const MainArguments& _args)
{
    __DCL_ASSERT(!_args.values().isEmpty());
    String path = _args.values()[0];
    if (_args.summary()) {
        TagCounter counter;
        if (Files::isDirectory(path)) {
            TagCounter::readDir(path, counter, _args);
        }
        else {
            TagCounter::read(path, counter);
        }
        
        _args.output() << L"[" << path << L"] " << counter.toString() << endl;
        return;
    }
 
    TagReader reader(_args);
    if (Files::isDirectory(path)) {
        reader.readDir(path);
    }
    else {
        reader.read(Files::dirname(path), Files::basename(path));
    }
}
 
__DCL_END_NAMESPACE
 
__DCL_USING_NAMESPACE
 
int main(int argc, char* argv[])
{
    // std::locale::global(std::locale(""));
    setlocale(LC_ALL, "");
 
    __DCL_INITIALIZE
    try {
        FileWriter output(STDOUT_HANDLE);
        FileWriter errout(STDERR_HANDLE);
#if __DCL_DEBUG
        DCLDebugSetGlobalReport(&errout);
#endif
        MainArguments args(output, errout);
        if (args.parse(argc, argv)) {
            String sep(L'=', 80);
            output << L"Start [" << DateTime::getCurrentLocalTime().toString()
                << L"]" << endl;
            if (args.verbose()) {
                output << L"      [" << args.toString() << L"]" << endl;
            }
            output << sep << endl;
 
            MediaInfo::initialize();
            try {
                __main(args);
            }
            catch (Exception* e) {
                errout << e->toStringAll() << endl;
                e->destroy();
            }
            MediaInfo::cleanup();
 
            output << sep << endl
                << L"Finish [" << DateTime::getCurrentLocalTime().toString()
                << L"]" << endl;
        }
#if __DCL_DEBUG
#if __DCL_HAVE_ALLOC_DEBUG
        DCLDebugDumpGlobalMemoryLeak(DCL_ALLOC_DUMP_ALL, &errout);
#endif
        DCLDebugSetGlobalReport(NULL);
#endif      // __DCL_DEBUG
    }
    catch(Exception* e)    {
        String s = e->toStringAll();
        e->destroy();
        fwprintf(stderr, L"Warning!! Final Exception: %ls\n", s.data());
    }
    __DCL_CLEANUP
 
    return 0;
}