weni-cli 0.1.0__tar.gz

weni_cli-0.1.0/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.1
+Name: weni-cli
+Version: 0.1.0
+Requires-Dist: Click
+Requires-Dist: requests
+Requires-Dist: flask
+Requires-Dist: waitress
+Requires-Dist: pyyaml
+Requires-Dist: python-slugify

weni_cli-0.1.0/README.md

@@ -0,0 +1 @@
+# Weni-CLI

weni_cli-0.1.0/pyproject.toml

@@ -0,0 +1,24 @@
+[tool.poetry]
+name = "weni-cli"
+version = "0.1.0"
+description = ""
+authors = ["Paulo Bernardo <paulo.bernardo@weni.ai>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.12"
+click = "^8.1.8"
+requests = "^2.32.3"
+flask = "^3.1.0"
+waitress = "^3.0.2"
+pyyaml = "^6.0.2"
+python-slugify = "^8.0.4"
+setuptools = "^75.7.0"
+twine = "^6.0.1"
+
+[tool.poetry.scripts]
+weni = "weni_cli.cli:cli"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

weni_cli-0.1.0/setup.cfg

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

weni_cli-0.1.0/setup.py

@@ -0,0 +1,21 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="weni-cli",
+    version="0.1.0",
+    packages=find_packages(),
+    include_package_data=True,
+    install_requires=[
+        "Click",
+        "requests",
+        "flask",
+        "waitress",
+        "pyyaml",
+        "python-slugify",
+    ],
+    entry_points={
+        "console_scripts": [
+            "weni = weni_cli.cli:cli",
+        ],
+    },
+)

weni_cli-0.1.0/weni_cli/auth.py

@@ -0,0 +1,52 @@
+import requests
+
+from weni_cli.store import (
+    STORE_KEYCLOAK_CLIENT_ID,
+    STORE_KEYCLOAK_REALM,
+    STORE_KEYCLOAK_URL,
+    Store,
+)
+
+DEFAULT_KEYCLOAK_URL = "https://accounts.weni.ai/auth"
+DEFAULT_KEYCLOAK_REALM = "weni"
+DEFAULT_KEYCLOAK_CLIENT_ID = "weni-cli"
+
+
+class Auth:
+    keycloak_url = None
+    realm = None
+    client_id = None
+    response_type = "code"
+    redirect_uri = "http://localhost:8081/sso-callback"
+
+    def __init__(self):
+        store = Store()
+        self.keycloak_url = store.get(STORE_KEYCLOAK_URL, DEFAULT_KEYCLOAK_URL)
+        self.realm = store.get(STORE_KEYCLOAK_REALM, DEFAULT_KEYCLOAK_REALM)
+        self.client_id = store.get(STORE_KEYCLOAK_CLIENT_ID, DEFAULT_KEYCLOAK_CLIENT_ID)
+
+    def get_login_url(self) -> str:
+        return f"{self.keycloak_url}/realms/{self.realm}/protocol/openid-connect/auth?client_id={self.client_id}&redirect_uri={self.redirect_uri}&response_type={self.response_type}"
+
+    def exchange_code(self, code) -> str:
+        token_url = (
+            f"{self.keycloak_url}/realms/{self.realm}/protocol/openid-connect/token"
+        )
+
+        data = {
+            "client_id": self.client_id,
+            "grant_type": "authorization_code",
+            "code": code,
+            "redirect_uri": self.redirect_uri,
+        }
+
+        response = requests.post(
+            token_url,
+            data=data,
+            headers={"Content-Type": "application/x-www-form-urlencoded"},
+        )
+
+        if response.status_code != 200:
+            return None
+
+        return response.json().get("access_token", None)

weni_cli-0.1.0/weni_cli/cli.py

@@ -0,0 +1,74 @@
+import click
+
+
+# Main CLI Group
+@click.group()
+def cli():
+    """Weni CLI"""
+
+
+# Login Command
+@cli.command("login")
+def login():
+    """Login with your Weni account"""
+    from weni_cli.commands.login import LoginHandler
+
+    LoginHandler().execute()
+
+
+# Nested CLI Project Group
+@cli.group()
+def project():
+    """Project commands"""
+
+
+# Project Group Commands
+@project.command("list")
+@click.option("--org", "-o", help="Filter by organization", type=click.UUID)
+def list_projects(org):
+    """List projects"""
+    from weni_cli.commands.project_list import ProjectListHandler
+
+    ProjectListHandler().execute(org_uuid=org)
+
+
+@project.command("use")
+@click.argument("project_uuid", required=True)
+def use_project(project_uuid):
+    """Set the project to be used in the CLI
+
+    PROJECT_UUID: The UUID of the project to be used
+    """
+    from weni_cli.commands.project_use import ProjectUseHandler
+
+    ProjectUseHandler().execute(project_uuid=project_uuid)
+
+
+@project.command("current")
+def current_project():
+    """Show current selected project"""
+    from weni_cli.commands.project_current import ProjectCurrentHandler
+
+    ProjectCurrentHandler().execute()
+
+
+@project.command("push")
+@click.argument(
+    "definition", required=True, type=click.Path(exists=True, dir_okay=False)
+)
+@click.option("--force-update", is_flag=True, help="Force update to the project")
+def push_project(definition, force_update):
+    """Push an Agent definition to the current project
+
+    DEFINITION: The path to the YAML agent definition file
+    """
+    from weni_cli.commands.project_push import ProjectPushHandler
+
+    try:
+        ProjectPushHandler().execute(definition=definition, force_update=force_update)
+    except Exception as e:
+        click.echo(f"Error: {e}")
+
+
+if __name__ == "__main__":
+    cli()

