mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
8a780a58aa
This change fixes the pipeline behavior when producing targets: * produce and extract now return rp_buffer, which has a start pointer and size * the python api `revng.api` will return on produce/extract either str or bytes depending on the mime type of the container * the GraphQL api will encode bytes in base64
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
from pathlib import Path
|
|
from typing import Callable, Generator, Optional, TypeVar, Union
|
|
|
|
from ._capi import ffi
|
|
|
|
|
|
def make_python_string(s: ffi.CData) -> str:
|
|
if s == ffi.NULL:
|
|
return ""
|
|
bytestring = ffi.string(s)
|
|
return bytestring.decode("utf-8")
|
|
|
|
|
|
def make_c_string(s: Union[bytes, str]):
|
|
if isinstance(s, bytes):
|
|
return ffi.new("char[]", s)
|
|
if isinstance(s, str):
|
|
return ffi.new("char[]", s.encode("utf-8"))
|
|
raise TypeError(f"Invalid type: {s.__class__}")
|
|
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
def make_generator(count: int, get_idx: Callable[[int], Optional[T]]) -> Generator[T, None, None]:
|
|
for i in range(count):
|
|
ret = get_idx(i)
|
|
if ret is not None:
|
|
yield ret
|
|
|
|
|
|
def save_file(path: Path, content: Union[bytes, str]):
|
|
if isinstance(content, str):
|
|
with open(path.resolve(), "w", encoding="utf-8") as str_f:
|
|
str_f.write(content)
|
|
else:
|
|
with open(path.resolve(), "wb") as bytes_f:
|
|
bytes_f.write(content)
|
|
|
|
|
|
def convert_buffer(ptr: ffi.CData, size: int, mime: str) -> str | bytes:
|
|
if ptr == ffi.NULL:
|
|
return b""
|
|
|
|
data = ffi.unpack(ptr, size)
|
|
if mime.startswith("text/") or mime == "image/svg":
|
|
return data.decode("utf-8")
|
|
else:
|
|
return data
|