janet-cli 0.2.7__py3-none-any.whl → 0.2.33__py3-none-any.whl

janet/__init__.py

@@ -1,3 +1,3 @@
 """Janet AI CLI - Sync tickets to local markdown files."""
 
-__version__ = "0.2.7"
+__version__ = "0.2.33"

janet/api/client.py

@@ -126,3 +126,39 @@ class APIClient:
             raise NetworkError(f"Request timeout after {self.timeout}s")
         except Exception as e:
             raise NetworkError(f"Network error: {e}")
+
+    def put(
+        self, endpoint: str, data: Optional[Dict] = None, include_org: bool = False, **kwargs
+    ) -> Dict:
+        """
+        Make PUT request.
+
+        Args:
+            endpoint: API endpoint (relative to base_url)
+            data: Request body data
+            include_org: Whether to include organization ID header
+            **kwargs: Additional arguments for httpx.put
+
+        Returns:
+            Response JSON
+
+        Raises:
+            NetworkError: If request fails
+        """
+        url = f"{self.base_url}{endpoint}"
+        headers = self._get_headers(include_org=include_org)
+
+        try:
+            response = httpx.put(
+                url, headers=headers, json=data, timeout=self.timeout, **kwargs
+            )
+            response.raise_for_status()
+            return response.json()
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 401:
+                raise AuthenticationError("Authentication failed. Please log in again.")
+            raise NetworkError(f"API request failed: {e.response.status_code} {e.response.text}")
+        except httpx.TimeoutException:
+            raise NetworkError(f"Request timeout after {self.timeout}s")
+        except Exception as e:
+            raise NetworkError(f"Network error: {e}")

janet/api/projects.py

@@ -55,3 +55,23 @@ class ProjectAPI(APIClient):
             raise Exception(response.get("error", "Failed to fetch project"))
 
         return response.get("project", {})
+
+    def get_project_columns(self, project_id: str) -> List[Dict]:
+        """
+        Get kanban columns (valid statuses) for a project.
+
+        Args:
+            project_id: Project ID
+
+        Returns:
+            List of column dictionaries with status_value, is_resolved, column_order
+
+        Raises:
+            NetworkError: If API request fails
+        """
+        response = self.get(f"/api/v1/projects/{project_id}/columns", include_org=True)
+
+        if not response.get("success"):
+            raise Exception(response.get("error", "Failed to fetch project columns"))
+
+        return response.get("columns", [])

janet/api/tickets.py

@@ -75,6 +75,30 @@ class TicketAPI(APIClient):
 
         return response.get("ticket", {})
 
+    def cli_batch_fetch(self, ticket_ids: List[str]) -> List[Dict]:
+        """
+        Fetch multiple tickets using CLI unlimited endpoint - NO 500 LIMIT.
+
+        Args:
+            ticket_ids: List of ticket IDs (unlimited)
+
+        Returns:
+            List of ticket dictionaries
+
+        Raises:
+            NetworkError: If API request fails
+        """
+        if not ticket_ids:
+            return []
+
+        data = {"ticket_ids": ticket_ids}
+        response = self.post("/api/v1/cli/tickets/batch", data=data, include_org=True)
+
+        if not response.get("success"):
+            raise Exception(response.get("error", "Failed to batch fetch tickets"))
+
+        return response.get("tickets", [])
+
     def batch_fetch(self, ticket_ids: List[str]) -> List[Dict]:
         """
         Fetch multiple tickets in one request.
@@ -141,7 +165,7 @@ class TicketAPI(APIClient):
             NetworkError: If API request fails
         """
         response = self.get(
-            f"/api/v1/tickets/{ticket_id}/attachments", include_org=True
+            f"/api/v1/attachments/record/ticket/{ticket_id}", include_org=True
         )
 
         if not response.get("success"):
@@ -151,3 +175,146 @@ class TicketAPI(APIClient):
             "direct_attachments": response.get("direct_attachments", []),
             "indirect_attachments": response.get("indirect_attachments", []),
         }
