ns-omp-provider-kiro 0.3.0 → 0.3.1

package/dist/index.js

@@ -292,6 +292,7 @@ __export(kiro_cli_exports, {
   getKiroCliCredentials: () => getKiroCliCredentials,
   getKiroCliCredentialsAllowExpired: () => getKiroCliCredentialsAllowExpired,
   getKiroCliDbPath: () => getKiroCliDbPath,
+  getKiroCliExternalIdpCredentials: () => getKiroCliExternalIdpCredentials,
   getKiroCliModelRates: () => getKiroCliModelRates,
   getKiroCliSocialToken: () => getKiroCliSocialToken,
   getKiroCliSocialTokenAllowExpired: () => getKiroCliSocialTokenAllowExpired,
@@ -315,6 +316,16 @@ function getKiroCliDbPath() {
     dbPath = join2(homedir2(), ".local", "share", "kiro-cli", "data.sqlite3");
   return existsSync(dbPath) ? dbPath : void 0;
 }
+function getKiroCliExternalIdpCredentials() {
+  const dbPath = getKiroCliDbPath();
+  if (!dbPath)
+    return void 0;
+  try {
+    return tryKiroCliToken(dbPath, EXTERNAL_IDP_TOKEN_KEY, "external-idp", true);
+  } catch {
+    return void 0;
+  }
+}
 function getNodeSqlite() {
   try {
     return require2("node:sqlite");
@@ -887,17 +898,21 @@ var init_oauth = __esm({
 
 // ../kiro-core/dist/token-type.js
 function isExternalIdpAccessToken(accessToken) {
-  if (typeof accessToken !== "string")
+  if (typeof accessToken !== "string" || !accessToken)
     return false;
   const segments = accessToken.split(".");
-  if (segments.length !== 3)
-    return false;
   const payloadSegment = segments[1];
-  if (!payloadSegment)
-    return false;
+  if (segments.length === 3 && payloadSegment) {
+    try {
+      const payload = JSON.parse(Buffer.from(payloadSegment.replace(/-/g, "+").replace(/_/g, "/"), "base64").toString("utf-8"));
+      if (payload.aud === EXTERNAL_IDP_AUDIENCE || Array.isArray(payload.aud) && payload.aud.includes(EXTERNAL_IDP_AUDIENCE)) {
+        return true;
+      }
+    } catch {
+    }
+  }
   try {
-    const payload = JSON.parse(Buffer.from(payloadSegment.replace(/-/g, "+").replace(/_/g, "/"), "base64").toString("utf-8"));
-    return payload.aud === EXTERNAL_IDP_AUDIENCE;
+    return getKiroCliExternalIdpCredentials()?.access === accessToken;
   } catch {
     return false;
   }
@@ -909,6 +924,7 @@ var EXTERNAL_IDP_AUDIENCE;
 var init_token_type = __esm({
   "../kiro-core/dist/token-type.js"() {
     "use strict";
+    init_kiro_cli();
     EXTERNAL_IDP_AUDIENCE = "api://kiro";
   }
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ns-omp-provider-kiro",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "OMP (oh-my-pi) provider plugin: adds Kiro (AWS CodeWhisperer/Q) models via your kiro-cli login",
   "type": "module",
   "main": "./dist/index.js",
@@ -37,7 +37,7 @@
     "README.md"
   ],
   "dependencies": {
-    "ns-kiro-core": "0.3.0"
+    "ns-kiro-core": "0.3.1"
   },
   "peerDependencies": {
     "@oh-my-pi/pi-ai": ">=18.2.0",