main.py

import time
import os
from utils import find_md_files
from notion_api import upload_markdown_file_to_notion, delete_notion_page_if_missing
from markdown_parser import replace_md_links
from config import BASE_DIR, RED, YELLOW, GREEN, RESET

def sync_markdown_to_notion():
    """
    Finds local Markdown files, syncs them to Notion by creating or updating pages,
    and updates internal links in each file to point to the corresponding Notion page.
    """

    # Start the timer to measure execution time.
    start_time = time.time()

    # Phase 0: Locate all Markdown (.md) files in the base directory.
    md_files = find_md_files(BASE_DIR)
    total_files = len(md_files)
    if total_files == 0:
        print(f"{YELLOW}No Markdown (.md) files found.{RESET}")
        return

    # Archive Notion pages corresponding to markdown files that no longer exist.
    deleted_pages = delete_notion_page_if_missing(md_files)
    if deleted_pages:
        print(f"{YELLOW}Archived {deleted_pages} missing pages.{RESET}")
    print(f"Found {total_files} Markdown file(s). Starting sync...")

    # Phase 1: Upload markdown files to Notion and build a mapping (filename -> Notion URL).
    md_to_notion = {}  # Mapping: filename -> Notion URL

    # Initialize counters for summary statistics.
    new_pages = 0
    updated_pages = 0
    skipped_pages = 0
    failed_uploads = 0
    failed_files = []

    # Process each markdown file for initial upload or update.
    for md_file in md_files:
        result = upload_markdown_file_to_notion(md_file, update_content=False)
        if isinstance(result, tuple):
            status, page_id = result
        else:
            status, page_id = result, None

        # Update counters based on the upload result.
        if status == "created":
            new_pages += 1
        elif status == "updated":
            updated_pages += 1
        elif status == "skipped":
            skipped_pages += 1
        elif status == "failed":
            failed_uploads += 1
            failed_files.append(md_file)
            continue  # Skip mapping for failed uploads.

        # If a valid page ID is returned, construct and store the Notion URL.
        if page_id:
            filename = os.path.basename(md_file)
            # Construct a Notion URL using filename.
            notion_url = f"https://www.notion.so/{filename}-{page_id.replace('-', '')}"
            md_to_notion[filename] = notion_url
            print(f"Mapped {filename} -> {notion_url}")
        else:
            print(f"{RED}Warning: No page ID returned for {md_file}.{RESET}")

    # Phase 2: Update page content by replacing local .md links with Notion URLs.
    for md_file in md_files:
        try:
            with open(md_file, "r", encoding="utf-8") as f:
                md_content = f.read()
        except Exception as e:
            print(f"{RED}Error reading {md_file}: {e}{RESET}")
            continue

        # Replace local .md links in the markdown content using the mapping.
        updated_content = replace_md_links(md_content, md_to_notion)

        # Update the page content on Notion.
        result = upload_markdown_file_to_notion(md_file, update_content=True, new_content=updated_content)
        if isinstance(result, tuple):
            status, _ = result
        else:
            status = result

        # Handle update result.
        if status == "updated":
            updated_pages += 1
            print(f"Updated content for {os.path.basename(md_file)}")
        elif status == "failed":
            failed_uploads += 1
            failed_files.append(md_file)

    # Calculate the total execution time.
    end_time = time.time()
    elapsed_time = end_time - start_time

    # Recalculate skipped count.
    skipped_pages = total_files - (new_pages + updated_pages + failed_uploads)

    # Display the final sync summary.
    print("\n======Sync Summary======")
    print(f"Total Files: {total_files}")

    if new_pages > 0:
        print(f"{GREEN}Created: {new_pages}{RESET}")
    if updated_pages > 0:
        print(f"{GREEN}Updated: {updated_pages}{RESET}")
    if skipped_pages > 0:
        print(f"Skipped: {skipped_pages} (Already up to date)")
    if deleted_pages > 0:
        print(f"{YELLOW}Deleted: {deleted_pages} (Archived in Notion){RESET}")

    # Report any failed uploads.
    if failed_uploads > 0:
        print(f"{RED}Failed: {failed_uploads}/{total_files}{RESET}")
        print("\nThe following files failed to sync:")
        for file in failed_files:
            print(f"{RED} - {str(file)}{RESET}")

    # Success message when all files have synced without failures.
    if (new_pages > 0 or updated_pages > 0 or deleted_pages > 0 or skipped_pages > 0) and failed_uploads == 0:
        print(f"{GREEN}All files synced successfully!{RESET}")

    # Total time taken
    hours, remainder = divmod(elapsed_time, 3600)
    minutes, seconds = divmod(remainder, 60)
    print(f"\nTotal time taken: {int(hours)}h {int(minutes)}m {seconds:.2f}s")

if __name__ == "__main__":
    try:
        sync_markdown_to_notion()
    except KeyboardInterrupt:
        print(f"\n{RED}Aborted{RESET}")