Utils/Console/LineEditor.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2009
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief LineEditor non-inline non-template implementation */
  25
  26 #include "LineEditor.hh"
  27 //#include "LineEditor.ih"
  28
  29 // Custom includes
  30 #include "../Logger/SenfLog.hh"
  31 #include "../../Utils/range.hh"
  32
  33 //#include "LineEditor.mpp"
  34 #define prefix_
  35 ///////////////////////////////cc.p////////////////////////////////////////
  36
  37 ///////////////////////////////////////////////////////////////////////////
  38 // senf::console::detail::LineEditorSwitcher
  39
  40 prefix_ senf::console::detail::LineEditorSwitcher::LineEditorSwitcher(Client & client)
  41     : ClientReader(client),
  42       reader_ (new LineEditorClientReader(client, *this))
  43 {}
  44
  45 prefix_ void senf::console::detail::LineEditorSwitcher::editorSetupFailed()
  46 {
  47     // We need to delete the old reader *before* creating the new one such that all old scheduler
  48     // events are removed before installing the new ones.
  49     reader_.reset(0);
  50     reader_.reset(new DumbClientReader(client()));
  51 }
  52
  53 prefix_ void senf::console::detail::LineEditorSwitcher::v_disablePrompt()
  54 {
  55     reader_->disablePrompt();
  56 }
  57
  58 prefix_ void senf::console::detail::LineEditorSwitcher::v_enablePrompt()
  59 {
  60     reader_->enablePrompt();
  61 }
  62
  63 prefix_ void senf::console::detail::LineEditorSwitcher::v_write(std::string const & data)
  64 {
  65     reader_->write(data);
  66 }
  67
  68 prefix_ unsigned senf::console::detail::LineEditorSwitcher::v_width()
  69     const
  70 {
  71     return reader_->width();
  72 }
  73
  74 ///////////////////////////////////////////////////////////////////////////
  75 // senf::console::detail::LineEditorClientReader
  76
  77 prefix_ senf::console::detail::LineEditorClientReader::
  78 LineEditorClientReader(Client & client, LineEditorSwitcher & switcher)
  79     : term::BaseTelnetProtocol(client.handle()), ClientReader(client),
  80       editor_ (*this, senf::membind(&LineEditorClientReader::executeLine, this)),
  81       switcher_ (&switcher)
  82 {
  83     editor_.prompt(promptString());
  84     editor_.defineKey(senf::term::KeyParser::Ctrl('D'),
  85                       senf::membind(&LineEditorClientReader::deleteCharOrExit, this));
  86     editor_.defineKey(senf::term::KeyParser::Tab,
  87                       boost::bind(&term::bindings::complete,
  88                                   _1,
  89                                   senf::membind(&LineEditorClientReader::completePath, this)));
  90 }
  91
  92 prefix_ void senf::console::detail::LineEditorClientReader::v_setupFailed()
  93 {
  94     // Commits suicide
  95     switcher_->editorSetupFailed();
  96 }
  97
  98 prefix_ void senf::console::detail::LineEditorClientReader::v_eof()
  99 {
 100     stopClient();
 101 }
 102
 103 prefix_ void senf::console::detail::LineEditorClientReader::v_disablePrompt()
 104 {
 105     editor_.hide();
 106 }
 107
 108 prefix_ void senf::console::detail::LineEditorClientReader::v_enablePrompt()
 109 {
 110     editor_.show();
 111 }
 112
 113 prefix_ void senf::console::detail::LineEditorClientReader::v_write(std::string const & data)
 114 {
 115     BaseTelnetProtocol::write(data);
 116 }
 117
 118 prefix_ unsigned senf::console::detail::LineEditorClientReader::v_width()
 119     const
 120 {
 121     return editor_.width();
 122 }
 123
 124 prefix_ void
 125 senf::console::detail::LineEditorClientReader::executeLine(std::string const & text)
 126 {
 127     handleInput(text);
 128     stream() << std::flush;
 129     editor_.prompt(promptString());
 130     editor_.show();
 131 }
 132
 133 prefix_ void
 134 senf::console::detail::LineEditorClientReader::deleteCharOrExit(term::LineEditor & editor)
 135 {
 136     if (editor.text().empty())
 137         ClientReader::handle().facet<TCPSocketProtocol>().shutdown(TCPSocketProtocol::ShutRD);
 138     else
 139         term::bindings::deleteChar(editor);
 140 }
 141
 142 prefix_ void senf::console::detail::LineEditorClientReader::
 143 completePath(term::LineEditor & editor, unsigned b, unsigned e,
 144              std::vector<std::string> & completions)
 145 {
 146     std::string base (editor.text().substr(b,e));
 147     CommandParser parser;
 148     ParseCommandInfo cmd;
 149     try {
 150         parser.parsePath(base, cmd);
 151     }
 152     catch (CommandParser::ParserErrorException & ex) {
 153         return;
 154     }
 155
 156     ParseCommandInfo::TokensRange path (cmd.commandPath());
 157     if (path.empty()) {
 158         DirectoryNode::ChildrenRange cs (client().cwd().children());
 159         for (DirectoryNode::ChildrenRange::iterator i (cs.begin()); i != cs.end(); ++i)
 160             completions.push_back(i->first + (i->second->followLink().isDirectory() ? "/" : " "));
 161         return;
 162     }
 163
 164     ParseCommandInfo::TokensRange::const_iterator i (path.begin());
 165     ParseCommandInfo::TokensRange::const_iterator const i_end (boost::prior(path.end()));
 166     DirectoryNode * dir (& client().cwd());
 167     std::string basePath;
 168     for (; i != i_end; ++i)
 169         if (*i == NoneToken()) {
 170             if (i == path.begin()) {
 171                 dir = & client().root();
 172                 basePath = "/";
 173             }
 174         }
 175         else if (*i == WordToken("..")) {
 176             DirectoryNode * parent (dir->parent().get());
 177             if (parent) dir = parent;
 178             basePath += "../";
 179         }
 180         else if (*i == WordToken("."))
 181             basePath += "./";
 182         else {
 183             if (dir->hasChild(i->value())) {
 184                 try {
 185                     dir = & dir->getDirectory(i->value());
 186                 }
 187                 catch (std::bad_cast &) {
 188                     return;
 189                 }
 190                 basePath += i->value() + "/";
 191             }
 192             else {
 193                 DirectoryNode::ChildrenRange cs (dir->completions(i->value()));
 194                 if (has_one_elt(cs)) {
 195                     GenericNode & node (cs.begin()->second->followLink());
 196                     if (!node.isDirectory())
 197                         return;
 198                     dir = static_cast<DirectoryNode*>(&node);
 199                     basePath += cs.begin()->first + "/";
 200                 }
 201                 else
 202                     return;
 203             }
 204         }
 205
 206     DirectoryNode::ChildrenRange cs (dir->completions(i->value()));
 207     for (DirectoryNode::ChildrenRange::iterator j (cs.begin()); j != cs.end(); ++j)
 208         completions.push_back(basePath + j->first
 209                               + (j->second->followLink().isDirectory() ? "/" : " "));
 210 }
 211
 212 ///////////////////////////////cc.e////////////////////////////////////////
 213 #undef prefix_
 214 //#include "LineEditor.mpp"
 215
 216 \f
 217 // Local Variables:
 218 // mode: c++
 219 // fill-column: 100
 220 // comment-column: 40
 221 // c-file-style: "senf"
 222 // indent-tabs-mode: nil
 223 // ispell-local-dictionary: "american"
 224 // compile-command: "scons -u test"
 225 // End: