197 lines
6.8 KiB
Python
Executable File
197 lines
6.8 KiB
Python
Executable File
#!/usr/local/bin/python3
|
|
import os
|
|
import requests
|
|
import time
|
|
import subprocess
|
|
import json
|
|
import logging
|
|
from inotify_simple import INotify, flags
|
|
from systemd import journal
|
|
from utilities import *
|
|
from device_info import *
|
|
from drive_info import *
|
|
from application_info import *
|
|
from bitcoin_info import *
|
|
from lightning_info import *
|
|
|
|
|
|
PREMIUM_PLUS_CONNECT_URL = "https://www.mynodebtc.com/device_api/premium_plus_connect.php"
|
|
|
|
log = logging.getLogger('premium_plus_connect')
|
|
log.addHandler(journal.JournaldLogHandler())
|
|
log.setLevel(logging.INFO)
|
|
set_logger(log)
|
|
|
|
# Helper functions
|
|
def get_premium_plus_device_info():
|
|
info = {}
|
|
settings = get_premium_plus_settings()
|
|
|
|
# Basic Info
|
|
info["serial"] = get_device_serial()
|
|
info["device_type"] = get_device_type()
|
|
info["device_arch"] = get_device_arch()
|
|
info["drive_size"] = get_mynode_drive_size()
|
|
info["data_drive_usage"] = get_data_drive_usage()
|
|
info["os_drive_usage"] = get_os_drive_usage()
|
|
info["temperature"] = get_device_temp()
|
|
info["total_ram"] = get_device_ram()
|
|
info["ram_usage"] = get_ram_usage()
|
|
info["swap_usage"] = get_swap_usage()
|
|
info["uptime"] = get_system_uptime()
|
|
|
|
# App status info
|
|
if settings['sync_status']:
|
|
info["app_info"] = get_all_applications_from_json_cache()
|
|
|
|
return info
|
|
|
|
def get_premium_plus_bitcoin_info():
|
|
info = {}
|
|
settings = get_premium_plus_settings()
|
|
|
|
if settings['sync_bitcoin_and_lightning']:
|
|
#log_message("BITCOIN " + str(get_bitcoin_json_cache()))
|
|
info = get_bitcoin_json_cache()
|
|
return info
|
|
|
|
def get_premium_plus_lightning_info():
|
|
info = {}
|
|
settings = get_premium_plus_settings()
|
|
|
|
if settings['sync_bitcoin_and_lightning']:
|
|
#log_message("LIGHTNING " + str(get_lightning_json_cache()))
|
|
info = get_lightning_json_cache()
|
|
return info
|
|
|
|
|
|
def on_connect_success(connect_response_data):
|
|
settings = get_premium_plus_settings()
|
|
|
|
try:
|
|
# Check for lnd watchtower, setup this device to use Premium+ Watchtower
|
|
try:
|
|
if "watchtower_uri" in connect_response_data:
|
|
w = connect_response_data["watchtower_uri"]
|
|
parts = w.split("@")
|
|
pubkey = parts[0]
|
|
output = run_lncli_command("lncli wtclient towers")
|
|
info = json.loads(output)
|
|
towers = info["towers"]
|
|
|
|
found_watchtower = False
|
|
for t in towers:
|
|
#log_message("EXISTING TOWER: {} active={}".format(t["pubkey"], t["active_session_candidate"]))
|
|
if t["pubkey"] == pubkey and t["active_session_candidate"] == True:
|
|
if settings["watchtower"]:
|
|
log_message("Found Premium+ Tower")
|
|
else:
|
|
log_message("Removing Premium+ Tower {}".format(pubkey))
|
|
run_lncli_command("lncli wtclient remove {}".format(pubkey))
|
|
found_watchtower = True
|
|
if not found_watchtower:
|
|
log_message("Adding Premium+ Tower {}".format(w))
|
|
run_lncli_command("lncli wtclient add {}".format(w))
|
|
|
|
except Exception as e:
|
|
log_message("on_connect_success exception: {}".format(str(e)))
|
|
except Exception as e:
|
|
log_message("on_connect_success exception: {}".format(str(e)))
|
|
return
|
|
|
|
def clear_response_data():
|
|
os.system("rm -f /tmp/premium_plus_response.json")
|
|
def save_response_data(data):
|
|
try:
|
|
with open("/tmp/premium_plus_response.json", "w") as file:
|
|
json.dump(data, file, indent=4, sort_keys=True)
|
|
except Exception as e:
|
|
log_message("save_response_data exception: failed to save response - {}".format(str(e)))
|
|
|
|
# Update hourly
|
|
def premium_plus_connect():
|
|
|
|
# Check in
|
|
data = {
|
|
"serial": get_device_serial(),
|
|
"version": get_current_version(),
|
|
"token": get_premium_plus_token(),
|
|
"settings": json.dumps(get_premium_plus_settings()),
|
|
"device_info": json.dumps(get_premium_plus_device_info()),
|
|
"bitcoin_info": json.dumps(get_premium_plus_bitcoin_info()),
|
|
"lightning_info": json.dumps(get_premium_plus_lightning_info()),
|
|
"product_key": get_product_key(),
|
|
}
|
|
|
|
response = make_tor_request(PREMIUM_PLUS_CONNECT_URL, data)
|
|
update_premium_plus_last_sync_time()
|
|
if response == None:
|
|
clear_response_data()
|
|
set_premium_plus_token_status("CONNECTION_ERROR")
|
|
log_message("Premium+ Connect Error: Connection Failed")
|
|
return False
|
|
|
|
if response.status_code != 200:
|
|
clear_response_data()
|
|
set_premium_plus_token_status("CONNECTION_ERROR")
|
|
log_message("Premium+ Connect Error: Status Code {}".format(response.status_code))
|
|
return False
|
|
|
|
try:
|
|
info = json.loads(response.text)
|
|
except Exception as e:
|
|
log_message("Premium+ Connect Error: Error Parsing JSON - {}".format(str(e)))
|
|
return False
|
|
|
|
save_response_data(info)
|
|
|
|
if "error" in info:
|
|
set_premium_plus_token_status(info["error"])
|
|
return False
|
|
elif "status" in info:
|
|
set_premium_plus_token_status(info["status"])
|
|
if info["status"] == "OK":
|
|
log_message("Premium+ Connect Success!")
|
|
log_message(response.text)
|
|
on_connect_success(info)
|
|
else:
|
|
log_message("Premium+ Connect Error: Status ({})".format(info["status"]))
|
|
else:
|
|
log_message("Premium+ Connect Error: Missing Info {}".format(response.text))
|
|
return False
|
|
|
|
return True
|
|
|
|
# Run premium plus update consistently
|
|
if __name__ == "__main__":
|
|
update_in_min = 15 # Update every 15 minutes
|
|
update_ms = update_in_min * 60 * 1000
|
|
|
|
while True:
|
|
try:
|
|
# Wait for drive to be mounted
|
|
while not is_mynode_drive_mounted():
|
|
log_message("Checking if drive mounted...")
|
|
time.sleep(10)
|
|
log_message("Drive mounted!")
|
|
|
|
# Wait on token
|
|
log_message("Looking for Premium+ Token...")
|
|
while not has_premium_plus_token():
|
|
time.sleep(10)
|
|
log_message("Token found!")
|
|
premium_plus_connect()
|
|
|
|
# Watch for updates
|
|
log_message("")
|
|
log_message("Watching for file changes or 1hr...")
|
|
inotify = INotify()
|
|
watch_flags = flags.CREATE | flags.DELETE | flags.MODIFY | flags.DELETE_SELF
|
|
wd = inotify.add_watch('/home/bitcoin/.mynode/', watch_flags)
|
|
for event in inotify.read(timeout=update_ms):
|
|
log_message("File changed: " + str(event))
|
|
log_message("Running connect again")
|
|
except Exception as e:
|
|
log_message("Error: {}".format(e))
|
|
time.sleep(60)
|