/*
 * Copyright (c) 2010-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <Swiften/Base/API.h>
#include <Swiften/FileTransfer/WriteBytestream.h>

namespace Swift {
    class SWIFTEN_API ByteArrayWriteBytestream : public WriteBytestream {
        public:
            ByteArrayWriteBytestream() {
            }

            virtual bool write(const std::vector<unsigned char>& bytes) {
                data.insert(data.end(), bytes.begin(), bytes.end());
                onWrite(bytes);
                return true;
            }

            const std::vector<unsigned char>& getData() const {
                return data;
            }

        private:
            std::vector<unsigned char> data;
    };
}