fluzzi32
32fe18d27a
- Implement lazy `@property` getters for `configfile.privatekey` and `configfile.publickey` to defer RSA key loading and pycryptodome imports until explicitly accessed. - Move top-level `Crypto` imports inside local methods in `configfile.py`, `core.py`, and `config_service.py`. - Remove top-level `from rich.markdown import Markdown` across CLI handlers (`node_handler.py`, `ai_handler.py`, `terminal_ui.py`) and `ai.py`, deferring parser load to active rendering calls. - Reduce CLI startup load latency from ~172 ms to ~49 ms (>3.5x speedup / ~71% latency reduction). - Regenerate HTML documentation in `docs/` via `pdoc`. - Bump version to v6.1.1.
271 lines
9.7 KiB
Python
271 lines
9.7 KiB
Python
import sys
|
|
import yaml
|
|
|
|
from .. import printer
|
|
from ..services.exceptions import ConnpyError, InvalidConfigurationError
|
|
from .helpers import choose
|
|
from .help_text import get_instructions
|
|
|
|
class NodeHandler:
|
|
def __init__(self, app):
|
|
self.app = app
|
|
self._forms = None
|
|
|
|
@property
|
|
def forms(self):
|
|
if self._forms is None:
|
|
from .forms import Forms
|
|
self._forms = Forms(self.app)
|
|
return self._forms
|
|
|
|
@forms.setter
|
|
def forms(self, value):
|
|
self._forms = value
|
|
|
|
def _filter_exact_match(self, matches, query):
|
|
if not query or len(matches) <= 1:
|
|
return matches
|
|
|
|
exact_matches = []
|
|
for m in matches:
|
|
if self.app.case:
|
|
if m == query:
|
|
exact_matches.append(m)
|
|
else:
|
|
if m.lower() == query.lower():
|
|
exact_matches.append(m)
|
|
|
|
if len(exact_matches) == 1:
|
|
return exact_matches
|
|
return matches
|
|
|
|
def dispatch(self, args):
|
|
if not self.app.case and args.data != None:
|
|
args.data = args.data.lower()
|
|
actions = {"version": self.version, "connect": self.connect, "add": self.add, "del": self.delete, "mod": self.modify, "show": self.show}
|
|
return actions.get(args.action)(args)
|
|
|
|
def version(self, args):
|
|
from .._version import __version__
|
|
printer.info(f"Connpy {__version__}")
|
|
|
|
def connect(self, args):
|
|
if args.data == None:
|
|
try:
|
|
matches = self.app.services.nodes.list_nodes()
|
|
except Exception as e:
|
|
printer.error(f"Failed to list nodes: {e}")
|
|
sys.exit(1)
|
|
|
|
if len(matches) == 0:
|
|
printer.warning("There are no nodes created")
|
|
printer.info("try: connpy --help")
|
|
sys.exit(9)
|
|
else:
|
|
try:
|
|
matches = self.app.services.nodes.list_nodes(args.data)
|
|
matches = self._filter_exact_match(matches, args.data)
|
|
except Exception:
|
|
matches = []
|
|
|
|
if len(matches) == 0:
|
|
printer.error(f"{args.data} not found")
|
|
sys.exit(2)
|
|
elif len(matches) > 1:
|
|
matches[0] = choose(self.app, matches, "node", "connect")
|
|
|
|
if matches[0] == None:
|
|
sys.exit(7)
|
|
|
|
try:
|
|
self.app.services.nodes.connect_node(
|
|
matches[0],
|
|
sftp=args.sftp,
|
|
debug=args.debug,
|
|
logger=self.app._service_logger
|
|
)
|
|
except (ConnpyError, ValueError) as e:
|
|
printer.error(str(e))
|
|
sys.exit(1)
|
|
|
|
def delete(self, args):
|
|
if args.data == None:
|
|
printer.error("Missing argument node")
|
|
sys.exit(3)
|
|
|
|
is_folder = args.data.startswith("@")
|
|
try:
|
|
if is_folder:
|
|
matches = self.app.services.nodes.list_folders(args.data)
|
|
else:
|
|
matches = self.app.services.nodes.list_nodes(args.data)
|
|
matches = self._filter_exact_match(matches, args.data)
|
|
except Exception:
|
|
matches = []
|
|
|
|
if len(matches) == 0:
|
|
printer.error(f"{args.data} not found")
|
|
sys.exit(2)
|
|
|
|
printer.info(f"Removing: {matches}")
|
|
import inquirer
|
|
question = [inquirer.Confirm("delete", message="Are you sure you want to continue?")]
|
|
confirm = inquirer.prompt(question)
|
|
if confirm == None or not confirm["delete"]:
|
|
sys.exit(7)
|
|
|
|
try:
|
|
for i, item in enumerate(matches):
|
|
save_on_last = (i == len(matches) - 1)
|
|
self.app.services.nodes.delete_node(item, is_folder=is_folder, save=save_on_last)
|
|
|
|
if len(matches) == 1:
|
|
printer.success(f"{matches[0]} deleted successfully")
|
|
else:
|
|
printer.success(f"{len(matches)} items deleted successfully")
|
|
except ConnpyError as e:
|
|
printer.error(str(e))
|
|
sys.exit(1)
|
|
|
|
def add(self, args):
|
|
try:
|
|
args.data = self.app._type_node(args.data)
|
|
except ValueError as e:
|
|
printer.error(str(e))
|
|
sys.exit(3)
|
|
|
|
if args.data == None:
|
|
printer.error("Missing argument node")
|
|
sys.exit(3)
|
|
|
|
is_folder = args.data.startswith("@")
|
|
try:
|
|
if is_folder:
|
|
uniques = self.app.services.nodes.explode_unique(args.data)
|
|
if not uniques:
|
|
raise InvalidConfigurationError(f"Invalid folder {args.data}")
|
|
self.app.services.nodes.add_node(args.data, {}, is_folder=True)
|
|
printer.success(f"{args.data} added successfully")
|
|
else:
|
|
if args.data in self.app.nodes_list:
|
|
printer.error(f"Node '{args.data}' already exists.")
|
|
sys.exit(1)
|
|
uniques = self.app.services.nodes.explode_unique(args.data)
|
|
|
|
# Fast fail if parent folder does not exist
|
|
self.app.services.nodes.validate_parent_folder(args.data)
|
|
|
|
from rich.markdown import Markdown
|
|
printer.console.print(Markdown(get_instructions()))
|
|
|
|
new_node_data = self.forms.questions_nodes(args.data, uniques)
|
|
if not new_node_data:
|
|
sys.exit(7)
|
|
self.app.services.nodes.add_node(args.data, new_node_data)
|
|
printer.success(f"{args.data} added successfully")
|
|
except ConnpyError as e:
|
|
printer.error(str(e))
|
|
sys.exit(1)
|
|
|
|
def show(self, args):
|
|
if args.data == None:
|
|
printer.error("Missing argument node")
|
|
sys.exit(3)
|
|
|
|
try:
|
|
matches = self.app.services.nodes.list_nodes(args.data)
|
|
matches = self._filter_exact_match(matches, args.data)
|
|
except Exception:
|
|
matches = []
|
|
|
|
if len(matches) == 0:
|
|
printer.error(f"{args.data} not found")
|
|
sys.exit(2)
|
|
elif len(matches) > 1:
|
|
matches[0] = choose(self.app, matches, "node", "show")
|
|
|
|
if matches[0] == None:
|
|
sys.exit(7)
|
|
|
|
try:
|
|
node = self.app.services.nodes.get_node_details(matches[0])
|
|
yaml_output = yaml.dump(node, sort_keys=False, default_flow_style=False)
|
|
printer.data(matches[0], yaml_output)
|
|
except ConnpyError as e:
|
|
printer.error(str(e))
|
|
sys.exit(1)
|
|
|
|
def modify(self, args):
|
|
if args.data == None:
|
|
printer.error("Missing argument node")
|
|
sys.exit(3)
|
|
|
|
try:
|
|
matches = self.app.services.nodes.list_nodes(args.data)
|
|
matches = self._filter_exact_match(matches, args.data)
|
|
except Exception:
|
|
matches = []
|
|
|
|
if len(matches) == 0:
|
|
printer.error(f"No connection found with filter: {args.data}")
|
|
sys.exit(2)
|
|
|
|
unique = matches[0] if len(matches) == 1 else None
|
|
uniques = self.app.services.nodes.explode_unique(unique) if unique else {"id": None, "folder": None}
|
|
|
|
printer.info(f"Editing: {matches}")
|
|
node_details = {}
|
|
for i in matches:
|
|
node_details[i] = self.app.services.nodes.get_node_details(i)
|
|
|
|
edits = self.forms.questions_edit()
|
|
if edits == None:
|
|
sys.exit(7)
|
|
|
|
# Use first match as base for defaults if multiple matches exist
|
|
base_unique = matches[0]
|
|
base_uniques = self.app.services.nodes.explode_unique(base_unique)
|
|
updatenode = self.forms.questions_nodes(base_unique, base_uniques, edit=edits)
|
|
if not updatenode:
|
|
sys.exit(7)
|
|
|
|
try:
|
|
if len(matches) == 1:
|
|
# Comparison for "Nothing to do"
|
|
current = node_details[matches[0]].copy()
|
|
current.update(uniques)
|
|
current["type"] = "connection"
|
|
if sorted(updatenode.items()) == sorted(current.items()):
|
|
printer.info("Nothing to do here")
|
|
return
|
|
self.app.services.nodes.update_node(matches[0], updatenode)
|
|
printer.success(f"{args.data} edited successfully")
|
|
else:
|
|
changed_items = []
|
|
for k in matches:
|
|
updated_item = self.app.services.nodes.explode_unique(k)
|
|
updated_item["type"] = "connection"
|
|
updated_item.update(node_details[k])
|
|
|
|
this_item_changed = False
|
|
for key, should_edit in edits.items():
|
|
if should_edit:
|
|
this_item_changed = True
|
|
updated_item[key] = updatenode[key]
|
|
|
|
if this_item_changed:
|
|
changed_items.append((k, updated_item))
|
|
|
|
editcount = len(changed_items)
|
|
for i, (k, updated_item) in enumerate(changed_items):
|
|
save_on_last = (i == editcount - 1)
|
|
self.app.services.nodes.update_node(k, updated_item, save=save_on_last)
|
|
|
|
if editcount == 0:
|
|
printer.info("Nothing to do here")
|
|
else:
|
|
printer.success(f"{matches} edited successfully")
|
|
except ConnpyError as e:
|
|
printer.error(str(e))
|
|
sys.exit(1)
|