Skip to content

Commit

Permalink
app-parser-generator: extract AppObjectGenerator base class
Browse files Browse the repository at this point in the history 
Signed-off-by: Balazs Scheidler <[email protected]>
  • Loading branch information
@bazsi
bazsi committed May 15, 2024
1 parent cd6984f commit ab83465
Showing 1 changed file with 145 additions and 117 deletions.
262 changes: 145 additions & 117 deletions modules/appmodel/app-parser-generator.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,18 +27,143 @@

#include <string.h>

typedef struct _AppParserGenerator
typedef struct _AppObjectGenerator AppObjectGenerator;

struct _AppObjectGenerator
{
CfgBlockGenerator super;
GString *block;
const gchar *topic;
gboolean (*parse_arguments)(AppObjectGenerator *self, CfgArgs *args, const gchar *reference);
void (*generate_config)(AppObjectGenerator *self, GlobalConfig *cfg, GString *result);
const gchar *included_apps;
const gchar *excluded_apps;
gboolean is_parsing_enabled;
};

static gboolean
_is_application_included(AppObjectGenerator *self, const gchar *app_name)
{
/* include everything if we don't have the option */
if (!self->included_apps)
return TRUE;
return strstr(self->included_apps, app_name) != NULL;
}

static gboolean
_is_application_excluded(AppObjectGenerator *self, const gchar *app_name)
{
if (!self->excluded_apps)
return FALSE;
return strstr(self->excluded_apps, app_name) != NULL;
}

static gboolean
_parse_auto_parse_arg(AppObjectGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "auto-parse");

if (v)
self->is_parsing_enabled = cfg_process_yesno(v);
else
self->is_parsing_enabled = TRUE;
return TRUE;
}

static gboolean
_parse_auto_parse_exclude_arg(AppObjectGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "auto-parse-exclude");
if (!v)
return TRUE;
self->excluded_apps = g_strdup(v);
return TRUE;
}

static gboolean
_parse_auto_parse_include_arg(AppObjectGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "auto-parse-include");
if (!v)
return TRUE;
self->included_apps = g_strdup(v);
return TRUE;
}


static gboolean
app_object_generator_parse_arguments_method(AppObjectGenerator *self, CfgArgs *args, const gchar *reference)
{
g_assert(args != NULL);

if (!_parse_auto_parse_arg(self, args, reference))
return FALSE;
if (!_parse_auto_parse_exclude_arg(self, args, reference))
return FALSE;
if (!_parse_auto_parse_include_arg(self, args, reference))
return FALSE;
return TRUE;
}

static gboolean
_generate(CfgBlockGenerator *s, GlobalConfig *cfg, gpointer args, GString *result, const gchar *reference)
{
AppObjectGenerator *self = (AppObjectGenerator *) s;
CfgArgs *cfgargs = (CfgArgs *)args;

if (!self->parse_arguments(self, cfgargs, reference))
return FALSE;

self->generate_config(self, cfg, result);

return TRUE;
}

void
app_object_generator_init_instance(AppObjectGenerator *self, gint context, const gchar *name)
{
cfg_block_generator_init_instance(&self->super, context, name);
self->super.generate = _generate;
self->parse_arguments = app_object_generator_parse_arguments_method;
}

/* app-parser() */

typedef struct _AppParserGenerator
{
AppObjectGenerator super;
const gchar *topic;
GString *block;
gboolean first_app_generated;
gboolean allow_overlaps;
} AppParserGenerator;

static gboolean
_parse_topic_arg(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
self->topic = cfg_args_get(args, "topic");
if (!self->topic)
{
msg_error("app-parser() requires a topic() argument",
evt_tag_str("reference", reference));
return FALSE;
}
return TRUE;
}

static gboolean
app_parser_generator_parse_arguments(AppObjectGenerator *s, CfgArgs *args, const gchar *reference)
{
AppParserGenerator *self = (AppParserGenerator *) s;
g_assert(args != NULL);

if (!_parse_topic_arg(self, args, reference))
return FALSE;

if (!app_object_generator_parse_arguments_method(&self->super, args, reference))
return FALSE;

return TRUE;
}

static void
_generate_filter(AppParserGenerator *self, const gchar *filter_expr)
{
Expand Down Expand Up @@ -73,36 +198,18 @@ _generate_action(AppParserGenerator *self, Application *app)
}
}

static gboolean
_is_application_included(AppParserGenerator *self, Application *app)
{
/* include everything if we don't have the option */
if (!self->included_apps)
return TRUE;
return strstr(self->included_apps, app->super.name) != NULL;
}

static gboolean
_is_application_excluded(AppParserGenerator *self, Application *app)
{
if (!self->excluded_apps)
return FALSE;
return strstr(self->excluded_apps, app->super.name) != NULL;
}

