thinknagent 0.1.20 → 0.1.22

package/thinknagent.sh

@@ -1,554 +0,0 @@
-#!/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
-)
-
-    local metrics_res=$(curl -s -X POST "${server}/devops/api/agent/metrics" \
-      -H "Content-Type: application/json" \
-      -d "$payload" 2>/dev/null || echo '{}')
-
-    if [[ "$metrics_res" == *"Invalid"* || "$metrics_res" == *"not approved"* || "$metrics_res" == *"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"
-      fi
-    fi
-
-    # 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_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 "$@" ;;
-  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