--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/rust/hg-cpython/src/dirstate/status.rs Tue Oct 08 08:45:55 2019 +0200
@@ -0,0 +1,82 @@
+// status.rs
+//
+// Copyright 2019, Raphaël Gomès <rgomes@octobus.net>
+//
+// This software may be used and distributed according to the terms of the
+// GNU General Public License version 2 or any later version.
+
+//! Bindings for the `hg::status` module provided by the
+//! `hg-core` crate. From Python, this will be seen as `rustext.dirstate.status`.
+//!
+
+use crate::dirstate::DirstateMap;
+use cpython::exc::ValueError;
+use cpython::{
+ PyBytes, PyErr, PyList, PyObject, PyResult, Python, PythonObject,
+ ToPyObject,
+};
+use hg::utils::files::get_path_from_bytes;
+
+use hg::utils::hg_path::HgPath;
+use hg::{status, utils::hg_path::HgPathBuf};
+
+/// This will be useless once trait impls for collection are added to `PyBytes`
+/// upstream.
+fn collect_pybytes_list<P: AsRef<HgPath>>(
+ py: Python,
+ collection: &[P],
+) -> PyList {
+ let list = PyList::new(py, &[]);
+
+ for (i, path) in collection.iter().enumerate() {
+ list.insert_item(
+ py,
+ i,
+ PyBytes::new(py, path.as_ref().as_bytes()).into_object(),
+ )
+ }
+
+ list
+}
+
+pub fn status_wrapper(
+ py: Python,
+ dmap: DirstateMap,
+ root_dir: PyObject,
+ files: PyList,
+ list_clean: bool,
+ last_normal_time: i64,
+ check_exec: bool,
+) -> PyResult<(PyList, PyList, PyList, PyList, PyList, PyList, PyList)> {
+ let bytes = root_dir.extract::<PyBytes>(py)?;
+ let root_dir = get_path_from_bytes(bytes.data(py));
+
+ let dmap: DirstateMap = dmap.to_py_object(py);
+ let dmap = dmap.get_inner(py);
+
+ let files: PyResult<Vec<HgPathBuf>> = files
+ .iter(py)
+ .map(|f| Ok(HgPathBuf::from_bytes(f.extract::<PyBytes>(py)?.data(py))))
+ .collect();
+ let files = files?;
+
+ let (lookup, status_res) = status(
+ &dmap,
+ &root_dir,
+ &files,
+ list_clean,
+ last_normal_time,
+ check_exec,
+ )
+ .map_err(|e| PyErr::new::<ValueError, _>(py, e.to_string()))?;
+
+ let modified = collect_pybytes_list(py, status_res.modified.as_ref());
+ let added = collect_pybytes_list(py, status_res.added.as_ref());
+ let removed = collect_pybytes_list(py, status_res.removed.as_ref());
+ let deleted = collect_pybytes_list(py, status_res.deleted.as_ref());
+ let clean = collect_pybytes_list(py, status_res.clean.as_ref());
+ let lookup = collect_pybytes_list(py, lookup.as_ref());
+ let unknown = PyList::new(py, &[]);
+
+ Ok((lookup, modified, added, removed, deleted, unknown, clean))
+}