llm
Download
107 lines 3.4 KiBCModified 2026-08-31 raw
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
// Type info and methods for CString datatype (char*)

#include <err.h>
#include <gc.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>

#include "integers.h"
#include "list.h"
#include "metamethods.h"
#include "siphash.h"
#include "stdlib.h"
#include "text.h"
#include "util.h"

public
Text_t CString$as_text(const void *c_string, bool colorize, const TypeInfo_t *info) {
    (void)info;
    if (!c_string) return Text("CString");
    Text_t text = Text$from_str(*(const char **)c_string);
    return Text$as_text(&text, colorize, &Text$info);
}

PUREFUNC public int32_t CString$compare(const void *x, const void *y, const TypeInfo_t *info) {
    (void)info;
    if (x == y) return 0;

    if (!*(const char **)x != !*(const char **)y) return (!*(const char **)y) - (!*(const char **)x);

    const char *x_str = *(const char **)x, *y_str = *(const char **)y;
    return (x_str == NULL) || (y_str == NULL) ? (x_str != NULL) - (y_str != NULL) : strcmp(x_str, y_str);
}

PUREFUNC public bool CString$equal(const void *x, const void *y, const TypeInfo_t *type) {
    return CString$compare(x, y, type) == 0;
}

PUREFUNC public uint64_t CString$hash(const void *c_str, const TypeInfo_t *info) {
    (void)info;
    if (!*(const char **)c_str) return 0;
    return siphash24(*(void **)c_str, strlen(*(const char **)c_str));
}

PUREFUNC public bool CString$is_none(const void *c_str, const TypeInfo_t *info) {
    (void)info;
    return *(const char **)c_str == NULL;
}

public
List_t CString$bytes(const char *str) {
    if (!str || str[0] == '\0') return EMPTY_ATOMIC_LIST;
    size_t len = strlen(str);
    uint8_t *buf = GC_MALLOC_ATOMIC(sizeof(uint8_t[len]));
    memcpy(buf, str, len);
    return (List_t){.data = buf, .length = (uint64_t)len, .stride = 1, .atomic = 1};
}

public
void CString$serialize(const void *obj, FILE *out, Table_t *pointers, const TypeInfo_t *info) {
    (void)info;
    const char *str = *(const char **)obj;
    int64_t len = (int64_t)strlen(str);
    Int64$serialize(&len, out, pointers, &Int64$info);
    fwrite(str, sizeof(char), (size_t)len, out);
}

public
void CString$deserialize(FILE *in, void *out, List_t *pointers, const TypeInfo_t *info) {
    (void)info;
    int64_t len = -1;
    Int64$deserialize(in, &len, pointers, &Int64$info);
    if (len < 0 || len > deserialization_bytes_remaining(in)) deserialization_failed();
    char *str = GC_MALLOC_ATOMIC((size_t)len + 1);
    if (str == NULL) deserialization_failed();
    if (fread(str, sizeof(char), (size_t)len, in) != (size_t)len) deserialization_failed();
    str[len] = '\0';
    *(const char **)out = str;
}

public
const char *CString$join(const char *glue, List_t strings) {
    Text_t ret = EMPTY_TEXT;
    Text_t glue_text = Text$from_str(glue);
    for (int64_t i = 0; i < (int64_t)strings.length; i++) {
        if (i > 0) ret = Text$concat(ret, glue_text);
        ret = Text$concat(ret, Text$from_str(*(const char **)(strings.data + i * strings.stride)));
    }
    return Text$as_c_string(ret);
}

public
const TypeInfo_t CString$info = {
    .size = sizeof(const char *),
    .align = __alignof__(const char *),
    .metamethods =
        {
            .hash = CString$hash,
            .compare = CString$compare,
            .equal = CString$equal,
            .as_text = CString$as_text,
            .is_none = CString$is_none,
            .serialize = CString$serialize,
            .deserialize = CString$deserialize,
        },
};