bitarchiveeditor.hpp

/*
 * bit7z - A C++ static library to interface with the 7-zip shared libraries.
 * Copyright (c) Riccardo Ostani - All Rights Reserved.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */
 
#ifndef BITARCHIVEEDITOR_HPP
#define BITARCHIVEEDITOR_HPP
 
#include "bit7zlibrary.hpp"
#include "bitabstractarchivecreator.hpp"
#include "bitarchivewriter.hpp"
#include "bitdefines.hpp"
#include "bitformat.hpp"
#include "bititemsvector.hpp"
#include "bitpropvariant.hpp"
#include "bitoutputarchive.hpp"
#include "bittypes.hpp"
#include "bitwindows.hpp"
 
#include <cstdint>
#include <istream>
#include <unordered_map>
 
namespace bit7z {
 
using EditedItems = std::unordered_map< std::uint32_t, BitItemsVector::value_type >;

enum struct DeletePolicy : std::uint8_t {
    ItemOnly,    
    RecurseDirs  
};

class BIT7Z_MAYBE_UNUSED BitArchiveEditor final : public BitArchiveWriter {
    public:
        BitArchiveEditor(
            const Bit7zLibrary& lib,
            const tstring& inFile,
            const BitInOutFormat& format,
            const tstring& password = {}
        );
 
        BitArchiveEditor( const BitArchiveEditor& ) = delete;
 
        BitArchiveEditor( BitArchiveEditor&& ) = delete;
 
        auto operator=( const BitArchiveEditor& ) -> BitArchiveEditor& = delete;
 
        auto operator=( BitArchiveEditor&& ) -> BitArchiveEditor& = delete;
 
        ~BitArchiveEditor() override;

        void setUpdateMode( UpdateMode mode ) override;

        void renameItem( std::uint32_t index, const tstring& newPath );

        void renameItem( const tstring& oldPath, const tstring& newPath );

        void updateItem( std::uint32_t index, const tstring& inFile );

        void updateItem( std::uint32_t index, const buffer_t& inBuffer );

        void updateItem( std::uint32_t index, buffer_t&& inBuffer ) = delete;

        void updateItem( std::uint32_t index, std::istream& inStream );

        void updateItem( const tstring& itemPath, const tstring& inFile );

        void updateItem( const tstring& itemPath, const buffer_t& inBuffer );

        void updateItem( const tstring& itemPath, buffer_t&& inBuffer ) = delete;

        void updateItem( const tstring& itemPath, std::istream& inStream );

        void deleteItem( std::uint32_t index, DeletePolicy policy = DeletePolicy::ItemOnly );

        void deleteItem( const tstring& itemPath, DeletePolicy policy = DeletePolicy::ItemOnly );

        void applyChanges();
 
    private:
        EditedItems mEditedItems;
 
        auto findItem( const tstring& itemPath ) const -> std::uint32_t;
 
        void checkIndex( std::uint32_t index ) const;
 
        auto itemProperty( InputIndex index, BitProperty property ) const -> BitPropVariant override;
 
        auto itemStream( InputIndex index, ISequentialInStream** inStream ) const -> HRESULT override;
 
        auto hasNewData( std::uint32_t index ) const noexcept -> bool override;
 
        auto hasNewProperties( std::uint32_t index ) const noexcept -> bool override;
 
        void markItemAsDeleted( std::uint32_t index );
 
        void setEditedItem( std::uint32_t index, BitInputItem&& item );
};
 
} // namespace bit7z
 
#endif //BITARCHIVEEDITOR_HPP