faststep 0.0.1

data/ext/faststep/db.c

@@ -0,0 +1,56 @@
+#include "db.h"
+#include "mongo.h"
+#include "bson.h"
+#include "connection.h"
+#include "collection.h"
+#include "bson_ruby_conversion.h"
+#include "faststep_defines.h"
+
+void faststep_db_main() {
+  rb_cFaststepDb = rb_define_class_under(rb_mFaststep, "Db", rb_cObject);
+
+  rb_define_attr(rb_cFaststepDb, "name", 1, 0);
+  rb_define_attr(rb_cFaststepDb, "connection", 1, 0);
+
+  rb_define_alias(rb_cFaststepDb, "[]", "collection");
+
+  rb_define_method(rb_cFaststepDb, "initialize", faststep_db_init, 2);
+  rb_define_method(rb_cFaststepDb, "drop",       faststep_db_drop, 0);
+  rb_define_method(rb_cFaststepDb, "command",    faststep_db_command, 1);
+}
+
+static VALUE faststep_db_init(VALUE self, VALUE name, VALUE connection) {
+  rb_iv_set(self, "@name", name);
+  rb_iv_set(self, "@connection", connection);
+
+  return self;
+}
+
+static VALUE faststep_db_drop(VALUE self) {
+  mongo_connection* conn = GetFaststepConnection(rb_iv_get(self, "@connection"));
+
+  return bool_to_ruby(mongo_cmd_drop_db(conn, RSTRING_PTR(rb_iv_get(self, "@name"))));
+}
+
+static VALUE faststep_db_command(VALUE self, VALUE command) {
+  mongo_connection* conn = GetFaststepConnection(rb_iv_get(self, "@connection"));
+
+  bson* result = bson_malloc(sizeof(bson));
+  bson_init(result, "", 1);
+
+  bson* bson_command = create_bson_from_ruby_hash(command);
+
+  char ns[500] = "";
+  build_collection_ns(ns, RSTRING_PTR(rb_iv_get(self, "@name")), "$cmd");
+
+  mongo_find_one(conn, ns, bson_command, NULL, result);
+
+  bson_destroy(bson_command);
+
+  VALUE hash = ruby_hash_from_bson(result);
+  bson_destroy(result);
+
+  ensure_document_ok(hash);
+
+  return hash;
+}

data/ext/faststep/db.h

@@ -0,0 +1,8 @@
+#ifndef DB_H
+#include <ruby.h>
+#define DB_H
+void  faststep_db_main();
+static VALUE faststep_db_init(VALUE, VALUE, VALUE);
+static VALUE faststep_db_drop(VALUE);
+static VALUE faststep_db_command(VALUE, VALUE);
+#endif

data/ext/faststep/exceptions.c

@@ -0,0 +1,7 @@
+#include "exceptions.h"
+#include "faststep_defines.h"
+
+void faststep_exceptions_main() {
+  rb_cFaststepConnectionFailure = rb_define_class_under(rb_mFaststep, "ConnectionFailure", rb_eStandardError);
+  rb_cFaststepOperationFailure  = rb_define_class_under(rb_mFaststep, "OperationFailure", rb_eStandardError);
+}

data/ext/faststep/exceptions.h

@@ -0,0 +1,5 @@
+#ifndef EXCEPTIONS_H
+#include <ruby.h>
+#define EXCEPTIONS_H
+void faststep_exceptions_main();
+#endif

data/ext/faststep/extconf.rb

@@ -0,0 +1,3 @@
+require "mkmf"
+$CFLAGS = "--std=c99"
+create_makefile("faststep/faststep")

data/ext/faststep/faststep.c

@@ -0,0 +1,30 @@
+#include "faststep.h"
+#include "connection.h"
+#include "db.h"
+#include "collection.h"
+#include "exceptions.h"
+#include "cursor.h"
+#include "support.h"
+
+VALUE rb_mBson;
+VALUE rb_mFaststep;
+VALUE rb_cFaststepConnection;
+VALUE rb_cFaststepDb;
+VALUE rb_cFaststepCollection;
+VALUE rb_cFaststepCursor;
+VALUE rb_mFaststepSupport;
+
+VALUE rb_cFaststepConnectionFailure;
+VALUE rb_cFaststepOperationFailure;
+
+void Init_faststep() {
+  rb_mFaststep = rb_define_module("Faststep");
+  rb_mBson     = rb_const_get(rb_cObject, rb_intern("BSON"));
+
+  faststep_connection_main();
+  faststep_db_main();
+  faststep_collection_main();
+  faststep_cursor_main();
+  faststep_support_main();
+  faststep_exceptions_main();
+}

data/ext/faststep/faststep.h

@@ -0,0 +1,4 @@
+#ifndef FASTSTEP_H
+#include <ruby.h>
+#define FASTSTEP_H
+#endif

data/ext/faststep/faststep_defines.h

@@ -0,0 +1,11 @@
+RUBY_EXTERN VALUE rb_mFaststep;
+RUBY_EXTERN VALUE rb_mFaststepSupport;
+RUBY_EXTERN VALUE rb_cFaststepConnection;
+RUBY_EXTERN VALUE rb_cFaststepDb;
+RUBY_EXTERN VALUE rb_cFaststepCollection;
+RUBY_EXTERN VALUE rb_cFaststepCursor;
+
+RUBY_EXTERN VALUE rb_cFaststepConnectionFailure;
+RUBY_EXTERN VALUE rb_cFaststepOperationFailure;
+
+RUBY_EXTERN VALUE rb_mBson;