wsd.cc

00001 #include "wvfdstream.h"
00002 #include "wvistreamlist.h"
00003 #include "wvstrutils.h"
00004 #include "wvunixsocket.h"
00005 #include <readline/readline.h>
00006 #include <readline/history.h>
00007 
00008 
00009 class WvReadLineStream : public WvStream
00010 {
00011     static WvReadLineStream *me;
00012     WvStream *base;
00013     WvString prompt;
00014     WvDynBuf line_buf;
00015     WvStringList commands;
00016 
00017     virtual size_t uread(void *_buf, size_t count)
00018     {
00019         size_t result = 0;
00020         char *buf = (char *)_buf;
00021         while (count > 0 && line_buf.used() > 0)
00022         {
00023             size_t chunk = line_buf.optgettable();
00024             if (chunk > count)
00025                 chunk = count;
00026             memcpy(buf, line_buf.get(chunk), chunk);
00027             count -= chunk;
00028             buf += chunk;
00029             result += chunk;
00030         }
00031         return result;
00032     }
00033 
00034     virtual size_t uwrite(const void *_buf, size_t count)
00035     {
00036         const char *buf = (const char *)_buf;
00037         for (size_t i=0; i<count; ++i)
00038         {
00039             if (buf[i] == '\n')
00040                 rl_crlf();
00041             else
00042                 rl_show_char(buf[i]);
00043         }
00044         return count;        
00045     }
00046 
00047     static void readline_callback(char *str)
00048     {
00049         if (str == NULL)
00050             return;
00051         size_t len = strlen(str);
00052         if (len == 0)
00053             return;
00054         me->line_buf.put(str, len);
00055         me->line_buf.putch('\n');
00056         add_history(str);
00057     }
00058 
00059     static int readline_getc(FILE *)
00060     {
00061         char ch;
00062         assert(me->base->read(&ch, 1) == 1);
00063         return ch;
00064     }
00065 
00066     static char *readline_command_completion_function(const char *text, int state)
00067     {
00068         static int skip = 0;
00069         if (state == 0)
00070             skip = 0;
00071         int my_skip = skip;
00072         size_t len = strlen(text);
00073         WvStringList::Iter i(me->commands);
00074         for (i.rewind(); i.next(); )
00075         {
00076             if (my_skip-- > 0)
00077                 continue;
00078             ++skip;
00079             if (i->len() >= len && strncmp(*i, text, len) == 0)
00080                 return strdup(*i);
00081         }
00082         return NULL;       
00083     }
00084 
00085     virtual void pre_select(SelectInfo &si)
00086     {
00087         if (si.wants.readable && line_buf.used() > 0)
00088             si.msec_timeout = 0;
00089         
00090         base->pre_select(si);
00091     }
00092 
00093     virtual bool post_select(SelectInfo &si)
00094     {
00095         bool now = false;
00096         if (si.wants.readable && line_buf.used() > 0)
00097             now = true;
00098 
00099         while (base->isreadable())
00100             rl_callback_read_char();
00101         return base->post_select(si) || now;
00102     }
00103 
00104 public:
00105 
00106     WvReadLineStream(WvStream *_base, WvStringParm _prompt)
00107     {
00108         base = _base;
00109         prompt = _prompt;
00110 
00111         assert(!me);
00112         me = this;
00113         set_wsname("readline on %s", base->wsname());
00114         rl_already_prompted = 1;
00115         rl_completion_entry_function = readline_command_completion_function;
00116         rl_callback_handler_install(prompt, readline_callback);
00117         rl_getc_function = readline_getc;
00118     }
00119 
00120     ~WvReadLineStream()
00121     {
00122         rl_getc_function = NULL;
00123         rl_callback_handler_remove();
00124         me = NULL;
00125     }
00126 
00127     virtual bool isok() const
00128     {
00129         return WvStream::isok() && base->isok();
00130     }
00131 
00132     void display_prompt()
00133     {
00134         base->print("%s", prompt);
00135         rl_already_prompted = 1;
00136     }
00137 
00138     void set_commands(const WvStringList &_commands)
00139     {
00140         commands.zap();
00141         WvStringList::Iter i(_commands);
00142         for (i.rewind(); i.next(); )
00143             commands.append(*i);
00144     }
00145 
00146     const char *wstype() const { return "WvReadLineStream"; }
00147 };
00148 
00149 
00150 WvReadLineStream *WvReadLineStream::me = NULL;
00151 
00152 
00153 void remote_cb(WvStream &remote, void *_local)
00154 {
00155     WvReadLineStream &local = *(WvReadLineStream *)_local;
00156 
00157     const char *line = remote.getline();
00158     if (line == NULL)
00159         return;
00160 
00161     WvStringList words;
00162     wvtcl_decode(words, line);
00163 
00164     WvString first = words.popstr();
00165     bool last_line = !!first && first != "-";
00166     if (last_line)
00167         local.print("%s ", first);
00168     local.print("%s\n", words.join(" "));
00169     if (last_line)
00170         local.display_prompt();
00171 
00172     if (words.popstr() == "Commands availible:")
00173         local.set_commands(words);
00174 }
00175 
00176 
00177 void local_cb(WvStream &_local, void  *_remote)
00178 {
00179     WvReadLineStream &local = (WvReadLineStream &)_local;
00180     WvStream &remote = *(WvStream *)_remote;
00181 
00182     const char *line = local.getline();
00183     if (line == NULL)
00184         return;
00185 
00186     if (strcmp(line, "quit") == 0)
00187         remote.close();
00188 
00189     remote.print("%s\n", line);
00190 }
00191 
00192 
00193 int main(int argc, char **argv)
00194 {
00195     WvReadLineStream readlinestream(wvcon, "> ");
00196 
00197     const char *sockname = "/tmp/weaver.wsd";
00198     if (argc >= 2)
00199         sockname = argv[1];
00200 
00201     WvUnixConn *s = new WvUnixConn(sockname);
00202     if (!s->isok())
00203     {
00204         wverr->print("Failed to connect to %s: %s\n",
00205                 sockname, s->errstr());
00206         return 1;
00207     }
00208     s->set_wsname("%s", sockname);
00209     s->print("help\n");
00210 
00211     s->setcallback(remote_cb, &readlinestream);
00212     WvIStreamList::globallist.append(s, true);
00213 
00214     readlinestream.setcallback(local_cb, s);
00215     WvIStreamList::globallist.append(&readlinestream, false);
00216 
00217     while (s->isok() && readlinestream.isok())
00218         WvIStreamList::globallist.runonce();
00219 
00220     return 0;
00221 }
00222 

Generated on Thu Jan 24 16:50:55 2008 for WvStreams by  doxygen 1.5.4