fix(water-log): remove platform-login dependency from PIN flows
Deploy to route.crispygoat.com / deploy (push) Successful in 4m12s

Field users at /water were blocked because every server action called
getSession() (Neon Auth) even though the field path is PIN-authenticated.
A missing or expired platform session hung or rejected PIN submissions,
so users without a platform login could never reach the water log entry
screen.

Centralize water-log auth in src/actions/water-log/auth.ts with three
helpers — requireFieldSession, requireWaterAdminSession, and
requireWaterAdminPermission — and migrate all server actions off the
stray getSession() calls. Auth.ts deliberately does NOT import
getSession; a static-source unit test enforces that.

Changes:
- src/actions/water-log/auth.ts (new): dedicated auth layer, no Neon Auth
- src/actions/water-log/field.ts: -10 await getSession(), use helpers
- src/actions/water-log/admin.ts: -18 await getSession(), use helpers
- src/actions/water-log/settings.ts: -4 await getSession(), resilient
  to missing platform admin
- 3x admin pages: update getWaterAdminSession import path
- tests/unit/water-log-auth.test.ts (new): 17 tests incl. regression
  guard that auth.ts never imports getSession
- tests/water-log.spec.ts: 3 E2E regression tests (no platform login)
- vitest.config.ts: alias for deep @/db/schema/water-log path

