@quantiya/codevibe-gemini-plugin 1.0.0

package/hooks/after-agent.sh

@@ -0,0 +1,122 @@
+#!/bin/bash
+
+# AfterAgent Hook
+# Triggered when Gemini finishes responding to a prompt
+# Sends assistant response from hook input and extracts reasoning from transcript
+
+# Source common utilities
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+source "$SCRIPT_DIR/common.sh"
+
+log "INFO" "AfterAgent hook triggered"
+
+# Read JSON input from stdin
+INPUT=$(read_json_input)
+
+log "DEBUG" "Input: $INPUT"
+
+# Extract fields from hook input
+SESSION_ID=$(echo "$INPUT" | jq -r '.session_id // empty')
+PROMPT_RESPONSE=$(echo "$INPUT" | jq -r '.prompt_response // empty')
+TRANSCRIPT_PATH=$(echo "$INPUT" | jq -r '.transcript_path // empty')
+
+EVENTS_SENT=0
+
+# 1. Send assistant response
+# Prefer transcript content (accurate) over prompt_response (has streaming duplication bug)
+RESPONSE_CONTENT=""
+
+if [ -n "$TRANSCRIPT_PATH" ] && [ -f "$TRANSCRIPT_PATH" ]; then
+  # Extract content from the last gemini message in transcript
+  RESPONSE_CONTENT=$(jq -r '
+    [.messages[] | select(.type == "gemini")] | last |
+    if .content | type == "string" then .content
+    elif .content | type == "array" then [.content[] | .text // empty] | join("")
+    else empty end
+  ' "$TRANSCRIPT_PATH" 2>/dev/null)
+  log "DEBUG" "Got response from transcript: ${RESPONSE_CONTENT:0:100}..."
+fi
+
+# Fall back to prompt_response if transcript extraction failed
+if [ -z "$RESPONSE_CONTENT" ] || [ "$RESPONSE_CONTENT" = "null" ]; then
+  RESPONSE_CONTENT="$PROMPT_RESPONSE"
+  log "DEBUG" "Using prompt_response fallback: ${RESPONSE_CONTENT:0:100}..."
+fi
+
+if [ -n "$RESPONSE_CONTENT" ]; then
+  EVENT_PAYLOAD=$(jq -n \
+    --arg session_id "$SESSION_ID" \
+    --arg content "$RESPONSE_CONTENT" \
+    --arg hook_event_name "AfterAgent" \
+    --arg type "ASSISTANT_RESPONSE" \
+    --arg source "DESKTOP" \
+    '{
+      session_id: $session_id,
+      hook_event_name: $hook_event_name,
+      type: $type,
+      source: $source,
+      content: $content
+    }')
+
+  log "DEBUG" "Sending assistant response: ${RESPONSE_CONTENT:0:100}..."
+
+  send_to_mcp "event" "$EVENT_PAYLOAD" "$SESSION_ID"
+
+  if [ $? -eq 0 ]; then
+    EVENTS_SENT=$((EVENTS_SENT + 1))
+    log "INFO" "Assistant response sent successfully"
+  else
+    log "ERROR" "Failed to send assistant response"
+  fi
+else
+  log "WARN" "No assistant response content available"
+fi
+
+# 2. Extract reasoning/thoughts from transcript (secondary - for rich context)
+# Gemini transcript format: single JSON with .messages[] array
+# Each gemini message may have .thoughts[] with {subject, description, timestamp}
+if [ -n "$TRANSCRIPT_PATH" ] && [ -f "$TRANSCRIPT_PATH" ]; then
+  log "DEBUG" "Reading transcript for thoughts: $TRANSCRIPT_PATH"
+
+  # Get thoughts from the last gemini message in the transcript
+  THOUGHTS=$(jq -r '
+    [.messages[] | select(.type == "gemini")] | last |
+    .thoughts // [] |
+    map(.subject + ": " + .description) |
+    join("\n\n")
+  ' "$TRANSCRIPT_PATH" 2>/dev/null)
+
+  if [ -n "$THOUGHTS" ] && [ "$THOUGHTS" != "null" ]; then
+    REASONING_PAYLOAD=$(jq -n \
+      --arg session_id "$SESSION_ID" \
+      --arg content "$THOUGHTS" \
+      --arg hook_event_name "AfterAgent" \
+      --arg type "REASONING" \
+      --arg source "DESKTOP" \
+      '{
+        session_id: $session_id,
+        hook_event_name: $hook_event_name,
+        type: $type,
+        source: $source,
+        content: $content
+      }')
+
+    log "DEBUG" "Sending reasoning: ${THOUGHTS:0:100}..."
+
+    send_to_mcp "event" "$REASONING_PAYLOAD" "$SESSION_ID"
+
+    if [ $? -eq 0 ]; then
+      EVENTS_SENT=$((EVENTS_SENT + 1))
+      log "INFO" "Reasoning event sent successfully"
+    else
+      log "ERROR" "Failed to send reasoning event"
+    fi
+  else
+    log "DEBUG" "No thoughts found in transcript"
+  fi
+else
+  log "DEBUG" "No transcript path available for reasoning extraction"
+fi
+
+log "INFO" "AfterAgent hook completed. Events sent: $EVENTS_SENT"
+exit 0

package/hooks/after-tool.sh

@@ -0,0 +1,71 @@
+#!/bin/bash
+
+# Gemini CLI AfterTool hook
+# Called after a tool is executed - captures tool results
+#
+# Input (JSON via stdin):
+# {
+#   "session_id": "uuid",
+#   "hook_event_name": "AfterTool",
+#   "tool_name": "shell" | "edit" | "write" | etc,
+#   "tool_input": { ... tool-specific input ... },
+#   "tool_response": { ... tool output/result ... },
+#   ...
+# }
+
+# Get script directory and source common utilities
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+source "$SCRIPT_DIR/common.sh"
+
+log "INFO" "AfterTool hook triggered"
+
+# Read JSON input from stdin
+INPUT=$(read_json_input)
+
+if [ -z "$INPUT" ]; then
+  log "ERROR" "No input received"
+  exit 0
+fi
+
+log "DEBUG" "Received input: $INPUT"
+
+# Extract fields
+SESSION_ID=$(echo "$INPUT" | grep -o '"session_id"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*"\([^"]*\)"$/\1/')
+TOOL_NAME=$(echo "$INPUT" | grep -o '"tool_name"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*"\([^"]*\)"$/\1/')
+
+if [ -z "$SESSION_ID" ]; then
+  log "ERROR" "No session_id in input"
+  exit 0
+fi
+
+log "INFO" "Session: $SESSION_ID, Tool: $TOOL_NAME"
+
+# Check if MCP server is running
+if ! check_mcp_server "$SESSION_ID"; then
+  log "WARN" "MCP server not running, cannot send tool result"
+  exit 0
+fi
+
+# Build event payload with the full hook input
+# The server will extract tool_input and tool_response
+EVENT_PAYLOAD=$(cat <<PAYLOAD
+{
+  "session_id": "$SESSION_ID",
+  "hook_event_name": "AfterTool",
+  "type": "TOOL_USE",
+  "source": "DESKTOP",
+  "content": "Tool executed: $TOOL_NAME",
+  "metadata": {
+    "hook_event_name": "AfterTool",
+    "tool_name": "$TOOL_NAME",
+    "hookInput": $INPUT
+  }
+}
+PAYLOAD
+)
+
+# Send to MCP server
+send_to_mcp "event" "$EVENT_PAYLOAD" "$SESSION_ID"
+
+log "INFO" "AfterTool hook completed"
+exit 0

package/hooks/before-agent.sh

@@ -0,0 +1,46 @@
+#!/bin/bash
+
+# BeforeAgent Hook
+# Triggered when user submits a prompt in Gemini CLI
+# Captures user prompts and sends to mobile app as USER_PROMPT events
+
+# Source common utilities
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+source "$SCRIPT_DIR/common.sh"
+
+log "INFO" "BeforeAgent hook triggered"
+
+# Read JSON input from stdin
+INPUT=$(read_json_input)
+
+log "DEBUG" "Input: $INPUT"
+
+# Extract fields from hook input
+SESSION_ID=$(echo "$INPUT" | jq -r '.session_id // empty')
+PROMPT=$(echo "$INPUT" | jq -r '.prompt // empty')
+
+# Build explicit event payload with type and content
+EVENT_PAYLOAD=$(jq -n \
+  --arg session_id "$SESSION_ID" \
+  --arg content "$PROMPT" \
+  --arg hook_event_name "BeforeAgent" \
+  --arg type "USER_PROMPT" \
+  '{
+    session_id: $session_id,
+    hook_event_name: $hook_event_name,
+    type: $type,
+    content: $content
+  }')
+
+# Send to MCP server
+send_to_mcp "event" "$EVENT_PAYLOAD" "$SESSION_ID"
+
+EXIT_CODE=$?
+
+if [ $EXIT_CODE -eq 0 ]; then
+  log "INFO" "BeforeAgent event sent successfully"
+else
+  log "ERROR" "Failed to send BeforeAgent event (exit code: $EXIT_CODE)"
+fi
+
+exit $EXIT_CODE

package/hooks/before-tool.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+# Gemini CLI BeforeTool hook
+# Called before a tool is executed.
+#
+# Actual approval prompts are handled via the Notification/ToolPermission hook,
+# which fires only when Gemini CLI's permission mode requires user approval.
+# This hook simply allows all tools to proceed.
+#
+# Output (JSON to stdout):
+# {"decision": "allow"}
+
+# Read stdin to consume it (Gemini CLI expects it to be read)
+cat > /dev/null
+
+echo '{"decision":"allow"}'
+exit 0

package/hooks/common.sh

@@ -0,0 +1,220 @@
+#!/bin/bash
+
+# Common utilities for CodeVibe Gemini hooks
+# This file provides shared functions for all hook scripts
+
+# Configuration
+# Use TMPDIR if set (macOS sets this to user-specific temp), otherwise /tmp
+CODEVIBE_TMPDIR="${TMPDIR:-/tmp}"
+LOG_FILE="${LOG_FILE:-${CODEVIBE_TMPDIR}/codevibe-gemini-hooks.log}"
+DEFAULT_PORT="${DEFAULT_PORT:-3457}"
+
+# Timeout for interactive prompts (5 minutes = 300 seconds)
+INTERACTIVE_PROMPT_TIMEOUT="${INTERACTIVE_PROMPT_TIMEOUT:-300}"
+# Poll interval for checking prompt response (1 second)
+POLL_INTERVAL="${POLL_INTERVAL:-1}"
+
+# Log a message with timestamp
+log() {
+  local level="$1"
+  shift
+  local message="$*"
+  echo "[$(date -u +"%Y-%m-%dT%H:%M:%SZ")] [$level] [gemini-hook] $message" >> "$LOG_FILE"
+}
+
+# Get the MCP server URL for a given session
+# Priority: session-specific port file → default port file → DEFAULT_PORT
+get_mcp_server_url() {
+  local session_id="$1"
+  local port="$DEFAULT_PORT"
+
+  # Try session-specific port file first
+  if [ -n "$session_id" ]; then
+    local port_file="${CODEVIBE_TMPDIR}/codevibe-gemini-${session_id}.port"
+    if [ -f "$port_file" ]; then
+      port=$(cat "$port_file")
+      log "DEBUG" "Using port from session file: $port"
+      echo "http://localhost:${port}"
+      return
+    fi
+  fi
+
+  # Fallback: default port file (written at server startup before any session)
+  local default_port_file="${CODEVIBE_TMPDIR}/codevibe-gemini-default.port"
+  if [ -f "$default_port_file" ]; then
+    port=$(cat "$default_port_file")
+    log "DEBUG" "Using port from default file: $port"
+  else
+    log "DEBUG" "No port files found, using default port $port"
+  fi
+
+  echo "http://localhost:${port}"
+}
+
+# Read JSON input from stdin
+read_json_input() {
+  # Read all input from stdin
+  cat
+}
+
+# Send event to MCP server via HTTP POST
+# Usage: send_to_mcp <endpoint> <json_data> [session_id]
+send_to_mcp() {
+  local endpoint="$1"
+  local json_data="$2"
+  local session_id="$3"
+
+  # Get the MCP server URL for this session
+  local mcp_url=$(get_mcp_server_url "$session_id")
+
+  log "INFO" "Sending to MCP server: POST $mcp_url/$endpoint"
+  log "DEBUG" "Payload: $json_data"
+
+  # Send HTTP POST request to MCP server
+  local response
+  local http_code
+
+  response=$(curl -s -w "\n%{http_code}" \
+    -X POST \
+    -H "Content-Type: application/json" \
+    -d "$json_data" \
+    "$mcp_url/$endpoint" 2>&1)
+
+  http_code=$(echo "$response" | tail -n 1)
+  # Use sed instead of head -n -1 for macOS compatibility
+  local body=$(echo "$response" | sed '$d')
+
+  if [ "$http_code" = "200" ]; then
+    log "INFO" "Successfully sent to MCP server (HTTP $http_code)"
+    log "DEBUG" "Response: $body"
+    echo "$body"
+    return 0
+  else
+    log "ERROR" "Failed to send to MCP server (HTTP $http_code)"
+    log "ERROR" "Response: $body"
+    return 1
+  fi
+}
+
+# Send interactive prompt and wait for response
+# Usage: send_interactive_prompt <json_data> <session_id>
+# Returns: JSON with decision field or exits with error
+send_interactive_prompt() {
+  local json_data="$1"
+  local session_id="$2"
+
+  local mcp_url=$(get_mcp_server_url "$session_id")
+
+  log "INFO" "Sending interactive prompt to MCP server"
+
+  # Send the interactive prompt request
+  local response
+  local http_code
+
+  response=$(curl -s -w "\n%{http_code}" \
+    -X POST \
+    -H "Content-Type: application/json" \
+    -d "$json_data" \
+    "$mcp_url/interactive-prompt" 2>&1)
+
+  http_code=$(echo "$response" | tail -n 1)
+  local body=$(echo "$response" | sed '$d')
+
+  if [ "$http_code" != "200" ]; then
+    log "ERROR" "Failed to send interactive prompt (HTTP $http_code)"
+    log "ERROR" "Response: $body"
+    # Return default "ask" decision on error
+    echo '{"decision":"ask","reason":"Failed to communicate with MCP server"}'
+    return 1
+  fi
+
+  # Extract prompt ID from response
+  local prompt_id=$(echo "$body" | grep -o '"promptId"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*"\([^"]*\)"$/\1/')
+
+  if [ -z "$prompt_id" ]; then
+    log "ERROR" "No promptId in response: $body"
+    echo '{"decision":"ask","reason":"No prompt ID received"}'
+    return 1
+  fi
+
+  log "INFO" "Got prompt ID: $prompt_id, polling for response..."
+
+  # Poll for response
+  local elapsed=0
+  while [ $elapsed -lt $INTERACTIVE_PROMPT_TIMEOUT ]; do
+    sleep $POLL_INTERVAL
+    elapsed=$((elapsed + POLL_INTERVAL))
+
+    # Check for response
+    response=$(curl -s -w "\n%{http_code}" \
+      "$mcp_url/prompt-response/$prompt_id" 2>&1)
+
+    http_code=$(echo "$response" | tail -n 1)
+    body=$(echo "$response" | sed '$d')
+
+    if [ "$http_code" = "200" ]; then
+      # Check if we have a decision
+      local decision=$(echo "$body" | grep -o '"decision"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*"\([^"]*\)"$/\1/')
+
+      if [ -n "$decision" ] && [ "$decision" != "pending" ]; then
+        log "INFO" "Got decision: $decision after ${elapsed}s"
+        echo "$body"
+        return 0
+      fi
+    elif [ "$http_code" = "404" ]; then
+      # Prompt not found or expired
+      log "WARN" "Prompt not found or expired"
+      echo '{"decision":"ask","reason":"Prompt expired or not found"}'
+      return 1
+    fi
+
+    log "DEBUG" "Still waiting for response... (${elapsed}s / ${INTERACTIVE_PROMPT_TIMEOUT}s)"
+  done
+
+  # Timeout reached
+  log "WARN" "Timeout waiting for interactive prompt response after ${INTERACTIVE_PROMPT_TIMEOUT}s"
+  echo '{"decision":"ask","reason":"Timeout waiting for mobile response"}'
+  return 1
+}
+
+# Extract field from JSON using grep/sed (simple alternative to jq)
+# This is a basic implementation - for complex JSON, consider using jq
+extract_field() {
+  local json="$1"
+  local field="$2"
+
+  # Simple regex-based extraction (works for simple string values)
+  echo "$json" | grep -o "\"$field\"[[:space:]]*:[[:space:]]*\"[^\"]*\"" | sed "s/\"$field\"[[:space:]]*:[[:space:]]*\"\([^\"]*\)\"/\1/"
+}
+
+# Check if MCP server is running
+# Usage: check_mcp_server [session_id]
+check_mcp_server() {
+  local session_id="$1"
+  local mcp_url=$(get_mcp_server_url "$session_id")
+
+  log "DEBUG" "Checking if MCP server is running at $mcp_url"
+
+  local response
+  response=$(curl -s "$mcp_url/health" 2>&1)
+
+  if [ $? -eq 0 ]; then
+    log "DEBUG" "MCP server is running"
+    return 0
+  else
+    log "WARN" "MCP server is not responding at $mcp_url"
+    return 1
+  fi
+}
+
+# Export variables and functions for use in other scripts
+export CODEVIBE_TMPDIR
+export INTERACTIVE_PROMPT_TIMEOUT
+export POLL_INTERVAL
+export -f log
+export -f get_mcp_server_url
+export -f read_json_input
+export -f send_to_mcp
+export -f send_interactive_prompt
+export -f extract_field
+export -f check_mcp_server

package/hooks/hooks.json

@@ -0,0 +1,81 @@
+{
+  "hooks": {
+    "SessionStart": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/session-start.sh"
+          }
+        ]
+      }
+    ],
+    "SessionEnd": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/session-end.sh"
+          }
+        ]
+      }
+    ],
+    "BeforeAgent": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/before-agent.sh"
+          }
+        ]
+      }
+    ],
+    "AfterAgent": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/after-agent.sh"
+          }
+        ]
+      }
+    ],
+    "AfterTool": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/after-tool.sh"
+          }
+        ]
+      }
+    ],
+    "BeforeTool": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/before-tool.sh"
+          }
+        ]
+      }
+    ],
+    "Notification": [
+      {
+        "matcher": "*",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "bash ${GEMINI_EXTENSION_PATH}/hooks/notification.sh"
+          }
+        ]
+      }
+    ]
+  }
+}

