cadabra2-cli.hh

Go to the documentation of this file.

#include <string>
#include <memory>
 
// Work around MSVC linking problem
#ifdef _DEBUG
#define CADABRA_CLI_DEBUG_MARKER
#undef _DEBUG
#endif
//#include <Python.h>
#include <pybind11/pybind11.h>
#include <pybind11/embed.h>
#ifdef CADABRA_CLI_DEBUG_MARKER
#define _DEBUG
#undef CADABRA_CLI_DEBUG_MARKER
#endif
 
class Shell
{
public:
    enum class Flags : unsigned int
    {
        None             = 0x00,
        NoBanner         = 0x01,
        IgnoreSemicolons = 0x02,
        NoColour         = 0x04,
        NoReadline       = 0x08,
    };
 
    Shell(Flags flags);
    ~Shell();
 
    void restart();
    void interact();
    pybind11::object evaluate(const std::string& code, const std::string& filename = "<stdin>");
    void execute(const std::string& code, const std::string& filename = "<stdin>");
    void execute_file(const std::string& filename, bool preprocess = true);
    void interact_file(const std::string& filename, bool preprocess = true);
 
    void write_stdout(const std::string& text, const std::string& end = "\n", bool flush = false);
    void write_stderr(const std::string& text, const std::string& end = "\n", bool flush = false);
 
private:
    void set_histfile();
    std::string histfile;
    std::string site_path;
 
    std::string str(const pybind11::handle& obj);
    std::string repr(const pybind11::handle& obj);
    std::string sanitize(std::string s);
 
    void process_ps1(const std::string& line);
    void process_ps2(const std::string& line);
    void set_completion_callback(const char* buffer, std::vector<std::string>& completions);
 
    std::string get_ps1();
    std::string get_ps2();
 
    void handle_error();
    void handle_error(pybind11::error_already_set& err);
 
    pybind11::dict globals;
    pybind11::object sys;
    pybind11::object py_stdout, py_stderr;
    std::string collect;
 
    const char* colour_error;
    const char* colour_warning;
    const char* colour_info;
    const char* colour_success;
    const char* colour_reset;
    Flags flags;
    };
 
class ExitRequest : public std::exception
{
public:
    ExitRequest();
    ExitRequest(int code);
    ExitRequest(const std::string& message);
 
    virtual const char* what() const noexcept override;
 
    int code;
    std::string message;
};
 
 
Shell::Flags& operator |= (Shell::Flags& lhs, Shell::Flags rhs);
Shell::Flags operator | (Shell::Flags lhs, Shell::Flags rhs);
bool operator & (Shell::Flags lhs, Shell::Flags rhs);