recce-nightly 1.2.0.20250506__py3-none-any.whl → 1.26.0.20251124__py3-none-any.whl

recce/server.py

@@ -7,30 +7,66 @@ import uuid
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import datetime, timedelta
+from enum import Enum
 from pathlib import Path
-from typing import Optional, Any, Set, Annotated, Literal, Dict
-
-from fastapi import FastAPI, HTTPException, Request, WebSocket, UploadFile, Response, BackgroundTasks, Form
+from typing import Annotated, Any, Literal, Optional, Set
+
+from fastapi import (
+    BackgroundTasks,
+    FastAPI,
+    Form,
+    HTTPException,
+    Request,
+    Response,
+    UploadFile,
+    WebSocket,
+)
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import PlainTextResponse
 from fastapi.staticfiles import StaticFiles
-from pydantic import ValidationError, BaseModel
+from pydantic import BaseModel, ValidationError
 from pytz import utc
 from starlette.middleware.gzip import GZipMiddleware
 from starlette.middleware.sessions import SessionMiddleware
 from starlette.websockets import WebSocketDisconnect
 
-from . import __version__, event, __latest_version__
+from . import __latest_version__, __version__, event, is_recce_cloud_instance
 from .apis.check_api import check_router
 from .apis.run_api import run_router
 from .config import RecceConfig
-from .core import load_context, default_context, RecceContext
-from .event import log_api_event, log_single_env_event
+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 get_recce_api_token, log_api_event, log_single_env_event
 from .exceptions import RecceException
+from .github import is_github_codespace
+from .models.types import CllData
 from .run import load_preset_checks
-from .state import RecceStateLoader, RecceShareStateManager
+from .state import RecceShareStateManager, RecceStateLoader
+
+logger = logging.getLogger("uvicorn")
+
+# Idle timeout check interval bounds (in seconds)
+MAX_CHECK_INTERVAL = 30
+MIN_CHECK_INTERVAL = 1
+
 
-logger = logging.getLogger('uvicorn')
+class RecceServerMode(str, Enum):
+    server = "server"
+    preview = "preview"
+    read_only = "read-only"
+
+    def __str__(self):
+        return self.value
+
+    @staticmethod
+    def available_members() -> Set[str]:
+        return ["server", "preview", "read-only"]
 
 
 @dataclass
@@ -42,24 +78,82 @@ class AppState:
     auth_options: Optional[dict] = None
     lifetime: Optional[int] = None
     lifetime_expired_at: Optional[datetime] = None
+    idle_timeout: Optional[int] = None
+    last_activity: Optional[dict] = None
+    share_url: Optional[str] = None
+    organization_name: Optional[str] = None
+    web_url: Optional[str] = None
+    host: Optional[str] = None
+    port: Optional[int] = None
 
 
 def schedule_lifetime_termination(app_state):
     def terminating_server():
         pid = os.getpid()
-        logger.info(f"Terminating server process [{pid}] manually")
+        logger.info(f"Terminating server process [{pid}] manually due to lifetime expiration")
         os.kill(pid, signal.SIGINT)
 
     # Terminate the server process after the specified lifetime
-    logger.info(f'[Configuration] The lifetime of the server is {app_state.lifetime} seconds')
+    logger.info(f"[Configuration] The lifetime of the server is {app_state.lifetime} seconds")
     app.state.lifetime_expired_at = datetime.now(utc) + timedelta(seconds=app_state.lifetime)
     asyncio.get_running_loop().call_later(app_state.lifetime, terminating_server)
 
 
