agent-uplink 0.0.1__tar.gz

agent_uplink-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Huw
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

agent_uplink-0.0.1/PKG-INFO

@@ -0,0 +1,38 @@
+Metadata-Version: 2.4
+Name: agent-uplink
+Version: 0.0.1
+Summary: Claude Sandbox
+Author: Huw
+License: MIT License
+        
+        Copyright (c) 2024 Huw
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Project-URL: Homepage, https://github.com/hoo29/agent-uplink
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Provides-Extra: tests
+Requires-Dist: pytest; extra == "tests"
+Provides-Extra: build
+Requires-Dist: build; extra == "build"
+Dynamic: license-file
+
+# agent-uplink

agent_uplink-0.0.1/README.md

@@ -0,0 +1 @@
+# agent-uplink

agent_uplink-0.0.1/agent_uplink/__main__.py

@@ -0,0 +1,115 @@
+import argparse
+import logging
+import signal
+import json
+import os
+import uuid
+
+
+from pathlib import Path
+
+from .utils import SESSION_DIRS, create_aws_profile_file_contents, get_free_port, run_command, run_command_background, shutdown_handler
+
+
+LOGGER = logging.getLogger("agent-uplink")
+
+
+def get_bedrock_aws_profile_name() -> str:
+    with open(f"{Path.home()}/.claude/settings.json", "r", encoding="utf8") as f:
+        settings = json.loads(f.read())
+        name = settings.get("env", {}).get("AWS_PROFILE")
+        if not name:
+            raise RuntimeError("AWS_PROFILE not found in claude settings env")
+        return name
+
+
+def _main(socket_dir, mitm_dir):
+    parser = argparse.ArgumentParser(
+        description="Trust is a weakness", formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser.add_argument("-a", "--aws-profiles", type=str, nargs="*",
+                        action="extend", default=[],
+                        help="AWS profiles to provide credentials for")
+    parser.add_argument("-m", "--mitmproxy-image", type=str,
+                        default="mitmproxy/mitmproxy")
+    parser.add_argument("-c", "--claude-image", type=str,
+                        default="why")
+    args = parser.parse_args()
+    logging.basicConfig(level=logging.DEBUG, format="%(message)s")
+    LOGGER.info("hi")
+
+    bedrock_aws_profile_name = get_bedrock_aws_profile_name()
+    aws_profile_names = args.aws_profiles
+    aws_profile_names.append(bedrock_aws_profile_name)
+    aws_profile_file_contents = create_aws_profile_file_contents(
+        aws_profile_names)
+
+    free_port = get_free_port()
+
+    socket_path = socket_dir / "uplink.sock"
+    run_command_background(
+        f"socat UNIX-LISTEN:{socket_path},fork,mode=666 TCP:127.0.0.1:{free_port}")
+
+    mitmproxy_image = args.mitmproxy_image
+    mitm_container_command = " ".join([
+        "docker run",
+        "--cap-add=DAC_OVERRIDE",
+        "--cap-add=SETGID",
+        "--cap-add=SETUID",
+        "--cap-drop=ALL",
+        "--cpus 1",
+        "--ipc private",
+        "--memory=0.5g",
+        "--pids-limit 100",
+        "--read-only",
+        "--rm",
+        "--security-opt no-new-privileges",
+        f"-v {mitm_dir}:/home/mitmproxy/.mitmproxy",
+        f"-p {free_port}:8080",
+        mitmproxy_image,
+        "mitmdump",
+    ])
+    run_command_background(mitm_container_command)
+
+    claude_image = args.claude_image
+    claude_container_command = " ".join([
+        "docker run",
+        "--cap-drop=ALL",
+        "--cpus 1",
+        "--ipc private",
+        "--memory=0.5g",
+        "--network none",
+        "--pids-limit 100",
+        "--read-only",
+        "--rm",
+        "--security-opt no-new-privileges",
+        "--tmpfs /tmp:rw,noexec,nosuid,size=200m",
+        "-it",
+        f"-v {socket_path}:/mounts/socket/uplink.sock",
+        f"-v {mitm_dir}:/mounts/certs:ro",
+        claude_image,
+    ])
+    run_command(claude_container_command, interactive=True)
+
+    idk = 1
+
+
+def main():
+    script_dir = Path.home() / ".agent_uplink" / uuid.uuid4().hex
+    SESSION_DIRS.append(script_dir)
+    socket_dir = script_dir / "sockets"
+    mitm_dir = script_dir / "mitm"
+    for dir in [socket_dir, mitm_dir]:
+        os.makedirs(dir, exist_ok=True)
+
+    signal.signal(signal.SIGINT, shutdown_handler)
+    signal.signal(signal.SIGTERM, shutdown_handler)
+    try:
+        _main(socket_dir, mitm_dir)
+    except Exception:
+        LOGGER.fatal("oh no", exc_info=True)
+    finally:
+        shutdown_handler(None, None)
+
+
+if __name__ == "__main__":
+    main()

agent_uplink-0.0.1/agent_uplink.egg-info/PKG-INFO

@@ -0,0 +1,38 @@
+Metadata-Version: 2.4
+Name: agent-uplink
+Version: 0.0.1
+Summary: Claude Sandbox
+Author: Huw
+License: MIT License
+        
+        Copyright (c) 2024 Huw
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Project-URL: Homepage, https://github.com/hoo29/agent-uplink
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Provides-Extra: tests
+Requires-Dist: pytest; extra == "tests"
+Provides-Extra: build
+Requires-Dist: build; extra == "build"
+Dynamic: license-file
+
+# agent-uplink

agent_uplink-0.0.1/agent_uplink.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+LICENSE
+README.md
+pyproject.toml
+agent_uplink/__init__.py
+agent_uplink/__main__.py
+agent_uplink.egg-info/PKG-INFO
+agent_uplink.egg-info/SOURCES.txt
+agent_uplink.egg-info/dependency_links.txt
+agent_uplink.egg-info/entry_points.txt
+agent_uplink.egg-info/requires.txt
+agent_uplink.egg-info/top_level.txt

agent_uplink-0.0.1/agent_uplink.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

agent_uplink-0.0.1/agent_uplink.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+agent-uplink = agent_uplink.__main__:main

agent_uplink-0.0.1/agent_uplink.egg-info/requires.txt

@@ -0,0 +1,6 @@
+
+[build]
+build
+
+[tests]
+pytest

agent_uplink-0.0.1/agent_uplink.egg-info/top_level.txt

@@ -0,0 +1 @@
+agent_uplink

agent_uplink-0.0.1/pyproject.toml

@@ -0,0 +1,27 @@
+[build-system]
+requires = ["setuptools>=61.0.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "agent-uplink"
+version = "0.0.1"
+description = "Claude Sandbox"
+readme = "README.md"
+authors = [{ name = "Huw" }]
+license = { file = "LICENSE" }
+
+dependencies = []
+requires-python = ">=3.9"
+
+[project.urls]
+Homepage = "https://github.com/hoo29/agent-uplink"
+
+[project.scripts]
+agent-uplink = "agent_uplink.__main__:main"
+
+[project.optional-dependencies]
+tests = ["pytest"]
+build = ["build"]
+
+[tool.setuptools]
+packages = ["agent_uplink"]

agent_uplink-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+