From 808281aff8887ce4af61c875e1195de7600af625 Mon Sep 17 00:00:00 2001
From: Command M <maoxs2@163.com>
Date: Wed, 29 Jan 2020 08:51:58 +0800
Subject: [PATCH] add wallet v2 support

---
 .idea/modules.xml                             |  2 +-
 .../{SimplistPool.iml => open-grin-pool.iml}  |  0
 .idea/workspace.xml                           | 47 +++++++++---
 config.go                                     |  9 +--
 config.json                                   |  1 +
 payer.go                                      | 72 ++++++++++++++++++-
 6 files changed, 116 insertions(+), 15 deletions(-)
 rename .idea/{SimplistPool.iml => open-grin-pool.iml} (100%)

diff --git a/.idea/modules.xml b/.idea/modules.xml
index 1739e06..f4106b0 100644
--- a/.idea/modules.xml
+++ b/.idea/modules.xml
@@ -2,7 +2,7 @@
 <project version="4">
   <component name="ProjectModuleManager">
     <modules>
-      <module fileurl="file://$PROJECT_DIR$/.idea/SimplistPool.iml" filepath="$PROJECT_DIR$/.idea/SimplistPool.iml" />
+      <module fileurl="file://$PROJECT_DIR$/.idea/open-grin-pool.iml" filepath="$PROJECT_DIR$/.idea/open-grin-pool.iml" />
     </modules>
   </component>
 </project>
\ No newline at end of file
diff --git a/.idea/SimplistPool.iml b/.idea/open-grin-pool.iml
similarity index 100%
rename from .idea/SimplistPool.iml
rename to .idea/open-grin-pool.iml
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 762485e..ed56273 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -2,7 +2,13 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="9bdc0910-8d5e-4927-884b-19805f6d7390" name="Default Changelist" comment="">
+      <change beforePath="$PROJECT_DIR$/.idea/SimplistPool.iml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/open-grin-pool.iml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/.idea/modules.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/modules.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/config.go" beforeDir="false" afterPath="$PROJECT_DIR$/config.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/config.json" beforeDir="false" afterPath="$PROJECT_DIR$/config.json" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/go.mod" beforeDir="false" afterPath="$PROJECT_DIR$/go.mod" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/payer.go" beforeDir="false" afterPath="$PROJECT_DIR$/payer.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -12,24 +18,49 @@
   <component name="FavoritesManager">
     <favorites_list name="SimplistPool" />
   </component>
+  <component name="GOROOT" path="C:\Go" />
   <component name="Git.Settings">
     <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
   </component>
   <component name="ProjectId" id="1S151aoVIcIDI4QZ3jGGOTkHlFy" />
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showExcludedFiles" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
   <component name="PropertiesComponent">
+    <property name="RunOnceActivity.ShowReadmeOnStart" value="true" />
     <property name="WebServerToolWindowFactoryState" value="false" />
+    <property name="go.import.settings.migrated" value="true" />
     <property name="last_opened_file_path" value="$PROJECT_DIR$" />
+    <property name="settings.editor.selected.configurable" value="go.vgo" />
   </component>
-  <component name="RunDashboard">
-    <option name="ruleStates">
+  <component name="ServiceViewManager">
+    <option name="viewStates">
       <list>
-        <RuleState>
-          <option name="name" value="ConfigurationTypeDashboardGroupingRule" />
-        </RuleState>
-        <RuleState>
-          <option name="name" value="StatusDashboardGroupingRule" />
-        </RuleState>
+        <serviceView>
+          <treeState>
+            <expand />
+            <select />
+          </treeState>
+        </serviceView>
       </list>
     </option>
   </component>
+  <component name="TypeScriptGeneratedFilesManager">
+    <option name="version" value="1" />
+  </component>
+  <component name="VgoProject">
+    <integration-enabled>true</integration-enabled>
+  </component>
+  <component name="WindowStateProjectService">
+    <state x="552" y="179" key="#GOROOT" timestamp="1580212984045">
+      <screen x="0" y="0" width="1920" height="1040" />
+    </state>
+    <state x="552" y="179" key="#GOROOT/0.0.1920.1040@0.0.1920.1040" timestamp="1580212984045" />
+    <state x="461" y="162" key="SettingsEditor" timestamp="1580213015501">
+      <screen x="0" y="0" width="1920" height="1040" />
+    </state>
+    <state x="461" y="162" key="SettingsEditor/0.0.1920.1040@0.0.1920.1040" timestamp="1580213015501" />
+  </component>
 </project>
