@mrtrinhvn/ag-kit 1.1.7 → 1.1.9

package/bin/cli.js

@@ -1,12 +1,32 @@
 #!/usr/bin/env node
-
 const { program } = require('commander');
 const fs = require('fs');
 const path = require('path');
+const net = require('net');
 
 const TEMPLATE_DIR = path.join(__dirname, '..', 'template');
 const TARGET_AGENT_DIR = path.join(process.cwd(), '.agent');
 
+function isPortAvailable(port) {
+  return new Promise((resolve) => {
+    const server = net.createServer();
+    server.once('error', () => resolve(false));
+    server.once('listening', () => {
+      server.close();
+      resolve(true);
+    });
+    server.listen(port, '127.0.0.1');
+  });
+}
+
+async function findNextAvailablePort(startPort) {
+  let port = startPort;
+  while (!(await isPortAvailable(port))) {
+    port++;
+  }
+  return port;
+}
+
 // Helper copy đệ quy
 function copyRecursiveSync(src, dest) {
   const exists = fs.existsSync(src);
@@ -77,18 +97,33 @@ program.command('init')
       fs.writeFileSync(path.join(TARGET_AGENT_DIR, '.version'), pkg.version);
       console.log(`\x1b[32m✅ Đã cài ráp thành công thư mục .agent (phiên bản v${pkg.version}) vào Project!\x1b[0m`);
       
-      // Khởi tạo .env nếu chưa có
+      // Khởi động .env nếu chưa có
       const envPath = path.join(process.cwd(), '.env');
-      if (!fs.existsSync(envPath)) {
-        fs.writeFileSync(envPath, 'CDP_PORT=9555\n');
-        console.log('\x1b[34mℹ️  Đã tạo file .env mẫu với CDP_PORT=9555\x1b[0m');
-      } else {
-        const envContent = fs.readFileSync(envPath, 'utf-8');
-        if (!envContent.includes('CDP_PORT=')) {
-          fs.appendFileSync(envPath, '\nCDP_PORT=9555\n');
-          console.log('\x1b[34mℹ️  Đã thêm cấu hình CDP_PORT=9555 vào file .env hiện tại\x1b[0m');
+      (async () => {
+        const idePort = await findNextAvailablePort(9555);
+        const bridgePort = await findNextAvailablePort(idePort + 1);
+
+        if (!fs.existsSync(envPath)) {
+          const envTemplate = `# --- Remote Orchestration (Golden Combo) ---\nIDE_PORT=${idePort}\nBRIDGE_PORT=${bridgePort}\n\nPROJECT_NAME=ag-project\n`;
+          fs.writeFileSync(envPath, envTemplate);
+          console.log(`\x1b[34mℹ️  Đã tạo .env với IDE_PORT=${idePort} và BRIDGE_PORT=${bridgePort}\x1b[0m`);
+        } else {
+          let envContent = fs.readFileSync(envPath, 'utf-8');
+          let changed = false;
+          if (!envContent.includes('IDE_PORT=')) {
+            envContent += `\nIDE_PORT=${idePort}\n`;
+            changed = true;
+          }
+          if (!envContent.includes('BRIDGE_PORT=')) {
+            envContent += `BRIDGE_PORT=${bridgePort}\n`;
+            changed = true;
+          }
+          if (changed) {
+            fs.writeFileSync(envPath, envContent);
+            console.log(`\x1b[34mℹ️  Đã bổ sung cổng tự động: IDE_PORT=${idePort}, BRIDGE_PORT=${bridgePort}\x1b[0m`);
+          }
         }
-      }
+      })();
     } else {
       console.log('\x1b[33m⚠️ Thư mục .agent đã tồn tại.\x1b[0m');
       const vPath = path.join(TARGET_AGENT_DIR, '.version');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mrtrinhvn/ag-kit",
-  "version": "1.1.7",
+  "version": "1.1.9",
   "description": "Antigravity Kit Base Framework - Generic Agentic AI Programming Core",
   "main": "index.js",
   "bin": {

package/template/.agent/scripts/port_utils.sh

@@ -0,0 +1,67 @@
+#!/bin/bash
+
+# --- Port Management Utility (Golden V4) ---
+# Finds free ports and updates .env to ensure consistent project-wide communication.
+
+ENV_FILE=".env"
+
+# Logic tìm cổng trống
+function find_free_port() {
+    local port=$1
+    while netstat -atn | grep -q ":$port "; do
+        port=$((port + 1))
+    done
+    echo $port
+}
+
+# Cập nhật hoặc thêm cổng vào .env
+function update_env_port() {
+    local key=$1
+    local value=$2
+    
+    if grep -q "^${key}=" "$ENV_FILE"; then
+        # OS-compatible sed (Linux/macOS)
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i '' "s/^${key}=.*/${key}=${value}/" "$ENV_FILE"
+        else
+            sed -i "s/^${key}=.*/${key}=${value}/" "$ENV_FILE"
+        fi
+    else
+        echo "${key}=${value}" >> "$ENV_FILE"
+    fi
+}
+
+# --- Main Logic ---
+# 1. Đọc cổng hiện tại
+if [ -f "$ENV_FILE" ]; then
+    source "$ENV_FILE"
+fi
+
+ORIGINAL_IDE_PORT=${IDE_PORT:-9555}
+ORIGINAL_BRIDGE_PORT=${BRIDGE_PORT:-9556}
+
+# 2. Kiểm tra xung đột (IDE_PORT)
+# Nếu cổng IDE bận, và không phải là do IDE của chính dự án này chiếm giữ (kiểm tra qua /json endpoint)
+PORT_TAKEN=false
+if netstat -atn | grep -q ":$ORIGINAL_IDE_PORT "; then
+    # Kiểm tra xem folder dự án hiện tại có nằm trong danh sách tab/window của IDE không
+    if ! curl -s "http://127.0.0.1:$ORIGINAL_IDE_PORT/json" | grep -q "$PROJECT_DIR" > /dev/null 2>&1; then
+        PORT_TAKEN=true
+    fi
+fi
+
+if [ "$PORT_TAKEN" = true ]; then
+    echo "⚠️  Xung đột PORT phaát hieện! Đang tìm cổng mới..."
+    NEW_IDE_PORT=$(find_free_port $ORIGINAL_IDE_PORT)
+    NEW_BRIDGE_PORT=$(find_free_port $((NEW_IDE_PORT + 1)))
+    
+    update_env_port "IDE_PORT" "$NEW_IDE_PORT"
+    update_env_port "BRIDGE_PORT" "$NEW_BRIDGE_PORT"
+    
+    echo "✅ Đã cập nhật .env: IDE_PORT=$NEW_IDE_PORT, BRIDGE_PORT=$NEW_BRIDGE_PORT"
+    export IDE_PORT=$NEW_IDE_PORT
+    export BRIDGE_PORT=$NEW_BRIDGE_PORT
+else
+    export IDE_PORT=$ORIGINAL_IDE_PORT
+    export BRIDGE_PORT=$ORIGINAL_BRIDGE_PORT
+fi

package/template/.agent/scripts/receptionist_down.sh

@@ -0,0 +1,37 @@
+#!/bin/bash
+
+# --- Antigravity Unified Shutdown (Agent-Defined V3) ---
+# Dọn dẹp trạm gác theo Port Isolation.
+
+PROJECT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
+cd "$PROJECT_DIR"
+
+if [ -f .env ]; then
+    source .env
+fi
+
+BRIDGE_PORT=${BRIDGE_PORT:-9556}
+
+echo "════════════════════════════════════════════════════"
+echo "  🚑 SHUTTING DOWN GATEWAY [$BRIDGE_PORT]..."
+echo "════════════════════════════════════════════════════"
+
+# 1. Tắt Portal Bridge (HUD)
+BRIDGE_PID_FILE=".portal_bridge_${BRIDGE_PORT}.pid"
+if [ -f "$BRIDGE_PID_FILE" ]; then
+    PID=$(cat "$BRIDGE_PID_FILE")
+    echo "[1/2] 🧪 Stopping Portal Bridge [$BRIDGE_PORT] (PID: $PID)..."
+    kill $PID 2>/dev/null && rm "$BRIDGE_PID_FILE"
+    echo "      ✅ Bridge stopped."
+else
+    echo "[1/2] 🧪 Portal Bridge on port $BRIDGE_PORT not running."
+fi
+
+# 2. Tắt Docker Backend (Bot)
+echo "[2/2] 🤖 Stopping Receptionist (Docker)..."
+docker compose -f docker-compose.dev.yml stop ck-backend-dev
+echo "      ✅ Bot stopped."
+
+echo "════════════════════════════════════════════════════"
+echo "✅ SYSTEM IDLE."
+echo "════════════════════════════════════════════════════"

package/template/.agent/scripts/receptionist_up.sh

@@ -0,0 +1,53 @@
+#!/bin/bash
+
+# --- Antigravity Unified Receptionist (Agent-Defined V4) ---
+# Trạm gác hợp nhất: Bot + Portal Bridge (HUD) + Monitor Dashboard.
+# Hỗ trợ Đa dự án & Tự động xử lý Cổng (Smart Port Orchestration).
+
+# 1. Load Cấu hình & Xử lý Cổng Thông minh
+PROJECT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
+cd "$PROJECT_DIR"
+
+if [ -f .agent/scripts/port_utils.sh ]; then
+    source .agent/scripts/port_utils.sh
+else
+    source .env
+    export IDE_PORT=${IDE_PORT:-9555}
+    export BRIDGE_PORT=${BRIDGE_PORT:-9556}
+fi
+
+echo "════════════════════════════════════════════════════"
+echo "  🚑 AGENTIC GATEWAY - PORTAL [$BRIDGE_PORT]"
+echo "════════════════════════════════════════════════════"
+
+# 1. Kiểm tra Cổng IDE (CDP)
+echo -n "[1/4] 🛰️  Checking IDE Portal (Port $IDE_PORT)... "
+if curl -s http://127.0.0.1:$IDE_PORT/json/version > /dev/null; then
+    echo "✅ READY."
+else
+    echo "❌ NOT FOUND."
+    echo "      👉 Please start Antigravity in Debug mode on port $IDE_PORT"
+fi
+
+# 2. Khởi động Portal Bridge (HUD Injector)
+echo "[2/4] 🧪 Injected Golden HUD Bridge... "
+BRIDGE_PID_FILE=".portal_bridge_${BRIDGE_PORT}.pid"
+
+# Tắt bridge cũ nếu đang chạy trên port này
+if [ -f "$BRIDGE_PID_FILE" ]; then
+    kill $(cat "$BRIDGE_PID_FILE") 2>/dev/null
+fi
+
+nohup node scripts/ag_portal_bridge.js > ".portal_bridge_${BRIDGE_PORT}.log" 2>&1 &
+echo $! > "$BRIDGE_PID_FILE"
+echo "      ✅ Bridge running (PID: $(cat "$BRIDGE_PID_FILE") on Port $BRIDGE_PORT)"
+
+# 3. Khởi động Bot (Tùy chỉnh theo dự án)
+echo "[3/4] 🤖 Waking up Agent Bot... "
+# [TEMPLATE_NOTE]: Sếp hãy gõ lệnh khởi động Bot của mình vào đây.
+# Ví dụ: docker compose up -d (nếu dùng docker) hoặc npx tsx src/index.ts (nếu dùng local)
+echo "      ⚠️  Vui lòng cấu hình lệnh khởi động Bot trong .agent/scripts/receptionist_up.sh"
+
+# 4. Monitor
+echo "[4/4] 📊 Ready."
+echo "----------------------------------------------------"

package/template/.env.example

@@ -12,11 +12,15 @@ ALLOWED_USER_IDS="123456789,987654321"
 # ID Chat/Group được phép hoạt động (Để tránh Bot bị kéo vào group lạ)
 ALLOWED_CHAT_IDS="-100123456789"
 
-# [Cấu Hình IDE]
+# [Cấu Hình IDE & Remote]
 # Cổng Debug của Antigravity (Mặc định là 9555)
 # Phải khớp với tham số --remote-debugging-port khi mở Editor
 IDE_PORT=9555
 
+# Cổng của Portal Bridge (Mặc định là 9556)
+# Phải khớp với script start.sh và bot callback
+BRIDGE_PORT=9556
+
 # Tên dự án để Bot tự động tìm kiếm target phù hợp trong CDP
 PROJECT_NAME="ceogravity"
 

package/template/scripts/ag_hud.js

@@ -0,0 +1,162 @@
+/**
+ * Antigravity Golden HUD - Medical/Trading Theme
+ * Purpose: Inject a premium status dashboard into the IDE.
+ */
+export const AG_HUD_JS = `(async () => {
+    const HUD_ID = 'antigravity-golden-hud';
+    if (document.getElementById(HUD_ID)) return "ALREADY_EXISTS";
+
+    // 1. Create Styles
+    const style = document.createElement('style');
+    style.innerHTML = \`
+        #\${HUD_ID} {
+            position: fixed;
+            top: 15px;
+            right: 15px;
+            z-index: 999999;
+            background: rgba(10, 10, 10, 0.85);
+            backdrop-filter: blur(12px);
+            border: 1px solid rgba(255, 255, 255, 0.1);
+            border-radius: 12px;
+            padding: 12px;
+            width: 280px;
+            color: #fff;
+            font-family: 'Inter', system-ui, sans-serif;
+            box-shadow: 0 8px 32px rgba(0,0,0,0.5);
+            transition: all 0.3s ease;
+            user-select: none;
+        }
+        #\${HUD_ID}:hover {
+            border-color: #4caf50;
+            box-shadow: 0 0 20px rgba(76, 175, 80, 0.2);
+        }
+        #\${HUD_ID} .header {
+            display: flex;
+            align-items: center;
+            justify-content: space-between;
+            margin-bottom: 10px;
+            font-weight: 700;
+            letter-spacing: 0.5px;
+        }
+        #\${HUD_ID} .status-dot {
+            width: 8px;
+            height: 8px;
+            border-radius: 50%;
+            background: #4caf50;
+            display: inline-block;
+            margin-right: 6px;
+            animation: pulse 2s infinite;
+        }
+        @keyframes pulse {
+            0% { transform: scale(1); opacity: 1; }
+            50% { transform: scale(1.5); opacity: 0.5; }
+            100% { transform: scale(1); opacity: 1; }
+        }
+        #\${HUD_ID} .metric {
+            display: flex;
+            justify-content: space-between;
+            font-size: 12px;
+            margin-bottom: 6px;
+            color: #ccc;
+        }
+        #\${HUD_ID} .metric b { color: #fff; }
+        #\${HUD_ID} .action-btn {
+            background: #1a1a1a;
+            border: 1px solid #333;
+            color: #4caf50;
+            font-size: 11px;
+            padding: 6px;
+            border-radius: 6px;
+            width: 100%;
+            margin-top: 8px;
+            cursor: pointer;
+            text-align: center;
+            font-weight: 600;
+            transition: all 0.2s;
+        }
+        #\${HUD_ID} .action-btn:hover {
+            background: #4caf50;
+            color: #000;
+        }
+        #\${HUD_ID} .surgery-alert {
+            background: rgba(244, 67, 54, 0.2);
+            border: 1px solid #f44336;
+            color: #f44336;
+            padding: 8px;
+            border-radius: 8px;
+            font-size: 11px;
+            margin-top: 10px;
+            display: none;
+            animation: shake 0.5s infinite;
+        }
+        @keyframes shake {
+            0% { transform: translateX(0); }
+            25% { transform: translateX(2px); }
+            50% { transform: translateX(0); }
+            75% { transform: translateX(-2px); }
+            100% { transform: translateX(0); }
+        }
+    \`;
+    document.head.appendChild(style);
+
+    // 2. Create UI
+    const hud = document.createElement('div');
+    hud.id = HUD_ID;
+    hud.innerHTML = \`
+        <div class="header">
+            <span>🧠 CORE AGENT B1</span>
+            <span id="ag-status"><span class="status-dot"></span>ONLINE</span>
+        </div>
+        <div class="metric">AI Evolution: <b id="ag-gen">GEN-439</b></div>
+        <div class="metric">Win Rate: <b id="ag-win">47.7%</b></div>
+        <div class="metric">Current Phase: <b id="ag-phase">5m VSA Trading</b></div>
+        
+        <div class="surgery-alert" id="ag-alert">
+            🛑 <b>DOCTOR CALL REQUIRED!</b><br>AI Learning Stalled. Surgery needed.
+        </div>
+        
+        <div id="ag-toast" style="display:none; background:rgba(33, 150, 243, 0.9); color:#fff; padding:8px; border-radius:8px; margin-top:10px; font-size:11px; border-left:4px solid #0d47a1; backdrop-filter: blur(8px); box-shadow: 0 4px 12px rgba(0,0,0,0.3);">
+            <b>💬 TELEGRAM:</b> <span id="ag-toast-msg"></span>
+        </div>
+        
+        <div class="action-btn" id="ag-btn-surgery">🏥 START BRAIN SURGERY</div>
+    \`;
+    document.body.appendChild(hud);
+
+    // 3. Logic to update from Bridge
+    window.addEventListener('ag-portal-update', (e) => {
+        const data = e.detail;
+        if (data.gen) document.getElementById('ag-gen').innerText = data.gen;
+        if (data.win) document.getElementById('ag-win').innerText = data.win;
+        if (data.phase) document.getElementById('ag-phase').innerText = data.phase;
+        
+        const alert = document.getElementById('ag-alert');
+        if (data.status === 'DOCTOR_REQUIRED' || data.status === 'STALLED') {
+            alert.style.display = 'block';
+            document.getElementById('ag-status').innerHTML = '<span class="status-dot" style="background:#f44336"></span>REPAIR';
+        } else {
+            alert.style.display = 'none';
+            document.getElementById('ag-status').innerHTML = '<span class="status-dot" style="background:#4caf50"></span>ONLINE';
+        }
+    });
+
+    window.addEventListener('ag-portal-alert', (e) => {
+        const data = e.detail;
+        const toast = document.getElementById('ag-toast');
+        const toastMsg = document.getElementById('ag-toast-msg');
+        
+        toastMsg.innerText = data.message || "New message from Telegram.";
+        toast.style.display = 'block';
+        if (data.color) toast.style.background = data.color;
+        
+        // Auto-hide after 10 seconds
+        setTimeout(() => { toast.style.display = 'none'; }, 10000);
+    });
+
+    document.getElementById('ag-btn-surgery').onclick = () => {
+        console.log("AG_PORTAL_ACTION:START_SURGERY");
+        document.getElementById('ag-btn-surgery').innerText = "🚑 REQUESTING SURGEON...";
+    };
+
+    return "GOLDEN_HUD_INJECTED";
+})()`;

package/template/scripts/ag_portal_bridge.js

@@ -0,0 +1,141 @@
+import WebSocket from 'ws';
+import fs from 'fs';
+import path from 'path';
+import fetch from 'node-fetch';
+import http from 'http';
+import { fileURLToPath } from 'url';
+import dotenv from 'dotenv';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+dotenv.config({ path: path.join(__dirname, '../.env') });
+
+const HUD_JS_FILE = path.join(__dirname, 'ag_hud.js');
+const STATE_FILE = path.join(__dirname, '../backend/data/ai_state.json');
+const REQUEST_FILE = path.join(__dirname, '../backend/data/remote_requests.json');
+const IDE_PORT = process.env.IDE_PORT || 9555;
+const BRIDGE_PORT = process.env.BRIDGE_PORT || 9556;
+
+let globalWs = null;
+
+async function getWsUrl() {
+    try {
+        const resp = await fetch(`http://127.0.0.1:${IDE_PORT}/json/version`);
+        const data = await resp.json();
+        return data.webSocketDebuggerUrl;
+    } catch (err) {
+        return null;
+    }
+}
+
+async function injectHud(ws) {
+    const hudJsContent = fs.readFileSync(HUD_JS_FILE, 'utf8');
+    const script = hudJsContent.match(/export const AG_HUD_JS = `([\s\S]*)`;/)[1];
+
+    const message = {
+        id: 1,
+        method: "Runtime.evaluate",
+        params: {
+            expression: script,
+            awaitPromise: true,
+            userGesture: true
+        }
+    };
+    ws.send(JSON.stringify(message));
+}
+
+async function updateHud(ws, state) {
+    const updateEvent = {
+        id: Date.now(),
+        method: "Runtime.evaluate",
+        params: {
+            expression: `window.dispatchEvent(new CustomEvent('ag-portal-update', { detail: ${JSON.stringify(state)} }));`
+        }
+    };
+    ws.send(JSON.stringify(updateEvent));
+}
+
+async function pushInstantAlert(ws, alertData) {
+    const alertEvent = {
+        id: Date.now(),
+        method: "Runtime.evaluate",
+        params: {
+            expression: `window.dispatchEvent(new CustomEvent('ag-portal-alert', { detail: ${JSON.stringify(alertData)} }));`
+        }
+    };
+    ws.send(JSON.stringify(alertEvent));
+}
+
+function startHttpServer() {
+    const server = http.createServer((req, res) => {
+        if (req.method === 'POST' && (req.url === '/alert' || req.url === '/update')) {
+            let body = '';
+            req.on('data', chunk => { body += chunk; });
+            req.on('end', async () => {
+                try {
+                    const data = JSON.parse(body);
+                    if (globalWs && globalWs.readyState === WebSocket.OPEN) {
+                        if (req.url === '/alert') {
+                            await pushInstantAlert(globalWs, data);
+                        } else {
+                            await updateHud(globalWs, data);
+                        }
+                        res.writeHead(200, { 'Content-Type': 'application/json' });
+                        res.end(JSON.stringify({ status: 'ok' }));
+                    } else {
+                        res.writeHead(503, { 'Content-Type': 'application/json' });
+                        res.end(JSON.stringify({ error: 'Bridge not connected to IDE' }));
+                    }
+                } catch (e) {
+                    res.writeHead(400, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify({ error: 'Invalid JSON' }));
+                }
+            });
+        } else {
+            res.writeHead(404);
+            res.end();
+        }
+    });
+
+    server.listen(BRIDGE_PORT, '0.0.0.0', () => {
+        console.log(`📡  Instant Alert Server listening on port ${BRIDGE_PORT}`);
+    });
+}
+
+async function startBridge() {
+    console.log(`🛰️  Connecting to Antigravity Portal on port ${IDE_PORT}...`);
+    
+    const wsUrl = await getWsUrl();
+    if (!wsUrl) {
+        console.log("❌ IDE is not running or port 9555 is closed.");
+        process.exit(1);
+    }
+
+    const ws = new WebSocket(wsUrl);
+    globalWs = ws;
+
+    ws.on('open', async () => {
+        console.log("✅ Bridge Established. Injecting Golden HUD...");
+        await injectHud(ws);
+        
+        // Start HTTP server for instant pushes
+        startHttpServer();
+        
+        // Background polling for legacy state compatibility
+        setInterval(async () => {
+            if (fs.existsSync(STATE_FILE)) {
+                const state = JSON.parse(fs.readFileSync(STATE_FILE, 'utf8'));
+                await updateHud(ws, state);
+            }
+        }, 5000);
+    });
+
+    ws.on('message', (data) => {
+        const msg = JSON.parse(data);
+    });
+
+    ws.on('error', (err) => {
+        console.error("❌ WS Error:", err);
+    });
+}
+
+startBridge();

package/template/start.sh

@@ -17,6 +17,6 @@ if [ $? -ne 0 ]; then
     echo "⚠️ Cảnh báo: Cổng CDP ${IDE_PORT:-9555} chưa mở. Đang chờ Antigravity khởi động..."
 fi
 
-# 4. Khởi động Bridge và Bot (Sử dụng Lễ tân/Receptionist pattern)
+# 4. Khởi động Bridge và Bot (Agent-Defined)
 echo "🚀 Khởi động Lễ tân trực chiến (Agentic Gateway)..."
-./receptionist_up.sh
+bash .agent/scripts/receptionist_up.sh

package/template/stop_bot.sh

@@ -3,7 +3,7 @@
 
 echo "🛑 Đang tắt toàn bộ trạm điều hành Agentic..."
 
-# Gọi script hạ cánh tiêu chuẩn
-./receptionist_down.sh
+# Gọi script hạ cánh tiêu chuẩn (Agent-Defined)
+bash .agent/scripts/receptionist_down.sh
 
 echo "✅ Đã dọn dẹp xong."