code_ostream.hpp

//
// Created by gabriele on 14/11/18.
//

#ifndef FORBCC_CODE_OSSTREAM_H
#define FORBCC_CODE_OSSTREAM_H

#include <ostream>
#include <sstream>

namespace forbcc {

    class code_ostream : public std::ostream {

        /* ********************************************** INNER CLASS *********************************************** */

        class code_stringbuf : public std::stringbuf {

            /* ********************************************* ATTRIBUTES ********************************************* */
            std::ostream &output;

            int indentation = 0;

            /* ******************************************** CONSTRUCTORS ******************************************** */
        public:
            explicit code_stringbuf(std::ostream &output) : output(output) {};

            /**********************************************************************************************************/

            code_stringbuf(code_stringbuf &&) = default;

            code_stringbuf &operator=(code_stringbuf &&) = delete;

            code_stringbuf(const code_stringbuf &) = delete;

            code_stringbuf &operator=(const code_stringbuf &) = delete;

            /**********************************************************************************************************/

            ~code_stringbuf() override {
                if (pbase() != pptr()) {
                    putOutput();
                }
            };

            int sync() override {
                putOutput();
                return 0;
            };

            void increment_indentation() { ++indentation; };

            void decrement_indentation() { --indentation; };

        private:
            void putOutput();

        };

        /* *********************************************** ATTRIBUTES *********************************************** */
    private:
        code_stringbuf buffer;

    public:
        /* ********************************************** CONSTRUCTORS ********************************************** */

        explicit code_ostream(std::ostream &output) : std::ostream(&buffer), buffer(output) {};

        /**************************************************************************************************************/

        ~code_ostream() override = default;

        code_ostream(code_ostream &&) = delete;

        code_ostream &operator=(code_ostream &&) = delete;

        code_ostream(const code_ostream &) = delete;

        code_ostream &operator=(const code_ostream &) = delete;

        /**************************************************************************************************************/

        void increment_indentation() { buffer.increment_indentation(); };

        void decrement_indentation() { buffer.decrement_indentation(); };
    };

}


#endif //FORBCC_CODE_OSSTREAM_H