+
+    def batch_fetch_attachments(self, ticket_ids: List[str]) -> Dict[str, Dict]:
+        """
+        Batch fetch attachments for multiple tickets in one request.
+
+        Args:
+            ticket_ids: List of ticket IDs
+
+        Returns:
+            Dictionary mapping ticket_id to attachments dict
+
+        Raises:
+            NetworkError: If API request fails
+        """
+        if not ticket_ids:
+            return {}
+
+        response = self.post(
+            "/api/v1/attachments/batch/tickets",
+            data={"ticket_ids": ticket_ids},
+            include_org=True,
+        )
+
+        if not response.get("success"):
+            raise Exception(response.get("error", "Failed to batch fetch attachments"))
+
+        return response.get("attachments_by_ticket", {})
+
+    def create_ticket(
+        self,
+        project_id: str,
+        title: str,
+        description: Optional[str] = None,
+        status: Optional[str] = None,
+        priority: Optional[str] = None,
+        issue_type: Optional[str] = None,
+        assignees: Optional[List[str]] = None,
+        labels: Optional[List[str]] = None,
+    ) -> Dict:
+        """
+        Create a new ticket.
+
+        Args:
+            project_id: Project ID to create ticket in
+            title: Ticket title (required)
+            description: Ticket description
+            status: Status (e.g., "To Do", "In Progress")
+            priority: Priority (Low, Medium, High, Critical)
+            issue_type: Type (Task, Bug, Story, Epic)
+            assignees: List of assignee emails
+            labels: List of label strings
+
+        Returns:
+            Dictionary with created ticket details including ticket_key
+
+        Raises:
+            NetworkError: If API request fails
+        """
+        payload: Dict = {
+            "project_id": project_id,
+            "title": title,
+        }
+
+        if description:
+            payload["description"] = description
+        if status:
+            payload["status"] = status
+        if priority:
+            payload["priority"] = priority
+        if issue_type:
+            payload["issue_type"] = issue_type
+        if assignees:
+            payload["assignees"] = assignees
+        if labels:
+            payload["labels"] = labels
+
+        response = self.post("/api/v1/tickets", data=payload, include_org=True)
+
+        if not response.get("success"):
+            raise Exception(response.get("error", "Failed to create ticket"))
+
+        return response
+
+    def update_ticket(
+        self,
+        ticket_id: str,
+        title: Optional[str] = None,
+        description: Optional[str] = None,
+        status: Optional[str] = None,
+        priority: Optional[str] = None,
+        issue_type: Optional[str] = None,
+        assignees: Optional[List[str]] = None,
+        labels: Optional[List[str]] = None,
+        due_date: Optional[str] = None,
+    ) -> Dict:
+        """
+        Update an existing ticket.
+
+        Args:
+            ticket_id: Ticket ID (UUID) to update
+            title: New ticket title
+            description: New ticket description
+            status: New status (e.g., "To Do", "In Progress", "Done")
+            priority: New priority (Low, Medium, High, Critical)
+            issue_type: New type (Task, Bug, Story, Epic)
+            assignees: New list of assignee emails (replaces existing)
+            labels: New list of labels (replaces existing)
+            due_date: New due date (ISO format)
+
+        Returns:
+            Dictionary with update result
+
+        Raises:
+            NetworkError: If API request fails
+        """
+        payload: Dict = {}
+
+        if title is not None:
+            payload["title"] = title
+        if description is not None:
+            payload["description"] = description
+        if status is not None:
+            payload["status"] = status
+        if priority is not None:
+            payload["priority"] = priority
+        if issue_type is not None:
+            payload["issue_type"] = issue_type
+        if assignees is not None:
+            payload["assignees"] = assignees
+        if labels is not None:
+            payload["labels"] = labels
+        if due_date is not None:
+            payload["due_date"] = due_date
+
+        if not payload:
+            raise Exception("No fields to update")
+
+        response = self.put(f"/api/v1/tickets/{ticket_id}", data=payload, include_org=True)
+
+        if not response.get("success"):
+            raise Exception(response.get("error", "Failed to update ticket"))
+
+        return response

