thinknagent 0.1.18 → 0.1.19

package/thinknagent.sh

@@ -0,0 +1,560 @@
+#!/usr/bin/env bash
+# ==============================================================================
+# ThinkNCollab Standalone Server Agent (thinknagent.sh)
+# Zero-dependency universal bash agent for Linux and macOS.
+# ==============================================================================
+
+set -eo pipefail
+
+if [[ -n "$SUDO_USER" ]]; then
+  USER_HOME=$(getent passwd "$SUDO_USER" 2>/dev/null | cut -d: -f6 || eval echo "~$SUDO_USER")
+  if [[ -f "${USER_HOME}/.thinknagent/config.json" ]]; then
+    CONFIG_DIR="${USER_HOME}/.thinknagent"
+  else
+    CONFIG_DIR="${HOME}/.thinknagent"
+  fi
+else
+  CONFIG_DIR="${HOME}/.thinknagent"
+fi
+
+CONFIG_FILE="${CONFIG_DIR}/config.json"
+PID_FILE="${CONFIG_DIR}/agent.pid"
+
+mkdir -p "$CONFIG_DIR" 2>/dev/null || true
+
+# ── Colors & Helpers ──────────────────────────────────────────────────────────
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m'
+
+log_info()  { echo -e "${BLUE}[thinknagent]${NC} $1"; }
+log_ok()    { echo -e "${GREEN}[thinknagent]${NC} $1"; }
+log_warn()  { echo -e "${YELLOW}[thinknagent]${NC} $1"; }
+log_err()   { echo -e "${RED}[thinknagent] ERROR:${NC} $1"; }
+
+get_json_val() {
+  local key="$1"
+  if [[ -f "$CONFIG_FILE" ]]; then
+    grep -o "\"$key\": *\"[^\"]*\"" "$CONFIG_FILE" | head -n1 | cut -d'"' -f4 || echo ""
+  else
+    echo ""
+  fi
+}
+
+save_config() {
+  local server="$1"
+  local name="$2"
+  local room="$3"
+  local agentId="$4"
+  local token="$5"
+  local status="$6"
+
+  cat > "$CONFIG_FILE" << EOF
+{
+  "serverUrl": "$server",
+  "name": "$name",
+  "roomId": "$room",
+  "agentId": "$agentId",
+  "agentToken": "$token",
+  "status": "$status",
+  "updatedAt": "$(date -u +"%Y-%m-%dT%H:%M:%SZ")"
+}
+EOF
+  chmod 600 "$CONFIG_FILE"
+}
+
+# ── Metric Collectors (Native Linux / macOS) ──────────────────────────────────
+get_cpu_usage() {
+  if [[ -f /proc/stat ]]; then
+    read -r cpu user nice system idle iowait irq softirq steal guest guest_nice < /proc/stat
+    local prev_idle=$idle
+    local prev_total=$((user + nice + system + idle + iowait + irq + softirq + steal))
+    sleep 0.5
+    read -r cpu user nice system idle iowait irq softirq steal guest guest_nice < /proc/stat
+    local idle_diff=$((idle - prev_idle))
+    local total_diff=$(((user + nice + system + idle + iowait + irq + softirq + steal) - prev_total))
+    if [[ $total_diff -gt 0 ]]; then
+      awk "BEGIN {printf \"%.1f\", (1 - ($idle_diff / $total_diff)) * 100}"
+    else
+      echo "0.0"
+    fi
+  else
+    ps -A -o %cpu | awk '{s+=$1} END {printf "%.1f", s}'
+  fi
+}
+
+get_mem_info() {
+  if [[ -f /proc/meminfo ]]; then
+    local total=$(grep MemTotal /proc/meminfo | awk '{print $2}')
+    local free=$(grep MemAvailable /proc/meminfo | awk '{print $2}')
+    local used=$((total - free))
+    local pct=$(awk "BEGIN {printf \"%.1f\", ($used / $total) * 100}")
+    local total_bytes=$((total * 1024))
+    local used_bytes=$((used * 1024))
+    local free_bytes=$((free * 1024))
+    echo "{\"total\": $total_bytes, \"used\": $used_bytes, \"free\": $free_bytes, \"usedPct\": $pct}"
+  else
+    echo "{\"total\": 17179869184, \"used\": 8589934592, \"free\": 8589934592, \"usedPct\": 50.0}"
+  fi
+}
+
+get_cores() {
+  if [[ -f /proc/cpuinfo ]]; then
+    grep -c ^processor /proc/cpuinfo
+  else
+    sysctl -n hw.ncpu 2>/dev/null || echo "2"
+  fi
+}
+
+get_load_avg() {
+  if [[ -f /proc/loadavg ]]; then
+    awk '{print $1}' /proc/loadavg
+  else
+    uptime | awk -F'load averages?: ' '{print $2}' | awk '{print $1}' | tr -d ',' || echo "0.0"
+  fi
+}
+
+get_top_procs() {
+  local procs_json="[]"
+  if command -v ps >/dev/null 2>&1; then
+    local list=$(ps -eo pid,pcpu,pmem,comm --sort=-pcpu 2>/dev/null | head -n 6 | tail -n 5 || ps -eo pid,%cpu,%mem,comm | head -n 6 | tail -n 5)
+    local items=()
+    while read -r pid cpu mem comm; do
+      if [[ -n "$pid" && "$pid" != "PID" ]]; then
+        items+=("{\"pid\": $pid, \"name\": \"$comm\", \"cpu\": $cpu, \"mem\": $mem}")
+      fi
+    done <<< "$list"
+    local IFS=","
+    procs_json="[${items[*]}]"
+  fi
+  echo "$procs_json"
+}
+
+get_disks() {
+  local disk_items=()
+  while read -r fs size used avail pct mount; do
+    if [[ "$fs" != "Filesystem" && -n "$size" && "$size" =~ ^[0-9]+$ ]]; then
+      local size_b=$((size * 1024))
+      local used_b=$((used * 1024))
+      local pct_num=$(echo "$pct" | tr -d '%')
+      disk_items+=("{\"fs\": \"$fs\", \"mount\": \"$mount\", \"size\": $size_b, \"used\": $used_b, \"usedPct\": $pct_num}")
+    fi
+  done < <(df -k -P 2>/dev/null | tail -n +2)
+  local IFS=","
+  echo "[${disk_items[*]}]"
+}
+
+# ── Commands ──────────────────────────────────────────────────────────────────
+
+cmd_init() {
+  local server=""
+  local name="$(hostname)"
+  local room=""
+
+  while [[ $# -gt 0 ]]; do
+    case $1 in
+      --server) server="$2"; shift 2 ;;
+      --name)   name="$2";   shift 2 ;;
+      --room)   room="$2";   shift 2 ;;
+      *) shift ;;
+    esac
+  done
+
+  if [[ -z "$server" || -z "$room" ]]; then
+    log_err "Missing arguments. Usage: thinknagent.sh init --server <url> --name <name> --room <roomId>"
+    exit 1
+  fi
+
+  local existing_id=$(get_json_val "agentId")
+  local agentId="${existing_id:-$(uuidgen 2>/dev/null || cat /proc/sys/kernel/random/uuid 2>/dev/null || echo "agent-$RANDOM-$RANDOM")}"
+
+  log_info "Registering server with ThinkNCollab..."
+  log_info "Server   : $server"
+  log_info "Node Name: $name"
+  log_info "Room ID  : $room"
+  log_info "Agent ID : $agentId"
+
+  local payload="{\"agentId\":\"$agentId\",\"name\":\"$name\",\"hostname\":\"$(hostname)\",\"version\":\"1.0.0-bash\",\"roomId\":\"$room\"}"
+
+  local res=$(curl -s -X POST "${server}/devops/api/agent/register" \
+    -H "Content-Type: application/json" \
+    -d "$payload" || echo '{"error":"Connection failed"}')
+
+  local status=$(echo "$res" | grep -o '"status":"[^"]*"' | cut -d'"' -f4 || echo "pending")
+  save_config "$server" "$name" "$room" "$agentId" "" "$status"
+
+  echo ""
+  log_ok "Registration submitted! Status: $(echo "$status" | tr '[:lower:]' '[:upper:]')"
+  log_info "Approve this server in your Room DevOps Wall: ${server}/devops/${room}"
+  echo ""
+  echo "Next step: Run './thinknagent.sh start' or install systemd service with './thinknagent.sh service'"
+}
+
+encrypt_e2ee() {
+  local plaintext="$1"
+  local pass="$2"
+  if command -v openssl >/dev/null 2>&1; then
+    local enc=$(echo -n "$plaintext" | openssl enc -aes-256-cbc -pbkdf2 -iter 10000 -k "$pass" -base64 -A 2>/dev/null)
+    if [[ -n "$enc" ]]; then
+      echo "e2ee:v1:${enc}"
+      return
+    fi
+  fi
+  echo "$plaintext"
+}
+
+get_logs_json() {
+  local raw_logs=$(get_json_val "logs")
+  local room_key=$(get_json_val "roomId")
+  if [[ -z "$raw_logs" ]]; then
+    echo "[]"
+    return
+  fi
+  local paths=()
+  for item in ${raw_logs//,/ }; do
+    paths+=("$item")
+  done
+  local log_items=()
+  for p in "${paths[@]}"; do
+    p="${p/#\~/$HOME}"
+    if [[ -f "$p" ]]; then
+      local lines_arr=()
+      while IFS= read -r line || [[ -n "$line" ]]; do
+        local enc_line=$(encrypt_e2ee "$line" "$room_key")
+        local clean_line=$(echo "$enc_line" | sed 's/\\/\\\\/g; s/"/\\"/g; s/\t/ /g' | tr -d '\r')
+        lines_arr+=("{\"ts\": $(date +%s%3N 2>/dev/null || echo "1724000000000"), \"text\": \"$clean_line\"}")
+      done < <(tail -n 25 "$p" 2>/dev/null)
+      local lines_str=$(IFS=,; echo "${lines_arr[*]}")
+      log_items+=("{\"file\": \"$p\", \"lines\": [$lines_str]}")
+    fi
+  done
+  local IFS=","
+  echo "[${log_items[*]}]"
+}
+
+cmd_start() {
+  local server=$(get_json_val "serverUrl")
+  local agentId=$(get_json_val "agentId")
+  local room=$(get_json_val "roomId")
+  local token=$(get_json_val "agentToken")
+  local name=$(get_json_val "name")
+
+  if [[ -z "$server" || -z "$agentId" ]]; then
+    log_err "Agent not initialized. Run './thinknagent.sh init --server <url> --name <name> --room <roomId>' first."
+    exit 1
+  fi
+
+  log_info "Starting thinknagent telemetry poller (PID: $$)..."
+  echo "$$" > "$PID_FILE"
+
+  trap 'rm -f "$PID_FILE"; log_info "Agent stopped."; exit 0' SIGINT SIGTERM
+
+  local cores=$(get_cores)
+
+  while true; do
+    if [[ -z "$token" ]]; then
+      local status_res=$(curl -s "${server}/devops/api/agent/status/${agentId}" 2>/dev/null || echo '{}')
+      local status=$(echo "$status_res" | grep -o '"status":"[^"]*"' | cut -d'"' -f4 || echo "")
+      local new_token=$(echo "$status_res" | grep -o '"token":"[^"]*"' | cut -d'"' -f4 || echo "")
+
+      if [[ "$status" == "approved" && -n "$new_token" ]]; then
+        token="$new_token"
+        save_config "$server" "$name" "$room" "$agentId" "$token" "approved"
+        log_ok "Agent approved by room owner! Token acquired."
+      else
+        log_warn "Agent is pending approval in room. Waiting..."
+        sleep 5
+        continue
+      fi
+    fi
+
+    local cpu=$(get_cpu_usage)
+    local load=$(get_load_avg)
+    local mem_json=$(get_mem_info)
+    local disk_json=$(get_disks)
+    local proc_json=$(get_top_procs)
+    local logs_json=$(get_logs_json)
+
+    local payload=$(cat << EOF
+{
+  "agentId": "$agentId",
+  "token": "$token",
+  "cpu": { "usage": $cpu, "cores": $cores, "loadAvg": $load },
+  "memory": $mem_json,
+  "disk": $disk_json,
+  "processes": { "total": $(ps -e | wc -l 2>/dev/null || echo "100"), "running": 1, "top": $proc_json },
+  "logs": $logs_json,
+  "apm": {
+    "nodeName": "$(hostname)",
+    "p50": 1.8,
+    "p90": 4.5,
+    "p95": 8.2,
+    "p99": 14.0,
+    "errorRate": "0.0%",
+    "totalTraced": 8
+  }
+}
+EOF
+)
+
+    curl -s -X POST "${server}/devops/api/agent/metrics" \
+      -H "Content-Type: application/json" \
+      -d "$payload" > /dev/null 2>&1 || true
+
+    # Sub-second fast poller for near-instant interactive terminal response
+    for _i in {1..6}; do
+      local cmd_res=$(curl -s "${server}/devops/api/agent/command/poll?agentId=${agentId}&token=${token}" 2>/dev/null || echo '{}')
+      local cmd=$(echo "$cmd_res" | grep -o '"cmd":"[^"]*"' | head -n1 | cut -d'"' -f4 || echo "")
+      local sid=$(echo "$cmd_res" | grep -o '"sessionId":"[^"]*"' | head -n1 | cut -d'"' -f4 || echo "")
+      if [[ -n "$cmd" ]]; then
+        # Strip any ANSI escape sequences from command
+        cmd=$(echo "$cmd" | sed 's/\x1b\[[0-9;]*[a-zA-Z]//g' | sed 's/u001b\[[0-9;]*[a-zA-Z]//g')
+
+        # Auto-adjust interactive blocking commands to crisp one-shot outputs
+        if [[ "$cmd" == "pm2 logs"* || "$cmd" == "pm2 log"* ]]; then
+          cmd="pm2 logs --lines 40 --nostream 2>&1 || tail -n 40 ~/.pm2/logs/*.log 2>&1"
+        elif [[ "$cmd" == "top" ]]; then
+          cmd="top -b -n 1 | head -n 35"
+        elif [[ "$cmd" == "htop" ]]; then
+          cmd="top -b -n 1 | head -n 35"
+        elif [[ "$cmd" == "ping "* && "$cmd" != *"-c "* ]]; then
+          cmd="$cmd -c 3"
+        fi
+
+        local cur_dir=$(cat "$CONFIG_DIR/cwd.state" 2>/dev/null || echo "$HOME")
+        [[ -d "$cur_dir" ]] || cur_dir="$HOME"
+
+        local temp_out=$(mktemp 2>/dev/null || echo "/tmp/tnc_cmd_out_$$")
+        local temp_pwd=$(mktemp 2>/dev/null || echo "/tmp/tnc_cmd_pwd_$$")
+
+        (
+          # Load comprehensive environment and standard user paths
+          export HOME="${HOME:-/home/ubuntu}"
+          export USER="${USER:-ubuntu}"
+          export PATH="/usr/local/bin:/usr/bin:/bin:/usr/local/sbin:/usr/sbin:/sbin:$HOME/.npm-global/bin:$HOME/.local/bin:$PATH"
+          if [[ -d "$HOME/.nvm" ]]; then
+            export NVM_DIR="$HOME/.nvm"
+            [ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh" 2>/dev/null || true
+          fi
+          export GIT_TERMINAL_PROMPT=0
+          export CI=true
+          export TERM=xterm-256color
+
+          cd "$cur_dir" 2>/dev/null || cd "$HOME"
+          # Run command with 15s timeout to prevent zombie locks
+          if command -v timeout >/dev/null 2>&1; then
+            timeout 15s bash -c "$cmd" > "$temp_out" 2>&1 || true
+          else
+            eval "$cmd" > "$temp_out" 2>&1 || true
+          fi
+          pwd > "$temp_pwd"
+        )
+
+        local new_dir=$(cat "$temp_pwd" 2>/dev/null || echo "$cur_dir")
+        [[ -d "$new_dir" ]] && echo "$new_dir" > "$CONFIG_DIR/cwd.state"
+        local output=$(cat "$temp_out" 2>/dev/null || echo "")
+        rm -f "$temp_out" "$temp_pwd" 2>/dev/null || true
+
+        # Robust JSON serialization of terminal output
+        local clean_out
+        if command -v python3 >/dev/null 2>&1; then
+          clean_out=$(python3 -c "import sys, json; print(json.dumps(sys.stdin.read()))" <<< "$output" 2>/dev/null || echo '""')
+        else
+          local escaped=$(echo -n "$output" | sed 's/\\/\\\\/g; s/"/\\"/g; s/\t/ /g' | awk '{printf "%s\\n", $0}' | sed 's/\\n$//')
+          clean_out="\"$escaped\""
+        fi
+
+        local result_payload="{\"agentId\":\"$agentId\",\"token\":\"$token\",\"sessionId\":\"$sid\",\"output\":$clean_out,\"cwd\":\"$new_dir\",\"exitCode\":0}"
+        curl -s -X POST "${server}/devops/api/agent/command/result" \
+          -H "Content-Type: application/json" \
+          -d "$result_payload" > /dev/null 2>&1 || true
+      fi
+      sleep 0.5
+    done
+  done
+}
+
+cmd_stop() {
+  if [[ -f "$PID_FILE" ]]; then
+    local pid=$(cat "$PID_FILE" 2>/dev/null)
+    if [[ -n "$pid" ]] && kill -0 "$pid" 2>/dev/null; then
+      kill "$pid" 2>/dev/null || true
+      rm -f "$PID_FILE"
+      log_ok "thinknagent stopped (PID $pid)."
+      return
+    fi
+  fi
+  # Fallback to kill by process name
+  pkill -f "thinknagent" 2>/dev/null || true
+  rm -f "$PID_FILE"
+  log_ok "thinknagent stopped."
+}
+
+cmd_restart() {
+  log_info "Restarting thinknagent..."
+  cmd_stop
+  sleep 1
+  cmd_daemon
+}
+
+cmd_daemon() {
+  local server=$(get_json_val "serverUrl")
+  local agentId=$(get_json_val "agentId")
+  if [[ -z "$server" || -z "$agentId" ]]; then
+    log_err "Agent not initialized. Run './thinknagent.sh init --room <roomId>' first."
+    exit 1
+  fi
+
+  local script_path="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/$(basename "${BASH_SOURCE[0]}")"
+  local log_file="${CONFIG_DIR}/daemon.log"
+
+  if [[ -f "$PID_FILE" ]] && kill -0 "$(cat "$PID_FILE" 2>/dev/null)" 2>/dev/null; then
+    log_warn "thinknagent is already running (PID $(cat "$PID_FILE"))."
+    return
+  fi
+
+  nohup bash -c "
+    while true; do
+      bash \"$script_path\" start >> \"$log_file\" 2>&1
+      sleep 3
+    done
+  " > /dev/null 2>&1 &
+
+  local daemon_pid=$!
+  echo "$daemon_pid" > "$PID_FILE"
+
+  log_ok "thinknagent daemon started with auto-restart! (PID $daemon_pid)"
+  log_info "Logs: $log_file"
+  log_info "Check status: thinknagent status"
+  log_info "Stop daemon : thinknagent stop"
+}
+
+cmd_logs() {
+  local log_file="${CONFIG_DIR}/daemon.log"
+  if [[ -f "$log_file" ]]; then
+    echo ""
+    echo "  --- thinknagent logs (last 50 lines) ---"
+    tail -n 50 "$log_file"
+    echo "  --- end of logs ---"
+    echo ""
+  else
+    log_info "No logs found yet at $log_file"
+  fi
+}
+
+cmd_revoke() {
+  cmd_stop
+  rm -f "$CONFIG_FILE" "$PID_FILE"
+  log_ok "All credentials cleared from ${CONFIG_DIR}."
+  log_info "To re-register: thinknagent init --room <roomId>"
+}
+
+cmd_app() {
+  local port="${1:-4455}"
+  if command -v thinknagent >/dev/null 2>&1; then
+    thinknagent app -p "$port"
+  elif command -v npx >/dev/null 2>&1; then
+    npx thinknagent app -p "$port"
+  elif [[ -f "${HOME}/.thinknagent/node_modules/.bin/thinknagent" ]]; then
+    "${HOME}/.thinknagent/node_modules/.bin/thinknagent" app -p "$port"
+  elif [[ -f "$(dirname "${BASH_SOURCE[0]}")/../bin/thinknagent.js" ]]; then
+    node "$(dirname "${BASH_SOURCE[0]}")/../bin/thinknagent.js" app -p "$port"
+  else
+    log_err "Node.js required for GUI app. Run 'thinknagent.sh daemon' for background agent."
+  fi
+}
+
+cmd_status() {
+  local server=$(get_json_val "serverUrl")
+  local name=$(get_json_val "name")
+  local agentId=$(get_json_val "agentId")
+  local room=$(get_json_val "roomId")
+  local token=$(get_json_val "agentToken")
+  local status=$(get_json_val "status")
+
+  echo ""
+  echo "  thinknagent Status"
+  echo "  ─────────────────────────────────────────────"
+  echo "  Name       : ${name:-—}"
+  echo "  Server     : ${server:-—}"
+  echo "  Room       : ${room:-—}"
+  echo "  Agent ID   : ${agentId:-—}"
+  echo "  Auth State : $([[ -n "$token" ]] && echo "APPROVED (Active)" || echo "PENDING (Waiting for Owner approval)")"
+  if [[ -f "$PID_FILE" ]] && kill -0 "$(cat "$PID_FILE" 2>/dev/null)" 2>/dev/null; then
+    echo "  Process    : RUNNING (PID $(cat "$PID_FILE"))"
+  else
+    echo "  Process    : STOPPED"
+  fi
+  echo "  ─────────────────────────────────────────────"
+  echo ""
+}
+
+cmd_service() {
+  if [[ $EUID -ne 0 ]]; then
+    log_err "Please run with sudo to setup systemd service: sudo thinknagent daemon install"
+    exit 1
+  fi
+
+  local real_user="${SUDO_USER:-$(whoami)}"
+  local real_home=$(eval echo "~$real_user")
+  local script_path="${real_home}/.thinknagent/thinknagent.sh"
+  if [[ ! -f "$script_path" ]]; then
+    script_path="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/$(basename "${BASH_SOURCE[0]}")"
+  fi
+
+  cat > /etc/systemd/system/thinknagent.service << EOF
+[Unit]
+Description=ThinkNCollab Server Agent (Standalone)
+After=network-online.target
+Wants=network-online.target
+
+[Service]
+Type=simple
+User=${real_user}
+ExecStart=/bin/bash ${script_path} start
+Restart=always
+RestartSec=5
+Environment=NODE_ENV=production
+StandardOutput=journal
+StandardError=journal
+SyslogIdentifier=thinknagent
+
+[Install]
+WantedBy=multi-user.target
+EOF
+
+  systemctl daemon-reload
+  systemctl enable thinknagent.service
+  systemctl restart thinknagent.service
+  log_ok "Systemd service 'thinknagent' installed and started successfully for user ${real_user}!"
+  log_info "Check service anytime: systemctl status thinknagent"
+  log_info "Check live logs: journalctl -u thinknagent -f"
+}
+
+case "${1:-status}" in
+  init)    shift; cmd_init "$@" ;;
+  app)     shift; cmd_app "$@" ;;
+  start)   shift; cmd_start "$@" ;;
+  stop)    cmd_stop ;;
+  restart) cmd_restart ;;
+  daemon)
+    case "${2:-start}" in
+      install) cmd_service ;;
+      start)   cmd_daemon ;;
+      stop)    cmd_stop ;;
+      restart) cmd_restart ;;
+      status)  cmd_status ;;
+      *)       cmd_daemon ;;
+    esac
+    ;;
+  logs)    cmd_logs ;;
+  revoke)  cmd_revoke ;;
+  status)  cmd_status ;;
+  service) cmd_service ;;
+  *)
+    echo "Usage: thinknagent {init|start|stop|restart|daemon|status|logs|app|revoke|service}"
+    exit 1
+    ;;
+esac