lnav/src/readline_curses.hh

/**
 * @file readline_curses.hh
 */

#ifndef __readline_curses_hh
#define __readline_curses_hh

#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/ioctl.h>

#include <map>
#include <set>
#include <string>
#include <vector>
#include <exception>

#include <readline/readline.h>
#include <readline/history.h>

#include "auto_fd.hh"
#include "vt52_curses.hh"

class readline_context {
public:
    typedef std::string (*command_t)(std::string cmdline,
				     std::vector<std::string> &args);
    typedef std::map<std::string, command_t> command_map_t;

    readline_context(command_map_t *commands = NULL)
    {
	if (commands != NULL) {
	    command_map_t::iterator iter;

	    for (iter = commands->begin(); iter != commands->end(); iter++) {
		std::string cmd = iter->first;

		this->rc_possibilities["__command"].insert(cmd);
		iter->second(cmd, this->rc_prototypes[cmd]);
	    }
	}
	memset(&this->rc_history, 0, sizeof(this->rc_history));
    };

    void load(void)
    {
	loaded_context = this;
	rl_attempted_completion_function = attempted_completion;
	history_set_history_state(&this->rc_history);
    };

    void save(void)
    {
	HISTORY_STATE *hs = history_get_history_state();

	this->rc_history = *hs;
	free(hs);
	hs = NULL;
    };

    void add_possibility(std::string type, std::string value)
    {
	this->rc_possibilities[type].insert(value);
	fprintf(stderr, "pos %d %p %s %s\n",
		this->rc_possibilities[type].size(),
		&this->rc_possibilities[type],
		type.c_str(),
		value.c_str());
    };

    void rem_possibility(std::string type, std::string value)
    {
	this->rc_possibilities[type].erase(value);
    };

private:
    static char **attempted_completion(const char *text, int start, int end);
    static char *completion_generator(const char *text, int state);

    static readline_context *loaded_context;
    static std::            set<std::string> *arg_possibilities;

    HISTORY_STATE rc_history;
    std::map<std::string, std::set<std::string> >    rc_possibilities;
    std::map<std::string, std::vector<std::string> > rc_prototypes;
};

class readline_curses
    : public vt52_curses {
public:
    typedef view_action<readline_curses> action;

    class error
	: public std::exception {
public:
	error(int err)
	    : e_err(err) { };

	int e_err;
    };

    static const int KEY_TIMEOUT = 750 * 1000;

    readline_curses();
    virtual ~readline_curses();

    void add_context(int id, readline_context &rc)
    {
	this->rc_contexts[id] = &rc;
    };

    void set_perform_action(action va) { this->rc_perform = va; };
    void set_timeout_action(action va) { this->rc_timeout = va; };

    void set_value(std::string value) { this->rc_value = value; };
    std::string get_value() { return this->rc_value; };

    int update_fd_set(fd_set &readfds)
    {
	FD_SET(this->rc_pty[RCF_MASTER], &readfds);
	FD_SET(this->rc_command_pipe[RCF_MASTER], &readfds);

	return std::max(this->rc_pty[RCF_MASTER].get(),
			this->rc_command_pipe[RCF_MASTER].get());
    };

    void handle_key(int ch);

    void check_fd_set(fd_set &ready_rfds);

    void focus(int context, const char *prompt);

    void start(void);

    void do_update(void);

    void window_change(void)
    {
	struct winsize ws;

	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == -1) {
	    throw error(errno);
	}
	if (ioctl(this->rc_pty[RCF_MASTER], TIOCSWINSZ, &ws) == -1) {
	    throw error(errno);
	}
    };

    void line_ready(const char *line);

    void add_possibility(int context, std::string type, std::string value);
    void rem_possibility(int context, std::string type, std::string value);

private:
    enum {
	RCF_MASTER,
	RCF_SLAVE,

	RCF_MAX_VALUE,
    };

    int     rc_active_context;
    pid_t   rc_child;
    auto_fd rc_pty[2];
    auto_fd rc_command_pipe[2];
    std::map<int, readline_context *> rc_contexts;
    std::string rc_value;

    action rc_perform;
    action rc_timeout;
};

#endif