@dainprotocol/cli 1.2.28 → 1.2.31

package/dist/utils.js

@@ -1,4 +1,15 @@
 "use strict";
+var __assign = (this && this.__assign) || function () {
+    __assign = Object.assign || function(t) {
+        for (var s, i = 1, n = arguments.length; i < n; i++) {
+            s = arguments[i];
+            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
+                t[p] = s[p];
+        }
+        return t;
+    };
+    return __assign.apply(this, arguments);
+};
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -39,6 +50,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_API_BASE_URL = exports.DEFAULT_PLATFORM_BASE_URL = exports.DEFAULT_TUNNEL_BASE_URL = void 0;
+exports.normalizeBaseUrl = normalizeBaseUrl;
+exports.joinUrl = joinUrl;
+exports.parseEnvContent = parseEnvContent;
 exports.getDainConfig = getDainConfig;
 exports.displayTunnelUrl = displayTunnelUrl;
 exports.setupProxy = setupProxy;
@@ -47,51 +62,101 @@ exports.logSuccess = logSuccess;
 exports.logInfo = logInfo;
 exports.getStaticFilesPath = getStaticFilesPath;
 exports.extractOrgId = extractOrgId;
+exports.resolveOrgId = resolveOrgId;
+exports.fetchWithTimeout = fetchWithTimeout;
+exports.buildPlatformUrl = buildPlatformUrl;
+exports.validatePlatformConfig = validatePlatformConfig;
+exports.platformRequest = platformRequest;
+exports.createIdempotentCleanup = createIdempotentCleanup;
 var fs_1 = __importDefault(require("fs"));
 var path_1 = __importDefault(require("path"));
 var client_1 = require("@dainprotocol/tunnel/client");
 var ora_1 = __importDefault(require("ora"));
 var chalk_1 = __importDefault(require("chalk"));
 var dotenv_1 = __importDefault(require("dotenv"));