janet/auth/callback_server.py

@@ -1,5 +1,6 @@
 """Local HTTP server for OAuth callback."""
 
+import os
 import threading
 import urllib.parse
 from http.server import BaseHTTPRequestHandler, HTTPServer
@@ -282,6 +283,8 @@ class CallbackServer:
             port: Port to listen on (default: 8765)
         """
         self.port = port
+        # Use localhost for dev (WorkOS staging), 127.0.0.1 for prod
+        self.host = os.getenv("OAUTH_CALLBACK_HOST", "127.0.0.1")
         self.server: Optional[HTTPServer] = None
         self.thread: Optional[threading.Thread] = None
 
@@ -292,10 +295,9 @@ class CallbackServer:
         CallbackHandler.error = None
 
         # Try to start server on specified port, fallback to next ports if busy
-        # Use 127.0.0.1 instead of localhost for WorkOS compatibility
         for attempt_port in range(self.port, self.port + 10):
             try:
-                self.server = HTTPServer(("127.0.0.1", attempt_port), CallbackHandler)
+                self.server = HTTPServer((self.host, attempt_port), CallbackHandler)
                 self.port = attempt_port
                 break
             except OSError:
@@ -356,5 +358,4 @@ class CallbackServer:
         Returns:
             Redirect URI string
         """
-        # WorkOS requires 127.0.0.1 (not localhost) for http in production
-        return f"http://127.0.0.1:{self.port}/callback"
+        return f"http://{self.host}:{self.port}/callback"

janet/auth/oauth_flow.py

@@ -267,7 +267,6 @@ class OAuthFlow:
 
             # Update tokens
             self._save_tokens(tokens)
-            print_info("Access token refreshed")
 
         except httpx.HTTPStatusError as e:
             error_detail = e.response.text

janet/auth/token_manager.py

@@ -19,16 +19,18 @@ class TokenManager:
         """
         self.config_manager = config_manager
 
-    def get_access_token(self) -> str:
+    def get_access_token(self, auto_refresh: bool = True) -> str:
         """
-        Get current access token.
+        Get current access token, refreshing if expired.
+
+        Args:
+            auto_refresh: Automatically refresh token if expired
 
         Returns:
             Access token string
 
         Raises:
-            AuthenticationError: If not authenticated
-            TokenExpiredError: If token is expired
+            AuthenticationError: If not authenticated or refresh fails
         """
         config = self.config_manager.get()
 
@@ -36,7 +38,12 @@ class TokenManager:
             raise AuthenticationError("Not authenticated. Run 'janet login' first.")
 
         if self.is_token_expired():
-            raise TokenExpiredError("Access token has expired. Attempting refresh...")
+            if auto_refresh:
+                self.refresh_access_token()
+                # Re-read config after refresh
+                config = self.config_manager.get()
+            else:
+                raise TokenExpiredError("Access token has expired.")
 
         return config.auth.access_token
 
@@ -90,3 +97,22 @@ class TokenManager:
         """
         config = self.config_manager.get()
         return config.auth.user_id
+
+    def refresh_access_token(self) -> None:
+        """
+        Refresh the access token using the refresh token.
+
+        Raises:
+            AuthenticationError: If refresh fails or no refresh token
+        """
+        from janet.auth.oauth_flow import OAuthFlow
+
+        config = self.config_manager.get()
+
+        if not config.auth.refresh_token:
+            raise AuthenticationError(
+                "No refresh token available. Please run 'janet login' to re-authenticate."
+            )
+
+        oauth_flow = OAuthFlow(self.config_manager)
+        oauth_flow.refresh_token()