openusage 0.1.0

package/plugins/copilot/icon.svg

@@ -0,0 +1,3 @@
+<svg width="100" height="100" viewBox="0 0 100 100" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M79.7057 21.6044C85.0857 27.2967 87.3915 35.1429 88.3522 46.0659C90.8885 46.0659 93.2711 46.6428 94.8851 48.8352L97.8826 52.9121C98.728 54.0659 99.1892 55.4505 99.1892 56.9121V67.9505C99.1833 68.6605 99.0137 69.3596 98.6938 69.9934C98.3739 70.6271 97.9122 71.1784 97.3446 71.6044C83.7791 81.5275 67.0241 89.5275 50 89.5275C31.1698 89.5275 12.2627 78.6428 2.65542 71.6044C2.08785 71.1784 1.62614 70.6271 1.30621 69.9934C0.986278 69.3596 0.816752 68.6605 0.810822 67.9505L0.810822 56.9121C0.810822 55.4505 1.27197 54.0659 2.11741 52.8736L5.11488 48.8352C6.7289 46.6428 9.1115 46.0659 11.6478 46.0659C12.6085 35.1429 14.8759 27.2967 20.2944 21.6044C30.5165 10.7582 44.0819 9.52747 49.8463 9.52747H50C55.6491 9.52747 69.3683 10.6429 79.7057 21.6044ZM50 39.7967C48.8471 39.7967 47.5021 39.8736 46.0418 40.0275C45.678 41.7486 44.8848 43.3496 43.7361 44.6813C42.1163 46.2842 40.1962 47.5511 38.0858 48.4092C35.9755 49.2672 33.7165 49.6995 31.4388 49.6813C28.8256 49.6813 26.0971 49.1044 23.8683 47.6813C21.7547 48.4121 19.7179 49.4121 19.5642 51.9121C19.3721 56.6044 19.3336 61.3352 19.3336 66.0659C19.3336 68.4121 19.3336 70.7967 19.2568 73.1813C19.2568 74.5659 20.1022 75.8352 21.3704 76.4121C31.5156 81.0275 41.1613 83.3736 50 83.3736C58.8387 83.3736 68.446 81.0659 78.6297 76.4121C79.2511 76.1286 79.7794 75.6746 80.1535 75.1028C80.5276 74.531 80.7321 73.8647 80.7432 73.1813C80.8585 66.1044 80.7432 58.9505 80.4358 51.9121C80.2821 49.3736 78.2454 48.4121 76.1318 47.6813C73.9029 49.1044 71.136 49.6429 68.5612 49.6429C66.2861 49.6662 64.0288 49.2399 61.9186 48.3885C59.8084 47.5371 57.8868 46.2773 56.2639 44.6813C55.1153 43.3496 54.322 41.7486 53.9582 40.0275C52.6516 39.8736 51.3066 39.8352 50 39.7967ZM39.6242 56.7198C41.853 56.7198 43.6592 58.489 43.6592 60.7198V68.1044C43.6592 69.1653 43.2381 70.1827 42.4886 70.9328C41.7391 71.683 40.7226 72.1044 39.6626 72.1044C38.6026 72.1044 37.5861 71.683 36.8366 70.9328C36.087 70.1827 35.666 69.1653 35.666 68.1044V60.6813C35.666 58.4505 37.4337 56.6813 39.6626 56.6813L39.6242 56.7198ZM60.1453 56.7198C62.3742 56.7198 64.1419 58.489 64.1419 60.7198V68.1044C64.1419 69.1653 63.7208 70.1827 62.9713 70.9328C62.2218 71.683 61.2052 72.1044 60.1453 72.1044C59.0853 72.1044 58.0688 71.683 57.3192 70.9328C56.5697 70.1827 56.1487 69.1653 56.1487 68.1044V60.6813C56.1487 58.4505 57.9548 56.6813 60.1453 56.6813V56.7198ZM32.0921 20.4121C27.788 20.8352 24.1757 22.2582 22.3311 24.2582C18.3345 28.6044 19.1799 39.6813 21.4857 42.0275C23.7655 43.9275 26.6682 44.9142 29.6326 44.7967C32.2458 44.7967 37.1647 44.2198 41.1998 40.1044C43.0059 38.3736 44.0819 34.0659 43.9666 29.7198C43.8514 26.2198 42.8522 23.2967 41.3919 22.0659C39.7779 20.6813 36.1655 20.0659 32.0921 20.4121ZM58.6081 22.0659C57.1478 23.2967 56.1487 26.2198 56.0334 29.7198C55.9181 34.0659 56.9941 38.3736 58.8003 40.1044C60.3121 41.6165 62.111 42.8106 64.0909 43.6164C66.0708 44.4223 68.1918 44.8236 70.329 44.7967C73.7492 44.7967 76.8619 43.6813 78.5144 42.0275C80.8201 39.6813 81.6655 28.6044 77.6689 24.2198C75.8243 22.2967 72.212 20.8352 67.9079 20.4121C63.8345 20.0275 60.2221 20.6813 58.6081 22.0659ZM50 31.0659C49.0009 31.0659 47.848 31.1429 46.5414 31.2582C46.6951 31.9121 46.7335 32.6813 46.8104 33.4505C46.8104 34.0275 46.8104 34.6044 46.7335 35.1813C47.9633 35.0659 49.0393 35.0659 50 35.0659C50.9992 35.0659 52.0367 35.0659 53.2665 35.1813C53.1896 34.5659 53.1896 34.0275 53.1896 33.4505C53.2665 32.6813 53.3049 31.9121 53.4586 31.2582C52.152 31.1429 50.9992 31.0659 50 31.0659Z" fill="white"/>
+</svg>

