aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/complete.cpp
diff options
context:
space:
mode:
authorGravatar David Adam <zanchey@ucc.gu.uwa.edu.au>2015-07-26 10:20:13 +0800
committerGravatar David Adam <zanchey@ucc.gu.uwa.edu.au>2015-07-26 10:20:13 +0800
commit3929e9de0e69666b37df87347d5ce15663e81347 (patch)
treeb2701c439c0260840ce1c68beaebf7de1178cc53 /src/complete.cpp
parent793e1afa084982dac92c4fe19e50c25e326a79c2 (diff)
parentf4d1657c22c81a7720a91026f915b80d2d6aa6e8 (diff)
Merge branch 'master' into iwyu
Diffstat (limited to 'src/complete.cpp')
-rw-r--r--src/complete.cpp2287
1 files changed, 2287 insertions, 0 deletions
diff --git a/src/complete.cpp b/src/complete.cpp
new file mode 100644
index 00000000..877ac984
--- /dev/null
+++ b/src/complete.cpp
@@ -0,0 +1,2287 @@
+/** \file complete.c Functions related to tab-completion.
+
+ These functions are used for storing and retrieving tab-completion data, as well as for performing tab-completion.
+*/
+#include "config.h"
+
+#include <assert.h>
+#include <stdlib.h>
+#include <stddef.h>
+#include <wchar.h>
+#include <wctype.h>
+#include <pwd.h>
+#include <pthread.h>
+#include <algorithm>
+#include <list>
+#include <map>
+#include <set>
+#include <string>
+#include <utility>
+
+#include "fallback.h" // IWYU pragma: keep
+#include "util.h"
+
+#include "wildcard.h"
+#include "proc.h"
+#include "parser.h"
+#include "function.h"
+#include "complete.h"
+#include "builtin.h"
+#include "env.h"
+#include "exec.h"
+#include "expand.h"
+#include "common.h"
+#include "parse_util.h"
+#include "wutil.h"
+#include "path.h"
+#include "parse_tree.h"
+#include "iothread.h"
+#include "autoload.h"
+#include "parse_constants.h"
+
+/*
+ Completion description strings, mostly for different types of files, such as sockets, block devices, etc.
+
+ There are a few more completion description strings defined in
+ expand.c. Maybe all completion description strings should be defined
+ in the same file?
+*/
+
+/**
+ Description for ~USER completion
+*/
+#define COMPLETE_USER_DESC _( L"Home for %ls" )
+
+/**
+ Description for short variables. The value is concatenated to this description
+*/
+#define COMPLETE_VAR_DESC_VAL _( L"Variable: %ls" )
+
+/**
+ The maximum number of commands on which to perform description
+ lookup. The lookup process is quite time consuming, so this should
+ be set to a pretty low number.
+*/
+#define MAX_CMD_DESC_LOOKUP 10
+
+/**
+ Condition cache value returned from hashtable when this condition
+ has not yet been tested. This value is NULL, so that when the hash
+ table returns NULL, this wil be seen as an untested condition.
+*/
+#define CC_NOT_TESTED 0
+
+/**
+ Condition cache value returned from hashtable when the condition is
+ met. This can be any value, that is a valid pointer, and that is
+ different from CC_NOT_TESTED and CC_FALSE.
+*/
+#define CC_TRUE L"true"
+
+/**
+ Condition cache value returned from hashtable when the condition is
+ not met. This can be any value, that is a valid pointer, and that
+ is different from CC_NOT_TESTED and CC_TRUE.
+
+*/
+#define CC_FALSE L"false"
+
+/**
+ The special cased translation macro for completions. The empty
+ string needs to be special cased, since it can occur, and should
+ not be translated. (Gettext returns the version information as the
+ response)
+*/
+#ifdef USE_GETTEXT
+static const wchar_t *C_(const wcstring &s)
+{
+ return s.empty() ? L"" : wgettext(s.c_str());
+}
+#else
+static const wcstring &C_(const wcstring &s)
+{
+ return s;
+}
+#endif
+
+/* Testing apparatus */
+const wcstring_list_t *s_override_variable_names = NULL;
+
+void complete_set_variable_names(const wcstring_list_t *names)
+{
+ s_override_variable_names = names;
+}
+
+static inline wcstring_list_t complete_get_variable_names(void)
+{
+ if (s_override_variable_names != NULL)
+ {
+ return *s_override_variable_names;
+ }
+ else
+ {
+ return env_get_names(0);
+ }
+}
+
+/**
+ Struct describing a completion option entry.
+
+ If short_opt and long_opt are both zero, the comp field must not be
+ empty and contains a list of arguments to the command.
+
+ If either short_opt or long_opt are non-zero, they specify a switch
+ for the command. If \c comp is also not empty, it contains a list
+ of non-switch arguments that may only follow directly after the
+ specified switch.
+*/
+typedef struct complete_entry_opt
+{
+ /** Short style option */
+ wchar_t short_opt;
+ /** Long style option */
+ wcstring long_opt;
+ /** Arguments to the option */
+ wcstring comp;
+ /** Description of the completion */
+ wcstring desc;
+ /** Condition under which to use the option */
+ wcstring condition;
+ /** Must be one of the values SHARED, NO_FILES, NO_COMMON,
+ EXCLUSIVE, and determines how completions should be performed
+ on the argument after the switch. */
+ int result_mode;
+ /** True if old style long options are used */
+ int old_mode;
+ /** Completion flags */
+ complete_flags_t flags;
+
+ const wcstring localized_desc() const
+ {
+ return C_(desc);
+ }
+} complete_entry_opt_t;
+
+/* Last value used in the order field of completion_entry_t */
+static unsigned int kCompleteOrder = 0;
+
+/**
+ Struct describing a command completion
+*/
+typedef std::list<complete_entry_opt_t> option_list_t;
+class completion_entry_t
+{
+public:
+ /** List of all options */
+ option_list_t options;
+
+ /** String containing all short option characters */
+ wcstring short_opt_str;
+
+public:
+
+ /** Command string */
+ const wcstring cmd;
+
+ /** True if command is a path */
+ const bool cmd_is_path;
+
+ /** True if no other options than the ones supplied are possible */
+ bool authoritative;
+
+ /** Order for when this completion was created. This aids in outputting completions sorted by time. */
+ const unsigned int order;
+
+ /** Getters for option list. */
+ const option_list_t &get_options() const;
+
+ /** Adds or removes an option. */
+ void add_option(const complete_entry_opt_t &opt);
+ bool remove_option(wchar_t short_opt, const wchar_t *long_opt, int old_mode);
+
+ /** Getter for short_opt_str. */
+ wcstring &get_short_opt_str();
+ const wcstring &get_short_opt_str() const;
+
+ completion_entry_t(const wcstring &c, bool type, const wcstring &options, bool author) :
+ short_opt_str(options),
+ cmd(c),
+ cmd_is_path(type),
+ authoritative(author),
+ order(++kCompleteOrder)
+ {
+ }
+};
+
+/** Set of all completion entries */
+struct completion_entry_set_comparer
+{
+ /** Comparison for std::set */
+ bool operator()(completion_entry_t *p1, completion_entry_t *p2) const
+ {
+ /* Paths always come last for no particular reason */
+ if (p1->cmd_is_path != p2->cmd_is_path)
+ {
+ return p1->cmd_is_path < p2->cmd_is_path;
+ }
+ else
+ {
+ return p1->cmd < p2->cmd;
+ }
+ }
+};
+typedef std::set<completion_entry_t *, completion_entry_set_comparer> completion_entry_set_t;
+static completion_entry_set_t completion_set;
+
+// Comparison function to sort completions by their order field
+static bool compare_completions_by_order(const completion_entry_t *p1, const completion_entry_t *p2)
+{
+ return p1->order < p2->order;
+}
+
+/** The lock that guards the list of completion entries */
+static pthread_mutex_t completion_lock = PTHREAD_MUTEX_INITIALIZER;
+
+/**
+ * The lock that guards the options list of individual completion entries.
+ * If both completion_lock and completion_entry_lock are to be taken,
+ * completion_lock must be taken first.
+ */
+static pthread_mutex_t completion_entry_lock = PTHREAD_MUTEX_INITIALIZER;
+
+
+void completion_entry_t::add_option(const complete_entry_opt_t &opt)
+{
+ ASSERT_IS_LOCKED(completion_entry_lock);
+ options.push_front(opt);
+}
+
+const option_list_t &completion_entry_t::get_options() const
+{
+ ASSERT_IS_LOCKED(completion_entry_lock);
+ return options;
+}
+
+wcstring &completion_entry_t::get_short_opt_str()
+{
+ ASSERT_IS_LOCKED(completion_entry_lock);
+ return short_opt_str;
+}
+
+const wcstring &completion_entry_t::get_short_opt_str() const
+{
+ ASSERT_IS_LOCKED(completion_entry_lock);
+ return short_opt_str;
+}
+
+completion_t::~completion_t()
+{
+}
+
+/* Clear the COMPLETE_AUTO_SPACE flag, and set COMPLETE_NO_SPACE appropriately depending on the suffix of the string */
+static complete_flags_t resolve_auto_space(const wcstring &comp, complete_flags_t flags)
+{
+ if (flags & COMPLETE_AUTO_SPACE)
+ {
+ flags = flags & ~COMPLETE_AUTO_SPACE;
+ size_t len = comp.size();
+ if (len > 0 && (wcschr(L"/=@:", comp.at(len-1)) != 0))
+ flags |= COMPLETE_NO_SPACE;
+ }
+ return flags;
+}
+
+/* completion_t functions. Note that the constructor resolves flags! */
+completion_t::completion_t(const wcstring &comp, const wcstring &desc, string_fuzzy_match_t mat, complete_flags_t flags_val) :
+ completion(comp),
+ description(desc),
+ match(mat),
+ flags(resolve_auto_space(comp, flags_val))
+{
+}
+
+completion_t::completion_t(const completion_t &him) : completion(him.completion), description(him.description), match(him.match), flags(him.flags)
+{
+}
+
+completion_t &completion_t::operator=(const completion_t &him)
+{
+ if (this != &him)
+ {
+ this->completion = him.completion;
+ this->description = him.description;
+ this->match = him.match;
+ this->flags = him.flags;
+ }
+ return *this;
+}
+
+bool completion_t::is_naturally_less_than(const completion_t &a, const completion_t &b)
+{
+ return wcsfilecmp(a.completion.c_str(), b.completion.c_str()) < 0;
+}
+
+bool completion_t::is_alphabetically_equal_to(const completion_t &a, const completion_t &b)
+{
+ return a.completion == b.completion;
+}
+
+
+/** Class representing an attempt to compute completions */
+class completer_t
+{
+ const completion_request_flags_t flags;
+ const wcstring initial_cmd;
+ std::vector<completion_t> completions;
+
+ /** Table of completions conditions that have already been tested and the corresponding test results */
+ typedef std::map<wcstring, bool> condition_cache_t;
+ condition_cache_t condition_cache;
+
+ enum complete_type_t
+ {
+ COMPLETE_DEFAULT,
+ COMPLETE_AUTOSUGGEST
+ };
+
+ complete_type_t type() const
+ {
+ return (flags & COMPLETION_REQUEST_AUTOSUGGESTION) ? COMPLETE_AUTOSUGGEST : COMPLETE_DEFAULT;
+ }
+
+ bool wants_descriptions() const
+ {
+ return !!(flags & COMPLETION_REQUEST_DESCRIPTIONS);
+ }
+
+ bool fuzzy() const
+ {
+ return !!(flags & COMPLETION_REQUEST_FUZZY_MATCH);
+ }
+
+ fuzzy_match_type_t max_fuzzy_match_type() const
+ {
+ /* If we are doing fuzzy matching, request all types; if not request only prefix matching */
+ return (flags & COMPLETION_REQUEST_FUZZY_MATCH) ? fuzzy_match_none : fuzzy_match_prefix_case_insensitive;
+ }
+
+
+public:
+ completer_t(const wcstring &c, completion_request_flags_t f) :
+ flags(f),
+ initial_cmd(c)
+ {
+ }
+
+ bool empty() const
+ {
+ return completions.empty();
+ }
+ const std::vector<completion_t> &get_completions(void)
+ {
+ return completions;
+ }
+
+ bool try_complete_variable(const wcstring &str);
+ bool try_complete_user(const wcstring &str);
+
+ bool complete_param(const wcstring &cmd_orig,
+ const wcstring &popt,
+ const wcstring &str,
+ bool use_switches);
+
+ void complete_param_expand(const wcstring &str, bool do_file, bool directories_only = false);
+
+ void complete_cmd(const wcstring &str,
+ bool use_function,
+ bool use_builtin,
+ bool use_command,
+ bool use_implicit_cd);
+
+ void complete_from_args(const wcstring &str,
+ const wcstring &args,
+ const wcstring &desc,
+ complete_flags_t flags);
+
+ void complete_cmd_desc(const wcstring &str);
+
+ bool complete_variable(const wcstring &str, size_t start_offset);
+
+ bool condition_test(const wcstring &condition);
+
+ void complete_strings(const wcstring &wc_escaped,
+ const wchar_t *desc,
+ wcstring(*desc_func)(const wcstring &),
+ std::vector<completion_t> &possible_comp,
+ complete_flags_t flags);
+
+ expand_flags_t expand_flags() const
+ {
+ /* Never do command substitution in autosuggestions. Sadly, we also can't yet do job expansion because it's not thread safe. */
+ expand_flags_t result = 0;
+ if (this->type() == COMPLETE_AUTOSUGGEST)
+ result |= EXPAND_SKIP_CMDSUBST;
+
+ /* Allow fuzzy matching */
+ if (this->fuzzy())
+ result |= EXPAND_FUZZY_MATCH;
+
+ return result;
+ }
+};
+
+/* Autoloader for completions */
+class completion_autoload_t : public autoload_t
+{
+public:
+ completion_autoload_t();
+ virtual void command_removed(const wcstring &cmd);
+};
+
+static completion_autoload_t completion_autoloader;
+
+/** Constructor */
+completion_autoload_t::completion_autoload_t() : autoload_t(L"fish_complete_path", NULL, 0)
+{
+}
+
+/** Callback when an autoloaded completion is removed */
+void completion_autoload_t::command_removed(const wcstring &cmd)
+{
+ complete_remove(cmd.c_str(), COMMAND, 0, 0, 0);
+}
+
+
+/** Create a new completion entry. */
+void append_completion(std::vector<completion_t> &completions, const wcstring &comp, const wcstring &desc, complete_flags_t flags, string_fuzzy_match_t match)
+{
+ /* If we just constructed the completion and used push_back, we would get two string copies. Try to avoid that by making a stubby completion in the vector first, and then copying our string in. Note that completion_t's constructor will munge 'flags' so it's important that we pass those to the constructor.
+
+ Nasty hack for #1241 - since the constructor needs the completion string to resolve AUTO_SPACE, and we aren't providing it with the completion, we have to do the resolution ourselves. We should get this resolving out of the constructor.
+ */
+ const wcstring empty;
+ completions.push_back(completion_t(empty, empty, match, resolve_auto_space(comp, flags)));
+ completion_t *last = &completions.back();
+ last->completion = comp;
+ last->description = desc;
+}
+
+/**
+ Test if the specified script returns zero. The result is cached, so
+ that if multiple completions use the same condition, it needs only
+ be evaluated once. condition_cache_clear must be called after a
+ completion run to make sure that there are no stale completions.
+*/
+bool completer_t::condition_test(const wcstring &condition)
+{
+ if (condition.empty())
+ {
+// fwprintf( stderr, L"No condition specified\n" );
+ return 1;
+ }
+
+ if (this->type() == COMPLETE_AUTOSUGGEST)
+ {
+ /* Autosuggestion can't support conditions */
+ return 0;
+ }
+
+ ASSERT_IS_MAIN_THREAD();
+
+ bool test_res;
+ condition_cache_t::iterator cached_entry = condition_cache.find(condition);
+ if (cached_entry == condition_cache.end())
+ {
+ /* Compute new value and reinsert it */
+ test_res = (0 == exec_subshell(condition, false /* don't apply exit status */));
+ condition_cache[condition] = test_res;
+ }
+ else
+ {
+ /* Use the old value */
+ test_res = cached_entry->second;
+ }
+ return test_res;
+}
+
+
+/** Search for an exactly matching completion entry. Must be called while locked. */
+static completion_entry_t *complete_find_exact_entry(const wcstring &cmd, const bool cmd_is_path)
+{
+ ASSERT_IS_LOCKED(completion_lock);
+ completion_entry_t *result = NULL;
+ completion_entry_t tmp_entry(cmd, cmd_is_path, L"", false);
+ completion_entry_set_t::iterator iter = completion_set.find(&tmp_entry);
+ if (iter != completion_set.end())
+ {
+ result = *iter;
+ }
+ return result;
+}
+
+/** Locate the specified entry. Create it if it doesn't exist. Must be called while locked. */
+static completion_entry_t *complete_get_exact_entry(const wcstring &cmd, bool cmd_is_path)
+{
+ ASSERT_IS_LOCKED(completion_lock);
+ completion_entry_t *c;
+
+ c = complete_find_exact_entry(cmd, cmd_is_path);
+
+ if (c == NULL)
+ {
+ c = new completion_entry_t(cmd, cmd_is_path, L"", false);
+ completion_set.insert(c);
+ }
+
+ return c;
+}
+
+
+void complete_set_authoritative(const wchar_t *cmd, bool cmd_is_path, bool authoritative)
+{
+ completion_entry_t *c;
+
+ CHECK(cmd,);
+ scoped_lock lock(completion_lock);
+ c = complete_get_exact_entry(cmd, cmd_is_path);
+ c->authoritative = authoritative;
+}
+
+
+void complete_add(const wchar_t *cmd,
+ bool cmd_is_path,
+ wchar_t short_opt,
+ const wchar_t *long_opt,
+ int old_mode,
+ int result_mode,
+ const wchar_t *condition,
+ const wchar_t *comp,
+ const wchar_t *desc,
+ complete_flags_t flags)
+{
+ CHECK(cmd,);
+
+ /* Lock the lock that allows us to edit the completion entry list */
+ scoped_lock lock(completion_lock);
+
+ /* Lock the lock that allows us to edit individual completion entries */
+ scoped_lock lock2(completion_entry_lock);
+
+ completion_entry_t *c;
+ c = complete_get_exact_entry(cmd, cmd_is_path);
+
+ /* Create our new option */
+ complete_entry_opt_t opt;
+ if (short_opt != L'\0')
+ {
+ int len = 1 + ((result_mode & NO_COMMON) != 0);
+
+ c->get_short_opt_str().push_back(short_opt);
+ if (len == 2)
+ {
+ c->get_short_opt_str().push_back(L':');
+ }
+ }
+
+ opt.short_opt = short_opt;
+ opt.result_mode = result_mode;
+ opt.old_mode=old_mode;
+
+ if (comp) opt.comp = comp;
+ if (condition) opt.condition = condition;
+ if (long_opt) opt.long_opt = long_opt;
+ if (desc) opt.desc = desc;
+ opt.flags = flags;
+
+ c->add_option(opt);
+}
+
+/**
+ Remove all completion options in the specified entry that match the
+ specified short / long option strings. Returns true if it is now
+ empty and should be deleted, false if it's not empty. Must be called while locked.
+*/
+bool completion_entry_t::remove_option(wchar_t short_opt, const wchar_t *long_opt, int old_mode)
+{
+ ASSERT_IS_LOCKED(completion_lock);
+ ASSERT_IS_LOCKED(completion_entry_lock);
+ if ((short_opt == 0) && (long_opt == 0))
+ {
+ this->options.clear();
+ }
+ else
+ {
+ for (option_list_t::iterator iter = this->options.begin(); iter != this->options.end();)
+ {
+ complete_entry_opt_t &o = *iter;
+ if ((short_opt && short_opt == o.short_opt) ||
+ (long_opt && long_opt == o.long_opt && old_mode == o.old_mode))
+ {
+ /* fwprintf( stderr,
+ L"remove option -%lc --%ls\n",
+ o->short_opt?o->short_opt:L' ',
+ o->long_opt );
+ */
+ if (o.short_opt)
+ {
+ wcstring &short_opt_str = this->get_short_opt_str();
+ size_t idx = short_opt_str.find(o.short_opt);
+ if (idx != wcstring::npos)
+ {
+ /* Consume all colons */
+ size_t first_non_colon = idx + 1;
+ while (first_non_colon < short_opt_str.size() && short_opt_str.at(first_non_colon) == L':')
+ first_non_colon++;
+ short_opt_str.erase(idx, first_non_colon - idx);
+ }
+ }
+
+ /* Destroy this option and go to the next one */
+ iter = this->options.erase(iter);
+ }
+ else
+ {
+ /* Just go to the next one */
+ ++iter;
+ }
+ }
+ }
+ return this->options.empty();
+}
+
+
+void complete_remove(const wchar_t *cmd,
+ bool cmd_is_path,
+ wchar_t short_opt,
+ const wchar_t *long_opt,
+ int old_mode)
+{
+ CHECK(cmd,);
+ scoped_lock lock(completion_lock);
+ scoped_lock lock2(completion_entry_lock);
+
+ completion_entry_t tmp_entry(cmd, cmd_is_path, L"", false);
+ completion_entry_set_t::iterator iter = completion_set.find(&tmp_entry);
+ if (iter != completion_set.end())
+ {
+ completion_entry_t *entry = *iter;
+ bool delete_it = entry->remove_option(short_opt, long_opt, old_mode);
+ if (delete_it)
+ {
+ /* Delete this entry */
+ completion_set.erase(iter);
+ delete entry;
+ }
+ }
+}
+
+/* Formats an error string by prepending the prefix and then appending the str in single quotes */
+static wcstring format_error(const wchar_t *prefix, const wcstring &str)
+{
+ wcstring result = prefix;
+ result.push_back(L'\'');
+ result.append(str);
+ result.push_back(L'\'');
+ return result;
+}
+
+/**
+ Find the full path and commandname from a command string 'str'.
+*/
+static void parse_cmd_string(const wcstring &str, wcstring &path, wcstring &cmd)
+{
+ if (! path_get_path(str, &path))
+ {
+ /** Use the empty string as the 'path' for commands that can not be found. */
+ path = L"";
+ }
+
+ /* Make sure the path is not included in the command */
+ size_t last_slash = str.find_last_of(L'/');
+ if (last_slash != wcstring::npos)
+ {
+ cmd = str.substr(last_slash + 1);
+ }
+ else
+ {
+ cmd = str;
+ }
+}
+
+int complete_is_valid_option(const wcstring &str,
+ const wcstring &opt,
+ wcstring_list_t *errors,
+ bool allow_autoload)
+{
+ wcstring cmd, path;
+ bool found_match = false;
+ bool authoritative = true;
+ int opt_found=0;
+ std::set<wcstring> gnu_match_set;
+ bool is_gnu_opt=false;
+ bool is_old_opt=false;
+ bool is_short_opt=false;
+ bool is_gnu_exact=false;
+ size_t gnu_opt_len=0;
+
+ if (opt.empty())
+ return false;
+
+ std::vector<bool> short_validated;
+ /*
+ Check some generic things like -- and - options.
+ */
+ switch (opt.size())
+ {
+
+ case 0:
+ case 1:
+ {
+ return true;
+ }
+
+ case 2:
+ {
+ if (opt == L"--")
+ {
+ return true;
+ }
+ break;
+ }
+ }
+
+ if (opt.at(0) != L'-')
+ {
+ if (errors)
+ errors->push_back(L"Option does not begin with a '-'");
+ return false;
+ }
+
+
+ short_validated.resize(opt.size(), 0);
+
+ is_gnu_opt = opt.at(1) == L'-';
+ if (is_gnu_opt)
+ {
+ size_t opt_end = opt.find(L'=');
+ if (opt_end != wcstring::npos)
+ {
+ gnu_opt_len = opt_end-2;
+ }
+ else
+ {
+ gnu_opt_len = opt.size() - 2;
+ }
+ }
+
+ parse_cmd_string(str, path, cmd);
+
+ /*
+ Make sure completions are loaded for the specified command
+ */
+ if (allow_autoload)
+ {
+ complete_load(cmd, false);
+ }
+
+ scoped_lock lock(completion_lock);
+ scoped_lock lock2(completion_entry_lock);
+ for (completion_entry_set_t::const_iterator iter = completion_set.begin(); iter != completion_set.end(); ++iter)
+ {
+ const completion_entry_t *i = *iter;
+ const wcstring &match = i->cmd_is_path ? path : cmd;
+
+ if (!wildcard_match(match, i->cmd))
+ {
+ continue;
+ }
+
+ found_match = true;
+
+ if (! i->authoritative)
+ {
+ authoritative = false;
+ break;
+ }
+
+ const option_list_t &options = i->get_options();
+ if (is_gnu_opt)
+ {
+ for (option_list_t::const_iterator iter = options.begin(); iter != options.end(); ++iter)
+ {
+ const complete_entry_opt_t &o = *iter;
+ if (o.old_mode)
+ {
+ continue;
+ }
+
+ if (opt.compare(2, gnu_opt_len, o.long_opt) == 0)
+ {
+ gnu_match_set.insert(o.long_opt);
+ if (opt.compare(2, o.long_opt.size(), o.long_opt))
+ {
+ is_gnu_exact = true;
+ }
+ }
+ }
+ }
+ else
+ {
+ /* Check for old style options */
+ for (option_list_t::const_iterator iter = options.begin(); iter != options.end(); ++iter)
+ {
+ const complete_entry_opt_t &o = *iter;
+
+ if (!o.old_mode)
+ continue;
+
+
+ if (opt.compare(1, wcstring::npos, o.long_opt)==0)
+ {
+ opt_found = true;
+ is_old_opt = true;
+ break;
+ }
+
+ }
+
+ if (is_old_opt)
+ break;
+
+ for (size_t opt_idx = 1; opt_idx < opt.size(); opt_idx++)
+ {
+ const wcstring &short_opt_str = i->get_short_opt_str();
+ size_t str_idx = short_opt_str.find(opt.at(opt_idx));
+ if (str_idx != wcstring::npos)
+ {
+ if (str_idx + 1 < short_opt_str.size() && short_opt_str.at(str_idx + 1) == L':')
+ {
+ /*
+ This is a short option with an embedded argument,
+ call complete_is_valid_argument on the argument.
+ */
+ const wcstring nopt = L"-" + opt.substr(1, 1);
+ short_validated.at(opt_idx) =
+ complete_is_valid_argument(str, nopt, opt.substr(2));
+ }
+ else
+ {
+ short_validated.at(opt_idx) = true;
+ }
+ }
+ }
+ }
+ }
+
+ if (authoritative)
+ {
+
+ if (!is_gnu_opt && !is_old_opt)
+ is_short_opt = 1;
+
+
+ if (is_short_opt)
+ {
+ opt_found=1;
+ for (size_t j=1; j<opt.size(); j++)
+ {
+ if (!short_validated.at(j))
+ {
+ if (errors)
+ {
+ const wcstring str = opt.substr(j, 1);
+ errors->push_back(format_error(_(L"Unknown option: "), str));
+ }
+
+ opt_found = 0;
+ break;
+ }
+
+ }
+ }
+
+ if (is_gnu_opt)
+ {
+ opt_found = is_gnu_exact || (gnu_match_set.size() == 1);
+ if (errors && !opt_found)
+ {
+ const wchar_t *prefix;
+ if (gnu_match_set.empty())
+ {
+ prefix = _(L"Unknown option: ");
+ }
+ else
+ {
+ prefix = _(L"Multiple matches for option: ");
+ }
+ errors->push_back(format_error(prefix, opt));
+ }
+ }
+ }
+
+ return (authoritative && found_match)?opt_found:true;
+}
+
+bool complete_is_valid_argument(const wcstring &str, const wcstring &opt, const wcstring &arg)
+{
+ return true;
+}
+
+
+/**
+ Copy any strings in possible_comp which have the specified prefix
+ to the completer's completion array. The prefix may contain wildcards.
+ The output will consist of completion_t structs.
+
+ There are three ways to specify descriptions for each
+ completion. Firstly, if a description has already been added to the
+ completion, it is _not_ replaced. Secondly, if the desc_func
+ function is specified, use it to determine a dynamic
+ completion. Thirdly, if none of the above are available, the desc
+ string is used as a description.
+
+ \param wc_escaped the prefix, possibly containing wildcards. The wildcard should not have been unescaped, i.e. '*' should be used for any string, not the ANY_STRING character.
+ \param desc the default description, used for completions with no embedded description. The description _may_ contain a COMPLETE_SEP character, if not, one will be prefixed to it
+ \param desc_func the function that generates a description for those completions witout an embedded description
+ \param possible_comp the list of possible completions to iterate over
+*/
+
+void completer_t::complete_strings(const wcstring &wc_escaped,
+ const wchar_t *desc,
+ wcstring(*desc_func)(const wcstring &),
+ std::vector<completion_t> &possible_comp,
+ complete_flags_t flags)
+{
+ wcstring tmp = wc_escaped;
+ if (! expand_one(tmp, EXPAND_SKIP_CMDSUBST | EXPAND_SKIP_WILDCARDS | this->expand_flags(), NULL))
+ return;
+
+ const wchar_t *wc = parse_util_unescape_wildcards(tmp.c_str());
+
+ for (size_t i=0; i< possible_comp.size(); i++)
+ {
+ wcstring temp = possible_comp.at(i).completion;
+ const wchar_t *next_str = temp.empty()?NULL:temp.c_str();
+
+ if (next_str)
+ {
+ wildcard_complete(next_str, wc, desc, desc_func, this->completions, this->expand_flags(), flags);
+ }
+ }
+
+ free((void *)wc);
+}
+
+/**
+ If command to complete is short enough, substitute
+ the description with the whatis information for the executable.
+*/
+void completer_t::complete_cmd_desc(const wcstring &str)
+{
+ ASSERT_IS_MAIN_THREAD();
+
+ const wchar_t *cmd_start;
+ int skip;
+
+ const wchar_t * const cmd = str.c_str();
+ cmd_start=wcsrchr(cmd, L'/');
+
+ if (cmd_start)
+ cmd_start++;
+ else
+ cmd_start = cmd;
+
+ /*
+ Using apropos with a single-character search term produces far
+ to many results - require at least two characters if we don't
+ know the location of the whatis-database.
+ */
+ if (wcslen(cmd_start) < 2)
+ return;
+
+ if (wildcard_has(cmd_start, 0))
+ {
+ return;
+ }
+
+ skip = 1;
+
+ for (size_t i=0; i< this->completions.size(); i++)
+ {
+ const completion_t &c = this->completions.at(i);
+
+ if (c.completion.empty() || (c.completion[c.completion.size()-1] != L'/'))
+ {
+ skip = 0;
+ break;
+ }
+
+ }
+
+ if (skip)
+ {
+ return;
+ }
+
+
+ wcstring lookup_cmd(L"__fish_describe_command ");
+ lookup_cmd.append(escape_string(cmd_start, 1));
+
+ std::map<wcstring, wcstring> lookup;
+
+ /*
+ First locate a list of possible descriptions using a single
+ call to apropos or a direct search if we know the location
+ of the whatis database. This can take some time on slower
+ systems with a large set of manuals, but it should be ok
+ since apropos is only called once.
+ */
+ wcstring_list_t list;
+ if (exec_subshell(lookup_cmd, list, false /* don't apply exit status */) != -1)
+ {
+
+ /*
+ Then discard anything that is not a possible completion and put
+ the result into a hashtable with the completion as key and the
+ description as value.
+
+ Should be reasonably fast, since no memory allocations are needed.
+ */
+ for (size_t i=0; i < list.size(); i++)
+ {
+ const wcstring &elstr = list.at(i);
+
+ const wcstring fullkey(elstr, wcslen(cmd_start));
+
+ size_t tab_idx = fullkey.find(L'\t');
+ if (tab_idx == wcstring::npos)
+ continue;
+
+ const wcstring key(fullkey, 0, tab_idx);
+ wcstring val(fullkey, tab_idx + 1);
+
+ /*
+ And once again I make sure the first character is uppercased
+ because I like it that way, and I get to decide these
+ things.
+ */
+ if (! val.empty())
+ val[0]=towupper(val[0]);
+
+ lookup[key] = val;
+ }
+
+ /*
+ Then do a lookup on every completion and if a match is found,
+ change to the new description.
+
+ This needs to do a reallocation for every description added, but
+ there shouldn't be that many completions, so it should be ok.
+ */
+ for (size_t i=0; i<this->completions.size(); i++)
+ {
+ completion_t &completion = this->completions.at(i);
+ const wcstring &el = completion.completion;
+ if (el.empty())
+ continue;
+
+ std::map<wcstring, wcstring>::iterator new_desc_iter = lookup.find(el);
+ if (new_desc_iter != lookup.end())
+ completion.description = new_desc_iter->second;
+ }
+ }
+
+}
+
+/**
+ Returns a description for the specified function, or an empty string if none
+*/
+static wcstring complete_function_desc(const wcstring &fn)
+{
+ wcstring result;
+ bool has_description = function_get_desc(fn, &result);
+ if (! has_description)
+ {
+ function_get_definition(fn, &result);
+ }
+ return result;
+}
+
+
+/**
+ Complete the specified command name. Search for executables in the
+ path, executables defined using an absolute path, functions,
+ builtins and directories for implicit cd commands.
+
+ \param cmd the command string to find completions for
+
+ \param comp the list to add all completions to
+*/
+void completer_t::complete_cmd(const wcstring &str_cmd, bool use_function, bool use_builtin, bool use_command, bool use_implicit_cd)
+{
+ /* Paranoia */
+ if (str_cmd.empty())
+ return;
+
+ std::vector<completion_t> possible_comp;
+
+ env_var_t cdpath = env_get_string(L"CDPATH");
+ if (cdpath.missing_or_empty())
+ cdpath = L".";
+
+ if (use_command)
+ {
+
+ if (expand_string(str_cmd, this->completions, ACCEPT_INCOMPLETE | EXECUTABLES_ONLY | this->expand_flags(), NULL) != EXPAND_ERROR)
+ {
+ if (this->wants_descriptions())
+ {
+ this->complete_cmd_desc(str_cmd);
+ }
+ }
+ }
+ if (use_implicit_cd)
+ {
+ (void)expand_string(str_cmd, this->completions, ACCEPT_INCOMPLETE | DIRECTORIES_ONLY | this->expand_flags(), NULL);
+ }
+ if (str_cmd.find(L'/') == wcstring::npos && str_cmd.at(0) != L'~')
+ {
+ if (use_command)
+ {
+
+ const env_var_t path = env_get_string(L"PATH");
+ if (!path.missing())
+ {
+ wcstring base_path;
+ wcstokenizer tokenizer(path, ARRAY_SEP_STR);
+ while (tokenizer.next(base_path))
+ {
+ if (base_path.empty())
+ continue;
+
+ /* Make sure the base path ends with a slash */
+ if (base_path.at(base_path.size() - 1) != L'/')
+ base_path.push_back(L'/');
+
+ wcstring nxt_completion = base_path;
+ nxt_completion.append(str_cmd);
+
+ size_t prev_count = this->completions.size();
+ if (expand_string(nxt_completion,
+ this->completions,
+ ACCEPT_INCOMPLETE | EXECUTABLES_ONLY | this->expand_flags(), NULL) != EXPAND_ERROR)
+ {
+ /* For all new completions, if COMPLETE_NO_CASE is set, then use only the last path component */
+ for (size_t i=prev_count; i< this->completions.size(); i++)
+ {
+ completion_t &c = this->completions.at(i);
+ if (c.flags & COMPLETE_REPLACES_TOKEN)
+ {
+
+ c.completion.erase(0, base_path.size());
+ }
+ }
+ }
+ }
+ if (this->wants_descriptions())
+ this->complete_cmd_desc(str_cmd);
+ }
+ }
+
+ if (use_function)
+ {
+ //function_get_names( &possible_comp, cmd[0] == L'_' );
+ wcstring_list_t names = function_get_names(str_cmd.at(0) == L'_');
+ for (size_t i=0; i < names.size(); i++)
+ {
+ append_completion(possible_comp, names.at(i));
+ }
+
+ this->complete_strings(str_cmd, 0, &complete_function_desc, possible_comp, 0);
+ }
+
+ possible_comp.clear();
+
+ if (use_builtin)
+ {
+ builtin_get_names(possible_comp);
+ this->complete_strings(str_cmd, 0, &builtin_get_desc, possible_comp, 0);
+ }
+
+ }
+}
+
+
+/**
+ Evaluate the argument list (as supplied by complete -a) and insert
+ any return matching completions. Matching is done using \c
+ copy_strings_with_prefix, meaning the completion may contain
+ wildcards. Logically, this is not always the right thing to do, but
+ I have yet to come up with a case where this matters.
+
+ \param str The string to complete.
+ \param args The list of option arguments to be evaluated.
+ \param desc Description of the completion
+ \param comp_out The list into which the results will be inserted
+*/
+void completer_t::complete_from_args(const wcstring &str,
+ const wcstring &args,
+ const wcstring &desc,
+ complete_flags_t flags)
+{
+
+ std::vector<completion_t> possible_comp;
+
+ bool is_autosuggest = (this->type() == COMPLETE_AUTOSUGGEST);
+ parser_t parser(is_autosuggest ? PARSER_TYPE_COMPLETIONS_ONLY : PARSER_TYPE_GENERAL, false /* don't show errors */);
+
+ /* If type is COMPLETE_AUTOSUGGEST, it means we're on a background thread, so don't call proc_push_interactive */
+ if (! is_autosuggest)
+ proc_push_interactive(0);
+
+ parser.expand_argument_list(args, possible_comp);
+
+ if (! is_autosuggest)
+ proc_pop_interactive();
+
+ this->complete_strings(escape_string(str, ESCAPE_ALL), desc.c_str(), 0, possible_comp, flags);
+}
+
+/**
+ Match against an old style long option
+*/
+static int param_match_old(const complete_entry_opt_t *e,
+ const wchar_t *optstr)
+{
+ return (optstr[0] == L'-') && (e->long_opt == &optstr[1]);
+}
+
+/**
+ Match a parameter
+*/
+static int param_match(const complete_entry_opt_t *e,
+ const wchar_t *optstr)
+{
+ if (e->short_opt != L'\0' &&
+ e->short_opt == optstr[1])
+ return 1;
+
+ if (!e->old_mode && (wcsncmp(L"--", optstr, 2) == 0))
+ {
+ if (e->long_opt == &optstr[2])
+ {
+ return 1;
+ }
+ }
+
+ return 0;
+}
+
+/**
+ Test if a string is an option with an argument, like --color=auto or -I/usr/include
+*/
+static wchar_t *param_match2(const complete_entry_opt_t *e,
+ const wchar_t *optstr)
+{
+ if (e->short_opt != L'\0' && e->short_opt == optstr[1])
+ return (wchar_t *)&optstr[2];
+ if (!e->old_mode && (wcsncmp(L"--", optstr, 2) == 0))
+ {
+ size_t len = e->long_opt.size();
+
+ if (wcsncmp(e->long_opt.c_str(), &optstr[2],len) == 0)
+ {
+ if (optstr[len+2] == L'=')
+ return (wchar_t *)&optstr[len+3];
+ }
+ }
+ return 0;
+}
+
+/**
+ Tests whether a short option is a viable completion
+*/
+static int short_ok(const wcstring &arg_str, wchar_t nextopt, const wcstring &allopt_str)
+{
+ const wchar_t *arg = arg_str.c_str();
+ const wchar_t *allopt = allopt_str.c_str();
+ const wchar_t *ptr;
+
+ if (arg[0] != L'-')
+ return arg[0] == L'\0';
+ if (arg[1] == L'-')
+ return 0;
+
+ if (wcschr(arg, nextopt) != 0)
+ return 0;
+
+ for (ptr = arg+1; *ptr; ptr++)
+ {
+ const wchar_t *tmp = wcschr(allopt, *ptr);
+ /* Unknown option */
+ if (tmp == 0)
+ {
+ /*fwprintf( stderr, L"Unknown option %lc", *ptr );*/
+
+ return 0;
+ }
+
+ if (*(tmp+1) == L':')
+ {
+ /* fwprintf( stderr, L"Woot %ls", allopt );*/
+ return 0;
+ }
+
+ }
+
+ return 1;
+}
+
+void complete_load(const wcstring &name, bool reload)
+{
+ completion_autoloader.load(name, reload);
+}
+
+// Performed on main thread, from background thread. Return type is ignored.
+static int complete_load_no_reload(wcstring *name)
+{
+ assert(name != NULL);
+ ASSERT_IS_MAIN_THREAD();
+ complete_load(*name, false);
+ return 0;
+}
+
+/**
+ complete_param: Given a command, find completions for the argument str of command cmd_orig with previous option popt.
+
+ Examples in format (cmd, popt, str):
+
+ echo hello world <tab> -> ("echo", "world", "")
+ echo hello world<tab> -> ("echo", "hello", "world")
+
+ Insert results into comp_out. Return true to perform file completion, false to disable it.
+*/
+struct local_options_t
+{
+ wcstring short_opt_str;
+ option_list_t options;
+};
+bool completer_t::complete_param(const wcstring &scmd_orig, const wcstring &spopt, const wcstring &sstr, bool use_switches)
+{
+ const wchar_t * const cmd_orig = scmd_orig.c_str();
+ const wchar_t * const popt = spopt.c_str();
+ const wchar_t * const str = sstr.c_str();
+
+ bool use_common=1, use_files=1;
+
+ wcstring cmd, path;
+ parse_cmd_string(cmd_orig, path, cmd);
+
+ if (this->type() == COMPLETE_DEFAULT)
+ {
+ ASSERT_IS_MAIN_THREAD();
+ complete_load(cmd, true);
+ }
+ else if (this->type() == COMPLETE_AUTOSUGGEST)
+ {
+ /* Maybe load this command (on the main thread) */
+ if (! completion_autoloader.has_tried_loading(cmd))
+ {
+ iothread_perform_on_main(complete_load_no_reload, &cmd);
+ }
+ }
+
+ /* Make a list of lists of all options that we care about */
+ std::vector<local_options_t> all_options;
+ {
+ scoped_lock lock(completion_lock);
+ scoped_lock lock2(completion_entry_lock);
+ for (completion_entry_set_t::const_iterator iter = completion_set.begin(); iter != completion_set.end(); ++iter)
+ {
+ const completion_entry_t *i = *iter;
+ const wcstring &match = i->cmd_is_path ? path : cmd;
+ if (! wildcard_match(match, i->cmd))
+ {
+ continue;
+ }
+
+ /* Copy all of their options into our list */
+ all_options.push_back(local_options_t());
+ all_options.back().short_opt_str = i->get_short_opt_str();
+ all_options.back().options = i->get_options(); //Oof, this is a lot of copying
+ }
+ }
+
+ /* Now release the lock and test each option that we captured above.
+ We have to do this outside the lock because callouts (like the condition) may add or remove completions.
+ See https://github.com/ridiculousfish/fishfish/issues/2 */
+ for (std::vector<local_options_t>::const_iterator iter = all_options.begin(); iter != all_options.end(); ++iter)
+ {
+ const option_list_t &options = iter->options;
+ use_common=1;
+ if (use_switches)
+ {
+
+ if (str[0] == L'-')
+ {
+ /* Check if we are entering a combined option and argument
+ (like --color=auto or -I/usr/include) */
+ for (option_list_t::const_iterator oiter = options.begin(); oiter != options.end(); ++oiter)
+ {
+ const complete_entry_opt_t *o = &*oiter;
+ wchar_t *arg;
+ if ((arg=param_match2(o, str))!=0 && this->condition_test(o->condition))
+ {
+ if (o->result_mode & NO_COMMON) use_common = false;
+ if (o->result_mode & NO_FILES) use_files = false;
+ complete_from_args(arg, o->comp, o->localized_desc(), o->flags);
+ }
+
+ }
+ }
+ else if (popt[0] == L'-')
+ {
+ /* Set to true if we found a matching old-style switch */
+ int old_style_match = 0;
+
+ /*
+ If we are using old style long options, check for them
+ first
+ */
+ for (option_list_t::const_iterator oiter = options.begin(); oiter != options.end(); ++oiter)
+ {
+ const complete_entry_opt_t *o = &*oiter;
+ if (o->old_mode)
+ {
+ if (param_match_old(o, popt) && this->condition_test(o->condition))
+ {
+ old_style_match = 1;
+ if (o->result_mode & NO_COMMON) use_common = false;
+ if (o->result_mode & NO_FILES) use_files = false;
+ complete_from_args(str, o->comp, o->localized_desc(), o->flags);
+ }
+ }
+ }
+
+ /*
+ No old style option matched, or we are not using old
+ style options. We check if any short (or gnu style
+ options do.
+ */
+ if (!old_style_match)
+ {
+ for (option_list_t::const_iterator oiter = options.begin(); oiter != options.end(); ++oiter)
+ {
+ const complete_entry_opt_t *o = &*oiter;
+ /*
+ Gnu-style options with _optional_ arguments must
+ be specified as a single token, so that it can
+ be differed from a regular argument.
+ */
+ if (!o->old_mode && ! o->long_opt.empty() && !(o->result_mode & NO_COMMON))
+ continue;
+
+ if (param_match(o, popt) && this->condition_test(o->condition))
+ {
+ if (o->result_mode & NO_COMMON) use_common = false;
+ if (o->result_mode & NO_FILES) use_files = false;
+ complete_from_args(str, o->comp, o->localized_desc(), o->flags);
+
+ }
+ }
+ }
+ }
+ }
+
+ if (use_common)
+ {
+
+ for (option_list_t::const_iterator oiter = options.begin(); oiter != options.end(); ++oiter)
+ {
+ const complete_entry_opt_t *o = &*oiter;
+ /*
+ If this entry is for the base command,
+ check if any of the arguments match
+ */
+
+ if (!this->condition_test(o->condition))
+ continue;
+
+
+ if ((o->short_opt == L'\0') && (o->long_opt[0]==L'\0'))
+ {
+ use_files = use_files && ((o->result_mode & NO_FILES)==0);
+ complete_from_args(str, o->comp, o->localized_desc(), o->flags);
+ }
+
+ if (wcslen(str) > 0 && use_switches)
+ {
+ /*
+ Check if the short style option matches
+ */
+ if (o->short_opt != L'\0' &&
+ short_ok(str, o->short_opt, iter->short_opt_str))
+ {
+ const wcstring desc = o->localized_desc();
+ wchar_t completion[2];
+ completion[0] = o->short_opt;
+ completion[1] = 0;
+
+ append_completion(this->completions, completion, desc, 0);
+
+ }
+
+ /*
+ Check if the long style option matches
+ */
+ if (o->long_opt[0] != L'\0')
+ {
+ int match=0, match_no_case=0;
+
+ wcstring whole_opt;
+ whole_opt.append(o->old_mode?L"-":L"--");
+ whole_opt.append(o->long_opt);
+
+ match = string_prefixes_string(str, whole_opt);
+
+ if (!match)
+ {
+ match_no_case = wcsncasecmp(str, whole_opt.c_str(), wcslen(str))==0;
+ }
+
+ if (match || match_no_case)
+ {
+ int has_arg=0; /* Does this switch have any known arguments */
+ int req_arg=0; /* Does this switch _require_ an argument */
+
+ size_t offset = 0;
+ complete_flags_t flags = 0;
+
+ if (match)
+ {
+ offset = wcslen(str);
+ }
+ else
+ {
+ flags = COMPLETE_REPLACES_TOKEN;
+ }
+
+ has_arg = ! o->comp.empty();
+ req_arg = (o->result_mode & NO_COMMON);
+
+ if (!o->old_mode && (has_arg && !req_arg))
+ {
+
+ /*
+ Optional arguments to a switch can
+ only be handled using the '=', so we
+ add it as a completion. By default
+ we avoid using '=' and instead rely
+ on '--switch switch-arg', since it
+ is more commonly supported by
+ homebrew getopt-like functions.
+ */
+ wcstring completion = format_string(L"%ls=", whole_opt.c_str()+offset);
+ append_completion(this->completions,
+ completion,
+ C_(o->desc),
+ flags);
+
+ }
+
+ append_completion(this->completions,
+ whole_opt.c_str() + offset,
+ C_(o->desc),
+ flags);
+ }
+ }
+ }
+ }
+ }
+ }
+
+ return use_files;
+}
+
+/**
+ Perform file completion on the specified string
+*/
+void completer_t::complete_param_expand(const wcstring &sstr, bool do_file, bool directories_only)
+{
+ const wchar_t * const str = sstr.c_str();
+ const wchar_t *comp_str;
+
+ if (string_prefixes_string(L"--", sstr) && (comp_str = wcschr(str, L'=')))
+ {
+ comp_str++;
+ }
+ else
+ {
+ comp_str = str;
+ }
+
+ expand_flags_t flags = EXPAND_SKIP_CMDSUBST | ACCEPT_INCOMPLETE | this->expand_flags();
+
+ if (! do_file)
+ flags |= EXPAND_SKIP_WILDCARDS;
+
+ if (directories_only && do_file)
+ flags |= DIRECTORIES_ONLY;
+
+ /* Squelch file descriptions per issue 254 */
+ if (this->type() == COMPLETE_AUTOSUGGEST || do_file)
+ flags |= EXPAND_NO_DESCRIPTIONS;
+
+ /* Don't do fuzzy matching for files if the string begins with a dash (#568). We could consider relaxing this if there was a preceding double-dash argument */
+ if (string_prefixes_string(L"-", sstr))
+ flags &= ~EXPAND_FUZZY_MATCH;
+
+ if (expand_string(comp_str,
+ this->completions,
+ flags, NULL) == EXPAND_ERROR)
+ {
+ debug(3, L"Error while expanding string '%ls'", comp_str);
+ }
+}
+
+/**
+ Complete the specified string as an environment variable
+*/
+bool completer_t::complete_variable(const wcstring &str, size_t start_offset)
+{
+ const wchar_t * const whole_var = str.c_str();
+ const wchar_t *var = &whole_var[start_offset];
+ size_t varlen = wcslen(var);
+ bool res = false;
+
+ const wcstring_list_t names = complete_get_variable_names();
+ for (size_t i=0; i<names.size(); i++)
+ {
+ const wcstring & env_name = names.at(i);
+
+ string_fuzzy_match_t match = string_fuzzy_match_string(var, env_name, this->max_fuzzy_match_type());
+ if (match.type == fuzzy_match_none)
+ {
+ // No match
+ continue;
+ }
+
+ wcstring comp;
+ int flags = 0;
+
+ if (! match_type_requires_full_replacement(match.type))
+ {
+ // Take only the suffix
+ comp.append(env_name.c_str() + varlen);
+ }
+ else
+ {
+ comp.append(whole_var, start_offset);
+ comp.append(env_name);
+ flags = COMPLETE_REPLACES_TOKEN | COMPLETE_DONT_ESCAPE;
+ }
+
+ wcstring desc;
+ if (this->wants_descriptions())
+ {
+ env_var_t value_unescaped = env_get_string(env_name);
+ if (value_unescaped.missing())
+ continue;
+
+ wcstring value = expand_escape_variable(value_unescaped);
+ if (this->type() != COMPLETE_AUTOSUGGEST)
+ desc = format_string(COMPLETE_VAR_DESC_VAL, value.c_str());
+ }
+
+ append_completion(this->completions, comp, desc, flags, match);
+
+ res = true;
+ }
+
+ return res;
+}
+
+bool completer_t::try_complete_variable(const wcstring &str)
+{
+ enum {e_unquoted, e_single_quoted, e_double_quoted} mode = e_unquoted;
+ const size_t len = str.size();
+
+ /* Get the position of the dollar heading a run of valid variable characters. -1 means none. */
+ size_t variable_start = -1;
+
+ for (size_t in_pos=0; in_pos<len; in_pos++)
+ {
+ wchar_t c = str.at(in_pos);
+ if (! wcsvarchr(c))
+ {
+ /* This character cannot be in a variable, reset the dollar */
+ variable_start = -1;
+ }
+
+ switch (c)
+ {
+ case L'\\':
+ in_pos++;
+ break;
+
+ case L'$':
+ if (mode == e_unquoted || mode == e_double_quoted)
+ {
+ variable_start = in_pos;
+ }
+ break;
+
+ case L'\'':
+ if (mode == e_single_quoted)
+ {
+ mode = e_unquoted;
+ }
+ else if (mode == e_unquoted)
+ {
+ mode = e_single_quoted;
+ }
+ break;
+
+ case L'"':
+ if (mode == e_double_quoted)
+ {
+ mode = e_unquoted;
+ }
+ else if (mode == e_unquoted)
+ {
+ mode = e_double_quoted;
+ }
+ break;
+ }
+ }
+
+ /* Now complete if we have a variable start that's also not the last character */
+ bool result = false;
+ if (variable_start != static_cast<size_t>(-1) && variable_start + 1 < len)
+ {
+ result = this->complete_variable(str, variable_start + 1);
+ }
+ return result;
+}
+
+/**
+ Try to complete the specified string as a username. This is used by
+ ~USER type expansion.
+
+ \return 0 if unable to complete, 1 otherwise
+*/
+bool completer_t::try_complete_user(const wcstring &str)
+{
+ const wchar_t *cmd = str.c_str();
+ const wchar_t *first_char=cmd;
+ int res=0;
+ double start_time = timef();
+
+ if (*first_char ==L'~' && !wcschr(first_char, L'/'))
+ {
+ const wchar_t *user_name = first_char+1;
+ const wchar_t *name_end = wcschr(user_name, L'~');
+ if (name_end == 0)
+ {
+ struct passwd *pw;
+ size_t name_len = wcslen(user_name);
+
+ setpwent();
+
+ while ((pw=getpwent()) != 0)
+ {
+ double current_time = timef();
+
+ if (current_time - start_time > 0.2)
+ {
+ return 1;
+ }
+
+ if (pw->pw_name)
+ {
+ const wcstring pw_name_str = str2wcstring(pw->pw_name);
+ const wchar_t *pw_name = pw_name_str.c_str();
+ if (wcsncmp(user_name, pw_name, name_len)==0)
+ {
+ wcstring desc = format_string(COMPLETE_USER_DESC, pw_name);
+ append_completion(this->completions,
+ &pw_name[name_len],
+ desc,
+ COMPLETE_NO_SPACE);
+
+ res=1;
+ }
+ else if (wcsncasecmp(user_name, pw_name, name_len)==0)
+ {
+ wcstring name = format_string(L"~%ls", pw_name);
+ wcstring desc = format_string(COMPLETE_USER_DESC, pw_name);
+
+ append_completion(this->completions,
+ name,
+ desc,
+ COMPLETE_REPLACES_TOKEN | COMPLETE_DONT_ESCAPE | COMPLETE_NO_SPACE);
+ res=1;
+ }
+ }
+ }
+ endpwent();
+ }
+ }
+
+ return res;
+}
+
+void complete(const wcstring &cmd_with_subcmds, std::vector<completion_t> &comps, completion_request_flags_t flags)
+{
+ /* Determine the innermost subcommand */
+ const wchar_t *cmdsubst_begin, *cmdsubst_end;
+ parse_util_cmdsubst_extent(cmd_with_subcmds.c_str(), cmd_with_subcmds.size(), &cmdsubst_begin, &cmdsubst_end);
+ assert(cmdsubst_begin != NULL && cmdsubst_end != NULL && cmdsubst_end >= cmdsubst_begin);
+ const wcstring cmd = wcstring(cmdsubst_begin, cmdsubst_end - cmdsubst_begin);
+
+ /* Make our completer */
+ completer_t completer(cmd, flags);
+
+ wcstring current_command;
+ const size_t pos = cmd.size();
+ bool done=false;
+ bool use_command = 1;
+ bool use_function = 1;
+ bool use_builtin = 1;
+ bool use_implicit_cd = 1;
+
+ //debug( 1, L"Complete '%ls'", cmd.c_str() );
+
+ const wchar_t *cmd_cstr = cmd.c_str();
+ const wchar_t *tok_begin = NULL, *prev_begin = NULL, *prev_end = NULL;
+ parse_util_token_extent(cmd_cstr, cmd.size(), &tok_begin, NULL, &prev_begin, &prev_end);
+
+ /**
+ If we are completing a variable name or a tilde expansion user
+ name, we do that and return. No need for any other completions.
+ */
+ const wcstring current_token = tok_begin;
+
+ /* Unconditionally complete variables and processes. This is a little weird since we will happily complete variables even in e.g. command position, despite the fact that they are invalid there. */
+ if (!done)
+ {
+ done = completer.try_complete_variable(current_token) || completer.try_complete_user(current_token);
+ }
+
+ if (!done)
+ {
+ //const size_t prev_token_len = (prev_begin ? prev_end - prev_begin : 0);
+ //const wcstring prev_token(prev_begin, prev_token_len);
+
+ parse_node_tree_t tree;
+ parse_tree_from_string(cmd, parse_flag_continue_after_error | parse_flag_accept_incomplete_tokens | parse_flag_include_comments, &tree, NULL);
+
+ /* Find any plain statement that contains the position. We have to backtrack past spaces (#1261). So this will be at either the last space character, or after the end of the string */
+ size_t adjusted_pos = pos;
+ while (adjusted_pos > 0 && cmd.at(adjusted_pos - 1) == L' ')
+ {
+ adjusted_pos--;
+ }
+
+ const parse_node_t *plain_statement = tree.find_node_matching_source_location(symbol_plain_statement, adjusted_pos, NULL);
+ if (plain_statement == NULL)
+ {
+ /* Not part of a plain statement. This could be e.g. a for loop header, case expression, etc. Do generic file completions (#1309). If we had to backtrack, it means there was whitespace; don't do an autosuggestion in that case. Also don't do it if we are just after a pipe, semicolon, or & (#1631), or in a comment.
+
+ Overall this logic is a total mess. A better approach would be to return the "possible next token" from the parse tree directly (this data is available as the first of the sequence of nodes without source locations at the very end of the parse tree). */
+ bool do_file = true;
+ if (flags & COMPLETION_REQUEST_AUTOSUGGESTION)
+ {
+ if (adjusted_pos < pos)
+ {
+ do_file = false;
+ }
+ else if (pos > 0)
+ {
+ // If the previous character is in one of these types, we don't do file suggestions
+ parse_token_type_t bad_types[] = {parse_token_type_pipe, parse_token_type_end, parse_token_type_background, parse_special_type_comment};
+ for (size_t i=0; i < sizeof bad_types / sizeof *bad_types; i++)
+ {
+ if (tree.find_node_matching_source_location(bad_types[i], pos - 1, NULL))
+ {
+ do_file = false;
+ break;
+ }
+ }
+ }
+ }
+ completer.complete_param_expand(current_token, do_file);
+ }
+ else
+ {
+ assert(plain_statement->has_source() && plain_statement->type == symbol_plain_statement);
+
+ /* Get the command node */
+ const parse_node_t *cmd_node = tree.get_child(*plain_statement, 0, parse_token_type_string);
+
+ /* Get the actual command string */
+ if (cmd_node != NULL)
+ current_command = cmd_node->get_source(cmd);
+
+ /* Check the decoration */
+ switch (tree.decoration_for_plain_statement(*plain_statement))
+ {
+ case parse_statement_decoration_none:
+ use_command = true;
+ use_function = true;
+ use_builtin = true;
+ use_implicit_cd = true;
+ break;
+
+ case parse_statement_decoration_command:
+ case parse_statement_decoration_exec:
+ use_command = true;
+ use_function = false;
+ use_builtin = false;
+ use_implicit_cd = false;
+ break;
+
+ case parse_statement_decoration_builtin:
+ use_command = false;
+ use_function = false;
+ use_builtin = true;
+ use_implicit_cd = false;
+ break;
+ }
+
+ if (cmd_node && cmd_node->location_in_or_at_end_of_source_range(pos))
+ {
+ /* Complete command filename */
+ completer.complete_cmd(current_token, use_function, use_builtin, use_command, use_implicit_cd);
+ }
+ else
+ {
+ /* Get all the arguments */
+ const parse_node_tree_t::parse_node_list_t all_arguments = tree.find_nodes(*plain_statement, symbol_argument);
+
+ /* See whether we are in an argument. We may also be in a redirection, or nothing at all. */
+ size_t matching_arg_index = -1;
+ for (size_t i=0; i < all_arguments.size(); i++)
+ {
+ const parse_node_t *node = all_arguments.at(i);
+ if (node->location_in_or_at_end_of_source_range(adjusted_pos))
+ {
+ matching_arg_index = i;
+ break;
+ }
+ }
+
+ bool had_ddash = false;
+ wcstring current_argument, previous_argument;
+ if (matching_arg_index != (size_t)(-1))
+ {
+ const wcstring matching_arg = all_arguments.at(matching_arg_index)->get_source(cmd);
+
+ /* If the cursor is in whitespace, then the "current" argument is empty and the previous argument is the matching one. But if the cursor was in or at the end of the argument, then the current argument is the matching one, and the previous argument is the one before it. */
+ bool cursor_in_whitespace = (adjusted_pos < pos);
+ if (cursor_in_whitespace)
+ {
+ current_argument = L"";
+ previous_argument = matching_arg;
+ }
+ else
+ {
+ current_argument = matching_arg;
+ if (matching_arg_index > 0)
+ {
+ previous_argument = all_arguments.at(matching_arg_index - 1)->get_source(cmd);
+ }
+ }
+
+ /* Check to see if we have a preceding double-dash */
+ for (size_t i=0; i < matching_arg_index; i++)
+ {
+ if (all_arguments.at(i)->get_source(cmd) == L"--")
+ {
+ had_ddash = true;
+ break;
+ }
+ }
+ }
+
+ /* If we are not in an argument, we may be in a redirection */
+ bool in_redirection = false;
+ if (matching_arg_index == (size_t)(-1))
+ {
+ const parse_node_t *redirection = tree.find_node_matching_source_location(symbol_redirection, adjusted_pos, plain_statement);
+ in_redirection = (redirection != NULL);
+ }
+
+ bool do_file = false, directories_only = false;
+ if (in_redirection)
+ {
+ do_file = true;
+ }
+ else
+ {
+ wcstring current_command_unescape, previous_argument_unescape, current_argument_unescape;
+ if (unescape_string(current_command, &current_command_unescape, UNESCAPE_DEFAULT) &&
+ unescape_string(previous_argument, &previous_argument_unescape, UNESCAPE_DEFAULT) &&
+ unescape_string(current_argument, &current_argument_unescape, UNESCAPE_INCOMPLETE))
+ {
+ // Have to walk over the command and its entire wrap chain
+ // If any command disables do_file, then they all do
+ do_file = true;
+ const wcstring_list_t wrap_chain = complete_get_wrap_chain(current_command_unescape);
+ for (size_t i=0; i < wrap_chain.size(); i++)
+ {
+ // Hackish, this. The first command in the chain is always the given command. For every command past the first, we need to create a transient commandline for builtin_commandline. But not for COMPLETION_REQUEST_AUTOSUGGESTION, which may occur on background threads.
+ builtin_commandline_scoped_transient_t *transient_cmd = NULL;
+ if (i == 0)
+ {
+ assert(wrap_chain.at(i) == current_command_unescape);
+ }
+ else if (! (flags & COMPLETION_REQUEST_AUTOSUGGESTION))
+ {
+ assert(cmd_node != NULL);
+ wcstring faux_cmdline = cmd;
+ faux_cmdline.replace(cmd_node->source_start, cmd_node->source_length, wrap_chain.at(i));
+ transient_cmd = new builtin_commandline_scoped_transient_t(faux_cmdline);
+ }
+ if (! completer.complete_param(wrap_chain.at(i),
+ previous_argument_unescape,
+ current_argument_unescape,
+ !had_ddash))
+ {
+ do_file = false;
+ }
+ delete transient_cmd; //may be null
+ }
+ }
+
+ /* If we have found no command specific completions at all, fall back to using file completions. */
+ if (completer.empty())
+ do_file = true;
+
+ /* Hack. If we're cd, do directories only (#1059) */
+ directories_only = (current_command_unescape == L"cd");
+
+ /* And if we're autosuggesting, and the token is empty, don't do file suggestions */
+ if ((flags & COMPLETION_REQUEST_AUTOSUGGESTION) && current_argument_unescape.empty())
+ {
+ do_file = false;
+ }
+ }
+
+ /* This function wants the unescaped string */
+ completer.complete_param_expand(current_token, do_file, directories_only);
+ }
+ }
+ }
+
+ comps = completer.get_completions();
+}
+
+
+
+/**
+ Print the GNU longopt style switch \c opt, and the argument \c
+ argument to the specified stringbuffer, but only if arguemnt is
+ non-null and longer than 0 characters.
+*/
+static void append_switch(wcstring &out,
+ const wcstring &opt,
+ const wcstring &argument)
+{
+ if (argument.empty())
+ return;
+
+ wcstring esc = escape_string(argument, 1);
+ append_format(out, L" --%ls %ls", opt.c_str(), esc.c_str());
+}
+
+void complete_print(wcstring &out)
+{
+ scoped_lock locker(completion_lock);
+ scoped_lock locker2(completion_entry_lock);
+
+ // Get a list of all completions in a vector, then sort it by order
+ std::vector<const completion_entry_t *> all_completions(completion_set.begin(), completion_set.end());
+ sort(all_completions.begin(), all_completions.end(), compare_completions_by_order);
+
+ for (std::vector<const completion_entry_t *>::const_iterator iter = all_completions.begin(); iter != all_completions.end(); ++iter)
+ {
+ const completion_entry_t *e = *iter;
+ const option_list_t &options = e->get_options();
+ for (option_list_t::const_iterator oiter = options.begin(); oiter != options.end(); ++oiter)
+ {
+ const complete_entry_opt_t *o = &*oiter;
+ const wchar_t *modestr[] =
+ {
+ L"",
+ L" --no-files",
+ L" --require-parameter",
+ L" --exclusive"
+ }
+ ;
+
+ append_format(out,
+ L"complete%ls",
+ modestr[o->result_mode]);
+
+ append_switch(out,
+ e->cmd_is_path ? L"path" : L"command",
+ escape_string(e->cmd, ESCAPE_ALL));
+
+
+ if (o->short_opt != 0)
+ {
+ append_format(out,
+ L" --short-option '%lc'",
+ o->short_opt);
+ }
+
+
+ append_switch(out,
+ o->old_mode?L"old-option":L"long-option",
+ o->long_opt);
+
+ append_switch(out,
+ L"description",
+ C_(o->desc));
+
+ append_switch(out,
+ L"arguments",
+ o->comp);
+
+ append_switch(out,
+ L"condition",
+ o->condition);
+
+ out.append(L"\n");
+ }
+ }
+
+ /* Append wraps. This is a wonky interface where even values are the commands, and odd values are the targets that they wrap. */
+ const wcstring_list_t wrap_pairs = complete_get_wrap_pairs();
+ assert(wrap_pairs.size() % 2 == 0);
+ for (size_t i=0; i < wrap_pairs.size(); )
+ {
+ const wcstring &cmd = wrap_pairs.at(i++);
+ const wcstring &target = wrap_pairs.at(i++);
+ append_format(out, L"complete --command %ls --wraps %ls\n", cmd.c_str(), target.c_str());
+ }
+}
+
+
+/* Completion "wrapper" support. The map goes from wrapping-command to wrapped-command-list */
+static pthread_mutex_t wrapper_lock = PTHREAD_MUTEX_INITIALIZER;
+typedef std::map<wcstring, wcstring_list_t> wrapper_map_t;
+static wrapper_map_t &wrap_map()
+{
+ ASSERT_IS_LOCKED(wrapper_lock);
+ // A pointer is a little more efficient than an object as a static because we can elide the thread-safe initialization
+ static wrapper_map_t *wrapper_map = NULL;
+ if (wrapper_map == NULL)
+ {
+ wrapper_map = new wrapper_map_t();
+ }
+ return *wrapper_map;
+}
+
+/* Add a new target that is wrapped by command. Example: sgrep (command) wraps grep (target). */
+bool complete_add_wrapper(const wcstring &command, const wcstring &new_target)
+{
+ if (command.empty() || new_target.empty())
+ {
+ return false;
+ }
+
+ scoped_lock locker(wrapper_lock);
+ wrapper_map_t &wraps = wrap_map();
+ wcstring_list_t *targets = &wraps[command];
+ // If it's already present, we do nothing
+ if (std::find(targets->begin(), targets->end(), new_target) == targets->end())
+ {
+ targets->push_back(new_target);
+ }
+ return true;
+}
+
+bool complete_remove_wrapper(const wcstring &command, const wcstring &target_to_remove)
+{
+ if (command.empty() || target_to_remove.empty())
+ {
+ return false;
+ }
+
+ scoped_lock locker(wrapper_lock);
+ wrapper_map_t &wraps = wrap_map();
+ bool result = false;
+ wrapper_map_t::iterator current_targets_iter = wraps.find(command);
+ if (current_targets_iter != wraps.end())
+ {
+ wcstring_list_t *targets = &current_targets_iter->second;
+ wcstring_list_t::iterator where = std::find(targets->begin(), targets->end(), target_to_remove);
+ if (where != targets->end())
+ {
+ targets->erase(where);
+ result = true;
+ }
+ }
+ return result;
+}
+
+wcstring_list_t complete_get_wrap_chain(const wcstring &command)
+{
+ if (command.empty())
+ {
+ return wcstring_list_t();
+ }
+ scoped_lock locker(wrapper_lock);
+ const wrapper_map_t &wraps = wrap_map();
+
+ wcstring_list_t result;
+ std::set<wcstring> visited; // set of visited commands
+ wcstring_list_t to_visit(1, command); // stack of remaining-to-visit commands
+
+ wcstring target;
+ while (! to_visit.empty())
+ {
+ // Grab the next command to visit, put it in target
+ target.swap(to_visit.back());
+ to_visit.pop_back();
+
+ // Try inserting into visited. If it was already present, we skip it; this is how we avoid loops.
+ if (! visited.insert(target).second)
+ {
+ continue;
+ }
+
+ // Insert the target in the result. Note this is the command itself, if this is the first iteration of the loop.
+ result.push_back(target);
+
+ // Enqueue its children
+ wrapper_map_t::const_iterator target_children_iter = wraps.find(target);
+ if (target_children_iter != wraps.end())
+ {
+ const wcstring_list_t &children = target_children_iter->second;
+ to_visit.insert(to_visit.end(), children.begin(), children.end());
+ }
+ }
+
+ return result;
+}
+
+wcstring_list_t complete_get_wrap_pairs()
+{
+ wcstring_list_t result;
+ scoped_lock locker(wrapper_lock);
+ const wrapper_map_t &wraps = wrap_map();
+ for (wrapper_map_t::const_iterator outer = wraps.begin(); outer != wraps.end(); ++outer)
+ {
+ const wcstring &cmd = outer->first;
+ const wcstring_list_t &targets = outer->second;
+ for (size_t i=0; i < targets.size(); i++)
+ {
+ result.push_back(cmd);
+ result.push_back(targets.at(i));
+ }
+ }
+ return result;
+}