zipios.cpp

Go to the documentation of this file.

/*
  Zipios++ - a small C++ library that provides easy access to .zip files.

  Copyright (C) 2000-2007  Thomas Sondergaard
  Copyright (C) 2015  Made to Order Software Corporation

  This library 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 2 of the License, or (at your option) any later version.

  This library 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 this library; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
*/

#include "zipios/zipfile.hpp"

#include <cstring>

#include <stdlib.h>


namespace
{

char *g_progname;


void usage()
{
    std::cout << "Usage:  " << g_progname << " [-opt] [file]" << std::endl;
    std::cout << "Where -opt is one or more of:" << std::endl;
    std::cout << "  --count                 count the number of files in a .zip archive" << std::endl;
    std::cout << "  --count-directories     count the number of files in a .zip archive" << std::endl;
    std::cout << "  --count-files           count the number of files in a .zip archive" << std::endl;
    std::cout << "  --help                  show this help screen" << std::endl;
    std::cout << "  --version               print the library version and exit" << std::endl;
    std::cout << "  --version-tool          print the tool version and exit" << std::endl;
    exit(1);
}


enum class func_t
{
    UNDEFINED,

    COUNT,

    COUNT_DIRECTORIES,

    COUNT_FILES
};

} // no name namespace


int main(int argc, char *argv[])
{
    // define program name
    g_progname = argv[0];
    char *e(strrchr(g_progname, '/'));
    if(e)
    {
        g_progname = e + 1;
    }
    e = strrchr(g_progname, '\\');
    if(e)
    {
        g_progname = e + 1;
    }

    // check the various command line options
    std::vector<std::string> files;
    func_t function(func_t::UNDEFINED);
    for(int i(1); i < argc; ++i)
    {
        if(argv[i][0] == '-')
        {
            if(strcmp(argv[i], "--help") == 0)
            {
                usage();
            }
            if(strcmp(argv[i], "--version") == 0)
            {
                // version of the .so library
                std::cout << zipios::getVersion() << std::endl;
                exit(0);
            }
            if(strcmp(argv[i], "--version-tool") == 0)
            {
                // version of this tool (compiled with this version)
                // it should be the same as the --version
                std::cout << ZIPIOS_VERSION_STRING << std::endl;
                exit(0);
            }
            if(strcmp(argv[i], "--count") == 0)
            {
                function = func_t::COUNT;
            }
            else if(strcmp(argv[i], "--count-directories") == 0)
            {
                function = func_t::COUNT_DIRECTORIES;
            }
            else if(strcmp(argv[i], "--count-files") == 0)
            {
                function = func_t::COUNT_FILES;
            }
        }
        else
        {
            files.push_back(argv[i]);
        }
    }

    switch(function)
    {
    case func_t::COUNT:
        for(auto it(files.begin()); it != files.end(); ++it)
        {
            zipios::ZipFile zf(*it);
            if(files.size() > 1)
            {
                // write filename in case there is more than one file
                std::cout << *it << ": ";
            }
            std::cout << zf.entries().size() << std::endl;
        }
        break;

    case func_t::COUNT_DIRECTORIES:
        for(auto it(files.begin()); it != files.end(); ++it)
        {
            zipios::ZipFile zf(*it);
            if(files.size() > 1)
            {
                // write filename in case there is more than one file
                std::cout << *it << ": ";
            }
            int count(0);
            zipios::FileEntry::vector_t entries(zf.entries());
            for(auto entry(entries.begin()); entry != entries.end(); ++entry)
            {
                if((*entry)->isDirectory())
                {
                    ++count;
                }
            }
            std::cout << count << std::endl;
        }
        break;

    case func_t::COUNT_FILES:
        for(auto it(files.begin()); it != files.end(); ++it)
        {
            zipios::ZipFile zf(*it);
            if(files.size() > 1)
            {
                // write filename in case there is more than one file
                std::cout << *it << ": ";
            }
            int count(0);
            zipios::FileEntry::vector_t entries(zf.entries());
            for(auto entry(entries.begin()); entry != entries.end(); ++entry)
            {
                if(!(*entry)->isDirectory())
                {
                    ++count;
                }
            }
            std::cout << count << std::endl;
        }
        break;

    default:
        std::cerr << g_progname << ":error: undefined function." << std::endl;
        usage();
        break;

    }

    return 0;
}


// Local Variables:
// mode: cpp
// indent-tabs-mode: nil
// c-basic-offset: 4
// tab-width: 4
// End:

// vim: ts=4 sw=4 et