diff options
Diffstat (limited to 'scripts/mcp.sh')
| -rw-r--r-- | scripts/mcp.sh | 176 |
1 files changed, 176 insertions, 0 deletions
diff --git a/scripts/mcp.sh b/scripts/mcp.sh new file mode 100644 index 0000000..e974476 --- /dev/null +++ b/scripts/mcp.sh @@ -0,0 +1,176 @@ +#!/usr/bin/env bash +set -e + +ROOT_DIR="$(cd -- "$( dirname -- "${BASH_SOURCE[0]}" )/.." &> /dev/null && pwd)" +BIN_DIR="$ROOT_DIR/bin" +MCP_DIR="$ROOT_DIR/cache/__mcp__" +MCP_JSON_PATH="$ROOT_DIR/mcp.json" +FUNCTIONS_JSON_PATH="$ROOT_DIR/functions.json" +MCP_BRIDGE_PORT="${MCP_BRIDGE_PORT:-8808}" + +# @cmd Start/Restart mcp bridge server +start() { + if [[ ! -f "$MCP_JSON_PATH" ]]; then + _die "error: not found mcp.json" + fi + stop + mkdir -p "$MCP_DIR" + index_js="$ROOT_DIR/mcp/bridge/index.js" + llm_functions_dir="$ROOT_DIR" + if _is_win; then + index_js="$(cygpath -w "$index_js")" + llm_functions_dir="$(cygpath -w "$llm_functions_dir")" + fi + echo "Run MCP Bridge server" + nohup node "$index_js" "$llm_functions_dir" > "$MCP_DIR/mcp-bridge.log" 2>&1 & + wait-for-server + echo "Merge MCP tools into functions.json" + merge-functions > "$MCP_DIR/functions.json" + cp -f "$MCP_DIR/functions.json" "$FUNCTIONS_JSON_PATH" + build-bin +} + +# @cmd Stop mcp bridge server +stop() { + pid="$(get-server-pid)" + if [[ -n "$pid" ]]; then + if _is_win; then + taskkill /PID "$pid" /F > /dev/null 2>&1 || true + else + kill -9 "$pid" > /dev/null 2>&1 || true + fi + fi + mkdir -p "$MCP_DIR" + unmerge-functions > "$MCP_DIR/functions.original.json" + cp -f "$MCP_DIR/functions.original.json" "$FUNCTIONS_JSON_PATH" +} + +# @cmd Call mcp tool +# @arg tool![`_choice_tool`] The tool name +# @arg json The json data +call() { + if [[ -z "$argc_json" ]]; then + declaration="$(build-declarations | jq --arg tool "$argc_tool" -r '.[] | select(.name == $tool)')" + if [[ -n "$declaration" ]]; then + _ask_json_data "$declaration" + fi + fi + if [[ -z "$argc_json" ]]; then + _die "error: no JSON data" + fi + bash "$ROOT_DIR/scripts/run-mcp-tool.sh" "$argc_tool" "$argc_json" +} + +# @cmd Show logs +# @flag -f --follow Follow mode +logs() { + args="" + if [[ -n "$argc_follow" ]]; then + args="$args -f" + fi + if [[ -f "$MCP_DIR/mcp-bridge.log" ]]; then + tail $args "$MCP_DIR/mcp-bridge.log" + fi +} + +# @cmd Build tools to bin +build-bin() { + tools=( $(build-declarations | jq -r '.[].name') ) + for tool in "${tools[@]}"; do + if _is_win; then + bin_file="$BIN_DIR/$tool.cmd" + _build_win_shim > "$bin_file" + else + bin_file="$BIN_DIR/$tool" + ln -s -f "$ROOT_DIR/scripts/run-mcp-tool.sh" "$bin_file" + fi + echo "Build bin/$tool" + done +} + +# @cmd Merge mcp tools into functions.json +merge-functions() { + jq --argjson json1 "$(unmerge-functions)" --argjson json2 "$(build-declarations)" -n '($json1 + $json2)' +} + +# @cmd Unmerge mcp tools from functions.json +unmerge-functions() { + functions="[]" + if [[ -f "$FUNCTIONS_JSON_PATH" ]]; then + functions="$(cat "$FUNCTIONS_JSON_PATH")" + fi + printf "%s" "$functions" | jq 'map(select(has("mcp") | not))' +} + +# @cmd Build tools to bin +build-declarations() { + curl -sS http://localhost:$MCP_BRIDGE_PORT/tools | jq '.[] |= . + {mcp: true}' +} + +# @cmd Wait for mcp bridge server to ready +wait-for-server() { + while true; do + if [[ "$(curl -fsS http://localhost:$MCP_BRIDGE_PORT/health 2>&1)" == "OK" ]]; then + break; + fi + sleep 1 + done +} + +# @cmd +get-server-pid() { + curl -fsSL http://localhost:$MCP_BRIDGE_PORT/pid 2>/dev/null || true +} + +_ask_json_data() { + declaration="$1" + echo 'Missing the JSON data but here are its properties:' + echo "$declaration" | ./scripts/declarations-util.sh pretty-print | sed -n '2,$s/^/>/p' + echo 'Generate placeholder data:' + data="$(echo "$declaration" | _declarations_json_data)" + echo "> $data" + read -e -r -p 'JSON data (Press ENTER to use placeholder): ' res + if [[ -z "$res" ]]; then + argc_json="$data" + else + argc_json="$res" + fi +} + +_declarations_json_data() { + ./scripts/declarations-util.sh generate-json | tail -n +2 +} + +_build_win_shim() { + run="\"$(argc --argc-shell-path)\" --noprofile --norc" + cat <<-EOF +@echo off +setlocal + +set "bin_dir=%~dp0" +for %%i in ("%bin_dir:~0,-1%") do set "script_dir=%%~dpi" +set "script_name=%~n0" + +$run "%script_dir%scripts\run-mcp-tool.sh" "%script_name%" %* +EOF +} + +_is_win() { + if [[ "$OS" == "Windows_NT" ]]; then + return 0 + else + return 1 + fi +} + +_choice_tool() { + build-declarations | jq -r '.[].name' +} + +_die() { + echo "$*" >&2 + exit 1 +} + +# See more details at https://github.com/sigoden/argc +eval "$(argc --argc-eval "$0" "$@")" |