feat(selection): persist validated update preferences atomically

2026-09-16 19:18:24 +02:00
1 parent 50cb82074e
commit 3a5790e67c
@@ -0,0 +1,77 @@
+import argparse
+import csv
+import json
+import os
+from pathlib import Path
+import sys
+import tempfile
+
+
+def validate(state):
+    if not isinstance(state, dict) or state.get('version') != 1:
+        raise ValueError('Unsupported saved selection.')
+    if state.get('platform') not in ('windows', 'linux') or state.get('client') not in ('codex', 'claude', 'both'):
+        raise ValueError('Invalid saved platform or client.')
+    for key in ('selected', 'deselected'):
+        values = state.get(key)
+        if not isinstance(values, list) or any(not isinstance(value, str) or not value or any(ord(c) < 32 for c in value) for value in values):
+            raise ValueError('Invalid saved plugin selection.')
+        if len(values) != len(set(values)):
+            raise ValueError('Duplicate saved plugin selection.')
+    if set(state['selected']) & set(state['deselected']):
+        raise ValueError('Conflicting saved plugin selection.')
+    return state
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('action', choices=('read', 'write'))
+    parser.add_argument('--home', required=True)
+    parser.add_argument('--manifest', required=True)
+    parser.add_argument('--format', choices=('json', 'tsv'), default='json')
+    parser.add_argument('--platform')
+    parser.add_argument('--client')
+    parser.add_argument('--selected', nargs='*', default=[])
+    parser.add_argument('--deselected', nargs='*', default=[])
+    args = parser.parse_args()
+    path = Path(args.home) / '.my-ai-configuration' / 'selection.json'
+    with open(args.manifest, encoding='utf-8-sig', newline='') as stream:
+        names = {row['name'] for row in csv.DictReader(stream, delimiter='\t')}
+    if args.action == 'read':
+        if not path.is_file():
+            raise ValueError('No saved selection. Run install or update once without Quick.')
+        state = validate(json.loads(path.read_text(encoding='utf-8-sig')))
+        for key in ('selected', 'deselected'):
+            state[key] = [name for name in state[key] if name in names]
+        if args.format == 'json':
+            print(json.dumps(state))
+        else:
+            print('platform\t' + state['platform'])
+            print('client\t' + state['client'])
+            for key in ('selected', 'deselected'):
+                for name in state[key]:
+                    print(key + '\t' + name)
+        return
+    state = validate(dict(version=1, platform=args.platform, client=args.client,
+                          selected=args.selected, deselected=args.deselected))
+    if (set(state['selected']) | set(state['deselected'])) - names:
+        raise ValueError('Unknown plugin in selection.')
+    path.parent.mkdir(parents=True, exist_ok=True)
+    temporary = None
+    try:
+        with tempfile.NamedTemporaryFile(mode='w', encoding='utf-8', dir=path.parent, delete=False) as stream:
+            temporary = Path(stream.name)
+            json.dump(state, stream, indent=2)
+            stream.write('\n')
+        os.replace(temporary, path)
+    finally:
+        if temporary is not None:
+            temporary.unlink(missing_ok=True)
+
+
+if __name__ == '__main__':
+    try:
+        main()
+    except (OSError, ValueError, KeyError) as error:
+        print(f'Selection state: {error}', file=sys.stderr)
+        sys.exit(1)