encoder.hpp

#pragma once

#include "berstream.hpp"
#include "bytestream.hpp"

#include <string>
#include <vector>

#include <cstring>

namespace dangerous { namespace snmp {

class Encoder {
public:
    template <typename EncodingClass>
    static std::string encodeToString( typename EncodingClass::value_type& value ) {
        ByteStream byteStream;
        BerStream berStream( &byteStream );

        // Encode the value.
        bool success = berStream.write<EncodingClass>( value );
        if( ! success ) {
            throw Exception( std::string("Could not encode ") + EncodingClass::NAME );
        }

        std::string returnValue;
        // Copy the encoded data from the byte stream to the string.
        byteStream.copyTo( returnValue );

        return returnValue;
    }

    template <typename EncodingClass>
    static std::vector<char> encodeToVector( typename EncodingClass::value_type& value ) {
        ByteStream byteStream;
        BerStream berStream( &byteStream );

        // Encode the value.
        bool success = berStream.write<EncodingClass>( value );
        if( ! success ) {
            throw Exception( std::string("Could not encode ") + EncodingClass::NAME );
        }

        std::vector<char> returnValue;
        // Copy the encoded data from the byte stream to the vector.
        byteStream.copyTo( returnValue );

        return returnValue;
    }
};

} }