pocketpy/src/error.c

#include "pocketpy/objects/error.h"
#include "pocketpy/common/strname.h"
#include "pocketpy/common/sstream.h"

void pkpy_Exception__ctor(pkpy_Exception* self, pkpy_StrName type){
    self->type = type;
    self->is_re = true;
    self->_ip_on_error = -1;
    self->_code_on_error = NULL;
    self->self = NULL;

    pkpy_Str__ctor(&self->msg, "");
    c11_vector__ctor(&self->stacktrace, sizeof(pkpy_ExceptionFrame));
}

void pkpy_Exception__dtor(pkpy_Exception* self){
    for(int i=0; i<self->stacktrace.count; i++){
        pkpy_ExceptionFrame* frame = c11__at(pkpy_ExceptionFrame, &self->stacktrace, i);
        PK_DECREF(frame->src);
        pkpy_Str__dtor(&frame->name);
    }
    pkpy_Str__dtor(&self->msg);
    c11_vector__dtor(&self->stacktrace);
}

void pkpy_Exception__stpush(pkpy_Exception* self, pkpy_SourceData_ src, int lineno, const char* cursor, const char* name){
    if(self->stacktrace.count >= 7) return;
    PK_INCREF(src);
    pkpy_ExceptionFrame* frame = c11_vector__emplace(&self->stacktrace);
    frame->src = src;
    frame->lineno = lineno;
    frame->cursor = cursor;
    pkpy_Str__ctor(&frame->name, name);
}

pkpy_Str pkpy_Exception__summary(pkpy_Exception* self){
    pkpy_SStream ss;
    pkpy_SStream__ctor(&ss);

    if(self->is_re){
        pkpy_SStream__write_cstr(&ss, "Traceback (most recent call last):\n");
    }
    for(int i=self->stacktrace.count-1; i >= 0; i--) {
        pkpy_ExceptionFrame* frame = c11__at(pkpy_ExceptionFrame, &self->stacktrace, i);
        pkpy_Str s = pkpy_SourceData__snapshot(frame->src, frame->lineno, frame->cursor, pkpy_Str__data(&frame->name));
        pkpy_SStream__write_Str(&ss, &s);
        pkpy_Str__dtor(&s);
        pkpy_SStream__write_cstr(&ss, "\n");
    }

    const char* name = pkpy_StrName__rmap(self->type);
    pkpy_SStream__write_cstr(&ss, name);

    if(self->msg.size > 0){
        pkpy_SStream__write_cstr(&ss, ": ");
        pkpy_SStream__write_Str(&ss, &self->msg);
    }
    return pkpy_SStream__submit(&ss);
}