weni_cli-0.1.0/weni_cli/handler.py

@@ -0,0 +1,10 @@
+import click
+
+
+class Handler:
+    def execute(self, **kwargs):
+        raise NotImplementedError()
+
+    def exit(self, error=None):
+        if error:
+            click.echo(f"An error occurred: {error}")

weni_cli-0.1.0/weni_cli/store.py

@@ -0,0 +1,46 @@
+import click
+import os
+import json
+
+from pathlib import Path
+
+STORE_TOKEN_KEY = "token"
+STORE_PROJECT_UUID_KEY = "project_uuid"
+STORE_WENI_BASE_URL = "weni_base_url"
+STORE_NEXUS_BASE_URL = "nexus_base_url"
+STORE_KEYCLOAK_URL = "keycloak_url"
+STORE_KEYCLOAK_REALM = "keycloak_realm"
+STORE_KEYCLOAK_CLIENT_ID = "keycloak_client_id"
+
+
+class Store:
+    file_path = f"{Path.home()}{os.sep}.weni_cli"
+
+    # Validates that the file exists, if it does not exist, it creates it with an empty dictionary
+    def __init__(self):
+        with click.open_file(self.file_path, "a+") as file:
+            file.seek(0)
+            content = file.read()
+            if not content:
+                file.write("{}")
+
+            file.close()
+
+    def get(self, key, default=None):
+        with click.open_file(self.file_path, "r") as file:
+            content = json.loads(file.read())
+            file.close()
+            return content.get(key, default)
+
+    def set(self, key, value):
+        content = {}
+        with click.open_file(self.file_path, "r") as file:
+            content = json.loads(file.read())
+            file.close()
+
+        with click.open_file(self.file_path, "w") as file:
+            content[key] = value
+            file.write(json.dumps(content))
+            file.close()
+
+        return True

weni_cli-0.1.0/weni_cli/wsgi.py

@@ -0,0 +1,33 @@
+import click
+
+from flask import Flask, request
+from multiprocessing import Process, Pipe
+
+app = Flask(__name__)
+server_process = None
+auth_parent_conn, auth_child_conn = Pipe()
+
+
+@app.route("/sso-callback", methods=["GET"])
+def sso_callback():
+    auth_child_conn.send(request.args.get("code"))
+    return "Successfully logged in, you can close this window now"
+
+
+def serve():
+    from waitress import serve
+
+    global server_process
+
+    server_process = Process(
+        target=serve,
+        kwargs={"app": app, "host": "0.0.0.0", "port": 8081, "_quiet": True},
+    )
+    server_process.start()
+
+
+def shutdown():
+    global server_process
+    if server_process is not None:
+        server_process.terminate()
+        server_process = None

weni_cli-0.1.0/weni_cli.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.1
+Name: weni-cli
+Version: 0.1.0
+Requires-Dist: Click
+Requires-Dist: requests
+Requires-Dist: flask
+Requires-Dist: waitress
+Requires-Dist: pyyaml
+Requires-Dist: python-slugify

weni_cli-0.1.0/weni_cli.egg-info/SOURCES.txt

@@ -0,0 +1,16 @@
+README.md
+pyproject.toml
+setup.py
+tests/__init__.py
+weni_cli/__init__.py
+weni_cli/auth.py
+weni_cli/cli.py
+weni_cli/handler.py
+weni_cli/store.py
+weni_cli/wsgi.py
+weni_cli.egg-info/PKG-INFO
+weni_cli.egg-info/SOURCES.txt
+weni_cli.egg-info/dependency_links.txt
+weni_cli.egg-info/entry_points.txt
+weni_cli.egg-info/requires.txt
+weni_cli.egg-info/top_level.txt

weni_cli-0.1.0/weni_cli.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

weni_cli-0.1.0/weni_cli.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+weni = weni_cli.cli:cli

weni_cli-0.1.0/weni_cli.egg-info/requires.txt

@@ -0,0 +1,6 @@
+Click
+requests
+flask
+waitress
+pyyaml
+python-slugify

weni_cli-0.1.0/weni_cli.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+tests
+weni_cli