initial upload

4 files changed, 400 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..a64a8fff9991
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,19 @@
+# Generated by mksrcinfo v8
+# Fri Dec 25 22:42:51 UTC 2015
+pkgbase = python2-leveldb_mcpe
+	pkgdesc = A python wrapper for leveldb_mcpe
+	pkgver = 1.8
+	pkgrel = 1
+	url = https://github.com/Mojang/leveldb-mcpe
+	arch = any
+	license = BSD
+	makedepends = python2
+	makedepends = python2-setuptools
+	depends = leveldb-mcpe
+	source = setup.py
+	source = leveldb_mcpe.cpp
+	md5sums = f8a2813ef64b7a06629f0d8e50bc35fd
+	md5sums = a9b6fcc1a582507045d87348e31f8ecb
+
+pkgname = python2-leveldb_mcpe
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..ac30dc6bfc10
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,25 @@
+# Maintainer: nic96
+pkgname=python2-leveldb_mcpe
+pkgver=1.8
+pkgrel=1
+pkgdesc="A python wrapper for leveldb_mcpe"
+arch=("any")
+url="https://github.com/Mojang/leveldb-mcpe"
+license=('BSD')
+depends=('leveldb-mcpe')
+makedepends=('python2' 'python2-setuptools')
+source=(setup.py
+       leveldb_mcpe.cpp)
+
+md5sums=('f8a2813ef64b7a06629f0d8e50bc35fd'
+         'a9b6fcc1a582507045d87348e31f8ecb')
+
+build() {
+  cd "$srcdir"
+  python2 setup.py build
+}
+
+package() {
+  cd "$srcdir"
+  python2 setup.py install --root=/$pkgdir/ --optimize=1
+}
diff --git a/leveldb_mcpe.cpp b/leveldb_mcpe.cpp
new file mode 100644
index 000000000000..7527a131fafa
--- /dev/null
+++ b/leveldb_mcpe.cpp
@@ -0,0 +1,326 @@
+//Rubisks trying to wrap leveldb-mcpe
+#include <leveldb/cache.h>
+#include <leveldb/comparator.h>
+#include <leveldb/db.h>
+#include <leveldb/env.h>
+#include <leveldb/filter_policy.h>
+#include <leveldb/write_batch.h>
+#include <leveldb/zlib_compressor.h>
+#define BOOST_PYTHON_STATIC_LIB
+#define BOOST_SYSTEM_STATIC_LIB
+#define BOOST_ALL_STATIC_LIB
+
+#include <boost/python/module.hpp>
+#include <boost/python/def.hpp>
+#include <boost/python/object.hpp>
+#include <boost/python/str.hpp>
+#include <boost/python/extract.hpp>
+#include <boost/system/config.hpp>
+#include <boost/python.hpp>
+#include <stdint.h>
+#include <stdio.h>
+
+namespace bp = boost::python;
+
+//Exception
+class LevelDBException {
+public:
+	LevelDBException(const std::string msg)
+		: message(msg) {}
+
+	std::string getMessage() const {
+		return message;
+	}
+private:
+	std::string message;
+};
+
+static void ExceptionTranslator(const LevelDBException &err) {
+	PyErr_SetString(PyExc_RuntimeError, err.getMessage().c_str());
+};
+
+struct IteratorWrapper {
+	leveldb::Iterator* _it;
+	
+	IteratorWrapper(leveldb::Iterator* it){
+		this->_it = it;
+	}
+
+  ~IteratorWrapper()
+  {
+    delete _it;
+  }
+
+	bool Valid()
+	{
+		return this->_it->Valid();
+	}
+
+	void SeekToFirst()
+	{
+		this->_it->SeekToFirst();
+	}
+
+	void SeekToLast()
+	{
+		this->_it->SeekToLast();
+	}
+
+	void Seek(PyObject* _target)
+	{
+		const std::string target = bp::extract<const std::string>(_target);
+		return this->_it->Seek(target);
+	}
+
+	void Next()
+	{
+		this->_it->Next();
+	}
+
+	void Prev()
+	{
+		this->_it->Prev();
+	}
+	
+	std::string key()
+	{
+		return this->_it->key().ToString();
+	}
+
+	std::string value()
+	{
+		return this->_it->value().ToString();
+	}
+
+	void status()
+	{
+		leveldb::Status s = this->_it->status();
+
+		if (!s.ok()){
+			throw LevelDBException(s.ToString());
+		}
+	}
+};
+
+//write_batch.h
+struct WriteBatchWrapper
+{
+	leveldb::WriteBatch* _wb;
+
+	WriteBatchWrapper(){
+		_wb = new leveldb::WriteBatch();
+        }
+
+  ~WriteBatchWrapper(){
+    delete _wb;
+  }
+
+	void Put(PyObject* _key, PyObject* _value)
+	{
+		const std::string key = bp::extract<const std::string>(_key);
+		const std::string value = bp::extract<const std::string>(_value);
+		this->_wb->Put(key, value);
+	}
+
+	void Delete(PyObject* _key)
+	{
+		const std::string key = bp::extract<const std::string>(_key);
+		this->_wb->Delete(key);
+	}
+};
+
+
+//leveldb::DB wrapper
+class DBWrap
+{
+public:
+  leveldb::DB * _db;
+
+  DBWrap(PyObject* _options, PyObject* _name) //Open(options, name, db)
+    : _db(NULL)
+  {
+    leveldb::Options options = bp::extract<leveldb::Options&>(_options);
+	options.compressors[0] = new leveldb::ZlibCompressor();
+    std::string name = bp::extract<std::string>(_name);
+    leveldb::Status s = leveldb::DB::Open(options, name, &_db);
+
+    if(!s.ok())
+    {
+      throw LevelDBException(s.ToString());
+    }
+  }
+
+  ~DBWrap()
+  {
+    delete _db;
+  }
+
+  void Put(PyObject* _options, PyObject* _key, PyObject* _value) //Put(options, key, value)
+  {
+	  const leveldb::WriteOptions& options = bp::extract<const leveldb::WriteOptions&>(_options);
+	  const std::string key = bp::extract<const std::string>(_key);
+	  const std::string value = bp::extract<const std::string>(_value);
+	  leveldb::Status s = this->_db->Put(options, key, value);
+
+	  if (!s.ok()){
+		  throw LevelDBException(s.ToString());
+	  }
+  }
+
+  void Delete(PyObject* _options, PyObject* _key) //Delete(options, key)
+  {
+	  const leveldb::WriteOptions& options = bp::extract<const leveldb::WriteOptions&>(_options);
+	  const std::string key = bp::extract<const std::string>(_key);
+	  leveldb::Status s = this->_db->Delete(options, key);
+
+	  if (!s.ok()){
+		  throw LevelDBException(s.ToString());
+	  }
+  }
+
+  std::string Get(PyObject* _options, PyObject* _key) //Delete(options, key)
+  {
+	  const leveldb::ReadOptions& options = bp::extract<const leveldb::ReadOptions&>(_options);
+	  const std::string key = bp::extract<const std::string>(_key);
+	  std::string value;
+	  leveldb::Status s = this->_db->Get(options, key, &value);
+
+	  if (!s.ok()){
+		  throw LevelDBException(s.ToString());
+	  }
+	  return value;
+  }
+
+
+  void Write(PyObject* _options, PyObject* _updates)
+  {
+	  const leveldb::WriteOptions& options = bp::extract<const leveldb::WriteOptions&>(_options);
+	  WriteBatchWrapper& __updates = bp::extract<WriteBatchWrapper&>(_updates);
+	  leveldb::Status s = this->_db->Write(options, __updates._wb);
+
+	  if (!s.ok()){
+		  throw LevelDBException(s.ToString());
+	  }
+  }
+
+
+  IteratorWrapper* NewIterator(const leveldb::ReadOptions& options)
+  {
+	  leveldb::Iterator* it = this->_db->NewIterator(options);
+	  return new IteratorWrapper(it);
+  }
+
+  const leveldb::Snapshot* GetSnapshot()
+  {
+	  return _db->GetSnapshot();
+  }
+
+  void GetApproximateSizes(const leveldb::Range* range, int n, uint64_t* sizes)
+  {
+    this->_db->GetApproximateSizes(range, n, sizes);
+  }
+
+
+  void ReleaseSnapshot(const leveldb::Snapshot* snapshot)
+  {
+    this->_db->ReleaseSnapshot(snapshot);
+  }
+
+  bool GetProperty(const leveldb::Slice& property, std::string* value)
+  {
+    return this->_db->GetProperty(property, value);
+  }
+
+  void CompactRange(const leveldb::Slice* begin, const leveldb::Slice* end)
+  {
+    this->_db->CompactRange(begin, end);
+  }
+};
+
+class RepairWrapper{
+public:
+	~RepairWrapper(){};
+
+	RepairWrapper(PyObject* _path, PyObject* _options)
+	{
+		const std::string path = bp::extract<const std::string>(_path);
+		const leveldb::Options& options = bp::extract<const leveldb::Options&>(_options);
+		leveldb::Status s = leveldb::RepairDB(path, options);
+
+		if (!s.ok()){
+			throw LevelDBException(s.ToString());
+		}
+	}
+};
+
+BOOST_PYTHON_MODULE(leveldb_mcpe)
+{
+  //Exceptions
+  bp::register_exception_translator<LevelDBException>(&ExceptionTranslator);
+
+  //leveldb/db.h
+  bp::class_<DBWrap, boost::noncopyable>("DB", bp::init<PyObject*, PyObject*>())
+    .def("Put", &DBWrap::Put)
+    .def("Delete", &DBWrap::Delete)
+    .def("Write", &DBWrap::Write)
+    .def("Get", &DBWrap::Get)
+    .def("NewIterator", &DBWrap::NewIterator, 
+		bp::return_value_policy<bp::manage_new_object>())
+    .def("GetSnapshot", &DBWrap::GetSnapshot,
+		bp::return_value_policy<bp::reference_existing_object>())
+    .def("ReleaseSnapshot", &DBWrap::ReleaseSnapshot)
+    .def("GetProperty", &DBWrap::GetProperty)
+	.def("GetApproximateSizes", &DBWrap::GetApproximateSizes)
+	.def("CompactRange", &DBWrap::CompactRange);
+
+  //leveldb/options.h
+  bp::class_<leveldb::Options>("Options", bp::init<>())
+	  //.def_readonly("comparator", &leveldb::Options::comparator) //Pointer, maybe needs better wrapper? Untested
+	  .def_readwrite("create_if_missing", &leveldb::Options::create_if_missing)
+	  .def_readwrite("error_if_exists", &leveldb::Options::error_if_exists)
+	  //.def_readwrite("env", &leveldb::Options::env) //Pointer, maybe needs better wrapper? Untested
+	  //.def_readwrite("info_log", &leveldb::Options::info_log) //Pointer, maybe needs better wrapper? Untested
+	  .def_readwrite("write_buffer_size", &leveldb::Options::write_buffer_size)
+	  .def_readwrite("max_open_files", &leveldb::Options::max_open_files)
+	  //.def_readwrite("block_cache", &leveldb::Options::block_cache) //Pointer, maybe needs better wrapper? Untested
+	  .def_readwrite("block_size", &leveldb::Options::block_size)
+	  .def_readwrite("block_restart_interval", &leveldb::Options::block_restart_interval)
+	  //TODO setup a way to include compressor array //Pointer, maybe needs better wrapper? Untested
+	  //.def_readonly("filter_policy", &leveldb::Options::filter_policy); //Pointer, maybe needs better wrapper? Untested
+	;
+  
+  bp::class_<leveldb::ReadOptions>("ReadOptions", bp::init<>())
+	  .def_readwrite("verify_checksums", &leveldb::ReadOptions::verify_checksums)
+	  .def_readwrite("fill_cache", &leveldb::ReadOptions::fill_cache)
+	  .add_property("snapshot",
+		bp::make_getter(&leveldb::ReadOptions::snapshot),
+		bp::make_setter(&leveldb::ReadOptions::snapshot))
+	;
+
+  bp::class_<leveldb::WriteOptions>("WriteOptions", bp::init<>())
+	  .def_readwrite("sync", &leveldb::WriteOptions::sync)
+	;
+
+  //leveldb/iterator.h
+  bp::class_<IteratorWrapper, boost::noncopyable>("Iterator", bp::no_init)
+	  .def("Valid", &IteratorWrapper::Valid)
+	  .def("SeekToFirst", &IteratorWrapper::SeekToFirst)
+	  .def("SeekToLast", &IteratorWrapper::SeekToLast)
+	  .def("Seek", &IteratorWrapper::Seek)
+	  .def("Next", &IteratorWrapper::Next)
+	  .def("Prev", &IteratorWrapper::Prev)
+	  .def("key", &IteratorWrapper::key)
+	  .def("value", &IteratorWrapper::value)
+	  .def("status", &IteratorWrapper::status)
+	  ;
+
+  //leveldb/write_batch.h
+  bp::class_<WriteBatchWrapper>("WriteBatch", bp::init<>())
+	  .def("Put", &WriteBatchWrapper::Put)
+	  .def("Delete", &WriteBatchWrapper::Delete)
+	  ;
+
+  bp::class_<leveldb::Snapshot, boost::noncopyable>("Snapshot", bp::no_init);
+
+  bp::class_<RepairWrapper>("RepairWrapper", bp::init<PyObject*, PyObject*>());
+}
diff --git a/setup.py b/setup.py
new file mode 100644
index 000000000000..f13b2089379c
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,30 @@
+from setuptools import setup
+from distutils.extension import Extension
+
+include_dirs = ['/usr/include', '.']
+library_dirs = ['/usr/lib', '.']
+libraries = ['boost_python', 'leveldb']
+define_macros = [("LINUX", None), ("_LINUX", None), ("LEVELDB_PLATFORM_POSIX", None),
+                    ('OS_LINUX', None)]
+extra_compile_args = ['-std=c++11', '-fPIC', "-DDLLX="]
+runtime_library_dirs = ['/usr/lib']
+extra_link_args = ['-v']
+files = ["leveldb_mcpe.cpp"]
+
+setup(name="leveldb_python_wrapper",
+      packages = ['leveldb_mcpe'],
+      package_dir = {'leveldb_mcpe': ''},
+      ext_modules=[
+          Extension(
+              "leveldb_mcpe",
+              files,
+              library_dirs=library_dirs,
+              libraries=libraries,
+              include_dirs=include_dirs,
+              depends=[],
+              define_macros=define_macros,
+              extra_compile_args=extra_compile_args,
+              extra_link_args=extra_link_args,
+              runtime_library_dirs=runtime_library_dirs)
+      ]
+      )