Subspace/subspace-cli.py at master · abitfan/Subspace · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
__author__ = 'chris'

import argparse
import string
import pickle

from twisted.internet import reactor
from txjsonrpc.netstring.jsonrpc import Proxy
from os.path import expanduser
from bitcoin import *

datafolder = expanduser("~") + "/.subspace/"
if os.path.isfile(datafolder + 'keys.pickle'):
    privkey = pickle.load(open(datafolder + "keys.pickle", "rb"))

def doContinue(value):
    pass

def printValue(value):
    print str(value)
    reactor.stop()

def printError(error):
    print 'error', error
    reactor.stop()

class Parser(object):

    def __init__(self, proxy):
        parser = argparse.ArgumentParser(
            description='Subspace v0.2',
            usage='''
    subspace <command> [<args>]
    subspace <command> --help

commands:
    getmessages      returns a list of your messages in json format
    getprivkey       returns your private encryption key
    getpubkey        returns your node's public encryption key
    send             sends a message to the given public key
    start            start the subspace daemon
    stop             close subspace and disconnect
''')
        parser.add_argument('command', help='Execute the given command')
        parser.add_argument('-n', '--noisy', action='store_true', help="show log output")
        args = parser.parse_args(sys.argv[1:2])
        if not hasattr(self, args.command):
            parser.print_help()
            exit(1)
        getattr(self, args.command)()
        self.proxy = proxy

    def send(self):
        parser = argparse.ArgumentParser(
            description="Send a message to the recipient's public key",
            usage='''usage:
    subspace send [-k PUBLIC KEY] [-m MESSAGE]''')
        parser.add_argument('-k', '--key', required=True, help="recipient's public key")
        parser.add_argument('-m', '--message', required=True,
                            help="the unencrypted message to send (will be encrypted)",
                            nargs='+')
        args = parser.parse_args(sys.argv[2:])
        if len(args.key) != 66 or all(c in string.hexdigits for c in args.key) is not True:
            print "Invalid key. Enter a 33 byte public key in either hexadecimal for base58check format."
            return

        d = proxy.callRemote('send', args.key, args.message)
        d.addCallbacks(printValue, printError)
        reactor.run()

    def getmessages(self):
        parser = argparse.ArgumentParser(
            description='Returns a list of your messages in json format',
            usage='''usage:
    subspace getmessages''')
        args = parser.parse_args(sys.argv[2:])
        d = proxy.callRemote('getmessages')
        d.addCallbacks(printValue, printError)
        reactor.run()

    def getprivkey(self):
        def printKey(key):
            if args.base58:
                print encode_privkey(key, "wif")
            else:
                print key
            reactor.stop()
        parser = argparse.ArgumentParser(
            description="Returns your private encryption key",
            usage='''usage:
    subspace getprivkey''')
        parser.add_argument('-b', '--base58', action='store_true', help="returns the key in base58check format")
        args = parser.parse_args(sys.argv[2:])
        d = proxy.callRemote('getprivkey')
        d.addCallbacks(printKey, printError)
        reactor.run()

    def getpubkey(self):
        def printKey(key):
            if args.base58:
                print hex_to_b58check(key, 0)
            else:
                print key
            reactor.stop()
        parser = argparse.ArgumentParser(
            description="Returns your node's public encryption key",
            usage='''usage:
    subspace getpubkey''')
        parser.add_argument('-b', '--base58', action='store_true', help="returns the key in base58check format")
        args = parser.parse_args(sys.argv[2:])
        d = proxy.callRemote('getpubkey')
        d.addCallbacks(printKey, printError)
        reactor.run()

proxy = Proxy('127.0.0.1', 7080)
Parser(proxy)