bitarchiveeditor.hpp

/*
 * bit7z - A C++ static library to interface with the 7-zip shared libraries.
 * Copyright (c) 2014-2023 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 <unordered_map>
 
#include "bitarchivewriter.hpp"
 
namespace bit7z {
 
using std::vector;
 
using EditedItems = std::unordered_map< 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( uint32_t index, const tstring& newPath );
 
        void renameItem( const tstring& oldPath, const tstring& newPath );
 
        void updateItem( uint32_t index, const tstring& inFile );
 
        void updateItem( uint32_t index, const std::vector< byte_t >& inBuffer );
 
        void updateItem( uint32_t index, std::istream& inStream );
 
        void updateItem( const tstring& itemPath, const tstring& inFile );
 
        void updateItem( const tstring& itemPath, const std::vector< byte_t >& inBuffer );
 
        void updateItem( const tstring& itemPath, istream& inStream );
 
        void deleteItem( 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 ) -> uint32_t;
 
        void checkIndex( uint32_t index );
 
        auto itemProperty( InputIndex index, BitProperty property ) const -> BitPropVariant override;
 
        auto itemStream( InputIndex index, ISequentialInStream** inStream ) const -> HRESULT override;
 
        auto hasNewData( uint32_t index ) const noexcept -> bool override;
 
        auto hasNewProperties( uint32_t index ) const noexcept -> bool override;
 
        void markItemAsDeleted( uint32_t index );
};
 
}  // namespace bit7z
 
#endif //BITARCHIVEEDITOR_HPP