orion-bridge 2.0.0__tar.gz

orion_bridge-2.0.0/PKG-INFO

@@ -0,0 +1,204 @@
+Metadata-Version: 2.4
+Name: orion-bridge
+Version: 2.0.0
+Summary: ORION Lab Bridge — Modular device bridge for industrial robotics
+Author-email: Leonardo <a01174639@tec.mx>
+License: MIT
+Project-URL: Homepage, https://github.com/FILL_USER/orion-bridge
+Project-URL: Documentation, https://github.com/FILL_USER/orion-bridge#readme
+Project-URL: Issues, https://github.com/FILL_USER/orion-bridge/issues
+Project-URL: Source, https://github.com/FILL_USER/orion-bridge
+Keywords: robotics,industrial,automation,xarm,abb,plc,ros2,mujoco,industry-4.0
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: MacOS
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: System :: Hardware
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: websockets>=12.0
+Requires-Dist: rich>=13.0
+Provides-Extra: xarm
+Requires-Dist: xarm-python-sdk; extra == "xarm"
+Provides-Extra: abb
+Provides-Extra: plc
+Requires-Dist: python-snap7; extra == "plc"
+Provides-Extra: gazebo
+Requires-Dist: roslibpy; extra == "gazebo"
+Provides-Extra: mujoco
+Requires-Dist: mujoco>=3.5.0; extra == "mujoco"
+Provides-Extra: supabase
+Requires-Dist: supabase; extra == "supabase"
+Provides-Extra: http
+Requires-Dist: flask; extra == "http"
+Provides-Extra: all
+Requires-Dist: xarm-python-sdk; extra == "all"
+Requires-Dist: python-snap7; extra == "all"
+Requires-Dist: roslibpy; extra == "all"
+Requires-Dist: mujoco>=3.5.0; extra == "all"
+Requires-Dist: supabase; extra == "all"
+Requires-Dist: flask; extra == "all"
+
+# ORION Bridge
+
+**Modular device bridge connecting industrial robotics hardware to the ORION AI agent platform.**
+
+[![PyPI version](https://badge.fury.io/py/orion-bridge.svg)](https://pypi.org/project/orion-bridge/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+
+---
+
+## What is ORION Bridge?
+
+ORION Bridge is a lightweight Python daemon that sits between physical (or simulated) lab equipment and the [ORION agent platform](https://orion-learning.com). It speaks the hardware protocols — xArm SDK, Snap7 for Siemens S7-1200 PLCs, ABB Robot Web Services — and exposes a unified WebSocket interface that the ORION AI understands.
+
+The bridge runs on any machine with network access to your devices: a lab PC next to the robot arm, a Raspberry Pi on the factory floor, or your laptop for simulation. Once running, ORION can read live telemetry, execute actions, and run multi-step routines through natural language — without any changes to your hardware configuration.
+
+Supported hardware backends out of the box: **UFACTORY xArm 6/7** (physical + MuJoCo simulation), **ABB robots** (via Robot Web Services), **Siemens S7-1200 PLCs** (via python-snap7), and a **local shell** for scripting and debugging. A ROS/Gazebo adapter is also available for simulation-heavy workflows.
+
+---
+
+## Quick Start
+
+```bash
+pip install orion-bridge[all]
+orion
+```
+
+The interactive menu guides you through device setup and server connection. No config file needed for the first run — it generates one for you.
+
+---
+
+## Installation Options
+
+Install only the drivers you need:
+
+| Command | Installs |
+|---|---|
+| `pip install orion-bridge` | Core only (WebSocket transport, CLI) |
+| `pip install orion-bridge[xarm]` | + UFACTORY xArm SDK |
+| `pip install orion-bridge[plc]` | + Siemens S7-1200 (python-snap7) |
+| `pip install orion-bridge[mujoco]` | + MuJoCo simulation engine |
+| `pip install orion-bridge[gazebo]` | + ROS/Gazebo (roslibpy) |
+| `pip install orion-bridge[http]` | + PLC HTTP sidecar API (Flask) |
+| `pip install orion-bridge[all]` | Everything above |
+
+---
+
+## First Run
+
+```
+$ orion
+```
+
+The CLI opens an interactive menu:
+
+```
+[1] Add device       — register an xArm, ABB, PLC, or shell device
+[2] Remove device    — remove a device from the list
+[3] Configure server — pick a lab preset or enter a custom server URL + token
+[4] Show status      — display current devices and connection info
+[5] Save config      — write lab_config.json to disk
+[6] Save & Start     — save and launch the bridge
+[0] Exit
+```
+
+**Typical first-run flow:**
+
+1. Choose **3** to configure the server — pick a preset or enter your ORION server URL and bridge token.
+2. Choose **1** to add devices — select the device type (xArm, ABB, PLC, shell) and provide the IP address.
+3. Choose **6** to save the configuration and start the bridge.
+
+The bridge connects to all registered devices, then opens a WebSocket to the ORION server. The ORION AI can now see your devices and send commands.
+
+**Skip the menu with a saved config:**
+
+```bash
+orion --config lab_config.json
+```
+
+**Run without a server (local testing):**
+
+```bash
+orion --local
+```
+
+---
+
+## Configuration
+
+After the first run, configuration is saved to `lab_config.json` in your working directory. You can edit this file directly or regenerate it through the menu.
+
+Example `lab_config.json`:
+
+```json
+{
+  "server": "wss://your-orion-server/ws/robot",
+  "token": "your-bridge-token",
+  "lab_name": "My Lab",
+  "devices": [
+    { "type": "xarm", "id": "xarm-lab1", "ip": "192.168.1.185" },
+    { "type": "plc",  "id": "plc-station1", "ips": ["192.168.1.10"] }
+  ]
+}
+```
+
+Environment variables override the config file:
+
+| Variable | Purpose |
+|---|---|
+| `WS_SERVER_URL` | Override server WebSocket URL |
+| `WS_TOKEN` | Override bridge authentication token |
+| `BRIDGE_ID` | Override bridge identifier (default: `lab-bridge-01`) |
+
+---
+
+## MuJoCo Simulation
+
+Install with `pip install orion-bridge[mujoco]` to use the built-in MuJoCo physics simulation. xArm 6 and xArm 7 MJCF models are included in the package.
+
+```json
+{
+  "type": "xarm",
+  "id": "xarm7-sim",
+  "handler": "mujoco",
+  "viewer": true,
+  "mjcf_path": "sim/models/xarm7/scene_xarm7.xml"
+}
+```
+
+Set `"viewer": true` to open the MuJoCo interactive viewer window.
+
+---
+
+## Requirements
+
+- **Python**: 3.10 or higher
+- **OS**: Windows 10/11, macOS 12+, Linux (Ubuntu 20.04+)
+- **Network**: Access to device IPs and the ORION server endpoint
+
+Hardware-specific requirements are installed via optional extras (see Installation Options above).
+
+---
+
+## License
+
+MIT License — see [LICENSE](LICENSE) for details.
+
+---
+
+## Links
+
+- **ORION Platform**: [orion-learning.com](https://orion-learning.com)
+- **PyPI**: [pypi.org/project/orion-bridge](https://pypi.org/project/orion-bridge/)
+- **Issues**: [github.com/FILL_USER/orion-bridge/issues](https://github.com/FILL_USER/orion-bridge/issues)
+- **Source**: [github.com/FILL_USER/orion-bridge](https://github.com/FILL_USER/orion-bridge)

orion_bridge-2.0.0/README.md

@@ -0,0 +1,155 @@
+# ORION Bridge
+
+**Modular device bridge connecting industrial robotics hardware to the ORION AI agent platform.**
+
+[![PyPI version](https://badge.fury.io/py/orion-bridge.svg)](https://pypi.org/project/orion-bridge/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+
+---
+
+## What is ORION Bridge?
+
+ORION Bridge is a lightweight Python daemon that sits between physical (or simulated) lab equipment and the [ORION agent platform](https://orion-learning.com). It speaks the hardware protocols — xArm SDK, Snap7 for Siemens S7-1200 PLCs, ABB Robot Web Services — and exposes a unified WebSocket interface that the ORION AI understands.
+
+The bridge runs on any machine with network access to your devices: a lab PC next to the robot arm, a Raspberry Pi on the factory floor, or your laptop for simulation. Once running, ORION can read live telemetry, execute actions, and run multi-step routines through natural language — without any changes to your hardware configuration.
+
+Supported hardware backends out of the box: **UFACTORY xArm 6/7** (physical + MuJoCo simulation), **ABB robots** (via Robot Web Services), **Siemens S7-1200 PLCs** (via python-snap7), and a **local shell** for scripting and debugging. A ROS/Gazebo adapter is also available for simulation-heavy workflows.
+
+---
+
+## Quick Start
+
+```bash
+pip install orion-bridge[all]
+orion
+```
+
+The interactive menu guides you through device setup and server connection. No config file needed for the first run — it generates one for you.
+
+---
+
+## Installation Options
+
+Install only the drivers you need:
+
+| Command | Installs |
+|---|---|
+| `pip install orion-bridge` | Core only (WebSocket transport, CLI) |
+| `pip install orion-bridge[xarm]` | + UFACTORY xArm SDK |
+| `pip install orion-bridge[plc]` | + Siemens S7-1200 (python-snap7) |
+| `pip install orion-bridge[mujoco]` | + MuJoCo simulation engine |
+| `pip install orion-bridge[gazebo]` | + ROS/Gazebo (roslibpy) |
+| `pip install orion-bridge[http]` | + PLC HTTP sidecar API (Flask) |
+| `pip install orion-bridge[all]` | Everything above |
+
+---
+
+## First Run
+
+```
+$ orion
+```
+
+The CLI opens an interactive menu:
+
+```
+[1] Add device       — register an xArm, ABB, PLC, or shell device
+[2] Remove device    — remove a device from the list
+[3] Configure server — pick a lab preset or enter a custom server URL + token
+[4] Show status      — display current devices and connection info
+[5] Save config      — write lab_config.json to disk
+[6] Save & Start     — save and launch the bridge
+[0] Exit
+```
+
+**Typical first-run flow:**
+
+1. Choose **3** to configure the server — pick a preset or enter your ORION server URL and bridge token.
+2. Choose **1** to add devices — select the device type (xArm, ABB, PLC, shell) and provide the IP address.
+3. Choose **6** to save the configuration and start the bridge.
+
+The bridge connects to all registered devices, then opens a WebSocket to the ORION server. The ORION AI can now see your devices and send commands.
+
+**Skip the menu with a saved config:**
+
+```bash
+orion --config lab_config.json
+```
+
+**Run without a server (local testing):**
+
+```bash
+orion --local
+```
+
+---
+
+## Configuration
+
+After the first run, configuration is saved to `lab_config.json` in your working directory. You can edit this file directly or regenerate it through the menu.
+
+Example `lab_config.json`:
+
+```json
+{
+  "server": "wss://your-orion-server/ws/robot",
+  "token": "your-bridge-token",
+  "lab_name": "My Lab",
+  "devices": [
+    { "type": "xarm", "id": "xarm-lab1", "ip": "192.168.1.185" },
+    { "type": "plc",  "id": "plc-station1", "ips": ["192.168.1.10"] }
+  ]
+}
+```
+
+Environment variables override the config file:
+
+| Variable | Purpose |
+|---|---|
+| `WS_SERVER_URL` | Override server WebSocket URL |
+| `WS_TOKEN` | Override bridge authentication token |
+| `BRIDGE_ID` | Override bridge identifier (default: `lab-bridge-01`) |
+
+---
+
+## MuJoCo Simulation
+
+Install with `pip install orion-bridge[mujoco]` to use the built-in MuJoCo physics simulation. xArm 6 and xArm 7 MJCF models are included in the package.
+
+```json
+{
+  "type": "xarm",
+  "id": "xarm7-sim",
+  "handler": "mujoco",
+  "viewer": true,
+  "mjcf_path": "sim/models/xarm7/scene_xarm7.xml"
+}
+```
+
+Set `"viewer": true` to open the MuJoCo interactive viewer window.
+
+---
+
+## Requirements
+
+- **Python**: 3.10 or higher
+- **OS**: Windows 10/11, macOS 12+, Linux (Ubuntu 20.04+)
+- **Network**: Access to device IPs and the ORION server endpoint
+
+Hardware-specific requirements are installed via optional extras (see Installation Options above).
+
+---
+
+## License
+
+MIT License — see [LICENSE](LICENSE) for details.
+
+---
+
+## Links
+
+- **ORION Platform**: [orion-learning.com](https://orion-learning.com)
+- **PyPI**: [pypi.org/project/orion-bridge](https://pypi.org/project/orion-bridge/)
+- **Issues**: [github.com/FILL_USER/orion-bridge/issues](https://github.com/FILL_USER/orion-bridge/issues)
+- **Source**: [github.com/FILL_USER/orion-bridge](https://github.com/FILL_USER/orion-bridge)

orion_bridge-2.0.0/orion_bridge/__init__.py

@@ -0,0 +1,4 @@
+"""ORION Lab Bridge — Modular device bridge for industrial robotics."""
+
+__version__ = "2.0.0"
+__codename__ = "unified"

orion_bridge-2.0.0/orion_bridge/__main__.py

@@ -0,0 +1,3 @@
+"""Allow running as: python -m orion_bridge"""
+from orion_bridge.cli import main
+main()

orion_bridge-2.0.0/orion_bridge/cli.py

@@ -0,0 +1,296 @@
+"""
+orion_bridge/cli.py — Interactive CLI and bridge entry point.
+
+Extracted from bridge.py (Phase 3, Step 7).
+Connects all modules: config, ui, dispatcher, transport, logger, http.
+
+Usage:
+    orion                       # interactive menu
+    orion --local               # skip server, local only
+    orion --config lab.json     # autostart from saved config
+    python -m orion_bridge      # same as 'orion'
+"""
+
+import os
+import sys
+import time
+import asyncio
+import argparse
+import logging
+import threading
+
+from orion_bridge import __version__, __codename__
+from orion_bridge.config import (
+    LABS, DEVICE_TYPES, CONFIG_FILE,
+    load_config, save_config,
+)
+from orion_bridge.ui import (
+    RICH_UI,
+    print_header, print_devices, print_connection_panel,
+    print_menu, print_device_types, pick_lab, prompt,
+    msg_success, msg_error, msg_warning, msg_info,
+    print_bridge_active,
+)
+from orion_bridge.dispatcher import build_dispatcher
+from orion_bridge.logger import sb_logger
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(name)s — %(message)s",
+    datefmt="%H:%M:%S",
+)
+log = logging.getLogger("orion_bridge")
+
+try:
+    import roslibpy
+except ImportError:
+    roslibpy = None
+
+
+# ── Back exception for menu navigation ────────────────────────
+
+class _BackException(Exception):
+    pass
+
+
+def _prompt(label="", default=""):
+    """Prompt with 'back' support."""
+    value = prompt(label, default)
+    if value.lower() in ("back", "b"):
+        raise _BackException()
+    return value
+
+
+# ── PLC HTTP sidecar ──────────────────────────────────────────
+
+PLC_HTTP_PORT    = int(os.getenv("PLC_HTTP_PORT", "5000"))
+PLC_HTTP_ENABLED = os.getenv("PLC_HTTP_ENABLED", "true").lower() == "true"
+_plc_ref = None
+
+
+def _start_plc_http(dispatcher):
+    global _plc_ref
+    plc_handlers = dispatcher.handlers.get("plc", {})
+    if not plc_handlers:
+        log.info("PLC HTTP API: no PLCs, skipping")
+        return
+    _plc_ref = next(iter(plc_handlers.values()))
+    try:
+        from orion_bridge.http.plc_api import make_plc_app
+        app = make_plc_app(_plc_ref)
+        t = threading.Thread(
+            target=lambda: app.run(
+                host="0.0.0.0", port=PLC_HTTP_PORT,
+                debug=False, use_reloader=False,
+            ),
+            daemon=True,
+        )
+        t.start()
+        log.info(f"PLC HTTP API on :{PLC_HTTP_PORT}")
+    except ImportError:
+        log.warning("Flask not installed — PLC HTTP API off")
+    except Exception as e:
+        log.error(f"PLC HTTP API failed: {e}")
+
+
+# ── Menu actions ──────────────────────────────────────────────
+
+def _add_device(config):
+    try:
+        print_device_types(DEVICE_TYPES)
+        choice = _prompt("Select type")
+        dt = DEVICE_TYPES.get(choice)
+        if not dt:
+            msg_error("Invalid type.")
+            return
+
+        dtype = dt["type"]
+        name  = _prompt("Device name/ID (e.g. 'xarm-lab1')")
+        if not name:
+            msg_error("Name required.")
+            return
+
+        existing = [d.get("id") for d in config.get("devices", [])]
+        if name in existing:
+            msg_error(f"'{name}' already exists.")
+            return
+
+        dev = {"type": dtype, "id": name}
+
+        if dtype == "xarm":
+            dev["ip"] = _prompt("xArm IP", "192.168.1.185")
+        elif dtype == "abb":
+            dev["ip"]   = _prompt("ABB IP", "192.168.125.1")
+            dev["port"] = int(_prompt("Port", "1025"))
+        elif dtype == "plc":
+            msg_info("Enter PLC IPs one by one (empty to finish):")
+            ips = []
+            while True:
+                ip = _prompt("PLC IP")
+                if not ip:
+                    break
+                ips.append(ip)
+            if not ips:
+                msg_error("Need at least one IP.")
+                return
+            dev["ips"] = ips
+        elif dtype == "shell":
+            dev["ip"] = "localhost"
+
+        config.setdefault("devices", []).append(dev)
+        msg_success(f"Added {dtype} '{name}'")
+    except _BackException:
+        msg_info("Cancelled.")
+
+
+def _remove_device(config):
+    devices = config.get("devices", [])
+    if not devices:
+        msg_info("Nothing to remove.")
+        return
+    try:
+        print_devices(config)
+        idx = int(_prompt("Device # to remove (0=cancel)")) - 1
+        if idx < 0 or idx >= len(devices):
+            msg_info("Cancelled.")
+            return
+        removed = devices.pop(idx)
+        msg_success(f"Removed {removed['type']}/{removed['id']}")
+    except _BackException:
+        msg_info("Cancelled.")
+    except ValueError:
+        msg_error("Invalid number.")
+
+
+def _configure_server(config):
+    try:
+        lab = pick_lab(LABS)
+
+        if lab is None:
+            config.update({"server": None, "token": None, "lab_name": "Local"})
+            msg_success("Local mode")
+            return
+
+        config["server"]   = lab["server"]
+        config["token"]    = lab.get("token", "")
+        config["lab_name"] = lab.get("lab_name", lab.get("name", "Custom"))
+
+        if "devices" in lab:
+            config["devices"] = lab["devices"]
+            device_summary = ", ".join(
+                d.get("id", d.get("type", "?")) for d in lab["devices"]
+            )
+            msg_success(f"Lab: {config['lab_name']}")
+            msg_info(f"Devices: {device_summary}")
+        else:
+            msg_success(f"Lab: {config['lab_name']}")
+
+    except _BackException:
+        msg_info("Cancelled.")
+
+
+def _save(config):
+    save_config(config)
+    msg_success(f"Config saved → {CONFIG_FILE}")
+
+
+# ── Interactive menu ──────────────────────────────────────────
+
+def interactive_cli():
+    config = load_config()
+    print_header()
+    print_connection_panel(config)
+    print_devices(config)
+
+    while True:
+        print_menu()
+        choice = prompt()
+
+        if choice == "1":
+            _add_device(config)
+        elif choice == "2":
+            _remove_device(config)
+        elif choice == "3":
+            _configure_server(config)
+        elif choice == "4":
+            print_devices(config)
+            print_connection_panel(config)
+        elif choice == "5":
+            _save(config)
+        elif choice == "6":
+            _save(config)
+            return config
+        elif choice == "0":
+            save_q = prompt("Save before exit? (y/n)", "y")
+            if save_q != "n":
+                _save(config)
+            return None
+        elif choice.lower() in ("back", "b"):
+            msg_info("Already at main menu.")
+        else:
+            msg_error("Invalid option.")
+
+
+# ── Bridge startup ────────────────────────────────────────────
+
+def start_bridge(config, local=False):
+    if local:
+        config["server"] = None
+
+    dispatcher = build_dispatcher(config.get("devices", []), sb_logger=sb_logger)
+    dispatcher.connect_all()
+
+    if PLC_HTTP_ENABLED:
+        _start_plc_http(dispatcher)
+
+    status = dispatcher.get_status()
+    print_bridge_active(
+        config, status,
+        plc_http_port=PLC_HTTP_PORT if PLC_HTTP_ENABLED and _plc_ref else None,
+    )
+
+    try:
+        if config.get("server") and config.get("token"):
+            from orion_bridge.transport import ws_all_devices
+            asyncio.run(
+                ws_all_devices(
+                    dispatcher,
+                    config["server"],
+                    config["token"],
+                    config.get("devices", []),
+                )
+            )
+        else:
+            print("  Running local. Ctrl+C to stop.\n")
+            while True:
+                time.sleep(1)
+    except KeyboardInterrupt:
+        print("\n\nStopping...")
+    finally:
+        dispatcher.disconnect_all()
+        print("Done.")
+
+
+# ── Entry point ───────────────────────────────────────────────
+
+def main():
+    parser = argparse.ArgumentParser(description="ORION Lab Bridge")
+    parser.add_argument("--local",  action="store_true", help="No server, local only")
+    parser.add_argument("--config", type=str,            help="Config file (skip interactive menu)")
+    args = parser.parse_args()
+
+    if args.config:
+        import json
+        with open(args.config) as f:
+            config = json.load(f)
+        start_bridge(config, local=args.local)
+    else:
+        config = interactive_cli()
+        if config:
+            start_bridge(config, local=args.local)
+        else:
+            print("  Bye.")
+
+
+if __name__ == "__main__":
+    main()