replypier/backend/app/services/shopify_auth_service.py

"""
Copyright 2024-2026 ReplyPier

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""

from sqlalchemy.orm import Session
from typing import Optional, Tuple, List, Dict, Any
from app.core.logger import get_logger
from app.repositories.agent_shopify_config_repository import AgentShopifyConfigRepository
from app.repositories.widget import WidgetRepository
from app.services.shopify import ShopifyService
from app.repositories.agent import AgentRepository
from fastapi import Request
from fastapi.responses import RedirectResponse, HTMLResponse
from app.models.organization import Organization
from app.core.config import settings
from urllib.parse import quote
import hmac
import hashlib
import base64

logger = get_logger(__name__)


class ShopifyAuthService:
    """Service for handling Shopify authentication and HTML generation"""
    
    @staticmethod
    async def get_shop_success_data(db: Session, shop_id: str) -> Tuple[int, Optional[str]]:
        """
        Get success page data for a shop (agent count and widget ID).
        
        Args:
            db: Database session
            shop_id: Shop ID
            
        Returns:
            Tuple of (agents_connected, widget_id)
        """
        agent_config_repository = AgentShopifyConfigRepository(db)
        configs = agent_config_repository.get_configs_by_shop(shop_id, enabled_only=True)
        agents_connected = len(configs) if configs else 0
        
        # Get widget ID from the first connected agent
        widget_id = None
        if configs and len(configs) > 0:
            widget_repo = WidgetRepository(db)
            widgets = widget_repo.get_widgets_by_agent(configs[0].agent_id)
            if widgets and len(widgets) > 0:
                widget_id = str(widgets[0].id)
        
        return agents_connected, widget_id
    
    @staticmethod
    async def verify_shop_token(db: Session, db_shop) -> bool:
        """
        Verify if a shop's access token is valid.
        
        Args:
            db: Database session
            db_shop: Shop database record
            
        Returns:
            True if token is valid, False otherwise
        """
        shopify_service = ShopifyService(db)
        validation_query = "query { shop { name } }"
        validation_result = await shopify_service._execute_graphql_async(db_shop, validation_query)
        return validation_result.get("success", False)