1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3"""
4UI/UX Pro Max Search - BM25 search engine for UI/UX style guides
5Usage: python search.py "<query>" [--domain <domain>] [--stack <stack>] [--max-results 3]
6       python search.py "<query>" --design-system [-p "Project Name"]
7       python search.py "<query>" --design-system --persist [-p "Project Name"] [--page "dashboard"]
8 
9Domains: style, prompt, color, chart, landing, product, ux, typography, google-fonts
10Stacks: react, nextjs, vue, svelte, astro, swiftui, react-native, flutter, nuxtjs, nuxt-ui, html-tailwind, shadcn, jetpack-compose, threejs
11 
12Persistence (Master + Overrides pattern):
13  --persist    Save design system to design-system/MASTER.md
14  --page       Also create a page-specific override file in design-system/pages/
15"""
16 
17import argparse
18import sys
19import io
20from core import CSV_CONFIG, AVAILABLE_STACKS, MAX_RESULTS, search, search_stack
21from design_system import generate_design_system, persist_design_system
22 
23# Force UTF-8 for stdout/stderr to handle emojis on Windows (cp1252 default)
24if sys.stdout.encoding and sys.stdout.encoding.lower() != 'utf-8':
25    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
26if sys.stderr.encoding and sys.stderr.encoding.lower() != 'utf-8':
27    sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')
28 
29 
30def format_output(result):
31    """Format results for Claude consumption (token-optimized)"""
32    if "error" in result:
33        return f"Error: {result['error']}"
34 
35    output = []
36    if result.get("stack"):
37        output.append(f"## UI Pro Max Stack Guidelines")
38        output.append(f"**Stack:** {result['stack']} | **Query:** {result['query']}")
39    else:
40        output.append(f"## UI Pro Max Search Results")
41        output.append(f"**Domain:** {result['domain']} | **Query:** {result['query']}")
42    output.append(f"**Source:** {result['file']} | **Found:** {result['count']} results\n")
43 
44    for i, row in enumerate(result['results'], 1):
45        output.append(f"### Result {i}")
46        for key, value in row.items():
47            value_str = str(value)
48            if len(value_str) > 300:
49                value_str = value_str[:300] + "..."
50            output.append(f"- **{key}:** {value_str}")
51        output.append("")
52 
53    return "\n".join(output)
54 
55 
56if __name__ == "__main__":
57    parser = argparse.ArgumentParser(description="UI Pro Max Search")
58    parser.add_argument("query", help="Search query")
59    parser.add_argument("--domain", "-d", choices=list(CSV_CONFIG.keys()), help="Search domain")
60    parser.add_argument("--stack", "-s", choices=AVAILABLE_STACKS, help=f"Stack-specific search. Available: {', '.join(AVAILABLE_STACKS)}")
61    parser.add_argument("--max-results", "-n", type=int, default=MAX_RESULTS, help="Max results (default: 3)")
62    parser.add_argument("--json", action="store_true", help="Output as JSON")
63    # Design system generation
64    parser.add_argument("--design-system", "-ds", action="store_true", help="Generate complete design system recommendation")
65    parser.add_argument("--project-name", "-p", type=str, default=None, help="Project name for design system output")
66    parser.add_argument("--format", "-f", choices=["ascii", "markdown"], default="ascii", help="Output format for design system")
67    # Persistence (Master + Overrides pattern)
68    parser.add_argument("--persist", action="store_true", help="Save design system to design-system/MASTER.md (creates hierarchical structure)")
69    parser.add_argument("--page", type=str, default=None, help="Create page-specific override file in design-system/pages/")
70    parser.add_argument("--output-dir", "-o", type=str, default=None, help="Output directory for persisted files (default: current directory)")
71 
72    args = parser.parse_args()
73 
74    # Design system takes priority
75    if args.design_system:
76        result = generate_design_system(
77            args.query, 
78            args.project_name, 
79            args.format,
80            persist=args.persist,
81            page=args.page,
82            output_dir=args.output_dir
83        )
84        print(result)
85        
86        # Print persistence confirmation
87        if args.persist:
88            project_slug = args.project_name.lower().replace(' ', '-') if args.project_name else "default"
89            print("\n" + "=" * 60)
90            print(f"✅ Design system persisted to design-system/{project_slug}/")
91            print(f"   📄 design-system/{project_slug}/MASTER.md (Global Source of Truth)")
92            if args.page:
93                page_filename = args.page.lower().replace(' ', '-')
94                print(f"   📄 design-system/{project_slug}/pages/{page_filename}.md (Page Overrides)")
95            print("")
96            print(f"📖 Usage: When building a page, check design-system/{project_slug}/pages/[page].md first.")
97            print(f"   If exists, its rules override MASTER.md. Otherwise, use MASTER.md.")
98            print("=" * 60)
99    # Stack search
100    elif args.stack:
101        result = search_stack(args.query, args.stack, args.max_results)
102        if args.json:
103            import json
104            print(json.dumps(result, indent=2, ensure_ascii=False))
105        else:
106            print(format_output(result))
107    # Domain search
108    else:
109        result = search(args.query, args.domain, args.max_results)
110        if args.json:
111            import json
112            print(json.dumps(result, indent=2, ensure_ascii=False))
113        else:
114            print(format_output(result))
115