Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

hide private info && complete vm types && auto downloading #2

Closed
wants to merge 3 commits into from
Closed

hide private info && complete vm types && auto downloading #2

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
7c1555a
feat(nan): hide private info after import wif
Dec 30, 2021
e41f622
feat(neo-cli): auto downloading dependency file
Dec 30, 2021
cfb8fdc
feat(nan): complete the neovm types
Dec 30, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
165 changes: 110 additions & 55 deletions src/nan/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import base64
from json import loads
from json import dumps
from getpass import getpass
import json
from telnetlib import Telnet
from subprocess import PIPE
from subprocess import DEVNULL
Expand Down Expand Up @@ -38,61 +40,42 @@ class Any:
def __init__(self, val) -> None:
if val is None:
self.VAL = None
return
if isinstance(val, NEOVM.Any):
elif isinstance(val, NEOVM.Any):
self.VAL = val.VAL
return
if isinstance(val, bool):
elif isinstance(val, bool):
self.VAL = NEOVM.Boolean(val)
return
if isinstance(val, NEOVM.Boolean):
elif isinstance(val, NEOVM.Boolean):
self.VAL = val
return
if isinstance(val, int):
elif isinstance(val, int):
self.VAL = NEOVM.Integer(val)
return
if isinstance(val, NEOVM.Integer):
elif isinstance(val, NEOVM.Integer):
self.VAL = val
return
if isinstance(val, bytes):
elif isinstance(val, bytes):
self.VAL = NEOVM.ByteArray(val)
return
if isinstance(val, NEOVM.ByteArray):
elif isinstance(val, NEOVM.ByteArray):
self.VAL = val
return
if isinstance(val, str):
elif isinstance(val, str):
self.VAL = NEOVM.String(val)
return
if isinstance(val, NEOVM.String):
elif isinstance(val, NEOVM.String):
self.VAL = val
return
if isinstance(val, NEOVM.Hash160):
elif isinstance(val, NEOVM.Hash160):
self.VAL = val
return
if isinstance(val, NEOVM.Hash256):
elif isinstance(val, NEOVM.Hash256):
self.VAL = val
return
if isinstance(val, NEOVM.PublicKey):
elif isinstance(val, NEOVM.PublicKey):
self.VAL = val
return
if isinstance(val, NEOVM.Signature):
elif isinstance(val, NEOVM.Signature):
self.VAL = val
return
if isinstance(val, list):
elif isinstance(val, list):
self.VAL = NEOVM.Array(val)
return
if isinstance(val, NEOVM.Array):
elif isinstance(val, NEOVM.Array):
self.VAL = val
return
if isinstance(val, dict):
elif isinstance(val, dict):
self.VAL = NEOVM.Map(val)
return
if isinstance(val, NEOVM.Map):
elif isinstance(val, NEOVM.Map):
self.VAL = val
return
if isinstance(val, NEOVM.InteropInterface):
elif isinstance(val, NEOVM.InteropInterface):
self.VAL = val
return
raise Exception()

def __str__(self) -> str:
Expand All @@ -101,22 +84,29 @@ def __str__(self) -> str:
return str(self.VAL)

class Boolean:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Boolean):
self.VAL = val.VAL
elif isinstance(val, bool):
self.VAL = val
elif isinstance(val, str):
self.VAL = val.lower() in ['true', '1', 't', 'y', 'yes', 'yeah', 'yup', 'certainly', 'uh-huh']
else:
raise Exception()

def __str__(self) -> str:
return str(self.VAL)

class Integer:
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Integer):
self.VAL = val.VAL
return
if isinstance(val, int):
elif isinstance(val, int):
self.VAL = val
return
if isinstance(val, float):
elif isinstance(val, float):
self.VAL = int(val)
return
if isinstance(val, str):
elif isinstance(val, str):
self.VAL = int(val)
return
raise Exception()

def __repr__(self) -> str:
Expand All @@ -129,20 +119,36 @@ class ByteArray:
def __init__(self, val) -> None:
if isinstance(val, NEOVM.ByteArray):
self.VAL = val.VAL
if isinstance(val, bytes):
elif isinstance(val, bytes):
self.VAL = val
else:
raise Exception()

def __str__(self) -> str:
return dumps({'type': 'ByteArray', 'value': base64.b64encode(self.VAL)})

