#!/usr/bin/env bash
# OpenHands CLI 1.16+ with AI Model Mark GPT-6 Sol. No API key is saved to disk.
set -euo pipefail

if ! command -v openhands >/dev/null 2>&1; then
  echo 'OpenHands CLI chua duoc cai dat: https://github.com/OpenHands/OpenHands-CLI' >&2
  exit 1
fi

profile="${AIMM_OPENHANDS_PROFILE:-$HOME/.openhands-aimodelmark-sol}"
mkdir -p -- "$profile"
chmod 700 -- "$profile"
settings="$profile/agent_settings.json"

if [[ ! -e "$settings" ]]; then
  umask 077
  cat > "$settings" <<'JSON'
{
  "kind": "Agent",
  "llm": {
    "model": "openai/gpt-6-sol",
    "base_url": "https://aimodelmark.com/v1",
    "max_output_tokens": 8192,
    "usage_id": "agent"
  },
  "tools": [
    {"name": "terminal", "params": {}},
    {"name": "file_editor", "params": {}},
    {"name": "task_tracker", "params": {}},
    {"name": "task_tool_set", "params": {}}
  ],
  "condenser": null
}
JSON
else
  if ! python3 - "$settings" <<'PY'
import json, sys
try:
    agent = json.load(open(sys.argv[1], encoding='utf-8'))
    llm = agent['llm']
    assert llm['model'] == 'openai/gpt-6-sol'
    assert llm['base_url'] == 'https://aimodelmark.com/v1'
    assert 0 < llm['max_output_tokens'] <= 16384
    assert not llm.get('api_key')
except (AssertionError, KeyError, OSError, ValueError, TypeError):
    sys.exit(1)
PY
  then
    echo "Profile $settings da ton tai voi thiet lap khac. Hay kiem tra thu cong; script khong ghi de." >&2
    exit 1
  fi
fi

if [[ -z "${AI_MODEL_MARK_KEY:-}" ]]; then
  read -r -s -p 'Nhap AI Model Mark API key: ' AI_MODEL_MARK_KEY
  echo
fi
if [[ -z "${AI_MODEL_MARK_KEY:-}" ]]; then
  echo 'Thieu API key.' >&2
  exit 1
fi

export OPENHANDS_PERSISTENCE_DIR="$profile"
export OPENHANDS_CONVERSATIONS_DIR="$profile/conversations"
export OPENHANDS_WORK_DIR="${OPENHANDS_WORK_DIR:-$PWD}"
export LLM_MODEL='openai/gpt-6-sol'
export LLM_BASE_URL='https://aimodelmark.com/v1'
export LLM_API_KEY="$AI_MODEL_MARK_KEY"
export OPENHANDS_SUPPRESS_BANNER=1
exec openhands --override-with-envs "$@"