shell_command.c 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. /*
  2. * shell_word_generator.c
  3. */
  4. #include <string.h>
  5. #include "private.h"
  6. #include "lub/string.h"
  7. #include "lub/argv.h"
  8. /*-------------------------------------------------------- */
  9. void clish_shell_iterator_init(clish_shell_iterator_t * iter,
  10. clish_nspace_visibility_t field)
  11. {
  12. iter->last_cmd = NULL;
  13. iter->field = field;
  14. }
  15. /*--------------------------------------------------------- */
  16. const clish_command_t *clish_shell_resolve_command(const clish_shell_t * this,
  17. const char *line)
  18. {
  19. clish_command_t *cmd, *result;
  20. /* Search the current view */
  21. result = clish_view_resolve_command(clish_shell__get_view(this), line, BOOL_TRUE);
  22. /* Search the global view */
  23. cmd = clish_view_resolve_command(this->global, line, BOOL_TRUE);
  24. result = clish_command_choose_longest(result, cmd);
  25. return result;
  26. }
  27. /*--------------------------------------------------------- */
  28. const clish_command_t *clish_shell_resolve_prefix(const clish_shell_t * this,
  29. const char *line)
  30. {
  31. clish_command_t *cmd, *result;
  32. /* Search the current view */
  33. result = clish_view_resolve_prefix(clish_shell__get_view(this), line, BOOL_TRUE);
  34. /* Search the global view */
  35. cmd = clish_view_resolve_prefix(this->global, line, BOOL_TRUE);
  36. result = clish_command_choose_longest(result, cmd);
  37. return result;
  38. }
  39. /*-------------------------------------------------------- */
  40. const clish_command_t *clish_shell_find_next_completion(const clish_shell_t *
  41. this, const char *line, clish_shell_iterator_t * iter)
  42. {
  43. const clish_command_t *result, *cmd;
  44. /* ask the local view for next command */
  45. result = clish_view_find_next_completion(clish_shell__get_view(this),
  46. iter->last_cmd, line, iter->field, BOOL_TRUE);
  47. /* ask the global view for next command */
  48. cmd = clish_view_find_next_completion(this->global,
  49. iter->last_cmd, line, iter->field, BOOL_TRUE);
  50. if (clish_command_diff(result, cmd) > 0)
  51. result = cmd;
  52. if (!result)
  53. iter->last_cmd = NULL;
  54. else
  55. iter->last_cmd = clish_command__get_name(result);
  56. return result;
  57. }
  58. /*--------------------------------------------------------- */
  59. void clish_shell_param_generator(clish_shell_t *this, lub_argv_t *matches,
  60. const clish_command_t *cmd, const char *line, unsigned offset)
  61. {
  62. const char *name = clish_command__get_name(cmd);
  63. char *text = lub_string_dup(&line[offset]);
  64. clish_ptype_t *ptype;
  65. unsigned idx = lub_string_wordcount(name);
  66. /* get the index of the current parameter */
  67. unsigned index = lub_string_wordcount(line) - idx;
  68. clish_context_t context;
  69. if ((0 != index) || (offset && line[offset - 1] == ' ')) {
  70. lub_argv_t *argv = lub_argv_new(line, 0);
  71. clish_pargv_t *pargv = clish_pargv_new();
  72. clish_pargv_t *completion = clish_pargv_new();
  73. unsigned completion_index = 0;
  74. const clish_param_t *param = NULL;
  75. /* if there is some text for the parameter then adjust the index */
  76. if ((0 != index) && (text[0] != '\0'))
  77. index--;
  78. /* Parse command line to get completion pargv's */
  79. /* Prepare context */
  80. clish_context_init(&context, this);
  81. clish_context__set_cmd(&context, cmd);
  82. clish_context__set_pargv(&context, pargv);
  83. clish_shell_parse_pargv(pargv, cmd, &context,
  84. clish_command__get_paramv(cmd),
  85. argv, &idx, completion, index + idx);
  86. lub_argv_delete(argv);
  87. while ((param = clish_pargv__get_param(completion,
  88. completion_index++))) {
  89. char *result;
  90. /* The param is args so it has no completion */
  91. if (param == clish_command__get_args(cmd))
  92. continue;
  93. /* The switch has no completion string */
  94. if (CLISH_PARAM_SWITCH == clish_param__get_mode(param))
  95. continue;
  96. /* The subcommand is identified by it's value */
  97. if (CLISH_PARAM_SUBCOMMAND ==
  98. clish_param__get_mode(param)) {
  99. result = clish_param__get_value(param);
  100. if (result)
  101. lub_argv_add(matches, result);
  102. }
  103. /* The 'completion' field of PARAM */
  104. if (clish_param__get_completion(param)) {
  105. char *str, *q;
  106. char *saveptr = NULL;
  107. str = clish_shell_expand(
  108. clish_param__get_completion(param), SHELL_VAR_ACTION, &context);
  109. if (str) {
  110. for (q = strtok_r(str, " \n", &saveptr);
  111. q; q = strtok_r(NULL, " \n", &saveptr)) {
  112. if (q == strstr(q, text))
  113. lub_argv_add(matches, q);
  114. }
  115. lub_string_free(str);
  116. }
  117. }
  118. /* The common PARAM. Let ptype do the work */
  119. if ((ptype = clish_param__get_ptype(param)))
  120. clish_ptype_word_generator(ptype, matches, text);
  121. }
  122. clish_pargv_delete(completion);
  123. clish_pargv_delete(pargv);
  124. }
  125. lub_string_free(text);
  126. }
  127. /*--------------------------------------------------------- */