weread

Step 1: Check login status
    ├── Not logged in → Prompt user to scan QR code → Wait for login completion
    └── Logged in → Proceed to Step 2

Step 2: Check reading progress
    ├── No progress or finished → Pick first book from New Book ranking → Record status → Proceed to Step 3
    └── Unfinished book → Resume reading → Proceed to Step 3

Step 3: Auto reading (1 minute per session)
    ├── Every 3-5s scroll down 100px
    ├── See "Next Chapter" → Click, continue reading
    ├── See "Book Complete" + "Mark as Finished" → Click mark as finished → Record status as finished
    └── Time's up → Save progress, wait for next session

https://weread.qq.com/web/reader/{BOOK_ID}k{CHAPTER_HASH}

# Check if login is needed
login_needed = js("""
    const loginBtn = document.querySelector('[class*=login], [class*=Login]');
    const qrCode = document.querySelector('[class*=qrcode], [class*=QRCode]');
    return !!(loginBtn || qrCode);
""")

# Wait for login to complete
import time

def wait_for_login(timeout=120):
    start = time.time()
    while time.time() - start < timeout:
        # Check if still on login page
        on_login = js("""
            const loginBtn = document.querySelector('[class*=login], [class*=Login]');
            return !!loginBtn;
        """)
        if not on_login:
            return True
        time.sleep(2)
    return False

{
  "book": {
    "title": "book title",
    "author": "author",
    "url": "current chapter URL"
  },
  "progress": {
    "status": "reading | finished",
    "currentChapter": "chapter name",
    "completedChapters": ["list of chapters"],
    "lastReadTime": "2026-05-06"
  }
}

import json
import os

PROGRESS_FILE = "progress.json"

def load_progress():
    if not os.path.exists(PROGRESS_FILE):
        return None
    with open(PROGRESS_FILE, "r", encoding="utf-8") as f:
        return json.load(f)

def save_progress(book_title, author, url, chapter, completed_chapters, status="reading"):
    progress = {
        "book": {
            "title": book_title,
            "author": author,
            "url": url
        },
        "progress": {
            "status": status,
            "currentChapter": chapter,
            "completedChapters": completed_chapters,
            "lastReadTime": time.strftime("%Y-%m-%d")
        }
    }
    with open(PROGRESS_FILE, "w", encoding="utf-8") as f:
        json.dump(progress, f, ensure_ascii=False, indent=2)

def should_pick_new_book():
    progress = load_progress()
    if progress is None:
        return True  # never read before
    if progress["progress"]["status"] == "finished":
        return True  # already finished
    return False  # has an unfinished book

def pick_book_from_new_ranking():
    # Navigate to New Book ranking
    new_tab("https://weread.qq.com/web/category/newbook")
    wait_for_load()
    time.sleep(2)

    # Scroll to top
    js("window.scrollTo(0, 0)")
    time.sleep(1)

    # Click the first book
    first_book = js("""
        const allElements = document.querySelectorAll("[class*=title]");
        for (const el of allElements) {
            const text = el.textContent.trim();
            if (text.length > 3 && text.length < 50 && !text.includes("榜")) {
                const rect = el.getBoundingClientRect();
                return {
                    title: text,
                    x: rect.x + rect.width / 2,
                    y: rect.y + rect.height / 2
                };
            }
        }
        return null;
    """)

    if first_book:
        click_at_xy(first_book["x"], first_book["y"])
        wait_for_load()
        time.sleep(1)

    return first_book["title"] if first_book else None

import random

def scroll_reading(duration=360):
    start_time = time.time()
    chapters_read = []

    while time.time() - start_time < duration:
        # Scroll down 100px
        js("window.scrollBy(0, 100)")

        # Random wait 3-5 seconds
        wait_time = random.uniform(3, 5)
        time.sleep(wait_time)

        # Check if "Next Chapter" button is visible
        next_chapter = find_next_chapter_button()
        if next_chapter and next_chapter["visible"]:
            # Record current chapter
            current = get_current_chapter()
            chapters_read.append(current)
            # Click next chapter
            click_at_xy(next_chapter["x"], next_chapter["y"])
            wait_for_load()
            time.sleep(1)
            continue

        # Check if "Book Complete" and "Mark as Finished"
        finished = check_book_finished()
        if finished:
            click_mark_finished()
            return chapters_read, True  # True = book finished

    return chapters_read, False  # False = time's up, book not finished

def find_next_chapter_button():
    buttons = js("""
        const items = [];
        const elements = document.querySelectorAll("button, a, [role=button], [class*=next], [class*=Next]");
        elements.forEach(el => {
            const text = el.textContent.trim();
            if (text.includes("下一章")) {
                const rect = el.getBoundingClientRect();
                items.push({
                    text: text,
                    x: rect.x + rect.width/2,
                    y: rect.y + rect.height/2,
                    visible: rect.top < window.innerHeight && rect.bottom > 0
                });
            }
        });
        return items;
    """)
    return buttons[0] if buttons else None

def get_current_chapter():
    return js("""
        const title = document.title;
        const parts = title.split(" - ");
        return parts.length > 1 ? parts[1] : "未知章节";
    """)

def check_book_finished():
    return js("""
        const elements = document.querySelectorAll("[class*=finish], [class*=complete], [class*=end]");
        for (const el of elements) {
            const text = el.textContent.trim();
            if (text.includes("全书完") || text.includes("已读完")) {
                return true;
            }
        }
        return false;
    """)

def click_mark_finished():
    button = js("""
        const elements = document.querySelectorAll("button, [role=button]");
        for (const el of elements) {
            const text = el.textContent.trim();
            if (text.includes("标记读完") || text.includes("标记已读")) {
                const rect = el.getBoundingClientRect();
                return {x: rect.x + rect.width/2, y: rect.y + rect.height/2};
            }
        }
        return null;
    """)
    if button:
        click_at_xy(button["x"], button["y"])
        wait_for_load()
        time.sleep(1)