From 152c235d87fcc2cc5b4a0ebb08213d845cc6ebcf Mon Sep 17 00:00:00 2001
From: Christopher Jeffrey <chjjeffrey@gmail.com>
Date: Tue, 28 Oct 2014 14:09:55 -0700
Subject: [PATCH] add get and set generate.

---
 lib/bitcoind.js   |  8 +++++
 src/bitcoindjs.cc | 83 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 91 insertions(+)

diff --git a/lib/bitcoind.js b/lib/bitcoind.js
index 16beb73c..7a41a03f 100644
--- a/lib/bitcoind.js
+++ b/lib/bitcoind.js
@@ -357,6 +357,14 @@ Bitcoin.prototype.getProgress = function(callback) {
   return bitcoindjs.getProgress(callback);
 };
 
+Bitcoin.prototype.setGenerate = function(options) {
+  return bitcoindjs.setGenerate(options || {});
+};
+
+Bitcoin.prototype.getGenerate = function(options) {
+  return bitcoindjs.getGenerate(options || {});
+};
+
 Bitcoin.prototype.log =
 Bitcoin.prototype.info = function() {
   if (typeof arguments[0] !== 'string') {
diff --git a/src/bitcoindjs.cc b/src/bitcoindjs.cc
index f5c7e6c0..aee937b6 100644
--- a/src/bitcoindjs.cc
+++ b/src/bitcoindjs.cc
@@ -168,6 +168,9 @@ NAN_METHOD(GetInfo);
 NAN_METHOD(GetPeerInfo);
 NAN_METHOD(GetAddresses);
 NAN_METHOD(GetProgress);
+NAN_METHOD(SetGenerate);
+NAN_METHOD(GetGenerate);
+
 NAN_METHOD(GetBlockHex);
 NAN_METHOD(GetTxHex);
 NAN_METHOD(BlockFromHex);
@@ -1562,6 +1565,84 @@ async_get_progress_after(uv_work_t *req) {
   delete req;
 }
 
+/**
+ * SetGenerate()
+ * bitcoindjs.setGenerate(options)
+ * Set coin generation / mining
+ */
+
+NAN_METHOD(SetGenerate) {
+  NanScope();
+
+  if (args.Length() < 1 || !args[0]->IsObject()) {
+    return NanThrowError(
+      "Usage: bitcoindjs.setGenerate(options)");
+  }
+
+  Local<Object> options = Local<Object>::Cast(args[0]);
+
+  if (pwalletMain == NULL) {
+    return NanThrowError("Method not found (disabled)");
+  }
+
+  bool fGenerate = true;
+  if (options->Get(NanNew<String>("generate"))->IsBoolean()) {
+    fGenerate = options->Get(NanNew<String>("generate"))->ToBoolean()->IsTrue();
+  }
+
+  int nGenProcLimit = -1;
+  if (options->Get(NanNew<String>("limit"))->IsNumber()) {
+    nGenProcLimit = options->Get(NanNew<String>("limit"))->ToInt32();
+    if (nGenProcLimit == 0) {
+      fGenerate = false;
+    }
+  }
+
+  // -regtest mode: don't return until nGenProcLimit blocks are generated
+  if (fGenerate && Params().MineBlocksOnDemand()) {
+    int nHeightStart = 0;
+    int nHeightEnd = 0;
+    int nHeight = 0;
+    int nGenerate = (nGenProcLimit > 0 ? nGenProcLimit : 1);
+    { // Don't keep cs_main locked
+      LOCK(cs_main);
+      nHeightStart = chainActive.Height();
+      nHeight = nHeightStart;
+      nHeightEnd = nHeightStart+nGenerate;
+    }
+    int nHeightLast = -1;
+    while (nHeight < nHeightEnd) {
+      if (nHeightLast != nHeight) {
+        nHeightLast = nHeight;
+        GenerateBitcoins(fGenerate, pwalletMain, 1);
+      }
+      MilliSleep(1);
+      {   // Don't keep cs_main locked
+        LOCK(cs_main);
+        nHeight = chainActive.Height();
+      }
+    }
+  } else { // Not -regtest: start generate thread, return immediately
+    mapArgs["-gen"] = (fGenerate ? "1" : "0");
+    mapArgs ["-genproclimit"] = itostr(nGenProcLimit);
+    GenerateBitcoins(fGenerate, pwalletMain, nGenProcLimit);
+  }
+
+  NanReturnValue(True());
+}
+
+/**
+ * GetGenerate()
+ * bitcoindjs.GetGenerate()
+ * Get coin generation / mining
+ */
+
+NAN_METHOD(GetGenerate) {
+  NanScope();
+  bool generate = GetBoolArg("-gen", false);
+  NanReturnValue(NanNew<Boolean>(generate));
+}
+
 /**
  * GetBlockHex()
  * bitcoindjs.getBlockHex(callback)
@@ -4279,6 +4360,8 @@ init(Handle<Object> target) {
   NODE_SET_METHOD(target, "getPeerInfo", GetPeerInfo);
   NODE_SET_METHOD(target, "getAddresses", GetAddresses);
   NODE_SET_METHOD(target, "getProgress", GetProgress);
+  NODE_SET_METHOD(target, "setGenerate", SetGenerate);
+  NODE_SET_METHOD(target, "getGenerate", GetGenerate);
   NODE_SET_METHOD(target, "getBlockHex", GetBlockHex);
   NODE_SET_METHOD(target, "getTxHex", GetTxHex);
   NODE_SET_METHOD(target, "blockFromHex", BlockFromHex);