1
0

nspace.c 9.3 KB

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