install.sh

#!/bin/bash

set -e

REPO_URL="https://github.com/waveplate/kibash.git"
INSTALL_DIR="$HOME/.kibash"
BASHRC="$HOME/.bashrc"

echo "Installing kibash..."

# Check if we are installing from the local source directory
if [ -f "init.sh" ] && [ -f "kiba.sh" ] && [ -f "README.md" ]; then
    IS_LOCAL_INSTALL=true
else
    IS_LOCAL_INSTALL=false
fi

if [ "$IS_LOCAL_INSTALL" = false ]; then
    # Check if git is installed
    if ! command -v git &> /dev/null; then
        echo "Error: git is not installed."
        exit 1
    fi

    # Clone or update the repository
    if [ -d "$INSTALL_DIR" ]; then
        echo "Updating existing installation in $INSTALL_DIR..."
        cd "$INSTALL_DIR"
        git pull
    else
        echo "Cloning kibash to $INSTALL_DIR..."
        git clone "$REPO_URL" "$INSTALL_DIR"
    fi
else
    # Local install: copy files to INSTALL_DIR
    echo "Installing from local directory to $INSTALL_DIR..."
    mkdir -p "$INSTALL_DIR"
    cp -r . "$INSTALL_DIR"
fi

# Initialize env directory with defaults if they don't exist
ENV_DIR="$INSTALL_DIR/env"
mkdir -p "$ENV_DIR"

if [ ! -f "$ENV_DIR/KIBASH_PROVIDER" ]; then
    echo "groq" > "$ENV_DIR/KIBASH_PROVIDER"
fi

if [ ! -f "$ENV_DIR/KIBASH_MODEL" ]; then
    echo "gpt-oss-20b" > "$ENV_DIR/KIBASH_MODEL"
fi

if [ ! -f "$ENV_DIR/KIBASH_SYSTEM" ]; then
    echo "Output a valid linux shell command which matches the user description. Output only JSON, do not use codeblocks" > "$ENV_DIR/KIBASH_SYSTEM"
fi

# Add source line to .bashrc if not present
INIT_SCRIPT="$INSTALL_DIR/init.sh"
SOURCE_LINE=". $INIT_SCRIPT"

if grep -Fxq "$SOURCE_LINE" "$BASHRC"; then
    echo "kibash is already sourced in $BASHRC"
else
    echo "Adding source line to $BASHRC..."
    echo "" >> "$BASHRC"
    echo "# kibash" >> "$BASHRC"
    echo "$SOURCE_LINE" >> "$BASHRC"
fi

echo "Installation complete!"
echo "Please restart your terminal or run: source $BASHRC"