modelstat 0.0.15 → 0.0.17

package/dist/cli.mjs

@@ -7735,6 +7735,10 @@ async function fetchDeviceMe(secret) {
     method: "GET",
     headers: { authorization: `Bearer ${secret}` }
   });
+  if (res.statusCode === 401) {
+    await res.body.dump();
+    throw new DeviceMeUnauthorized();
+  }
   if (res.statusCode >= 300) {
     throw new Error(`devices/me failed: ${res.statusCode} ${await res.body.text()}`);
   }
@@ -7829,13 +7833,19 @@ async function uploadBatch(batch) {
     batch_id: result.response.batch_id
   };
 }
-var _ingest;
+var DeviceMeUnauthorized, _ingest;
 var init_api = __esm({
   "src/api.ts"() {
     "use strict";
     init_http();
     init_logger();
     init_config2();
+    DeviceMeUnauthorized = class extends Error {
+      constructor() {
+        super("devices/me returned 401: device_secret not recognised");
+        this.name = "DeviceMeUnauthorized";
+      }
+    };
     _ingest = null;
   }
 });
@@ -9007,12 +9017,37 @@ async function cmdConnect(opts) {
     process.stdout.write(`  \x1B[33m\u26A0\x1B[0m ${msg}
 `);
   };
+  const wipeAndSelfRegister = async (reason) => {
+    warn(`${reason} \u2014 re-registering this device`);
+    state.setBearer(null);
+    state.setDeviceId(null);
+    state.setDeviceUuid(null);
+    state.setClaimCode(null);
+    state.setClaimUrl(null);
+    await cmdSelfRegister();
+  };
   if (!state.deviceUuid || !state.bearer || !state.deviceId) {
     step("Registering this device with modelstat.ai");
     await cmdSelfRegister();
   } else {
     step("Re-using existing device identity");
     ok(`device ${state.deviceId}`);
+    try {
+      const me = await fetchDeviceMe(state.bearer);
+      if (me.claim_code && me.claim_code !== state.claimCode) {
+        state.setClaimCode(me.claim_code);
+        ok(`claim code refreshed from server`);
+      }
+      if (me.claim_url && me.claim_url !== state.claimUrl) {
+        state.setClaimUrl(me.claim_url);
+      }
+    } catch (e) {
+      if (e instanceof DeviceMeUnauthorized) {
+        await wipeAndSelfRegister("cached credentials no longer valid");
+      } else {
+        warn(`couldn't refresh device state: ${e.message}`);
+      }
+    }
   }
   const claimCode = state.claimCode ?? "(unknown)";
   const claimUrl = state.claimUrl ?? `https://modelstat.ai/device/${claimCode}`;