100 lines
2.5 KiB
Python
100 lines
2.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import errno
|
||
|
|
import re
|
||
|
|
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
item_regex = re.compile(r'.*(?P<type>\w):"(?P<key>.+?)"=(?P<value>.*)$', re.DOTALL)
|
||
|
|
|
||
|
|
def parse_ini(path: str) -> dict:
|
||
|
|
values = []
|
||
|
|
with open(path) as f:
|
||
|
|
for line in f.readlines():
|
||
|
|
line = line.strip()
|
||
|
|
if line.startswith("\ufeff"):
|
||
|
|
line = line[1:]
|
||
|
|
if len(line) > 2 and (line[1] == ":" or line[2] == ":"):
|
||
|
|
values.append(line)
|
||
|
|
elif line == "":
|
||
|
|
continue
|
||
|
|
else:
|
||
|
|
values[-1] += "\n" + line
|
||
|
|
|
||
|
|
matches = []
|
||
|
|
for i in values:
|
||
|
|
try:
|
||
|
|
match = item_regex.match(i)
|
||
|
|
matches.append(match.groupdict())
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
result = {}
|
||
|
|
for match in matches:
|
||
|
|
t = match["type"]
|
||
|
|
k = match["key"]
|
||
|
|
v = match["value"]
|
||
|
|
|
||
|
|
if t == "S":
|
||
|
|
result[k] = v
|
||
|
|
if t == "D":
|
||
|
|
result[k] = int(v, 16)
|
||
|
|
# We don't care about the other types
|
||
|
|
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
def gen_cmdlines_ssh(ini: dict, cfgdir: str) -> List[str]:
|
||
|
|
hostname = ini["Hostname"]
|
||
|
|
port = ini.get("[SSH2] Port", 22)
|
||
|
|
username = ini["Username"]
|
||
|
|
keyfile = ini.get("Identity Filename V2", None)
|
||
|
|
if keyfile:
|
||
|
|
keyfile = os.path.normpath(
|
||
|
|
keyfile \
|
||
|
|
.replace('\\', '/') \
|
||
|
|
.replace('${VDS_CONFIG_PATH}', cfgdir)
|
||
|
|
) \
|
||
|
|
.replace('"', '\\"')
|
||
|
|
|
||
|
|
cmdline = []
|
||
|
|
if keyfile:
|
||
|
|
cmdline += ["-i", f'"{keyfile}"']
|
||
|
|
if port != 22:
|
||
|
|
cmdline += ["-p", str(port)]
|
||
|
|
|
||
|
|
cmdline.append(f"{username}@{hostname}")
|
||
|
|
|
||
|
|
ssh = ["ssh"] + cmdline
|
||
|
|
scp = ["scp"] + cmdline
|
||
|
|
|
||
|
|
return [' '.join(ssh), ' '.join(scp)]
|
||
|
|
|
||
|
|
|
||
|
|
def gen_cmdlines(ini: dict, cfgdir: str) -> List[str]:
|
||
|
|
if ini["Protocol Name"].startswith("SSH"):
|
||
|
|
return gen_cmdlines_ssh(ini, cfgdir)
|
||
|
|
return []
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) < 2:
|
||
|
|
print(f"Usage: {sys.argv[0]} path/to/Config path/to/session.ini", file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
cfg = sys.argv[1]
|
||
|
|
ini = sys.argv[2]
|
||
|
|
|
||
|
|
if not os.path.exists(ini):
|
||
|
|
raise OSError(f"Session file does not exist: '{ini}'", errno=errno.ENOENT)
|
||
|
|
|
||
|
|
if not os.path.isfile(ini):
|
||
|
|
raise OSError(f"Not a file: '{ini}'", errno=errno.ENFILE)
|
||
|
|
|
||
|
|
inidict = parse_ini(ini)
|
||
|
|
cmdlines = gen_cmdlines(inidict, cfg)
|
||
|
|
print('\n'.join(cmdlines))
|