pocketpy/include/pocketpy/objects/sourcedata.hpp

#pragma once

#include "pocketpy/common/utils.h"
#include "pocketpy/common/str.hpp"
#include "pocketpy/objects/sourcedata.h"

namespace pkpy {

struct SourceData : public pkpy_SourceData {
    SourceData(std::string_view source, const Str& filename, CompileMode mode) {
        pkpy_SourceData__ctor(this, {source.data(), (int)source.size()}, &filename, mode);
    }

    ~SourceData() {
        pkpy_SourceData__dtor(this);
    }

    std::string_view get_line(int lineno) const {
        const char *st, *ed;
        if (pkpy_SourceData__get_line(this, lineno, &st, &ed)) {
            return std::string_view(st, ed - st);
        }
        return "<?>";
    }

    Str snapshot(int lineno, const char* cursor, std::string_view name) const {
        return pkpy_SourceData__snapshot(this, lineno, cursor, name.empty() ? nullptr : name.data());
    }
};

}  // namespace pkpy