clone_and_link.sh

#!/bin/bash

set -e  # Exit on any error

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

echo_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

echo_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

echo_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

is_legacy_pi_repo_symlink() {
    local pi_dir="$HOME/.pi"
    local repo_pi_dir="$HOME/.dot-files/files/.pi"
    local link_target

    if [ ! -L "$pi_dir" ]; then
      return 1
    fi

    link_target=$(readlink "$pi_dir")
    [ "$link_target" = ".dot-files/files/.pi" ] || [ "$link_target" = "$repo_pi_dir" ]
}

repair_pi_directory_if_needed() {
    local pi_dir="$HOME/.pi"
    local repo_pi_dir="$HOME/.dot-files/files/.pi"
    local temp_pi_dir

    if ! is_legacy_pi_repo_symlink; then
      return 0
    fi

    echo_warn "Detected legacy ~/.pi symlink to dot-files; converting it back to a real directory"

    temp_pi_dir=$(mktemp -d "${TMPDIR:-/tmp}/dotfiles-pi.XXXXXX")
    cp -R "$pi_dir/." "$temp_pi_dir/"

    rm -f "$pi_dir"
    mkdir -p "$pi_dir"
    cp -R "$temp_pi_dir/." "$pi_dir/"
    rm -rf "$temp_pi_dir"

    if [ -f "$repo_pi_dir/agent/auth.json" ] && [ -f "$pi_dir/agent/auth.json" ]; then
      rm -f "$repo_pi_dir/agent/auth.json"
      echo_info "✓ Moved Pi auth.json out of the dot-files checkout"
    fi
}

install_pi_settings_file() {
    local src="$HOME/.dot-files/files/.pi/agent/settings.json"
    local dest="$HOME/.pi/agent/settings.json"
    local dest_dir

    if [ ! -e "$src" ]; then
      return 0
    fi

    dest_dir=$(dirname "$dest")

    echo_info "Processing Pi settings file"
    mkdir -p "$dest_dir"

    if ln -vsf "$src" "$dest" 2>/dev/null; then
      echo_info "✓ Created Pi settings symlink"
      return 0
    fi

    echo_warn "Symlink failed for Pi settings file"
    echo_info "Attempting to copy Pi settings file instead..."

    if [ -d "$dest" ]; then
      echo_error "Refusing to replace directory with file: $dest"
      return 1
    fi

    if [ -f "$dest" ] || [ -L "$dest" ]; then
      rm -f "$dest"
    fi

    if cp "$src" "$dest"; then
      echo_info "✓ Copied Pi settings file"
    else
      echo_error "Failed to copy Pi settings file"
      return 1
    fi
}

pushd "$HOME"

  # Clone or update dot-files
  if [ -d ".dot-files" ]; then
    echo_info "Updating existing dot-files repository..."
    pushd ".dot-files"
      git pull --rebase
    popd
  else
    echo_info "Cloning dot-files repository..."
    git clone "git@github.com:colmarius/dot-files.git" ".dot-files"
  fi

  # Process each file/directory
  while IFS= read -r -d '' f; do
    # Skip git internals
    [ "$f" == '.dot-files/files/.git' ] && continue

    basename_f=$(basename "$f")

    if [ "$basename_f" = ".pi" ]; then
      echo_info "Skipping top-level .pi entry; Pi settings are managed separately"
      continue
    fi

    if [ -d "$f" ]; then
      echo_info "Processing directory: $basename_f"

      # Try to create symlink for directory
      if ln -vsf "$f" . 2>/dev/null; then
        echo_info "✓ Created symlink for directory: $basename_f"
      else
        echo_warn "Symlink failed for directory: $basename_f"
        echo_info "Attempting to copy directory instead..."

        # Remove existing directory if it exists
        if [ -d "$basename_f" ]; then
          rm -rf "$basename_f"
        fi

        # Copy the directory
        if cp -r "$f" .; then
          echo_info "✓ Copied directory: $basename_f"
        else
          echo_error "Failed to copy directory: $basename_f"
        fi
      fi
    else
      echo_info "Processing file: $basename_f"

      # For files, always try symlink first
      if ln -vsf "$f" . 2>/dev/null; then
        echo_info "✓ Created symlink for file: $basename_f"
      else
        echo_warn "Symlink failed for file: $basename_f"
        echo_info "Attempting to copy file instead..."

        # Remove existing file if it exists
        if [ -f "$basename_f" ]; then
          rm -f "$basename_f"
        fi

        # Copy the file
        if cp "$f" .; then
          echo_info "✓ Copied file: $basename_f"
        else
          echo_error "Failed to copy file: $basename_f"
        fi
      fi
    fi
  done < <(find .dot-files/files -mindepth 1 -maxdepth 1 -print0)

  repair_pi_directory_if_needed
  install_pi_settings_file

  echo_info "Dot-files setup complete!"

popd