chore: finish bindings

Author: Brooooooklyn

.gitignore

@@ -1,3 +1,6 @@
 /target
 **/*.rs.bk
 Cargo.lock
+*.pyc
+build
+*.so

Cargo.toml

@@ -10,3 +10,6 @@ crate-type = ["cdylib"]
 [dependencies]
 pyo3 = { version="0.4", features=["extension-module"] }
 source-map-mappings = { version="0.5" }
+serde = "1.0"
+serde_derive = "1.0"
+serde_json = "1.0"

py_sourcemap/__init__.py

@@ -0,0 +1,3 @@
+from .py_sourcemap import SourcemapParser
+
+__all__ = ["SourcemapParser"]

requirements-dev.txt

@@ -0,0 +1 @@
+setuptools-rust==0.10.2

setup.py

@@ -0,0 +1,11 @@
+from setuptools import setup
+from setuptools_rust import Binding, RustExtension
+
+setup(name='py-sourcemap',
+  version='0.1',
+  rust_extensions=[RustExtension('py_sourcemap.py_sourcemap',
+                                  'Cargo.toml', binding=Binding.PyO3)],
+  packages=['py_sourcemap'],
+  # rust extensions are not zip safe, just like C-extensions.
+  zip_safe=False
+)

src/lib.rs

@@ -2,35 +2,66 @@
 
 extern crate pyo3;
 extern crate source_map_mappings;
+#[macro_use] extern crate serde_derive;
+extern crate serde;
+extern crate serde_json;
 
 use std::fs::File;
-use std::io::{BufReader, Read};
 
 use pyo3::prelude::*;
-use source_map_mappings::{Bias, Mappings, parse_mappings};
+use source_map_mappings::{Bias, Mapping, Mappings, parse_mappings};
+
+#[derive(Serialize, Deserialize, Debug)]
+#[serde(rename_all = "camelCase")]
+struct RawSourceMap {
+  version: u8,
+  sources: Vec<String>,
+  names: Vec<String>,
+  mappings: String,
+  file: String,
+  source_root: Option<String>,
+}
 
 #[pyclass]
 struct SourcemapParser {
-  parsed_map: Mappings<()>
+  parsed_map: Mappings<()>,
+  sources: Vec<String>,
+  names: Vec<String>,
 }
 
 #[pymethods]
 impl SourcemapParser {
   #[new]
   fn __new__(obj: &PyRawObject, path: &str) -> PyResult<()> {
     let file = File::open(path).map_err(PyErr::from)?;
-    let mut buffers = vec![];
-    let mut reader = BufReader::new(file);
-    reader.read_to_end(&mut buffers).map_err(PyErr::from)?;
-    let mappings = parse_mappings(&buffers).map_err(|_| PyErr::new::<exc::TypeError, _>(format!("Parse Sourcemap failed: {}", path)))?;
-    obj.init(|_| SourcemapParser {
+    let raw_sourcemap: RawSourceMap = serde_json::from_reader(file).map_err(|e| {
+      PyErr::new::<exc::TypeError, _>(format!("{:?}", e))
+    })?;
+    let mapping_bytes = raw_sourcemap.mappings.as_bytes();
+    let mappings = parse_mappings(mapping_bytes).map_err(|_| {
+      PyErr::new::<exc::TypeError, _>(format!("Parse Sourcemap failed: {}", path))
+    })?;
+    let sources = raw_sourcemap.sources;
+    let names = raw_sourcemap.names;
+    obj.init(move |_| SourcemapParser {
       parsed_map: mappings,
+      sources,
+      names,
     })
   }
 
-  fn original_location_for(&self, generated_line: u32, generated_column: u32) -> PyResult<(u32, u32)> {
-    if let Some(mapping) = self.parsed_map.original_location_for(generated_line, generated_column, Bias::LeastUpperBound) {
-      return Ok((mapping.generated_line, mapping.generated_column))
+  fn original_location_for(&self, generated_line: u32, generated_column: u32) -> PyResult<(u32, u32, Option<String>, Option<String>)> {
+    if let Some(Mapping { original, .. }) = self.parsed_map.original_location_for(generated_line, generated_column, Bias::LeastUpperBound) {
+      match original {
+        Some(location) => {
+          let name = location.name.and_then(|index| {
+            self.names.get(index as usize).map(|str_slice| str_slice.to_string())
+          });
+          let source = self.sources.get(location.source as usize).map(|str_slice| str_slice.to_string());
+          return Ok((location.original_line, location.original_column, source, name))
+        },
+        None => return Err(PyErr::new::<exc::TypeError, _>("No original lines"))
+      };
     }
     Err(PyErr::new::<exc::TypeError, _>("No sources found"))
   }

tests/index.js.map

@@ -0,0 +1,7 @@
+from py_sourcemap import SourcemapParser
+
+sourcemap = SourcemapParser("./tests/index.js.map")
+
+result = sourcemap.original_location_for(0, 195302)
+
+print(result)