package/plugins/copilot/plugin.js

@@ -0,0 +1,264 @@
+(function () {
+  const KEYCHAIN_SERVICE = "OpenUsage-copilot";
+  const GH_KEYCHAIN_SERVICE = "gh:github.com";
+  const USAGE_URL = "https://api.github.com/copilot_internal/user";
+
+  function readJson(ctx, path) {
+    try {
+      if (!ctx.host.fs.exists(path)) return null;
+      const text = ctx.host.fs.readText(path);
+      return ctx.util.tryParseJson(text);
+    } catch (e) {
+      ctx.host.log.warn("readJson failed for " + path + ": " + String(e));
+      return null;
+    }
+  }
+
+  function writeJson(ctx, path, value) {
+    try {
+      ctx.host.fs.writeText(path, JSON.stringify(value));
+    } catch (e) {
+      ctx.host.log.warn("writeJson failed for " + path + ": " + String(e));
+    }
+  }
+
+  function saveToken(ctx, token) {
+    try {
+      ctx.host.keychain.writeGenericPassword(
+        KEYCHAIN_SERVICE,
+        JSON.stringify({ token: token }),
+      );
+    } catch (e) {
+      ctx.host.log.warn("keychain write failed: " + String(e));
+    }
+    writeJson(ctx, ctx.app.pluginDataDir + "/auth.json", { token: token });
+  }
+
+  function clearCachedToken(ctx) {
+    try {
+      ctx.host.keychain.deleteGenericPassword(KEYCHAIN_SERVICE);
+    } catch (e) {
+      ctx.host.log.info("keychain delete failed: " + String(e));
+    }
+    writeJson(ctx, ctx.app.pluginDataDir + "/auth.json", null);
+  }
+
+  function loadTokenFromKeychain(ctx) {
+    try {
+      const raw = ctx.host.keychain.readGenericPassword(KEYCHAIN_SERVICE);
+      if (raw) {
+        const parsed = ctx.util.tryParseJson(raw);
+        if (parsed && parsed.token) {
+          ctx.host.log.info("token loaded from OpenUsage keychain");
+          return { token: parsed.token, source: "keychain" };
+        }
+      }
+    } catch (e) {
+      ctx.host.log.info("OpenUsage keychain read failed: " + String(e));
+    }
+    return null;
+  }
+
+  function loadTokenFromGhCli(ctx) {
+    try {
+      const raw = ctx.host.keychain.readGenericPassword(GH_KEYCHAIN_SERVICE);
+      if (raw) {
+        let token = raw;
+        if (
+          typeof token === "string" &&
+          token.indexOf("go-keyring-base64:") === 0
+        ) {
+          token = ctx.base64.decode(token.slice("go-keyring-base64:".length));
+        }
+        if (token) {
+          ctx.host.log.info("token loaded from gh CLI keychain");
+          return { token: token, source: "gh-cli" };
+        }
+      }
+    } catch (e) {
+      ctx.host.log.info("gh CLI keychain read failed: " + String(e));
+    }
+    return null;
+  }
+
+  function loadTokenFromStateFile(ctx) {
+    const data = readJson(ctx, ctx.app.pluginDataDir + "/auth.json");
+    if (data && data.token) {
+      ctx.host.log.info("token loaded from state file");
+      return { token: data.token, source: "state" };
+    }
+    return null;
+  }
+
+  function loadToken(ctx) {
+    return (
+      loadTokenFromKeychain(ctx) ||
+      loadTokenFromGhCli(ctx) ||
+      loadTokenFromStateFile(ctx)
+    );
+  }
+
+  function fetchUsage(ctx, token) {
+    return ctx.util.request({
+      method: "GET",
+      url: USAGE_URL,
+      headers: {
+        Authorization: "token " + token,
+        Accept: "application/json",
+        "Editor-Version": "vscode/1.96.2",
+        "Editor-Plugin-Version": "copilot-chat/0.26.7",
+        "User-Agent": "GitHubCopilotChat/0.26.7",
+        "X-Github-Api-Version": "2025-04-01",
+      },
+      timeoutMs: 10000,
+    });
+  }
+
+  function makeProgressLine(ctx, label, snapshot, resetDate) {
+    if (!snapshot || typeof snapshot.percent_remaining !== "number")
+      return null;
+    const usedPercent = Math.min(100, Math.max(0, 100 - snapshot.percent_remaining));
+    return ctx.line.progress({
+      label: label,
+      used: usedPercent,
+      limit: 100,
+      format: { kind: "percent" },
+      resetsAt: ctx.util.toIso(resetDate),
+      periodDurationMs: 30 * 24 * 60 * 60 * 1000,
+    });
+  }
+
+  function makeLimitedProgressLine(ctx, label, remaining, total, resetDate) {
+    if (typeof remaining !== "number" || typeof total !== "number" || total <= 0)
+      return null;
+    const used = total - remaining;
+    const usedPercent = Math.min(100, Math.max(0, Math.round((used / total) * 100)));
+    return ctx.line.progress({
+      label: label,
+      used: usedPercent,
+      limit: 100,
+      format: { kind: "percent" },
+      resetsAt: ctx.util.toIso(resetDate),
+      periodDurationMs: 30 * 24 * 60 * 60 * 1000,
+    });
+  }
+
+  function probe(ctx) {
+    const cred = loadToken(ctx);
+    if (!cred) {
+      throw "Not logged in. Run `gh auth login` first.";
+    }
+
+    let token = cred.token;
+    let source = cred.source;
+
+    let resp;
+    try {
+      resp = fetchUsage(ctx, token);
+    } catch (e) {
+      ctx.host.log.error("usage request exception: " + String(e));
+      throw "Usage request failed. Check your connection.";
+    }
+
+    if (resp.status === 401 || resp.status === 403) {
+      // If cached token is stale, clear it and try fallback sources
+      if (source === "keychain") {
+        ctx.host.log.info("cached token invalid, trying fallback sources");
+        clearCachedToken(ctx);
+        const fallback = loadTokenFromGhCli(ctx);
+        if (fallback) {
+          try {
+            resp = fetchUsage(ctx, fallback.token);
+          } catch (e) {
+            ctx.host.log.error("fallback usage request exception: " + String(e));
+            throw "Usage request failed. Check your connection.";
+          }
+          if (resp.status >= 200 && resp.status < 300) {
+            // Fallback worked, persist the new token
+            saveToken(ctx, fallback.token);
+            token = fallback.token;
+            source = fallback.source;
+          }
+        }
+      }
+      // Still failing after retry
+      if (resp.status === 401 || resp.status === 403) {
+        throw "Token invalid. Run `gh auth login` to re-authenticate.";
+      }
+    }
+
+    if (resp.status < 200 || resp.status >= 300) {
+      ctx.host.log.error("usage returned error: status=" + resp.status);
+      throw (
+        "Usage request failed (HTTP " +
+        String(resp.status) +
+        "). Try again later."
+      );
+    }
+
+    // Persist gh-cli token to OpenUsage keychain for future use
+    if (source === "gh-cli") {
+      saveToken(ctx, token);
+    }
+
+    const data = ctx.util.tryParseJson(resp.bodyText);
+    if (data === null) {
+      throw "Usage response invalid. Try again later.";
+    }
+
+    ctx.host.log.info("usage fetch succeeded");
+
+    const lines = [];
+    let plan = null;
+    if (data.copilot_plan) {
+      plan = ctx.fmt.planLabel(data.copilot_plan);
+    }
+
+    // Paid tier: quota_snapshots
+    const snapshots = data.quota_snapshots;
+    if (snapshots) {
+      const premiumLine = makeProgressLine(
+        ctx,
+        "Premium",
+        snapshots.premium_interactions,
+        data.quota_reset_date,
+      );
+      if (premiumLine) lines.push(premiumLine);
+
+      const chatLine = makeProgressLine(
+        ctx,
+        "Chat",
+        snapshots.chat,
+        data.quota_reset_date,
+      );
+      if (chatLine) lines.push(chatLine);
+    }
+
+    // Free tier: limited_user_quotas
+    if (data.limited_user_quotas && data.monthly_quotas) {
+      const lq = data.limited_user_quotas;
+      const mq = data.monthly_quotas;
+      const resetDate = data.limited_user_reset_date;
+
+      const chatLine = makeLimitedProgressLine(ctx, "Chat", lq.chat, mq.chat, resetDate);
+      if (chatLine) lines.push(chatLine);
+
+      const completionsLine = makeLimitedProgressLine(ctx, "Completions", lq.completions, mq.completions, resetDate);
+      if (completionsLine) lines.push(completionsLine);
+    }
+
+    if (lines.length === 0) {
+      lines.push(
+        ctx.line.badge({
+          label: "Status",
+          text: "No usage data",
+          color: "#a3a3a3",
+        }),
+      );
+    }
+
+    return { plan: plan, lines: lines };
+  }
+
+  globalThis.__openusage_plugin = { id: "copilot", probe };
+})();

package/plugins/copilot/plugin.json

@@ -0,0 +1,20 @@
+{
+  "schemaVersion": 1,
+  "id": "copilot",
+  "name": "Copilot",
+  "version": "0.0.1",
+  "entry": "plugin.js",
+  "icon": "icon.svg",
+  "brandColor": "#A855F7",
+  "cli": {
+    "category": "cli",
+    "binaryName": "gh",
+    "installCmd": "sudo apt install gh",
+    "loginCmd": "gh auth login"
+  },
+  "lines": [
+    { "type": "progress", "label": "Premium", "scope": "overview", "primaryOrder": 1 },
+    { "type": "progress", "label": "Chat", "scope": "overview", "primaryOrder": 2 },
+    { "type": "progress", "label": "Completions", "scope": "overview" }
+  ]
+}