import os
import requests
import argparse
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# --- Configuration ---
WIKI_URL = os.getenv("WIKI_URL")
API_KEY = os.getenv("WIKI_API_KEY")

# --- GraphQL Queries & Mutations ---
GRAPHQL_LIST_ALL_PAGES_QUERY = """
query AllPages {
  pages {
    list {
      id
      path
    }
  }
}
"""

GRAPHQL_UPDATE_MUTATION = """
mutation UpdatePage($id: Int!, $content: String, $description: String, $tags: [String], $title: String) {
  pages {
    update(id: $id, content: $content, description: $description, tags: $tags, title: $title) {
      responseResult { succeeded, errorCode, slug, message }
      page { id, path, title }
    }
  }
}
"""

def run_graphql_query(query, variables):
    """Helper function to run a GraphQL query."""
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    try:
        response = requests.post(f"{WIKI_URL}/graphql", headers=headers, json={"query": query, "variables": variables}, timeout=15)
        response.raise_for_status()
        result = response.json()
        if 'errors' in result:
            print("❌ GraphQL API returned errors:")
            for error in result['errors']:
                print(f"  - {error.get('message')}")
            return None
        return result
    except requests.exceptions.RequestException as e:
        print(f"❌ An error occurred: {e}")
        return None

def get_page_id(path_to_find, locale):
    """Get the ID of a page by fetching all pages and searching for its path."""
    print("🔎 Fetching all pages to find the correct ID...")
    result = run_graphql_query(GRAPHQL_LIST_ALL_PAGES_QUERY, {})
    if not result:
        print("❌ Failed to fetch the page list.")
        return None

    all_pages = result.get('data', {}).get('pages', {}).get('list', [])
    if not all_pages:
        print("❌ No pages were found on the wiki.")
        return None

    for page in all_pages:
        if page.get('path') == path_to_find:
            page_id = page.get('id')
            print(f"📄 Found page with ID: {page_id} for path: '{path_to_find}'")
            return page_id

    print(f"❌ A page with the path '{path_to_find}' was not found.")
    return None

def edit_wiki_page(path, content_file, new_title, locale):
    """Updates an existing wiki page."""
    if not all([API_KEY, WIKI_URL]):
        print("❌ Error: WIKI_API_KEY and/or WIKI_URL not found in environment variables.")
        return

    page_id = get_page_id(path, locale)
    if not page_id:
        return

    variables = {
        "id": page_id,
        "description": f"Page at path {path} updated via script.",
        "tags": []
    }

    if content_file:
        try:
            with open(content_file, 'r') as f:
                variables['content'] = f.read()
            print(f"🔄 Preparing to update content from '{content_file}'...")
        except FileNotFoundError:
            print(f"❌ Error: Content file not found at '{content_file}'")
            return
    
    if new_title:
        variables['title'] = new_title
        print(f"🔄 Preparing to update title to '{new_title}'...")

    if 'content' not in variables and 'title' not in variables:
        print("✨ Nothing to update. Please provide a new title or content file.")
        return

    result = run_graphql_query(GRAPHQL_UPDATE_MUTATION, variables)
    if not result:
        return

    op_result = result.get('data', {}).get('pages', {}).get('update', {})
    response_result = op_result.get('responseResult', {})

    if response_result.get('succeeded'):
        page_info = op_result.get('page', {})
        print(f"✅ Success! Page '{page_info.get('title')}' updated.")
        print(f"🌐 View it at: {WIKI_URL}/{page_info.get('path')}")
    else:
        print(f"❌ API call failed: {response_result.get('message')}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Edit an existing page in Wiki.js.")
    parser.add_argument('--path', required=True, help='The path of the page to edit (e.g., \"home\").')
    parser.add_argument('--content-file', help='Path to a markdown file with the new content.')
    parser.add_argument('--title', help='A new title for the page.')
    parser.add_argument('--locale', default='en', help='The locale of the page (default: en).')

    args = parser.parse_args()
    edit_wiki_page(args.path, args.content_file, args.title, args.locale)