static void
_generate_application(AppModelObject *object, gpointer user_data)
_generate_application(Application *app, gpointer user_data)
{
Application *app = (Application *) object;
AppParserGenerator *self = (AppParserGenerator *) user_data;

if (strcmp(self->topic, app->super.instance) != 0)
return;

if (!_is_application_included(self, app))
if (!_is_application_included(&self->super, app->super.name))
return;

if (_is_application_excluded(self, app))
if (_is_application_excluded(&self->super, app->super.name))
return;

if (self->first_app_generated)
Expand Down Expand Up @@ -133,21 +240,21 @@ _generate_application(AppModelObject *object, gpointer user_data)
}

static void
_generate_applications(AppParserGenerator *self, AppModelContext *appmodel)
_generate_applications(AppParserGenerator *self, GlobalConfig *cfg)
{
appmodel_context_iter_objects(appmodel, APPLICATION_TYPE_NAME, _generate_application, self);
appmodel_iter_applications(cfg, _generate_application, self);
}

static void
_generate_framing(AppParserGenerator *self, AppModelContext *appmodel)
_generate_framing(AppParserGenerator *self, GlobalConfig *cfg)
{
g_string_append(self->block,
"\nchannel {\n");

self->first_app_generated = FALSE;
if (!self->allow_overlaps)
{
_generate_applications(self, appmodel);
_generate_applications(self, cfg);
if (self->first_app_generated)
g_string_append(self->block, " else {\n");
else
Expand All @@ -159,7 +266,7 @@ _generate_framing(AppParserGenerator *self, AppModelContext *appmodel)
}
else
{
_generate_applications(self, appmodel);
_generate_applications(self, cfg);
if (self->first_app_generated)
g_string_append(self->block, " ;\n");
}
Expand All @@ -172,106 +279,27 @@ _generate_empty_frame(AppParserGenerator *self)
g_string_append(self->block, "channel { filter { tags('.app.doesnotexist'); }; };");
}

static gboolean
_parse_auto_parse_arg(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "auto-parse");

if (v)
self->is_parsing_enabled = cfg_process_yesno(v);
else
self->is_parsing_enabled = TRUE;
return TRUE;
}

static gboolean
_parse_auto_parse_exclude_arg(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "auto-parse-exclude");
if (!v)
return TRUE;
self->excluded_apps = g_strdup(v);
return TRUE;
}

static gboolean
_parse_auto_parse_include_arg(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "auto-parse-include");
if (!v)
return TRUE;
self->included_apps = g_strdup(v);
return TRUE;
}

static gboolean
_parse_topic_arg(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
self->topic = cfg_args_get(args, "topic");
if (!self->topic)
{
msg_error("app-parser() requires a topic() argument",
evt_tag_str("reference", reference));
return FALSE;
}
return TRUE;
}

static gboolean
_parse_allow_overlaps(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
const gchar *v = cfg_args_get(args, "allow-overlaps");
if (v)
self->allow_overlaps = cfg_process_yesno(v);
else
self->allow_overlaps = FALSE;
return TRUE;
}

static gboolean
_parse_arguments(AppParserGenerator *self, CfgArgs *args, const gchar *reference)
{
g_assert(args != NULL);

if (!_parse_topic_arg(self, args, reference))
return FALSE;
if (!_parse_auto_parse_arg(self, args, reference))
return FALSE;
if (!_parse_auto_parse_exclude_arg(self, args, reference))
return FALSE;
if (!_parse_auto_parse_include_arg(self, args, reference))
return FALSE;
if (!_parse_allow_overlaps(self, args, reference))
return FALSE;
return TRUE;
}

static gboolean
_generate(CfgBlockGenerator *s, GlobalConfig *cfg, gpointer args, GString *result, const gchar *reference)
void
app_parser_generate_config(AppObjectGenerator *s, GlobalConfig *cfg, GString *result)
{
AppParserGenerator *self = (AppParserGenerator *) s;
AppModelContext *appmodel = appmodel_get_context(cfg);
CfgArgs *cfgargs = (CfgArgs *)args;

if (!_parse_arguments(self, cfgargs, reference))
return FALSE;

self->block = result;
if (self->is_parsing_enabled)
_generate_framing(self, appmodel);
if (self->super.is_parsing_enabled)
_generate_framing(self, cfg);
else
_generate_empty_frame(self);
self->block = NULL;

return TRUE;
}


CfgBlockGenerator *
app_parser_generator_new(gint context, const gchar *name)
{
AppParserGenerator *self = g_new0(AppParserGenerator, 1);

cfg_block_generator_init_instance(&self->super, context, name);
self->super.generate = _generate;
return &self->super;
app_object_generator_init_instance(&self->super, context, name);
self->super.parse_arguments = app_parser_generator_parse_arguments;
self->super.generate_config = app_parser_generate_config;
return &self->super.super;
}

0 comments on commit ab83465

Please sign in to comment.