Rollback = revert this commit. No DB migration; no schema change.
Existing rows/cookies unchanged.
This commit is contained in:
Tyler
2026-07-01 17:27:37 -06:00
parent 015eb33291
commit 658f6a5b8b
10 changed files with 780 additions and 209 deletions
+28 -72
View File
@@ -16,22 +16,21 @@
*/
"use server";
import { cookies } from "next/headers";
import { and, desc, eq, gte, lte, sql, SQL } from "drizzle-orm";
import { getAdminUser } from "@/lib/admin-permissions";
import { and, desc, eq, sql } from "drizzle-orm";
import { withBrand } from "@/db/client";
import {
waterHeadgates,
waterIrrigators,
waterLogEntries,
waterAdminSessions,
type WaterHeadgate,
type WaterIrrigator,
type WaterLogEntry,
} from "@/db/schema/water-log";
import { hashPin, generatePin, verifyPin } from "@/lib/water-log-pin";
import { logAuditEvent, logAlert } from "@/lib/water-log-audit";
import { getSession } from "@/lib/auth";
import { hashPin, generatePin } from "@/lib/water-log-pin";
import { logAuditEvent } from "@/lib/water-log-audit";
import {
requireWaterAdminPermission,
} from "@/actions/water-log/auth";
// ── Types used by both server and client ────────────────────────────────────
@@ -108,54 +107,11 @@ export type AlertLogEntry = {
};
// ── Auth helpers ───────────────────────────────────────────────────────────
async function requireWaterAdminPermission() {
const adminUser = await getAdminUser();
if (!adminUser) return { ok: false as const, error: "Not authenticated" };
if (
!adminUser.can_manage_water_log &&
adminUser.role !== "platform_admin"
) {
return { ok: false as const, error: "Not authorized" };
}
return { ok: true as const, adminUser };
}
/**
* For the `/water/admin` (PIN) portal. Returns the admin + brand if a
* valid session cookie is present. Falls back to a 401-shaped error.
*/
async function requireWaterAdminSession(): Promise<
| { ok: true; adminUserId: string; brandId: string }
| { ok: false; error: string }
> {
await getSession(); const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_admin_session")?.value;
if (!sessionId) return { ok: false, error: "Not signed in" };
return withBrand(globalThis.__TUXEDO_BRAND_ID__ ?? "", async () => {
// session lookup must use platform-admin to read across brands
const { withPlatformAdmin } = await import("@/db/client");
return withPlatformAdmin(async (db) => {
const rows = await db
.select()
.from(waterAdminSessions)
.where(eq(waterAdminSessions.id, sessionId))
.limit(1);
const row = rows[0];
if (!row) return { ok: false as const, error: "Session not found" };
if (row.expiresAt.getTime() < Date.now()) {
return { ok: false as const, error: "Session expired" };
}
return {
ok: true as const,
adminUserId: row.adminUserId,
brandId: row.brandId,
};
});
});
}
//
// `requireWaterAdminPermission` and `requireWaterAdminSession` live in
// `@/actions/water-log/auth`. The site-admin gate calls `getAdminUser()`;
// the brand-admin-PIN gate reads `wl_admin_session` directly. Neither
// calls the platform-level `getSession()` from `@/lib/auth`.
// Global hint used by the field admin portal: water log is currently
// scoped to the Tuxedo brand only. Surfacing this in one place makes it
@@ -229,7 +185,7 @@ export async function createWaterHeadgate(
options: { highThreshold?: number | null; lowThreshold?: number | null; notes?: string | null } = {},
): Promise<{ success: boolean; headgate?: AdminHeadgate; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const trimmed = name?.trim();
if (!trimmed) return { success: false, error: "Name is required" };
@@ -283,7 +239,7 @@ export async function updateWaterHeadgate(
status?: AdminHeadgate["status"],
): Promise<{ success: boolean; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
// Headgates aren't directly brand-scoped in the URL — look up brand first.
@@ -337,7 +293,7 @@ export async function deleteWaterHeadgate(
headgateId: string,
): Promise<{ success: boolean; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const brand = await withPlatformAdminBrandOf(headgateId);
if (!brand) return { success: false, error: "Headgate not found" };
@@ -363,7 +319,7 @@ export async function regenerateHeadgateToken(
headgateId: string,
): Promise<{ success: boolean; token?: string; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const brand = await withPlatformAdminBrandOf(headgateId);
if (!brand) return { success: false, error: "Headgate not found" };
@@ -391,7 +347,7 @@ export async function getWaterHeadgatesAdmin(
brandId: string,
): Promise<AdminHeadgate[]> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return [];
return withBrand(brandId, async (db) => {
const rows = await db
@@ -407,7 +363,7 @@ await getSession(); const auth = await requireWaterAdminPermission();
export async function getWaterIrrigators(brandId: string): Promise<AdminIrrigator[]> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return [];
return withBrand(brandId, async (db) => {
const rows = await db
@@ -434,7 +390,7 @@ export async function createWaterUser(
phone?: string | null,
): Promise<CreateUserResult> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const trimmed = name?.trim();
if (!trimmed) return { success: false, error: "Name is required" };
@@ -494,7 +450,7 @@ export async function createWaterIrrigator(
language: string = "en",
): Promise<CreateUserResult> {
await getSession(); return createWaterUser(brandId, name, "irrigator", language);
return createWaterUser(brandId, name, "irrigator", language);
}
export async function updateWaterIrrigator(
@@ -507,7 +463,7 @@ export async function updateWaterIrrigator(
notes?: string | null,
): Promise<{ success: boolean; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const brand = await withPlatformAdminBrandOfIrrigator(irrigatorId);
if (!brand) return { success: false, error: "User not found" };
@@ -547,7 +503,7 @@ export async function deleteWaterUser(
userId: string,
): Promise<{ success: boolean; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const brand = await withPlatformAdminBrandOfIrrigator(userId);
if (!brand) return { success: false, error: "User not found" };
@@ -573,7 +529,7 @@ export async function resetWaterIrrigatorPin(
userId: string,
): Promise<{ success: boolean; pin?: string; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const brand = await withPlatformAdminBrandOfIrrigator(userId);
if (!brand) return { success: false, error: "User not found" };
@@ -609,7 +565,7 @@ export async function getWaterEntries(
limit: number = 50,
): Promise<AdminEntry[]> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return [];
const safeLimit = Math.min(Math.max(limit, 1), 1000);
return withBrand(brandId, async (db) => {
@@ -664,7 +620,7 @@ await getSession(); const auth = await requireWaterAdminPermission();
export async function getWaterEntryById(entryId: string): Promise<AdminEntry | null> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return null;
const brand = await withPlatformAdminBrandOfEntry(entryId);
if (!brand) return null;
@@ -727,7 +683,7 @@ export async function updateWaterEntry(
method?: AdminEntry["method"],
): Promise<{ success: boolean; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
if (!Number.isFinite(measurement) || measurement < 0) {
return { success: false, error: "Measurement must be a non-negative number" };
@@ -763,7 +719,7 @@ export async function deleteWaterEntry(
entryId: string,
): Promise<{ success: boolean; error?: string }> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return { success: false, error: auth.error };
const brand = await withPlatformAdminBrandOfEntry(entryId);
if (!brand) return { success: false, error: "Entry not found" };
@@ -791,7 +747,7 @@ export async function getWaterDisplaySummary(
brandId: string,
): Promise<WaterDisplaySummary | null> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return null;
return withBrand(brandId, async (db) => {
// Headgates with their latest entry
@@ -923,7 +879,7 @@ export async function getWaterAlertLog(
limit: number = 50,
): Promise<AlertLogEntry[]> {
await getSession(); const auth = await requireWaterAdminPermission();
const auth = await requireWaterAdminPermission();
if (!auth.ok) return [];
return withBrand(brandId, async (db) => {
const rows = await db.execute<{
+243
View File
@@ -0,0 +1,243 @@
/**
* Water Log — auth layer.
*
* Centralizes the three auth gates the water-log module uses:
*
* 1. `requireFieldSession()` — read wl_session cookie, look up
* the session row in `water_sessions`,
* return user/brand/role. Gates every
* irrigator PIN action (/water).
* 2. `requireWaterAdminSession()` — read wl_admin_session cookie, look
* up the row in `water_admin_sessions`.
* Gates brand-admin PIN paths
* (/water/admin/*).
* 3. `requireWaterAdminPermission()`— site-admin gate (Neon Auth +
* admin_users.can_manage_water_log).
* Gates /admin/water-log/*.
*
* **Crucially, the two PIN-only helpers (1) and (2) never call
* `getSession()` from `@/lib/auth` and never call `getAdminUser()`.**
* The water log module is intentionally PIN-based and self-contained;
* a prior version of this code carried spurious `getSession()` calls
* that hung PIN submission for users with no platform login. The static
* "does not import getSession" test in `tests/unit/water-log-auth.test.ts`
* guards against that regression.
*/
"use server";
import { cookies } from "next/headers";
import { eq } from "drizzle-orm";
import { withPlatformAdmin } from "@/db/client";
import {
waterSessions,
waterIrrigators,
waterAdminSessions,
} from "@/db/schema/water-log";
import { getAdminUser } from "@/lib/admin-permissions";
// ── Result types ────────────────────────────────────────────────────────
export type FieldSession =
| { ok: true; userId: string; brandId: string; role: "irrigator" | "water_admin" }
| { ok: false; error: string };
export type AdminPinSession =
| { ok: true; sessionId: string; brandId: string; adminUserId: string }
| { ok: false; error: string };
export type AdminPinSessionInfo = {
sessionId: string;
brandId: string;
adminUserId: string;
role: "water_admin";
};
export type SiteAdminPermission =
| { ok: true; adminUser: NonNullable<Awaited<ReturnType<typeof getAdminUser>>> }
| { ok: false; error: string };
// ── Helpers ─────────────────────────────────────────────────────────────
/**
* Field session gate — /water (irrigator PIN).
*
* Reads the `wl_session` cookie, joins it to `water_sessions` and the
* linked `water_irrigators` row, and returns the user/brand/role.
*
* Returns `{ ok: false, error }` for:
* - missing cookie → "Not logged in"
* - cookie with no matching row → "Session not found"
* - row past `expires_at` → "Session expired" (best-effort
* cleanup of the stale row)
* - irrigator marked `active = false`→ "User is inactive"
*
* Never calls `getSession()`. Never calls `getAdminUser()`. The only
* network round-trip is the local DB query.
*/
export async function requireFieldSession(): Promise<FieldSession> {
const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_session")?.value;
if (!sessionId) return { ok: false, error: "Not logged in" };
return withPlatformAdmin(async (db) => {
const rows = await db
.select({
session: waterSessions,
irrigator: waterIrrigators,
})
.from(waterSessions)
.innerJoin(
waterIrrigators,
eq(waterIrrigators.id, waterSessions.irrigatorId),
)
.where(eq(waterSessions.id, sessionId))
.limit(1);
const row = rows[0];
if (!row) return { ok: false as const, error: "Session not found" };
if (row.session.expiresAt.getTime() < Date.now()) {
// Best-effort cleanup
try {
await db.delete(waterSessions).where(eq(waterSessions.id, sessionId));
} catch {
// ignore — the row may have been deleted concurrently
}
return { ok: false as const, error: "Session expired" };
}
if (!row.irrigator.active) {
return { ok: false as const, error: "User is inactive" };
}
return {
ok: true as const,
userId: row.irrigator.id,
brandId: row.irrigator.brandId,
role:
(row.irrigator.role as "irrigator" | "water_admin") ?? "irrigator",
};
});
}
/**
* Brand-admin PIN session gate — /water/admin/*.
*
* Reads the `wl_admin_session` cookie, joins it to
* `water_admin_sessions`, and returns brand/admin info. Same error
* semantics as `requireFieldSession` but for the admin-PIN cookie.
*/
export async function requireWaterAdminSession(): Promise<AdminPinSession> {
const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_admin_session")?.value;
if (!sessionId) return { ok: false, error: "Not signed in" };
return withPlatformAdmin(async (db) => {
const rows = await db
.select({
id: waterAdminSessions.id,
brandId: waterAdminSessions.brandId,
adminUserId: waterAdminSessions.adminUserId,
expiresAt: waterAdminSessions.expiresAt,
})
.from(waterAdminSessions)
.where(eq(waterAdminSessions.id, sessionId))
.limit(1);
const row = rows[0];
if (!row) return { ok: false as const, error: "Session not found" };
if (row.expiresAt.getTime() < Date.now()) {
return { ok: false as const, error: "Session expired" };
}
return {
ok: true as const,
sessionId: row.id,
brandId: row.brandId,
adminUserId: row.adminUserId,
};
});
}
/**
* "Return null on miss" variant of `requireWaterAdminSession`. Used by
* pages that want to render conditionally based on whether the user
* has a valid admin-PIN session, without forcing a hard error.
*/
export async function getWaterAdminSession(): Promise<AdminPinSessionInfo | null> {
const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_admin_session")?.value;
if (!sessionId) return null;
return withPlatformAdmin(async (db) => {
const rows = await db
.select({
id: waterAdminSessions.id,
brandId: waterAdminSessions.brandId,
adminUserId: waterAdminSessions.adminUserId,
expiresAt: waterAdminSessions.expiresAt,
})
.from(waterAdminSessions)
.where(eq(waterAdminSessions.id, sessionId))
.limit(1);
const row = rows[0];
if (!row) return null;
if (row.expiresAt.getTime() <= Date.now()) return null;
return {
sessionId: row.id,
brandId: row.brandId,
adminUserId: row.adminUserId,
role: "water_admin" as const,
};
});
}
/**
* Read the current field session user (or null). Cheap, idempotent.
* Used by the admin settings page to render the "logged in as" badge
* without forcing a hard error when the session is gone.
*/
export async function getFieldSessionUser(): Promise<{
userId: string;
name: string;
brandId: string;
role: "irrigator" | "water_admin";
} | null> {
const s = await requireFieldSession();
if (!s.ok) return null;
return withPlatformAdmin(async (db) => {
const rows = await db
.select({
name: waterIrrigators.name,
role: waterIrrigators.role,
})
.from(waterIrrigators)
.where(eq(waterIrrigators.id, s.userId))
.limit(1);
const row = rows[0];
if (!row) return null;
return {
userId: s.userId,
name: row.name,
brandId: s.brandId,
role: (row.role as "irrigator" | "water_admin") ?? "irrigator",
};
});
}
/**
* Site-admin gate — /admin/water-log/*.
*
* This is the only helper in this module that calls
* `getAdminUser()`, and that's intentional: it backs the
* `/admin/water-log/*` pages which sit behind Neon Auth in the
* middleware. The two PIN-only helpers above deliberately do not.
*/
export async function requireWaterAdminPermission(): Promise<SiteAdminPermission> {
const adminUser = await getAdminUser();
if (!adminUser) return { ok: false, error: "Not signed in" };
if (
!adminUser.can_manage_water_log &&
adminUser.role !== "platform_admin"
) {
return { ok: false, error: "Not authorized" };
}
// `adminUser` is non-null here (the first guard returned). Cast keeps
// the success-branch type narrow (non-nullable) so callers can read
// `auth.adminUser.foo` without an extra null check.
return { ok: true, adminUser: adminUser as NonNullable<typeof adminUser> };
}
+11 -128
View File
@@ -30,7 +30,12 @@ import {
} from "@/db/schema/water-log";
import { verifyPin, validatePin } from "@/lib/water-log-pin";
import { logAlert } from "@/lib/water-log-audit";
import { getSession } from "@/lib/auth";
import {
requireFieldSession,
type FieldSession,
} from "@/actions/water-log/auth";
export type { FieldSession };
// Field sessions last 8h — a working day in the field. Long enough
// that a single sign-in covers a morning shift + afternoon shift.
@@ -70,8 +75,7 @@ export async function getWaterHeadgates(
_brandId: string,
activeOnly: boolean = false,
): Promise<FieldHeadgate[]> {
await getSession(); return withBrand(TUXEDO_BRAND_ID, async (db) => {
return withBrand(TUXEDO_BRAND_ID, async (db) => {
const where = activeOnly
? and(
eq(waterHeadgates.brandId, TUXEDO_BRAND_ID),
@@ -101,8 +105,7 @@ export async function verifyWaterPin(
_brandId: string,
pin: string,
): Promise<VerifyPinResult> {
await getSession(); // Validate format first (cheap, no DB).
// Validate format first (cheap, no DB).
const formatError = validatePin(pin);
if (formatError) return { success: false, error: formatError };
@@ -185,8 +188,7 @@ export async function submitWaterEntry(
longitude?: number,
_headgateLocked?: boolean,
): Promise<SubmitEntryResult> {
await getSession(); // ── Auth ──
// ── Auth ──
const session = await requireFieldSession();
if (!session.ok) return { success: false, error: session.error };
@@ -281,78 +283,10 @@ await getSession(); // ── Auth ──
});
}
// ── Session helpers (shared with admin.ts via requireFieldSession) ────────
type FieldSession =
| { ok: true; userId: string; brandId: string; role: "irrigator" | "water_admin" }
| { ok: false; error: string };
async function requireFieldSession(): Promise<FieldSession> {
await getSession(); const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_session")?.value;
if (!sessionId) return { ok: false, error: "Not logged in" };
return withPlatformAdmin(async (db) => {
const rows = await db
.select({
session: waterSessions,
irrigator: waterIrrigators,
})
.from(waterSessions)
.innerJoin(waterIrrigators, eq(waterIrrigators.id, waterSessions.irrigatorId))
.where(eq(waterSessions.id, sessionId))
.limit(1);
const row = rows[0];
if (!row) return { ok: false as const, error: "Session not found" };
if (row.session.expiresAt.getTime() < Date.now()) {
// Best-effort cleanup
await db.delete(waterSessions).where(eq(waterSessions.id, sessionId));
return { ok: false as const, error: "Session expired" };
}
if (!row.irrigator.active) {
return { ok: false as const, error: "User is inactive" };
}
return {
ok: true as const,
userId: row.irrigator.id,
brandId: row.irrigator.brandId,
role: (row.irrigator.role as "irrigator" | "water_admin") ?? "irrigator",
};
});
}
async function getFieldSessionUser(): Promise<{
userId: string;
name: string;
brandId: string;
role: "irrigator" | "water_admin";
} | null> {
await getSession(); const s = await requireFieldSession();
if (!s.ok) return null;
return withPlatformAdmin(async (db) => {
const rows = await db
.select({ name: waterIrrigators.name, role: waterIrrigators.role })
.from(waterIrrigators)
.where(eq(waterIrrigators.id, s.userId))
.limit(1);
const row = rows[0];
if (!row) return null;
return {
userId: s.userId,
name: row.name,
brandId: s.brandId,
role: (row.role as "irrigator" | "water_admin") ?? "irrigator",
};
});
}
// ── Cookie/language helpers ───────────────────────────────────────────────
export async function logoutWater(): Promise<void> {
await getSession(); const cookieStore = await cookies();
const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_session")?.value;
if (sessionId) {
// Best-effort DB cleanup
@@ -368,8 +302,7 @@ await getSession(); const cookieStore = await cookies();
}
export async function logoutWaterAdmin(): Promise<void> {
await getSession(); const cookieStore = await cookies();
const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_admin_session")?.value;
if (sessionId) {
try {
@@ -385,58 +318,8 @@ await getSession(); const cookieStore = await cookies();
cookieStore.delete("wl_admin_session");
}
async function getWaterSession(): Promise<string | null> {
await getSession(); const cookieStore = await cookies();
return cookieStore.get("wl_session")?.value ?? null;
}
/**
* Resolves the current `/water/admin` session.
*
* Returns the admin role + brandId on success, or `null` if the cookie
* is missing / expired / points at a deleted session row. Used by
* admin pages (entries/[id], headgates/[id], users/[id]) as a *second*
* gate on top of `getAdminUser()` — a site admin can edit entries
* directly, but a PIN-authenticated water admin can too.
*/
export async function getWaterAdminSession(): Promise<{
sessionId: string;
brandId: string;
adminUserId: string;
role: "water_admin";
} | null> {
await getSession(); const cookieStore = await cookies();
const sessionId = cookieStore.get("wl_admin_session")?.value;
if (!sessionId) return null;
return withPlatformAdmin(async (db) => {
const rows = await db
.select({
id: waterAdminSessions.id,
brandId: waterAdminSessions.brandId,
adminUserId: waterAdminSessions.adminUserId,
expiresAt: waterAdminSessions.expiresAt,
})
.from(waterAdminSessions)
.where(eq(waterAdminSessions.id, sessionId))
.limit(1);
const row = rows[0];
if (!row) return null;
if (row.expiresAt.getTime() <= Date.now()) return null;
return {
sessionId: row.id,
brandId: row.brandId,
adminUserId: row.adminUserId,
role: "water_admin" as const,
};
});
}
export async function setWaterLang(lang: string): Promise<void> {
if (!["en", "es"].includes(lang)) return;
await getSession();
const cookieStore = await cookies();
cookieStore.set("wl_lang", lang, {
httpOnly: false,
+18 -6
View File
@@ -24,7 +24,10 @@ import {
import { getAdminUser } from "@/lib/admin-permissions";
import { hashPin, verifyPin, validatePin, generatePin } from "@/lib/water-log-pin";
import { logAuditEvent } from "@/lib/water-log-audit";
import { getSession } from "@/lib/auth";
// Note: we deliberately do NOT import `getSession` from `@/lib/auth`.
// The water-admin PIN entry flow (`verifyWaterAdminPin`) is PIN-based
// and self-contained — see docs/superpowers/specs/2026-07-01-water-log-no-platform-login-design.md.
export type AdminSettings = {
enabled: boolean;
@@ -56,7 +59,7 @@ export async function getWaterAdminSettings(
brandId: string,
): Promise<AdminSettings | null> {
await getSession(); const adminUser = await getAdminUser();
const adminUser = await getAdminUser();
if (!adminUser) return null;
if (
!adminUser.can_manage_water_log &&
@@ -88,7 +91,7 @@ export async function saveWaterAdminSettings(
settings: Partial<AdminSettings>,
): Promise<{ success: boolean; settings?: AdminSettings; error?: string }> {
await getSession(); const adminUser = await getAdminUser();
const adminUser = await getAdminUser();
if (!adminUser) return { success: false, error: "Not authenticated" };
if (
!adminUser.can_manage_water_log &&
@@ -152,7 +155,7 @@ export async function regenerateAdminPin(
brandId: string,
): Promise<{ success: boolean; pin?: string; error?: string }> {
await getSession(); const adminUser = await getAdminUser();
const adminUser = await getAdminUser();
if (!adminUser) return { success: false, error: "Not authenticated" };
if (
!adminUser.can_manage_water_log &&
@@ -199,7 +202,7 @@ export async function verifyWaterAdminPin(
pin: string,
): Promise<{ success: boolean; session_id?: string; error?: string }> {
await getSession(); const formatError = validatePin(pin);
const formatError = validatePin(pin);
if (formatError) return { success: false, error: formatError };
return withBrand(brandId, async (db) => {
@@ -224,7 +227,16 @@ await getSession(); const formatError = validatePin(pin);
}
// Tie the session to the calling site admin (best-effort).
const adminUser = await getAdminUser();
// The `wl_admin_session` is valid whether or not a platform admin
// is signed in — this call only attaches an `adminUserId` for
// audit. If `getAdminUser()` fails or returns null, we fall back
// to a zero-UUID placeholder.
let adminUser: Awaited<ReturnType<typeof getAdminUser>> = null;
try {
adminUser = await getAdminUser();
} catch {
// ignore — the session is valid regardless
}
const expiresAt = new Date(
Date.now() + settings.sessionDurationHours * 60 * 60 * 1000,
@@ -1,5 +1,5 @@
import { getAdminUser } from "@/lib/admin-permissions";
import { getWaterAdminSession } from "@/actions/water-log/field";
import { getWaterAdminSession } from "@/actions/water-log/auth";
import { getWaterEntryById } from "@/actions/water-log/admin";
import WaterLogEntryEditForm from "@/components/admin/WaterLogEntryEditForm";
import Link from "next/link";
@@ -1,6 +1,6 @@
import { getAdminUser } from "@/lib/admin-permissions";
import { redirect } from "next/navigation";
import { getWaterAdminSession } from "@/actions/water-log/field";
import { getWaterAdminSession } from "@/actions/water-log/auth";
import { getWaterHeadgatesAdmin } from "@/actions/water-log/admin";
import HeadgateEditForm from "@/components/admin/HeadgateEditForm";
import Link from "next/link";
+1 -1
View File
@@ -1,6 +1,6 @@
import { getAdminUser } from "@/lib/admin-permissions";
import { redirect } from "next/navigation";
import { getWaterAdminSession } from "@/actions/water-log/field";
import { getWaterAdminSession } from "@/actions/water-log/auth";
import { getWaterIrrigators } from "@/actions/water-log/admin";
import WaterUserEditForm from "@/components/admin/WaterUserEditForm";
import Link from "next/link";
+403
View File
@@ -0,0 +1,403 @@
/**
* Unit tests for `src/actions/water-log/auth.ts` — the centralized auth
* helpers for the water log module.
*
* These helpers are the gate for every field/admin-PIN server action
* (`/water`, `/water/admin/*`). Critically, **they must not call
* `getSession()` from `@/lib/auth`** (the Neon Auth wrapper) — that
* was the source of the bug where PIN submission hung for users with
* no platform login.
*
* If a future refactor accidentally re-introduces a `getSession()` call
* in this module, the `does not import getSession from @/lib/auth` test
* below will fail.
*/
import { describe, it, expect, vi, beforeEach } from "vitest";
// ── Hoisted state (visible inside vi.mock factory closures) ────────────
const state = vi.hoisted(() => ({
cookieValue: undefined as string | undefined,
mockDb: { select: vi.fn() } as { select: ReturnType<typeof vi.fn> },
mockWithPlatformAdmin: null as unknown as ReturnType<typeof vi.fn>,
mockGetAdminUser: null as unknown as ReturnType<typeof vi.fn>,
}));
state.mockWithPlatformAdmin = vi.fn(
async (fn: (db: typeof state.mockDb) => Promise<unknown>) => fn(state.mockDb),
);
state.mockGetAdminUser = vi.fn();
vi.mock("server-only", () => ({}));
vi.mock("@/db/client", () => ({
withPlatformAdmin: state.mockWithPlatformAdmin,
withBrand: vi.fn(),
withDb: vi.fn(),
}));
vi.mock("@/lib/admin-permissions", () => ({
getAdminUser: state.mockGetAdminUser,
}));
vi.mock("next/headers", () => ({
cookies: () =>
Promise.resolve({
get: () =>
state.cookieValue !== undefined
? { value: state.cookieValue }
: undefined,
}),
headers: () => Promise.resolve(new Headers()),
}));
beforeEach(() => {
state.cookieValue = undefined;
state.mockDb.select.mockReset();
state.mockWithPlatformAdmin.mockClear();
state.mockGetAdminUser.mockReset();
});
// ── Module guards (the regression this whole PR exists for) ─────────────
describe("water-log/auth — module surface", () => {
it("does not import getSession from @/lib/auth", async () => {
// Strip comments (block + line) and collapse whitespace before checking
// — the JSDoc deliberately references these symbols as documentation
// of what the module does NOT do.
const fs = await import("node:fs/promises");
const path = await import("node:path");
const raw = await fs.readFile(
path.resolve(process.cwd(), "src/actions/water-log/auth.ts"),
"utf8",
);
const stripped = raw
.replace(/\/\*[\s\S]*?\*\//g, "")
.replace(/^\s*\/\/.*$/gm, "")
.replace(/\s+/g, " ");
expect(stripped).not.toMatch(/from\s+["']@\/lib\/auth["']/);
expect(stripped).not.toMatch(/\bgetSession\s*\(/);
});
});
// ── requireFieldSession ─────────────────────────────────────────────────
describe("requireFieldSession()", () => {
it("returns { ok: false, error: 'Not logged in' } when wl_session cookie is missing", async () => {
state.cookieValue = undefined;
const { requireFieldSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireFieldSession();
expect(result).toEqual({ ok: false, error: "Not logged in" });
expect(state.mockWithPlatformAdmin).not.toHaveBeenCalled();
});
it("returns { ok: false, error: 'Session not found' } when cookie has no matching row", async () => {
state.cookieValue = "missing-session-id";
state.mockDb.select.mockReturnValueOnce({
from: () => ({
innerJoin: () => ({
where: () => ({
limit: async () => [],
}),
}),
}),
});
const { requireFieldSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireFieldSession();
expect(result).toEqual({ ok: false, error: "Session not found" });
});
it("returns { ok: false, error: 'Session expired' } and best-effort deletes the row when past expires_at", async () => {
state.cookieValue = "expired-id";
const past = new Date(Date.now() - 60_000);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
innerJoin: () => ({
where: () => ({
limit: async () => [
{
session: { id: "expired-id", expiresAt: past },
irrigator: {
id: "u1",
brandId: "b1",
role: "irrigator",
active: true,
},
},
],
}),
}),
}),
});
const mockDelete = vi.fn().mockReturnValue({
where: async () => undefined,
});
(state.mockDb as unknown as { select: typeof vi.fn; delete: typeof vi.fn }).delete =
mockDelete;
const { requireFieldSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireFieldSession();
expect(result).toEqual({ ok: false, error: "Session expired" });
expect(mockDelete).toHaveBeenCalled();
});
it("returns { ok: false, error: 'User is inactive' } when irrigator is soft-deleted", async () => {
state.cookieValue = "valid-id";
const future = new Date(Date.now() + 60_000);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
innerJoin: () => ({
where: () => ({
limit: async () => [
{
session: { id: "valid-id", expiresAt: future },
irrigator: {
id: "u1",
brandId: "b1",
role: "irrigator",
active: false,
},
},
],
}),
}),
}),
});
const { requireFieldSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireFieldSession();
expect(result).toEqual({ ok: false, error: "User is inactive" });
});
it("returns { ok: true, userId, brandId, role } on a happy-path session", async () => {
state.cookieValue = "valid-id";
const future = new Date(Date.now() + 60_000);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
innerJoin: () => ({
where: () => ({
limit: async () => [
{
session: { id: "valid-id", expiresAt: future },
irrigator: {
id: "u1",
brandId: "b1",
role: "irrigator",
active: true,
},
},
],
}),
}),
}),
});
const { requireFieldSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireFieldSession();
expect(result).toEqual({
ok: true,
userId: "u1",
brandId: "b1",
role: "irrigator",
});
});
});
// ── requireWaterAdminSession ────────────────────────────────────────────
describe("requireWaterAdminSession()", () => {
it("returns { ok: false, error: 'Not signed in' } when wl_admin_session cookie is missing", async () => {
state.cookieValue = undefined;
const { requireWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminSession();
expect(result).toEqual({ ok: false, error: "Not signed in" });
});
it("returns { ok: false, error: 'Session expired' } when row is past expires_at", async () => {
state.cookieValue = "expired-admin";
const past = new Date(Date.now() - 60_000);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
where: () => ({
limit: async () => [
{
id: "expired-admin",
brandId: "b1",
adminUserId: "au1",
expiresAt: past,
},
],
}),
}),
});
const { requireWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminSession();
expect(result).toEqual({ ok: false, error: "Session expired" });
});
it("returns { ok: true, sessionId, brandId, adminUserId } on a happy-path admin session", async () => {
state.cookieValue = "valid-admin";
const future = new Date(Date.now() + 60_000);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
where: () => ({
limit: async () => [
{
id: "valid-admin",
brandId: "b1",
adminUserId: "au1",
expiresAt: future,
},
],
}),
}),
});
const { requireWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminSession();
expect(result).toEqual({
ok: true,
sessionId: "valid-admin",
brandId: "b1",
adminUserId: "au1",
});
});
});
// ── getWaterAdminSession (the "return null on miss" variant) ────────────
describe("getWaterAdminSession()", () => {
it("returns null when cookie is missing", async () => {
state.cookieValue = undefined;
const { getWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
expect(await getWaterAdminSession()).toBeNull();
});
it("returns null when session row is missing", async () => {
state.cookieValue = "missing";
state.mockDb.select.mockReturnValueOnce({
from: () => ({
where: () => ({ limit: async () => [] }),
}),
});
const { getWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
expect(await getWaterAdminSession()).toBeNull();
});
it("returns null when past expires_at", async () => {
state.cookieValue = "expired";
const past = new Date(Date.now() - 1);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
where: () => ({
limit: async () => [
{
id: "expired",
brandId: "b1",
adminUserId: "au1",
expiresAt: past,
},
],
}),
}),
});
const { getWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
expect(await getWaterAdminSession()).toBeNull();
});
it("returns the session shape on a valid row", async () => {
state.cookieValue = "valid";
const future = new Date(Date.now() + 60_000);
state.mockDb.select.mockReturnValueOnce({
from: () => ({
where: () => ({
limit: async () => [
{
id: "valid",
brandId: "b1",
adminUserId: "au1",
expiresAt: future,
},
],
}),
}),
});
const { getWaterAdminSession } = await import(
"@/actions/water-log/auth"
);
expect(await getWaterAdminSession()).toEqual({
sessionId: "valid",
brandId: "b1",
adminUserId: "au1",
role: "water_admin",
});
});
});
// ── requireWaterAdminPermission (site-admin gate) ───────────────────────
describe("requireWaterAdminPermission()", () => {
it("returns { ok: false, error: 'Not signed in' } when getAdminUser is null", async () => {
state.mockGetAdminUser.mockResolvedValue(null);
const { requireWaterAdminPermission } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminPermission();
expect(result).toEqual({ ok: false, error: "Not signed in" });
});
it("returns { ok: false, error: 'Not authorized' } when admin lacks can_manage_water_log and is not platform_admin", async () => {
state.mockGetAdminUser.mockResolvedValue({
user_id: "u1",
role: "brand_admin",
can_manage_water_log: false,
});
const { requireWaterAdminPermission } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminPermission();
expect(result).toEqual({ ok: false, error: "Not authorized" });
});
it("returns { ok: true, adminUser } when platform_admin", async () => {
const adminUser = { user_id: "u1", role: "platform_admin" };
state.mockGetAdminUser.mockResolvedValue(adminUser);
const { requireWaterAdminPermission } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminPermission();
expect(result).toEqual({ ok: true, adminUser });
});
it("returns { ok: true, adminUser } when brand_admin with can_manage_water_log", async () => {
const adminUser = {
user_id: "u1",
role: "brand_admin",
can_manage_water_log: true,
};
state.mockGetAdminUser.mockResolvedValue(adminUser);
const { requireWaterAdminPermission } = await import(
"@/actions/water-log/auth"
);
const result = await requireWaterAdminPermission();
expect(result).toEqual({ ok: true, adminUser });
});
});
+72
View File
@@ -66,6 +66,78 @@ test.describe("Water Log — public surfaces", () => {
});
});
test.describe("Water Log — no platform login required (regression)", () => {
// The water log module is PIN-based and self-contained. A prior
// version of the server actions called getSession() (Neon Auth) on
// every PIN-screen interaction, which caused `/water` to hang or
// fail for users with no platform login. These tests assert that
// a fresh, cookie-free browser context can complete the public
// surfaces without ever touching /login.
test("/water full flow does not redirect to /login with no cookies", async ({ browser }) => {
// Fresh context: no storageState, no cookies, no dev_session.
const ctx = await browser.newContext();
const page = await ctx.newPage();
const hits: string[] = [];
page.on("request", (r) => {
const u = r.url();
if (u.startsWith(BASE)) hits.push(u);
});
const res = await page.goto(`${BASE}/water`, { waitUntil: "domcontentloaded" });
expect(res?.status()).toBe(200);
// Walk through language → role. The exact button labels depend on
// the i18n, but the Irrigator path is the green button labeled
// "Irrigator" and the language step is the first English/Español
// choice.
await page.getByRole("button", { name: /^english$/i }).click({ timeout: 5_000 }).catch(async () => {
// Some renderings show "English" as a label rather than exact
// text — fall back to the first button in the language picker.
await page.locator("button").filter({ hasText: /english/i }).first().click();
});
await page.getByRole("button", { name: /^irrigator$/i }).click({ timeout: 5_000 });
// The PIN input is now visible.
const pin = page.locator('input[type="password"][inputmode="numeric"]').first();
await expect(pin).toBeVisible({ timeout: 10_000 });
// We must never have hit /login.
const loginHits = hits.filter((u) => u.includes("/login"));
expect(loginHits).toEqual([]);
await ctx.close();
});
test("GET /api/water-admin-auth works with no cookies (returns 401, not 302/500)", async () => {
const ctx = await request.newContext({
baseURL: BASE,
storageState: { cookies: [], origins: [] },
});
// Any 4-digit PIN, no cookies — must NOT 302 to /login and must
// NOT 500 (server error). 401/403 are the correct gates.
const res = await ctx.post("/api/water-admin-auth", {
data: { brandId: "64294306-5f42-463d-a5e8-2ad6c81a96de", pin: "0000" },
maxRedirects: 0,
});
expect([200, 401, 403]).toContain(res.status());
await ctx.dispose();
});
test("/water responds 200 to a no-cookie GET (no Neon Auth hang)", async () => {
const ctx = await request.newContext({
baseURL: BASE,
storageState: { cookies: [], origins: [] },
});
// Bounded timeout — if the server-action pre-call is hanging on
// Neon Auth, this will throw. 8s is generous for a static page
// render and tight enough to fail fast on a hang.
const res = await ctx.get("/water", { timeout: 8_000 });
expect(res.status()).toBe(200);
await ctx.dispose();
});
});
test.describe("Water Log — site-admin gate", () => {
test("/admin/water-log redirects unauthenticated users", async ({ page }) => {
await page.goto(`${BASE}/admin/water-log`, { waitUntil: "domcontentloaded" });
+2
View File
@@ -7,6 +7,8 @@ export default defineConfig({
resolve: {
alias: [
{ find: /^@\/(?!db)/, replacement: path.resolve(__dirname, "src") + "/" },
// Specific sub-paths must come before the general @/db catch-all.
{ find: "@/db/schema/water-log", replacement: path.resolve(__dirname, "db/schema/water-log.ts") },
{ find: "@/db/client", replacement: path.resolve(__dirname, "db/client.ts") },
{ find: "@/db/schema", replacement: path.resolve(__dirname, "db/schema/index.ts") },
{ find: "@/db", replacement: path.resolve(__dirname, "db") },