initial config parsing

Author: Wh1t3Fox

MANIFEST.in

@@ -0,0 +1,3 @@
+include rete/VERSION
+include rete/config/*.yml
+global-exclude *.py[cod] __pycache__ *.so

rete/VERSION

@@ -0,0 +1 @@
+1.0.0

rete/__init__.py

@@ -0,0 +1,49 @@
+#!/usr/bin/python3
+# -*- coding:utf-8 -*-
+
+import coloredlogs
+import logging
+import logging.config
+import elevate
+import os
+
+REPO_NAME = "retenet"
+
+with open(os.path.join(os.path.dirname(__file__), "VERSION")) as fr:
+    VERSION = fr.read().strip()
+
+try:
+    USER_CONFIG_PATH = f"{os.environ['XDG_CONFIG_HOME']}/rete"
+except KeyError:
+    USER_CONFIG_PATH = f"{os.environ['HOME']}/.config/rete"
+if not os.path.exists(USER_CONFIG_PATH):
+    os.makedirs(USER_CONFIG_PATH)
+
+try:
+    LOG_PATH = f"{os.environ['XDG_DATA_HOME']}/rete"
+except KeyError:
+    LOG_PATH = f"{os.environ['HOME']}/.local/share/rete"
+if not os.path.exists(LOG_PATH):
+    os.makedirs(LOG_PATH)
+
+BROWSERS = ["brave", "chromium", "firefox", "opera", "tbb"]
+
+logging.config.dictConfig(
+    {
+        "version": 1,
+        "disable_existing_loggers": False,
+        "formatters": {"standard": {"format": "%(message)s"},},
+        "handlers": {
+            "screen": {"level": "INFO", "class": "logging.StreamHandler",},
+            "file": {
+                "level": "DEBUG",
+                "class": "logging.FileHandler",
+                "filename": f"{LOG_PATH}/rete.log",
+            },
+        },
+        "loggers": {
+            "": {"handlers": ["screen", "file"], "level": "DEBUG", "propagate": True}
+        },
+    }
+)
+coloredlogs.install()

rete/__main__.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+from rete.cli import main
+
+if __name__ == "__main__":
+    main()

rete/cli.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+import argparse
+import logging
+import elevate
+import docker
+import yaml
+import grp
+import os
+
+from rete import BROWSERS, USER_CONFIG_PATH, VERSION
+from rete.utils import parse_config, run_container, pull_image
+
+logger = logging.getLogger(__name__)
+
+
+def get_args():
+    cfg = parse_config()
+
+    parser = argparse.ArgumentParser(f"rete version {VERSION}")
+    parser.add_argument(
+        "browser",
+        nargs="?",
+        choices=BROWSERS,
+        default=cfg["browser"]["name"],
+        help="Supported Browsers",
+    )
+
+    parser.add_argument(
+        "-p", "--profile", default="", required=False, help="Profile Name"
+    )
+    parser.add_argument("-t", action="store_true", help="Temporary Profile")
+    parser.add_argument("--vpn", required=False, help="Use a VPN")
+    parser.add_argument("--proxy", required=False, help="PROTO://IP:PORT")
+
+    group = parser.add_mutually_exclusive_group()
+    group.add_argument("--config", required=False, help="Open Config for Editing")
+    group.add_argument("--rm", required=False, help="Stop and Remove ALL Browsers")
+    group.add_argument("--update", required=False, help="Check for Upates")
+
+    args = parser.parse_args()
+
+    if not args.browser:
+        logger.error("No Browser Specified")
+        exit(1)
+
+    return args
+
+
+def main():
+
+    args = get_args()
+
+    user_grps = [g.gr_name for g in grp.getgrall() if os.environ["USER"] in g.gr_mem]
+    if "docker" not in user_grps:
+        elevate.elevate(graphical=False)
+
+    client = docker.from_env()
+    # pull_image(client, args.browser)
+
+
+if __name__ == "__main__":
+    main()

rete/config/rete.yml

@@ -0,0 +1,13 @@
+browser:
+  name: firefox
+  proxy:
+  dns:
+    host:
+    doh: false
+
+profile:
+  default: personal
+  list: [htb, lan, media, personal, shopping, work]
+
+vpn:
+  ovpn: 

rete/config/rete_schema.yml

@@ -0,0 +1,46 @@
+type: object
+properties:
+  browser:
+    type: object
+    required: 
+      - name
+    properties:
+      name:
+        type: string
+        enum: 
+          - brave
+          - chromium
+          - firefox
+          - opeara
+          - tbb
+      proxy:
+        type: 
+          - string
+          - 'null'
+        pattern: '^([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-]{0,61}[a-zA-Z0-9])(\.([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-]{0,61}[a-zA-Z0-9]))*$'
+      dns:
+        type: object
+        properties:
+          host:
+            type: 
+              - string
+              - 'null'
+          doh:
+            type: boolean
+  profile:
+    type: object
+    properties:
+      default:
+        type: string
+      list: 
+        type: array
+        items:
+          type: string
+  vpn:
+    type: object
+    properties:
+      ovpn:
+        type: 
+          - string
+          - 'null'
+  

rete/utils.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+import jsonschema
+import logging
+import docker
+import shutil
+import yaml
+import os
+
+from rete import REPO_NAME, USER_CONFIG_PATH
+
+logger = logging.getLogger(__name__)
+
+
+def parse_config():
+    config_path = f"{USER_CONFIG_PATH}/rete.yml"
+    if not os.path.exists(config_path):
+        shutil.copy(f"{os.path.dirname(__file__)}/config/rete.yml", config_path)
+
+    with open(f"{USER_CONFIG_PATH}/rete.yml") as fr, open(
+        f"{os.path.dirname(__file__)}/config/rete_schema.yml"
+    ) as fr2:
+        try:
+            cfg = yaml.safe_load(fr)
+            jsonschema.validate(cfg, yaml.safe_load(fr2))
+        except jsonschema.exceptions.ValidationError as e:
+            if "Failed validating 'pattern'" in str(e):
+                logger.error(f"Invalid Proxy in Config File: {USER_CONFIG_PATH}/rete.yml")
+            else:
+                logger.error(f"Invalid Option in Config File: {USER_CONFIG_PATH}/rete.yml")
+                logger.error(e)
+            exit(1)
+
+    return cfg
+
+
+def run_container(client, browser, detach=False):
+    cntr = client.containers.run(f"{REPO_NAME}/{browser}", detach=detach)
+    logger.info(cntr)
+
+
+def pull_image(client, browser):
+    cntr = client.images.pull(f"{REPO_NAME}/{browser}")
+    logger.info(cntr)

setup.cfg

@@ -0,0 +1,3 @@
+[metadata]
+description-file = README.md
+long_description_content_type = text/markdown

setup.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+import os
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+with open(os.path.join("rete", "VERSION"), "r", encoding="utf-8") as f:
+    version = f.read()
+
+with open("README.md", "r", encoding="utf-8") as f:
+    long_description = f.read()
+
+setup(
+    name="rete",
+    version=version,
+    description="",
+    license="GPLv3",
+    long_description=long_description,
+    author="retenet",
+    author_email="[email protected]",
+    classifiers=[
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: End Users/Desktop",
+        "License :: OSI Approved :: MIT License",
+        "Programming Language :: Python :: 3",
+        "Operating System :: POSIX :: Linux",
+        "Environment :: Console",
+    ],
+    keywords="browser",
+    packages=find_packages(exclude=["browsers", "contrib", "docs", "tests"]),
+    install_requires=[
+        "colorama==0.4.3",
+        "coloredlogs==14.0",
+        "docker==4.2.2",
+        "elevate==0.1.3",
+        "jsonschema==3.2.0",
+        "PyYAML==5.3.1",
+    ],
+    use_scm_version=True,
+    setup_required=["setuptools_scm"],
+    extras_require={"dev": ["black==19.10b0", "ipython==7.13.0", "pytest==4.6.7"],},
+    entry_points={"console_scripts": ["rete = rete.cli:main",],},
+    include_package_data=True,
+)