Making the code base easy to read and maintain and making it a oneline install
This commit is contained in:
Tyler
Executable
+242
@@ -0,0 +1,242 @@
|
||||
#!/usr/bin/env bash
|
||||
set -e
|
||||
|
||||
REPO_URL="https://github.com/TySP-Dev/ollama-ai-answers-searxng"
|
||||
REPO_DIR="/tmp/ollama-ai-answers-searxng"
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 1: Clone or update repo ==="
|
||||
# ============================================================
|
||||
if [ -d "$REPO_DIR/.git" ]; then
|
||||
echo "Repository already exists — pulling latest changes..."
|
||||
git -C "$REPO_DIR" pull
|
||||
else
|
||||
git clone "$REPO_URL" "$REPO_DIR"
|
||||
fi
|
||||
cd "$REPO_DIR"
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 2: Build dist file ==="
|
||||
# ============================================================
|
||||
if ! python3 build.py; then
|
||||
echo "ERROR: build.py failed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 3: Detect SearXNG Docker Compose installation ==="
|
||||
# ============================================================
|
||||
CANDIDATES=(
|
||||
"$HOME/searxng/docker-compose.yml"
|
||||
"$HOME/docker/searxng/docker-compose.yml"
|
||||
"/opt/searxng/docker-compose.yml"
|
||||
"/etc/searxng/docker-compose.yml"
|
||||
)
|
||||
|
||||
FOUND=()
|
||||
for f in "${CANDIDATES[@]}"; do
|
||||
if [ -f "$f" ] && grep -q "searxng" "$f" 2>/dev/null; then
|
||||
FOUND+=("$f")
|
||||
fi
|
||||
done
|
||||
|
||||
# Also do a broader search
|
||||
while IFS= read -r f; do
|
||||
# Avoid duplicates
|
||||
already=false
|
||||
for existing in "${FOUND[@]}"; do
|
||||
[ "$f" = "$existing" ] && already=true && break
|
||||
done
|
||||
$already || FOUND+=("$f")
|
||||
done < <(find "$HOME" -maxdepth 6 -name "docker-compose.yml" -exec grep -l "searxng" {} \; 2>/dev/null)
|
||||
|
||||
if [ ${#FOUND[@]} -eq 0 ]; then
|
||||
echo "ERROR: No docker-compose.yml containing 'searxng' was found."
|
||||
echo "Please set COMPOSE_FILE manually and re-run."
|
||||
exit 1
|
||||
elif [ ${#FOUND[@]} -eq 1 ]; then
|
||||
COMPOSE_FILE="${FOUND[0]}"
|
||||
echo "Found: $COMPOSE_FILE"
|
||||
else
|
||||
echo "Multiple docker-compose.yml files found:"
|
||||
for i in "${!FOUND[@]}"; do
|
||||
echo " [$((i+1))] ${FOUND[$i]}"
|
||||
done
|
||||
read -rp "Select a file [1-${#FOUND[@]}]: " choice
|
||||
idx=$((choice - 1))
|
||||
if [ "$idx" -lt 0 ] || [ "$idx" -ge "${#FOUND[@]}" ]; then
|
||||
echo "ERROR: Invalid selection"
|
||||
exit 1
|
||||
fi
|
||||
COMPOSE_FILE="${FOUND[$idx]}"
|
||||
fi
|
||||
|
||||
COMPOSE_DIR="$(dirname "$COMPOSE_FILE")"
|
||||
echo "Using: $COMPOSE_FILE"
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 4: Create plugins directory ==="
|
||||
# ============================================================
|
||||
PLUGINS_DIR="$COMPOSE_DIR/plugins"
|
||||
mkdir -p "$PLUGINS_DIR"
|
||||
echo "Plugins directory: $PLUGINS_DIR"
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 5: Copy built plugin ==="
|
||||
# ============================================================
|
||||
cp dist/ollama_answers.py "$PLUGINS_DIR/ollama_answers.py"
|
||||
echo "Plugin installed to $PLUGINS_DIR/ollama_answers.py"
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 6: Update docker-compose.yml volume mount ==="
|
||||
# ============================================================
|
||||
python3 - "$COMPOSE_FILE" <<'PYEOF'
|
||||
import re, sys
|
||||
compose_file = sys.argv[1]
|
||||
content = open(compose_file).read()
|
||||
mount = ' - ./plugins/ollama_answers.py:/usr/local/searxng/searx/plugins/ollama_answers.py:Z'
|
||||
if 'ollama_answers.py' not in content:
|
||||
content = re.sub(
|
||||
r'(volumes:(?:\n\s+- [^\n]+)*)',
|
||||
lambda m: m.group(0) + '\n' + mount,
|
||||
content, count=1
|
||||
)
|
||||
open(compose_file, 'w').write(content)
|
||||
print('Added volume mount to docker-compose.yml')
|
||||
else:
|
||||
print('Volume mount already present')
|
||||
PYEOF
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 7: Find and update settings.yml ==="
|
||||
# ============================================================
|
||||
SETTINGS_CANDIDATES=(
|
||||
"$COMPOSE_DIR/core-config/settings.yml"
|
||||
"$COMPOSE_DIR/config/settings.yml"
|
||||
"$COMPOSE_DIR/searxng/settings.yml"
|
||||
)
|
||||
|
||||
SETTINGS_FOUND=()
|
||||
for f in "${SETTINGS_CANDIDATES[@]}"; do
|
||||
[ -f "$f" ] && SETTINGS_FOUND+=("$f")
|
||||
done
|
||||
|
||||
# Broader search within COMPOSE_DIR
|
||||
while IFS= read -r f; do
|
||||
already=false
|
||||
for existing in "${SETTINGS_FOUND[@]}"; do
|
||||
[ "$f" = "$existing" ] && already=true && break
|
||||
done
|
||||
$already || SETTINGS_FOUND+=("$f")
|
||||
done < <(find "$COMPOSE_DIR" -name "settings.yml" 2>/dev/null | head -5)
|
||||
|
||||
if [ ${#SETTINGS_FOUND[@]} -eq 0 ]; then
|
||||
echo "WARNING: settings.yml not found. Add this manually:"
|
||||
echo ""
|
||||
echo " plugins:"
|
||||
echo " searx.plugins.ollama_answers.SXNGPlugin:"
|
||||
echo " active: true"
|
||||
echo ""
|
||||
SETTINGS_FILE=""
|
||||
elif [ ${#SETTINGS_FOUND[@]} -eq 1 ]; then
|
||||
SETTINGS_FILE="${SETTINGS_FOUND[0]}"
|
||||
echo "Found: $SETTINGS_FILE"
|
||||
else
|
||||
echo "Multiple settings.yml files found:"
|
||||
for i in "${!SETTINGS_FOUND[@]}"; do
|
||||
echo " [$((i+1))] ${SETTINGS_FOUND[$i]}"
|
||||
done
|
||||
read -rp "Select a file [1-${#SETTINGS_FOUND[@]}]: " schoice
|
||||
sidx=$((schoice - 1))
|
||||
if [ "$sidx" -lt 0 ] || [ "$sidx" -ge "${#SETTINGS_FOUND[@]}" ]; then
|
||||
echo "ERROR: Invalid selection"
|
||||
exit 1
|
||||
fi
|
||||
SETTINGS_FILE="${SETTINGS_FOUND[$sidx]}"
|
||||
fi
|
||||
|
||||
if [ -n "$SETTINGS_FILE" ]; then
|
||||
python3 - "$SETTINGS_FILE" <<'PYEOF'
|
||||
import sys
|
||||
settings_file = sys.argv[1]
|
||||
content = open(settings_file).read()
|
||||
entry = 'searx.plugins.ollama_answers.SXNGPlugin:\n active: true'
|
||||
if 'ollama_answers' not in content:
|
||||
if 'plugins:' in content:
|
||||
content = content.replace('plugins:', 'plugins:\n\n ' + entry, 1)
|
||||
else:
|
||||
content += '\nplugins:\n\n ' + entry + '\n'
|
||||
open(settings_file, 'w').write(content)
|
||||
print('Added plugin entry to settings.yml')
|
||||
else:
|
||||
print('Plugin entry already present')
|
||||
PYEOF
|
||||
fi
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 8: Configure environment variables ==="
|
||||
# ============================================================
|
||||
python3 - "$COMPOSE_FILE" <<'PYEOF'
|
||||
import sys
|
||||
compose_file = sys.argv[1]
|
||||
content = open(compose_file).read()
|
||||
has_llm_url = 'LLM_URL' in content
|
||||
has_llm_model = 'LLM_MODEL' in content
|
||||
sys.exit(0 if (has_llm_url and has_llm_model) else 1)
|
||||
PYEOF
|
||||
ENV_ALREADY=$?
|
||||
|
||||
if [ $ENV_ALREADY -eq 0 ]; then
|
||||
echo "LLM_URL and LLM_MODEL already present in docker-compose.yml"
|
||||
else
|
||||
read -rp "Enter Ollama URL (default: http://ollama:11434/v1/chat/completions): " llm_url
|
||||
llm_url="${llm_url:-http://ollama:11434/v1/chat/completions}"
|
||||
|
||||
read -rp "Enter model name (default: qwen3.5:9b): " llm_model
|
||||
llm_model="${llm_model:-qwen3.5:9b}"
|
||||
|
||||
python3 - "$COMPOSE_FILE" "$llm_url" "$llm_model" <<'PYEOF'
|
||||
import re, sys
|
||||
compose_file, llm_url, llm_model = sys.argv[1], sys.argv[2], sys.argv[3]
|
||||
content = open(compose_file).read()
|
||||
env_lines = f' - LLM_URL={llm_url}\n - LLM_MODEL={llm_model}'
|
||||
if 'LLM_URL' not in content:
|
||||
content = re.sub(
|
||||
r'(environment:(?:\n\s+- [^\n]+)*)',
|
||||
lambda m: m.group(0) + '\n' + env_lines,
|
||||
content, count=1
|
||||
)
|
||||
open(compose_file, 'w').write(content)
|
||||
print(f'Added LLM_URL={llm_url} and LLM_MODEL={llm_model} to docker-compose.yml')
|
||||
else:
|
||||
print('LLM_URL already present, skipping')
|
||||
PYEOF
|
||||
fi
|
||||
|
||||
# ============================================================
|
||||
echo "=== Step 9: Restart SearXNG ==="
|
||||
# ============================================================
|
||||
read -rp "Restart SearXNG now? (y/n): " restart_choice
|
||||
if [[ "$restart_choice" =~ ^[Yy]$ ]]; then
|
||||
echo "Restarting SearXNG..."
|
||||
cd "$COMPOSE_DIR"
|
||||
docker compose up -d --force-recreate core
|
||||
echo "SearXNG restarted."
|
||||
else
|
||||
echo "Skipping restart. Run manually:"
|
||||
echo " cd \"$COMPOSE_DIR\" && docker compose up -d --force-recreate core"
|
||||
fi
|
||||
|
||||
# ============================================================
|
||||
echo ""
|
||||
echo "=== Installation complete ==="
|
||||
echo ""
|
||||
echo "Summary:"
|
||||
echo " Plugin: $PLUGINS_DIR/ollama_answers.py"
|
||||
echo " Compose: $COMPOSE_FILE"
|
||||
[ -n "$SETTINGS_FILE" ] && echo " Settings: $SETTINGS_FILE"
|
||||
echo ""
|
||||
echo "Environment variables to verify in docker-compose.yml:"
|
||||
echo " LLM_URL — Ollama endpoint"
|
||||
echo " LLM_MODEL — Model name (e.g. qwen3.5:9b)"
|
||||
echo ""
|
||||
Reference in New Issue
Block a user