class String:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.String):
self.VAL = val.VAL
elif isinstance(val, str):
self.VAL = val
else:
raise Exception()

def __str__(self) -> str:
return dumps({'type': 'String', 'value': self.VAL})

class Hash160:
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Hash160):
self.VAL = val.VAL
if isinstance(val, str) and len(val) == 42 and val.startswith('0x'):
elif isinstance(val, str) and len(val) == 42 and val.startswith('0x'):
self.VAL = val
if isinstance(val, bytes) and len(val) == 42 and val.startswith(b'0x'):
elif isinstance(val, bytes) and len(val) == 42 and val.startswith(b'0x'):
self.VAL = val.decode()
else:
raise Exception()

def __repr__(self) -> str:
return repr(self.VAL)
Expand All @@ -151,19 +157,69 @@ def __str__(self) -> str:
return dumps({'type': 'Hash160', 'value': self.VAL})

class Hash256:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Hash256):
self.VAL = val.VAL
elif isinstance(val, str) and len(val) == 66 and val.startswith('0x'):
self.VAL = val
elif isinstance(val, bytes) and len(val) == 66 and val.startswith(b'0x'):
self.VAL = val.decode()
else:
raise Exception()

def __repr__(self) -> str:
return repr(self.VAL)

def __str__(self) -> str:
return dumps({'type': 'Hash256', 'value': self.VAL})

class PublicKey:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.PublicKey):
self.VAL = val.VAL
elif isinstance(val, str) and len(val) == 66:
self.VAL = val
else:
raise Exception()

def __str__(self) -> str:
return dumps({'type': 'PublicKey', 'value': self.VAL})

class Signature:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Signature):
self.VAL = val.VAL
elif isinstance(val, bytes):
self.VAL = val
else:
raise Exception()

def __str__(self) -> str:
return dumps({'type': 'Signature', 'value': base64.b64encode(self.VAL)})

class Array:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Array):
self.VAL = val.VAL
elif isinstance(val, list):
self.VAL = val
else:
raise Exception()

def __str__(self) -> str:
return dumps({'type': 'Array', 'value': str(self.VAL)})

class Map:
pass
def __init__(self, val) -> None:
if isinstance(val, NEOVM.Map):
self.VAL = val.VAL
elif isinstance(val, dict):
self.VAL = val
else:
raise Exception()

def __str__(self) -> str:
return dumps({'type': 'Map', 'value': str([{'key':k, 'value':v} for k, v in self.VAL.items() ])})

class InteropInterface:
pass
Expand Down Expand Up @@ -251,7 +307,6 @@ def __init__(self) -> None:

def AddWifByNep6(self, filename: str, name: str = None) -> None:
wif = self.GetWifByNEP6(filename)
print(wif)
address = self.GetAddressByNEP6(filename)
name = name or address
setattr(nan, name, Wif(wif))
Expand Down
28 changes: 28 additions & 0 deletions src/nan/__main__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,33 @@
from . import cmd
from . import nan


import os
if not os.path.exists(os.path.expanduser("~/.nan/neo-cli")):
if __import__('platform').machine() != "x86_64":
print("Your machine is x86 which is not supported!")
exit(1)
url = ""
if __import__('platform').system() == "Linux":
url = "https://github.com/lazynode/NeoArchiveNode/archive/refs/tags/neo-cli-linux-x64-3.1.0.tar.gz"
dir_name = "NeoArchiveNode-neo-cli-linux-x64-3.1.0"
else:
print("Your system is not supported!")
exit(1)
print("Downloading neo-cli, pelease wait a minute......")
import urllib.request
import tarfile
import shutil
file_name, _ = urllib.request.urlretrieve(url)
print("Decompressing {}, pelease wait a minute......".format(file_name))
with tarfile.open(file_name) as file:
file.extractall(file_name+"_extracted")
shutil.move(file_name+"_extracted/"+dir_name, os.path.expanduser("~/.nan/neo-cli"))
print("neo-cli preparation compelete, enjoy your nan")
del urllib.request, tarfile, shutil, url, file_name
del os


__import__('rlcompleter')

try:
Expand Down Expand Up @@ -42,6 +69,7 @@
),
)


__import__('readline').parse_and_bind("tab: complete")


Expand Down