PartCheckerStream.h

Go to the documentation of this file.

/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/SmingHub/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * PartCheckerStream.h
 *
 * @author: 2021 - Slavey Karadzhov <slav@attachix.com>
 *
 ****/

#pragma once

#include <Data/Stream/StreamWrapper.h>
#include <Network/Http/HttpHeaders.h>
#include <Delegate.h>

class PartCheckerStream : public StreamWrapper
{
public:
    struct FilePart {
        String name;
        String fileName;
        String mime;
        int length; // -1 if not available, >0 otherwise
    };

    using CheckerCallback =
        Delegate<bool(const HttpHeaders& headers, ReadWriteStream* source, const FilePart& filePart)>;

    PartCheckerStream(ReadWriteStream* source, CheckerCallback callback) : StreamWrapper(source), callback(callback)
    {
    }

    bool checkHeaders(const HttpHeaders& headers, const FilePart& part)
    {
        save = callback(headers, getSource(), part);
        return save;
    }

    bool isValid() const override
    {
        return save;
    }

    StreamType getStreamType() const override
    {
        return eSST_HeaderChecker;
    }

    size_t write(const uint8_t* buffer, size_t size) override
    {
        if(!save) {
            debug_d("Discarding %d bytes", size);
            return size;
        }

        return getSource()->write(buffer, size);
    }

    uint16_t readMemoryBlock(char* data, int bufSize) override
    {
        return getSource()->readMemoryBlock(data, bufSize);
    }

    bool isFinished() override
    {
        return getSource()->isFinished();
    }

private:
    bool save{true};
    CheckerCallback callback;
};