mirror of
https://github.com/StijnBTC/Ringtools.git
synced 2025-02-28 15:55:27 +01:00
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
|
import os
|
||
|
|
import grpc
|
||
|
|
from time import sleep
|
||
|
|
|
||
|
|
from output import format_channel_error, format_error, clear_screen, format_channel
|
||
|
|
from yachalk import chalk
|
||
|
|
|
||
|
|
LOOP_SLEEP_TIME = 10
|
||
|
|
|
||
|
|
|
||
|
|
class CheckRing:
|
||
|
|
def __init__(self, lnd, output, pubkeys_file, show_fees):
|
||
|
|
self.lnd = lnd
|
||
|
|
self.output = output
|
||
|
|
self.pubkeys_file = pubkeys_file
|
||
|
|
self.show_fees = show_fees
|
||
|
|
|
||
|
|
def read_file(self, file):
|
||
|
|
if not os.path.isfile(file):
|
||
|
|
self.handle_error("File does not exist")
|
||
|
|
else:
|
||
|
|
with open(self.pubkeys_file) as file:
|
||
|
|
return file.read().splitlines()
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
self.once()
|
||
|
|
|
||
|
|
|
||
|
|
def once(self):
|
||
|
|
pubkeys = self.read_file(self.pubkeys_file)
|
||
|
|
# response = self.lnd.get_node(pubkeys[0].split(',')[0])
|
||
|
|
# print(response)
|
||
|
|
for idx, pubkeyInfo in enumerate(pubkeys):
|
||
|
|
pubkey = pubkeyInfo.split(',')
|
||
|
|
try:
|
||
|
|
#print(pubkey[0], idx)
|
||
|
|
response = self.lnd.get_node_channels(pubkey[0])
|
||
|
|
|
||
|
|
print("%s" %
|
||
|
|
(chalk.yellow(response.node.alias)))
|
||
|
|
|
||
|
|
# print("Number of channels: %s" % len(response.channels))
|
||
|
|
channelTo = pubkeys[(idx+1) % (len(pubkeys))].split(',')[0]
|
||
|
|
hasChannel = False
|
||
|
|
channelId = 0
|
||
|
|
channelInfo = {}
|
||
|
|
|
||
|
|
for channel in response.channels:
|
||
|
|
if (channel.node1_pub == channelTo) or (channel.node2_pub == channelTo):
|
||
|
|
hasChannel = True
|
||
|
|
channelId = channel.channel_id
|
||
|
|
channelInfo = channel
|
||
|
|
|
||
|
|
if hasChannel:
|
||
|
|
outputHas = chalk.green('✅')
|
||
|
|
else:
|
||
|
|
outputHas = chalk.red('🙌🏻')
|
||
|
|
print("%s %s\r\n%s" %
|
||
|
|
(outputHas, pubkey[1], pubkey[0]))
|
||
|
|
|
||
|
|
if hasChannel:
|
||
|
|
print(chalk.green("Channel is open with ID: %s") % channelId)
|
||
|
|
|
||
|
|
if self.show_fees:
|
||
|
|
response = self.lnd.get_edge(int(channelId))
|
||
|
|
node1 = self.lnd.get_node(response.node1_pub)
|
||
|
|
node2 = self.lnd.get_node(response.node2_pub)
|
||
|
|
disabled = response.node1_policy.disabled or response.node2_policy.disabled
|
||
|
|
self.print_channel(
|
||
|
|
channelInfo, node1.alias, node2.alias, disabled)
|
||
|
|
else:
|
||
|
|
print(chalk.red("Should open to node with pubkey %s") % channelTo)
|
||
|
|
except grpc.RpcError as e:
|
||
|
|
self.output.print_line(format_channel_error(pubkey, repr(e)))
|
||
|
|
except Exception as error:
|
||
|
|
self.output.print_line(
|
||
|
|
format_channel_error(pubkey, repr(error)))
|
||
|
|
|
||
|
|
def print_channel(self, channel, node1_alias, node2_alias, chan_disabled):
|
||
|
|
self.output.print_line(format_channel(
|
||
|
|
channel, node1_alias, node2_alias, chan_disabled, self.show_fees))
|
||
|
|
|
||
|
|
def handle_error(self, error):
|
||
|
|
self.output.print_line(format_error(error))
|