Coverage for mcp/feature_flags.py: 100%

1"""Feature-flag evaluation for the GCO MCP server. 

2 

3Convention: a flag is enabled if and only if (a) the umbrella flag 

4GCO_ENABLE_ALL_TOOLS is set to "true", OR (b) the case-insensitive, 

5whitespace-stripped value of os.environ.get(<flag>, "") equals the 

6literal "true". Anything else (unset, empty, "false", "0", "yes", "1", 

7"True\n" without strip, ...) is disabled. 

8 

9The umbrella flag is mutually inclusive — setting GCO_ENABLE_ALL_TOOLS=true 

10overrides per-flag values, even per-flag values explicitly set to "false". 

11""" 

12 

13import os 

14 

15# Known flag constants. Tool modules import these by name. 

16FLAG_ALL_TOOLS = "GCO_ENABLE_ALL_TOOLS" 

17FLAG_CAPACITY_PURCHASE = "GCO_ENABLE_CAPACITY_PURCHASE" 

18FLAG_MODEL_UPLOAD = "GCO_ENABLE_MODEL_UPLOAD" 

19FLAG_IMAGE_PUBLISH = "GCO_ENABLE_IMAGE_PUBLISH" 

20FLAG_INFRASTRUCTURE_DEPLOY = "GCO_ENABLE_INFRASTRUCTURE_DEPLOY" 

21FLAG_INFRASTRUCTURE_DESTROY = "GCO_ENABLE_INFRASTRUCTURE_DESTROY" 

22FLAG_DESTRUCTIVE_OPERATIONS = "GCO_ENABLE_DESTRUCTIVE_OPERATIONS" 

23FLAG_MISSION = "GCO_ENABLE_MISSION" 

24 

25# Per-tool flags. The umbrella is intentionally not in this tuple — callers 

26# iterating ALL_FLAGS for "what gates this tool?" lookups should not see 

27# the umbrella, only the per-tool flags. 

28ALL_FLAGS = ( 

29 FLAG_CAPACITY_PURCHASE, 

30 FLAG_MODEL_UPLOAD, 

31 FLAG_IMAGE_PUBLISH, 

32 FLAG_INFRASTRUCTURE_DEPLOY, 

33 FLAG_INFRASTRUCTURE_DESTROY, 

34 FLAG_DESTRUCTIVE_OPERATIONS, 

35 FLAG_MISSION, 

36) 

37 

38 

39def _raw(flag_name: str) -> bool: 

40 """Return True iff the env var equals literal "true" (case-insensitive, stripped).""" 

41 return os.environ.get(flag_name, "").strip().lower() == "true" 

42 

43 

44def is_enabled(flag_name: str) -> bool: 

45 """Return True iff the umbrella flag is set OR the named flag is set.""" 

46 return _raw(FLAG_ALL_TOOLS) or _raw(flag_name) 

47 

48 

49def all_tools_enabled() -> bool: 

50 """Return True iff GCO_ENABLE_ALL_TOOLS is set. Used by emit_startup_log.""" 

51 return _raw(FLAG_ALL_TOOLS)