#!/usr/bin/env python3
import json
import sys

def search_chinese_books(keyword):
    with open('/root/chinese_books.json', 'r', encoding='utf-8') as f:
        chinese_books = json.load(f)
    
    results = []
    keyword_lower = keyword.lower()
    
    for book in chinese_books:
        title = book.get('title', '').lower()
        author = book.get('author', '').lower()
        publisher = book.get('publisher', '').lower()
        
        if (keyword_lower in title or 
            keyword_lower in author or 
            keyword_lower in publisher):
            results.append(book)
    
    print(f"🔍 搜索关键词: '{keyword}'")
    print(f"找到 {len(results)} 本相关书籍\n")
    
    for i, book in enumerate(results[:20], 1):  # 只显示前20个结果
        print(f"{i:2d}. 《{book.get('title', '未知标题')}》")
        print(f"    作者: {book.get('author', '未知')}")
        print(f"    出版社: {book.get('publisher', '未知')}")
        print(f"    年份: {book.get('year', '未知')} | 格式: {book.get('extension', '').upper()}")
        if book.get('description'):
            desc = book['description'][:100] + "..." if len(book['description']) > 100 else book['description']
            print(f"    简介: {desc}")
        print()
    
    if len(results) > 20:
        print(f"... 还有 {len(results) - 20} 本书籍未显示")

if __name__ == "__main__":
    if len(sys.argv) > 1:
        keyword = " ".join(sys.argv[1:])
        search_chinese_books(keyword)
    else:
        print("用法: python3 search_books.py <关键词>")
        print("示例: python3 search_books.py 三体")
        print("      python3 search_books.py 刘慈欣")