You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

99 lines
2.5 KiB

#!/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))