import os
import sys
from dotenv import load_dotenv

# Add parent dir to path so we can import backend packages
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from backend.wb_api import WildberriesAPI
from backend.wb_api.exceptions import WBAPIError

def test_wb_api():
    # Load environment variables
    env_path = os.path.join(os.path.dirname(__file__), '.env')
    load_dotenv(env_path)
    
    # Check if key is loaded
    api_token = os.getenv('WB_API_TOKEN')
    
    if not api_token:
        print("Error: WB_API_TOKEN must be in backend/.env")
        return

    print("Initializing WildberriesAPI...")
    api = WildberriesAPI()
    
    try:
        # Test 1: Get Content (Products)
        print("\n--- Testing Content API (Products) ---")
        cards_response = api.content.get_cards_list(limit=5)
        print("Success! Response keys:", cards_response.keys())
        
        cards = cards_response.get("cards", [])
        print(f"Found {len(cards)} products.")
        
        if cards:
            nm_id = cards[0].get("nmID")
            vendor_code = cards[0].get("vendorCode")
            print(f"First product - nmID: {nm_id}, Vendor Code: {vendor_code}")
            
        # Test 2: Get Marketplace (Orders)
        print("\n--- Testing Marketplace API (New Orders) ---")
        orders_response = api.marketplace.get_new_orders()
        orders = orders_response.get("orders", [])
        print(f"Found {len(orders)} new FBS orders.")
            
    except WBAPIError as e:
        print(f"API Error: {e}")
    except Exception as e:
        print(f"Unexpected Error: {e}")

if __name__ == '__main__':
    test_wb_api()