Coverage for gco_mcp/feature_flags.py: 100.00%

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# Read-only-but-sensitive metric readers: local-filesystem reads and per-call 

25# LLM scoring are default-off for security / cost reasons even though they 

26# never mutate AWS state. They are full members of the registry like any 

27# other per-tool flag. 

28FLAG_LOCAL_METRICS = "GCO_ENABLE_LOCAL_METRICS" 

29# Reads from or writes to the MCP host and can upload data to S3. 

30FLAG_LOCAL_STORAGE_SYNC = "GCO_ENABLE_LOCAL_STORAGE_SYNC" 

31FLAG_SEMANTIC_PROGRESS = "GCO_ENABLE_SEMANTIC_PROGRESS" 

32 

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

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

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

36ALL_FLAGS = ( 

37 FLAG_CAPACITY_PURCHASE, 

38 FLAG_MODEL_UPLOAD, 

39 FLAG_IMAGE_PUBLISH, 

40 FLAG_INFRASTRUCTURE_DEPLOY, 

41 FLAG_INFRASTRUCTURE_DESTROY, 

42 FLAG_DESTRUCTIVE_OPERATIONS, 

43 FLAG_MISSION, 

44 FLAG_LOCAL_METRICS, 

45 FLAG_LOCAL_STORAGE_SYNC, 

46 FLAG_SEMANTIC_PROGRESS, 

47) 

48 

49 

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

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

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

53 

54 

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

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

57 return _raw(FLAG_ALL_TOOLS) or _raw(flag_name) 

58 

59 

60def all_tools_enabled() -> bool: 

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

62 return _raw(FLAG_ALL_TOOLS)