#!/usr/bin/env bash
set -euo pipefail

REPO="${PLUXX_EXA_REPO:-orchidautomation/pluxx}"
REF="${PLUXX_EXA_REF:-main}"
PLUGIN_NAME="exa-research-example"
PLUGIN_ROOT_DIR="${PLUXX_EXA_OPENCODE_PLUGIN_ROOT_DIR:-$HOME/.config/opencode/plugins}"
INSTALL_DIR="${PLUXX_EXA_OPENCODE_INSTALL_DIR:-$PLUGIN_ROOT_DIR/$PLUGIN_NAME}"
ENTRY_PATH="${PLUXX_EXA_OPENCODE_ENTRY_PATH:-$PLUGIN_ROOT_DIR/$PLUGIN_NAME.ts}"
SKILLS_ROOT="${PLUXX_EXA_OPENCODE_SKILLS_ROOT:-$HOME/.config/opencode/skills}"
ARCHIVE_URL="${PLUXX_EXA_ARCHIVE_URL:-https://codeload.github.com/$REPO/tar.gz/$REF}"

need_cmd() {
  if ! command -v "$1" >/dev/null 2>&1; then
    echo "Missing required command: $1" >&2
    exit 1
  fi
}

need_cmd curl
need_cmd mktemp
need_cmd node
need_cmd npm
need_cmd rm
need_cmd tar

TMP_DIR="$(mktemp -d)"
cleanup() {
  rm -rf "$TMP_DIR"
}
trap cleanup EXIT

echo "Downloading $REPO@$REF ..."
ARCHIVE_PATH="$TMP_DIR/pluxx.tar.gz"
curl -fsSL "$ARCHIVE_URL" -o "$ARCHIVE_PATH"
tar -xzf "$ARCHIVE_PATH" -C "$TMP_DIR"

SOURCE_ROOT="$(find "$TMP_DIR" -mindepth 1 -maxdepth 1 -type d -name 'pluxx-*' | head -n1)"
if [[ -z "$SOURCE_ROOT" ]]; then
  echo "Could not locate extracted Pluxx source." >&2
  exit 1
fi

echo "Trust note: this example installs a session-start shell hook (scripts/check-exa-setup.sh)."
echo "The hook only reports whether EXA_API_KEY is set and prints setup guidance, but running this installer means trusting that local hook."

echo "Building Pluxx ..."
(cd "$SOURCE_ROOT" && npm ci && npm run build)

echo "Building Exa Research Example ..."
(cd "$SOURCE_ROOT/example/exa-plugin" && node ../../bin/pluxx.js build)

BUNDLE_DIR="$SOURCE_ROOT/example/exa-plugin/dist/opencode"
PLUGIN_PACKAGE="$BUNDLE_DIR/package.json"

if [[ ! -f "$PLUGIN_PACKAGE" ]]; then
  echo "Built Exa example is missing an OpenCode package.json." >&2
  exit 1
fi

mkdir -p "$(dirname "$INSTALL_DIR")" "$SKILLS_ROOT"
rm -rf "$INSTALL_DIR"
cp -R "$BUNDLE_DIR" "$INSTALL_DIR"

export ENTRY_PATH
export PLUGIN_NAME

node <<'NODE'
const fs = require('fs')

const entryPath = process.env.ENTRY_PATH
const pluginName = process.env.PLUGIN_NAME
const exportName = pluginName
  .split(/[^A-Za-z0-9]+/)
  .filter(Boolean)
  .map((segment) => segment.charAt(0).toUpperCase() + segment.slice(1))
  .join('')

const content = [
  'import type { Plugin } from "@opencode-ai/plugin"',
  'import { join } from "path"',
  '',
  'import * as PluginModule from "./' + pluginName + '/index.ts"',
  '',
  '// OpenCode auto-loads plugin files placed directly in ~/.config/opencode/plugins.',
  '// Proxy into the installed plugin bundle while preserving its expected root.',
  'const pluginFactory = Object.values(PluginModule).find((value): value is Plugin => typeof value === "function")',
  '',
  'if (!pluginFactory) {',
  '  throw new Error("OpenCode plugin bundle for ' + pluginName + ' did not export a plugin function.")',
  '}',
  '',
  'export const ' + exportName + ': Plugin = async (context) =>',
  '  pluginFactory({',
  '    ...context,',
  '    directory: join(context.directory, "' + pluginName + '"),',
  '  })',
  '',
].join('\n')

fs.writeFileSync(entryPath, content)
NODE

if [[ -d "$INSTALL_DIR/skills" ]]; then
  for skill_dir in "$INSTALL_DIR"/skills/*; do
    [[ -d "$skill_dir" ]] || continue
    skill_name="$(basename "$skill_dir")"
    installed_skill_dir="$SKILLS_ROOT/${PLUGIN_NAME}-${skill_name}"
    rm -rf "$installed_skill_dir"
    cp -R "$skill_dir" "$installed_skill_dir"

    export SKILL_PATH="$installed_skill_dir/SKILL.md"
    export SKILL_NAME="$skill_name"
    export PLUGIN_NAME

    node <<'NODE'
const fs = require('fs')

const filepath = process.env.SKILL_PATH
const pluginName = process.env.PLUGIN_NAME
const fallbackName = process.env.SKILL_NAME

if (!fs.existsSync(filepath)) process.exit(0)

const content = fs.readFileSync(filepath, 'utf8')
const match = content.match(/^---\n([\s\S]*?)\n---\n?/)
const namespacedName = pluginName + '/' + fallbackName

if (!match) {
  fs.writeFileSync(filepath, '---\nname: ' + namespacedName + '\n---\n\n' + content)
  process.exit(0)
}

const frontmatter = match[1]
const nameMatch = frontmatter.match(/^name:\s*(.+)$/m)
const existingName = nameMatch ? nameMatch[1].trim().replace(/^['"]|['"]$/g, '') : fallbackName
const nextName = existingName.startsWith(pluginName + '/') ? existingName : pluginName + '/' + existingName
const nextFrontmatter = nameMatch
  ? frontmatter.replace(/^name:\s*.+$/m, 'name: ' + nextName)
  : 'name: ' + nextName + '\n' + frontmatter

fs.writeFileSync(
  filepath,
  content.slice(0, match.index) + '---\n' + nextFrontmatter + '\n---\n' + content.slice(match[0].length),
)
NODE
  done
fi

echo "Installed $PLUGIN_NAME plugin code to $INSTALL_DIR"
echo "Installed OpenCode wrapper at $ENTRY_PATH"
echo "Synced namespaced skills into $SKILLS_ROOT"
echo "Optional: export EXA_API_KEY for higher limits."
echo "If OpenCode is already open, restart or reload it so the plugin is picked up."