use cpython::exc::ValueError;

use cpython::{PyBytes, PyDict, PyErr, PyObject, PyResult, PyTuple, Python};

use hg::revlog::Node;

use std::convert::TryFrom;

#[allow(unused)]

pub fn print_python_trace(py: Python) -> PyResult<PyObject> {

    eprintln!("===============================");

    eprintln!("Printing Python stack from Rust");

    eprintln!("===============================");

    let traceback = py.import("traceback")?;

    let sys = py.import("sys")?;

    let kwargs = PyDict::new(py);

    kwargs.set_item(py, "file", sys.get(py, "stderr")?)?;

    traceback.call(py, "print_stack", PyTuple::new(py, &[]), Some(&kwargs))

}

// Necessary evil for the time being, could maybe be moved to

// a TryFrom in Node itself

const NODE_BYTES_LENGTH: usize = 20;

type NodeData = [u8; NODE_BYTES_LENGTH];

/// Copy incoming Python bytes given as `PyObject` into `Node`,

/// doing the necessary checks

pub fn node_from_py_object<'a>(

    py: Python,

    bytes: &'a PyObject,

) -> PyResult<Node> {

    let as_py_bytes: &'a PyBytes = bytes.extract(py)?;

    node_from_py_bytes(py, as_py_bytes)

}

/// Clone incoming Python bytes given as `PyBytes` as a `Node`,

/// doing the necessary checks.

pub fn node_from_py_bytes(py: Python, bytes: &PyBytes) -> PyResult<Node> {

    <NodeData>::try_from(bytes.data(py))

        .map_err(|_| {

            PyErr::new::<ValueError, _>(

                py,

                format!("{}-byte hash required", NODE_BYTES_LENGTH),

            )

        })

        .map(Into::into)

}