167 lines · 4.0 KB · C++
RawDownload
#include "config.h" #include <algorithm>#include <functional> #include <dirent.h>#include <sys/stat.h>#include <sys/types.h>#include <torrent/utils/log.h> #include "path_input.h" namespace { template <typename _Value>struct compare_base {  bool operator () (const _Value& complete, const _Value& base) const {    return !complete.compare(0, base.size(), base);  }}; // Count the number of elements from the start of the containers to// the first inequal element.template <typename _InputIter1, typename _InputIter2>typename std::iterator_traits<_InputIter1>::difference_typecount_base(_InputIter1 __first1, _InputIter1 __last1,	   _InputIter2 __first2, _InputIter2 __last2) {   typename std::iterator_traits<_InputIter1>::difference_type __n = 0;   for ( ;__first1 != __last1 && __first2 != __last2; ++__first1, ++__first2, ++__n)    if (*__first1 != *__first2)      return __n;   return __n;} template <typename _Return, typename _InputIter, typename _Ftor>_Returnmake_base(_InputIter __first, _InputIter __last, _Ftor __ftor) {  if (__first == __last)    return "";   _Return __base = __ftor(*__first++);   for ( ;__first != __last; ++__first) {    auto __pos = count_base(__base.begin(), __base.end(),                            __ftor(*__first).begin(), __ftor(*__first).end());     if (__pos < (typename std::iterator_traits<_InputIter>::difference_type)__base.size())      __base.resize(__pos);  }   return __base;} } namespace input { boolPathInput::pressed(int key) {  // Consider binding tab in m_bindings instead.   if (key != '\t') {    m_showNext = false;    return TextInput::pressed(key);   } else if (m_showNext) {    for (auto& itr : m_signal_show_next)      itr();   } else {    receive_do_complete();  }   return true;} voidPathInput::receive_do_complete() {  lt_log_print(torrent::LOG_UI_EVENTS, "path_input: received completion");   size_type dirEnd = find_last_delim();   utils::Directory dir(dirEnd != 0 ? str().substr(0, dirEnd) : "./");   if (!dir.update(utils::Directory::update_sort | utils::Directory::update_hide_dot) || dir.empty()) {    mark_dirty();     return;  }   for (auto& entry : dir) {#ifdef __sun__    if (entry.s_type & S_IFDIR)#else    if (entry.s_type == DT_DIR)#endif      entry.s_name += '/';  }   range_type r = find_incomplete(dir, str().substr(dirEnd, get_pos() - dirEnd));   if (r.first == r.second)    return; // Show some nice colors here.   std::string base = make_base<std::string>(r.first, r.second, [](auto& de) { return de.s_name; });   // Clear the path after the cursor to make this code cleaner. It's  // not really nessesary to add the complexity just because someone  // might start tab-completeing in the middle of a path.  str().resize(dirEnd);  str().insert(dirEnd, base);   set_pos(dirEnd + base.size());   mark_dirty();   // Only emit if there are more than one option.  m_showNext = ++utils::Directory::iterator(r.first) != r.second;   if (m_showNext) {    lt_log_print(torrent::LOG_UI_EVENTS, "path_input: show next page");     for (auto& itr : m_signal_show_range)      itr(r.first, r.second);  }} PathInput::size_typePathInput::find_last_delim() {  size_type r = str().rfind('/', get_pos());   if (r == npos)    return 0;  else if (r == size())    return r;  else    return r + 1;} inline boolfind_complete_compare(const utils::directory_entry& complete, const std::string& base) {  return complete.s_name.compare(0, base.size(), base);} inline boolfind_complete_not_compare(const utils::directory_entry& complete, const std::string& base) {  return !complete.s_name.compare(0, base.size(), base);} PathInput::range_typePathInput::find_incomplete(utils::Directory& d, const std::string& f) {  range_type r;   r.first  = std::find_if(d.begin(), d.end(), [f](const utils::directory_entry& de) {       return find_complete_not_compare(de, f);   });  r.second = std::find_if(r.first,   d.end(), [f](const utils::directory_entry& de) {       return find_complete_compare(de, f);   });   return r;} }