123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263 |
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #include <assert.h>
- #include <faux/str.h>
- #include <faux/list.h>
- #include <faux/error.h>
- #include <klish/khelper.h>
- #include <klish/kparam.h>
- #include <klish/kaction.h>
- #include <klish/kcommand.h>
- struct kcommand_s {
- char *name;
- char *help;
- faux_list_t *params;
- faux_list_t *actions;
- };
- // Simple attributes
- // Name
- KGET_STR(command, name);
- KSET_STR_ONCE(command, name);
- // Help
- KGET_STR(command, help);
- KSET_STR(command, help);
- // PARAM list
- static KCMP_NESTED(command, param, name);
- static KCMP_NESTED_BY_KEY(command, param, name);
- KADD_NESTED(command, param);
- KFIND_NESTED(command, param);
- // ACTION list
- KADD_NESTED(command, action);
- static kcommand_t *kcommand_new_empty(void)
- {
- kcommand_t *command = NULL;
- command = faux_zmalloc(sizeof(*command));
- assert(command);
- if (!command)
- return NULL;
- // Initialize
- command->name = NULL;
- command->help = NULL;
- // PARAM list
- command->params = faux_list_new(FAUX_LIST_UNSORTED, FAUX_LIST_UNIQUE,
- kcommand_param_compare, kcommand_param_kcompare,
- (void (*)(void *))kparam_free);
- assert(command->params);
- // ACTION list
- command->actions = faux_list_new(FAUX_LIST_UNSORTED, FAUX_LIST_NONUNIQUE,
- NULL, NULL, (void (*)(void *))kaction_free);
- assert(command->actions);
- return command;
- }
- kcommand_t *kcommand_new(const icommand_t *info, kcommand_error_e *error)
- {
- kcommand_t *command = NULL;
- command = kcommand_new_empty();
- assert(command);
- if (!command) {
- if (error)
- *error = KCOMMAND_ERROR_ALLOC;
- return NULL;
- }
- if (!info)
- return command;
- if (!kcommand_parse(command, info, error)) {
- kcommand_free(command);
- return NULL;
- }
- return command;
- }
- void kcommand_free(kcommand_t *command)
- {
- if (!command)
- return;
- faux_str_free(command->name);
- faux_str_free(command->help);
- faux_list_free(command->params);
- faux_list_free(command->actions);
- faux_free(command);
- }
- const char *kcommand_strerror(kcommand_error_e error)
- {
- const char *str = NULL;
- switch (error) {
- case KCOMMAND_ERROR_OK:
- str = "Ok";
- break;
- case KCOMMAND_ERROR_INTERNAL:
- str = "Internal error";
- break;
- case KCOMMAND_ERROR_ALLOC:
- str = "Memory allocation error";
- break;
- case KCOMMAND_ERROR_ATTR_NAME:
- str = "Illegal 'name' attribute";
- break;
- case KCOMMAND_ERROR_ATTR_HELP:
- str = "Illegal 'help' attribute";
- break;
- default:
- str = "Unknown error";
- break;
- }
- return str;
- }
- bool_t kcommand_parse(kcommand_t *command, const icommand_t *info, kcommand_error_e *error)
- {
- // Name [mandatory]
- if (faux_str_is_empty(info->name)) {
- if (error)
- *error = KCOMMAND_ERROR_ATTR_NAME;
- return BOOL_FALSE;
- } else {
- if (!kcommand_set_name(command, info->name)) {
- if (error)
- *error = KCOMMAND_ERROR_ATTR_NAME;
- return BOOL_FALSE;
- }
- }
- // Help
- if (!faux_str_is_empty(info->name)) {
- if (!kcommand_set_help(command, info->help)) {
- if (error)
- *error = KCOMMAND_ERROR_ATTR_HELP;
- return BOOL_FALSE;
- }
- }
- return BOOL_TRUE;
- }
- bool_t kcommand_nested_from_icommand(kcommand_t *kcommand, icommand_t *icommand,
- faux_error_t *error_stack)
- {
- bool_t retval = BOOL_TRUE;
- if (!kcommand || !icommand) {
- faux_error_add(error_stack,
- kcommand_strerror(KCOMMAND_ERROR_INTERNAL));
- return BOOL_FALSE;
- }
- // PARAM list
- if (icommand->params) {
- iparam_t **p_iparam = NULL;
- for (p_iparam = *icommand->params; *p_iparam; p_iparam++) {
- kparam_t *kparam = NULL;
- iparam_t *iparam = *p_iparam;
- kparam = kparam_from_iparam(iparam, error_stack);
- if (!kparam) {
- retval = BOOL_FALSE;
- continue;
- }
- if (!kcommand_add_param(kcommand, kparam)) {
- // Search for PARAM duplicates
- if (kcommand_find_param(kcommand,
- kparam_name(kparam))) {
- faux_error_sprintf(error_stack,
- "COMMAND: "
- "Can't add duplicate PARAM "
- "\"%s\"",
- kparam_name(kparam));
- } else {
- faux_error_sprintf(error_stack,
- "COMMAND: "
- "Can't add PARAM \"%s\"",
- kparam_name(kparam));
- }
- kparam_free(kparam);
- retval = BOOL_FALSE;
- continue;
- }
- }
- }
- // ACTION list
- if (icommand->actions) {
- iaction_t **p_iaction = NULL;
- for (p_iaction = *icommand->actions; *p_iaction; p_iaction++) {
- kaction_t *kaction = NULL;
- iaction_t *iaction = *p_iaction;
- kaction = kaction_from_iaction(iaction, error_stack);
- if (!kaction) {
- retval = BOOL_FALSE;
- continue;
- }
- if (!kcommand_add_action(kcommand, kaction)) {
- faux_error_sprintf(error_stack, "COMMAND: "
- "Can't add ACTION #%d",
- faux_list_len(kcommand->actions) + 1);
- kaction_free(kaction);
- retval = BOOL_FALSE;
- continue;
- }
- }
- }
- if (!retval)
- faux_error_sprintf(error_stack,
- "COMMAND \"%s\": Illegal nested elements",
- kcommand_name(kcommand));
- return retval;
- }
- kcommand_t *kcommand_from_icommand(icommand_t *icommand, faux_error_t *error_stack)
- {
- kcommand_t *kcommand = NULL;
- kcommand_error_e kcommand_error = KCOMMAND_ERROR_OK;
- kcommand = kcommand_new(icommand, &kcommand_error);
- if (!kcommand) {
- faux_error_sprintf(error_stack, "COMMAND \"%s\": %s",
- icommand->name ? icommand->name : "(null)",
- kcommand_strerror(kcommand_error));
- return NULL;
- }
- // Parse nested elements
- if (!kcommand_nested_from_icommand(kcommand, icommand, error_stack)) {
- kcommand_free(kcommand);
- return NULL;
- }
- return kcommand;
- }
|