add code

Author: wulfmann

.gitignore

@@ -1,3 +1,16 @@
+# IDE
 .vscode
+
+# misc
 .DS_Store
+
+# test
 .coverage
+
+# python
+*.pyc
+*.egg-info
+__pycache__
+
+# poetry
+dist

README.md

@@ -1,11 +1,67 @@
-# AWS coco
+# AWS coco (console container)
 
 This tool allows you to manage AWS Console Sessions with Firefox Containers
 
+## Quickstart
+
+Ensure you've  met the [requirements](#requirements).
+
+```bash
+$ pip install aws_coco
+```
+
+Usage
+
+```bash
+$ coco -c green -i fingerprint
+```
+
+You should now have a new browser tab with your aws session!
+
+Continue reading for a more in-depth walkthrough of the setup.
+
 ## Requirements
 
 - [Firefox](https://www.mozilla.org/en-US/firefox/new/)
+- [Firefox Multi-Account Containers](https://addons.mozilla.org/en-US/firefox/addon/multi-account-containers/)
 - [Open URL in Container Extension](https://addons.mozilla.org/en-US/firefox/addon/open-url-in-container/)
 - [Python >= 3.7](http://python.org/)
 
 If you don't wish to install the extension through the marketplace, the source for the extension can be found [here](https://github.com/honsiorovskyi/open-url-in-container).
+
+## Installation
+
+```bash
+$ pip install aws_coco
+```
+
+## Usage
+
+```bash
+$ coco 
+```
+
+## Development
+
+```bash
+$ git clone https://github.com/wulfmann/aws-coco.git
+$ git clone git@github.com:wulfmann/aws-coco.git
+```
+
+Install Dependencies
+
+```bash
+$ poetry install
+```
+
+Run the command
+
+```bash
+$ poetry run coco -c green -i fingerprint
+```
+
+Run tests
+
+```bash
+$ poetry run pytest
+```

aws_coco/__init__.py

@@ -0,0 +1,49 @@
+import sys
+import json
+import urllib
+
+import requests
+
+from aws_coco.lib.utils import dict_to_querystring
+
+
+def create_signin_token(session):
+    credentials = session.get_credentials()
+
+    if not credentials.token:
+        print(
+            f"coco does not work without a session token. make sure you are running this from within a temporary session"
+        )
+        sys.exit(1)
+
+    session_parameters = {
+        "sessionId": credentials.access_key,
+        "sessionKey": credentials.secret_key,
+        "sessionToken": credentials.token,
+    }
+
+    response = requests.get(
+        "https://signin.aws.amazon.com/federation",
+        params={"Action": "getSigninToken", "Session": json.dumps(session_parameters)},
+    )
+
+    return response.json().get("SigninToken")
+
+
+def create_console_url(session, destination=None):
+    if destination is None:
+        destination = "https://console.aws.amazon.com/"
+
+    signin_token = create_signin_token(session)
+
+    base = "https://signin.aws.amazon.com/federation"
+    qs = dict_to_querystring(
+        {
+            "Action": "login",
+            "Issuer": "aws-coco",
+            "Destination": urllib.parse.quote_plus(destination),
+            "SigninToken": signin_token,
+        }
+    )
+
+    return f"{base}?{qs}"

aws_coco/lib/__init__.py

@@ -0,0 +1,25 @@
+import urllib
+
+# https://github.com/mozilla/multi-account-containers/blob/00a1ce9dca2c4f54c9622515a7a86b56442eb9f2/src/js/popup.js#L1302
+colors = ["blue", "turquoise", "green", "yellow", "orange", "red", "pink", "purple"]
+
+# https://github.com/mozilla/multi-account-containers/blob/00a1ce9dca2c4f54c9622515a7a86b56442eb9f2/src/js/popup.js#L1316
+icons = [
+    "fingerprint",
+    "briefcase",
+    "dollar",
+    "cart",
+    "vacation",
+    "gift",
+    "food",
+    "fruit",
+    "pet",
+    "tree",
+    "chill",
+    "circle",
+    "fence",
+]
+
+
+def format_for_container(url, name, color, icon):
+    return f"ext+container:name={name}&color={color}&icon={icon}&url={urllib.parse.quote_plus(url)}"

aws_coco/lib/aws.py

@@ -0,0 +1,13 @@
+import argparse
+
+
+class NegateAction(argparse.Action):
+    def __call__(self, parser, ns, values, option):
+        setattr(ns, self.dest, option[2:4] != "no")
+
+
+def dict_to_querystring(params):
+    qs = []
+    for key in params:
+        qs.append(f"{key}={params[key]}")
+    return "&".join(qs)

aws_coco/lib/firefox.py

@@ -0,0 +1,94 @@
+import argparse
+import webbrowser
+
+import boto3
+
+from aws_coco.lib.firefox import colors
+from aws_coco.lib.firefox import icons
+from aws_coco.lib.firefox import format_for_container
+from aws_coco.lib.aws import create_console_url
+from aws_coco.lib.utils import NegateAction
+
+
+def construct_parser():
+    parser = argparse.ArgumentParser(
+        description="A utility for managing AWS Console Sessions"
+    )
+
+    parser.add_argument("--profile", "-p", help="The AWS profile to use", default=None)
+
+    parser.add_argument(
+        "--container",
+        "--no-container",
+        dest="container",
+        action=NegateAction,
+        help="Open URL in Firefox container",
+        default=True,
+        nargs=0,
+    )
+
+    parser.add_argument(
+        "--open",
+        "--no-open",
+        dest="open",
+        action=NegateAction,
+        help="Open URL automatically",
+        default=True,
+        nargs=0,
+    )
+
+    parser.add_argument("--name", "-n", help="The container tab name")
+    parser.add_argument("--color", "-c", choices=colors, help="The container tab color")
+    parser.add_argument("--icon", "-i", choices=icons, help="The container tab icon")
+    parser.add_argument(
+        "--destination",
+        "-d",
+        choices=icons,
+        help="The destination URL in the AWS console",
+        default=None,
+    )
+
+    return parser
+
+
+def validate_args(args):
+    if args.container:
+        if not args.color:
+            raise Exception(f"--color is required if --container is passed")
+        if not args.icon:
+            raise Exception(f"--icon is required if --container is passed")
+
+
+def main(args):
+    validate_args(args)
+
+    session = boto3.session.Session(profile_name=args.profile)
+    url = create_console_url(session, args.destination)
+
+    if args.container:
+        firefox = webbrowser.get("firefox")
+
+        if firefox is None:
+            raise Exception(f"firefox is required if --container is passed")
+
+        container_name = args.profile
+        if args.name:
+            container_name = args.name
+
+        url = format_for_container(url, container_name, args.color, args.icon)
+
+        if args.open:
+            firefox.open_new_tab(url)
+        else:
+            print(url)
+    else:
+        if args.open:
+            webbrowser.open_new(url)
+        else:
+            print(url)
+
+
+def run():
+    parser = construct_parser()
+    args = parser.parse_args()
+    main(args)