filecollection.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-2022  Made to Order Software Corp.  All Rights Reserved

  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.1 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
*/

#include "zipios/filecollection.hpp"

#include "zipios/zipiosexceptions.hpp"

#include <algorithm>


namespace zipios
{



namespace
{

char const * g_default_filename = "-";


class MatchName
{
public:
    explicit MatchName(std::string const & name)
        : m_name(name)
    {
    }

    bool operator() (FileEntry::pointer_t entry) const
    {
        return entry->getName() == m_name;
    }

private:
    std::string const       m_name;
};


class MatchFileName
{
public:
    explicit MatchFileName(std::string const & name)
        : m_name(name)
    {
    }

    bool operator() (FileEntry::pointer_t entry) const
    {
        return entry->getFileName() == m_name;
    }

private:
    std::string const       m_name;
};


} // no name namespace



FileCollection::FileCollection(std::string const & filename)
    : m_filename(filename.empty() ? g_default_filename : filename)
{
}


FileCollection::FileCollection(FileCollection const & rhs)
    : m_filename(rhs.m_filename)
    , m_valid(rhs.m_valid)
{
    m_entries.reserve(rhs.m_entries.size());
    for(auto it = rhs.m_entries.begin(); it != rhs.m_entries.end(); ++it)
    {
        m_entries.push_back((*it)->clone());
    }
}


FileCollection & FileCollection::operator = (FileCollection const & rhs)
{
    if(this != &rhs)
    {
        m_filename = rhs.m_filename;

        m_entries.clear();
        m_entries.reserve(rhs.m_entries.size());
        for(auto it(rhs.m_entries.begin()); it != rhs.m_entries.end(); ++it)
        {
            m_entries.push_back((*it)->clone());
        }

        m_valid = rhs.m_valid;
    }

    return *this;
}


FileCollection::~FileCollection()
{
}


void FileCollection::addEntry(FileEntry const & entry)
{
    m_entries.push_back(entry.clone());
}


void FileCollection::close()
{
    m_entries.clear();
    m_filename = g_default_filename;
    m_valid = false;
}


FileEntry::vector_t FileCollection::entries() const
{
    mustBeValid();

    return m_entries;
}


FileEntry::pointer_t FileCollection::getEntry(std::string const & name, MatchPath matchpath) const
{
    // make sure the entries were loaded if necessary
    entries();

    mustBeValid();

    FileEntry::vector_t::const_iterator iter;
    if(matchpath == MatchPath::MATCH)
    {
        iter = std::find_if(m_entries.begin(), m_entries.end(), MatchName(name));
    }
    else
    {
        iter = std::find_if(m_entries.begin(), m_entries.end(), MatchFileName(name));
    }

    return iter == m_entries.end() ? FileEntry::pointer_t() : *iter;
}


std::string FileCollection::getName() const
{
    mustBeValid();
    return m_filename;
}


size_t FileCollection::size() const
{
    // make sure the entries were loaded if necessary
    entries();

    mustBeValid();
    return m_entries.size();
}


bool FileCollection::isValid() const
{
    return m_valid;
}


void FileCollection::mustBeValid() const
{
    if(!m_valid)
    {
        throw InvalidStateException("Attempted to access an invalid FileCollection");
    }
}


void FileCollection::setMethod(
      std::size_t limit
    , StorageMethod small_storage_method
    , StorageMethod large_storage_method)
{
    // make sure the entries were loaded if necessary
    entries();

    mustBeValid();

    for(auto it(m_entries.begin()); it != m_entries.end(); ++it)
    {
        if((*it)->getSize() > limit)
        {
            (*it)->setMethod(large_storage_method);
        }
        else
        {
            (*it)->setMethod(small_storage_method);
        }
    }
}


void FileCollection::setLevel(
      std::size_t limit
    , FileEntry::CompressionLevel small_compression_level
    , FileEntry::CompressionLevel large_compression_level)
{
    // make sure the entries were loaded if necessary
    entries();

    mustBeValid();

    for(auto it(m_entries.begin()); it != m_entries.end(); ++it)
    {
        if((*it)->getSize() > limit)
        {
            (*it)->setLevel(large_compression_level);
        }
        else
        {
            (*it)->setLevel(small_compression_level);
        }
    }
}


std::ostream & operator << (std::ostream & os, FileCollection const & collection)
{
    os << "collection '" << collection.getName() << "' {";
    FileEntry::vector_t entries(collection.entries());
    char const *sep("");
    for(auto it = entries.begin(); it != entries.end(); ++it)
    {
        os << sep;
        sep = ", ";
        os << (*it)->getName();
    }
    os << "}";
    return os;
}


} // zipios namespace

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

// vim: ts=4 sw=4 et