install.sh

#!/bin/bash
#
# mmerd - Mermaid ERD Generator Installation Script
# Automatically generates Entity Relationship Diagrams from SQL files
#

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Installation directory
INSTALL_DIR="$HOME/.local/bin"

echo -e "${BLUE}╔══════════════════════════════════════════╗${NC}"
echo -e "${BLUE}║      mmerd Installation Script           ║${NC}"
echo -e "${BLUE}╚══════════════════════════════════════════╝${NC}"
echo

# Check for required dependencies
check_dependency() {
    if ! command -v "$1" &> /dev/null; then
        echo -e "${RED}✗ $1 is not installed${NC}"
        return 1
    else
        echo -e "${GREEN}✓ $1 is installed${NC}"
        return 0
    fi
}

echo -e "${YELLOW}Checking dependencies...${NC}"

# Check for Node.js and npm
if ! check_dependency "node"; then
    echo -e "${RED}Node.js is required. Please install it first:${NC}"
    echo "  Ubuntu/Debian: sudo apt-get install nodejs"
    echo "  macOS: brew install node"
    echo "  Or visit: https://nodejs.org/"
    exit 1
fi

if ! check_dependency "npm"; then
    echo -e "${RED}npm is required. Please install Node.js which includes npm.${NC}"
    exit 1
fi

# Check for mermaid-cli
if ! check_dependency "mmdc"; then
    echo -e "${YELLOW}Installing mermaid-cli globally...${NC}"
    npm install -g @mermaid-js/mermaid-cli
    if [ $? -ne 0 ]; then
        echo -e "${RED}Failed to install mermaid-cli. You may need to use sudo:${NC}"
        echo "  sudo npm install -g @mermaid-js/mermaid-cli"
        exit 1
    fi
    echo -e "${GREEN}✓ mermaid-cli installed successfully${NC}"
else
    echo -e "${GREEN}✓ mermaid-cli is already installed${NC}"
fi

# Create installation directory if it doesn't exist
if [ ! -d "$INSTALL_DIR" ]; then
    echo -e "${YELLOW}Creating installation directory: $INSTALL_DIR${NC}"
    mkdir -p "$INSTALL_DIR"
fi

# Check if PATH includes $HOME/.local/bin
if [[ ":$PATH:" != *":$HOME/.local/bin:"* ]]; then
    echo -e "${YELLOW}Adding $HOME/.local/bin to PATH...${NC}"
    
    # Detect shell and update appropriate config file
    if [ -n "$BASH_VERSION" ]; then
        echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.bashrc
        echo -e "${GREEN}✓ Added to ~/.bashrc${NC}"
        echo -e "${YELLOW}  Run 'source ~/.bashrc' to update current session${NC}"
    elif [ -n "$ZSH_VERSION" ]; then
        echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.zshrc
        echo -e "${GREEN}✓ Added to ~/.zshrc${NC}"
        echo -e "${YELLOW}  Run 'source ~/.zshrc' to update current session${NC}"
    else
        echo -e "${YELLOW}Please add the following to your shell configuration:${NC}"
        echo '  export PATH="$HOME/.local/bin:$PATH"'
    fi
fi

# Install mmerd scripts
echo -e "${YELLOW}Installing mmerd scripts...${NC}"

# Get the directory where this script is located
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

# Copy scripts to installation directory
for script in mmerd mmviz; do
    if [ -f "$SCRIPT_DIR/bin/$script" ]; then
        cp "$SCRIPT_DIR/bin/$script" "$INSTALL_DIR/"
        chmod +x "$INSTALL_DIR/$script"
        echo -e "${GREEN}✓ Installed $script${NC}"
    else
        echo -e "${RED}✗ $script not found in $SCRIPT_DIR/bin/${NC}"
    fi
done

# Create mmsvg script
cat > "$INSTALL_DIR/mmsvg" << 'EOF'
#!/bin/bash
# mmsvg: Generate SVG (vector, scalable) from .mermaid file

if [ -z "$1" ]; then
    echo "Usage: mmsvg <file.mermaid>"
    exit 1
fi

input_file="$1"
output_file="${input_file%.mermaid}.svg"

echo "🎨 Generating SVG from $input_file..."
mmdc -i "$input_file" -o "$output_file" -b white

if [ $? -eq 0 ]; then
    echo "✓ Created: $output_file"
    echo "🖼️ Opening diagram..."
    xdg-open "$output_file" 2>/dev/null || open "$output_file" 2>/dev/null || echo "Please open $output_file manually"
else
    echo "✗ Failed to generate diagram"
    exit 1
fi
EOF

chmod +x "$INSTALL_DIR/mmsvg"
echo -e "${GREEN}✓ Installed mmsvg${NC}"

# Create an alias for mmview
cat > "$INSTALL_DIR/mmview" << 'EOF'
#!/bin/bash
# mmview: Quick view any image file

if [ -z "$1" ]; then
    echo "Usage: mmview <image-file>"
    exit 1
fi

xdg-open "$1" 2>/dev/null || open "$1" 2>/dev/null || echo "Please open $1 manually"
EOF

chmod +x "$INSTALL_DIR/mmview"
echo -e "${GREEN}✓ Installed mmview${NC}"

# Clean up and summary
echo

# Success message
echo
echo -e "${GREEN}╔══════════════════════════════════════════╗${NC}"
echo -e "${GREEN}║    Installation completed successfully!   ║${NC}"
echo -e "${GREEN}╚══════════════════════════════════════════╝${NC}"
echo
echo -e "${BLUE}Available commands:${NC}"
echo "  mmerd  - Generate ERD from SQL files in current directory"
echo "  mmviz  - Convert .mermaid file to high-res PNG"
echo "  mmsvg  - Convert .mermaid file to scalable SVG"
echo "  mmview - Quick view any image file"
echo
echo -e "${YELLOW}Quick start:${NC}"
echo "  1. cd to a directory with .sql files"
echo "  2. Run: mmerd"
echo "  3. View generated database_erd.png"
echo
echo -e "${BLUE}Example files available in:${NC} $SCRIPT_DIR/examples/"