nspace.c 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371
  1. /*
  2. * nspace.c
  3. *
  4. * This file provides the implementation of the "nspace" class
  5. */
  6. #include "private.h"
  7. #include "lub/string.h"
  8. #include <assert.h>
  9. #include <stdlib.h>
  10. #include <string.h>
  11. #include <stdio.h>
  12. #include <sys/types.h>
  13. #include <regex.h>
  14. /*---------------------------------------------------------
  15. * PRIVATE METHODS
  16. *--------------------------------------------------------- */
  17. static void clish_nspace_init(clish_nspace_t * this, clish_view_t * view)
  18. {
  19. this->view = view;
  20. /* set up defaults */
  21. this->prefix = NULL;
  22. this->help = BOOL_FALSE;
  23. this->completion = BOOL_TRUE;
  24. this->context_help = BOOL_FALSE;
  25. this->inherit = BOOL_TRUE;
  26. this->prefix_cmd = NULL;
  27. /* initialise the tree of commands links for this nspace */
  28. lub_bintree_init(&this->tree,
  29. clish_command_bt_offset(),
  30. clish_command_bt_compare, clish_command_bt_getkey);
  31. }
  32. /*--------------------------------------------------------- */
  33. static void clish_nspace_fini(clish_nspace_t * this)
  34. {
  35. clish_command_t *cmd;
  36. /* deallocate the memory for this instance */
  37. if (this->prefix) {
  38. free(this->prefix);
  39. regfree(&this->prefix_regex);
  40. }
  41. /* delete each command link held by this nspace */
  42. while ((cmd = lub_bintree_findfirst(&this->tree))) {
  43. /* remove the command from the tree */
  44. lub_bintree_remove(&this->tree, cmd);
  45. /* release the instance */
  46. clish_command_delete(cmd);
  47. }
  48. /* Delete prefix pseudo-command */
  49. if (this->prefix_cmd) {
  50. clish_command_delete(this->prefix_cmd);
  51. this->prefix_cmd = NULL;
  52. }
  53. }
  54. /*--------------------------------------------------------- */
  55. clish_command_t * clish_nspace_create_prefix_cmd(clish_nspace_t * this,
  56. const char * name, const char * help)
  57. {
  58. if (this->prefix_cmd) {
  59. clish_command_delete(this->prefix_cmd);
  60. this->prefix_cmd = NULL;
  61. }
  62. return (this->prefix_cmd = clish_command_new(name, help));
  63. }
  64. /*--------------------------------------------------------- */
  65. static clish_command_t *clish_nspace_find_create_command(clish_nspace_t * this,
  66. const char *prefix, const clish_command_t * ref)
  67. {
  68. clish_command_t *cmd;
  69. char *name = NULL;
  70. const char *help = NULL;
  71. clish_command_t *tmp = NULL;
  72. const char *str = NULL;
  73. assert(prefix);
  74. if (!ref) {
  75. assert(this->prefix_cmd);
  76. name = lub_string_dup(prefix);
  77. ref = this->prefix_cmd;
  78. help = clish_command__get_text(this->prefix_cmd);
  79. } else {
  80. lub_string_catn(&name, prefix, strlen(prefix));
  81. lub_string_catn(&name, " ", 1);
  82. lub_string_catn(&name, clish_command__get_name(ref),
  83. strlen(clish_command__get_name(ref)));
  84. help = clish_command__get_text(ref);
  85. }
  86. /* The command is cached already */
  87. if ((cmd = lub_bintree_find(&this->tree, name))) {
  88. free(name);
  89. return cmd;
  90. }
  91. cmd = clish_command_new_link(name, help, ref);
  92. free(name);
  93. assert(cmd);
  94. /* The command was created dynamically */
  95. clish_command__set_dynamic(cmd, BOOL_TRUE);
  96. /* Delete proxy commands with another prefixes */
  97. tmp = lub_bintree_findfirst(&this->tree);
  98. if (tmp)
  99. str = clish_command__get_name(tmp);
  100. if (str && (lub_string_nocasestr(str, prefix) != str)) {
  101. do {
  102. lub_bintree_remove(&this->tree, tmp);
  103. clish_command_delete(tmp);
  104. } while ((tmp = lub_bintree_findfirst(&this->tree)));
  105. }
  106. /* Insert command link into the tree */
  107. if (-1 == lub_bintree_insert(&this->tree, cmd)) {
  108. clish_command_delete(cmd);
  109. cmd = NULL;
  110. }
  111. return cmd;
  112. }
  113. /*---------------------------------------------------------
  114. * PUBLIC META FUNCTIONS
  115. *--------------------------------------------------------- */
  116. clish_nspace_t *clish_nspace_new(clish_view_t * view)
  117. {
  118. clish_nspace_t *this = malloc(sizeof(clish_nspace_t));
  119. if (this)
  120. clish_nspace_init(this, view);
  121. return this;
  122. }
  123. /*---------------------------------------------------------
  124. * PUBLIC METHODS
  125. *--------------------------------------------------------- */
  126. void clish_nspace_delete(clish_nspace_t * this)
  127. {
  128. clish_nspace_fini(this);
  129. free(this);
  130. }
  131. /*--------------------------------------------------------- */
  132. static const char *clish_nspace_after_prefix(const regex_t *prefix_regex,
  133. const char *line, char **real_prefix)
  134. {
  135. const char *in_line = NULL;
  136. regmatch_t pmatch[1];
  137. int res;
  138. if (!line)
  139. return NULL;
  140. /* Compile regular expression */
  141. res = regexec(prefix_regex, line, 1, pmatch, 0);
  142. if (res || (0 != pmatch[0].rm_so))
  143. return NULL;
  144. /* Empty match */
  145. if (0 == pmatch[0].rm_eo)
  146. return NULL;
  147. in_line = line + pmatch[0].rm_eo;
  148. lub_string_catn(real_prefix, line, pmatch[0].rm_eo);
  149. return in_line;
  150. }
  151. /*--------------------------------------------------------- */
  152. clish_command_t *clish_nspace_find_command(clish_nspace_t * this, const char *name)
  153. {
  154. clish_command_t *cmd = NULL, *retval = NULL;
  155. clish_view_t *view = clish_nspace__get_view(this);
  156. const char *in_line;
  157. char *real_prefix = NULL;
  158. if (!clish_nspace__get_prefix(this))
  159. return clish_view_find_command(view, name, this->inherit);
  160. if (!(in_line = clish_nspace_after_prefix(
  161. clish_nspace__get_prefix_regex(this), name, &real_prefix)))
  162. return NULL;
  163. /* If prefix is followed by space */
  164. if (in_line[0] == ' ')
  165. in_line++;
  166. if (in_line[0] != '\0') {
  167. cmd = clish_view_find_command(view, in_line, this->inherit);
  168. if (!cmd) {
  169. lub_string_free(real_prefix);
  170. return NULL;
  171. }
  172. }
  173. retval = clish_nspace_find_create_command(this, real_prefix, cmd);
  174. lub_string_free(real_prefix);
  175. return retval;
  176. }
  177. /*--------------------------------------------------------- */
  178. const clish_command_t *clish_nspace_find_next_completion(clish_nspace_t * this,
  179. const char *iter_cmd, const char *line,
  180. clish_nspace_visibility_t field)
  181. {
  182. const clish_command_t *cmd = NULL, *retval = NULL;
  183. clish_view_t *view = clish_nspace__get_view(this);
  184. const char *in_iter = "";
  185. const char *in_line;
  186. char *real_prefix = NULL;
  187. if (!clish_nspace__get_prefix(this))
  188. return clish_view_find_next_completion(view, iter_cmd,
  189. line, field, this->inherit);
  190. if (!(in_line = clish_nspace_after_prefix(
  191. clish_nspace__get_prefix_regex(this), line, &real_prefix)))
  192. return NULL;
  193. if (in_line[0] != '\0') {
  194. /* If prefix is followed by space */
  195. if (in_line[0] == ' ')
  196. in_line++;
  197. if (iter_cmd &&
  198. (lub_string_nocasestr(iter_cmd, real_prefix) == iter_cmd) &&
  199. (lub_string_nocasecmp(iter_cmd, real_prefix)))
  200. in_iter = iter_cmd + strlen(real_prefix) + 1;
  201. cmd = clish_view_find_next_completion(view,
  202. in_iter, in_line, field, this->inherit);
  203. if (!cmd) {
  204. lub_string_free(real_prefix);
  205. return NULL;
  206. }
  207. }
  208. /* If prefix was already returned. */
  209. if (!cmd && iter_cmd && !lub_string_nocasecmp(iter_cmd, real_prefix)) {
  210. lub_string_free(real_prefix);
  211. return NULL;
  212. }
  213. retval = clish_nspace_find_create_command(this, real_prefix, cmd);
  214. lub_string_free(real_prefix);
  215. return retval;
  216. }
  217. /*--------------------------------------------------------- */
  218. void clish_nspace_clean_proxy(clish_nspace_t * this)
  219. {
  220. clish_command_t *cmd = NULL;
  221. /* Recursive proxy clean */
  222. clish_view_clean_proxy(this->view);
  223. /* Delete each command proxy held by this nspace */
  224. while ((cmd = lub_bintree_findfirst(&this->tree))) {
  225. /* remove the command from the tree */
  226. lub_bintree_remove(&this->tree, cmd);
  227. /* release the instance */
  228. clish_command_delete(cmd);
  229. }
  230. }
  231. /*---------------------------------------------------------
  232. * PUBLIC ATTRIBUTES
  233. *--------------------------------------------------------- */
  234. clish_view_t *clish_nspace__get_view(const clish_nspace_t * this)
  235. {
  236. return this->view;
  237. }
  238. /*--------------------------------------------------------- */
  239. void clish_nspace__set_prefix(clish_nspace_t * this, const char *prefix)
  240. {
  241. assert(!this->prefix);
  242. this->prefix = lub_string_dup(prefix);
  243. regcomp(&this->prefix_regex, prefix, REG_EXTENDED | REG_ICASE);
  244. }
  245. /*--------------------------------------------------------- */
  246. const char *clish_nspace__get_prefix(const clish_nspace_t * this)
  247. {
  248. return this->prefix;
  249. }
  250. /*--------------------------------------------------------- */
  251. const regex_t *clish_nspace__get_prefix_regex(const clish_nspace_t * this)
  252. {
  253. if (!this->prefix)
  254. return NULL;
  255. return &this->prefix_regex;
  256. }
  257. /*--------------------------------------------------------- */
  258. void clish_nspace__set_help(clish_nspace_t * this, bool_t help)
  259. {
  260. this->help = help;
  261. }
  262. /*--------------------------------------------------------- */
  263. bool_t clish_nspace__get_help(const clish_nspace_t * this)
  264. {
  265. return this->help;
  266. }
  267. /*--------------------------------------------------------- */
  268. void clish_nspace__set_completion(clish_nspace_t * this, bool_t completion)
  269. {
  270. this->completion = completion;
  271. }
  272. /*--------------------------------------------------------- */
  273. bool_t clish_nspace__get_completion(const clish_nspace_t * this)
  274. {
  275. return this->completion;
  276. }
  277. /*--------------------------------------------------------- */
  278. void clish_nspace__set_context_help(clish_nspace_t * this, bool_t context_help)
  279. {
  280. this->context_help = context_help;
  281. }
  282. /*--------------------------------------------------------- */
  283. bool_t clish_nspace__get_context_help(const clish_nspace_t * this)
  284. {
  285. return this->context_help;
  286. }
  287. /*--------------------------------------------------------- */
  288. void clish_nspace__set_inherit(clish_nspace_t * this, bool_t inherit)
  289. {
  290. this->inherit = inherit;
  291. }
  292. /*--------------------------------------------------------- */
  293. bool_t clish_nspace__get_inherit(const clish_nspace_t * this)
  294. {
  295. return this->inherit;
  296. }
  297. /*--------------------------------------------------------- */
  298. bool_t clish_nspace__get_visibility(const clish_nspace_t * instance,
  299. clish_nspace_visibility_t field)
  300. {
  301. bool_t result = BOOL_FALSE;
  302. switch (field) {
  303. case CLISH_NSPACE_HELP:
  304. result = clish_nspace__get_help(instance);
  305. break;
  306. case CLISH_NSPACE_COMPLETION:
  307. result = clish_nspace__get_completion(instance);
  308. break;
  309. case CLISH_NSPACE_CHELP:
  310. result = clish_nspace__get_context_help(instance);
  311. break;
  312. default:
  313. break;
  314. }
  315. return result;
  316. }
  317. /*--------------------------------------------------------- */