doveadm-print-formatted.c revision 44aebbf6ddecda29a52d5a0fd6e766a1653a6a38
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "lib.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "array.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "str.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "ostream.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "client-connection.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "doveadm-server.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "doveadm-print.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "doveadm-print-private.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen#include "var-expand.h"
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstruct doveadm_print_formatted_context {
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen pool_t pool;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen const char *format;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen ARRAY(struct var_expand_table) headers;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen string_t *buf;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen string_t *vbuf;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen unsigned int idx;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen};
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstatic struct doveadm_print_formatted_context ctx;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenvoid doveadm_print_formatted_set_format(const char *format)
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen{
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen ctx.format = format;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen}
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstatic void doveadm_print_formatted_init(void)
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen{
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen memset(&ctx,0,sizeof(ctx));
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen ctx.pool = pool_alloconly_create("doveadm formatted print", 1024);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen ctx.buf = str_new(ctx.pool, 256);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen p_array_init(&ctx.headers, ctx.pool, 8);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen ctx.idx = 0;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen}
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstatic void
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainendoveadm_print_formatted_header(const struct doveadm_print_header *hdr)
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen{
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen struct var_expand_table entry;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen memset(&entry, 0, sizeof(entry));
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen entry.key = '\0';
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen entry.long_key = p_strdup(ctx.pool, hdr->key);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen entry.value = NULL;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen array_append(&ctx.headers, &entry, 1);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen}
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstatic void doveadm_print_formatted_flush(void)
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen{
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen o_stream_nsend(doveadm_print_ostream, str_data(ctx.buf), str_len(ctx.buf));
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen str_truncate(ctx.buf, 0);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen}
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstatic void doveadm_print_formatted_print(const char *value)
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen{
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen if (ctx.format == NULL) {
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen i_fatal("formatted formatter cannot be used without a format.");
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen }
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen struct var_expand_table *entry = array_idx_modifiable(&ctx.headers, ctx.idx++);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen entry->value = value;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen if (ctx.idx >= array_count(&ctx.headers)) {
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen var_expand(ctx.buf, ctx.format, array_idx(&ctx.headers,0));
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen doveadm_print_formatted_flush();
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen ctx.idx = 0;
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen }
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen}
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstatic void doveadm_print_formatted_deinit(void)
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen{
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen pool_unref(&ctx.pool);
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen}
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainenstruct doveadm_print_vfuncs doveadm_print_formatted_vfuncs = {
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen "formatted",
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen doveadm_print_formatted_init,
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen doveadm_print_formatted_deinit,
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen doveadm_print_formatted_header,
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen doveadm_print_formatted_print,
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen NULL,
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen doveadm_print_formatted_flush
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen};
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen
4654f788834c9d7920a351306b89cf5d1c21772eTimo Sirainen