update

2025-04-11 15:50:28 +08:00
1 parent 19ae1e6f91
commit 69f0301bef
@@ -1,107 +0,0 @@
-#!/usr/bin/env python3
-
-from __future__ import annotations
-
-import argparse
-import sys
-from pathlib import Path
-from collections import namedtuple
-
-
-class Entry(namedtuple('Entry', ['host', 'algorithm', 'hash'])):
-    separator: str = '.'
-
-    def __str__(self) -> str:
-        return f'{self.host} {self.algorithm} {self.hash}'
-
-    @property
-    def key(self) -> str:
-        return f'{self.algorithm}{self.separator}{self.hash}'
-
-
-def run(args: argparse.Namespace) -> None:
-    filename: Path = getattr(args, 'input', Path())
-    file_path = filename.resolve()
-    if not file_path.is_file():
-        sys.exit(1)
-    entryMap = {}
-
-    with file_path.open(encoding='utf-8') as fp:
-        for lineNumber, line in enumerate(fp):
-            content = line.strip().split()
-            if len(content) == 0:
-                continue  # empty line -> skip
-            if line[0].startswith('#') or len(content) != 3:
-                if not line[0].startswith('#'):
-                    sys.stderr.write(f'could not parse a line {lineNumber}: "{line}"\n')
-                sys.stdout.write(line)
-                continue  # comment or erroneous line -> skip
-            entry = Entry(*content)
-            entryMap.setdefault(entry.key, []).append(entry)
-
-    compactEntryList = []
-
-    for key, entryList in entryMap.items():
-        domainList = []
-        ipList = []
-        entry: Entry
-        for entry in entryList:
-            for domainOrIp in entry.host.split(','):
-                if len(domainOrIp) < 2:
-                    sys.stderr.write(f'could not parse {domainOrIp}\n')
-                    continue
-                start_loc = 0
-                if domainOrIp[0] == '[':
-                    start_loc = 1
-                if 'a' <= domainOrIp[start_loc].lower() <= 'z':
-                    domainList.append(domainOrIp)
-                else:
-                    ipList.append(domainOrIp)
-        host = ','.join(sorted(domainList) + sorted(ipList))
-        compactEntryList.append(Entry(host, *key.split(Entry.separator)))
-
-    output: Path | None = getattr(args, 'output', None)
-    output_lines = '\n'.join(
-        map(
-            str,
-            sorted(
-                compactEntryList,
-                key=lambda entry: (
-                    entry.host[1:] if entry.host.startswith('[') else entry.host
-                ),
-            ),
-        )
-    )
-    if output is None:
-        for line in output_lines:
-            print(line, end='')
-    else:
-        output_file = output.resolve()
-        if output_file.is_dir():
-            print('Can not write to directory.')
-            sys.exit(1)
-        if not output_file.parent.exists():
-            output_file.parent.mkdir(parents=True)
-        with output_file.open('w', encoding='utf-8') as wfp:
-            wfp.writelines(output_lines)
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser(
-        formatter_class=argparse.ArgumentDefaultsHelpFormatter
-    )
-    parser.add_argument(
-        '-i',
-        '--input',
-        type=Path,
-        default=f'{Path("~").expanduser()}/.ssh/known_hosts',
-        help='file to read',
-    )
-    parser.add_argument('-o', '--output', type=Path, help='file to write')
-    parser.set_defaults(func=run)
-    args = parser.parse_args()
-    args.func(args)
-
-
-if __name__ == '__main__':
-    main()