@relipa/ai-flow-kit 0.0.5 → 0.0.6-beta.0

package/scripts/telemetry/crypto.js

@@ -1,20 +1,20 @@
-const crypto = require('crypto');
-
-/**
- * Sign an event object using HMAC-SHA256 and return a string <b64_json>.<b64_hmac_sig>
- */
-function signEvent(eventObj, teamSecret) {
-  const jsonStr = JSON.stringify(eventObj);
-  const b64_json = Buffer.from(jsonStr).toString('base64');
-  
-  // If no team_secret is provided, fallback to empty string to ensure HMAC doesn't throw,
-  // but it will fail verification on server.
-  const secret = teamSecret || '';
-  const b64_sig = crypto.createHmac('sha256', secret).update(jsonStr).digest('base64');
-  
-  return `${b64_json}.${b64_sig}`;
-}
-
-module.exports = {
-  signEvent
-};
+const crypto = require('crypto');
+
+/**
+ * Sign an event object using HMAC-SHA256 and return a string <b64_json>.<b64_hmac_sig>
+ */
+function signEvent(eventObj, teamSecret) {
+  const jsonStr = JSON.stringify(eventObj);
+  const b64_json = Buffer.from(jsonStr).toString('base64');
+  
+  // If no team_secret is provided, fallback to empty string to ensure HMAC doesn't throw,
+  // but it will fail verification on server.
+  const secret = teamSecret || '';
+  const b64_sig = crypto.createHmac('sha256', secret).update(jsonStr).digest('base64');
+  
+  return `${b64_json}.${b64_sig}`;
+}
+
+module.exports = {
+  signEvent
+};

package/scripts/telemetry/flush.js