+exports.DEFAULT_TUNNEL_BASE_URL = "wss://tunnel.dain-local.com";
+exports.DEFAULT_PLATFORM_BASE_URL = "https://codegen-deploy-service.dainapp.com";
+exports.DEFAULT_API_BASE_URL = "https://dain-platform-ochre.vercel.app";
+function normalizeBaseUrl(value) {
+    var trimmed = value.trim();
+    if (!trimmed)
+        return trimmed;
+    try {
+        return new URL(trimmed).toString().replace(/\/+$/, "");
+    }
+    catch (_a) {
+        return trimmed.replace(/\/+$/, "");
+    }
+}
+function joinUrl(baseUrl, pathname) {
+    return "".concat(normalizeBaseUrl(baseUrl), "/").concat(pathname.replace(/^\/+/, ""));
+}
+function parseEnvContent(envContent) {
+    return Object.entries(dotenv_1.default.parse(envContent))
+        .map(function (_a) {
+        var name = _a[0], value = _a[1];
+        return ({ name: name.trim(), value: value });
+    })
+        .filter(function (env) { return env.name !== "" && env.value !== ""; });
+}
+function parseDainApiKey(apiKey) {
+    if (!(apiKey === null || apiKey === void 0 ? void 0 : apiKey.startsWith("sk_agent_")))
+        return null;
+    var parts = apiKey.split("_");
+    if (parts.length < 5)
+        return null;
+    var isOrgFormat = parts[2] === "org";
+    if (isOrgFormat) {
+        var orgId_1 = parts[3];
+        var isAgentKeyword = parts[4] === "agent";
+        var agentId_1 = parts[5];
+        var secret_1 = parts.slice(6).join("_");
+        if (!orgId_1 || !isAgentKeyword || !agentId_1 || !secret_1)
+            return null;
+        return { agentId: agentId_1, orgId: orgId_1, secret: secret_1 };
+    }
+    var agentId = parts[2];
+    var orgId = parts[3];
+    var secret = parts.slice(4).join("_");
+    if (!agentId || !orgId || !secret)
+        return null;
+    return { agentId: agentId, orgId: orgId, secret: secret };
+}
 function loadEnvFiles() {
     var envFiles = [".env.local", ".env", ".env.development", ".env.production"];
-    envFiles.forEach(function (file) {
+    for (var _i = 0, envFiles_1 = envFiles; _i < envFiles_1.length; _i++) {
+        var file = envFiles_1[_i];
         var envPath = path_1.default.join(process.cwd(), file);
         if (fs_1.default.existsSync(envPath)) {
             dotenv_1.default.config({ path: envPath });
         }
-    });
+    }
 }
 function getDainConfig(configFile) {
-    loadEnvFiles(); // Loads all env files first
-    var defaultConfigPath = path_1.default.join(process.cwd(), "dain.json");
-    var configPath = configFile ? path_1.default.join(process.cwd(), configFile) : defaultConfigPath;
+    loadEnvFiles();
+    var configPath = path_1.default.join(process.cwd(), configFile || "dain.json");
     if (!fs_1.default.existsSync(configPath)) {
         logError("Configuration file not found: ".concat(configPath));
         process.exit(1);
     }
     try {
-        var configData = fs_1.default.readFileSync(configPath, "utf8");
-        var config = JSON.parse(configData);
-        // Validate required fields
-        if (!config["main-file"]) {
+        var config = JSON.parse(fs_1.default.readFileSync(configPath, "utf8"));
+        if (!config["main-file"])
             throw new Error("Configuration must include 'main-file'");
-        }
-        // Set default values for optional fields
-        config["environment"] = config["environment"] || "development";
-        config["version"] = config["version"] || "1.0.0";
-        config["out-dir"] = config["out-dir"] || "dist"; // Default to 'dist' if not specified
-        config["tunnel-base-url"] = config["tunnel-base-url"] || "wss://tunnel.dain-local.com"; // Default value if not specified
-        config["runtime"] = config["runtime"] || "node"; // Add this line
-        // Handle API key
-        if (!config["api-key"] ||
-            config["api-key"] === "env" ||
-            config["api-key"] === "MUST PUT IN .env.development as DAIN_API_KEY=YOUR_API_KEY") {
+        config.environment || (config.environment = "development");
+        config.version || (config.version = "1.0.0");
+        config["out-dir"] || (config["out-dir"] = "dist");
+        config.runtime || (config.runtime = "node");
+        config["tunnel-base-url"] = normalizeBaseUrl(config["tunnel-base-url"] || exports.DEFAULT_TUNNEL_BASE_URL);
+        if (config["platform-base-url"])
+            config["platform-base-url"] = normalizeBaseUrl(config["platform-base-url"]);
+        if (config["api-base-url"])
+            config["api-base-url"] = normalizeBaseUrl(config["api-base-url"]);
+        if (!config["api-key"] || config["api-key"] === "env" || config["api-key"].startsWith("MUST PUT")) {
             config["api-key"] = process.env.DAIN_API_KEY;
         }
-        if (!config["api-key"]) {
+        if (!config["api-key"])
             throw new Error("API key is not set in config or DAIN_API_KEY environment variable");
-        }
         return config;
     }
     catch (error) {
@@ -99,13 +164,7 @@ function getDainConfig(configFile) {
     }
 }
 function displayTunnelUrl(tunnelUrl) {
-    var divider = chalk_1.default.green("------------------------------------------------------------");
-    var header = chalk_1.default.green("Your service is available publicly at:");
-    var url = chalk_1.default.cyan.underline(tunnelUrl);
-    var info = chalk_1.default.yellow("This service URL can be connected to by a DAIN client or assistant");
-    var subInfo = chalk_1.default.yellow("(such as butterfly in development mode)");
-    var warning = chalk_1.default.red("You should not visit this URL directly");
-    console.log("\n".concat(divider, "\n").concat(header, "\n").concat(url, "\n\n").concat(info, "\n").concat(subInfo, "\n\n").concat(warning, "\n").concat(divider, "\n"));
+    console.log("\n".concat(chalk_1.default.green("------------------------------------------------------------"), "\n").concat(chalk_1.default.green("Your service is available publicly at:"), "\n").concat(chalk_1.default.cyan.underline(tunnelUrl), "\n\n").concat(chalk_1.default.yellow("This service URL can be connected to by a DAIN client or assistant"), "\n").concat(chalk_1.default.yellow("(such as butterfly in development mode)"), "\n\n").concat(chalk_1.default.red("You should not visit this URL directly"), "\n").concat(chalk_1.default.green("------------------------------------------------------------"), "\n"));
 }
 function setupProxy(port, apiKey, config) {
     return __awaiter(this, void 0, void 0, function () {
@@ -117,12 +176,11 @@ function setupProxy(port, apiKey, config) {
                     _a.label = 1;
                 case 1:
                     _a.trys.push([1, 3, , 4]);
-                    client = new client_1.DainTunnel(config["tunnel-base-url"] || "wss://tunnel.dain-local.com", apiKey);
+                    client = new client_1.DainTunnel(config["tunnel-base-url"] || exports.DEFAULT_TUNNEL_BASE_URL, apiKey);
                     return [4 /*yield*/, client.start(parseInt(port))];
                 case 2:
                     tunnelUrl = _a.sent();
                     spinner.succeed("Proxy setup complete");
-                    displayTunnelUrl(tunnelUrl);
                     return [2 /*return*/, { client: client, tunnelUrl: tunnelUrl }];
                 case 3:
                     error_1 = _a.sent();
@@ -136,9 +194,8 @@ function setupProxy(port, apiKey, config) {
 }
 function logError(message, error) {
     console.error(chalk_1.default.red("\nError: ".concat(message)));
-    if (error) {
+    if (error)
         console.error(chalk_1.default.red(error));
-    }
 }
 function logSuccess(message) {
     console.log(chalk_1.default.green("\nSuccess: ".concat(message)));
@@ -150,6 +207,83 @@ function getStaticFilesPath() {
     return path_1.default.join(process.cwd(), "static");
 }
 function extractOrgId(apiKey) {
-    var apiKeySplit = apiKey === null || apiKey === void 0 ? void 0 : apiKey.split("_");
-    return apiKeySplit ? apiKeySplit[2] : "";
+    var _a;
+    return ((_a = parseDainApiKey(apiKey)) === null || _a === void 0 ? void 0 : _a.orgId) || "";
+}
+function resolveOrgId(config) {
+    return config["org-id"] || process.env.DAIN_ORG_ID || extractOrgId(config["api-key"]);
+}
+function fetchWithTimeout(url_1, options_1) {
+    return __awaiter(this, arguments, void 0, function (url, options, timeoutMs) {
+        var controller, timeoutId;
+        if (timeoutMs === void 0) { timeoutMs = 30000; }
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    controller = new AbortController();
+                    timeoutId = setTimeout(function () { return controller.abort(); }, timeoutMs);
+                    _a.label = 1;
+                case 1:
+                    _a.trys.push([1, , 3, 4]);
+                    return [4 /*yield*/, fetch(url, __assign(__assign({}, options), { signal: controller.signal }))];
+                case 2: return [2 /*return*/, _a.sent()];
+                case 3:
+                    clearTimeout(timeoutId);
+                    return [7 /*endfinally*/];
+                case 4: return [2 /*return*/];
+            }
+        });
+    });
+}
+function buildPlatformUrl(config, orgId, endpoint, overrides) {
+    var baseUrl = config["platform-base-url"] || exports.DEFAULT_PLATFORM_BASE_URL;
+    var deploymentId = (overrides === null || overrides === void 0 ? void 0 : overrides.deploymentId) || config["deployment-id"];
+    var environment = (overrides === null || overrides === void 0 ? void 0 : overrides.environment) || config["environment"];
+    var serviceId = (overrides === null || overrides === void 0 ? void 0 : overrides.serviceId) || config["service-id"];
+    return joinUrl(baseUrl, "/codegen-deploy/".concat(endpoint, "/").concat(orgId, "/").concat(serviceId, "/").concat(deploymentId, "/").concat(environment));
+}
+function validatePlatformConfig(config, overrides) {
+    var orgId = resolveOrgId(config);
+    var apiKey = config["api-key"];
+    var deploymentId = (overrides === null || overrides === void 0 ? void 0 : overrides.deploymentId) || config["deployment-id"];
+    var serviceId = (overrides === null || overrides === void 0 ? void 0 : overrides.serviceId) || config["service-id"];
+    if (!orgId) {
+        logError("Org ID not found. Ensure your API key or DAIN_ORG_ID is set.");
+        process.exit(1);
+    }
+    if (!deploymentId || !serviceId) {
+        logError("Deployment ID or service ID not found");
+        process.exit(1);
+    }
+    if (!apiKey) {
+        logError("API key not found");
+        process.exit(1);
+    }
+    return { config: config, orgId: orgId, apiKey: apiKey, deploymentId: deploymentId, serviceId: serviceId };
+}
+function platformRequest(ctx_1, endpoint_1) {
+    return __awaiter(this, arguments, void 0, function (ctx, endpoint, method, overrides) {
+        var url;
+        if (method === void 0) { method = "GET"; }
+        return __generator(this, function (_a) {
+            url = buildPlatformUrl(ctx.config, ctx.orgId, endpoint, {
+                deploymentId: ctx.deploymentId,
+                serviceId: ctx.serviceId,
+                environment: overrides === null || overrides === void 0 ? void 0 : overrides.environment,
+            });
+            return [2 /*return*/, fetchWithTimeout(url, {
+                    method: method,
+                    headers: { "Content-Type": "application/json", Authorization: "Bearer ".concat(ctx.apiKey) },
+                })];
+        });
+    });
+}
+function createIdempotentCleanup(cleanupFn) {
+    var hasRun = false;
+    return function idempotentCleanup() {
+        if (hasRun)
+            return;
+        hasRun = true;
+        cleanupFn();
+    };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dainprotocol/cli",
-  "version": "1.2.28",
+  "version": "1.2.31",
   "private": false,
   "publishConfig": {
     "access": "public"
@@ -23,8 +23,8 @@
   ],
   "dependencies": {
     "@ai-sdk/anthropic": "^0.0.50",
-    "@dainprotocol/service-sdk": "^2.0.77",
-    "@dainprotocol/tunnel": "^1.1.26",
+    "@dainprotocol/service-sdk": "^2.0.79",
+    "@dainprotocol/tunnel": "^1.1.29",
     "@types/fs-extra": "^11.0.4",
     "@types/localtunnel": "^2.0.4",
     "ai": "^3.3.41",