recce-nightly 1.9.0.20250622__py3-none-any.whl → 1.9.0.20250624__py3-none-any.whl

recce/server.py

@@ -8,7 +8,7 @@ from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from pathlib import Path
-from typing import Annotated, Any, Dict, Literal, Optional, Set
+from typing import Annotated, Any, Literal, Optional, Set
 
 from fastapi import (
     BackgroundTasks,
@@ -33,8 +33,15 @@ from . import __latest_version__, __version__, event
 from .apis.check_api import check_router
 from .apis.run_api import run_router
 from .config import RecceConfig
+from .connect_to_cloud import (
+    connect_to_cloud_background_task,
+    generate_key_pair,
+    get_connection_url,
+    is_callback_server_running,
+    prepare_connection_url,
+)
 from .core import RecceContext, default_context, load_context
-from .event import log_api_event, log_single_env_event
+from .event import get_recce_api_token, log_api_event, log_single_env_event
 from .exceptions import RecceException
 from .models.types import CllData
 from .run import load_preset_checks
@@ -247,8 +254,7 @@ async def recce_instance_info():
     read_only = flag.get("read_only", False)
     single_env = flag.get("single_env_onboarding", False)
 
-    auth_options = app_state.auth_options or {}
-    api_token = auth_options.get("api_token")
+    api_token = get_recce_api_token()
 
     return {
         "read_only": read_only,
@@ -271,13 +277,6 @@ async def config_flag():
     return flag
 
 
-@app.post("/api/onboarding/completed", status_code=204)
-async def mark_onboarding_completed():
-    context = default_context()
-    context.mark_onboarding_completed()
-    app.state.flag["show_onboarding_guide"] = False
-
-
 @app.post("/api/relaunch-hint/completed", status_code=204)
 async def mark_relaunch_hint_completed():
     app.state.flag["show_relaunch_hint"] = False
@@ -335,7 +334,12 @@ async def get_info():
 
 
 class CllIn(BaseModel):
-    params: Dict
+    node_id: Optional[str] = None
+    column: Optional[str] = None
+    change_analysis: Optional[bool] = False
+    cll: Optional[bool] = False
+    upstream: Optional[bool] = False
+    downstream: Optional[bool] = False
 
 
 class CllOutput(BaseModel):
@@ -347,27 +351,13 @@ async def column_level_lineage_by_node(cll_input: CllIn):
     from recce.adapter.dbt_adapter import DbtAdapter
 
     dbt_adapter: DbtAdapter = default_context().adapter
-    node_id = cll_input.params.get("node_id")
-    column = cll_input.params.get("column")
-
-    cll = dbt_adapter.get_cll(node_id, column)
-
-    return CllOutput(current=cll)
-
-
-class ImpactRadiusIn(BaseModel):
-    node_id: str
-
-
-@app.post("/api/impact-radius", response_model=CllOutput)
-async def impact_radius_by_node(impact_input: ImpactRadiusIn):
-    from recce.adapter.dbt_adapter import DbtAdapter
-
-    context = default_context()
-    dbt_adapter: DbtAdapter = context.adapter
-    node_id = impact_input.node_id
-
-    cll = dbt_adapter.get_impact_radius(node_id)
+    cll = dbt_adapter.get_cll(
+        node_id=cll_input.node_id,
+        column=cll_input.column,
+        change_analysis=cll_input.change_analysis,
+        upstream=cll_input.upstream,
+        downstream=cll_input.downstream,
+    )
 
     return CllOutput(current=cll)
 
@@ -670,6 +660,20 @@ async def broadcast(data: str):
         await client.send_text(data)
 
 
+@app.post("/api/connect")
+async def generate_connect_to_cloud_url(background_tasks: BackgroundTasks):
+    if is_callback_server_running():
+        return {"connection_url": get_connection_url()}
+
+    private_key, public_key = generate_key_pair()
+    connection_url, callback_port = prepare_connection_url(public_key)
+
+    background_tasks.add_task(connect_to_cloud_background_task, private_key, callback_port, connection_url)
+    return {
+        "connection_url": connection_url,
+    }
+
+
 api_prefix = "/api"
 app.include_router(check_router, prefix=api_prefix)
 app.include_router(run_router, prefix=api_prefix)

recce/state.py

@@ -16,6 +16,7 @@ import botocore.exceptions
 from pydantic import BaseModel, Field
 
 from recce import get_version
+from recce.event import get_recce_api_token
 from recce.git import current_branch
 from recce.models import CheckDAO
 from recce.models.types import Check, Run
@@ -766,7 +767,7 @@ class RecceShareStateManager:
         self.hint_message = None
 
     def verify(self) -> bool:
-        if self.auth_options.get("api_token") is None:
+        if get_recce_api_token() is None:
             self.error_message = RECCE_API_TOKEN_MISSING.error_message
             self.hint_message = RECCE_API_TOKEN_MISSING.hint_message
             return False
@@ -781,5 +782,5 @@ class RecceShareStateManager:
 
         with tempfile.NamedTemporaryFile() as tmp:
             state.to_file(tmp.name, file_type=SupportedFileTypes.FILE)
-            response = RecceCloud(token=self.auth_options.get("api_token")).share_state(file_name, open(tmp.name, "rb"))
+            response = RecceCloud(token=get_recce_api_token()).share_state(file_name, open(tmp.name, "rb"))
             return response

recce/util/api_token.py

@@ -7,8 +7,6 @@ from recce.exceptions import RecceConfigException
 from recce.util.recce_cloud import (
     RECCE_CLOUD_BASE_URL,
     RecceCloud,
-    get_recce_cloud_onboarding_state,
-    set_recce_cloud_onboarding_state,
 )
 
 console = Console()
@@ -54,7 +52,7 @@ def prepare_api_token(
             event.log_connected_to_cloud()
     else:
         # No api_token provided
-        if interaction is True:
+        if interaction:
             console.print(
                 "An API token is required for this feature. This can be obtained in your user account settings.\n"
                 f"{RECCE_CLOUD_BASE_URL}/settings#tokens\n"
@@ -70,10 +68,4 @@ def prepare_api_token(
                 "You no longer need to append --api-token to the recce command"
             )
 
-    if api_token:
-        cloud_onboarding_state = get_recce_cloud_onboarding_state(api_token)
-        if cloud_onboarding_state == "new":
-            # Mark the onboarding state as "launched" if the user is new
-            set_recce_cloud_onboarding_state(api_token, "launched")
-
     return api_token

recce/util/lineage.py

@@ -1,9 +1,9 @@
-from typing import Dict, Set, Tuple
+from typing import Dict, Iterable, Set, Tuple
 
 from recce.models.types import CllColumn, CllNode
 
 
-def find_upstream(node, parent_map):
+def find_upstream(node_ids: Iterable, parent_map):
     visited = set()
     upstream = set()
 
@@ -17,11 +17,13 @@ def find_upstream(node, parent_map):
             upstream.add(parent)
             dfs(parent)
 
-    dfs(node)
+    for node_id in node_ids:
+        dfs(node_id)
+
     return upstream
 
 
-def find_downstream(node, child_map):
+def find_downstream(node_ids: Iterable, child_map):
     visited = set()
     downstream = set()
 
@@ -35,13 +37,15 @@ def find_downstream(node, child_map):
             downstream.add(child)
             dfs(child)
 
-    dfs(node)
+    for node_id in node_ids:
+        dfs(node_id)
+
     return downstream
 
 
 def find_column_dependencies(node_column_id: str, parent_map: Dict, child_map: Dict) -> Tuple[Set, Set]:
-    upstream_cols = find_upstream(node_column_id, parent_map)
-    downstream_cols = find_downstream(node_column_id, child_map)
+    upstream_cols = find_upstream([node_column_id], parent_map)
+    downstream_cols = find_downstream([node_column_id], child_map)
     return upstream_cols, downstream_cols
 
 
@@ -63,16 +67,16 @@ def filter_lineage_vertices(
 
 
 def filter_dependency_maps(
-    parent_map: Dict, child_map: Dict, relevant_columns: Set
+    parent_map: Dict, child_map: Dict, relevant_ids: Set
 ) -> Tuple[Dict[str, Set], Dict[str, Set]]:
     p_map = {}
     c_map = {}
     for node_id, parents in parent_map.items():
-        if node_id in relevant_columns:
-            p_map[node_id] = {p for p in parents if p in relevant_columns}
+        if node_id in relevant_ids:
+            p_map[node_id] = {p for p in parents if p in relevant_ids}
 
     for node_id, children in child_map.items():
-        if node_id in relevant_columns:
-            c_map[node_id] = {c for c in children if c in relevant_columns}
+        if node_id in relevant_ids:
+            c_map[node_id] = {c for c in children if c in relevant_ids}
 
     return p_map, c_map

recce/util/onboarding_state.py

@@ -0,0 +1,45 @@
+from enum import Enum
+from typing import Union
+
+from recce.util.recce_cloud import (
+    get_recce_cloud_onboarding_state,
+    set_recce_cloud_onboarding_state,
+)
+
+
+class OnboardingState(Enum):
+    SIGNUP_SUCCESSFUL = "signup_successful"
+    LAUNCHED_WITH_TOKEN = "launched_with_token"
+    CONFIGURE_TWO_ENV = "configure_two_env"
+
+
+def update_onboarding_state(api_token: Union[str, bool, None], is_single_env: bool) -> OnboardingState:
+    if api_token:
+        # existing onboarding_state values -> new, launched, launched_with_two_envs
+        #   new -> launched
+        #   new -> launched_with_two_envs
+        #   launched -> launched_with_two_envs
+        cloud_onboarding_state = get_recce_cloud_onboarding_state(api_token)
+
+        if cloud_onboarding_state == OnboardingState.SIGNUP_SUCCESSFUL.value:
+            # User has an API Token and is a "new" user
+            if is_single_env:
+                # Mark the onboarding state as "launched" if the user is new
+                set_recce_cloud_onboarding_state(api_token, OnboardingState.LAUNCHED_WITH_TOKEN.value)
+                return OnboardingState.LAUNCHED_WITH_TOKEN
+            else:
+                set_recce_cloud_onboarding_state(api_token, OnboardingState.CONFIGURE_TWO_ENV.value)
+                return OnboardingState.CONFIGURE_TWO_ENV
+        elif cloud_onboarding_state == OnboardingState.LAUNCHED_WITH_TOKEN.value:
+            # User has an API Token and has Two Environments
+            if is_single_env:
+                # Just return the current state
+                return OnboardingState.LAUNCHED_WITH_TOKEN
+            else:
+                set_recce_cloud_onboarding_state(api_token, OnboardingState.CONFIGURE_TWO_ENV.value)
+                return OnboardingState.CONFIGURE_TWO_ENV
+        elif cloud_onboarding_state == OnboardingState.CONFIGURE_TWO_ENV.value:
+            # Just return the current state
+            return OnboardingState.CONFIGURE_TWO_ENV
+
+    return OnboardingState.SIGNUP_SUCCESSFUL

recce/util/recce_cloud.py

@@ -180,13 +180,13 @@ class RecceCloud:
 
     def set_onboarding_state(self, state: str):
         api_url = f"{self.base_url}/users/onboarding-state"
-        response = self._request("PUT", api_url, json={"state": state})
-        if response.status_code != 200:
-            raise RecceCloudException(
-                message="Failed to update onboarding state in Recce Cloud.",
-                reason=response.text,
-                status_code=response.status_code,
-            )
+        try:
+            response = self._request("PUT", api_url, json={"state": state})
+            response.raise_for_status()
+        except requests.exceptions.HTTPError as e:
+            # Don't Raise an exception if setting onboarding_state fails
+            logger.warning(f"Failed to set Onboarding State in Recce Cloud. Reason: {str(e)}")
+        return
 
 
 def get_recce_cloud_onboarding_state(token: str) -> str:

{recce_nightly-1.9.0.20250622.dist-info → recce_nightly-1.9.0.20250624.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: recce-nightly
-Version: 1.9.0.20250622
+Version: 1.9.0.20250624
 Summary: Environment diff tool for dbt
 Home-page: https://github.com/InfuseAI/recce
 Author: InfuseAI Dev Team

{recce_nightly-1.9.0.20250622.dist-info → recce_nightly-1.9.0.20250624.dist-info}/RECORD

@@ -1,8 +1,9 @@
-recce/VERSION,sha256=Ko4z6NZFs-uR9cVLSn1qtggpU1S5lipxndZ_bYd4hgw,15
+recce/VERSION,sha256=PM9mUjBu33bctG1WGN9EvPjwMgQRl2NheT822LUdTlc,15
 recce/__init__.py,sha256=yNb0QT-yoStex0VZALNJvUwtPLommoVCStcow31guqo,2392
 recce/artifact.py,sha256=tKQAHSrLRjiR3ppOI4sym8SxYiiLTuD3DPMYh4DWQdA,6506
-recce/cli.py,sha256=bhnVGR7K6cU3tyP-GbX2QHJgi4gFX_5xH4XqKWOOSY0,39710
+recce/cli.py,sha256=4uFTt2vWlLikySC2TsV3j_obUyfeg0L2vEW1cqXQprY,40529
 recce/config.py,sha256=fs22mpFj8CFIxftGbhFAV5xIsPLX2xNTwWSer3UYn5k,4658
+recce/connect_to_cloud.py,sha256=_SkX2pdyXa9FNyaHvItyYVPF2nZxy2JnCyd_o_psh2Y,4750
 recce/core.py,sha256=3Nv2QWBGz3yj1QI0wyPNRaDj1rH80DsX2qRQn9ydpg0,10747
 recce/diff.py,sha256=L2_bzQ3__PO-0aeir8PHF8FvSOUmQ8WcDXgML1-mHdY,748
 recce/exceptions.py,sha256=SclQ678GrHGjw7p_ZFJ3vZaL_yMU5xABeIAm2u_W2bk,592
@@ -10,25 +11,26 @@ recce/git.py,sha256=8Eg-6NzL-KjA3rT-ibbAyaCwGlzV0JqH3yGikrJNMDA,2344
 recce/github.py,sha256=PEpM6ZRiinsPbXSWj4aJCKbZrN1jUXzpzAfJq_CGah4,7420
 recce/pull_request.py,sha256=aW0B1NE2LUKTam1S4TQ7smXB9KLE1DV8GnyBqNXA6j8,3832
 recce/run.py,sha256=LAjbWUF8loZ9cL25d_maQELwROHHmfaFn7iqFRy5O1o,13567
-recce/server.py,sha256=Nr5y4jTk4CnHCTtcPHdJkWXjRybmSYBFAmxJ1t74Zlg,20667
-recce/state.py,sha256=VPzKyEdmyBeZ45BQArB587QN5C3Sfr7O5Oh_JRBcKrc,30616
+recce/server.py,sha256=RFbZQP9RAsbtGEgtP-ltLc9ZvBZZM6T9qEn9nIwh3HY,20944
+recce/state.py,sha256=fOzy09rn96XqOQuA0RcMW-k2KpjNVH9FQhihEXLvlA0,30634
 recce/summary.py,sha256=Mbxvxr9KazR5o9icqhhjiGHsoAiWxQU4PdN7HytBJ1c,19154
 recce/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 recce/adapter/base.py,sha256=T_JNeLHgiHSaegw-DbrvHOaYjMyZcjj2Qtg5cWh_fco,3548
 recce/adapter/sqlmesh_adapter.py,sha256=IU3N-F6ToDoO7_bV5vsG8pmTuDcbFtewTIuCxedTaRM,5046
-recce/adapter/dbt_adapter/__init__.py,sha256=Bnh-6ZRO5k1yxJxjGJcf50NcMMTe6or2DoLaeFFMaZM,60211
+recce/adapter/dbt_adapter/__init__.py,sha256=TqxYCxIe675XswFmx57hW55GV-YgC9pGMZW4mikVvV4,62852
 recce/adapter/dbt_adapter/dbt_version.py,sha256=M7aedZIWslXnJsryK8Ki4OL_t2oAKxy4uE2pRwfWIkk,1228
 recce/apis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 recce/apis/check_api.py,sha256=KMCXSMl1qqzx2jQgRqCrD4j_cY3EHBbM3H2-t-6saAU,6227
 recce/apis/check_func.py,sha256=gktbCcyk3WGvWRJJ-wDnwv7NrIny2nTHWLl1-kdiVRo,4183
 recce/apis/run_api.py,sha256=eOaxOxXDkH59uqGCd4blld7edavUx7JU_DCd2WAYrL8,3416
 recce/apis/run_func.py,sha256=6wC8TDU-h7TLr2VZH7HNsWaUVlQ9HBN5N_dwqfi4lMY,7440
-recce/data/404.html,sha256=BhzZcWhFy9Im2YToG8rlIVHaIqetOX-tXyW0nv3FhLk,26574
+recce/data/404.html,sha256=ZSuzOByvWRvnqWk8gxsrw8_5FmBt0mvHqXbSowXEK70,26574
+recce/data/auth_callback.html,sha256=2QnhHRc2eLPiaxCYl-HndNsGDGmM6jZAMbqoMlehal4,94783
 recce/data/favicon.ico,sha256=B2mBumUOnzvUrXrqNkrc5QfdDXjzEXRcWkWur0fJ6sM,2565
-recce/data/index.html,sha256=sYlLEVgPoApkCbCFQd7r4MUJ0XWEi61y_Gg9NP_AsFQ,42155
-recce/data/index.txt,sha256=Q68kY89E_Iqmx1OSXbO9-8OxKFacmn5m5Jx5oR5Cq4Q,4462
-recce/data/_next/static/M3_t2h-wA4_IXOtKHdJqV/_buildManifest.js,sha256=sjy6FmOqWlkXWmod2oXwqKv7L582_MTycs9THCE6G-c,224
-recce/data/_next/static/M3_t2h-wA4_IXOtKHdJqV/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
+recce/data/index.html,sha256=MybT9NbtYA-o_HIVCKS6agiqjoATOg2BFI19JZCBcvs,42155
+recce/data/index.txt,sha256=f5XOwdv9lEn96-mYbI4hur7bUvreWiTlQX1vdf43l9o,4462
+recce/data/_next/static/Iga3yMdHBVJ_Vhluh2SdE/_buildManifest.js,sha256=sjy6FmOqWlkXWmod2oXwqKv7L582_MTycs9THCE6G-c,224
+recce/data/_next/static/Iga3yMdHBVJ_Vhluh2SdE/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
 recce/data/_next/static/chunks/181-acc61ddada3bc0ca.js,sha256=2zRVkcaI906lYW_aM2mNqnt2HbLxmUaYFjft7eNcuTQ,310191
 recce/data/_next/static/chunks/1bff33f1-1ef85cf5e658a751.js,sha256=wTH-_PbE1XdP-wmGaQStF5y2yH6XcZXeYX_hDaGMfh4,153777
 recce/data/_next/static/chunks/217-879a84d70f7a907c.js,sha256=kNJkLcwaIlWnQY7akbrFimbcHHdK6lMVTbTYCNKXIjc,124102
@@ -45,7 +47,7 @@ recce/data/_next/static/chunks/7a8a3e83-edf6dc64b5d5f0a5.js,sha256=NfgBK9op5SzWm
 recce/data/_next/static/chunks/7f27ae6c-d5f0438edd5c2a5b.js,sha256=ewMLioJbTUUDeqV0J9NDKE0Bj-XCFg8QqMLOmLPgpcc,777
 recce/data/_next/static/chunks/86730205-cfb14e3f051bab35.js,sha256=qKkzgWsNqSrn-PUGBloZopIlacGCy8hH0BZA9tpqQoQ,172837
 recce/data/_next/static/chunks/8d700b6a.8bb140898499c512.js,sha256=NPGWvqv8Y_rbNaKxYnoq16mi6FDoQHxOOjpH4tJnUkE,119972
-recce/data/_next/static/chunks/92-7ab55ae02606193c.js,sha256=f0YhpleWOSyqlxfS32YC25CQnb0EJAujrN4BWncRKdY,80497
+recce/data/_next/static/chunks/92-607cd1af83c41f43.js,sha256=0ThWZlPjuwxRRFkwB0qN9Q6QdMxJzdGhuIyLAOHO4lw,80466
 recce/data/_next/static/chunks/9746af58-a42b7d169cacadf0.js,sha256=lV-V6BYpd4HEt-pFtT03G6qQD2Wa-2L_jp6aH9VgoeM,4177
 recce/data/_next/static/chunks/a30376cd-de84559016d7e133.js,sha256=qzgyjrX5PdPKGeSwb-Gh0RfM_NEF2zjMvMK14ixq228,864
 recce/data/_next/static/chunks/b63b1b3f-4282bdcf459e075c.js,sha256=s2EZ59zDBJDmy2kh0cHkQ_pAWxiQr_yCPlnttCj0x_k,572
@@ -60,7 +62,7 @@ recce/data/_next/static/chunks/main-b5b3ae20a1405261.js,sha256=28hJH2VsiTdxZGZrQ
 recce/data/_next/static/chunks/polyfills-42372ed130431b0a.js,sha256=CXPB1kyIrcjjyVBBDLWLKI9yEY1ZZbeASUON648vloM,112594
 recce/data/_next/static/chunks/webpack-7b49d5ba7e3a434d.js,sha256=lcK59Juk4EZqi3jIsY35lQqAlB5TLTPy-neri0BIZL0,3929
 recce/data/_next/static/chunks/app/layout-177a410a97e0d018.js,sha256=Hmy6Lmrt4GPYYa0f1qhxTiMj4K47PCmkAkrm1Rbnd7w,1731
-recce/data/_next/static/chunks/app/page-707b8e8d98d7f780.js,sha256=4R2WXYq5x46aRuNVKckw4f5YmkbU5XI0fFSOeEhGt_E,156467
+recce/data/_next/static/chunks/app/page-3f79c2e6d6f03a09.js,sha256=N3h3lkuP_o97aLrOd0lZp--oJqjuw_SOv733KWqOCM0,156413
 recce/data/_next/static/chunks/app/_not-found/page-01ed58b7f971d311.js,sha256=_l3JSAaIJxInWqyqdDsobkhJ3JP6DZuwXkw-MsRakQA,1751
 recce/data/_next/static/chunks/pages/_app-437c455677d62394.js,sha256=FaxURD1wuAmnMU1zJlsQKT9rRdfuspR7yx7BckcoCLY,284
 recce/data/_next/static/chunks/pages/_error-e7650df18ca04bde.js,sha256=0Vzx_hCqw4RcptsFzHfUigpI13sToBnFw38owTP7ruo,250
@@ -91,7 +93,7 @@ recce/event/track.py,sha256=xDyDWblhR6mp0tezEYhkA0aOyZLqXBpJyHZwhI2xhU8,4803
 recce/models/__init__.py,sha256=F7cgALtdWnwv37R0eEgKZ_yBsMwxWnUfo3jAZ3u6qyU,209
 recce/models/check.py,sha256=jjR1SGyczjrqyK-0Zas6ikLIGSgVp54lvfcQA19AniI,1588
 recce/models/run.py,sha256=QK2gvOWvko9YYhd2NLs3BPt5l4MSCZGwpzTAiqx9zJw,1161
-recce/models/types.py,sha256=g4w5RNNeeNRUDyLWd1Sqv7wPvt0ue4d0-_-KZ585GJI,4193
+recce/models/types.py,sha256=0YbhuL_MiDinhtqq9UQJQYu5_zTNWOjm7G5t-WKbDV0,4390
 recce/tasks/__init__.py,sha256=b553AtDHjYROgmMePv_Hv_X3fjh4iEn11gzzpUJz6_o,610
 recce/tasks/core.py,sha256=JFYa1CfgOiRPQ7KVTwMuxJjhMB-pvCwB-xezVt-h3RU,4080
 recce/tasks/dataframe.py,sha256=03UBWwt0DFTXlaEOtnV5i_mxdRKD7UbRayewEL2Ub48,3650
@@ -104,21 +106,23 @@ recce/tasks/schema.py,sha256=HHrSvhd_ZJdrNj2QKi9W8vmig0NuYci5cgsKFvp2bu4,2274
 recce/tasks/top_k.py,sha256=vY3VCBmg61E8I4V-9-hJOv5RCYCmhxl6sHiKR9Zv7eE,5521
 recce/tasks/valuediff.py,sha256=XJWkA307B5qTerT87fJRhJxPCqjmXn5eILYCXDfPtSQ,16439
 recce/util/__init__.py,sha256=cDvL3WT32cJR9CUPIJmibwJoyYcB3ZaKBFbOZDyvqP8,95
-recce/util/api_token.py,sha256=w_NI4pzcRv2FpBkYPW9Bg9TtQa50pB-hV1ukriIrQdg,3102
+recce/util/api_token.py,sha256=D1fM5romXnxXIvmedr8_0r_jspRRjWsmDK-lM7VSrk4,2737
 recce/util/breaking.py,sha256=evT-xlDWMgIbuPQH6w5yNoDjOqopKzBXnCF9_xTn1S0,12591
 recce/util/cache.py,sha256=QB6wzxe0M3jNTwP0M27Ys8F2hF-oda4-LyXXG9THuZQ,646
 recce/util/cll.py,sha256=VXhtyoPIcPEW3_SmNUTbZWwOmmv8_JRfAB9CLJ4S2Wg,13315
 recce/util/io.py,sha256=fNiIafNxPlTF6NNXlugzeJjfVGGrP3RXb39ERzRWD3Q,3459
-recce/util/lineage.py,sha256=IF1CNrYt0u_bNkB4XQaueteD1JqGKDEnNQV-Qba26ZE,2090
+recce/util/lineage.py,sha256=WdHxFTViqykBT8hBIbhGOovpEVnu9hbaW0VEQJUiCmE,2186
 recce/util/logger.py,sha256=6UgLFkRiur9jJfu2ZRdo4LUvMw4f75V-l-1HT1-sgKo,747
+recce/util/onboarding_state.py,sha256=kwFirKlfXdl5WFkR_nmilqGKFyELNcSPMqYq-by35fk,1991
 recce/util/pydantic_model.py,sha256=KumKuyCjbTzEMsKLE4-b-eZfp0gLhYDdmVtw1-hxiJw,587
-recce/util/recce_cloud.py,sha256=g7uhWkknR9ulTuGxaz14erOGiUwIv3q3C0mfCEG3gl8,7925
+recce/util/recce_cloud.py,sha256=YgRs9PhER3EhESaJInJo6mf3uwvnp0OavHBY1OYHPXw,7958
 recce/util/singleton.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
 recce/yaml/__init__.py,sha256=EgXYlFeJZchatUClRDXbIC5Oqb2_nBvB2NqItYVihio,1292
-recce_nightly-1.9.0.20250622.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
+recce_nightly-1.9.0.20250624.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_cli.py,sha256=_ikuZW4bYEGyj5A_V7nDSyBa1tVZqc0TVm-TJ0rEYVo,5467
 tests/test_config.py,sha256=ODDFe_XF6gphmSmmc422dGLBaCCmG-IjDzTkD5SJsJE,1557
+tests/test_connect_to_cloud.py,sha256=b2fgV8L1iQBdEwh6RumMsIIyYg7GtMOMRz1dvE3WRPg,3059
 tests/test_core.py,sha256=jfn9dV-k89O2TywUYHmMTSdvyfBUips1ful1cIY2oFY,799
 tests/test_dbt.py,sha256=VzXvdoJNwwEmKNhJJDNB1N_qZYwp1aoJQ1sLcoyRBmk,1316
 tests/test_pull_request.py,sha256=HmZo5MoDaoKSgPwbLxJ3Ur3ajZ7IxhkzJxaOmhg6bwE,3562
@@ -130,7 +134,7 @@ tests/adapter/dbt_adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 tests/adapter/dbt_adapter/conftest.py,sha256=WlMVytJcyfLJlMBeYhpk8Q-nyCDR9x_7qgsy6Uh9ECw,386
 tests/adapter/dbt_adapter/dbt_test_helper.py,sha256=o_7m0cnVeGwPwEFXBCx3Me5QqLzQMOBR3dXD-MS2eaM,10406
 tests/adapter/dbt_adapter/test_dbt_adapter.py,sha256=Y5TSlqkXGjJC6hC7YYs45bnRht5aou_KLjPMjpAGxNM,830
-tests/adapter/dbt_adapter/test_dbt_cll.py,sha256=W3VKg6GKUMGyR_qomwbOpVcSyOTpYBb_KMhS4NfGvS8,14249
+tests/adapter/dbt_adapter/test_dbt_cll.py,sha256=IvFIImI6SQF-F-Ti0tHvUEZkAiKOmdpb9ZeJ_5JOw9w,14294
 tests/adapter/dbt_adapter/test_selector.py,sha256=uKFm0QxrxG-xyUEXGobCMueCRUssgnZMyo-S1Nlk3_s,6856
 tests/tasks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/tasks/conftest.py,sha256=lyZFx4jEY3To3xVhfMQ-G45egYOCttKN2lxeyExACkY,157
@@ -143,8 +147,8 @@ tests/tasks/test_row_count.py,sha256=21PaP2aq-x8-pqwzWHRT1sixhQ8g3CQNRWOZTTmbK0s
 tests/tasks/test_schema.py,sha256=7ds4Vx8ixaiIWDR49Lvjem4xlPkRP1cXazDRY3roUak,3121
 tests/tasks/test_top_k.py,sha256=YR_GS__DJsbDlQVaEEdJvNQ3fh1VmV5Nb3G7lb0r6YM,1779
 tests/tasks/test_valuediff.py,sha256=_xQJGgxsXoy2NYk_Z6Hsw2FlVh6zk2nN_iUueyRN1e8,2046
-recce_nightly-1.9.0.20250622.dist-info/METADATA,sha256=FrkMZ6YqX3pe92_YA77jSUcNGl1gAYiNACnMlt7ztl0,9397
-recce_nightly-1.9.0.20250622.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-recce_nightly-1.9.0.20250622.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
-recce_nightly-1.9.0.20250622.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
-recce_nightly-1.9.0.20250622.dist-info/RECORD,,
+recce_nightly-1.9.0.20250624.dist-info/METADATA,sha256=okPHlV3RrR0VRdIJj1C7cXeoRRADck7xIb4fhNbsuDc,9397
+recce_nightly-1.9.0.20250624.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+recce_nightly-1.9.0.20250624.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
+recce_nightly-1.9.0.20250624.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
+recce_nightly-1.9.0.20250624.dist-info/RECORD,,

tests/adapter/dbt_adapter/test_dbt_cll.py

@@ -51,11 +51,11 @@ def test_cll_basic(dbt_test_helper):
     )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
 
-    result = adapter.get_cll_by_node_id("model.model2")
+    result = adapter.get_cll("model.model2", no_filter=True)
     assert_model(result, "model.model2", [("model.model1", "c")])
     assert_column(result, "model.model2", "c", "passthrough", [("model.model1", "c")])
 
-    result = adapter.get_cll_by_node_id("model.model3")
+    result = adapter.get_cll("model.model3", no_filter=True)
     assert_model(result, "model.model3", [("model.model1", "c")])
     assert_column(result, "model.model3", "c", "passthrough", [("model.model1", "c")])
 
@@ -75,7 +75,7 @@ def test_cll_table_alisa(dbt_test_helper):
         depends_on=["model.model1"],
     )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
-    result = adapter.get_cll_by_node_id("model.model1")
+    result = adapter.get_cll("model.model1", no_filter=True)
     assert_column(result, "model.model2", "c", "passthrough", [("model.model1", "c")])
 
 
@@ -103,7 +103,7 @@ def test_seed(dbt_test_helper):
     )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
 
-    result = adapter.get_cll_by_node_id("model.model1")
+    result = adapter.get_cll("model.model1", no_filter=True)
     assert_model(result, "seed.seed1", [])
     assert_column(result, "seed.seed1", "customer_id", "source", [])
     assert_model(result, "model.model1", [("seed.seed1", "age")])
@@ -134,7 +134,7 @@ def test_python_model(dbt_test_helper):
     assert not adapter.is_python_model("model1")
     assert adapter.is_python_model("model2")
 
-    result = adapter.get_cll_by_node_id("model1")
+    result = adapter.get_cll("model1", no_filter=True)
     assert_column(result, "model2", "customer_id", "unknown", [])
 
 
@@ -161,9 +161,9 @@ def test_source(dbt_test_helper):
         depends_on=["source.source1.table1"],
     )
     adapter: DbtAdapter = dbt_test_helper.context.adapter
-    result = adapter.get_cll_by_node_id("source.source1.table1")
+    result = adapter.get_cll("source.source1.table1", no_filter=True)
     assert_column(result, "source.source1.table1", "customer_id", "source", [])
-    result = adapter.get_cll_by_node_id("model.model1")
+    result = adapter.get_cll("model.model1", no_filter=True)
     assert_column(result, "model.model1", "customer_id", "passthrough", [("source.source1.table1", "customer_id")])
 
 
@@ -171,14 +171,14 @@ def test_parse_error(dbt_test_helper):
     dbt_test_helper.create_model("model1", curr_sql="select 1 as c", curr_columns={"c": "int"})
     dbt_test_helper.create_model("model2", curr_sql="this is not a valid sql", curr_columns={"c": "int"})
     adapter: DbtAdapter = dbt_test_helper.context.adapter
-    result = adapter.get_cll_by_node_id("model2")
+    result = adapter.get_cll("model2", no_filter=True)
     assert_column(result, "model2", "c", "unknown", [])
 
 
 def test_model_without_catalog(dbt_test_helper):
     dbt_test_helper.create_model("model1", curr_sql="select 1 as c")
     adapter: DbtAdapter = dbt_test_helper.context.adapter
-    result = adapter.get_cll_by_node_id("model1")
+    result = adapter.get_cll("model1", no_filter=True)
     assert not result.nodes["model1"].columns
 
 

tests/test_connect_to_cloud.py

@@ -0,0 +1,82 @@
+import base64
+import unittest
+from unittest.mock import patch
+from urllib.parse import quote
+
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.primitives.asymmetric import padding
+
+from recce.connect_to_cloud import (
+    connect_to_cloud_background_task,
+    decrypt_code,
+    generate_key_pair,
+    is_callback_server_running,
+    prepare_connection_url,
+)
+
+
+class ConnectToCloudTests(unittest.TestCase):
+
+    def test_generate_key_pair(self):
+        private_key, public_key = generate_key_pair()
+        self.assertIsNotNone(private_key)
+        self.assertIsNotNone(public_key)
+        self.assertEqual(private_key.public_key().public_numbers(), public_key.public_numbers())
+
+    def test_prepare_connection_url(self):
+        _, public_key = generate_key_pair()
+        url, port = prepare_connection_url(public_key)
+        self.assertIn("connect?", url)
+        self.assertTrue(port >= 10000 and port <= 15000)
+
+    def test_decrypt_code(self):
+        private_key, public_key = generate_key_pair()
+        test_string = "recce-api-token-123"
+        ciphertext = public_key.encrypt(
+            test_string.encode(),
+            padding.OAEP(mgf=padding.MGF1(algorithm=hashes.SHA1()), algorithm=hashes.SHA1(), label=None),
+        )
+        b64_ciphertext = base64.b64encode(ciphertext).decode()
+        result = decrypt_code(private_key, b64_ciphertext)
+        self.assertEqual(result, test_string)
+
+    @patch("recce.connect_to_cloud.update_recce_api_token")
+    @patch("recce.connect_to_cloud.update_onboarding_state")
+    @patch("recce.connect_to_cloud.RecceCloud")
+    def test_handle_callback_request_success(self, mock_recce_cloud, mock_update_state, mock_update_token):
+        private_key, public_key = generate_key_pair()
+
+        # Prepare encrypted token
+        test_token = "recce-api-token-xyz"
+        ciphertext = public_key.encrypt(
+            test_token.encode(),
+            padding.OAEP(mgf=padding.MGF1(algorithm=hashes.SHA1()), algorithm=hashes.SHA1(), label=None),
+        )
+        encrypted_b64 = base64.b64encode(ciphertext).decode()
+
+        # Set up mocks
+        mock_recce_cloud.return_value.verify_token.return_value = True
+
+        from recce.connect_to_cloud import handle_callback_request
+
+        result = handle_callback_request(f"code={quote(encrypted_b64)}", private_key)
+
+        assert result == test_token
+        mock_update_token.assert_called_once_with(test_token)
+        mock_update_state.assert_called_once_with(test_token, False)
+
+    def test_is_callback_server_running(self):
+        # Should return False by default
+        self.assertFalse(is_callback_server_running())
+
+    @patch("recce.connect_to_cloud.run_one_time_http_server")
+    def test_connect_to_cloud_background_task_runs(self, mock_server):
+        private_key, public_key = generate_key_pair()
+        url, port = prepare_connection_url(public_key)
+
+        connect_to_cloud_background_task(private_key, port, url)
+        mock_server.assert_called_once()
+
+
+if __name__ == "__main__":
+    unittest.main()