feat: unify app and third-party wloc features

2026-09-21 22:30:46 +08:00 · 2026-08-10 13:35:47 +08:00
parent 27af50375d
commit 41dfe7a8ed
36 changed files with 1702 additions and 73 deletions
@@ -0,0 +1,85 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import {
+  MOTION_ACTIVITY_CONFIDENCE, MOTION_ACTIVITY_TYPE, internals, parseFields, patchWlocBody
+} from "../src/core.js";
+
+const { concat, writeVarint, writeTag, writeLengthDelimited } = internals;
+
+function location(withMotion = false) {
+  const fields = [
+    concat(writeTag(1, 0), writeVarint(100)),
+    concat(writeTag(2, 0), writeVarint(200)),
+    concat(writeTag(3, 0), writeVarint(25))
+  ];
+  if (withMotion) {
+    fields.push(concat(writeTag(11, 0), writeVarint(7)));
+    fields.push(concat(writeTag(12, 0), writeVarint(88)));
+  }
+  return concat(...fields);
+}
+
+function wifiPayload(value = location()) {
+  const device = concat(
+    writeLengthDelimited(1, new TextEncoder().encode("aa:bb:cc:dd:ee:ff")),
+    writeLengthDelimited(2, value)
+  );
+  return writeLengthDelimited(2, device);
+}
+
+function frame(payload) {
+  return concat(Uint8Array.from([0, 1, 0, 0, 0, 1, 0, 0]),
+    Uint8Array.from([payload.length >> 8, payload.length & 0xff]), payload);
+}
+
+const config = {
+  latitude: 31.230416,
+  longitude: 121.473701,
+  accuracy: 50,
+  motionSimulationEnabled: false
+};
+
+test("patches synthetic Wi-Fi response", () => {
+  const result = patchWlocBody(frame(wifiPayload()), config);
+  assert.equal(result.stats.wifi, 1);
+  assert.equal(result.stats.locations, 1);
+});
+
+test("preserves motion fields while disabled", () => {
+  const result = patchWlocBody(frame(wifiPayload(location(true))), config);
+  const payload = result.data.slice(10);
+  assert.ok(payload.includes(7));
+  assert.ok(payload.includes(88));
+});
+
+test("replaces motion fields while enabled", () => {
+  const result = patchWlocBody(frame(wifiPayload(location(true))), {
+    ...config, motionSimulationEnabled: true
+  });
+  const root = parseFields(result.data.slice(10));
+  const device = parseFields(root[0].value);
+  const fields = parseFields(device.find((field) => field.number === 2).value);
+  const motionType = fields.find((field) => field.number === 11);
+  const motionConfidence = fields.find((field) => field.number === 12);
+  assert.deepEqual(motionType.value, writeVarint(MOTION_ACTIVITY_TYPE));
+  assert.deepEqual(motionConfidence.value, writeVarint(MOTION_ACTIVITY_CONFIDENCE));
+});
+
+test("patches CellTower fields 22 and 24", () => {
+  for (const number of [22, 24]) {
+    const cell = writeLengthDelimited(5, location());
+    const result = patchWlocBody(frame(writeLengthDelimited(number, cell)), config);
+    assert.equal(result.stats.cell, 1);
+  }
+});
+
+test("encodes signed int64 coordinates without BigInt", () => {
+  assert.deepEqual(
+    Array.from(writeVarint(-18_000_000_000)),
+    [128, 152, 247, 248, 188, 255, 255, 255, 255, 1]
+  );
+  assert.deepEqual(
+    Array.from(writeVarint(18_000_000_000)),
+    [128, 232, 136, 135, 67]
+  );
+});