@@ -1,162 +1,162 @@
-const fs = require("fs");
-const path = require("path");
-const os = require("os");
-const https = require("https");
-const { loadConfig, saveConfig } = require("./config");
-
-const GLOBAL_AIFLOW_DIR = path.join(os.homedir(), ".aiflow");
-const TELEMETRY_DIR = path.join(GLOBAL_AIFLOW_DIR, "telemetry");
-const BUFFER_PATH = path.join(TELEMETRY_DIR, "buffer.jsonl");
-const LOCK_PATH = path.join(TELEMETRY_DIR, "flush.lock");
-
-const MAX_BATCH_SIZE = 100;
-
-function sendData(url, payload) {
-  return new Promise((resolve, reject) => {
-    const data = JSON.stringify(payload);
-    let urlObj;
-    try {
-      urlObj = new URL(url);
-    } catch (e) {
-      return reject(new Error("Invalid URL"));
-    }
-
-    const options = {
-      hostname: urlObj.hostname,
-      path: urlObj.pathname + urlObj.search,
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        "Content-Length": Buffer.byteLength(data),
-      },
-      timeout: 5000,
-    };
-
-    const req = https.request(options, (res) => {
-      // Google Apps Script POST to /exec returns 302 *after* doPost has already executed.
-      // Treat 302 as success immediately — following the redirect is best-effort only.
-      // Never retry on 302: the data was already written; retrying causes duplicates.
-      if (res.statusCode === 302 || res.statusCode === 301) {
-        res.resume(); // drain response body
-        const redirectUrl = res.headers.location;
-        if (redirectUrl) {
-          https
-            .get(redirectUrl, (redirectRes) => {
-              redirectRes.resume(); // drain, ignore body
-            })
-            .on("error", () => {}); // best-effort, do not reject
-        }
-        resolve(true); // GAS already processed the request
-      } else {
-        let body = "";
-        res.on("data", (chunk) => (body += chunk));
-        res.on("end", () => {
-          if (body === "OK") resolve(true);
-          else reject(new Error(`Status ${res.statusCode}: ${body.substring(0, 300)}`));
-        });
-      }
-    });
-
-    req.on("timeout", () => {
-      req.destroy();
-      reject(new Error("Timeout"));
-    });
-    req.on("error", reject);
-    req.write(data);
-    req.end();
-  });
-}
-
-const wait = (ms) => new Promise((r) => setTimeout(r, ms));
-
-async function main() {
-  setTimeout(() => {
-    try { if (fs.existsSync(LOCK_PATH)) fs.unlinkSync(LOCK_PATH); } catch (e) {}
-    process.exit(0);
-  }, 40000); // 40s hard exit — always release lock before dying
-
-  // Debounce sleep: Wait 10s to accumulate rapid command executions
-  await wait(10000);
-
-  const cfg = loadConfig();
-  if (!cfg.enabled || !cfg.apps_script_url) return;
-
-  if (!fs.existsSync(BUFFER_PATH)) return;
-
-  const content = fs.readFileSync(BUFFER_PATH, "utf-8");
-  const lines = content.split("\n").filter((l) => l.trim().length > 0);
-
-  if (lines.length === 0) return;
-
-  // Validate line format <b64_json>.<b64_sig>
-  const validLines = lines.filter((l) => l.includes("."));
-
-  if (validLines.length === 0) {
-    fs.unlinkSync(BUFFER_PATH);
-    return;
-  }
-
-  // Take first MAX_BATCH_SIZE
-  const batch = validLines.slice(0, MAX_BATCH_SIZE);
-  const remaining = validLines.slice(MAX_BATCH_SIZE);
-
-  const payload = {
-    email: cfg.email,
-    install_id: cfg.install_id,
-    lines: batch,
-  };
-
-  let success = false;
-  let lastError = null;
-  const retries = [2000, 5000, 15000];
-
-  for (let i = 0; i <= retries.length; i++) {
-    try {
-      await sendData(cfg.apps_script_url, payload);
-      success = true;
-      break;
-    } catch (err) {
-      lastError = err;
-      if (
-        typeof err === "object" &&
-        err !== null &&
-        String(err.message).includes("Status 4")
-      ) {
-        // 4xx error (e.g. 403 Forbidden because of invalid HMAC) -> no point retrying
-        break;
-      }
-      if (i < retries.length) {
-        await wait(retries[i]);
-      }
-    }
-  }
-
-  if (success) {
-    saveConfig({ last_flush_ts: Date.now() });
-    if (remaining.length > 0) {
-      // Atomic write remaining
-      const tmpPath = BUFFER_PATH + ".tmp";
-      fs.writeFileSync(tmpPath, remaining.join("\n") + "\n", "utf-8");
-      fs.renameSync(tmpPath, BUFFER_PATH);
-    } else {
-      fs.unlinkSync(BUFFER_PATH);
-    }
-  } else if (lastError) {
-    try {
-      const errLog = path.join(path.dirname(BUFFER_PATH), "errors.log");
-      fs.appendFileSync(
-        errLog,
-        `[${new Date().toISOString()}] FLUSH ERROR: ${lastError.message}\n`,
-      );
-    } catch (e) {}
-  }
-
-  // Remove lock so next record() can spawn a new flusher
-  try {
-    if (fs.existsSync(LOCK_PATH)) {
-      fs.unlinkSync(LOCK_PATH);
-    }
-  } catch (e) {}
-}
-
-main().catch(() => process.exit(0));
+const fs = require("fs");
+const path = require("path");
+const os = require("os");
+const https = require("https");
+const { loadConfig, saveConfig } = require("./config");
+
+const GLOBAL_AIFLOW_DIR = path.join(os.homedir(), ".aiflow");
+const TELEMETRY_DIR = path.join(GLOBAL_AIFLOW_DIR, "telemetry");
+const BUFFER_PATH = path.join(TELEMETRY_DIR, "buffer.jsonl");
+const LOCK_PATH = path.join(TELEMETRY_DIR, "flush.lock");
+
+const MAX_BATCH_SIZE = 100;
+
+function sendData(url, payload) {
+  return new Promise((resolve, reject) => {
+    const data = JSON.stringify(payload);
+    let urlObj;
+    try {
+      urlObj = new URL(url);
+    } catch (e) {
+      return reject(new Error("Invalid URL"));
+    }
+
+    const options = {
+      hostname: urlObj.hostname,
+      path: urlObj.pathname + urlObj.search,
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Content-Length": Buffer.byteLength(data),
+      },
+      timeout: 5000,
+    };
+
+    const req = https.request(options, (res) => {
+      // Google Apps Script POST to /exec returns 302 *after* doPost has already executed.
+      // Treat 302 as success immediately — following the redirect is best-effort only.
+      // Never retry on 302: the data was already written; retrying causes duplicates.
+      if (res.statusCode === 302 || res.statusCode === 301) {
+        res.resume(); // drain response body
+        const redirectUrl = res.headers.location;
+        if (redirectUrl) {
+          https
+            .get(redirectUrl, (redirectRes) => {
+              redirectRes.resume(); // drain, ignore body
+            })
+            .on("error", () => {}); // best-effort, do not reject
+        }
+        resolve(true); // GAS already processed the request
+      } else {
+        let body = "";
+        res.on("data", (chunk) => (body += chunk));
+        res.on("end", () => {
+          if (body === "OK") resolve(true);
+          else reject(new Error(`Status ${res.statusCode}: ${body.substring(0, 300)}`));
+        });
+      }
+    });
+
+    req.on("timeout", () => {
+      req.destroy();
+      reject(new Error("Timeout"));
+    });
+    req.on("error", reject);
+    req.write(data);
+    req.end();
+  });
+}
+
+const wait = (ms) => new Promise((r) => setTimeout(r, ms));
+
+async function main() {
+  setTimeout(() => {
+    try { if (fs.existsSync(LOCK_PATH)) fs.unlinkSync(LOCK_PATH); } catch (e) {}
+    process.exit(0);
+  }, 40000); // 40s hard exit — always release lock before dying
+
+  // Debounce sleep: Wait 10s to accumulate rapid command executions
+  await wait(10000);
+
+  const cfg = loadConfig();
+  if (!cfg.enabled || !cfg.apps_script_url) return;
+
+  if (!fs.existsSync(BUFFER_PATH)) return;
+
+  const content = fs.readFileSync(BUFFER_PATH, "utf-8");
+  const lines = content.split("\n").filter((l) => l.trim().length > 0);
+
+  if (lines.length === 0) return;
+
+  // Validate line format <b64_json>.<b64_sig>
+  const validLines = lines.filter((l) => l.includes("."));
+
+  if (validLines.length === 0) {
+    fs.unlinkSync(BUFFER_PATH);
+    return;
+  }
+
+  // Take first MAX_BATCH_SIZE
+  const batch = validLines.slice(0, MAX_BATCH_SIZE);
+  const remaining = validLines.slice(MAX_BATCH_SIZE);
+
+  const payload = {
+    email: cfg.email,
+    install_id: cfg.install_id,
+    lines: batch,
+  };
+
+  let success = false;
+  let lastError = null;
+  const retries = [2000, 5000, 15000];
+
+  for (let i = 0; i <= retries.length; i++) {
+    try {
+      await sendData(cfg.apps_script_url, payload);
+      success = true;
+      break;
+    } catch (err) {
+      lastError = err;
+      if (
+        typeof err === "object" &&
+        err !== null &&
+        String(err.message).includes("Status 4")
+      ) {
+        // 4xx error (e.g. 403 Forbidden because of invalid HMAC) -> no point retrying
+        break;
+      }
+      if (i < retries.length) {
+        await wait(retries[i]);
+      }
+    }
+  }
+
+  if (success) {
+    saveConfig({ last_flush_ts: Date.now() });
+    if (remaining.length > 0) {
+      // Atomic write remaining
+      const tmpPath = BUFFER_PATH + ".tmp";
+      fs.writeFileSync(tmpPath, remaining.join("\n") + "\n", "utf-8");
+      fs.renameSync(tmpPath, BUFFER_PATH);
+    } else {
+      fs.unlinkSync(BUFFER_PATH);
+    }
+  } else if (lastError) {
+    try {
+      const errLog = path.join(path.dirname(BUFFER_PATH), "errors.log");
+      fs.appendFileSync(
+        errLog,
+        `[${new Date().toISOString()}] FLUSH ERROR: ${lastError.message}\n`,
+      );
+    } catch (e) {}
+  }
+
+  // Remove lock so next record() can spawn a new flusher
+  try {
+    if (fs.existsSync(LOCK_PATH)) {
+      fs.unlinkSync(LOCK_PATH);
+    }
+  } catch (e) {}
+}
+
+main().catch(() => process.exit(0));

