diff options
Diffstat (limited to '')
-rw-r--r-- | src/doveadm/doveadm-print-flow.c | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/src/doveadm/doveadm-print-flow.c b/src/doveadm/doveadm-print-flow.c new file mode 100644 index 0000000..7f4d01e --- /dev/null +++ b/src/doveadm/doveadm-print-flow.c @@ -0,0 +1,111 @@ +/* Copyright (c) 2010-2018 Dovecot authors, see the included COPYING file */ + +#include "lib.h" +#include "array.h" +#include "ostream.h" +#include "doveadm-print-private.h" + +struct doveadm_print_flow_header { + const char *title; + enum doveadm_print_header_flags flags; +}; + +struct doveadm_print_flow_context { + pool_t pool; + ARRAY(struct doveadm_print_flow_header) headers; + unsigned int header_idx; + + bool streaming:1; +}; + +static struct doveadm_print_flow_context *ctx; + +static void +doveadm_print_flow_header(const struct doveadm_print_header *hdr) +{ + struct doveadm_print_flow_header *fhdr; + + fhdr = array_append_space(&ctx->headers); + fhdr->title = p_strdup(ctx->pool, hdr->title); + fhdr->flags = hdr->flags; +} + +static void flow_next_hdr(void) +{ + if (++ctx->header_idx < array_count(&ctx->headers)) + o_stream_nsend(doveadm_print_ostream, " ", 1); + else { + ctx->header_idx = 0; + o_stream_nsend(doveadm_print_ostream, "\n", 1); + } +} + +static void doveadm_print_flow_print_header(const struct doveadm_print_flow_header *hdr) +{ + if ((hdr->flags & DOVEADM_PRINT_HEADER_FLAG_HIDE_TITLE) == 0) { + o_stream_nsend_str(doveadm_print_ostream, hdr->title); + o_stream_nsend(doveadm_print_ostream, "=", 1); + } +} + +static void doveadm_print_flow_print(const char *value) +{ + const struct doveadm_print_flow_header *hdr = + array_idx(&ctx->headers, ctx->header_idx); + + doveadm_print_flow_print_header(hdr); + o_stream_nsend_str(doveadm_print_ostream, value); + flow_next_hdr(); +} + +static void +doveadm_print_flow_print_stream(const unsigned char *value, size_t size) +{ + const struct doveadm_print_flow_header *hdr = + array_idx(&ctx->headers, ctx->header_idx); + + if (!ctx->streaming) { + ctx->streaming = TRUE; + doveadm_print_flow_print_header(hdr); + } + o_stream_nsend(doveadm_print_ostream, value, size); + if (size == 0) { + flow_next_hdr(); + ctx->streaming = FALSE; + } +} + +static void doveadm_print_flow_init(void) +{ + pool_t pool; + + pool = pool_alloconly_create("doveadm print flow", 1024); + ctx = p_new(pool, struct doveadm_print_flow_context, 1); + ctx->pool = pool; + p_array_init(&ctx->headers, pool, 16); +} + +static void doveadm_print_flow_flush(void) +{ + if (ctx->header_idx != 0) { + o_stream_nsend(doveadm_print_ostream, "\n", 1); + ctx->header_idx = 0; + } +} + +static void doveadm_print_flow_deinit(void) +{ + pool_unref(&ctx->pool); + ctx = NULL; +} + +struct doveadm_print_vfuncs doveadm_print_flow_vfuncs = { + "flow", + + doveadm_print_flow_init, + doveadm_print_flow_deinit, + doveadm_print_flow_header, + doveadm_print_flow_print, + doveadm_print_flow_print_stream, + doveadm_print_flow_flush +}; |