path: root/src/chunk.h

#ifndef CMARK_CHUNK_H
#define CMARK_CHUNK_H

#include <string.h>
#include <ctype.h>
#include <stdlib.h>
#include <assert.h>
#include "buffer.h"

typedef struct {
    const unsigned char *data;
    int len;
    int alloc;
} cmark_chunk;

static inline void cmark_chunk_free(cmark_chunk *c)
{
    if (c->alloc)
        free((char *)c->data);

    c->data = NULL;
    c->alloc = 0;
    c->len = 0;
}

static inline void cmark_chunk_ltrim(cmark_chunk *c)
{
    assert(!c->alloc);

    while (c->len && isspace(c->data[0])) {
        c->data++;
        c->len--;
    }
}

static inline void cmark_chunk_rtrim(cmark_chunk *c)
{
    while (c->len > 0) {
        if (!isspace(c->data[c->len - 1]))
            break;

        c->len--;
    }
}

static inline void cmark_chunk_trim(cmark_chunk *c)
{
    cmark_chunk_ltrim(c);
    cmark_chunk_rtrim(c);
}

static inline int cmark_chunk_strchr(cmark_chunk *ch, int c, int offset)
{
    const unsigned char *p = (unsigned char *)memchr(ch->data + offset, c, ch->len - offset);
    return p ? (int)(p - ch->data) : ch->len;
}

static inline unsigned char *cmark_chunk_to_cstr(cmark_chunk *c)
{
    unsigned char *str;

    str = (unsigned char *)calloc(c->len + 1, sizeof(*str));
    if(str != NULL) {
        memcpy(str, c->data, c->len);
        str[c->len] = 0;
    }
    return str;
}

static inline cmark_chunk cmark_chunk_literal(const char *data)
{
    cmark_chunk c = {(const unsigned char *)data, data ? strlen(data) : 0, 0};
    return c;
}

static inline cmark_chunk cmark_chunk_dup(const cmark_chunk *ch, int pos, int len)
{
    cmark_chunk c = {ch->data + pos, len, 0};
    return c;
}

static inline cmark_chunk cmark_chunk_buf_detach(cmark_strbuf *buf)
{
    cmark_chunk c;

    c.len = buf->size;
    c.data = cmark_strbuf_detach(buf);
    c.alloc = 1;

    return c;
}

#ifndef CMARK_NO_SHORT_NAMES
  #define chunk             cmark_chunk
  #define chunk_free        cmark_chunk_free
  #define chunk_ltrim       cmark_chunk_ltrim
  #define chunk_rtrim       cmark_chunk_rtrim
  #define chunk_trim        cmark_chunk_trim
  #define chunk_strchr      cmark_chunk_strchr
  #define chunk_to_cstr     cmark_chunk_to_cstr
  #define chunk_literal     cmark_chunk_literal
  #define chunk_dup         cmark_chunk_dup
  #define chunk_buf_detach  cmark_chunk_buf_detach
#endif

#endif