pocketpy/src/interpreter/py_compile.c

#include "pocketpy/pocketpy.h"
#include "pocketpy/interpreter/vm.h"
#include <errno.h>

bool py_compilefile(const char* src_path, const char* dst_path) {
    // read
    FILE* fp = fopen(src_path, "rb");
    if(fp == NULL) {
        const char* msg = strerror(errno);
        return OSError("[Errno %d] %s: '%s'", errno, msg, src_path);
    }
    fseek(fp, 0, SEEK_END);
    long size = ftell(fp);
    fseek(fp, 0, SEEK_SET);
    char* buffer = PK_MALLOC(size + 1);
    size = fread(buffer, 1, size, fp);
    buffer[size] = 0;
    fclose(fp);
    // compile
    bool ok = py_compile(buffer, src_path, EXEC_MODE, false);
    PK_FREE(buffer);
    if(!ok) return false;
    // dump
    py_assign(py_pushtmp(), py_retval());
    int bc_size;
    void* bc_data = CodeObject__dumps(py_touserdata(py_peek(-1)), &bc_size);
    py_pop();
    // write
    fp = fopen(dst_path, "wb");
    if(fp == NULL) {
        PK_FREE(bc_data);
        const char* msg = strerror(errno);
        return OSError("[Errno %d] %s: '%s'", errno, msg, dst_path);
    }
    fwrite(bc_data, 1, bc_size, fp);
    fclose(fp);
    PK_FREE(bc_data);
    return true;
}