PyPI - tinybird - Versions diffs - 0.0.1.dev12__tar.gz → 0.0.1.dev13__tar.gz - Mend

tinybird 0.0.1.dev12tar.gz → 0.0.1.dev13tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (92) hide show

{tinybird-0.0.1.dev12 → tinybird-0.0.1.dev13}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev12
+Version: 0.0.1.dev13
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev12 → tinybird-0.0.1.dev13}/tinybird/tb/cli.py RENAMED Viewed

@@ -14,6 +14,7 @@ import tinybird.tb.modules.create
 import tinybird.tb.modules.datasource
 import tinybird.tb.modules.fmt
 import tinybird.tb.modules.job
+import tinybird.tb.modules.login
 import tinybird.tb.modules.mock
 import tinybird.tb.modules.pipe
 import tinybird.tb.modules.tag

tinybird-0.0.1.dev13/tinybird/tb/modules/login.py ADDED Viewed

@@ -0,0 +1,166 @@
+import http.server
+import socketserver
+import threading
+import time
+import urllib.parse
+import webbrowser
+from urllib.parse import urlencode
+import click
+import requests
+from tinybird.feedback_manager import FeedbackManager
+from tinybird.tb.modules.cli import CLIConfig, cli
+class AuthHandler(http.server.SimpleHTTPRequestHandler):
+    def do_GET(self):
+        # The access_token is in the URL fragment, which is not sent to the server
+        # We'll send a small HTML page that extracts the token and sends it back to the server
+        self.send_response(200)
+        self.send_header("Content-type", "text/html")
+        self.end_headers()
+        self.wfile.write(b"""
+        <html>
+        <head>
+            <style>
+                body {
+                    font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
+                    background: #f5f5f5;
+                    display: flex;
+                    align-items: center;
+                    justify-content: center;
+                    height: 100vh;
+                    margin: 0;
+                }
+                .message {
+                    background: white;
+                    padding: 2rem 3rem;
+                    border-radius: 8px;
+                    box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+                    text-align: center;
+                    display: flex;
+                    flex-direction: column;
+                    align-items: center;
+                }
+                h1 {
+                    color: #25283D;
+                    font-size: 1.2rem;
+                    margin: 0;
+                    font-weight: 500;
+                }
+            </style>
+        </head>
+        <body>
+            <div class="message">
+                <h1>Authenticating...</h1>
+            </div>
+            <script>
+                var hash = window.location.hash.substr(1);
+                var access_token = new URLSearchParams(hash).get('access_token');
+                window.history.pushState({}, '', '/');
+                fetch('/?token=' + access_token, {method: 'POST'})
+                    .then(() => {
+                        document.querySelector('.message h1').textContent = 'Authentication successful! You can close this window.';
+                    });
+            </script>
+        </body>
+        </html>
+        """)
+    def do_POST(self):
+        parsed_path = urllib.parse.urlparse(self.path)
+        query_params = urllib.parse.parse_qs(parsed_path.query)
+        if "token" in query_params:
+            token = query_params["token"][0]
+            self.server.auth_callback(token)
+            self.send_response(200)
+            self.end_headers()
+        else:
+            self.send_error(400, "Missing 'token' parameter")
+        self.server.shutdown()
+    def log_message(self, format, *args):
+        # Suppress log messages
+        return
+class AuthServer(socketserver.TCPServer):
+    allow_reuse_address = True
+    def __init__(self, server_address, RequestHandlerClass, auth_callback):
+        self.auth_callback = auth_callback
+        super().__init__(server_address, RequestHandlerClass)
+def start_server(auth_callback):
+    with AuthServer(("", 3000), AuthHandler, auth_callback) as httpd:
+        httpd.timeout = 30
+        start_time = time.time()
+        while time.time() - start_time < 60:  # Run for a maximum of 60 seconds
+            httpd.handle_request()
+@cli.command()
+@click.option(
+    "--host",
+    help="Set custom host if it's different than https://api.tinybird.co. Check https://www.tinybird.co/docs/api-reference/overview#regions-and-endpoints for the available list of regions",
+)
+@click.option(
+    "--workspace",
+    help="Set the workspace to authenticate to. If not set, the default workspace will be used.",
+)
+def login(host: str, workspace: str):
+    """Authenticate via browser."""
+    auth_event = threading.Event()
+    auth_code = [None]  # Using a list to store the code, as it's mutable
+    config = CLIConfig.get_project_config()
+    host = host or "https://api.tinybird.co"
+    def auth_callback(code):
+        auth_code[0] = code
+        auth_event.set()
+    click.echo("Opening browser for authentication...")
+    # Start the local server in a separate thread
+    server_thread = threading.Thread(target=start_server, args=(auth_callback,))
+    server_thread.daemon = True
+    server_thread.start()
+    # Open the browser to the auth page
+    client_id = "T6excMo8IKguvUw4vFNYfqlt9pe6msCU"
+    callback_url = "http://localhost:3000"
+    params = {
+        "client_id": client_id,
+        "redirect_uri": callback_url,
+        "response_type": "token",
+        "scope": "openid profile email",
+    }
+    auth_url = f"https://auth.tinybird.co/authorize?{urlencode(params)}"
+    webbrowser.open(auth_url)
+    # Wait for the authentication to complete or timeout
+    if auth_event.wait(timeout=60):  # Wait for up to 60 seconds
+        params = {}
+        if workspace:
+            params["workspace_id"] = workspace
+        response = requests.get(
+            f"https://api.tinybird.co/v0/user/tokens?{urlencode(params)}",
+            headers={"Authorization": f"Bearer {auth_code[0]}"},
+        )
+        data = response.json()
+        cli_config = CLIConfig.get_project_config()
+        workspace_token = data["workspace_token"]
+        user_token = data["user_token"]
+        cli_config.set_token(workspace_token)
+        cli_config.set_token_for_host(workspace_token, host)
+        cli_config.set_user_token(user_token)
+        config.set_host(host)
+        cli_config.persist_to_file()
+        click.echo(FeedbackManager.success(message="✓ Authentication successful!"))
+    else:
+        click.echo(FeedbackManager.error(message="Authentication failed or timed out."))

{tinybird-0.0.1.dev12 → tinybird-0.0.1.dev13}/tinybird.egg-info/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev12
+Version: 0.0.1.dev13
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev12 → tinybird-0.0.1.dev13}/tinybird.egg-info/SOURCES.txt RENAMED Viewed

@@ -39,6 +39,7 @@ tinybird/tb/modules/fmt.py
 tinybird/tb/modules/job.py
 tinybird/tb/modules/llm.py
 tinybird/tb/modules/local.py
+tinybird/tb/modules/login.py
 tinybird/tb/modules/mock.py
 tinybird/tb/modules/pipe.py
 tinybird/tb/modules/prompts.py