+def schedule_idle_timeout_check(app_state):
+    """
+    Schedule periodic checks for idle timeout.
+    If the server has been idle for longer than idle_timeout, terminate it.
+    """
+    # Track last activity time in app_state
+    app_state.last_activity = {"time": datetime.now(utc)}
+
+    def terminating_server_idle():
+        pid = os.getpid()
+        logger.info(f"Terminating server process [{pid}] manually due to idle timeout")
+        os.kill(pid, signal.SIGINT)
+
+    async def check_idle_timeout():
+        """Periodically check if the server has been idle for too long"""
+        # Use smaller check interval if idle_timeout is very short
+        # Check at least every MAX_CHECK_INTERVAL seconds, but also check when idle_timeout is approaching
+        check_interval = min(MAX_CHECK_INTERVAL, max(MIN_CHECK_INTERVAL, app_state.idle_timeout // 3))
+
+        logger.debug(f"[Idle Timeout] Starting idle timeout checker with {check_interval}s check interval")
+
+        while True:
+            await asyncio.sleep(check_interval)
+
+            idle_seconds = (datetime.now(utc) - app_state.last_activity["time"]).total_seconds()
+            remaining_seconds = app_state.idle_timeout - idle_seconds
+
+            # Always log the countdown for debugging
+            if remaining_seconds > 0:
+                logger.debug(
+                    f"[Idle Timeout] Server idle for {idle_seconds:.1f}s / {app_state.idle_timeout}s "
+                    f"(remaining: {remaining_seconds:.1f}s)"
+                )
+
+            if idle_seconds >= app_state.idle_timeout:
+                logger.info(
+                    f"[Idle Timeout] Threshold reached! Server has been idle for {idle_seconds:.0f} seconds "
+                    f"(threshold: {app_state.idle_timeout} seconds)"
+                )
+                terminating_server_idle()
+                break
+
+    # Start the idle timeout check task
+    logger.info(f"[Configuration] The idle timeout of the server is {app_state.idle_timeout} seconds")
+
+    # Create task using asyncio.create_task which works in async context
+    task = asyncio.create_task(check_idle_timeout())
+    logger.debug(f"[Idle Timeout] Background task created: {task}")
+
+
 def setup_server(app_state: AppState) -> RecceContext:
-    from .core import load_context
     from rich.console import Console
 
+    from .core import load_context
+
     console = Console()
     state_loader = app_state.state_loader
     kwargs = app_state.kwargs
@@ -73,56 +167,87 @@ def setup_server(app_state: AppState) -> RecceContext:
         log_single_env_event()
 
     # Initialize Recce Config
-    config = RecceConfig(config_file=kwargs.get('config'))
+    config = RecceConfig(config_file=kwargs.get("config"))
     if state_loader.state is None:
-        preset_checks = config.get('checks', [])
+        preset_checks = config.get("checks", [])
         if preset_checks and len(preset_checks) > 0:
             console.rule("Loading Preset Checks")
             load_preset_checks(preset_checks)
 
     from recce.event import log_load_state
-    log_load_state(command='server', single_env=single_env)
 
-    if app_state.lifetime is not None and app_state.lifetime > 0:
-        schedule_lifetime_termination(app_state)
+    log_load_state(command="server", single_env=single_env)
 
     return ctx
 
 
 def teardown_server(app_state: AppState, ctx: RecceContext):
-    state_loader = app_state.state_loader
+    # pull latest state, merge runs/checks and pick the newer artifacts
+    state_loader = ctx.state_loader
+    state_loader.refresh()
+    if state_loader.state:
+        ctx.import_state(state_loader.state, merge=True)
     state_loader.export(ctx.export_state())
-
     ctx.stop_monitor_artifacts()
     if app_state.flag.get("single_env_onboarding", False):
         ctx.stop_monitor_base_env()
 
 
 def setup_ready_only(app_state: AppState):
-    if app_state.lifetime is not None and app_state.lifetime > 0:
-        schedule_lifetime_termination(app_state)
+    pass
 
 
 def teardown_ready_only(app_state: AppState):
     pass
 
 
+def setup_preview(app_state: AppState):
+    state_loader = app_state.state_loader
+    kwargs = app_state.kwargs
+    ctx = load_context(**kwargs, state_loader=state_loader)
+    return ctx
+
+
+def teardown_preview(app_state: AppState, ctx: RecceContext):
+    state_loader = app_state.state_loader
+    state_loader.export(ctx.export_state())
+    pass
+
+
 @asynccontextmanager
 async def lifespan(fastapi: FastAPI):
     ctx = None
     app_state: AppState = app.state
 
-    if app_state.command == 'server':
+    # Ensure logger is at DEBUG level if debug mode is enabled
+    # This is needed because uvicorn might reset logger configuration
+    if hasattr(app_state, "kwargs") and app_state.kwargs.get("debug"):
+        logger.setLevel(logging.DEBUG)
+        logger.debug("Debug mode enabled - logger set to DEBUG level")
+
+    if app_state.command == "server":
         ctx = setup_server(app_state)
-    elif app_state.command == 'read_only':
+    elif app_state.command == "read-only":
         setup_ready_only(app_state)
+    elif app_state.command == "preview":
+        ctx = setup_preview(app_state)
+
+    if app_state.lifetime is not None and app_state.lifetime > 0:
+        schedule_lifetime_termination(app_state)
+
+    # Schedule idle timeout check if idle_timeout is set
+    if app_state.idle_timeout is not None and app_state.idle_timeout > 0:
+        logger.debug(f"[Idle Timeout] Scheduling idle timeout check with {app_state.idle_timeout} seconds")
+        schedule_idle_timeout_check(app_state)
 
     yield
 
-    if app_state.command == 'server':
+    if app_state.command == "server":
         teardown_server(app_state, ctx)
-    elif app_state.command == 'read_only':
+    elif app_state.command == "read_only":
         teardown_ready_only(app_state)
+    elif app_state.command == "preview":
+        teardown_preview(app_state, ctx)
 
 
 app = FastAPI(lifespan=lifespan)
@@ -130,7 +255,7 @@ app = FastAPI(lifespan=lifespan)
 
 def verify_json_file(file_path: str) -> bool:
     try:
-        with open(file_path, 'r') as f:
+        with open(file_path, "r", encoding="utf-8") as f:
             json.load(f)
     except Exception:
         return False
@@ -143,19 +268,15 @@ def dbt_artifacts_updated_callback(file_changed_event: Any):
     file_name = src_path.name
 
     if not verify_json_file(file_changed_event.src_path):
-        logger.debug('Skip to refresh the artifacts because the file is not updated completely.')
+        logger.debug("Skip to refresh the artifacts because the file is not updated completely.")
         return
 
-    logger.info(
-        f'Detect {target_type} file {file_changed_event.event_type}: {file_name}')
+    logger.info(f"Detect {target_type} file {file_changed_event.event_type}: {file_name}")
     ctx = load_context()
     ctx.refresh_manifest(file_changed_event.src_path)
     broadcast_command = {
-        'command': 'refresh',
-        'event': {
-            'eventType': file_changed_event.event_type,
-            'srcPath': file_changed_event.src_path
-        }
+        "command": "refresh",
+        "event": {"eventType": file_changed_event.event_type, "srcPath": file_changed_event.src_path},
     }
     payload = json.dumps(broadcast_command)
     asyncio.run(broadcast(payload))
@@ -164,7 +285,7 @@ def dbt_artifacts_updated_callback(file_changed_event: Any):
 def dbt_env_updated_callback():
     logger.info("Detect 'manifest.json' and 'catalog.json' are generated under 'target-base' directory")
     broadcast_command = {
-        'command': 'relaunch',
+        "command": "relaunch",
     }
     payload = json.dumps(broadcast_command)
     asyncio.run(broadcast(payload))
@@ -191,11 +312,32 @@ app.add_middleware(
 )
 
 
+@app.middleware("http")
+async def track_activity_for_idle_timeout(request: Request, call_next):
+    """Track activity time for idle timeout check"""
+    # Exclude paths that should not reset idle timer
+    # Health checks and monitoring endpoints don't count as user activity
+    excluded_paths = ["/api/health", "/api/ws"]
+
+    # Update last activity time BEFORE processing request if idle timeout is enabled
+    # This ensures long-running requests don't get terminated mid-execution
+    app_state: AppState = app.state
+    if app_state.last_activity is not None:
+        if request.url.path not in excluded_paths:
+            app_state.last_activity["time"] = datetime.now(utc)
+            logger.debug(f"[Idle Timeout] ✓ Activity detected: {request.method} {request.url.path} - Timer reset")
+        else:
+            logger.debug(f"[Idle Timeout] Excluded path (no timer reset): {request.method} {request.url.path}")
+
+    response = await call_next(request)
+    return response
+
+
 @app.middleware("http")
 async def set_context_by_cookie(request: Request, call_next):
     response = await call_next(request)
 
-    user_id_in_cookie = request.cookies.get('recce_user_id')
+    user_id_in_cookie = request.cookies.get("recce_user_id")
     user_id = event.get_user_id()
 
     if event.is_anonymous_tracking() is False:
@@ -203,7 +345,7 @@ async def set_context_by_cookie(request: Request, call_next):
         user_id = None
 
     if user_id_in_cookie is None or user_id_in_cookie != user_id:
-        response.set_cookie(key='recce_user_id', value=user_id)
+        response.set_cookie(key="recce_user_id", value=user_id)
     return response
 
 
@@ -212,8 +354,8 @@ async def disable_cache(request: Request, call_next):
     response = await call_next(request)
 
     # disable cache for '/' and '/index.html'
-    if request.url.path in ['/', '/index.html']:
-        response.headers['Cache-Control'] = 'no-store'
+    if request.url.path in ["/", "/index.html"]:
+        response.headers["Cache-Control"] = "no-store"
 
     return response
 
@@ -224,24 +366,40 @@ async def health_check(request: Request):
 
 
 class RecceInstanceInfoOut(BaseModel):
+    server_mode: RecceServerMode
     read_only: bool
+    preview: bool
+    single_env: bool
     authed: bool
+    cloud_instance: bool
     lifetime_expired_at: Optional[datetime] = None
+    share_url: Optional[str] = None
+    session_id: Optional[str] = None
+    organization_name: Optional[str] = None
+    web_url: Optional[str] = None
 
 
 @app.get("/api/instance-info", response_model=RecceInstanceInfoOut, response_model_exclude_none=True)
 async def recce_instance_info():
     app_state: AppState = app.state
     flag = app_state.flag
-    read_only = flag.get('read_only', False)
+    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 {
+        "server_mode": app_state.command,
         "read_only": read_only,
+        "preview": flag.get("preview", False),
+        "single_env": single_env,
         "authed": True if api_token else False,
+        "cloud_instance": is_recce_cloud_instance(),
         "lifetime_expired_at": app_state.lifetime_expired_at,  # UTC timezone
+        "share_url": app_state.share_url,
+        "session_id": app_state.state_loader.session_id if app_state.state_loader else None,
+        "organization_name": app_state.organization_name,
+        "web_url": app_state.web_url,
         # TODO: Add more instance info which won't change during the instance lifecycle
         # review_mode
         # cloud_mode
@@ -257,16 +415,9 @@ 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
+    app.state.flag["show_relaunch_hint"] = False
 
 
 @app.get("/api/info")
@@ -275,7 +426,8 @@ async def get_info():
     Get the information of the current context.
     """
     context = default_context()
-    demo = os.environ.get('DEMO', False)
+    demo = os.environ.get("DEMO", False)
+    is_codespace = is_github_codespace()
 
     if demo:
         state = context.export_demo_state()
@@ -293,25 +445,27 @@ async def get_info():
 
     try:
         info = {
-            'state_metadata': state_metadata,
-            'adapter_type': context.adapter_type,
-            'review_mode': context.review_mode,
-            'git': state.git.to_dict() if state.git else None,
-            'pull_request': state.pull_request.to_dict() if state.pull_request else None,
-            'lineage': lineage_diff,
-            'demo': bool(demo),
-            'cloud_mode': context.state_loader.cloud_mode,
-            'file_mode': context.state_loader.state_file is not None,
-            'filename': filename,
-            'support_tasks': support_tasks,
+            "state_metadata": state_metadata,
+            "adapter_type": context.adapter_type,
+            "review_mode": context.review_mode,
+            "git": state.git.to_dict() if state.git else None,
+            "pull_request": state.pull_request.to_dict() if state.pull_request else None,
+            "lineage": lineage_diff,
+            "demo": bool(demo),
+            "codespace": bool(is_codespace),
+            "cloud_mode": context.state_loader.cloud_mode,
+            "file_mode": context.state_loader.state_file is not None,
+            "filename": filename,
+            "support_tasks": support_tasks,
         }
 
-        if context.adapter_type == 'sqlmesh':
+        if context.adapter_type == "sqlmesh":
             from recce.adapter.sqlmesh_adapter import SqlmeshAdapter
+
             sqlmesh_adapter: SqlmeshAdapter = context.adapter
-            info['sqlmesh'] = {
-                'base_env': sqlmesh_adapter.base_env.name,
-                'current_env': sqlmesh_adapter.curr_env.name,
+            info["sqlmesh"] = {
+                "base_env": sqlmesh_adapter.base_env.name,
+                "current_env": sqlmesh_adapter.curr_env.name,
             }
 
         return info
@@ -320,32 +474,40 @@ async def get_info():
 
 
 class CllIn(BaseModel):
-    params: Dict
+    node_id: Optional[str] = None
+    column: Optional[str] = None
+    change_analysis: Optional[bool] = False
+    no_cll: Optional[bool] = False
+    no_upstream: Optional[bool] = False
+    no_downstream: Optional[bool] = False
 
 
 class CllOutput(BaseModel):
-    current: Dict
+    current: CllData
 
 
 @app.post("/api/cll", response_model=CllOutput)
 async def column_level_lineage_by_node(cll_input: CllIn):
     from recce.adapter.dbt_adapter import DbtAdapter
-    dbt_adapter: DbtAdapter = default_context().adapter
 
-    try:
-        # TODO: Add support for by the node and column
-        result = dbt_adapter.get_cll_by_node_id(cll_input.params.get('node_id'))
-    except Exception as e:
-        raise HTTPException(status_code=400, detail=str(e))
+    dbt_adapter: DbtAdapter = default_context().adapter
+    cll = dbt_adapter.get_cll(
+        node_id=cll_input.node_id,
+        column=cll_input.column,
+        change_analysis=cll_input.change_analysis,
+        no_upstream=cll_input.no_upstream,
+        no_downstream=cll_input.no_downstream,
+        no_cll=cll_input.no_cll,
+    )
 
-    return CllOutput(current=result)
+    return CllOutput(current=cll)
 
 
 class SelectNodesInput(BaseModel):
     select: Optional[str] = None
     exclude: Optional[str] = None
     packages: Optional[list[str]] = None
-    view_mode: Optional[Literal['all', 'changed_models']] = None
+    view_mode: Optional[Literal["all", "changed_models"]] = None
 
 
 class SelectNodesOutput(BaseModel):
@@ -356,8 +518,8 @@ class SelectNodesOutput(BaseModel):
 async def select_nodes(input: SelectNodesInput):
     context = default_context()
 
-    if context.adapter_type != 'dbt':
-        raise HTTPException(status_code=400, detail='Only dbt adapter is supported')
+    if context.adapter_type != "dbt":
+        raise HTTPException(status_code=400, detail="Only dbt adapter is supported")
 
     try:
         nodes = context.adapter.select_nodes(
@@ -366,7 +528,7 @@ async def select_nodes(input: SelectNodesInput):
             packages=input.packages,
             view_mode=input.view_mode,
         )
-        nodes = [node for node in nodes if not node.startswith('test.')]
+        nodes = [node for node in nodes if not node.startswith("test.")]
         return SelectNodesOutput(nodes=nodes)
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
@@ -377,9 +539,9 @@ async def get_columns(model_id: str):
     context = default_context()
     try:
         return {
-            'model': {
-                'base': context.get_model(model_id, base=True),
-                'current': context.get_model(model_id, base=False)
+            "model": {
+                "base": context.get_model(model_id, base=True),
+                "current": context.get_model(model_id, base=False),
             }
         }
     except Exception as e:
@@ -394,12 +556,12 @@ async def save_handler():
     try:
         # Sync the state file
         context = default_context()
-        log_api_event('save', dict(state_loader_mode=context.state_loader_mode()))
+        log_api_event("save", dict(state_loader_mode=context.state_loader_mode()))
         state_loader = context.state_loader
         if not state_loader.cloud_mode and state_loader.state_file is None:
-            raise RecceException('Not file mode or cloud mode')
+            raise RecceException("Not file mode or cloud mode")
 
-        context.sync_state('overwrite')
+        context.sync_state("overwrite")
     except RecceException as e:
         raise HTTPException(status_code=400, detail=e.message)
 
@@ -415,33 +577,33 @@ def saveas_or_rename(input: SaveAsOrRenameInput, rename: bool = False):
     context = default_context()
     state_loader = context.state_loader
     if state_loader.cloud_mode:
-        raise RecceException('Cloud mode does not support rename')
+        raise RecceException("Cloud mode does not support rename")
 
     new_filename = input.filename
     if os.path.dirname(new_filename):
-        raise RecceException('The new filename should not contain directory')
-    if not new_filename.endswith('.json'):
-        raise RecceException('The new filename should end with .json')
+        raise RecceException("The new filename should not contain directory")
+    if not new_filename.endswith(".json"):
+        raise RecceException("The new filename should end with .json")
 
     old_path = state_loader.state_file
     if old_path:
         old_dir = os.path.dirname(state_loader.state_file)
         old_filename = os.path.basename(state_loader.state_file)
         if old_filename == new_filename:
-            raise RecceException('The new filename is the same as the current filename')
+            raise RecceException("The new filename is the same as the current filename")
         new_path = os.path.join(old_dir, new_filename)
     else:
         new_path = new_filename
 
     if os.path.exists(new_path):
         if os.path.isdir(new_path):
-            raise HTTPException(status_code=400, detail=f'The file {new_path} exists and is a directory')
+            raise HTTPException(status_code=400, detail=f"The file {new_path} exists and is a directory")
 
         if not input.overwrite:
-            raise HTTPException(status_code=409, detail=f'The file {new_filename} already exists')
+            raise HTTPException(status_code=409, detail=f"The file {new_filename} already exists")
 
     state_loader.state_file = new_path
-    context.sync_state('overwrite')
+    context.sync_state("overwrite")
     if rename and os.path.exists(old_path):
         os.remove(old_path)
 
@@ -453,7 +615,7 @@ async def save_as_handler(input: SaveAsOrRenameInput):
     """
     context = default_context()
     try:
-        log_api_event('saveas', dict(state_loader_mode=context.state_loader_mode()))
+        log_api_event("saveas", dict(state_loader_mode=context.state_loader_mode()))
         saveas_or_rename(input, rename=False)
     except RecceException as e:
         raise HTTPException(status_code=400, detail=e.message)
@@ -466,7 +628,7 @@ async def rename_handler(input: SaveAsOrRenameInput):
     """
     context = default_context()
     try:
-        log_api_event('rename', dict(state_loader_mode=context.state_loader_mode()))
+        log_api_event("rename", dict(state_loader_mode=context.state_loader_mode()))
         saveas_or_rename(input, rename=True)
     except RecceException as e:
         raise HTTPException(status_code=400, detail=e.message)
@@ -479,7 +641,7 @@ async def export_handler():
     """
     context = default_context()
     try:
-        log_api_event('export', dict(state_loader_mode=context.state_loader_mode()))
+        log_api_event("export", dict(state_loader_mode=context.state_loader_mode()))
         return context.export_state().to_json()
     except RecceException as e:
         raise HTTPException(status_code=400, detail=e.message)
@@ -487,17 +649,16 @@ async def export_handler():
 
 @app.post("/api/import", status_code=200)
 async def import_handler(
-    file: Annotated[UploadFile, Form()],
-    checks_only: Annotated[bool, Form()],
-    background_tasks: BackgroundTasks
+    file: Annotated[UploadFile, Form()], checks_only: Annotated[bool, Form()], background_tasks: BackgroundTasks
 ):
     """
     Import the recce state from the client.
     """
     from recce.state import RecceState
+
     context = default_context()
     try:
-        log_api_event('import', dict(state_loader_mode=context.state_loader_mode()))
+        log_api_event("import", dict(state_loader_mode=context.state_loader_mode()))
         content = await file.read()
         state = RecceState.from_json(content)
 
@@ -531,16 +692,19 @@ async def sync_handler(input: SyncStateInput, response: Response, background_tas
     context = default_context()
     state_loader = context.state_loader
     method = input.method
-    log_api_event('sync', dict(
-        state_loader_mode=context.state_loader_mode(),
-        method=method,
-    ))
+    log_api_event(
+        "sync",
+        dict(
+            state_loader_mode=context.state_loader_mode(),
+            method=method,
+        ),
+    )
 
     if not method:
         is_conflict = state_loader.check_conflict()
         if is_conflict:
-            raise HTTPException(status_code=409, detail='Conflict detected')
-        method = 'overwrite'
+            raise HTTPException(status_code=409, detail="Conflict detected")
+        method = "overwrite"
 
     is_syncing = state_loader.state_lock.locked()
     if is_syncing:
@@ -590,7 +754,7 @@ async def share_state():
     context = default_context()
     state_loader = context.state_loader
 
-    file_name = 'recce_state.json'
+    file_name = "recce_state.json"
     if state_loader.state_file:
         file_name = os.path.basename(state_loader.state_file)
 
@@ -626,8 +790,8 @@ async def websocket_endpoint(websocket: WebSocket):
     try:
         while True:
             data = await websocket.receive_text()
-            if data == 'ping':
-                await websocket.send_text('pong')
+            if data == "ping":
+                await websocket.send_text("pong")
     except WebSocketDisconnect:
         clients.remove(websocket)
 
@@ -637,9 +801,37 @@ async def broadcast(data: str):
         await client.send_text(data)
 
 
-api_prefix = '/api'
+@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,
+    }
+
+
+@app.get("/api/users")
+async def get_user_info():
+    from recce.connect_to_cloud import RecceCloud
+
+    context = default_context()
+    user_token = get_recce_api_token() or context.state_loader.token
+    cloud = RecceCloud(user_token)
+    try:
+        user_info = cloud.get_user_info()
+        return user_info
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+api_prefix = "/api"
 app.include_router(check_router, prefix=api_prefix)
 app.include_router(run_router, prefix=api_prefix)
 
-static_folder_path = Path(__file__).parent / 'data'
+static_folder_path = Path(__file__).parent / "data"
 app.mount("/", StaticFiles(directory=static_folder_path, html=True), name="static")