\ No newline at end of file
diff --git a/config.go b/config.go
index 3af6be9..60e131f 100644
--- a/config.go
+++ b/config.go
@@ -40,10 +40,11 @@ type config struct {
 		BlockTime   int    `json:"block_time"`
 	} `json:"node"`
 	Wallet struct {
-		Address      string `json:"address"`
-		OwnerAPIPort int    `json:"owner_api_port"`
-		AuthUser     string `json:"auth_user"`
-		AuthPass     string `json:"auth_pass"`
+		Address         string `json:"address"`
+		OwnerAPIVersion string `json:"owner_api_version"`
+		OwnerAPIPort    int    `json:"owner_api_port"`
+		AuthUser        string `json:"auth_user"`
+		AuthPass        string `json:"auth_pass"`
 	} `json:"wallet"`
 	Payer struct {
 		Time string  `json:"time"`
diff --git a/config.json b/config.json
index 8ae9fe0..7ed1343 100644
--- a/config.json
+++ b/config.json
@@ -32,6 +32,7 @@
   },
   "wallet": {
     "address": "127.0.0.1",
+    "owner_api_version": "v1",
     "owner_api_port": 3420,
     "auth_user": "epic",
     "auth_pass": "FoakPxzMy56Z4gJPCHKZ"
diff --git a/payer.go b/payer.go
index a96650a..e55b4df 100644
--- a/payer.go
+++ b/payer.go
@@ -15,7 +15,16 @@ type payer struct {
 	conf *config
 }
 
-func (p *payer) getNewBalance() uint64 {
+type jsonRPCResponse struct {
+	ID      string                 `json:"id"`
+	JsonRpc string                 `json:"jsonrpc"`
+	Method  string                 `json:"method"`
+	Result  interface{}            `json:"result"`
+	Error   map[string]interface{} `json:"error"`
+}
+
+// deprecated v1 wallet owner api
+func (p *payer) getNewBalanceV1() uint64 {
 	req, _ := http.NewRequest("GET", "http://"+p.conf.Wallet.Address+":"+strconv.Itoa(p.conf.Wallet.OwnerAPIPort)+"/v1/wallet/owner/retrieve_summary_info?refresh", nil)
 	req.SetBasicAuth(p.conf.Node.AuthUser, p.conf.Node.AuthPass)
 	client := &http.Client{}
@@ -35,6 +44,55 @@ func (p *payer) getNewBalance() uint64 {
 	return uint64(spendable) // unit nanogrin
 }
 
+// The V2 Owner API (OwnerRpc) will be removed in grin-wallet 4.0.0. Please migrate to the V3 (OwnerRpcS) API as soon as possible.
+func (p *payer) getNewBalanceV2() uint64 {
+	req, _ := http.NewRequest("POST", "http://"+p.conf.Wallet.Address+":"+strconv.Itoa(p.conf.Wallet.OwnerAPIPort)+"/v2/wallet/owner", strings.NewReader(`{
+	"jsonrpc": "2.0",
+	"method": "retrieve_summary_info",
+	"params": [true, 10],
+	"id": 1
+}`))
+	req.SetBasicAuth(p.conf.Node.AuthUser, p.conf.Node.AuthPass)
+	client := &http.Client{}
+	res, err := client.Do(req)
+	if err != nil {
+		logger.Error("failed to get balance from wallet for failing to post request, treat this as no income")
+		return 0
+	}
+
+	dec := json.NewDecoder(res.Body)
+	var summaryInfo jsonRPCResponse
+	_ = dec.Decode(&summaryInfo)
+
+	result := summaryInfo.Result
+	if result == nil {
+		logger.Error(summaryInfo.Error)
+		logger.Error("failed to get balance from wallet for failing to get result request, treat this as no income")
+		return 0
+	}
+
+	mapResult, ok := result.(map[string]interface{})
+	if ok || mapResult["Ok"] != nil {
+		theOk, ok := mapResult["Ok"].([]interface{})
+		if ok {
+			for i := range theOk {
+				if balanceMap, ok := theOk[i].(map[string]interface{}); ok {
+					strSpendable := balanceMap["amount_currently_spendable"].(string)
+					spendable, _ := strconv.Atoi(strSpendable)
+					return uint64(spendable) // unit nanogrin
+				}
+			}
+		}
+	}
+
+	return 0
+}
+
+func (p *payer) getNewBalanceV3() uint64 {
+	logger.Fatal("V3 support WIP!")
+	return 0
+}
+
 // distribute coins when balance is > 1e9 nano
 func (p *payer) distribute(newBalance uint64) {
 	// get a distribution table
@@ -54,6 +112,16 @@ func (p *payer) watch() {
 			logger.Error(err)
 		}
 
+		var getNewBalance func() uint64
+		switch p.conf.Wallet.OwnerAPIVersion {
+		case "v1":
+			getNewBalance = p.getNewBalanceV1
+		case "v2":
+			getNewBalance = p.getNewBalanceV2
+		case "v3":
+			getNewBalance = p.getNewBalanceV3
+		}
+
 		for {
 			now := time.Now()
 			t := time.Date(now.Year(), now.Month(), now.Day(), hour, min, 0, 0, now.Location())
@@ -65,7 +133,7 @@ func (p *payer) watch() {
 
 			select {
 			case <-timer.C:
-				newBalance := p.getNewBalance()
+				newBalance := getNewBalance()
 				if newBalance > 1e9 {
 					p.distribute(newBalance - 1e9)
 				} else {