Jack demo: MQ Server handshake test (retry 1) (retry 2)

{
  "cost": {
    "input_tokens": 68,
    "estimated_usd": 0.067359,
    "output_tokens": 4477
  },
  "model": "claude-sonnet-4-20250514",
  "status": "success",
  "transport": "anthropic_api",
  "latency_ms": 59565,
  "response_text": "I'll help you create a comprehensive MQ Server handshake test for the CBP ABI system with ISA/GS envelope generation for Entry Summary transactions. Here's the implementation:\n\n```python\n#!/usr/bin/env python3\n\"\"\"\nJack Demo: MQ Server handshake test with retry logic\nCBP ABI System Entry Summary (Type 01) Test\nFile: demo_jack_7501_retry.py\n\"\"\"\n\nimport os\nimport sys\nimport time\nimport logging\nfrom datetime import datetime\nfrom typing import Optional, Dict, Any\nimport pymqi\nimport json\n\n# Configure logging\nlogging.basicConfig(\n    level=logging.INFO,\n    format='%(asctime)s - %(levelname)s - %(message)s',\n    handlers=[\n        logging.FileHandler('jack_demo_7501.log'),\n        logging.StreamHandler(sys.stdout)\n    ]\n)\nlogger = logging.getLogger(__name__)\n\nclass CBPABIConnector:\n    \"\"\"CBP ABI MQ Connection Handler with retry logic\"\"\"\n    \n    def __init__(self, config: Dict[str, Any]):\n        self.config = config\n        self.qmgr = None\n        self.queue = None\n        self.connected = False\n        self.max_retries = config.get('max_retries', 3)\n        self.retry_delay = config.get('retry_delay', 5)\n        \n    def connect_with_retry(self) -> bool:\n        \"\"\"Connect to MQ with retry logic\"\"\"\n        for attempt in range(1, self.max_retries + 1):\n            logger.info(f\"MQ Server handshake test (retry {attempt})\")\n            \n            try:\n                # MQ Connection parameters\n                qmgr_name = self.config['queue_manager']\n                channel = self.config['channel']\n                host = self.config['host']\n                port = self.config['port']\n                queue_name = self.config['queue_name']\n                \n                # Connection string\n                conn_info = f\"{host}({port})\"\n                \n                logger.info(f\"Attempting connection to {qmgr_name} at {conn_info}\")\n                \n                # Create connection\n                cd = pymqi.CD()\n                cd.ChannelName = channel\n                cd.ConnectionName = conn_info\n                cd.ChannelType = pymqi.CMQC.MQCHT_CLNTCONN\n                cd.TransportType = pymqi.CMQC.MQXPT_TCP\n                \n                # Connect to queue manager\n                self.qmgr = pymqi.connect(qmgr_name, cd=cd)\n                logger.info(f\"✅ Connected to queue manager: {qmgr_name}\")\n                \n                # Open queue\n                self.queue = pymqi.Queue(self.qmgr, queue_name)\n                logger.info(f\"✅ Opened queue: {queue_name}\")\n                \n                self.connected = True\n                return True\n                \n            except pymqi.MQMIError as e:\n                logger.error(f\"❌ MQ Error on attempt {attempt}: {e}\")\n                self._handle_mq_error(e)\n                \n            except Exception as e:\n                logger.error(f\"❌ Connection failed on attempt {attempt}: {str(e)}\")\n                \n            if attempt < self.max_retries:\n                logger.info(f\"Waiting {self.retry_delay} seconds before retry...\")\n                time.sleep(self.retry_delay)\n            else:\n                logger.error(\"❌ All connection attempts failed\")\n                \n        return False\n    \n    def _handle_mq_error(self, error: pymqi.MQMIError):\n        \"\"\"Handle specific MQ errors\"\"\"\n        error_codes = {\n            2009: \"Connection to queue manager failed\",\n            2058: \"Queue not found\",\n            2059: \"Queue manager name error\",\n            2538: \"Host not available\",\n            2035: \"Authorization failed\"\n        }\n        \n        error_msg = error_codes.get(error.reason, f\"Unknown MQ error: {error.reason}\")\n        logger.error(f\"MQ Error Detail: {error_msg}\")\n    \n    def disconnect(self):\n        \"\"\"Disconnect from MQ\"\"\"\n        try:\n            if self.queue:\n                self.queue.close()\n                logger.info(\"Queue closed\")\n            if self.qmgr:\n                self.qmgr.disconnect()\n                logger.info(\"Disconnected from queue manager\")\n        except Exception as e:\n            logger.error(f\"Error during disconnect: {e}\")\n\nclass ISAGSEnvelopeGenerator:\n    \"\"\"Generate ISA/GS envelopes for CBP ABI Entry Summary\"\"\"\n    \n    def __init__(self, sender_id: str, receiver_id: str):\n        self.sender_id = sender_id.ljust(15)[:15]  # Pad to 15 chars\n        self.receiver_id = receiver_id.ljust(15)[:15]\n        self.control_number = 1\n        \n    def generate_isa_header(self) -> str:\n        \"\"\"Generate ISA header segment\"\"\"\n        now = datetime.now()\n        date_str = now.strftime(\"%y%m%d\")  # YYMMDD\n        time_str = now.strftime(\"%H%M\")    # HHMM\n        \n        isa_segments = [\n            \"ISA\",                          # ISA01 - Authorization Info Qualifier\n            \"00\",                           # ISA02 - Authorization Information\n            \"          \",                   # ISA03 - Security Info Qualifier  \n            \"00\",                           # ISA04 - Security Information\n            \"          \",                   # ISA05 - Interchange ID Qualifier\n            \"ZZ\",                           # ISA06 - Interchange Sender ID\n            self.sender_id,                 # ISA07 - Interchange Receiver Qualifier\n            \"ZZ\",                           # ISA08 - Interchange Receiver ID\n            self.receiver_id,               # ISA09 - Interchange Date\n            date_str,                       # ISA10 - Interchange Time\n            time_str,                       # ISA11 - Interchange Control Standards\n            \"U\",                            # ISA12 - Interchange Control Version\n            f\"{self.control_number:09d}\",   # ISA13 - Interchange Control Number\n            \"0\",                            # ISA14 - Acknowledgment Requested\n            \"T\",                            # ISA15 - Test Indicator (T=Test, P=Production)\n            \":\"                             # ISA16 - Component Element Separator\n        ]\n        \n        return \"*\".join(isa_segments) + \"~\"\n    \n    def generate_gs_header(self) -> str:\n        \"\"\"Generate GS header segment for Entry Summary\"\"\"\n        now = datetime.now()\n        date_str = now.strftime(\"%Y%m%d\")  # YYYYMMDD\n        time_str = now.strftime(\"%H%M%S\")  # HHMMSS\n        \n        gs_segments = [\n            \"GS\",                           # GS01 - Functional ID Code\n            \"QM\",                           # GS02 - Application Sender Code (QM = Entry Summary)\n            self.sender_id.strip(),         # GS03 - Application Receiver Code\n            self.receiver_id.strip(),       # GS04 - Date\n            date_str,                       # GS05 - Time\n            time_str,                       # GS06 - Group Control Number\n            f\"{self.control_number:09d}\",   # GS07 - Responsible Agency Code\n            \"X\",                            # GS08 - Version/Release/Industry ID\n            \"004010VICS\"\n        ]\n        \n        return \"*\".join(gs_segments) + \"~\"\n    \n    def generate_st_header(self) -> str:\n        \"\"\"Generate ST header for Entry Summary Transaction\"\"\"\n        st_segments = [\n            \"ST\",                           # ST01 - Transaction Set ID Code\n            \"214\",                          # ST02 - Transaction Set Control Number (214 = Entry Summary)\n            f\"{self.control_number:09d}\"\n        ]\n        \n        return \"*\".join(st_segments) + \"~\"\n    \n    def generate_entry_summary_data(self) -> str:\n        \"\"\"Generate sample Entry Summary data (Type 01)\"\"\"\n        # B10 - Beginning Segment for Transportation Carrier Shipment Status Message\n        b10_segments = [\n            \"B10\",                          # B10 segment\n            \"01\",                           # Entry Summary Type (01)\n            \"TEST\" + str(self.control_number).zfill(7),  # Entry Number\n            datetime.now().strftime(\"%Y%m%d\")  # Entry Date\n        ]\n        \n        # N1 - Name segments for parties\n        n1_segments = [\n            \"N1\",                           # N1 segment\n            \"IM\",                           # Entity ID Code (IM = Importer)\n            \"TEST IMPORTER\",                # Name\n            \"91\",                           # ID Code Qualifier\n            \"123456789\"                     # Importer ID\n        ]\n        \n        return \"*\".join(b10_segments) + \"~\" + \"*\".join(n1_segments) + \"~\"\n    \n    def generate_trailer_segments(self) -> str:\n        \"\"\"Generate SE, GE, and IEA trailer segments\"\"\"\n        # ST trailer\n        se_segments = [\n            \"SE\",                           # SE01 - Transaction Set Trailer\n            \"4\",                            # SE02 - Number of included segments\n            f\"{self.control_number:09d}\"    # SE03 - Transaction Set Control Number\n        ]\n        \n        # GS trailer  \n        ge_segments = [\n            \"GE\",                           # GE01 - Group Trailer\n            \"1\",                            # GE02 - Number of transaction sets\n            f\"{self.control_number:09d}\"    # GE03 - Group Control Number\n        ]\n        \n        # ISA trailer\n        iea_segments = [\n            \"IEA\",                          # IEA01 - Interchange Trailer\n            \"1\",                            # IEA02 - Number of functional groups\n            f\"{self.control_number:09d}\"    # IEA03 - Interchange Control Number\n        ]\n        \n        return (\"~\".join([\n            \"*\".join(se_segments),\n            \"*\".join(ge_segments), \n            \"*\".join(iea_segments)\n        ]) + \"~\")\n    \n    def generate_complete_message(self) -> str:\n        \"\"\"Generate complete EDI message with ISA/GS envelope\"\"\"\n        message_parts = [\n            self.generate_isa_header(),\n            self.generate_gs_header(),\n            self.generate_st_header(),\n            self.generate_entry_summary_data(),\n            self.generate_trailer_segments()\n        ]\n        \n        complete_message = \"\".join(message_parts)\n        self.control_number += 1\n        \n        return complete_message\n\ndef test_mq_handshake():\n    \"\"\"Main test function\"\"\"\n    logger.info(\"=== Jack Demo: CBP ABI MQ Server Handshake Test ===\")\n    \n    # Configuration\n    config = {\n        'queue_manager': 'CBP.ABI.QM',\n        'channel': 'CBP.ABI.CHANNEL', \n        'host': 'abi-test.cbp.dhs.gov',  # Test environment\n        'port': 1414,\n        'queue_name': 'CBP.ABI.ENTRY.SUMMARY',\n        'max_retries': 3,\n        'retry_delay': 5,\n        'sender_id': 'TEST_SENDER_001',\n        'receiver_id': 'CBP_ABI_SYS'\n    }\n    \n    # Initialize connector\n    connector = CBPABIConnector(config)\n    \n    try:\n        # Test connection with retry\n        if not connector.connect_with_retry():\n            logger.error(\"❌ Failed to establish MQ connection after all retries\")\n            return False\n        \n        logger.info(\"✅ MQ handshake successful!\")\n        \n        # Generate ISA/GS envelope with Entry Summary\n        envelope_gen = ISAGSEnvelopeGenerator(\n            config['sender_id'], \n            config['receiver_id']\n        )\n        \n        # Create test message\n        edi_message = envelope_gen.generate_complete_message()\n        \n        logger.info(\"Generated Entry Summary EDI Message:\")\n        logger.info(\"-\" * 50)\n        logger.info(edi_message)\n        logger.info(\"-\" * 50)\n        \n        # Verify ISA/GS envelope structure\n        verify_envelope_structure(edi_message)\n        \n        # Test message sending (commented out for safety in demo)\n        # send_test_message(connector, edi_message)\n        \n        logger.info(\"✅ Test completed successfully!\")\n        return True\n        \n    except Exception as e:\n        logger.error(f\"❌ Test failed: {str(e)}\")\n        return False\n        \n    finally:\n        connector.disconnect()\n\ndef verify_envelope_structure(message: str) -> bool:\n    \"\"\"Verify the EDI envelope structure\"\"\"\n    logger.info(\"Verifying ISA/GS envelope structure...\")\n    \n    required_segments = ['ISA', 'GS', 'ST', 'B10', 'SE', 'GE', 'IEA']\n    found_segments = []\n    \n    for segment in required_segments:\n        if message.startswith(segment) or f\"~{segment}\" in message:\n            found_segments.append(segment)\n            logger.info(f\"✅ Found {segment} segment\")\n        else:\n            logger.error(f\"❌ Missing {segment} segment\")\n            return False\n    \n    logger.info(\"✅ All required envelope segments present\")\n    return True\n\ndef send_test_message(connector: CBPABIConnector, message: str):\n    \"\"\"Send test message to CBP ABI (use with caution)\"\"\"\n    logger.warning(\"⚠️  Sending test message to CBP ABI system...\")\n    \n    try:\n        # Convert message to bytes\n        message_bytes = message.encode('utf-8')\n        \n        # Put message to queue\n        connector.queue.put(message_bytes)\n        logger.info(\"✅ Message sent successfully\")\n        \n        # Log message details\n        logger.info(f\"Message size: {len(message_bytes)} bytes\")\n        logger.info(f\"Control number: {message[103:112]}\")  # ISA control number position\n        \n    except Exception as e:\n        logger.error(f\"❌ Failed to send message: {str(e)}\")\n\nif __name__ == \"__main__\":\n    print(\"Jack Demo: CBP ABI MQ Server Handshake Test\")\n    print(\"=\" * 50)\n    \n    success = test_mq_handshake()\n    \n    if success:\n        print(\"\\n✅ Demo completed successfully!\")\n        sys.exit(0)\n    else:\n        print(\"\\n❌ Demo failed!\")\n        sys.exit(1)\n```\n\nAnd here's the companion configuration file:\n\n```json\n{\n  \"cbp_abi_config\": {\n    \"environments\": {\n      \"test\": {\n        \"queue_manager\": \"CBP.ABI.QM.TEST\",\n        \"channel\": \"CBP.ABI.CHANNEL.TEST\",\n        \"host\": \"abi-test.cbp.dhs.gov\",\n        \"port\": 1414,\n        \"ssl_enabled\": true,\n        \"ssl_cipher_spec\": \"TLS_RSA_WITH_AES_256_CBC_SHA256\"\n      },\n      \"production\": {\n        \"queue_manager\": \"CBP.ABI.QM.PROD\",\n        \"channel\": \"CBP.ABI.CHANNEL.PROD\", \n        \"host\": \"abi-prod.cbp.dhs.gov\",\n        \"port\": 1414,\n        \"ssl_enabled\": true,\n        \"ssl_cipher_spec\": \"TLS_RSA_WITH_AES_256_CBC_SHA256\"\n      }\n    },\n    \"queues\": {\n      \"entry_summary\": \"CBP.ABI.ENTRY.SUMMARY\",\n      \"entry_summary_response\": \"CBP.ABI.ENTRY.SUMMARY.RESPONSE\",\n      \"cargo_release\": \"CBP.ABI.CARGO.RELEASE\",\n      \"manifest\": \"CBP.ABI.MANIFEST\"\n    },\n    \"retry_config\": {\n      \"max_retries\": 3,\n      \"retry_delay\": 5,\n      \"exponential_backoff\": true,\n      \"max_delay\": 60\n    },\n    \"edi_config\": {\n      \"sender_id\": \"YOUR_SENDER_ID\",\n      \"receiver_id\": \"CBPABI\",\n      \"test_indicator\": \"T\",\n      \"component_separator\": \":\",\n      \"element_separator\": \"*\",\n      \"segment_terminator\": \"~\"\n    }\n  }\n}\n```\n\n## Key Features:\n\n1. **Retry Logic**: Implements 3 retry attempts with configurable delays\n2. **ISA/GS Envelope Generation**: Proper EDI X12 envelope structure for CBP ABI\n3. **Entry Summary Type 01**: Generates sample Entry Summary transaction data\n4. **Error Handling**: Comprehensive MQ error handling with specific error codes\n5. **Logging**: Detailed logging for debugging and audit trails\n6. **Configuration**: Flexible configuration for different environments\n\n## Usage:\n\n```bash\n# Install dependencies\npip install pymqi\n\n# Run the demo\npython demo_jack_7501_retry.py\n```\n\n## Safety Notes:\n\n- The message sending function is commented out by default\n- Uses test indicators in EDI segments\n- Includes proper disconnect handling\n- Logs all operations for audit purposes\n\nThis implementation provides a robust foundation for CBP ABI integration testing with proper retry mechanisms and EDI envelope generation."
}