tinybird 0.0.1.dev304__tar.gz → 0.0.1.dev305__tar.gz

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev304
+Version: 0.0.1.dev305
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -39,6 +39,7 @@ Requires-Dist: packaging<24,>=23.1
 Requires-Dist: llm>=0.19
 Requires-Dist: thefuzz==0.22.1
 Requires-Dist: python-dotenv==1.1.0
+Requires-Dist: pyjwt[crypto]==2.9.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/tinybird/sql_template.py

@@ -2385,6 +2385,14 @@ def render_sql_template(
                 documentation="/cli/advanced-templates.html",
             )
         raise SQLTemplateException(str(e), documentation="/cli/advanced-templates.html")
+    except IndexError as e:
+        # This happens when trying to access string indices on empty strings
+        if "string index out of range" in str(e):
+            raise SQLTemplateException(
+                "String index out of range. Check that string parameters have values before accessing specific characters (e.g., param[0]). Provide default values or add length checks in your template.",
+                documentation="/cli/advanced-templates.html",
+            )
+        raise SQLTemplateException(str(e), documentation="/cli/advanced-templates.html")
     except Exception as e:
         # errors might vary here, we need to support as much as possible
         # https://gitlab.com/tinybird/analytics/-/issues/943

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev304'
-__revision__ = '9a414a1'
+__version__ = '0.0.1.dev305'
+__revision__ = 'cc453b3'

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/tinybird/tb/modules/local.py

@@ -1,11 +1,15 @@
+import json
+import uuid
 from pathlib import Path
+from typing import Any
 
 import click
+import jwt
 import requests
 from docker.client import DockerClient
 
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import update_cli
+from tinybird.tb.modules.common import echo_json, update_cli
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.local_common import (
     TB_CONTAINER_NAME,
@@ -125,7 +129,21 @@ def remove() -> None:
     is_flag=True,
     help="Skip pulling the latest Tinybird Local image. Use directly your current local image.",
 )
-def start(use_aws_creds: bool, volumes_path: str, skip_new_version: bool) -> None:
+@click.option(
+    "--user-token",
+    default=None,
+    envvar="TB_LOCAL_USER_TOKEN",
+    help="User token to use for the Tinybird Local container.",
+)
+@click.option(
+    "--workspace-token",
+    default=None,
+    envvar="TB_LOCAL_WORKSPACE_TOKEN",
+    help="Workspace token to use for the Tinybird Local container.",
+)
+def start(
+    use_aws_creds: bool, volumes_path: str, skip_new_version: bool, user_token: str, workspace_token: str
+) -> None:
     """Start Tinybird Local"""
     if volumes_path is not None:
         absolute_path = Path(volumes_path).absolute()
@@ -134,7 +152,7 @@ def start(use_aws_creds: bool, volumes_path: str, skip_new_version: bool) -> Non
 
     click.echo(FeedbackManager.highlight(message="» Starting Tinybird Local..."))
     docker_client = get_docker_client()
-    start_tinybird_local(docker_client, use_aws_creds, volumes_path, skip_new_version)
+    start_tinybird_local(docker_client, use_aws_creds, volumes_path, skip_new_version, user_token, workspace_token)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local is ready!"))
 
 
@@ -183,3 +201,35 @@ def version() -> None:
     """Show Tinybird Local version"""
     response = requests.get(f"{TB_LOCAL_ADDRESS}/version")
     click.echo(FeedbackManager.success(message=f"✓ Tinybird Local version: {response.text}"))
+
+
+@local.command()
+@click.pass_context
+def generate_tokens(ctx: click.Context) -> None:
+    """Generate static tokens for initializing Tinybird Local"""
+    output = ctx.ensure_object(dict).get("output")
+    user_id = str(uuid.uuid4())
+    workspace_id = str(uuid.uuid4())
+    user_token_id = str(uuid.uuid4())
+    workspace_token_id = str(uuid.uuid4())
+    payload = {"u": user_id, "id": user_token_id, "host": None}
+    user_token = generate_token(payload)
+    payload = {"u": workspace_id, "id": workspace_token_id, "host": None}
+    workspace_token = generate_token(payload)
+
+    if output == "json":
+        echo_json({"workspace_token": workspace_token, "user_token": user_token})
+    else:
+        click.echo(FeedbackManager.gray(message="Workspace token: ") + FeedbackManager.info(message=workspace_token))
+        click.echo(FeedbackManager.gray(message="User token: ") + FeedbackManager.info(message=user_token))
+        click.echo(FeedbackManager.success(message="✓ Tinybird Local tokens generated!"))
+
+
+def generate_token(payload: dict[str, Any]) -> str:
+    algo = jwt.algorithms.get_default_algorithms()["HS256"]
+    msg = json.dumps(payload)
+    msg_base64 = jwt.utils.base64url_encode(msg.encode())
+    sign_key = algo.prepare_key("abcd")
+    signature = algo.sign(msg_base64, sign_key)
+    token = msg_base64 + b"." + jwt.utils.base64url_encode(signature)
+    return "p." + token.decode()

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/tinybird/tb/modules/local_common.py

@@ -234,7 +234,9 @@ def start_tinybird_local(
     docker_client: DockerClient,
     use_aws_creds: bool,
     volumes_path: Optional[str] = None,
-    skip_new_version: bool = False,
+    skip_new_version: bool = True,
+    user_token: Optional[str] = None,
+    workspace_token: Optional[str] = None,
 ) -> None:
     """Start the Tinybird container."""
     pull_show_prompt = False
@@ -261,7 +263,13 @@ def start_tinybird_local(
         if pull_required:
             docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
 
-    environment = get_use_aws_creds() if use_aws_creds else {}
+    environment = {}
+    if use_aws_creds:
+        environment.update(get_use_aws_creds())
+    if user_token:
+        environment["TB_LOCAL_USER_TOKEN"] = user_token
+    if workspace_token:
+        environment["TB_LOCAL_WORKSPACE_TOKEN"] = workspace_token
 
     container = get_existing_container_with_matching_env(docker_client, TB_CONTAINER_NAME, environment)
 

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev304
+Version: 0.0.1.dev305
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -39,6 +39,7 @@ Requires-Dist: packaging<24,>=23.1
 Requires-Dist: llm>=0.19
 Requires-Dist: thefuzz==0.22.1
 Requires-Dist: python-dotenv==1.1.0
+Requires-Dist: pyjwt[crypto]==2.9.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{tinybird-0.0.1.dev304 → tinybird-0.0.1.dev305}/tinybird.egg-info/requires.txt

@@ -30,3 +30,4 @@ packaging<24,>=23.1
 llm>=0.19
 thefuzz==0.22.1
 python-dotenv==1.1.0
+pyjwt[crypto]==2.9.0