#!/usr/bin/env bash

SCRIPT=$(readlink -f "$0")
SCRIPTPATH=$(dirname "$SCRIPT")
cd "$SCRIPTPATH" || exit

if [[ ! -x $(command -v aichat) ]]; then
	echo "aichat is not installed."
	if [[ $(docker images -q aichat-build 2>/dev/null) ]]; then
		container_id=$(docker create "aichat")
		docker cp "${container_id}:/usr/local/cargo/bin/aichat" "./tools/"
		docker rm "${container_id}"
	else
		cargo install aichat@0.29.0
	fi
fi

source .env
mkdir -p ~/.config/aichat/
cat src/aichat/config.yaml | sed "s/__LLM_API_KEY__/${LLM_API_KEY}/" >~/.config/aichat/config.yaml
OLLAMA_HOST=$(cat .env | grep OLLAMA_HOST | cut -d'=' -f2 | cut -d'/' -f3)
if [[ ! -z $OLLAMA_HOST ]]; then
	cat ~/.config/aichat/config.yaml | sed "s/http/https/" >~/.config/aichat/config.yaml.tmp
	cat ~/.config/aichat/config.yaml.tmp | sed "s/localhost:11434/${OLLAMA_HOST}/" >~/.config/aichat/config.yaml
fi

mkdir -p ~/.config/aichat/roles
cp src/aichat/roles/* ~/.config/aichat/roles/