package/scripts/telemetry/record.js

@@ -1,138 +1,138 @@
-const fs = require('fs');
-const path = require('path');
-const os = require('os');
-const crypto = require('crypto');
-const { spawn } = require('child_process');
-const { loadConfig, getMachineIdHash } = require('./config');
-const { signEvent } = require('./crypto');
-
-// Use relative path for getting package.json
-let PKG_VERSION = 'unknown';
-try {
-  const pkg = require('../../package.json');
-  PKG_VERSION = pkg.version;
-} catch (e) {}
-
-const GLOBAL_AIFLOW_DIR = path.join(os.homedir(), '.aiflow');
-const TELEMETRY_DIR = path.join(GLOBAL_AIFLOW_DIR, 'telemetry');
-const BUFFER_PATH = path.join(TELEMETRY_DIR, 'buffer.jsonl');
-const ERRORS_PATH = path.join(TELEMETRY_DIR, 'errors.log');
-
-const FLUSH_SCRIPT = path.join(__dirname, 'flush.js');
-
-function ensureDirectories() {
-  if (!fs.existsSync(TELEMETRY_DIR)) {
-    fs.mkdirSync(TELEMETRY_DIR, { recursive: true });
-  }
-}
-
-function findGitRoot(currentPath) {
-  if (!currentPath || currentPath === path.parse(currentPath).root) return null;
-  if (fs.existsSync(path.join(currentPath, '.git'))) return currentPath;
-  return findGitRoot(path.dirname(currentPath));
-}
-
-function isRepoOptedOut() {
-  const repoRoot = findGitRoot(process.cwd());
-  if (!repoRoot) return false;
-  return fs.existsSync(path.join(repoRoot, '.aiflow', 'no-telemetry'));
-}
-
-function detectRepoName() {
-  const repoRoot = findGitRoot(process.cwd());
-  if (repoRoot) return path.basename(repoRoot);
-  return path.basename(process.cwd());
-}
-
-function logInternalError(err) {
-  try {
-    ensureDirectories();
-    const ts = new Date().toISOString();
-    const msg = `[${ts}] ${err.name}: ${err.message}\n${err.stack || ''}\n`;
-    fs.appendFileSync(ERRORS_PATH, msg);
-    // Simple log rotation if > 1MB
-    const stats = fs.statSync(ERRORS_PATH);
-    if (stats.size > 1024 * 1024) {
-      fs.renameSync(ERRORS_PATH, ERRORS_PATH + '.1');
-    }
-  } catch (e) {}
-}
-
-function record(eventType, payload = {}) {
-  try {
-    const cfg = loadConfig();
-    if (!cfg.enabled) return;
-    if (isRepoOptedOut()) return;
-
-    ensureDirectories();
-
-    const event = {
-      event_id: crypto.randomUUID(),
-      ts: new Date().toISOString(),
-      email: cfg.email || '',
-      install_id: cfg.install_id || '',
-      machine_id: getMachineIdHash(),
-      repo_name: detectRepoName(),
-      pkg_version: PKG_VERSION,
-      os: process.platform,
-      node_version: process.version,
-      event_type: eventType,
-      ...payload
-    };
-
-    // Drop any full content payload keys immediately just in case caller included them
-    delete event.prompt_content;
-    delete event.response_content;
-
-    const lineStr = signEvent(event, cfg.team_secret);
-    fs.appendFileSync(BUFFER_PATH, lineStr + '\n');
-    
-    // Check if buffer is getting huge (> 10MB cap). Simple clear
-    try {
-      if (fs.statSync(BUFFER_PATH).size > 10 * 1024 * 1024) {
-        fs.unlinkSync(BUFFER_PATH);
-      }
-    } catch (e) {}
-
-    maybeSpawnFlusher(cfg);
-  } catch (err) {
-    logInternalError(err);
-  }
-}
-
-const LOCK_PATH = path.join(TELEMETRY_DIR, 'flush.lock');
-
-function safeStatSize(filePath) {
-  try { return fs.statSync(filePath).size; } catch (err) { return 0; }
-}
-
-function maybeSpawnFlusher(cfg) {
-  const bufferSize = safeStatSize(BUFFER_PATH);
-  if (bufferSize === 0) return;
-
-  // Check if a flusher is already running in background (lock < 15s old)
-  try {
-    if (fs.existsSync(LOCK_PATH)) {
-      const stat = fs.statSync(LOCK_PATH);
-      if (Date.now() - stat.mtimeMs < 15000) {
-        return; // Already debounced, existing process will flush our new data soon
-      }
-    }
-  } catch (err) {}
-
-  try {
-    // Touch lock file
-    fs.writeFileSync(LOCK_PATH, Date.now().toString());
-    const child = spawn(process.execPath, [FLUSH_SCRIPT], {
-      detached: true,
-      stdio: 'ignore',
-      windowsHide: true,
-      env: process.env
-    });
-    child.unref();
-  } catch (err) {
-    logInternalError(err);
-  }
-}
-
-module.exports = { record };
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const crypto = require('crypto');
+const { spawn } = require('child_process');
+const { loadConfig, getMachineIdHash } = require('./config');
+const { signEvent } = require('./crypto');
+
+// Use relative path for getting package.json
+let PKG_VERSION = 'unknown';
+try {
+  const pkg = require('../../package.json');
+  PKG_VERSION = pkg.version;
+} catch (e) {}
+
+const GLOBAL_AIFLOW_DIR = path.join(os.homedir(), '.aiflow');
+const TELEMETRY_DIR = path.join(GLOBAL_AIFLOW_DIR, 'telemetry');
+const BUFFER_PATH = path.join(TELEMETRY_DIR, 'buffer.jsonl');
+const ERRORS_PATH = path.join(TELEMETRY_DIR, 'errors.log');
+
+const FLUSH_SCRIPT = path.join(__dirname, 'flush.js');
+
+function ensureDirectories() {
+  if (!fs.existsSync(TELEMETRY_DIR)) {
+    fs.mkdirSync(TELEMETRY_DIR, { recursive: true });
+  }
+}
+
+function findGitRoot(currentPath) {
+  if (!currentPath || currentPath === path.parse(currentPath).root) return null;
+  if (fs.existsSync(path.join(currentPath, '.git'))) return currentPath;
+  return findGitRoot(path.dirname(currentPath));
+}
+
+function isRepoOptedOut() {
+  const repoRoot = findGitRoot(process.cwd());
+  if (!repoRoot) return false;
+  return fs.existsSync(path.join(repoRoot, '.aiflow', 'no-telemetry'));
+}
+
+function detectRepoName() {
+  const repoRoot = findGitRoot(process.cwd());
+  if (repoRoot) return path.basename(repoRoot);
+  return path.basename(process.cwd());
+}
+
+function logInternalError(err) {
+  try {
+    ensureDirectories();
+    const ts = new Date().toISOString();
+    const msg = `[${ts}] ${err.name}: ${err.message}\n${err.stack || ''}\n`;
+    fs.appendFileSync(ERRORS_PATH, msg);
+    // Simple log rotation if > 1MB
+    const stats = fs.statSync(ERRORS_PATH);
+    if (stats.size > 1024 * 1024) {
+      fs.renameSync(ERRORS_PATH, ERRORS_PATH + '.1');
+    }
+  } catch (e) {}
+}
+
+function record(eventType, payload = {}) {
+  try {
+    const cfg = loadConfig();
+    if (!cfg.enabled) return;
+    if (isRepoOptedOut()) return;
+
+    ensureDirectories();
+
+    const event = {
+      event_id: crypto.randomUUID(),
+      ts: new Date().toISOString(),
+      email: cfg.email || '',
+      install_id: cfg.install_id || '',
+      machine_id: getMachineIdHash(),
+      repo_name: detectRepoName(),
+      pkg_version: PKG_VERSION,
+      os: process.platform,
+      node_version: process.version,
+      event_type: eventType,
+      ...payload
+    };
+
+    // Drop any full content payload keys immediately just in case caller included them
+    delete event.prompt_content;
+    delete event.response_content;
+
+    const lineStr = signEvent(event, cfg.team_secret);
+    fs.appendFileSync(BUFFER_PATH, lineStr + '\n');
+    
+    // Check if buffer is getting huge (> 10MB cap). Simple clear
+    try {
+      if (fs.statSync(BUFFER_PATH).size > 10 * 1024 * 1024) {
+        fs.unlinkSync(BUFFER_PATH);
+      }
+    } catch (e) {}
+
+    maybeSpawnFlusher(cfg);
+  } catch (err) {
+    logInternalError(err);
+  }
+}
+
+const LOCK_PATH = path.join(TELEMETRY_DIR, 'flush.lock');
+
+function safeStatSize(filePath) {
+  try { return fs.statSync(filePath).size; } catch (err) { return 0; }
+}
+
+function maybeSpawnFlusher(cfg) {
+  const bufferSize = safeStatSize(BUFFER_PATH);
+  if (bufferSize === 0) return;
+
+  // Check if a flusher is already running in background (lock < 15s old)
+  try {
+    if (fs.existsSync(LOCK_PATH)) {
+      const stat = fs.statSync(LOCK_PATH);
+      if (Date.now() - stat.mtimeMs < 15000) {
+        return; // Already debounced, existing process will flush our new data soon
+      }
+    }
+  } catch (err) {}
+
+  try {
+    // Touch lock file
+    fs.writeFileSync(LOCK_PATH, Date.now().toString());
+    const child = spawn(process.execPath, [FLUSH_SCRIPT], {
+      detached: true,
+      stdio: 'ignore',
+      windowsHide: true,
+      env: process.env
+    });
+    child.unref();
+  } catch (err) {
+    logInternalError(err);
+  }
+}
+
+module.exports = { record };