add archivebox config command and move config into sections

archivebox/cli/archivebox_config.py

@@ -0,0 +1,117 @@
+#!/usr/bin/env python3
+
+__package__ = 'archivebox.cli'
+__command__ = 'archivebox config'
+__description__ = 'Get and set your ArchiveBox project configuration values'
+
+import sys
+import argparse
+
+from typing import Optional, List
+
+from ..legacy.util import SmartFormatter
+from ..legacy.config import (
+    check_data_folder,
+    OUTPUT_DIR,
+    write_config_file,
+    CONFIG,
+    ConfigDict,
+    stderr,
+)
+
+
+def main(args: List[str]=None, stdin: Optional[str]=None) -> None:
+    check_data_folder()
+    
+    args = sys.argv[1:] if args is None else args
+
+    parser = argparse.ArgumentParser(
+        prog=__command__,
+        description=__description__,
+        add_help=True,
+        formatter_class=SmartFormatter,
+    )
+    group = parser.add_mutually_exclusive_group()
+    group.add_argument(
+        '--get', #'-g',
+        action='store_true',
+        help="Get the value for the given config KEYs",
+    )
+    group.add_argument(
+        '--set', #'-s',
+        action='store_true',
+        help="Set the given KEY=VALUE config values",
+    )
+    parser.add_argument(
+        'config_options',
+        nargs='*',
+        type=str,
+        help='KEY or KEY=VALUE formatted config values to get or set',
+    )
+    command = parser.parse_args(args)
+
+    if stdin or not sys.stdin.isatty():
+        stdin_raw_text = stdin or sys.stdin.read()
+        if stdin_raw_text and command.config_options:
+            stderr(
+                '[X] You should either pass config values as an arguments '
+                'or via stdin, but not both.\n',
+                color='red',
+            )
+            raise SystemExit(1)
+
+        config_options = stdin_raw_text.split('\n')
+    else:
+        config_options = command.config_options
+
+    no_args = not (command.get or command.set or command.config_options)
+
+    matching_config: ConfigDict = {}
+    if command.get or no_args:
+        if config_options:
+            matching_config = {key: CONFIG[key] for key in config_options if key in CONFIG}
+            failed_config = [key for key in config_options if key not in CONFIG]
+            if failed_config:
+                stderr()
+                stderr('[X] These options failed to get', color='red')
+                stderr('    {}'.format('\n    '.join(config_options)))
+                raise SystemExit(1)
+        else:
+            matching_config = CONFIG
+        
+        print('\n'.join(f'{key}={val}' for key, val in matching_config.items()))
+        raise SystemExit(not matching_config)
+    elif command.set:
+        new_config = {}
+        failed_options = []
+        for line in config_options:
+            if line.startswith('#') or not line.strip():
+                continue
+            if '=' not in line:
+                stderr('[X] Config KEY=VALUE must have an = sign in it', color='red')
+                stderr(f'    {line}')
+                raise SystemExit(2)
+
+            key, val = line.split('=')
+            if key.upper().strip() in CONFIG:
+                new_config[key.upper().strip()] = val.strip()
+            else:
+                failed_options.append(line)
+
+        if new_config:
+            matching_config = write_config_file(new_config, out_dir=OUTPUT_DIR)
+            print('\n'.join(f'{key}={val}' for key, val in matching_config.items()))
+        if failed_options:
+            stderr()
+            stderr('[X] These options failed to set:', color='red')
+            stderr('    {}'.format('\n    '.join(failed_options)))
+        raise SystemExit(bool(failed_options))
+    else:
+        stderr('[X] You must pass either --get or --set, or no arguments to get the whole config.', color='red')
+        stderr('    archivebox config')
+        stderr('    archivebox config --get SOME_KEY')
+        stderr('    archivebox config --set SOME_KEY=SOME_VALUE')
+        raise SystemExit(2)
+
+if __name__ == '__main__':
+    main()