feat: manage all beta features (#2322)

Signed-off-by: Matteo Fari <matteofari06@gmail.com>

Author: MatteoFari

docs/tutorials/cli.md

@@ -202,6 +202,14 @@ trestle beta enable example-feature
 trestle beta disable example-feature
 ```
 
+Use `all` to enable every registered beta feature or disable every feature stored in the selected config. Features
+enabled by environment or by default remain enabled.
+
+```bash
+trestle beta enable all
+trestle beta disable all
+```
+
 Use `--beta` on a beta command to run it one time without writing beta state to config. Commands that are not beta
 features will warn that `--beta` is only effective for beta level commands, but will otherwise proceed normally.
 

tests/trestle/core/beta_features_test.py

@@ -76,6 +76,27 @@ def test_enable_disable_feature_persists_to_workspace_config(
     assert not beta_features.disable_feature('sample', tmp_trestle_dir)
 
 
+def test_enable_disable_all_features_preserves_unknown_config(
+    tmp_trestle_dir: pathlib.Path, monkeypatch: MonkeyPatch
+) -> None:
+    """Test bulk beta updates preserve unregistered config entries."""
+    patch_beta_features(monkeypatch, sample_feature('one'), sample_feature('two'))
+    config_path = beta_features.get_beta_config_path(tmp_trestle_dir)
+    config_path.write_text('[beta]\nenabled_features = legacy\n', encoding='utf-8')
+
+    assert beta_features.enable_all_features(tmp_trestle_dir) == 2
+    assert beta_features.get_enabled_features(tmp_trestle_dir) == {'one', 'two'}
+    assert beta_features.enable_all_features(tmp_trestle_dir) == 0
+
+    assert beta_features.disable_all_features(tmp_trestle_dir) == 2
+    assert not beta_features.get_enabled_features(tmp_trestle_dir)
+    assert beta_features.disable_all_features(tmp_trestle_dir) == 0
+
+    config = configparser.ConfigParser()
+    config.read(config_path)
+    assert config.get(beta_features.BETA_SECTION, beta_features.ENABLED_FEATURES_KEY) == 'legacy'
+
+
 def test_enable_feature_preserves_existing_config_sections(
     tmp_trestle_dir: pathlib.Path, monkeypatch: MonkeyPatch
 ) -> None:

tests/trestle/core/commands/beta_test.py

@@ -90,6 +90,38 @@ def test_beta_enable_query_disable_feature(
     assert not beta_features.is_beta_enabled('sample', tmp_trestle_dir)
 
 
+def test_beta_enable_disable_all(
+    tmp_trestle_dir: pathlib.Path, monkeypatch: MonkeyPatch, capsys: CaptureFixture[str]
+) -> None:
+    """Test enabling and disabling all registered beta features."""
+    patch_beta_features(monkeypatch, sample_feature('one'), sample_feature('two'))
+
+    execute_command_and_assert('trestle beta enable all', CmdReturnCodes.SUCCESS.value, monkeypatch)
+    output, _ = capsys.readouterr()
+    assert 'Enabled 2 beta feature(s).' in output
+    assert beta_features.get_enabled_features(tmp_trestle_dir) == {'one', 'two'}
+
+    execute_command_and_assert('trestle beta disable all', CmdReturnCodes.SUCCESS.value, monkeypatch)
+    output, _ = capsys.readouterr()
+    assert 'Disabled 2 beta feature(s).' in output
+    assert not beta_features.get_enabled_features(tmp_trestle_dir)
+
+
+def test_beta_disable_all_reports_environment_enabled_features(
+    tmp_trestle_dir: pathlib.Path, monkeypatch: MonkeyPatch, capsys: CaptureFixture[str]
+) -> None:
+    """Test disabling all reports features that config cannot disable."""
+    patch_beta_features(monkeypatch, sample_feature())
+    monkeypatch.setenv(beta_features.TRESTLE_BETA_FEATURES_ENV, 'sample')
+
+    execute_command_and_assert('trestle beta disable all', CmdReturnCodes.SUCCESS.value, monkeypatch)
+
+    output, _ = capsys.readouterr()
+    assert 'Disabled 0 beta feature(s).' in output
+    assert 'Features still enabled by environment or default: sample' in output
+    assert beta_features.is_beta_enabled('sample', tmp_trestle_dir)
+
+
 def test_beta_query_verbose(monkeypatch: MonkeyPatch, capsys: CaptureFixture[str]) -> None:
     """Test verbose beta query output."""
     patch_beta_features(monkeypatch, sample_feature())

trestle/core/beta_features.py

@@ -163,6 +163,26 @@ def disable_feature(feature_name: str, trestle_root: pathlib.Path) -> bool:
     return True
 
 
+def enable_all_features(trestle_root: pathlib.Path) -> int:
+    """Enable every registered beta feature and return the number changed."""
+    config_path = get_beta_config_path(trestle_root)
+    enabled_features = _read_config_enabled_features(config_path)
+    newly_enabled = set(BETA_FEATURES).difference(enabled_features)
+    if newly_enabled:
+        _write_config_enabled_features(config_path, enabled_features.union(newly_enabled))
+    return len(newly_enabled)
+
+
+def disable_all_features(trestle_root: pathlib.Path) -> int:
+    """Disable every config-enabled beta feature and return the number changed."""
+    config_path = get_beta_config_path(trestle_root)
+    enabled_features = _read_config_enabled_features(config_path)
+    configured_features = set(BETA_FEATURES).intersection(enabled_features)
+    if configured_features:
+        _write_config_enabled_features(config_path, enabled_features.difference(configured_features))
+    return len(configured_features)
+
+
 def beta_feature(feature_name: str) -> Callable[[BetaCallable], BetaCallable]:
     """Decorate a command run method so it requires an enabled beta feature."""
 

trestle/core/commands/beta.py

@@ -88,6 +88,8 @@ def _show_help(self) -> None:
         self.out('  trestle beta query             List beta features and their status')
         self.out('  trestle beta enable <feature>  Enable a beta feature')
         self.out('  trestle beta disable <feature> Disable a beta feature')
+        self.out('  trestle beta enable all        Enable all beta features')
+        self.out('  trestle beta disable all       Disable all config-enabled beta features')
         self.out('')
         self.out('Use "trestle beta query --verbose" for detailed descriptions.')
 
@@ -151,6 +153,10 @@ def _enable_feature(self, feature_name: str, trestle_root: pathlib.Path) -> None
         """Enable a beta feature."""
         if not feature_name:
             raise TrestleIncorrectArgsError('A beta feature name is required.')
+        if feature_name == 'all':
+            changed = beta_features.enable_all_features(trestle_root)
+            self.out(f'Enabled {changed} beta feature(s).')
+            return
 
         feature = self._get_feature(feature_name)
         changed = beta_features.enable_feature(feature_name, trestle_root)
@@ -176,6 +182,13 @@ def _disable_feature(self, feature_name: str, trestle_root: pathlib.Path) -> Non
         """Disable a beta feature."""
         if not feature_name:
             raise TrestleIncorrectArgsError('A beta feature name is required.')
+        if feature_name == 'all':
+            changed = beta_features.disable_all_features(trestle_root)
+            self.out(f'Disabled {changed} beta feature(s).')
+            remaining = beta_features.get_enabled_features(trestle_root)
+            if remaining:
+                self.out(f'Features still enabled by environment or default: {", ".join(sorted(remaining))}')
+            return
 
         feature = self._get_feature(feature_name)
         changed = beta_features.disable_feature(feature_name, trestle_root)