package/hooks/notification.sh

@@ -0,0 +1,32 @@
+#!/bin/bash
+
+# Notification Hook
+# Triggered when Gemini CLI shows a notification
+# Forwards notifications to mobile app
+
+# Source common utilities
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+source "$SCRIPT_DIR/common.sh"
+
+log "INFO" "Notification hook triggered"
+
+# Read JSON input from stdin
+INPUT=$(read_json_input)
+
+log "DEBUG" "Input: $INPUT"
+
+# Extract session ID for dynamic port lookup
+SESSION_ID=$(echo "$INPUT" | jq -r '.session_id // empty')
+
+# Send to MCP server
+send_to_mcp "event" "$INPUT" "$SESSION_ID"
+
+EXIT_CODE=$?
+
+if [ $EXIT_CODE -eq 0 ]; then
+  log "INFO" "Notification event sent successfully"
+else
+  log "ERROR" "Failed to send Notification event (exit code: $EXIT_CODE)"
+fi
+
+exit $EXIT_CODE

package/hooks/session-end.sh

@@ -0,0 +1,70 @@
+#!/bin/bash
+
+# Gemini CLI SessionEnd hook
+# Called when a Gemini session ends
+#
+# Input (JSON via stdin):
+# {
+#   "session_id": "uuid",
+#   "hook_event_name": "SessionEnd",
+#   "reason": "user_exit" | "error" | etc,
+#   ...
+# }
+
+# Get script directory and source common utilities
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+source "$SCRIPT_DIR/common.sh"
+
+log "INFO" "SessionEnd hook triggered"
+
+# Read JSON input from stdin
+INPUT=$(read_json_input)
+
+if [ -z "$INPUT" ]; then
+  log "ERROR" "No input received"
+  exit 0
+fi
+
+log "DEBUG" "Received input: $INPUT"
+
+# Extract session_id
+SESSION_ID=$(echo "$INPUT" | grep -o '"session_id"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*"\([^"]*\)"$/\1/')
+
+if [ -z "$SESSION_ID" ]; then
+  log "ERROR" "No session_id in input"
+  exit 0
+fi
+
+# Extract reason if available
+REASON=$(echo "$INPUT" | grep -o '"reason"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/.*"\([^"]*\)"$/\1/')
+REASON="${REASON:-unknown}"
+
+log "INFO" "Session ID: $SESSION_ID, Reason: $REASON"
+
+# Check if MCP server is running
+if ! check_mcp_server "$SESSION_ID"; then
+  log "WARN" "MCP server not running, cannot send SessionEnd"
+  exit 0
+fi
+
+# Build event payload
+EVENT_PAYLOAD=$(cat <<PAYLOAD
+{
+  "session_id": "$SESSION_ID",
+  "hook_event_name": "SessionEnd",
+  "type": "NOTIFICATION",
+  "source": "DESKTOP",
+  "content": "Session ended: $REASON",
+  "metadata": {
+    "hook_event_name": "SessionEnd",
+    "reason": "$REASON"
+  }
+}
+PAYLOAD
+)
+
+# Send to MCP server
+send_to_mcp "event" "$EVENT_PAYLOAD" "$SESSION_ID"
+
+log "INFO" "SessionEnd hook completed"
+exit 0