Add environment to output

Author: Chris Ballinger

cli/retention_message.py

@@ -111,10 +111,12 @@ def upload_message(args) -> None:
                 "status": "success",
                 "message_id": message_id,
                 "header": args.header,
-                "body": args.body
+                "body": args.body,
+                "environment": args.environment
             }))
         else:
             print(f"✓ Message uploaded successfully!")
+            print(f"  Environment: {args.environment}")
             print(f"  Message ID: {message_id}")
             if args.header:
                 print(f"  Header: {args.header}")
@@ -165,21 +167,22 @@ def list_messages(args) -> None:
                 for msg in response.messageIdentifiers:
                     messages.append({
                         "message_id": msg.messageIdentifier,
-                        "state": msg.state.value if msg.state else None
+                        "state": msg.messageState.value if msg.messageState else None
                     })
             print(json.dumps({
                 "status": "success",
                 "messages": messages,
-                "total_count": len(messages)
+                "total_count": len(messages),
+                "environment": args.environment
             }))
         else:
             if not response.messageIdentifiers or len(response.messageIdentifiers) == 0:
-                print("No retention messages found.")
+                print(f"No retention messages found in {args.environment}.")
             else:
-                print(f"Found {len(response.messageIdentifiers)} retention message(s):")
+                print(f"Found {len(response.messageIdentifiers)} retention message(s) in {args.environment}:")
                 print()
                 for msg in response.messageIdentifiers:
-                    state = msg.state.value if msg.state else "UNKNOWN"
+                    state = msg.messageState.value if msg.messageState else "UNKNOWN"
                     print(f"  Message ID: {msg.messageIdentifier}")
                     print(f"  State:      {state}")
                     print()
@@ -227,10 +230,12 @@ def delete_message(args) -> None:
             print(json.dumps({
                 "status": "success",
                 "message_id": args.message_id,
-                "action": "deleted"
+                "action": "deleted",
+                "environment": args.environment
             }))
         else:
             print(f"✓ Message deleted successfully!")
+            print(f"  Environment: {args.environment}")
             print(f"  Message ID: {args.message_id}")
 
     except APIException as e: