// :mode=c++:
/*
encode.h - c++ wrapper for a base64 encoding algorithm

This is part of the libb64 project, and has been placed in the public domain.
For details, see http://sourceforge.net/projects/libb64
*/
#ifndef BASE64_ENCODE_H
#define BASE64_ENCODE_H

#include <iostream>
#include <sstream>

#define BUFFERSIZE 8192

namespace base64 {
extern "C" {
#include "cencode.h"
}

struct encoder {
	base64_encodestate _state;
	int _buffersize;

	encoder(int buffersize_in = BUFFERSIZE) : _buffersize(buffersize_in) {}

	int encode(char value_in) { return base64_encode_value(value_in); }

	int encode(const char* code_in, const int length_in, char* plaintext_out) {
		return base64_encode_block(code_in, length_in, plaintext_out, &_state);
	}

	int encode_end(char* plaintext_out) { return base64_encode_blockend(plaintext_out, &_state); }

	void encode(std::istream& istream_in, std::ostream& ostream_in) {
		base64_init_encodestate(&_state);
		//
		const int N = _buffersize;
		char* plaintext = new char[N];
		char* code = new char[2 * N];
		int plainlength;
		int codelength;

		do {
			istream_in.read(plaintext, N);
			plainlength = istream_in.gcount();
			//
			codelength = encode(plaintext, plainlength, code);
			ostream_in.write(code, codelength);
		} while (istream_in.good() && plainlength > 0);

		codelength = encode_end(code);
		ostream_in.write(code, codelength);
		//
		base64_init_encodestate(&_state);

		delete[] code;
		delete[] plaintext;
	}

	static std::string from_string(std::string s) {
		std::stringstream in(s);
		std::stringstream out;
		encoder enc;
		enc.encode(in, out);
		return out.str();
	}
};

} // namespace base64

#endif // BASE64_ENCODE_H