98 lines
3.0 KiB
Python
98 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
|
import argparse
|
|
import getpass
|
|
import os
|
|
import posixpath
|
|
import shlex
|
|
import sys
|
|
import threading
|
|
import time
|
|
|
|
import paramiko
|
|
|
|
|
|
def pump_output(channel, stop_flag):
|
|
while not stop_flag["stop"]:
|
|
if channel.recv_ready():
|
|
data = channel.recv(4096)
|
|
if not data:
|
|
break
|
|
sys.stdout.buffer.write(data)
|
|
sys.stdout.buffer.flush()
|
|
continue
|
|
if channel.exit_status_ready():
|
|
if not channel.recv_ready():
|
|
break
|
|
time.sleep(0.05)
|
|
stop_flag["stop"] = True
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Upload deploy-neko.sh and run it interactively on a remote Linux server.")
|
|
parser.add_argument("--host", default="192.168.2.179")
|
|
parser.add_argument("--user", default="eric")
|
|
parser.add_argument("--password", default=os.environ.get("NEKO_REMOTE_PASSWORD"))
|
|
parser.add_argument("--local-script", default=os.path.join(os.path.dirname(__file__), "deploy-neko.sh"))
|
|
parser.add_argument("--remote-dir", default="~/neko-tools")
|
|
parser.add_argument("--args", default="", help="Optional arguments passed to deploy-neko.sh")
|
|
args = parser.parse_args()
|
|
|
|
if not args.password:
|
|
args.password = getpass.getpass("SSH password: ")
|
|
|
|
local_script = os.path.abspath(args.local_script)
|
|
if not os.path.isfile(local_script):
|
|
raise SystemExit(f"Script not found: {local_script}")
|
|
|
|
client = paramiko.SSHClient()
|
|
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
client.connect(hostname=args.host, username=args.user, password=args.password, timeout=15)
|
|
|
|
sftp = client.open_sftp()
|
|
remote_home = sftp.normalize(".")
|
|
remote_dir = args.remote_dir.replace("~", remote_home, 1)
|
|
try:
|
|
sftp.stat(remote_dir)
|
|
except IOError:
|
|
client.exec_command(f"mkdir -p {shlex.quote(remote_dir)}")
|
|
time.sleep(0.2)
|
|
remote_script = posixpath.join(remote_dir, "deploy-neko.sh")
|
|
sftp.put(local_script, remote_script)
|
|
sftp.chmod(remote_script, 0o755)
|
|
sftp.close()
|
|
|
|
channel = client.invoke_shell(width=180, height=40)
|
|
stop_flag = {"stop": False}
|
|
worker = threading.Thread(target=pump_output, args=(channel, stop_flag), daemon=True)
|
|
worker.start()
|
|
|
|
time.sleep(0.2)
|
|
quoted_remote_dir = shlex.quote(remote_dir)
|
|
quoted_remote_script = shlex.quote(remote_script)
|
|
trailing_args = f" {args.args.strip()}" if args.args.strip() else ""
|
|
channel.send(f"cd {quoted_remote_dir}\n")
|
|
channel.send(f"bash {quoted_remote_script}{trailing_args}\n")
|
|
|
|
try:
|
|
while not stop_flag["stop"]:
|
|
line = sys.stdin.readline()
|
|
if not line:
|
|
break
|
|
channel.send(line)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
finally:
|
|
time.sleep(0.2)
|
|
try:
|
|
channel.send("\n")
|
|
except Exception:
|
|
pass
|
|
stop_flag["stop"] = True
|
|
worker.join(timeout=1)
|
|
channel.close()
|
|
client.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|