@moderniser/plugin 0.1.5 → 0.1.6

package/dist/client.d.ts

@@ -2,3 +2,7 @@ export declare const API: () => string;
 export declare function setStoredKey(key: string): void;
 export declare function storedKey(): string | undefined;
 export declare function resolveKey(): string | undefined;
+export declare const COMPANY_HEADER = "x-moderniser-company";
+export declare function setStoredCompany(companyId: string): void;
+export declare function resolveCompany(): string | undefined;
+export declare function companyHeaders(): Record<string, string>;

package/dist/index.js

@@ -2,7 +2,7 @@ import { createRequire } from "node:module";
 var __require = /* @__PURE__ */ createRequire(import.meta.url);
 
 // packages/plugin/src/client.ts
-import { readFileSync } from "node:fs";
+import { mkdirSync, readFileSync, renameSync, writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
 var API = () => (process.env.MODERNISER_API_URL ?? "https://v2.moderniser.ai/api").replace(/\/+$/, "");
@@ -34,6 +34,44 @@ function fromAuthFile() {
 function resolveKey() {
   return process.env.MODERNISER_API_KEY ?? stored ?? fromAuthFile();
 }
+var COMPANY_HEADER = "x-moderniser-company";
+function companyFilePath() {
+  const data = process.env.XDG_DATA_HOME || join(homedir(), ".local", "share");
+  return join(data, "moderniser", "company.json");
+}
+var storedCompany;
+function setStoredCompany(companyId) {
+  storedCompany = companyId;
+  fileCompany = companyId;
+  try {
+    const path = companyFilePath();
+    mkdirSync(join(path, ".."), { recursive: true });
+    const temp = `${path}.${process.pid}.tmp`;
+    writeFileSync(temp, JSON.stringify({ companyId }, null, 2) + `
+`, { mode: 384 });
+    renameSync(temp, path);
+  } catch {}
+}
+var fileCompany;
+function companyFromFile() {
+  if (fileCompany !== undefined)
+    return fileCompany ?? undefined;
+  fileCompany = null;
+  try {
+    const parsed = JSON.parse(readFileSync(companyFilePath(), "utf8"));
+    if (typeof parsed?.companyId === "string" && parsed.companyId.length > 0) {
+      fileCompany = parsed.companyId;
+    }
+  } catch {}
+  return fileCompany ?? undefined;
+}
+function resolveCompany() {
+  return process.env.MODERNISER_COMPANY ?? storedCompany ?? companyFromFile();
+}
+function companyHeaders() {
+  const company = resolveCompany();
+  return company ? { [COMPANY_HEADER]: company } : {};
+}
 
 // packages/plugin/src/auth.ts
 async function start() {
@@ -67,7 +105,7 @@ async function poll(deviceCode, expiresIn, intervalSeconds) {
       continue;
     }
     if (body.token)
-      return body.token;
+      return { token: body.token, companyId: body.companyId };
     if (body.error === "authorization_pending")
       continue;
     if (body.error === "slow_down") {
@@ -92,11 +130,13 @@ function moderniserAuth() {
             instructions: `Check that your browser shows this code: ${started.userCode}. After signing in for the first time, restart to pick up the Moderniser tools.`,
             method: "auto",
             async callback() {
-              const token = await poll(started.deviceCode, started.expiresIn, started.interval);
-              if (!token)
+              const collected = await poll(started.deviceCode, started.expiresIn, started.interval);
+              if (!collected)
                 return { type: "failed" };
-              setStoredKey(token);
-              return { type: "success", key: token };
+              setStoredKey(collected.token);
+              if (collected.companyId)
+                setStoredCompany(collected.companyId);
+              return { type: "success", key: collected.token };
             }
           };
         }
@@ -158,7 +198,7 @@ async function call(path, body) {
   try {
     return await fetch(`${API2()}${path}`, {
       method: "POST",
-      headers: { "Content-Type": "application/json", Authorization: `Bearer ${key}` },
+      headers: { "Content-Type": "application/json", Authorization: `Bearer ${key}`, ...companyHeaders() },
       body: JSON.stringify(body),
       signal: AbortSignal.timeout(1e4)
     });
@@ -175,7 +215,7 @@ async function gatewayProduct() {
     return {};
   try {
     const res = await fetch(`${GATEWAY()}/models`, {
-      headers: { Authorization: `Bearer ${key}` },
+      headers: { Authorization: `Bearer ${key}`, ...companyHeaders() },
       signal: AbortSignal.timeout(5000)
     });
     if (!res.ok)
@@ -212,11 +252,16 @@ var ModerniserPlugin = async () => {
         }
       };
       const key = KEY();
+      const company = companyHeaders();
       config.provider ??= {};
       config.provider[PROVIDER_ID] ??= {
         name: "Moderniser",
         npm: "@ai-sdk/openai-compatible",
-        options: { baseURL: GATEWAY(), ...key ? { apiKey: key } : {} },
+        options: {
+          baseURL: GATEWAY(),
+          ...key ? { apiKey: key } : {},
+          ...Object.keys(company).length ? { headers: company } : {}
+        },
         models
       };
       config.enabled_providers ??= [PROVIDER_ID];
@@ -231,7 +276,7 @@ var ModerniserPlugin = async () => {
         config.mcp[PROVIDER_ID] ??= {
           type: "remote",
           url: `${API2()}/mcp`,
-          headers: { Authorization: `Bearer ${mcpKey}` },
+          headers: { Authorization: `Bearer ${mcpKey}`, ...companyHeaders() },
           oauth: false
         };
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@moderniser/plugin",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "type": "module",
   "description": "Moderniser plugin for opencode: points inference at the gateway, declares the Moderniser remote MCP server, and reports session telemetry",
   "license": "MIT",