From 6fbf60d84cbdf349023c53a98d029b875b2b2739 Mon Sep 17 00:00:00 2001
From: Marcin Jachymiak <marcin.jachymiak1@gmail.com>
Date: Tue, 3 Mar 2020 15:49:09 -0500
Subject: [PATCH] Start health check script

---
 setup-scripts/health-checker.py | 107 ++++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100755 setup-scripts/health-checker.py

diff --git a/setup-scripts/health-checker.py b/setup-scripts/health-checker.py
new file mode 100755
index 00000000..54a49edf
--- /dev/null
+++ b/setup-scripts/health-checker.py
@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+
+from urllib.request import urlopen, Request
+import urllib
+import json
+import sys
+
+
+# sc_precision is the number of hastings per siacoin
+sc_precision = 10 ** 24
+
+api_endpoint = "http://localhost:9980" # TODO get port from env/param
+
+
+# TODO: get values from a param
+min_balance = 10_000 * sc_precision
+min_unspent_renter_funds = 5_000 * sc_precision
+default_renter_funds = 20_000 * sc_precision
+
+#siac class provides wrappers for the necessary siac commands.
+class siac:
+    # initializes values for using the API (password and
+    # user-agent) so that all calls to urllib.request.urlopen have these set.
+    @staticmethod
+    def initialize():
+        # Setup a handler with the API password
+        username = ""
+        password_mgr = urllib.request.HTTPPasswordMgrWithDefaultRealm()
+        password_mgr.add_password(None, api_endpoint, username, siac.get_password())
+        handler = urllib.request.HTTPBasicAuthHandler(password_mgr)
+
+        # Setup an opener with the correct user agent
+        opener = urllib.request.build_opener(handler)
+        opener.addheaders = [('User-agent', 'Sia-Agent')]
+
+        # Install the opener.
+        # Now all calls to urllib.request.urlopen use our opener.
+        urllib.request.install_opener(opener)
+
+    @staticmethod
+    def get_password():
+        password_file = open("/home/marcin/.sia/apipassword")
+        password = password_file.readlines()
+        return password[0].strip()
+
+    # load_json reads the http response and decodes the JSON value
+    @staticmethod
+    def load_json(resp):
+        return json.loads(resp.decode("utf-8"))
+
+    @staticmethod
+    def get_wallet():
+        resp = urllib.request.urlopen(api_endpoint + "/wallet").read()
+        return siac.load_json(resp)
+
+    @staticmethod
+    def get_renter():
+        resp = urllib.request.urlopen(api_endpoint + "/renter").read()
+        return siac.load_json(resp)
+
+# Initialize the API opener
+siac.initialize()
+
+# fail should ping discord for failures in health checks that can't be managed automatically.
+def fail(err_msg):
+    print("Error message: ", err_msg)
+
+
+# check_wallet_health_check checks that the wallet is unlocked and has at least min_balance SC.
+def check_wallet_health():
+    wallet_get = siac.get_wallet()
+
+    if not wallet_get['unlocked']:
+        fail("Wallet not unlocked")
+
+    confirmed_coins = int(wallet_get['confirmedsiacoinbalance'])
+    unconfirmed_coins = int(wallet_get['unconfirmedincomingsiacoins'])
+    unconfirmed_outgoing_coins = int(wallet_get['unconfirmedoutgoingsiacoins'])
+    balance = confirmed_coins + unconfirmed_coins - unconfirmed_outgoing_coins
+
+    print("Balance: ", balance / sc_precision)
+    print("Min balance: ", min_balance / sc_precision)
+
+    if balance < min_balance:
+        fail("Wallet balance too low: {}".format(balance))
+
+def check_renter_health():
+    renter_get = siac.get_renter()
+
+    allowance = renter_get['settings']['allowance']
+    allowance_funds = int(allowance['funds'])
+    unspent_funds = int(renter_get['financialmetrics']['unspent'])
+
+    # if funds are under a certain amount, re-balance the wallet.
+    if unspent_funds < min_unspent_renter_funds:
+        #siac.set_renter_funds(default_renter_funds)
+        #TODO
+
+check_renter_health()
+
+try:
+    check_wallet_health()
+    check_renter_health()
+
+except: # catch all exceptions
+    e = sys.exc_info()[0]
+    fail(e)