1#!/usr/bin/env python3
2"""
3shadcn/ui Component Installer
4 
5Add shadcn/ui components to project with automatic dependency handling.
6Wraps shadcn CLI for programmatic component installation.
7"""
8 
9import argparse
10import json
11import subprocess
12import sys
13from pathlib import Path
14from typing import List, Optional
15 
16 
17class ShadcnInstaller:
18    """Handle shadcn/ui component installation."""
19 
20    def __init__(self, project_root: Optional[Path] = None, dry_run: bool = False):
21        """
22        Initialize installer.
23 
24        Args:
25            project_root: Project root directory (default: current directory)
26            dry_run: If True, show actions without executing
27        """
28        self.project_root = project_root or Path.cwd()
29        self.dry_run = dry_run
30        self.components_json = self.project_root / "components.json"
31 
32    def check_shadcn_config(self) -> bool:
33        """
34        Check if shadcn is initialized in project.
35 
36        Returns:
37            True if components.json exists
38        """
39        return self.components_json.exists()
40 
41    def get_installed_components(self) -> List[str]:
42        """
43        Get list of already installed components.
44 
45        Returns:
46            List of installed component names
47        """
48        if not self.check_shadcn_config():
49            return []
50 
51        try:
52            with open(self.components_json) as f:
53                config = json.load(f)
54 
55            components_dir = self.project_root / config.get("aliases", {}).get(
56                "components", "components"
57            ).replace("@/", "")
58            ui_dir = components_dir / "ui"
59 
60            if not ui_dir.exists():
61                return []
62 
63            return [f.stem for f in ui_dir.glob("*.tsx") if f.is_file()]
64        except (json.JSONDecodeError, KeyError, OSError):
65            return []
66 
67    def add_components(
68        self, components: List[str], overwrite: bool = False
69    ) -> tuple[bool, str]:
70        """
71        Add shadcn/ui components.
72 
73        Args:
74            components: List of component names to add
75            overwrite: If True, overwrite existing components
76 
77        Returns:
78            Tuple of (success, message)
79        """
80        if not components:
81            return False, "No components specified"
82 
83        if not self.check_shadcn_config():
84            return (
85                False,
86                "shadcn not initialized. Run 'npx shadcn@latest init' first",
87            )
88 
89        # Check which components already exist
90        installed = self.get_installed_components()
91        already_installed = [c for c in components if c in installed]
92 
93        if already_installed and not overwrite:
94            return (
95                False,
96                f"Components already installed: {', '.join(already_installed)}. "
97                "Use --overwrite to reinstall",
98            )
99 
100        # Build command
101        cmd = ["npx", "shadcn@latest", "add"] + components
102 
103        if overwrite:
104            cmd.append("--overwrite")
105 
106        if self.dry_run:
107            return True, f"Would run: {' '.join(cmd)}"
108 
109        # Execute command
110        try:
111            result = subprocess.run(
112                cmd,
113                cwd=self.project_root,
114                capture_output=True,
115                text=True,
116                check=True,
117            )
118 
119            success_msg = f"Successfully added components: {', '.join(components)}"
120            if result.stdout:
121                success_msg += f"\n\nOutput:\n{result.stdout}"
122 
123            return True, success_msg
124 
125        except subprocess.CalledProcessError as e:
126            error_msg = f"Failed to add components: {e.stderr or e.stdout or str(e)}"
127            return False, error_msg
128        except FileNotFoundError:
129            return False, "npx not found. Ensure Node.js is installed"
130 
131    def add_all_components(self, overwrite: bool = False) -> tuple[bool, str]:
132        """
133        Add all available shadcn/ui components.
134 
135        Args:
136            overwrite: If True, overwrite existing components
137 
138        Returns:
139            Tuple of (success, message)
140        """
141        if not self.check_shadcn_config():
142            return (
143                False,
144                "shadcn not initialized. Run 'npx shadcn@latest init' first",
145            )
146 
147        cmd = ["npx", "shadcn@latest", "add", "--all"]
148 
149        if overwrite:
150            cmd.append("--overwrite")
151 
152        if self.dry_run:
153            return True, f"Would run: {' '.join(cmd)}"
154 
155        try:
156            result = subprocess.run(
157                cmd,
158                cwd=self.project_root,
159                capture_output=True,
160                text=True,
161                check=True,
162            )
163 
164            success_msg = "Successfully added all components"
165            if result.stdout:
166                success_msg += f"\n\nOutput:\n{result.stdout}"
167 
168            return True, success_msg
169 
170        except subprocess.CalledProcessError as e:
171            error_msg = f"Failed to add all components: {e.stderr or e.stdout or str(e)}"
172            return False, error_msg
173        except FileNotFoundError:
174            return False, "npx not found. Ensure Node.js is installed"
175 
176    def list_installed(self) -> tuple[bool, str]:
177        """
178        List installed components.
179 
180        Returns:
181            Tuple of (success, message with component list)
182        """
183        if not self.check_shadcn_config():
184            return False, "shadcn not initialized"
185 
186        installed = self.get_installed_components()
187 
188        if not installed:
189            return True, "No components installed"
190 
191        return True, f"Installed components:\n" + "\n".join(f"  - {c}" for c in sorted(installed))
192 
193 
194def main():
195    """CLI entry point."""
196    parser = argparse.ArgumentParser(
197        description="Add shadcn/ui components to your project",
198        formatter_class=argparse.RawDescriptionHelpFormatter,
199        epilog="""
200Examples:
201  # Add single component
202  python shadcn_add.py button
203 
204  # Add multiple components
205  python shadcn_add.py button card dialog
206 
207  # Add all components
208  python shadcn_add.py --all
209 
210  # Overwrite existing components
211  python shadcn_add.py button --overwrite
212 
213  # Dry run (show what would be done)
214  python shadcn_add.py button card --dry-run
215 
216  # List installed components
217  python shadcn_add.py --list
218        """,
219    )
220 
221    parser.add_argument(
222        "components",
223        nargs="*",
224        help="Component names to add (e.g., button, card, dialog)",
225    )
226 
227    parser.add_argument(
228        "--all",
229        action="store_true",
230        help="Add all available components",
231    )
232 
233    parser.add_argument(
234        "--overwrite",
235        action="store_true",
236        help="Overwrite existing components",
237    )
238 
239    parser.add_argument(
240        "--dry-run",
241        action="store_true",
242        help="Show what would be done without executing",
243    )
244 
245    parser.add_argument(
246        "--list",
247        action="store_true",
248        help="List installed components",
249    )
250 
251    parser.add_argument(
252        "--project-root",
253        type=Path,
254        help="Project root directory (default: current directory)",
255    )
256 
257    args = parser.parse_args()
258 
259    # Initialize installer
260    installer = ShadcnInstaller(
261        project_root=args.project_root,
262        dry_run=args.dry_run,
263    )
264 
265    # Handle list command
266    if args.list:
267        success, message = installer.list_installed()
268        print(message)
269        sys.exit(0 if success else 1)
270 
271    # Handle add all command
272    if args.all:
273        success, message = installer.add_all_components(overwrite=args.overwrite)
274        print(message)
275        sys.exit(0 if success else 1)
276 
277    # Handle add specific components
278    if not args.components:
279        parser.print_help()
280        sys.exit(1)
281 
282    success, message = installer.add_components(
283        args.components,
284        overwrite=args.overwrite,
285    )
286 
287    print(message)
288    sys.exit(0 if success else 1)
289 
290 
291if __name__ == "__main__":
292    main()
293