add __repr__

This commit is contained in:
William Bell
2025-09-13 01:17:16 +01:00
parent 5846adf025
commit 4be8e8e32f
3 changed files with 72 additions and 1 deletions

View File

@@ -7,12 +7,54 @@
#include "string.h"
#include "../number/number.h"
#include "../object.h"
#include <ctype.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
ArgonObject *ARGON_STRING_TYPE = NULL;
char *c_quote_string(const char *input, size_t len) {
// Worst case: every byte becomes "\uXXXX" (6 chars) + quotes + NUL
size_t max_out = 2 + (len * 6) + 1;
char *out = malloc(max_out);
if (!out) return NULL;
size_t j = 0;
out[j++] = '"';
for (size_t i = 0; i < len; i++) {
unsigned char c = (unsigned char)input[i];
switch (c) {
case '\n':
out[j++] = '\\'; out[j++] = 'n'; break;
case '\t':
out[j++] = '\\'; out[j++] = 't'; break;
case '\r':
out[j++] = '\\'; out[j++] = 'r'; break;
case '\\':
out[j++] = '\\'; out[j++] = '\\'; break;
case '\"':
out[j++] = '\\'; out[j++] = '\"'; break;
default:
if (isprint(c)) {
out[j++] = c;
} else {
// write \uXXXX
j += sprintf(&out[j], "\\u%04X", c);
}
}
}
out[j++] = '"';
out[j] = '\0';
return out;
}
void init_string(ArgonObject*object,char *data, size_t length, uint64_t prehash,
uint64_t hash) {
add_builtin_field(object, field_length,

View File

@@ -10,6 +10,8 @@
extern ArgonObject *ARGON_STRING_TYPE;
char *c_quote_string(const char *input, size_t len);
void init_string(ArgonObject*object,char *data, size_t length, uint64_t prehash,
uint64_t hash);

View File

@@ -310,6 +310,18 @@ ArgonObject *BASE_CLASS___string__(size_t argc, ArgonObject **argv, ArErr *err,
return new_string_object_null_terminated(buffer);
}
ArgonObject *BASE_CLASS___repr__(size_t argc, ArgonObject **argv,
ArErr *err, RuntimeState *state) {
if (argc != 1) {
*err = create_err(0, 0, 0, "", "Runtime Error",
"__repr__ expects 1 arguments, got %" PRIu64, argc);
}
ArgonObject *string_method = get_builtin_field_for_class(
get_builtin_field(argv[0], __class__), __string__, argv[0]);
return argon_call(string_method, 0, NULL, err, state);
}
ArgonObject *BASE_CLASS___boolean__(size_t argc, ArgonObject **argv, ArErr *err,
RuntimeState *state) {
(void)argv;
@@ -434,6 +446,18 @@ ArgonObject *ARGON_STRING_TYPE___string__(size_t argc, ArgonObject **argv,
}
return argv[0];
}
ArgonObject *ARGON_STRING_TYPE___repr__(size_t argc, ArgonObject **argv,
ArErr *err, RuntimeState *state) {
(void)state;
if (argc != 1) {
*err = create_err(0, 0, 0, "", "Runtime Error",
"__repr__ expects 1 arguments, got %" PRIu64, argc);
}
char* quoted = c_quote_string(argv[0]->value.as_str->data,argv[0]->value.as_str->length);
ArgonObject* result = new_string_object_null_terminated(quoted);
free(quoted);
return result;
}
ArgonObject *ARGON_STRING_TYPE___hash__(size_t argc, ArgonObject **argv,
ArErr *err, RuntimeState *state) {
(void)state;
@@ -591,7 +615,7 @@ void bootstrap_types() {
create_argon_native_function("__string__", BASE_CLASS___string__));
add_builtin_field(
BASE_CLASS, __repr__,
create_argon_native_function("__repr__", BASE_CLASS___string__));
create_argon_native_function("__repr__", BASE_CLASS___repr__));
add_builtin_field(
ARGON_TYPE_TYPE, __call__,
create_argon_native_function("__call__", ARGON_TYPE_TYPE___call__));
@@ -619,6 +643,9 @@ void bootstrap_types() {
add_builtin_field(
ARGON_STRING_TYPE, __string__,
create_argon_native_function("__string__", ARGON_STRING_TYPE___string__));
add_builtin_field(
ARGON_STRING_TYPE, __repr__,
create_argon_native_function("__repr__", ARGON_STRING_TYPE___repr__));
add_builtin_field(
ARGON_BOOL_TYPE, __new__,
create_argon_native_function("__new__", ARGON_BOOL_TYPE___new__));